소스 검색

fix: create basic app causing internal error when default model is not exist

takatost 6 달 전
부모
커밋
f0637ba332
2개의 변경된 파일5개의 추가작업 그리고 2개의 파일을 삭제
  1. 1 1
      api/core/model_manager.py
  2. 4 1
      api/core/provider_manager.py

+ 1 - 1
api/core/model_manager.py

@@ -409,7 +409,7 @@ class ModelManager:
 
         return ModelInstance(provider_model_bundle, model)
 
-    def get_default_provider_model_name(self, tenant_id: str, model_type: ModelType) -> tuple[str, str]:
+    def get_default_provider_model_name(self, tenant_id: str, model_type: ModelType) -> tuple[str | None, str | None]:
         """
         Return first provider and the first model in the provider
         :param tenant_id: tenant id

+ 4 - 1
api/core/provider_manager.py

@@ -270,7 +270,7 @@ class ProviderManager:
             ),
         )
 
-    def get_first_provider_first_model(self, tenant_id: str, model_type: ModelType) -> tuple[str, str]:
+    def get_first_provider_first_model(self, tenant_id: str, model_type: ModelType) -> tuple[str | None, str | None]:
         """
         Get names of first model and its provider
 
@@ -283,6 +283,9 @@ class ProviderManager:
         # get available models from provider_configurations
         all_models = provider_configurations.get_models(model_type=model_type, only_active=False)
 
+        if not all_models:
+            return None, None
+
         return all_models[0].provider.provider, all_models[0].model
 
     def update_default_model_record(