Wider window

App/Chat/utils/RAG.py

@@ -19,7 +19,11 @@ class GenerativeAIAssistant:
         self.examples=[{"input": {"content": "hello"}, "output": {"content": "Hello to you too! How can I help you today?"}}]
         self.context = "You are a helpful assistant"
 
-    def generate_template(self,question,task_id):
+    def generate_template(self,question,task_id,summary=None):
+        if summary == None:
+            self.context = "You are a helpful assistant"
+        else:
+            self.context = summary
         contexts=search(question,task_id=task_id)
         context_yaml = ""        
         for context in contexts:

App/Embedding/utils/Initialize.py

@@ -70,7 +70,7 @@ def search(query: str, task_id: str):
     filtering_conditions = {
             "task_id": {"$eq": task_id},
         }
-    data =docsearch.similarity_search(query, k=4, filter=filtering_conditions)
+    data =docsearch.similarity_search(query, k=10, filter=filtering_conditions)
     return [
         {"text": d.page_content, "start": d.metadata["start"], "end": d.metadata["end"]}
         for d in data