fix: skip dataset icon (#3696)

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',