fixes

main.py

@@ -216,13 +216,6 @@ def register_toolkit_for_master_organisation():
             Organisation.id == marketplace_organisation_id).first()
         if marketplace_organisation is not None:
             register_marketplace_toolkits(session, marketplace_organisation)
-
-    def local_llm_model_config():
-        existing_models_config = session.query(ModelsConfig).filter(ModelsConfig.org_id == default_user.organisation_id, ModelsConfig.provider == 'Local LLM').first()
-        if existing_models_config is None:
-            models_config = ModelsConfig(org_id=default_user.organisation_id, provider='Local LLM', api_key="EMPTY")
-            session.add(models_config)
-            session.commit()
 
     IterationWorkflowSeed.build_single_step_agent(session)
     IterationWorkflowSeed.build_task_based_agents(session)
@@ -246,8 +239,7 @@ def local_llm_model_config():
     # AgentWorkflowSeed.doc_search_and_code(session)
     # AgentWorkflowSeed.build_research_email_workflow(session)
     replace_old_iteration_workflows(session)
-    local_llm_model_config()
-
+
     if env != "PROD":
         register_toolkit_for_all_organisation()
     else:

superagi/controllers/user.py

@@ -14,6 +14,8 @@
 from superagi.helper.auth import check_auth, get_current_user
 from superagi.lib.logger import logger
 
+from superagi.models.models_config import ModelsConfig
+
 # from superagi.types.db import UserBase, UserIn, UserOut
 
 router = APIRouter()
@@ -73,6 +75,10 @@
     organisation = Organisation.find_or_create_organisation(db.session, db_user)
     Project.find_or_create_default_project(db.session, organisation.id)
     logger.info("User created", db_user)
+
+    #adding local llm configuration
+    ModelsConfig.add_llm_config(db.session, organisation.id)
+
     return db_user
 
 

superagi/models/models_config.py

@@ -145,4 +145,12 @@
         if model is None:
             return {"error": "Model not found"}
         else:
-            return {"provider": model.provider}
+            return {"provider": model.provider}
+
+    @classmethod
+    def add_llm_config(cls, session, organisation_id):
+        existing_models_config = session.query(ModelsConfig).filter(ModelsConfig.org_id == organisation_id, ModelsConfig.provider == 'Local LLM').first()
+        if existing_models_config is None:
+            models_config = ModelsConfig(org_id=organisation_id, provider='Local LLM', api_key="EMPTY")
+            session.add(models_config)
+            session.commit()