AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 19:26:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
sino 2024-06-17 23:25:29 +08:00 committed by GitHub
parent c163521b9e
commit d7213b12cc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/core/workflow/nodes/parameter_extractor/parameter_extractor_node.py
View File

@ -95,7 +95,7 @@ class ParameterExtractorNode(LLMNode):
# fetch memory # fetch memory
memory = self._fetch_memory(node_data.memory, variable_pool, model_instance) 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': and node_data.reasoning_mode == 'function_call':
# use function call # use function call
prompt_messages, prompt_message_tools = self._generate_function_call_prompt( prompt_messages, prompt_message_tools = self._generate_function_call_prompt(