hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-10 19:36:53 +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
1 changed files with 16 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
api/services/agent_service.py
View File

@@ -82,6 +82,7 @@ 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', {})
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', ''),
@@ -95,6 +96,8 @@ class AgentService:
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',