refactor: list tools
parent
435e71eb60
commit
7a3e756020
@ -1,13 +1,65 @@
|
|||||||
|
from collections.abc import Generator
|
||||||
|
from typing import Any
|
||||||
|
|
||||||
|
from core.plugin.entities.plugin_daemon import PluginToolProviderEntity
|
||||||
from core.plugin.manager.base import BasePluginManager
|
from core.plugin.manager.base import BasePluginManager
|
||||||
from core.tools.entities.tool_entities import ToolProviderEntity
|
from core.tools.entities.tool_entities import ToolInvokeMessage
|
||||||
|
|
||||||
|
|
||||||
class PluginToolManager(BasePluginManager):
|
class PluginToolManager(BasePluginManager):
|
||||||
def fetch_tool_providers(self, tenant_id: str) -> list[ToolProviderEntity]:
|
def fetch_tool_providers(self, tenant_id: str) -> list[PluginToolProviderEntity]:
|
||||||
"""
|
"""
|
||||||
Fetch tool providers for the given asset.
|
Fetch tool providers for the given asset.
|
||||||
"""
|
"""
|
||||||
response = self._request_with_plugin_daemon_response(
|
response = self._request_with_plugin_daemon_response(
|
||||||
"GET", f"plugin/{tenant_id}/tools", list[ToolProviderEntity], params={"page": 1, "page_size": 256}
|
"GET", f"plugin/{tenant_id}/tools", list[PluginToolProviderEntity], params={"page": 1, "page_size": 256}
|
||||||
|
)
|
||||||
|
return response
|
||||||
|
|
||||||
|
def invoke(
|
||||||
|
self,
|
||||||
|
tenant_id: str,
|
||||||
|
user_id: str,
|
||||||
|
plugin_unique_identifier: str,
|
||||||
|
tool_provider: str,
|
||||||
|
tool_name: str,
|
||||||
|
credentials: dict[str, Any],
|
||||||
|
tool_parameters: dict[str, Any],
|
||||||
|
) -> Generator[ToolInvokeMessage, None, None]:
|
||||||
|
response = self._request_with_plugin_daemon_response_stream(
|
||||||
|
"POST",
|
||||||
|
f"plugin/{tenant_id}/tool/invoke",
|
||||||
|
ToolInvokeMessage,
|
||||||
|
data={
|
||||||
|
"plugin_unique_identifier": plugin_unique_identifier,
|
||||||
|
"user_id": user_id,
|
||||||
|
"data": {
|
||||||
|
"provider": tool_provider,
|
||||||
|
"tool": tool_name,
|
||||||
|
"credentials": credentials,
|
||||||
|
"tool_parameters": tool_parameters,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
)
|
||||||
|
return response
|
||||||
|
|
||||||
|
def validate_provider_credentials(
|
||||||
|
self, tenant_id: str, user_id: str, plugin_unique_identifier: str, provider: str, credentials: dict[str, Any]
|
||||||
|
) -> bool:
|
||||||
|
"""
|
||||||
|
validate the credentials of the provider
|
||||||
|
"""
|
||||||
|
response = self._request_with_plugin_daemon_response(
|
||||||
|
"POST",
|
||||||
|
f"plugin/{tenant_id}/tool/validate_credentials",
|
||||||
|
bool,
|
||||||
|
data={
|
||||||
|
"plugin_unique_identifier": plugin_unique_identifier,
|
||||||
|
"user_id": user_id,
|
||||||
|
"data": {
|
||||||
|
"provider": provider,
|
||||||
|
"credentials": credentials,
|
||||||
|
},
|
||||||
|
},
|
||||||
)
|
)
|
||||||
return response
|
return response
|
||||||
|
|||||||
@ -1,30 +0,0 @@
|
|||||||
|
|
||||||
|
|
||||||
from core.entities.provider_entities import ProviderConfig
|
|
||||||
from core.tools.__base.tool import Tool
|
|
||||||
from core.tools.__base.tool_provider import ToolProviderController
|
|
||||||
from core.tools.entities.tool_entities import ToolProviderType
|
|
||||||
|
|
||||||
|
|
||||||
class PluginToolProvider(ToolProviderController):
|
|
||||||
@property
|
|
||||||
def provider_type(self) -> ToolProviderType:
|
|
||||||
"""
|
|
||||||
returns the type of the provider
|
|
||||||
|
|
||||||
:return: type of the provider
|
|
||||||
"""
|
|
||||||
return ToolProviderType.PLUGIN
|
|
||||||
|
|
||||||
def get_tool(self, tool_name: str) -> Tool:
|
|
||||||
"""
|
|
||||||
return tool with given name
|
|
||||||
"""
|
|
||||||
return super().get_tool(tool_name)
|
|
||||||
|
|
||||||
def get_credentials_schema(self) -> dict[str, ProviderConfig]:
|
|
||||||
"""
|
|
||||||
get credentials schema
|
|
||||||
"""
|
|
||||||
return super().get_credentials_schema()
|
|
||||||
|
|
||||||
@ -0,0 +1,72 @@
|
|||||||
|
from typing import Any
|
||||||
|
|
||||||
|
from core.plugin.manager.tool import PluginToolManager
|
||||||
|
from core.tools.__base.tool_runtime import ToolRuntime
|
||||||
|
from core.tools.builtin_tool.provider import BuiltinToolProviderController
|
||||||
|
from core.tools.entities.tool_entities import ToolProviderEntityWithPlugin, ToolProviderType
|
||||||
|
from core.tools.errors import ToolProviderCredentialValidationError
|
||||||
|
from core.tools.plugin_tool.tool import PluginTool
|
||||||
|
|
||||||
|
|
||||||
|
class PluginToolProviderController(BuiltinToolProviderController):
|
||||||
|
entity: ToolProviderEntityWithPlugin
|
||||||
|
tenant_id: str
|
||||||
|
plugin_unique_identifier: str
|
||||||
|
|
||||||
|
def __init__(self, entity: ToolProviderEntityWithPlugin, tenant_id: str, plugin_unique_identifier: str) -> None:
|
||||||
|
self.entity = entity
|
||||||
|
self.tenant_id = tenant_id
|
||||||
|
self.plugin_unique_identifier = plugin_unique_identifier
|
||||||
|
|
||||||
|
@property
|
||||||
|
def provider_type(self) -> ToolProviderType:
|
||||||
|
"""
|
||||||
|
returns the type of the provider
|
||||||
|
|
||||||
|
:return: type of the provider
|
||||||
|
"""
|
||||||
|
return ToolProviderType.PLUGIN
|
||||||
|
|
||||||
|
def _validate_credentials(self, user_id: str, credentials: dict[str, Any]) -> None:
|
||||||
|
"""
|
||||||
|
validate the credentials of the provider
|
||||||
|
"""
|
||||||
|
manager = PluginToolManager()
|
||||||
|
if not manager.validate_provider_credentials(
|
||||||
|
tenant_id=self.tenant_id,
|
||||||
|
user_id=user_id,
|
||||||
|
plugin_unique_identifier=self.plugin_unique_identifier,
|
||||||
|
provider=self.entity.identity.name,
|
||||||
|
credentials=credentials,
|
||||||
|
):
|
||||||
|
raise ToolProviderCredentialValidationError("Invalid credentials")
|
||||||
|
|
||||||
|
def get_tool(self, tool_name: str) -> PluginTool:
|
||||||
|
"""
|
||||||
|
return tool with given name
|
||||||
|
"""
|
||||||
|
tool_entity = next(tool_entity for tool_entity in self.entity.tools if tool_entity.identity.name == tool_name)
|
||||||
|
|
||||||
|
if not tool_entity:
|
||||||
|
raise ValueError(f"Tool with name {tool_name} not found")
|
||||||
|
|
||||||
|
return PluginTool(
|
||||||
|
entity=tool_entity,
|
||||||
|
runtime=ToolRuntime(tenant_id=self.tenant_id),
|
||||||
|
tenant_id=self.tenant_id,
|
||||||
|
plugin_unique_identifier=self.plugin_unique_identifier,
|
||||||
|
)
|
||||||
|
|
||||||
|
def get_tools(self) -> list[PluginTool]:
|
||||||
|
"""
|
||||||
|
get all tools
|
||||||
|
"""
|
||||||
|
return [
|
||||||
|
PluginTool(
|
||||||
|
entity=tool_entity,
|
||||||
|
runtime=ToolRuntime(tenant_id=self.tenant_id),
|
||||||
|
tenant_id=self.tenant_id,
|
||||||
|
plugin_unique_identifier=self.plugin_unique_identifier,
|
||||||
|
)
|
||||||
|
for tool_entity in self.entity.tools
|
||||||
|
]
|
||||||
@ -0,0 +1,41 @@
|
|||||||
|
from collections.abc import Generator
|
||||||
|
from typing import Any
|
||||||
|
|
||||||
|
from core.plugin.manager.tool import PluginToolManager
|
||||||
|
from core.tools.__base.tool import Tool
|
||||||
|
from core.tools.__base.tool_runtime import ToolRuntime
|
||||||
|
from core.tools.entities.tool_entities import ToolEntity, ToolInvokeMessage, ToolProviderType
|
||||||
|
|
||||||
|
|
||||||
|
class PluginTool(Tool):
|
||||||
|
tenant_id: str
|
||||||
|
plugin_unique_identifier: str
|
||||||
|
|
||||||
|
def __init__(self, entity: ToolEntity, runtime: ToolRuntime, tenant_id: str, plugin_unique_identifier: str) -> None:
|
||||||
|
super().__init__(entity, runtime)
|
||||||
|
self.tenant_id = tenant_id
|
||||||
|
self.plugin_unique_identifier = plugin_unique_identifier
|
||||||
|
|
||||||
|
@property
|
||||||
|
def tool_provider_type(self) -> ToolProviderType:
|
||||||
|
return ToolProviderType.PLUGIN
|
||||||
|
|
||||||
|
def _invoke(self, user_id: str, tool_parameters: dict[str, Any]) -> Generator[ToolInvokeMessage, None, None]:
|
||||||
|
manager = PluginToolManager()
|
||||||
|
return manager.invoke(
|
||||||
|
tenant_id=self.tenant_id,
|
||||||
|
user_id=user_id,
|
||||||
|
plugin_unique_identifier=self.plugin_unique_identifier,
|
||||||
|
tool_provider=self.entity.identity.provider,
|
||||||
|
tool_name=self.entity.identity.name,
|
||||||
|
credentials=self.runtime.credentials,
|
||||||
|
tool_parameters=tool_parameters,
|
||||||
|
)
|
||||||
|
|
||||||
|
def fork_tool_runtime(self, runtime: ToolRuntime) -> "PluginTool":
|
||||||
|
return PluginTool(
|
||||||
|
entity=self.entity,
|
||||||
|
runtime=runtime,
|
||||||
|
tenant_id=self.tenant_id,
|
||||||
|
plugin_unique_identifier=self.plugin_unique_identifier,
|
||||||
|
)
|
||||||
Loading…
Reference in New Issue