handle generating without context

rag.py

@@ -79,6 +79,13 @@ def get_retreive_answer(enterprise_id,prompt,index,common_id):
             search_kwargs={"k": 3, "score_threshold": 0.6},
         )
 
+        enterprise_context = retriever.invoke(prompt)
+
+        if enterprise_context:
+            print("found enterprise context")
+        else:
+            print("no enterprise context")
+
         if common_id:
             vector_store_commun = PineconeVectorStore(index=index, embedding=embedding,namespace=common_id)
 
@@ -87,7 +94,13 @@ def get_retreive_answer(enterprise_id,prompt,index,common_id):
                 search_kwargs={"k": 3, "score_threshold": 0.1},
             )
 
-            response = retriever.invoke(prompt) + retriever_commun.invoke(prompt)
+            commun_context = retriever_commun.invoke(prompt)
+            if commun_context:
+                print("found commun context")
+            else:
+                print("no commun context")
+
+            response = enterprise_context + commun_context
 
         else:
             response = retriever.invoke(prompt)
@@ -145,10 +158,14 @@ def setup_rag(file_type,content):
 
 def prompt_reformatting(prompt:str,context,query:str,style="formel",tonality="neutre",enterprise_name=""):
     if context == "":
-        return prompt.format(context="Pas de contexte pertinent",messages="",query=query,style=style,tonality=tonality)
+        print("no context found for prompt reormatting")
+        return prompt.format(context="Pas de contexte pertinent",messages="",query=query,style=style,tonality=tonality,enterprise=enterprise_name)
     
+
     docs_names = []
+    print("context found for prompt reormatting")
     for chunk in context:
+        print(chunk.metadata)
         chunk_name = chunk.metadata["filename"]
         if chunk_name not in docs_names:
             docs_names.append(chunk_name)