From cdd610f94fd705f5e714c8642b8b9d9e7aead282 Mon Sep 17 00:00:00 2001 From: Yeuoly Date: Sat, 8 Feb 2025 15:26:51 +0800 Subject: [PATCH] fix: handle jina/gemini/siliconflow --- api/services/plugin/dependencies_analysis.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/api/services/plugin/dependencies_analysis.py b/api/services/plugin/dependencies_analysis.py index df0c321a8b..237ca3b629 100644 --- a/api/services/plugin/dependencies_analysis.py +++ b/api/services/plugin/dependencies_analysis.py @@ -13,6 +13,8 @@ class DependenciesAnalysisService: """ try: tool_provider_id = GenericProviderID(tool_id) + if tool_id in ["jina", "siliconflow"]: + tool_provider_id.plugin_name = tool_provider_id.plugin_name + "_tool" return tool_provider_id.plugin_id except Exception as e: raise e @@ -26,6 +28,9 @@ class DependenciesAnalysisService: """ try: generic_provider_id = GenericProviderID(model_provider_id) + if model_provider_id == "google": + generic_provider_id.plugin_name = "gemini" + return generic_provider_id.plugin_id except Exception as e: raise e