add personal info

Modules/rag.py

@@ -23,7 +23,7 @@ from langchain.retrievers import (
 from huggingface_hub import login
 login(token=os.getenv("HUGGING_FACE_TOKEN"))
 
-def load_chunk_persist_pdf() -> Chroma:
+def load_chunk_persist_pdf(task) -> Chroma:
     
     pdf_folder_path = os.path.join(os.getcwd(),Path(f"data/pdf/{task}"))
     documents = []
@@ -37,12 +37,13 @@ def load_chunk_persist_pdf() -> Chroma:
     os.makedirs("data/chroma_store/", exist_ok=True)
     vectorstore = Chroma.from_documents(
         documents=chunked_documents,
-        embedding=MistralAIEmbeddings(api_key=mistral_api_key),
+        embedding=MistralAIEmbeddings(),
         persist_directory= os.path.join(os.getcwd(),Path("data/chroma_store/"))
     )
     vectorstore.persist()
     return vectorstore
 
+personal_info_vectorstore = load_chunk_persist_pdf("personal_info")
 zero2hero_vectorstore = load_chunk_persist_pdf("zero2hero")
 bodyweight_vectorstore = load_chunk_persist_pdf("bodyweight")
 nutrition_vectorstore = load_chunk_persist_pdf("nutrition")
@@ -51,13 +52,14 @@ zero2hero_retriever = zero2hero_vectorstore.as_retriever()
 nutrition_retriever = nutrition_vectorstore.as_retriever()
 bodyweight_retriever = bodyweight_vectorstore.as_retriever()
 workout_retriever = workout_vectorstore.as_retriever()
+personal_info_retriever = personal_info_vectorstore.as_retriever()
 
 llm = ChatMistralAI(model="mistral-large-latest", mistral_api_key=mistral_api_key, temperature=0)
 
 prompt = ChatPromptTemplate.from_template(
     """
     You are a professional AI coach specialized in fitness, bodybuilding and nutrition.
-    You must adapt to the user : if he is a beginner, use simple words. You are gentle and motivative.
+    You must adapt to the user according to personal informations in the context. A You are gentle and motivative.
     Use the following pieces of retrieved context to answer the question.
     If you don't know the answer, use your common knowledge.
     Use three sentences maximum and keep the answer concise.
@@ -73,7 +75,7 @@ prompt = ChatPromptTemplate.from_template(
 def format_docs(docs):
         return "\n\n".join(doc.page_content for doc in docs)
     
-retriever = MergerRetriever(retrievers=[zero2hero_retriever, bodyweight_retriever, nutrition_retriever, workout_retriever])
+retriever = MergerRetriever(retrievers=[zero2hero_retriever, bodyweight_retriever, nutrition_retriever, workout_retriever, personal_info_retriever])
 
 rag_chain = (
         {"context": retriever | format_docs, "question": RunnablePassthrough()}
@@ -84,6 +86,6 @@ rag_chain = (
 
 
 
-print(rag_chain.invoke("What supplement could i buy to improve my sleep?"))
+print(rag_chain.invoke("WHi I'm Susan. Can you make a fitness program for me please?"))
 
 # print(rag_chain.invoke("I am a 45 years old woman and I have to loose weight for the summer. Provide me with a fitness program, and a nutrition program"))

data/Zero To Hero - Bienvenue dans ta nouvelle vie.pdf

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:1e59511fd335776dcb8b0e98e9e9afe6d22483df94c83755163d5d5d4a7f3809
-size 60211845

data/{RaptorBodyweight.pdf → pdf/personal_info/Susan_Thompson.pdf}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cbc90aebadbad36b8f47cd9ba45a958d278e38e7481037acb9ecee5e3805d771
-size 26474997
+oid sha256:25e4eaa98480003224e09c48011e82678f9947382fc37c16011b75fab80f93a2
+size 55515