Update DocRetriever to adjust k value and score threshold; add debug print for email subject

app/controllers/mail.py

@@ -85,6 +85,7 @@ def list_emails(service, messages):
                 metadata["to"] = header["value"]
             elif header["name"] == "Subject":
                 metadata["subject"] = header["value"]
+                print(f"subject: {metadata["subject"]}")
             elif header["name"] == "Cc":
                 metadata["cc"] = header["value"]
         metadata["date"] = datetime.fromtimestamp(int(msg["internalDate"]) / 1000).strftime(

app/retriever/__init__.py

@@ -22,19 +22,19 @@ class DocRetriever(BaseRetriever):
                 list: A list of Document objects with relevant metadata.
     """
     retriever: VectorStoreRetriever = None
-    k: int = 10
+    k: int = 6
 
-    def __init__(self, req, k: int = 10) -> None:
+    def __init__(self, req, k: int = 6) -> None:
         super().__init__()
         _filter={}
         _filter.update({"user_id": req.user_id})
         print(_filter)
         self.retriever = vectorstore.as_retriever(
-            search_type='similarity',
+            search_type='similarity_score_threshold',
             search_kwargs={
                 "k": k,
                 "filter": _filter,
-                # "score_threshold": .1
+                "score_threshold": .7
             }
         )