瀏覽代碼

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

tags/0.6.12
sino 1 年之前
父節點
當前提交
d7213b12cc
No account linked to committer's email address
共有 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(

Loading…
取消
儲存