mirror of
https://github.com/langgenius/dify.git
synced 2026-01-14 06:07:33 +08:00
69 lines
2.7 KiB
Python
69 lines
2.7 KiB
Python
from typing import Any
|
|
|
|
from core.plugin.entities.plugin import ToolProviderID
|
|
from core.plugin.entities.plugin_daemon import PluginToolProviderEntity, PluginTriggerProviderEntity
|
|
from core.plugin.impl.base import BasePluginClient
|
|
|
|
|
|
class PluginTriggerManager(BasePluginClient):
|
|
def fetch_trigger_providers(self, tenant_id: str) -> list[PluginTriggerProviderEntity]:
|
|
"""
|
|
Fetch tool providers for the given tenant.
|
|
"""
|
|
|
|
def transformer(json_response: dict[str, Any]) -> dict:
|
|
for provider in json_response.get("data", []):
|
|
declaration = provider.get("declaration", {}) or {}
|
|
provider_name = declaration.get("identity", {}).get("name")
|
|
for tool in declaration.get("tools", []):
|
|
tool["identity"]["provider"] = provider_name
|
|
|
|
return json_response
|
|
|
|
response = self._request_with_plugin_daemon_response(
|
|
"GET",
|
|
f"plugin/{tenant_id}/management/tools",
|
|
list[PluginToolProviderEntity],
|
|
params={"page": 1, "page_size": 256},
|
|
transformer=transformer,
|
|
)
|
|
|
|
for provider in response:
|
|
provider.declaration.identity.name = f"{provider.plugin_id}/{provider.declaration.identity.name}"
|
|
|
|
# override the provider name for each tool to plugin_id/provider_name
|
|
for tool in provider.declaration.tools:
|
|
tool.identity.provider = provider.declaration.identity.name
|
|
|
|
return response
|
|
|
|
def fetch_tool_provider(self, tenant_id: str, provider: str) -> PluginToolProviderEntity:
|
|
"""
|
|
Fetch tool provider for the given tenant and plugin.
|
|
"""
|
|
tool_provider_id = ToolProviderID(provider)
|
|
|
|
def transformer(json_response: dict[str, Any]) -> dict:
|
|
data = json_response.get("data")
|
|
if data:
|
|
for tool in data.get("declaration", {}).get("tools", []):
|
|
tool["identity"]["provider"] = tool_provider_id.provider_name
|
|
|
|
return json_response
|
|
|
|
response = self._request_with_plugin_daemon_response(
|
|
"GET",
|
|
f"plugin/{tenant_id}/management/tool",
|
|
PluginToolProviderEntity,
|
|
params={"provider": tool_provider_id.provider_name, "plugin_id": tool_provider_id.plugin_id},
|
|
transformer=transformer,
|
|
)
|
|
|
|
response.declaration.identity.name = f"{response.plugin_id}/{response.declaration.identity.name}"
|
|
|
|
# override the provider name for each tool to plugin_id/provider_name
|
|
for tool in response.declaration.tools:
|
|
tool.identity.provider = response.declaration.identity.name
|
|
|
|
return response
|