Unverified Commit 2d1cb076 authored by takatost's avatar takatost Committed by GitHub

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

parent 289c93d0
...@@ -79,6 +79,8 @@ class MultiDatasetRouterAgent(OpenAIFunctionsAgent): ...@@ -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)
......
...@@ -107,6 +107,8 @@ class StructuredMultiDatasetRouterAgent(StructuredChatAgent): ...@@ -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, "
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment