Skip to content

Commit 249ccdd

Browse files
author
d.rudenko
committed
Added support of strict mode to migrate.py + tests
1 parent 3366f14 commit 249ccdd

File tree

2 files changed

+6
-0
lines changed

2 files changed

+6
-0
lines changed

qdrant_client/migrate/migrate.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -125,6 +125,7 @@ def _recreate_collection(
125125
optimizers_config=models.OptimizersConfigDiff(**to_dict(src_config.optimizer_config)),
126126
wal_config=models.WalConfigDiff(**to_dict(src_config.wal_config)),
127127
quantization_config=src_config.quantization_config,
128+
strict_mode_config=src_config.strict_mode_config,
128129
)
129130

130131
_recreate_payload_schema(dest_client, collection_name, src_payload_schema)

tests/test_migrate.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -479,6 +479,10 @@ def test_recreate_collection(remote_client: QdrantClient):
479479
)
480480
)
481481

482+
strict_mode_config = models.StrictModeConfig(
483+
read_rate_limit=1000000
484+
)
485+
482486
remote_client.create_collection(
483487
collection_name,
484488
vectors_config=vectors_config,
@@ -490,6 +494,7 @@ def test_recreate_collection(remote_client: QdrantClient):
490494
optimizers_config=optimizers_config,
491495
wal_config=wal_config,
492496
quantization_config=general_quantization_config,
497+
strict_mode_config=strict_mode_config,
493498
)
494499

495500
remote_client.create_payload_index(

0 commit comments

Comments
 (0)