payload.py

"""Runner for Agents to output answers"""


def get_answer_payload_results_log(questions_data: dict) -> tuple:
    """Get Answer Payload, Results Log or Error Message"""
    results_log = []
    answers_payload = []
    submitted_answer = "Mock Final Answer Test Data"

    for item in questions_data:
        task_id = item.get("task_id")
        question_text = item.get("question")
        if not task_id or question_text is None:
            print(f"Skipping item with missing task_id or question: {item}")
            continue
        answers_payload.append(
            {"task_id": task_id, "submitted_answer": submitted_answer}
        )
        results_log.append(
            {
                "Task ID": task_id,
                "Question": question_text,
                "Submitted Answer": submitted_answer,
            }
        )
    return answers_payload, results_log, None