delete docs

rag.py

@@ -67,17 +67,29 @@ def get_vectorstore(text_chunks,filename, file_type,namespace,index):
         print(e)
         return False
     
-def get_retreive_answer(enterprise_id,prompt,index):
+def get_retreive_answer(enterprise_id,prompt,index,common_id):
     try:
         
         embedding = OpenAIEmbeddings(model="text-embedding-3-large")
         vector_store = PineconeVectorStore(index=index, embedding=embedding,namespace=enterprise_id)
-        
+
         retriever = vector_store.as_retriever(
             search_type="similarity_score_threshold",
             search_kwargs={"k": 3, "score_threshold": 0.6},
         )
-        response = retriever.invoke(prompt)
+
+        if common_id:
+            vector_store_commun = PineconeVectorStore(index=index, embedding=embedding,namespace=common_id)
+
+            retriever_commun = vector_store_commun.as_retriever(
+                search_type="similarity_score_threshold",
+                search_kwargs={"k": 3, "score_threshold": 0.6},
+            )
+
+            response = retriever.invoke(prompt) + retriever_commun.invoke(prompt)
+
+        else:
+            response = retriever.invoke(prompt)
 
         return response
     
@@ -88,21 +100,8 @@ def get_retreive_answer(enterprise_id,prompt,index):
 def generate_response_via_langchain(query: str, stream: bool = False, model: str = "gpt-4o-mini",context:str="",messages = [],style:str="formal",tonality:str="neutral"):
     # Define the prompt template
     template = "En tant qu'IA experte en marketing, réponds avec un style {style} et une tonalité {tonality} dans ta communcation, sachant le context suivant: {context}, et l'historique de la conversation: {messages}, {query}"
-    
-    template = '''
-                Vous êtes un spécialiste de la communication marketing responsable avec une expertise en stratégie de communication numérique, en gestion des médias sociaux, en création de contenu, en notoriété de la marque et en engagement communautaire.
-
-                Je vous fournirai des détails concernant le {style}, la {tonality}, mon entreprise, les contextes spécifiques et les conversations passées afin d'assurer une intégration transparente dans nos projets. Vos tâches comprennent le développement de matériel de marketing numérique engageant et informatif tel que des tweets, des posts LinkedIn, et des newsletters alignés sur Vegetaelis 2024 - le Salon de l'Innovation Végétale et de la Transition Agroécologique qui aura lieu à Pau les 21 et 22 novembre 2024.
 
-                Pour chaque tâche, suivez les étapes suivantes :
 
-                Analyser les préférences des utilisateurs et les exigences du projet à partir des paramètres donnés.
-                Effectuer des recherches sur le sujet et intégrer les tendances actuelles si nécessaire.
-                Créer un contenu captivant adapté au format demandé (message sur Twitter, article sur LinkedIn ou lettre d'information).
-                Relire attentivement les textes pour s'assurer qu'ils respectent la grammaire, la ponctuation et la cohérence stylistique.
-                Livrer le document final en veillant à ce qu'il réponde à toutes les attentes et à toutes les lignes directrices.
-                Le résultat doit être attrayant, informatif, rédigé de manière professionnelle, cohérent avec le style et la tonalité spécifiés, doté des hashtags appropriés si nécessaire, et représentatif des valeurs de l'entreprise de l'utilisateur et des normes de l'industrie.
-                '''
     prompt = PromptTemplate.from_template(template)
     
     # Initialize the OpenAI LLM with the specified model