diff --git a/src/upsonic/reliability_processor.py b/src/upsonic/reliability_processor.py index 7443f64cb3..dacdb6ca06 100644 --- a/src/upsonic/reliability_processor.py +++ b/src/upsonic/reliability_processor.py @@ -49,14 +49,14 @@ def process_result( validator_agent = AgentConfiguration("Information Validator Agent", model=llm_model, sub_task=False) copy_task = deepcopy(task) copy_task._response = result - validator_task = Task("Evaluate the old question and its answer for correctness, consistency, and completeness", context=[copy_task], response_format=ValidationResult) + validator_task = Task("Evaluate the old question and its answer for correctness, consistency, and completeness", context=[copy_task], response_format=ValidationResult, tools=task.tools) validator_agent.do(validator_task) if validator_task.response.result == True: return result else: editor_agent = AgentConfiguration("Information Editor Agent", model=llm_model, sub_task=False) - editor_task = Task("Edit the answer to the question to correct the inaccuracies by the feedback provided", context=[copy_task, validator_task], response_format=task.response_format) + editor_task = Task("Edit the answer to the question to correct the inaccuracies by the feedback provided", context=[copy_task, validator_task], response_format=task.response_format, tools=task.tools) editor_agent.do(editor_task) return editor_task.response