projecte-aina/EADOP-RAG

1import logging2import warnings3 4from dotenv import load_dotenv5 6 7from rag import RAG8 9USER_INPUT = 10010 11 12def setup():13    load_dotenv()14    warnings.filterwarnings("ignore")15 16    logging.addLevelName(USER_INPUT, "USER_INPUT")17    logging.basicConfig(format="[%(levelname)s]: %(message)s", level=logging.INFO)18 19 20def interactive(model: RAG):21    logging.info("Write `exit` when you want to stop the model.")22    print()23 24    query = ""25    while query.lower() != "exit":26        logging.log(USER_INPUT, "Write the query or `exit`:")27        query = input()28 29        if query.lower() == "exit":30            break31 32        response = model.get_response(query)33        print(response, end="\n\n")34