feat: Tortoise ORM migrations

backend/src/app.py

@@ -22,7 +22,7 @@ from chat import ChatView, Message, MESSAGE_PAGE_SIZE
 from app_mixins import ChatMixin
 
 class App(ChatMixin):
-    def __init__(self, data_directory, replicate_api_token=None):
+    def __init__(self, data_directory, replicate_token=None, openrouter_token=None):
         self.data_directory = data_directory
         self.user_connections = {}
         self.remote_connections = {}
@@ -34,16 +34,9 @@ class App(ChatMixin):
         # Clients
         self.qdrant_client = QdrantClient(path=f"{self.data_directory}/qdrant_data")
 
-        if replicate_api_token:
-            self.replicate_client = replicate.Client(api_token=replicate_api_token)
-        else:
-            self.replicate_client = None
-
-        openrouter_api_key = os.getenv("OPENROUTER_API_KEY")
-        if openrouter_api_key:
-            self.openrouter_client = OpenRouter(api_key=openrouter_api_key)
-        else:
-            self.openrouter_client = None
+        self.replicate_client = replicate_token and replicate.Client(api_token=replicate_token) or None
+        self.openrouter_client = openrouter_token and OpenRouter(api_key=openrouter_token) or None
+        
 
         self.agent_classes = [
             agents_mod.ExpertAgent,
@@ -350,10 +343,10 @@ class App(ChatMixin):
 # --- Main Entry Point ---
 
 async def main():
-    if len(sys.argv) > 1:
-        data_directory = sys.argv[1]
-    else:
-        data_directory = "."
+    # Get data directory from environment variable
+    data_directory = os.getenv("HORDE_DATA_DIR")
+    if not data_directory:
+        raise ValueError("HORDE_DATA_DIR environment variable must be set")
 
     # 1. Initialize Database First
     await db.init_database(data_directory)
@@ -362,7 +355,8 @@ async def main():
     # 2. Initialize App
     horde_app = App(
         data_directory=data_directory,
-        replicate_api_token=os.getenv('REPLICATE_API_TOKEN'),
+        replicate_token=os.getenv('REPLICATE_API_TOKEN'),
+        openrouter_token=os.getenv("OPENROUTER_API_KEY")
     )
 
     # 3. Start Server