SunnyQjm/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-06-05 15:40:14 +08:00
Code Issues Projects Releases Wiki Activity

refactor(tools): replace redundant dict[str, str] with EmojiIconDict (#34786)

Browse Source 
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
This commit is contained in:
Ygor Leal
2026-04-10 12:53:05 +02:00
committed by GitHub
parent c9f525a3b2
commit 8633b2f1f7
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
api/core/tools/tool_manager.py
+3 -3
View File
@@ -993,7 +993,7 @@ class ToolManager:
return {"background": "#252525", "content": "\ud83d\ude01"} return {"background": "#252525", "content": "\ud83d\ude01"}
@classmethod @classmethod
def generate_mcp_tool_icon_url(cls, tenant_id: str, provider_id: str) -> EmojiIconDict | dict[str, str] | str: def generate_mcp_tool_icon_url(cls, tenant_id: str, provider_id: str) -> EmojiIconDict | str:
try: try:
with Session(db.engine) as session: with Session(db.engine) as session:
mcp_service = MCPToolManageService(session=session) mcp_service = MCPToolManageService(session=session)
@@ -1001,7 +1001,7 @@ class ToolManager:
mcp_provider = mcp_service.get_provider_entity( mcp_provider = mcp_service.get_provider_entity(
provider_id=provider_id, tenant_id=tenant_id, by_server_id=True provider_id=provider_id, tenant_id=tenant_id, by_server_id=True
) )
return mcp_provider.provider_icon return cast(EmojiIconDict | str, mcp_provider.provider_icon)
except ValueError: except ValueError:
raise ToolProviderNotFoundError(f"mcp provider {provider_id} not found") raise ToolProviderNotFoundError(f"mcp provider {provider_id} not found")
except Exception: except Exception:
@@ -1013,7 +1013,7 @@ class ToolManager:
tenant_id: str, tenant_id: str,
provider_type: ToolProviderType, provider_type: ToolProviderType,
provider_id: str, provider_id: str,
) -> str | EmojiIconDict | dict[str, str]: ) -> str | EmojiIconDict:
""" """
get the tool icon get the tool icon