Merge pull request cheshire-cat-ai#1061 from primax79/qdrant_timeout

Added CCAT_QDRANT_CLIENT_TIMEOUT environment parameter to configure Q…

Author: pieroit

core/cat/env.py

@@ -24,6 +24,7 @@ def get_supported_env_variables():
         "CCAT_CORS_ENABLED": "true",
         "CCAT_CACHE_TYPE": "in_memory",
         "CCAT_CACHE_DIR": "/tmp",
+        "CCAT_QDRANT_CLIENT_TIMEOUT": None,
     }
 
 

core/cat/memory/vector_memory.py

@@ -65,6 +65,9 @@ def connect_to_vector_memory(self) -> None:
             qdrant_https = is_https(qdrant_host)
             qdrant_host = extract_domain_from_url(qdrant_host)
             qdrant_api_key = get_env("CCAT_QDRANT_API_KEY")
+            
+            qdrant_client_timeout = get_env("CCAT_QDRANT_CLIENT_TIMEOUT")
+            qdrant_client_timeout = int(qdrant_client_timeout) if qdrant_client_timeout is not None else None
 
             try:
                 s = socket.socket()
@@ -81,6 +84,7 @@ def connect_to_vector_memory(self) -> None:
                 port=qdrant_port,
                 https=qdrant_https,
                 api_key=qdrant_api_key,
+                timeout=qdrant_client_timeout
             )
 
     def delete_collection(self, collection_name: str):