| import os | |
| EMBEDDING_MODEL = "thenlper/gte-small" | |
| QDRANT_COLLECTION_NAME = "polars-documentation" | |
| QDRANT_URL = os.environ.get("QDRANT_URL", "") | |
| QDRANT_API_KEY = os.environ.get("QDRANT_API_KEY", "") | |
| CHAT_API_KEY = os.environ.get("CHAT_API_KEY", "") | |
| 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 | |