new: add migrate method to client's interface

qdrant_client/client_base.py

@@ -1,5 +1,11 @@
+import sys
 from typing import Any, Dict, Iterable, List, Mapping, Optional, Sequence, Tuple, Union
 
+if sys.version_info <= (3, 11):
+    from typing_extensions import Self
+else:
+    from typing import Self
+
 from qdrant_client.conversions import common_types as types
 from qdrant_client.http import models
 
@@ -326,3 +332,6 @@ def get_locks(self, **kwargs: Any) -> types.LocksOption:
 
     def close(self, **kwargs: Any) -> None:
         pass
+
+    def migrate(self, dest_client: Self, batch_size: int = 100) -> None:
+        raise NotImplementedError()

qdrant_client/migrate/migrate.py

@@ -1,15 +1,15 @@
 from typing import List
 
-from qdrant_client import QdrantClient
+from qdrant_client.client_base import QdrantBase
 from qdrant_client.http import models
 
 
-def migrate(source_client: QdrantClient, dest_client: QdrantClient, batch_size: int = 100) -> None:
+def migrate(source_client: QdrantBase, dest_client: QdrantBase, batch_size: int = 100) -> None:
     """Migrate all collections from source client to destination client
 
     Args:
-        source_client (QdrantClient): Source client
-        dest_client (QdrantClient): Destination client
+        source_client (QdrantBase): Source client
+        dest_client (QdrantBase): Destination client
         batch_size (int, optional): Batch size for scrolling and uploading vectors. Defaults to 100.
     """
     source_collections = source_client.get_collections().collections
@@ -34,15 +34,15 @@ def migrate(source_client: QdrantClient, dest_client: QdrantClient, batch_size:
 
 def _compare_collections(
     source_collection_names: List[str],
-    source_client: QdrantClient,
-    dest_client: QdrantClient,
+    source_client: QdrantBase,
+    dest_client: QdrantBase,
 ) -> bool:
     """Compare collections from source client and destination client
 
     Args:
         source_collection_names (list[str]): List of collection names
-        source_client (QdrantClient): Source client
-        dest_client (QdrantClient): Destination client
+        source_client (QdrantBase): Source client
+        dest_client (QdrantBase): Destination client
 
     Returns:
         bool: True if collections have the same vector and distance params
@@ -81,16 +81,16 @@ def _compare_collections(
 
 def _migrate_collection(
     collection_name: str,
-    source_client: QdrantClient,
-    dest_client: QdrantClient,
+    source_client: QdrantBase,
+    dest_client: QdrantBase,
     batch_size: int = 100,
 ) -> None:
     """Migrate collection from source client to destination client
 
     Args:
         collection_name (str): Collection name
-        source_client (QdrantClient): Source client
-        dest_client (QdrantClient): Destination client
+        source_client (QdrantBase): Source client
+        dest_client (QdrantBase): Destination client
         batch_size (int, optional): Batch size for scrolling and uploading vectors. Defaults to 100.
     """
     records, next_offset = source_client.scroll(

qdrant_client/qdrant_client.py

@@ -5,6 +5,7 @@
 from qdrant_client.conversions import common_types as types
 from qdrant_client.http import ApiClient, SyncApis
 from qdrant_client.local.qdrant_local import QdrantLocal
+from qdrant_client.migrate import migrate
 from qdrant_client.qdrant_remote import QdrantRemote
 
 
@@ -1708,3 +1709,6 @@ def get_locks(self, **kwargs: Any) -> types.LocksOption:
         assert len(kwargs) == 0, f"Unknown arguments: {list(kwargs.keys())}"
 
         return self._client.get_locks(**kwargs)
+
+    def migrate(self, dest_client: QdrantBase, batch_size: int = 100) -> None:
+        migrate(self, dest_client, batch_size=batch_size)

qdrant_client/migrate/tests/test_migrate.py → tests/test_migrate.py

@@ -3,7 +3,6 @@
 
 from qdrant_client import QdrantClient
 from qdrant_client.http import models
-from qdrant_client.migrate import migrate
 
 VECTOR_NUMBER = 1000
 
@@ -37,7 +36,7 @@ def test_single_vector_collection(source_client: QdrantClient, dest_client: Qdra
         ),
     )
 
-    migrate(source_client, dest_client)
+    source_client.migrate(dest_client)
 
 
 def test_multiple_vectors_collection(
@@ -65,6 +64,7 @@ def test_multiple_vectors_collection(
             ),
         },
     )
+    source_client.migrate(dest_client)
 
 
 def test_multiple_collections(source_client: QdrantClient, dest_client: QdrantClient) -> None:
@@ -78,7 +78,7 @@ def test_multiple_collections(source_client: QdrantClient, dest_client: QdrantCl
             vectors=np.random.randn(VECTOR_NUMBER, vector_params.size),
         )
 
-    migrate(source_client, dest_client)
+    source_client.migrate(dest_client)
 
     for collection_name in collection_names:
         source_vector_number = source_client.get_collection(collection_name).vectors_count
@@ -98,7 +98,7 @@ def test_different_distances(source_client: QdrantClient, dest_client: QdrantCli
     dest_client.recreate_collection(collection_name, vectors_config=euclid_params)
 
     with pytest.raises(AssertionError):
-        migrate(source_client, dest_client)
+        source_client.migrate(dest_client)
 
 
 def test_different_vector_sizes(source_client: QdrantClient, dest_client: QdrantClient) -> None:
@@ -110,7 +110,7 @@ def test_different_vector_sizes(source_client: QdrantClient, dest_client: Qdrant
     dest_client.recreate_collection(collection_name, vectors_config=big_vector_params)
 
     with pytest.raises(AssertionError):
-        migrate(source_client, dest_client)
+        source_client.migrate(dest_client)
 
 
 def test_single_vs_multiple_vectors(
@@ -127,4 +127,4 @@ def test_single_vs_multiple_vectors(
     dest_client.recreate_collection(collection_name, vectors_config=multiple_vectors_params)
 
     with pytest.raises(AssertionError):
-        migrate(source_client, dest_client)
+        source_client.migrate(dest_client)

tests/test_qdrant_client.py

@@ -1,5 +1,4 @@
 import os
-import random
 import uuid
 from pprint import pprint
 from tempfile import mkdtemp