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)