fix: empty model features (#2330)

fce64d76 · Yeuoly · GitHub · f0c9bb7c · fce64d76
Unverified Commit fce64d76 authored Feb 01, 2024 by Yeuoly Committed by GitHub Feb 01, 2024
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

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

No files found.
--- a/api/core/app_runner/assistant_app_runner.py
+++ b/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