From 15a56f705f8a53eb25ebdc8d56c6f33357124241 Mon Sep 17 00:00:00 2001 From: Yeuoly <45712896+Yeuoly@users.noreply.github.com> Date: Tue, 18 Feb 2025 20:18:36 +0800 Subject: [PATCH] fix: get tool provider (#13958) --- api/core/plugin/entities/plugin.py | 2 +- api/core/plugin/manager/tool.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/api/core/plugin/entities/plugin.py b/api/core/plugin/entities/plugin.py index 632c9727b0..732e10c191 100644 --- a/api/core/plugin/entities/plugin.py +++ b/api/core/plugin/entities/plugin.py @@ -180,7 +180,7 @@ class ToolProviderID(GenericProviderID): def __init__(self, value: str, is_hardcoded: bool = False) -> None: super().__init__(value, is_hardcoded) if self.organization == "langgenius": - if self.provider_name in ["jina", "siliconflow"]: + if self.provider_name in ["jina", "siliconflow", "stepfun"]: self.plugin_name = f"{self.provider_name}_tool" diff --git a/api/core/plugin/manager/tool.py b/api/core/plugin/manager/tool.py index 23a390f4aa..4c3abd3acf 100644 --- a/api/core/plugin/manager/tool.py +++ b/api/core/plugin/manager/tool.py @@ -3,7 +3,7 @@ from typing import Any, Optional from pydantic import BaseModel -from core.plugin.entities.plugin import GenericProviderID +from core.plugin.entities.plugin import GenericProviderID, ToolProviderID from core.plugin.entities.plugin_daemon import PluginBasicBooleanResponse, PluginToolProviderEntity from core.plugin.manager.base import BasePluginManager from core.tools.entities.tool_entities import ToolInvokeMessage, ToolParameter @@ -45,7 +45,7 @@ class PluginToolManager(BasePluginManager): """ Fetch tool provider for the given tenant and plugin. """ - tool_provider_id = GenericProviderID(provider) + tool_provider_id = ToolProviderID(provider) def transformer(json_response: dict[str, Any]) -> dict: data = json_response.get("data")