GitHub Actions Bot
Changes from ggruber193/polars-docu-chat-rag
c48903e
raw
history blame
447 Bytes
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