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

refactor: tool models

Browse Source
This commit is contained in:
Yeuoly
2024-08-30 15:55:10 +08:00
parent 1fa3b9cfd8
commit cf4e9f317e
6 changed files with 60 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/callback_handler/agent_tool_callback_handler.py
View File

@ -1,5 +1,5 @@
import os
from collections.abc import Mapping, Sequence
from collections.abc import Iterable, Mapping
from typing import Any, Optional, TextIO, Union
from pydantic import BaseModel
@ -55,7 +55,7 @@ class DifyAgentCallbackHandler(BaseModel):
self,
tool_name: str,
tool_inputs: Mapping[str, Any],
tool_outputs: Sequence[ToolInvokeMessage],
tool_outputs: Iterable[ToolInvokeMessage] | str,
message_id: Optional[str] = None,
timer: Optional[Any] = None,
trace_manager: Optional[TraceQueueManager] = None