Skip to content
11 changes: 11 additions & 0 deletions redisvl/migration/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,8 +6,13 @@
async_wait_for_index_ready,
)
from redisvl.migration.async_validation import AsyncMigrationValidator
from redisvl.migration.batch_executor import BatchMigrationExecutor
from redisvl.migration.batch_planner import BatchMigrationPlanner
from redisvl.migration.executor import MigrationExecutor
from redisvl.migration.models import (
BatchPlan,
BatchReport,
BatchState,
DiskSpaceEstimate,
FieldRename,
MigrationPlan,
Expand All @@ -29,6 +34,12 @@
"MigrationValidator",
"RenameOperations",
"SchemaPatch",
# Batch
"BatchMigrationExecutor",
"BatchMigrationPlanner",
"BatchPlan",
"BatchReport",
"BatchState",
# Async
"AsyncMigrationExecutor",
"AsyncMigrationPlanner",
Expand Down
3 changes: 2 additions & 1 deletion redisvl/migration/async_executor.py
Original file line number Diff line number Diff line change
Expand Up @@ -744,7 +744,8 @@ def _notify(step: str, detail: Optional[str] = None) -> None:
effective_changes = datatype_changes
if has_field_renames:
field_rename_map = {
fr.old_name: fr.new_name for fr in rename_ops.rename_fields
fr.old_name: fr.new_name
for fr in rename_ops.rename_fields
}
effective_changes = {
field_rename_map.get(k, k): v
Expand Down
Loading
Loading