fix: dataset direct output (#860)

7a859098 · takatost · GitHub · e8c14bb7 · 7a859098
Unverified Commit 7a859098 authored Aug 15, 2023 by takatost Committed by GitHub Aug 15, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

orchestrator_rule_parser.py api/core/orchestrator_rule_parser.py +1 -1

No files found.
--- a/api/core/orchestrator_rule_parser.py
+++ b/api/core/orchestrator_rule_parser.py
@@ -72,7 +72,7 @@ class OrchestratorRuleParser:

            # only OpenAI chat model (include Azure) support function call, use ReACT instead
            if agent_model_instance.model_mode != ModelMode.CHAT \
-                         or agent_model_instance.name not in ['openai', 'azure_openai']:
+                         or agent_model_instance.model_provider.provider_name not in ['openai', 'azure_openai']:
                if planning_strategy in [PlanningStrategy.FUNCTION_CALL, PlanningStrategy.MULTI_FUNCTION_CALL]:
                    planning_strategy = PlanningStrategy.REACT
                elif planning_strategy == PlanningStrategy.ROUTER: