reggpt/routers/controller.py

"""
 /*************************************************************************
 * 
 * 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
 *************************************************************************/
 """

import logging
logger = logging.getLogger(__name__)
from reggpt.configs.config import AVALIABLE_MODELS , MEMORY_WINDOW_K

# from qaPipeline import QAPipeline
# from qaPipeline_retriever_only import QAPipeline
# qaPipeline = QAPipeline()

from reggpt.routers.qaPipeline import run_agent

def get_QA_Answers(userQuery):
    # model=userQuery.model
    # query=userQuery.content
    query=userQuery.user_question

    # chat_history = userQuery.chat_history[-MEMORY_WINDOW_K:]
    
    # logger.info(f"model: {model} \n query : {query} \n chat_history : {chat_history}")
    logger.info(f"query : {query}")
    # answer= run_agent(query=query, model=model, chat_history=chat_history)
    answer= run_agent(query=query)
    logger.info(f"Response: {answer}")
    return answer


def get_avaliable_models():
    logger.info("getting avaliable models")
    return AVALIABLE_MODELS