This commit is contained in:
2025-04-17 15:55:54 -04:00
parent 5ca30d5e11
commit 79f9a1dbc2
12 changed files with 611 additions and 80 deletions

View File

@@ -3,12 +3,17 @@ from qdrant_client.http import models
from qdrant_client.models import Distance, VectorParams
client = QdrantClient(host="localhost", port=6333)
# %%
collection_head = "nuggets_clip"
collection_head = "nuggets_so400m"
out = client.scroll('nuggets_so400m',limit=10,with_vectors=True)
print(len(out[0][0].vector))
# %%
ou = np.asarray([x.vector for x in out[0]], dtype=np.float16)
# %%
for collection_name in [collection_head, collection_head +'_averaged']:
try:
client.create_collection(
@@ -33,3 +38,27 @@ for collection_name in [collection_head, collection_head +'_averaged']:
from prettyprinter import cpprint
cpprint(client.get_collection(collection_name).dict()['vectors_count'])
# %%
client.update_collection(
collection_name=f"{collection_name}",
hnsw_config=models.HnswConfigDiff(
on_disk=True
),
vectors_config={
"": models.VectorParamsDiff(
on_disk=True,
hnsw_config=models.HnswConfigDiff(
on_disk=True
),
)
}
)
# %%
cpprint(client.get_collection(collection_name).dict())