fix migration

Josh-XT · Jan 12, 2025 · 9f1048f · 9f1048f
1 parent fd0b6a7
commit 9f1048f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/agixt/DB.py b/agixt/DB.py
@@ -800,7 +800,10 @@ def migrate_company_agent_name():
                 logging.error(f"Error connecting to database: {e}")
                 time.sleep(5)
     # Create any missing tables
-    migrate_company_agent_name()
+    try:
+        migrate_company_agent_name()
+    except Exception as e:
+        logging.error(f"Error during migration: {e}")
     Base.metadata.create_all(engine)
     setup_default_roles()
     seed_data = str(getenv("SEED_DATA")).lower() == "true"