seed data on env var

Josh-XT · Jan 5, 2025 · 77a1847 · 77a1847
1 parent 8eaee21
commit 77a1847
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/agixt/DB.py b/agixt/DB.py
@@ -744,11 +744,12 @@ def setup_default_roles():
     # Create any missing tables
     Base.metadata.create_all(engine)
     setup_default_roles()
+    seed_data = str(getenv("SEED_DATA")).lower() == "true"
+    if seed_data:
+        # Import seed data
+        from SeedImports import import_all_data
 
-    # Import seed data
-    from SeedImports import import_all_data
-
-    import_all_data()
+        import_all_data()
     uvicorn.run(
         "app:app",
         host="0.0.0.0",

diff --git a/agixt/Globals.py b/agixt/Globals.py
@@ -36,6 +36,7 @@ def getenv(var_name: str, default_value: str = "") -> str:
         "REGISTRATION_DISABLED": "false",
         "CREATE_AGENT_ON_REGISTER": "true",
         "CREATE_AGIXT_AGENT": "true",
+        "SEED_DATA": "true",
     }
     if default_value != "":
         default_values[var_name] = default_value