fixes

GildedHonour · Nov 3, 2023 · b8b8b18 · b8b8b18
1 parent 3c43030
commit b8b8b18
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/superagi/models/models_config.py b/superagi/models/models_config.py
@@ -157,7 +157,7 @@ def fetch_model_by_id_marketplace(cls, session, model_provider_id):
             return {"provider": model.provider}
 
     @classmethod
-    def local_llm_model_config(cls, session, organisation_id):
+    def add_model_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")