AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 10:45:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: skip dataset icon (#3696)

Browse Source
This commit is contained in:
Yeuoly 2024-04-23 12:41:41 +08:00 committed by GitHub
parent 9525ccac4f
commit 2c50fab3dd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 16 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
api/services/agent_service.py
View File

@ -82,19 +82,22 @@ class AgentService:
tool_output = tool_outputs.get(tool_name, {}) tool_output = tool_outputs.get(tool_name, {})
tool_meta_data = tool_meta.get(tool_name, {}) tool_meta_data = tool_meta.get(tool_name, {})
tool_config = tool_meta_data.get('tool_config', {}) tool_config = tool_meta_data.get('tool_config', {})
tool_icon = ToolManager.get_tool_icon( if tool_config.get('tool_provider_type', '') != 'dataset-retrieval':
tenant_id=app_model.tenant_id, tool_icon = ToolManager.get_tool_icon(
provider_type=tool_config.get('tool_provider_type', ''), tenant_id=app_model.tenant_id,
provider_id=tool_config.get('tool_provider', ''), 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 not tool_icon:
if tool_entity: tool_entity = find_agent_tool(tool_name)
tool_icon = ToolManager.get_tool_icon( if tool_entity:
tenant_id=app_model.tenant_id, tool_icon = ToolManager.get_tool_icon(
provider_type=tool_entity.provider_type, tenant_id=app_model.tenant_id,
provider_id=tool_entity.provider_id, provider_type=tool_entity.provider_type,
) provider_id=tool_entity.provider_id,
)
else:
tool_icon = ''
tool_calls.append({ tool_calls.append({
'status': 'success' if not tool_meta_data.get('error') else 'error', 'status': 'success' if not tool_meta_data.get('error') else 'error',