technophyle commited on
Commit
ee4d71c
·
verified ·
1 Parent(s): 60b97da

Sync from GitHub via hub-sync

Browse files
Files changed (1) hide show
  1. src/vector_store.py +10 -2
src/vector_store.py CHANGED
@@ -15,8 +15,8 @@ class QdrantVectorStore:
15
  self._ensure_collection()
16
 
17
  def _create_client(self):
18
- url = os.getenv("QDRANT_URL")
19
- api_key = os.getenv("QDRANT_API_KEY")
20
  timeout = int(os.getenv("QDRANT_TIMEOUT_SECONDS", "120"))
21
  if url:
22
  return QdrantClient(
@@ -27,6 +27,14 @@ class QdrantVectorStore:
27
  )
28
  return QdrantClient(":memory:")
29
 
 
 
 
 
 
 
 
 
30
  def _ensure_collection(self):
31
  if not self.client.collection_exists(self.collection_name):
32
  self.client.create_collection(
 
15
  self._ensure_collection()
16
 
17
  def _create_client(self):
18
+ url = self._clean_env("QDRANT_URL")
19
+ api_key = self._clean_env("QDRANT_API_KEY")
20
  timeout = int(os.getenv("QDRANT_TIMEOUT_SECONDS", "120"))
21
  if url:
22
  return QdrantClient(
 
27
  )
28
  return QdrantClient(":memory:")
29
 
30
+ @staticmethod
31
+ def _clean_env(name: str) -> Optional[str]:
32
+ value = os.getenv(name)
33
+ if value is None:
34
+ return None
35
+ cleaned = value.strip()
36
+ return cleaned or None
37
+
38
  def _ensure_collection(self):
39
  if not self.client.collection_exists(self.collection_name):
40
  self.client.create_collection(