AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-12 04:29:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: dataset segment not exist return agent response (#1329)

Browse Source
This commit is contained in:
takatost 2023-10-12 17:40:20 +08:00 committed by GitHub
parent 289c93d081
commit 2d1cb076c6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/core/agent/agent/multi_dataset_router_agent.py
View File

@ -79,6 +79,8 @@ class MultiDatasetRouterAgent(OpenAIFunctionsAgent):
if isinstance(tool_inputs, dict) and 'query' in tool_inputs: if isinstance(tool_inputs, dict) and 'query' in tool_inputs:
tool_inputs['query'] = kwargs['input'] tool_inputs['query'] = kwargs['input']
agent_decision.tool_input = tool_inputs agent_decision.tool_input = tool_inputs
else:
agent_decision.return_values['output'] = ''
return agent_decision return agent_decision
except Exception as e: except Exception as e:
new_exception = self.model_instance.handle_exceptions(e) new_exception = self.model_instance.handle_exceptions(e)

2
api/core/agent/agent/structed_multi_dataset_router_agent.py
View File

@ -107,6 +107,8 @@ class StructuredMultiDatasetRouterAgent(StructuredChatAgent):
if isinstance(tool_inputs, dict) and 'query' in tool_inputs: if isinstance(tool_inputs, dict) and 'query' in tool_inputs:
tool_inputs['query'] = kwargs['input'] tool_inputs['query'] = kwargs['input']
agent_decision.tool_input = tool_inputs agent_decision.tool_input = tool_inputs
else:
agent_decision.return_values['output'] = ''
return agent_decision return agent_decision
except OutputParserException: except OutputParserException:
return AgentFinish({"output": "I'm sorry, the answer of model is invalid, " return AgentFinish({"output": "I'm sorry, the answer of model is invalid, "