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

refactor: tool message transformer

Browse Source
This commit is contained in:
Yeuoly
2024-08-29 13:42:31 +08:00
parent 4b4741f7ed
commit c28998a6f0
5 changed files with 52 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
api/core/tools/entities/tool_entities.py
View File

@ -96,6 +96,9 @@ class ToolInvokeMessage(BaseModel):
class JsonMessage(BaseModel):
json_object: dict
class BlobMessage(BaseModel):
blob: bytes
class MessageType(Enum):
TEXT = "text"
IMAGE = "image"
@ -109,7 +112,7 @@ class ToolInvokeMessage(BaseModel):
"""
plain text, image url or link url
"""
message: JsonMessage | TextMessage | None
message: JsonMessage | TextMessage | BlobMessage | None
meta: dict[str, Any] | None = None
save_as: str = ''
@ -321,7 +324,7 @@ class ToolRuntimeVariablePool(BaseModel):
self.pool.append(variable)
def set_file(self, tool_name: str, value: str, name: str = None) -> None:
def set_file(self, tool_name: str, value: str, name: Optional[str] = None) -> None:
"""
set an image variable