From 2c50fab3ddce50d8ef1ec49af0e225604cea1226 Mon Sep 17 00:00:00 2001 From: Yeuoly <45712896+Yeuoly@users.noreply.github.com> Date: Tue, 23 Apr 2024 12:41:41 +0800 Subject: [PATCH] fix: skip dataset icon (#3696) --- api/services/agent_service.py | 29 ++++++++++++++++------------- 1 file changed, 16 insertions(+), 13 deletions(-) diff --git a/api/services/agent_service.py b/api/services/agent_service.py index b553d3c884..a25a48dc35 100644 --- a/api/services/agent_service.py +++ b/api/services/agent_service.py @@ -82,19 +82,22 @@ class AgentService: tool_output = tool_outputs.get(tool_name, {}) tool_meta_data = tool_meta.get(tool_name, {}) tool_config = tool_meta_data.get('tool_config', {}) - tool_icon = ToolManager.get_tool_icon( - tenant_id=app_model.tenant_id, - provider_type=tool_config.get('tool_provider_type', ''), - provider_id=tool_config.get('tool_provider', ''), - ) - if not tool_icon: - tool_entity = find_agent_tool(tool_name) - if tool_entity: - tool_icon = ToolManager.get_tool_icon( - tenant_id=app_model.tenant_id, - provider_type=tool_entity.provider_type, - provider_id=tool_entity.provider_id, - ) + if tool_config.get('tool_provider_type', '') != 'dataset-retrieval': + tool_icon = ToolManager.get_tool_icon( + tenant_id=app_model.tenant_id, + provider_type=tool_config.get('tool_provider_type', ''), + provider_id=tool_config.get('tool_provider', ''), + ) + if not tool_icon: + tool_entity = find_agent_tool(tool_name) + if tool_entity: + tool_icon = ToolManager.get_tool_icon( + tenant_id=app_model.tenant_id, + provider_type=tool_entity.provider_type, + provider_id=tool_entity.provider_id, + ) + else: + tool_icon = '' tool_calls.append({ 'status': 'success' if not tool_meta_data.get('error') else 'error',