소스 검색

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

sino 10 달 전
부모
커밋
d7213b12cc
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      api/core/workflow/nodes/parameter_extractor/parameter_extractor_node.py

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