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

fix: api tool provider not found (#2782)

Browse Source
This commit is contained in:
Yeuoly 2024-03-11 18:21:41 +08:00 committed by GitHub
parent 6454e1d644
commit f8951d7f57
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 44 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/controllers/console/app/app.py
View File

@ -247,6 +247,7 @@ class AppApi(Resource):
for tool in agent_mode.get('tools') or []: for tool in agent_mode.get('tools') or []:
agent_tool_entity = AgentToolEntity(**tool) agent_tool_entity = AgentToolEntity(**tool)
# get tool # get tool
try:
tool_runtime = ToolManager.get_agent_tool_runtime( tool_runtime = ToolManager.get_agent_tool_runtime(
tenant_id=current_user.current_tenant_id, tenant_id=current_user.current_tenant_id,
agent_tool=agent_tool_entity, agent_tool=agent_tool_entity,
@ -268,6 +269,8 @@ class AppApi(Resource):
# override tool parameters # override tool parameters
tool['tool_parameters'] = masked_parameter tool['tool_parameters'] = masked_parameter
except Exception as e:
pass
# override agent mode # override agent mode
model_config.agent_mode = json.dumps(agent_mode) model_config.agent_mode = json.dumps(agent_mode)

6
api/controllers/console/app/model_config.py
View File

@ -54,6 +54,7 @@ class ModelConfigResource(Resource):
for tool in agent_mode.get('tools') or []: for tool in agent_mode.get('tools') or []:
agent_tool_entity = AgentToolEntity(**tool) agent_tool_entity = AgentToolEntity(**tool)
# get tool # get tool
try:
tool_runtime = ToolManager.get_agent_tool_runtime( tool_runtime = ToolManager.get_agent_tool_runtime(
tenant_id=current_user.current_tenant_id, tenant_id=current_user.current_tenant_id,
agent_tool=agent_tool_entity, agent_tool=agent_tool_entity,
@ -65,6 +66,8 @@ class ModelConfigResource(Resource):
provider_name=agent_tool_entity.provider_id, provider_name=agent_tool_entity.provider_id,
provider_type=agent_tool_entity.provider_type, provider_type=agent_tool_entity.provider_type,
) )
except Exception as e:
continue
# get decrypted parameters # get decrypted parameters
if agent_tool_entity.tool_parameters: if agent_tool_entity.tool_parameters:
@ -89,11 +92,14 @@ class ModelConfigResource(Resource):
if key in tool_map: if key in tool_map:
tool_runtime = tool_map[key] tool_runtime = tool_map[key]
else: else:
try:
tool_runtime = ToolManager.get_agent_tool_runtime( tool_runtime = ToolManager.get_agent_tool_runtime(
tenant_id=current_user.current_tenant_id, tenant_id=current_user.current_tenant_id,
agent_tool=agent_tool_entity, agent_tool=agent_tool_entity,
agent_callback=None agent_callback=None
) )
except Exception as e:
continue
manager = ToolParameterConfigurationManager( manager = ToolParameterConfigurationManager(
tenant_id=current_user.current_tenant_id, tenant_id=current_user.current_tenant_id,