Jelajahi Sumber

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

takatost 1 tahun lalu
induk
melakukan
2d1cb076c6

+ 2 - 0
api/core/agent/agent/multi_dataset_router_agent.py

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

+ 2 - 0
api/core/agent/agent/structed_multi_dataset_router_agent.py

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