Files
dify-plugin-sdks/python/examples/code_based_workflow/tools/upload_file.py
T
-LAN- b3e0612e53 chore: Apply format
Signed-off-by: -LAN- <laipz8200@outlook.com>
2024-12-02 21:18:36 +08:00

11 lines
382 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}")