Update App/Embedding/utils/Initialize.py

App/Embedding/utils/Initialize.py

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