main.py

"""Module to run the mail collection process."""
from dotenv import load_dotenv
from controllers import mail
from chain import RAGChain
from retriever import DocRetriever

# load_dotenv()

if __name__ == "__main__":
    mail.collect()
    mail.get_documents()
    req = {
        "query": "What is the latest news on the stock market?",
    }
    chain = RAGChain(DocRetriever(req=req))
    result = chain.invoke({"input": req['query']},
                       config={"configurable": {"session_id": "abc"}})
    print(result)