ConvoBot / src /tests /test_qdrant_integration.py
ashish-ninehertz
changes
e272f4f
raw
history blame contribute delete
466 Bytes
import pytest
from qdrant_client import QdrantClient
from qdrant_client.models import VectorParams, Distance
@pytest.fixture
def qdrant_client():
return QdrantClient(host="localhost", port=6333)
def test_collection_creation(qdrant_client):
test_collection = "test_collection"
qdrant_client.recreate_collection(test_collection, vectors_config=VectorParams(size=384, distance=Distance.COSINE))
assert qdrant_client.collection_exists(test_collection)