EMBEDDING_MODEL = "thenlper/gte-small" | |
QDRANT_COLLECTION_NAME = "polars-documentation" | |
def get_qdrant_config(): | |
from qdrant_client import models | |
QDRANT_COLLECTION_CONFIG = { | |
"collection_name": QDRANT_COLLECTION_NAME, | |
"vectors_config": models.VectorParams(size=384, distance=models.Distance.COSINE), # on_disk=True), | |
# "hnsw_config": models.HnswConfigDiff(on_disk=True) | |
} | |
return QDRANT_COLLECTION_CONFIG | |