Test / backend /services /qdrant.py
UKielz's picture
Upload 14 files
0bbe8e9 verified
raw
history blame contribute delete
380 Bytes
from qdrant_client import QdrantClient
from backend.config import QDRANT_URL, QDRANT_API_KEY, COLLECTION_NAME, TOP_K
qdrant = QdrantClient(url=QDRANT_URL, api_key=QDRANT_API_KEY, timeout=120.0)
def search_in_qdrant(query_vector, limit=TOP_K):
return qdrant.search(collection_name=COLLECTION_NAME, query_vector=query_vector.tolist(), limit=limit, with_payload=True)