File size: 548 Bytes
b5deaf1
 
c529966
e83b975
b5deaf1
 
 
c529966
b5deaf1
 
e83b975
 
b5deaf1
a4e857f
b5deaf1
 
 
a4e857f
 
e83b975
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
"""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": "Just give me an update?",
    }
    chain = RAGChain(DocRetriever(req=req))
    result = chain.invoke({"input": req['query']},
                       config={"configurable": {"session_id": "20250301"}})
    print(result)
    print(result.get("answer"))