Spaces:
Sleeping
Sleeping
Sync from GitHub via hub-sync
Browse files- 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 =
|
| 19 |
-
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(
|