youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-03 00:48:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: tool node

Browse Source
This commit is contained in:
Yeuoly
2024-03-11 13:54:11 +08:00
committed by takatost
parent f8cba2679e
commit 5e4bd9fc38
7 changed files with 334 additions and 109 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
api/core/agent/cot_agent_runner.py
View File

@ -25,6 +25,7 @@ from core.tools.errors import (
ToolProviderCredentialValidationError,
ToolProviderNotFoundError,
)
from core.tools.utils.message_transformer import ToolFileMessageTransformer
from models.model import Conversation, Message
@ -280,7 +281,12 @@ class CotAgentRunner(BaseAgentRunner):
tool_parameters=tool_call_args
)
# transform tool response to llm friendly response
tool_response = self.transform_tool_invoke_messages(tool_response)
tool_response = ToolFileMessageTransformer.transform_tool_invoke_messages(
messages=tool_response,
user_id=self.user_id,
tenant_id=self.tenant_id,
conversation_id=self.message.conversation_id
)
# extract binary data from tool invoke message
binary_files = self.extract_tool_response_binary(tool_response)
# create message file