from qdrant_client import QdrantClient

from article_embedding.utils import env_str

remote_url = env_str("CLUSTER_1_ENDPOINT")
remote_key = env_str("CLUSTER_1_API_KEY")
collection_name = "wsws-2"
local_url = env_str("QDRANT_URL")

# model = StellaEmbedder()

cloud_client = QdrantClient(url=remote_url, api_key=remote_key)

# cloud_client.recreate_collection(
#     collection_name=collection_name,
#     vectors_config=model.get_vectors_config(),
# )

local_client = QdrantClient(local_url)

local_client.migrate(cloud_client, collection_names=[collection_name])