mirror of
https://github.com/langgenius/dify-plugin-sdks.git
synced 2026-07-22 18:35:29 -04:00
Yeuoly
43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
from abc import abstractmethod
|
|
from typing import Generator
|
|
from dify_plugin.core.runtime import Session
|
|
from dify_plugin.entities.agent import AgentInvokeMessage
|
|
from dify_plugin.interfaces.tool import ToolLike, ToolProvider
|
|
|
|
|
|
class AgentProvider(ToolProvider):
|
|
def validate_credentials(self):
|
|
"""
|
|
Always permit the agent to run
|
|
"""
|
|
return True
|
|
|
|
def _validate_credentials(self):
|
|
pass
|
|
|
|
|
|
class AgentStrategy(ToolLike[AgentInvokeMessage]):
|
|
def __init__(
|
|
self,
|
|
session: Session,
|
|
):
|
|
self.session = session
|
|
self.response_type = AgentInvokeMessage
|
|
|
|
############################################################
|
|
# Methods that can be implemented by plugin #
|
|
############################################################
|
|
|
|
@abstractmethod
|
|
def _invoke(self, parameters: dict) -> Generator[AgentInvokeMessage, None, None]:
|
|
pass
|
|
|
|
############################################################
|
|
# For executor use only #
|
|
############################################################
|
|
|
|
def invoke(self, parameters: dict) -> Generator[AgentInvokeMessage, None, None]:
|
|
# convert parameters into correct types
|
|
parameters = self._convert_parameters(parameters)
|
|
return self._invoke(parameters)
|