File size: 380 Bytes
0bbe8e9 | 1 2 3 4 5 6 7 8 9 | 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)
|