fix: extract params by function calling for models supporting tool call (#5334)

2 years ago · d7213b12cc
parent c163521b9e
commit d7213b12cc
1 changed files with 1 additions and 1 deletions
--- a/api/core/workflow/nodes/parameter_extractor/parameter_extractor_node.py
+++ b/api/core/workflow/nodes/parameter_extractor/parameter_extractor_node.py
@ -95,7 +95,7 @@ class ParameterExtractorNode(LLMNode):
        # fetch memory
        memory = self._fetch_memory(node_data.memory, variable_pool, model_instance)
        
-        if set(model_schema.features or []) & set([ModelFeature.MULTI_TOOL_CALL, ModelFeature.MULTI_TOOL_CALL]) \
+        if set(model_schema.features or []) & set([ModelFeature.TOOL_CALL, ModelFeature.MULTI_TOOL_CALL]) \
            and node_data.reasoning_mode == 'function_call':
            # use function call 
            prompt_messages, prompt_message_tools = self._generate_function_call_prompt(