ソースを参照

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

sino 10 ヶ月 前
コミット
d7213b12cc

+ 1 - 1
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(