mirror of
https://github.com/langgenius/dify-plugin-sdks.git
synced 2026-07-22 18:35:29 -04:00
5858fe715e
Signed-off-by: -LAN- <laipz8200@outlook.com>
13 lines
396 B
Python
13 lines
396 B
Python
from collections.abc import Generator
|
|
|
|
from dify_plugin import Tool
|
|
from dify_plugin.entities.tool import ToolInvokeMessage
|
|
|
|
|
|
class UploadFileTool(Tool):
|
|
def _invoke(
|
|
self, tool_parameters: dict
|
|
) -> Generator[ToolInvokeMessage, None, None]:
|
|
response = self.session.file.upload("1.txt", b"", "text/plain")
|
|
yield self.create_text_message(f"file id: {response.id}")
|