Spaces:
Running
Running
File size: 1,336 Bytes
ca801ba a624e23 ca801ba a624e23 ca801ba |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
"""
/*************************************************************************
*
* 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.model import AVALIABLE_MODELS , MEMORY_WINDOW_K
# from qaPipeline import QAPipeline
# from qaPipeline_retriever_only import QAPipeline
# qaPipeline = QAPipeline()
from reggpt.controller.agent 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
|