mirror of
https://github.com/langgenius/dify-plugin-sdks.git
synced 2026-07-25 05:25:23 -04:00
Yeuoly
44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
from abc import ABC, abstractmethod
|
|
from collections.abc import Generator
|
|
|
|
from dify_plugin.core.runtime.request import RequestInterface
|
|
from dify_plugin.tool.entities import ToolInvokeMessage, ToolRuntime
|
|
|
|
|
|
class ToolProvider(ABC):
|
|
def validate_credentials(self, credentials: dict):
|
|
return self._validate_credentials(credentials)
|
|
|
|
@abstractmethod
|
|
def _validate_credentials(self, credentials: dict):
|
|
pass
|
|
|
|
class Tool(RequestInterface, ABC):
|
|
runtime: ToolRuntime
|
|
|
|
def __init__(self, runtime: ToolRuntime):
|
|
self.runtime = runtime
|
|
|
|
@classmethod
|
|
def from_credentials(cls, credentials: dict) -> 'Tool':
|
|
return cls(ToolRuntime(credentials=credentials, user_id=''))
|
|
|
|
def create_text_message(self, text: str) -> ToolInvokeMessage:
|
|
return ToolInvokeMessage(
|
|
type=ToolInvokeMessage.MessageType.TEXT,
|
|
message=ToolInvokeMessage.TextMessage(text=text)
|
|
)
|
|
|
|
def create_json_message(self, json: dict) -> ToolInvokeMessage:
|
|
return ToolInvokeMessage(
|
|
type=ToolInvokeMessage.MessageType.JSON,
|
|
message=ToolInvokeMessage.JsonMessage(json_object=json)
|
|
)
|
|
|
|
@abstractmethod
|
|
def _invoke(self, tool_parameters: dict) -> Generator[ToolInvokeMessage, None]:
|
|
pass
|
|
|
|
def invoke(self, tool_parameters: dict) -> Generator[ToolInvokeMessage, None]:
|
|
return self._invoke(tool_parameters)
|