Преглед на файлове

fix: empty model features (#2330)

Yeuoly преди 1 година
родител
ревизия
fce64d760b
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      api/core/app_runner/assistant_app_runner.py

+ 1 - 1
api/core/app_runner/assistant_app_runner.py

@@ -199,7 +199,7 @@ class AssistantApplicationRunner(AppRunner):
         llm_model = cast(LargeLanguageModel, model_instance.model_type_instance)
         model_schema = llm_model.get_model_schema(model_instance.model, model_instance.credentials)
 
-        if set([ModelFeature.MULTI_TOOL_CALL, ModelFeature.TOOL_CALL]).intersection(model_schema.features):
+        if set([ModelFeature.MULTI_TOOL_CALL, ModelFeature.TOOL_CALL]).intersection(model_schema.features or []):
             agent_entity.strategy = AgentEntity.Strategy.FUNCTION_CALLING
 
         # start agent runner