first commit, creating PR

UIUC-Chatbot · Oct 24, 2023 · 4c3e6a9 · 4c3e6a9
1 parent 89f61aa
commit 4c3e6a9
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/ai_ta_backend/vector_database.py b/ai_ta_backend/vector_database.py
@@ -935,7 +935,7 @@ def getAll(
     return distinct_dicts
 
   def vector_search(self, search_query, course_name):
-      top_n = 80
+      top_n = 5
       o = OpenAIEmbeddings() # type: ignore
       user_query_embedding = o.embed_query(search_query)
       myfilter = models.Filter(
@@ -955,6 +955,11 @@ def vector_search(self, search_query, course_name):
       )
 
       print("search_results", search_results)
+      print("\n")
+      for i in search_results:
+        print("scores: ", i.score)
+        print("payload: ", i.payload['metadata'])
+      exit()
       found_docs: list[Document] = []
       for d in search_results:
         metadata = d.payload.get('metadata') # type: ignore