reggpt/chains/qa_agent.py

"""
 /*************************************************************************
 *
 * CONFIDENTIAL
 * __________________
 *
 *  Copyright (2024-2025) AI Labs, IronOne Technologies, LLC
 *  All Rights Reserved
 *
 *  Author  : Theekshana Samaradiwakara
 *  Description : This file contains the server application routing logic. 
 *  CreatedDate : 17/10/2024
 *  LastModifiedDate : 
 *************************************************************************/
"""
import logging
logger = logging.getLogger(__name__)

from langchain.chains import LLMChain

from reggpt.llms.llm import get_model
from reggpt.prompts.qa_agent import qa_agent_prompt

def get_qa_agent_chain(model_type):
    logger.info("creating qa_agent")

    try:
        qa_agent_llm = get_model(model_type)
        qa_agent = LLMChain(llm=qa_agent_llm, prompt=qa_agent_prompt)

        logger.info("qa_agent created")
        return qa_agent

    except Exception as e:
        msg=f"Error : {e}"
        logger.exception(msg)
        raise e