Losita/mai-bot
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(migration): enhance migration progress reporting with detailed record and table tracking

Browse Source
This commit is contained in:
DrSmoothl
2026-03-31 09:29:58 +08:00
parent c2c992ff01
commit 5d410171d2
4 changed files with 422 additions and 238 deletions
Download Patch File Download Diff File Expand all files Collapse all files

467
src/common/database/migrations/legacy_v1_to_v2.py
View File

@@ -66,7 +66,7 @@ def migrate_legacy_v1_to_v2(context: MigrationExecutionContext) -> None:
_rename_legacy_v1_tables(context.connection, snapshot)
SQLModel.metadata.create_all(context.connection)
table_migration_jobs: List[Tuple[str, Callable[[Connection], int]]] = [
table_migration_jobs: List[Tuple[str, Callable[[MigrationExecutionContext], int]]] = [
("chat_sessions", _migrate_chat_sessions),
("llm_usage", _migrate_model_usage),
("images", _migrate_images),
@@ -81,10 +81,16 @@ def migrate_legacy_v1_to_v2(context: MigrationExecutionContext) -> None:
("thinking_questions", _migrate_thinking_questions),
]
migrated_counts: Dict[str, int] = {}
context.start_progress(total=len(table_migration_jobs), description="总迁移进度", unit_name="表")
total_record_count = _estimate_total_record_count(context.connection)
context.start_progress(
total_tables=len(table_migration_jobs),
total_records=total_record_count,
description="总迁移进度",
table_unit_name="表",
record_unit_name="记录",
)
for table_name, migration_handler in table_migration_jobs:
migrated_counts[table_name] = migration_handler(context.connection)
context.advance_progress(item_name=table_name)
migrated_counts[table_name] = migration_handler(context)
summary_text = ", ".join(f"{table_name}={count}" for table_name, count in migrated_counts.items())
logger.info(f"旧版数据库迁移完成: {summary_text}")
@@ -535,17 +541,77 @@ def _deduce_image_type_name(value: Any) -> str:
return "IMAGE"
def _migrate_chat_sessions(connection: Connection) -> int:
"""迁移旧版 ``chat_streams`` 到新版 ``chat_sessions``。
def _count_legacy_table_rows(connection: Connection, original_table_name: str) -> int:
"""统计单张旧版备份表中的记录总数。
Args:
connection: 当前数据库连接。
original_table_name: 旧版原始表名。
Returns:
int: 备份表中的记录数；若表不存在则返回 ``0``。
"""
backup_table_name = _legacy_backup_table_name(original_table_name)
schema_inspector = SQLiteSchemaInspector()
if not schema_inspector.table_exists(connection, backup_table_name):
return 0
row = connection.execute(
text(f"SELECT COUNT(*) FROM {_quote_identifier(backup_table_name)}")
).first()
if row is None:
return 0
return _normalize_int(row[0], default=0)
def _estimate_total_record_count(connection: Connection) -> int:
"""估算旧版迁移步骤需要处理的总记录数。
Args:
connection: 当前数据库连接。
Returns:
int: 本次迁移预计处理的总记录数。
"""
return (
_count_legacy_table_rows(connection, "chat_streams")
+ _count_legacy_table_rows(connection, "llm_usage")
+ _count_legacy_table_rows(connection, "emoji")
+ _count_legacy_table_rows(connection, "images")
+ _count_legacy_table_rows(connection, "messages")
+ _count_legacy_table_rows(connection, "action_records")
+ _count_legacy_table_rows(connection, "action_records")
+ _count_legacy_table_rows(connection, "online_time")
+ _count_legacy_table_rows(connection, "person_info")
+ _count_legacy_table_rows(connection, "expression")
+ _count_legacy_table_rows(connection, "jargon")
+ _count_legacy_table_rows(connection, "chat_history")
+ _count_legacy_table_rows(connection, "thinking_back")
)
def _complete_table_progress(context: MigrationExecutionContext, table_name: str) -> None:
"""标记单张表的迁移已经完成。
Args:
context: 当前迁移步骤执行上下文。
table_name: 已完成迁移的表名。
"""
context.advance_progress(completed_tables=1, item_name=table_name)
def _migrate_chat_sessions(context: MigrationExecutionContext) -> int:
"""迁移旧版 ``chat_streams`` 到新版 ``chat_sessions``。
Args:
context: 当前迁移步骤执行上下文。
Returns:
int: 迁移成功的记录数。
"""
connection = context.connection
legacy_table = _load_legacy_table_data(connection, "chat_streams")
if legacy_table is None:
_complete_table_progress(context, "chat_sessions")
return 0
migrated_count = 0
@@ -570,34 +636,37 @@ def _migrate_chat_sessions(connection: Connection) -> int:
)
for row in legacy_table.rows:
session_id = _normalize_required_text(row.get("stream_id"))
if not session_id:
continue
connection.execute(
insert_sql,
{
"session_id": session_id,
"created_timestamp": _coerce_datetime(row.get("create_time"), fallback_now=True),
"last_active_timestamp": _coerce_datetime(row.get("last_active_time"), fallback_now=True),
"user_id": _normalize_optional_text(row.get("user_id")),
"group_id": _normalize_optional_text(row.get("group_id")),
"platform": _normalize_required_text(row.get("platform"), default="unknown"),
},
)
migrated_count += 1
if session_id:
connection.execute(
insert_sql,
{
"session_id": session_id,
"created_timestamp": _coerce_datetime(row.get("create_time"), fallback_now=True),
"last_active_timestamp": _coerce_datetime(row.get("last_active_time"), fallback_now=True),
"user_id": _normalize_optional_text(row.get("user_id")),
"group_id": _normalize_optional_text(row.get("group_id")),
"platform": _normalize_required_text(row.get("platform"), default="unknown"),
},
)
migrated_count += 1
context.advance_progress(records=1)
_complete_table_progress(context, "chat_sessions")
return migrated_count
def _migrate_model_usage(connection: Connection) -> int:
def _migrate_model_usage(context: MigrationExecutionContext) -> int:
"""迁移旧版 ``llm_usage`` 到新版 ``llm_usage``。
Args:
connection: 当前数据库连接。
context: 当前迁移步骤执行上下文。
Returns:
int: 迁移成功的记录数。
"""
connection = context.connection
legacy_table = _load_legacy_table_data(connection, "llm_usage")
if legacy_table is None:
_complete_table_progress(context, "llm_usage")
return 0
migrated_count = 0
@@ -654,18 +723,21 @@ def _migrate_model_usage(connection: Connection) -> int:
},
)
migrated_count += 1
context.advance_progress(records=1)
_complete_table_progress(context, "llm_usage")
return migrated_count
def _migrate_images(connection: Connection) -> int:
def _migrate_images(context: MigrationExecutionContext) -> int:
"""迁移旧版 ``emoji`` 与 ``images`` 到新版 ``images``。
Args:
connection: 当前数据库连接。
context: 当前迁移步骤执行上下文。
Returns:
int: 迁移成功的记录数。
"""
connection = context.connection
migrated_count = 0
existing_keys: Set[Tuple[str, str, str]] = set()
existing_rows = connection.execute(
@@ -719,28 +791,28 @@ def _migrate_images(connection: Connection) -> int:
full_path = _normalize_required_text(row.get("full_path"))
image_hash = _normalize_required_text(row.get("emoji_hash"))
dedupe_key = (full_path, image_hash, "EMOJI")
if not full_path or dedupe_key in existing_keys:
continue
connection.execute(
insert_sql,
{
"image_hash": image_hash,
"description": _normalize_required_text(row.get("description")),
"full_path": full_path,
"image_type": "EMOJI",
"emotion": _normalize_optional_text(row.get("emotion")),
"query_count": _normalize_int(row.get("query_count"), default=0),
"is_registered": _normalize_bool(row.get("is_registered"), default=False),
"is_banned": _normalize_bool(row.get("is_banned"), default=False),
"no_file_flag": False,
"record_time": _coerce_datetime(row.get("record_time"), fallback_now=True),
"register_time": _coerce_datetime(row.get("register_time")),
"last_used_time": _coerce_datetime(row.get("last_used_time")),
"vlm_processed": False,
},
)
existing_keys.add(dedupe_key)
migrated_count += 1
if full_path and dedupe_key not in existing_keys:
connection.execute(
insert_sql,
{
"image_hash": image_hash,
"description": _normalize_required_text(row.get("description")),
"full_path": full_path,
"image_type": "EMOJI",
"emotion": _normalize_optional_text(row.get("emotion")),
"query_count": _normalize_int(row.get("query_count"), default=0),
"is_registered": _normalize_bool(row.get("is_registered"), default=False),
"is_banned": _normalize_bool(row.get("is_banned"), default=False),
"no_file_flag": False,
"record_time": _coerce_datetime(row.get("record_time"), fallback_now=True),
"register_time": _coerce_datetime(row.get("register_time")),
"last_used_time": _coerce_datetime(row.get("last_used_time")),
"vlm_processed": False,
},
)
existing_keys.add(dedupe_key)
migrated_count += 1
context.advance_progress(records=1)
legacy_images_table = _load_legacy_table_data(connection, "images")
if legacy_images_table is not None:
@@ -749,43 +821,46 @@ def _migrate_images(connection: Connection) -> int:
image_hash = _normalize_required_text(row.get("emoji_hash"))
image_type = _deduce_image_type_name(row.get("type"))
dedupe_key = (full_path, image_hash, image_type)
if not full_path or dedupe_key in existing_keys:
continue
connection.execute(
insert_sql,
{
"image_hash": image_hash,
"description": _normalize_required_text(row.get("description")),
"full_path": full_path,
"image_type": image_type,
"emotion": None,
"query_count": _normalize_int(row.get("count"), default=0),
"is_registered": False,
"is_banned": False,
"no_file_flag": False,
"record_time": _coerce_datetime(row.get("timestamp"), fallback_now=True),
"register_time": None,
"last_used_time": None,
"vlm_processed": _normalize_bool(row.get("vlm_processed"), default=False),
},
)
existing_keys.add(dedupe_key)
migrated_count += 1
if full_path and dedupe_key not in existing_keys:
connection.execute(
insert_sql,
{
"image_hash": image_hash,
"description": _normalize_required_text(row.get("description")),
"full_path": full_path,
"image_type": image_type,
"emotion": None,
"query_count": _normalize_int(row.get("count"), default=0),
"is_registered": False,
"is_banned": False,
"no_file_flag": False,
"record_time": _coerce_datetime(row.get("timestamp"), fallback_now=True),
"register_time": None,
"last_used_time": None,
"vlm_processed": _normalize_bool(row.get("vlm_processed"), default=False),
},
)
existing_keys.add(dedupe_key)
migrated_count += 1
context.advance_progress(records=1)
_complete_table_progress(context, "images")
return migrated_count
def _migrate_messages(connection: Connection) -> int:
def _migrate_messages(context: MigrationExecutionContext) -> int:
"""迁移旧版 ``messages`` 到新版 ``mai_messages``。
Args:
connection: 当前数据库连接。
context: 当前迁移步骤执行上下文。
Returns:
int: 迁移成功的记录数。
"""
connection = context.connection
legacy_table = _load_legacy_table_data(connection, "messages")
if legacy_table is None:
_complete_table_progress(context, "mai_messages")
return 0
migrated_count = 0
@@ -840,62 +915,65 @@ def _migrate_messages(connection: Connection) -> int:
)
for row in legacy_table.rows:
session_id = _normalize_optional_text(row.get("chat_id")) or _normalize_optional_text(row.get("chat_info_stream_id"))
if not session_id:
continue
processed_plain_text = _normalize_optional_text(row.get("processed_plain_text"))
display_message = _normalize_optional_text(row.get("display_message"))
connection.execute(
insert_sql,
{
"id": row.get("id"),
"message_id": _normalize_required_text(row.get("message_id"), default=""),
"timestamp": _coerce_datetime(row.get("time"), fallback_now=True),
"platform": _normalize_required_text(
row.get("chat_info_platform") or row.get("user_platform"),
default="unknown",
),
"user_id": _normalize_required_text(
row.get("user_id") or row.get("chat_info_user_id"),
default="",
),
"user_nickname": _normalize_required_text(
row.get("user_nickname") or row.get("chat_info_user_nickname"),
default="",
),
"user_cardname": _normalize_optional_text(
row.get("user_cardname") or row.get("chat_info_user_cardname")
),
"group_id": _normalize_optional_text(row.get("chat_info_group_id")),
"group_name": _normalize_optional_text(row.get("chat_info_group_name")),
"is_mentioned": _normalize_bool(row.get("is_mentioned"), default=False),
"is_at": _normalize_bool(row.get("is_at"), default=False),
"session_id": session_id,
"reply_to": _normalize_optional_text(row.get("reply_to")),
"is_emoji": _normalize_bool(row.get("is_emoji"), default=False),
"is_picture": _normalize_bool(row.get("is_picid"), default=False),
"is_command": _normalize_bool(row.get("is_command"), default=False),
"is_notify": _normalize_bool(row.get("is_notify"), default=False),
"raw_content": _build_message_raw_content(processed_plain_text, display_message),
"processed_plain_text": processed_plain_text,
"display_message": display_message,
"additional_config": _build_legacy_message_additional_config(row),
},
)
migrated_count += 1
if session_id:
processed_plain_text = _normalize_optional_text(row.get("processed_plain_text"))
display_message = _normalize_optional_text(row.get("display_message"))
connection.execute(
insert_sql,
{
"id": row.get("id"),
"message_id": _normalize_required_text(row.get("message_id"), default=""),
"timestamp": _coerce_datetime(row.get("time"), fallback_now=True),
"platform": _normalize_required_text(
row.get("chat_info_platform") or row.get("user_platform"),
default="unknown",
),
"user_id": _normalize_required_text(
row.get("user_id") or row.get("chat_info_user_id"),
default="",
),
"user_nickname": _normalize_required_text(
row.get("user_nickname") or row.get("chat_info_user_nickname"),
default="",
),
"user_cardname": _normalize_optional_text(
row.get("user_cardname") or row.get("chat_info_user_cardname")
),
"group_id": _normalize_optional_text(row.get("chat_info_group_id")),
"group_name": _normalize_optional_text(row.get("chat_info_group_name")),
"is_mentioned": _normalize_bool(row.get("is_mentioned"), default=False),
"is_at": _normalize_bool(row.get("is_at"), default=False),
"session_id": session_id,
"reply_to": _normalize_optional_text(row.get("reply_to")),
"is_emoji": _normalize_bool(row.get("is_emoji"), default=False),
"is_picture": _normalize_bool(row.get("is_picid"), default=False),
"is_command": _normalize_bool(row.get("is_command"), default=False),
"is_notify": _normalize_bool(row.get("is_notify"), default=False),
"raw_content": _build_message_raw_content(processed_plain_text, display_message),
"processed_plain_text": processed_plain_text,
"display_message": display_message,
"additional_config": _build_legacy_message_additional_config(row),
},
)
migrated_count += 1
context.advance_progress(records=1)
_complete_table_progress(context, "mai_messages")
return migrated_count
def _migrate_action_records(connection: Connection) -> int:
def _migrate_action_records(context: MigrationExecutionContext) -> int:
"""迁移旧版 ``action_records`` 到新版 ``action_records``。
Args:
connection: 当前数据库连接。
context: 当前迁移步骤执行上下文。
Returns:
int: 迁移成功的记录数。
"""
connection = context.connection
legacy_table = _load_legacy_table_data(connection, "action_records")
if legacy_table is None:
_complete_table_progress(context, "action_records")
return 0
migrated_count = 0
@@ -926,37 +1004,40 @@ def _migrate_action_records(connection: Connection) -> int:
)
for row in legacy_table.rows:
session_id = _normalize_optional_text(row.get("chat_id")) or _normalize_optional_text(row.get("chat_info_stream_id"))
if not session_id:
continue
connection.execute(
insert_sql,
{
"id": row.get("id"),
"action_id": _normalize_required_text(row.get("action_id")),
"timestamp": _coerce_datetime(row.get("time"), fallback_now=True),
"session_id": session_id,
"action_name": _normalize_required_text(row.get("action_name"), default="unknown"),
"action_reasoning": _normalize_optional_text(row.get("action_reasoning")),
"action_data": _normalize_optional_text(row.get("action_data")),
"action_builtin_prompt": None,
"action_display_prompt": _normalize_optional_text(row.get("action_prompt_display")),
},
)
migrated_count += 1
if session_id:
connection.execute(
insert_sql,
{
"id": row.get("id"),
"action_id": _normalize_required_text(row.get("action_id")),
"timestamp": _coerce_datetime(row.get("time"), fallback_now=True),
"session_id": session_id,
"action_name": _normalize_required_text(row.get("action_name"), default="unknown"),
"action_reasoning": _normalize_optional_text(row.get("action_reasoning")),
"action_data": _normalize_optional_text(row.get("action_data")),
"action_builtin_prompt": None,
"action_display_prompt": _normalize_optional_text(row.get("action_prompt_display")),
},
)
migrated_count += 1
context.advance_progress(records=1)
_complete_table_progress(context, "action_records")
return migrated_count
def _migrate_tool_records(connection: Connection) -> int:
def _migrate_tool_records(context: MigrationExecutionContext) -> int:
"""迁移旧版 ``action_records`` 到新版 ``tool_records``。
Args:
connection: 当前数据库连接。
context: 当前迁移步骤执行上下文。
Returns:
int: 迁移成功的记录数。
"""
connection = context.connection
legacy_table = _load_legacy_table_data(connection, "action_records")
if legacy_table is None:
_complete_table_progress(context, "tool_records")
return 0
migrated_count = 0
@@ -987,37 +1068,40 @@ def _migrate_tool_records(connection: Connection) -> int:
)
for row in legacy_table.rows:
session_id = _normalize_optional_text(row.get("chat_id")) or _normalize_optional_text(row.get("chat_info_stream_id"))
if not session_id:
continue
connection.execute(
insert_sql,
{
"id": row.get("id"),
"tool_id": _normalize_required_text(row.get("action_id")),
"timestamp": _coerce_datetime(row.get("time"), fallback_now=True),
"session_id": session_id,
"tool_name": _normalize_required_text(row.get("action_name"), default="unknown"),
"tool_reasoning": _normalize_optional_text(row.get("action_reasoning")),
"tool_data": _normalize_optional_text(row.get("action_data")),
"tool_builtin_prompt": None,
"tool_display_prompt": _normalize_optional_text(row.get("action_prompt_display")),
},
)
migrated_count += 1
if session_id:
connection.execute(
insert_sql,
{
"id": row.get("id"),
"tool_id": _normalize_required_text(row.get("action_id")),
"timestamp": _coerce_datetime(row.get("time"), fallback_now=True),
"session_id": session_id,
"tool_name": _normalize_required_text(row.get("action_name"), default="unknown"),
"tool_reasoning": _normalize_optional_text(row.get("action_reasoning")),
"tool_data": _normalize_optional_text(row.get("action_data")),
"tool_builtin_prompt": None,
"tool_display_prompt": _normalize_optional_text(row.get("action_prompt_display")),
},
)
migrated_count += 1
context.advance_progress(records=1)
_complete_table_progress(context, "tool_records")
return migrated_count
def _migrate_online_time(connection: Connection) -> int:
def _migrate_online_time(context: MigrationExecutionContext) -> int:
"""迁移旧版 ``online_time`` 到新版 ``online_time``。
Args:
connection: 当前数据库连接。
context: 当前迁移步骤执行上下文。
Returns:
int: 迁移成功的记录数。
"""
connection = context.connection
legacy_table = _load_legacy_table_data(connection, "online_time")
if legacy_table is None:
_complete_table_progress(context, "online_time")
return 0
migrated_count = 0
@@ -1050,20 +1134,24 @@ def _migrate_online_time(connection: Connection) -> int:
},
)
migrated_count += 1
context.advance_progress(records=1)
_complete_table_progress(context, "online_time")
return migrated_count
def _migrate_person_info(connection: Connection) -> int:
def _migrate_person_info(context: MigrationExecutionContext) -> int:
"""迁移旧版 ``person_info`` 到新版 ``person_info``。
Args:
connection: 当前数据库连接。
context: 当前迁移步骤执行上下文。
Returns:
int: 迁移成功的记录数。
"""
connection = context.connection
legacy_table = _load_legacy_table_data(connection, "person_info")
if legacy_table is None:
_complete_table_progress(context, "person_info")
return 0
migrated_count = 0
@@ -1123,20 +1211,24 @@ def _migrate_person_info(connection: Connection) -> int:
},
)
migrated_count += 1
context.advance_progress(records=1)
_complete_table_progress(context, "person_info")
return migrated_count
def _migrate_expressions(connection: Connection) -> int:
def _migrate_expressions(context: MigrationExecutionContext) -> int:
"""迁移旧版 ``expression`` 到新版 ``expressions``。
Args:
connection: 当前数据库连接。
context: 当前迁移步骤执行上下文。
Returns:
int: 迁移成功的记录数。
"""
connection = context.connection
legacy_table = _load_legacy_table_data(connection, "expression")
if legacy_table is None:
_complete_table_progress(context, "expressions")
return 0
migrated_count = 0
@@ -1187,20 +1279,24 @@ def _migrate_expressions(connection: Connection) -> int:
},
)
migrated_count += 1
context.advance_progress(records=1)
_complete_table_progress(context, "expressions")
return migrated_count
def _migrate_jargons(connection: Connection) -> int:
def _migrate_jargons(context: MigrationExecutionContext) -> int:
"""迁移旧版 ``jargon`` 到新版 ``jargons``。
Args:
connection: 当前数据库连接。
context: 当前迁移步骤执行上下文。
Returns:
int: 迁移成功的记录数。
"""
connection = context.connection
legacy_table = _load_legacy_table_data(connection, "jargon")
if legacy_table is None:
_complete_table_progress(context, "jargons")
return 0
migrated_count = 0
@@ -1259,20 +1355,24 @@ def _migrate_jargons(connection: Connection) -> int:
},
)
migrated_count += 1
context.advance_progress(records=1)
_complete_table_progress(context, "jargons")
return migrated_count
def _migrate_chat_history(connection: Connection) -> int:
def _migrate_chat_history(context: MigrationExecutionContext) -> int:
"""迁移旧版 ``chat_history`` 到新版 ``chat_history``。
Args:
connection: 当前数据库连接。
context: 当前迁移步骤执行上下文。
Returns:
int: 迁移成功的记录数。
"""
connection = context.connection
legacy_table = _load_legacy_table_data(connection, "chat_history")
if legacy_table is None:
_complete_table_progress(context, "chat_history")
return 0
migrated_count = 0
@@ -1307,39 +1407,42 @@ def _migrate_chat_history(connection: Connection) -> int:
)
for row in legacy_table.rows:
session_id = _normalize_required_text(row.get("chat_id"))
if not session_id:
continue
connection.execute(
insert_sql,
{
"id": row.get("id"),
"session_id": session_id,
"start_timestamp": _coerce_datetime(row.get("start_time"), fallback_now=True),
"end_timestamp": _coerce_datetime(row.get("end_time"), fallback_now=True),
"query_count": _normalize_int(row.get("count"), default=0),
"query_forget_count": _normalize_int(row.get("forget_times"), default=0),
"original_messages": _normalize_required_text(row.get("original_text")),
"participants": _normalize_required_text(row.get("participants"), default="[]"),
"theme": _normalize_required_text(row.get("theme"), default=""),
"keywords": _normalize_required_text(row.get("keywords"), default="[]"),
"summary": _normalize_required_text(row.get("summary"), default=""),
},
)
migrated_count += 1
if session_id:
connection.execute(
insert_sql,
{
"id": row.get("id"),
"session_id": session_id,
"start_timestamp": _coerce_datetime(row.get("start_time"), fallback_now=True),
"end_timestamp": _coerce_datetime(row.get("end_time"), fallback_now=True),
"query_count": _normalize_int(row.get("count"), default=0),
"query_forget_count": _normalize_int(row.get("forget_times"), default=0),
"original_messages": _normalize_required_text(row.get("original_text")),
"participants": _normalize_required_text(row.get("participants"), default="[]"),
"theme": _normalize_required_text(row.get("theme"), default=""),
"keywords": _normalize_required_text(row.get("keywords"), default="[]"),
"summary": _normalize_required_text(row.get("summary"), default=""),
},
)
migrated_count += 1
context.advance_progress(records=1)
_complete_table_progress(context, "chat_history")
return migrated_count
def _migrate_thinking_questions(connection: Connection) -> int:
def _migrate_thinking_questions(context: MigrationExecutionContext) -> int:
"""迁移旧版 ``thinking_back`` 到新版 ``thinking_questions``。
Args:
connection: 当前数据库连接。
context: 当前迁移步骤执行上下文。
Returns:
int: 迁移成功的记录数。
"""
connection = context.connection
legacy_table = _load_legacy_table_data(connection, "thinking_back")
if legacy_table is None:
_complete_table_progress(context, "thinking_questions")
return 0
migrated_count = 0
@@ -1381,4 +1484,6 @@ def _migrate_thinking_questions(connection: Connection) -> int:
},
)
migrated_count += 1
context.advance_progress(records=1)
_complete_table_progress(context, "thinking_questions")
return migrated_count