fix Qdrant
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import logging
|
||||
from functools import lru_cache
|
||||
from typing import Any
|
||||
|
||||
from qdrant_client import QdrantClient
|
||||
@@ -14,11 +15,13 @@ COLLECTION_FACTS = "memory_facts"
|
||||
COLLECTION_SUMMARIES = "session_summaries"
|
||||
COLLECTION_DOC_CHUNKS = "document_chunks"
|
||||
VECTOR_SIZE = 1536
|
||||
QDRANT_TIMEOUT_SEC = 30.0
|
||||
|
||||
|
||||
@lru_cache
|
||||
def _client() -> QdrantClient:
|
||||
settings = get_settings()
|
||||
return QdrantClient(url=settings.qdrant_url)
|
||||
return QdrantClient(url=settings.qdrant_url, timeout=QDRANT_TIMEOUT_SEC)
|
||||
|
||||
|
||||
def ensure_collections() -> None:
|
||||
|
||||
Reference in New Issue
Block a user