fix: skip dataset icon (langgenius#3696)

ocxers · Apr 23, 2024 · 2c50fab · 2c50fab
1 parent 9525cca
commit 2c50fab
Showing 1 changed file with 16 additions and 13 deletions.
diff --git a/api/services/agent_service.py b/api/services/agent_service.py
@@ -82,19 +82,22 @@ def find_agent_tool(tool_name: str):
                 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',