Increase Qdrant timeout from defualt 5s to 20s. Getting timeout err w…

…/ doc groups.
UIUC-Chatbot · Apr 1, 2024 · 65b642e · 65b642e
1 parent 5da717c
commit 65b642e
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/ai_ta_backend/database/vector.py b/ai_ta_backend/database/vector.py
@@ -23,11 +23,14 @@ def __init__(self):
     self.qdrant_client = QdrantClient(
         url=os.environ['QDRANT_URL'],
         api_key=os.environ['QDRANT_API_KEY'],
+        timeout=20,  # default is 5 seconds. Getting timeout errors w/ document groups.
     )
 
-    self.vectorstore = Qdrant(client=self.qdrant_client,
-                              collection_name=os.environ['QDRANT_COLLECTION_NAME'],
-                              embeddings=OpenAIEmbeddings(openai_api_type=OPENAI_API_TYPE))
+    self.vectorstore = Qdrant(
+        client=self.qdrant_client,
+        collection_name=os.environ['QDRANT_COLLECTION_NAME'],
+        embeddings=OpenAIEmbeddings(openai_api_type=OPENAI_API_TYPE),
+    )
 
   def vector_search(self, search_query, course_name, doc_groups: List[str], user_query_embedding, top_n):
     """