소스 검색

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

tags/0.3.27
takatost 2 년 전
부모
커밋
2d1cb076c6
No account linked to committer's email address
2개의 변경된 파일4개의 추가작업 그리고 0개의 파일을 삭제
  1. 2
    0
      api/core/agent/agent/multi_dataset_router_agent.py
  2. 2
    0
      api/core/agent/agent/structed_multi_dataset_router_agent.py

+ 2
- 0
api/core/agent/agent/multi_dataset_router_agent.py 파일 보기

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
- 0
api/core/agent/agent/structed_multi_dataset_router_agent.py 파일 보기

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, "

Loading…
취소
저장