From 957ab093c909ec15b46c32665617b9a9660faea2 Mon Sep 17 00:00:00 2001 From: Yeuoly Date: Sun, 29 Sep 2024 17:07:40 +0800 Subject: [PATCH] enhancement: reduce requests to plugin daemon --- api/core/tools/tool_manager.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/api/core/tools/tool_manager.py b/api/core/tools/tool_manager.py index fdd7e8385b..2f747a317e 100644 --- a/api/core/tools/tool_manager.py +++ b/api/core/tools/tool_manager.py @@ -123,12 +123,13 @@ class ToolManager: :return: the tool """ if provider_type == ToolProviderType.BUILT_IN: - builtin_tool = cls.get_builtin_tool(provider_id, tool_name, tenant_id) - if not builtin_tool: - raise ValueError(f"tool {tool_name} not found") - # check if the builtin tool need credentials provider_controller = cls.get_builtin_provider(provider_id, tenant_id) + + builtin_tool = provider_controller.get_tool(tool_name) + if not builtin_tool: + raise ToolProviderNotFoundError(f"builtin tool {tool_name} not found") + if not provider_controller.need_credentials: return cast( BuiltinTool,