"""
 /*************************************************************************
 * 
 * CONFIDENTIAL
 * __________________
 * 
 *  Copyright (2023-2024) AI Labs, IronOne Technologies, LLC
 *  All Rights Reserved
 *
 *  Author  : Theekshana Samaradiwakara
 *  Description :Python Backend API to chat with private data  
 *  CreatedDate : 14/11/2023
 *  LastModifiedDate : 18/03/2024
 *************************************************************************/
 """

def qa_chain_output_parser(result):
    return {
        "question": result["question"], 
        "answer": result["answer"],
        "source_documents": result["source_documents"]
    }

def general_qa_chain_output_parser(result):
    return {
        "question": result["question"], 
        "answer": result["text"],
        "source_documents": []
    }


def out_of_domain_chain_parser(query):
    return {
        "question": query, 
        "answer":"sorry this question is out of my domain.",
        "source_documents":[]
    }