refactor(stage-1): eliminate dead parse_text, simplify task progress SQL, remove prompt hacks
This commit is contained in:
+18
-48
@@ -49,54 +49,24 @@ def update_task_progress(
|
||||
"""
|
||||
Updates the progress and status of a running task.
|
||||
"""
|
||||
if total is not None and message is not None:
|
||||
conn.execute(
|
||||
"""
|
||||
UPDATE background_tasks
|
||||
SET status = 'running',
|
||||
progress_current = ?,
|
||||
progress_total = ?,
|
||||
message = ?,
|
||||
updated_at = CURRENT_TIMESTAMP
|
||||
WHERE id = ? AND status != 'cancelled'
|
||||
""",
|
||||
(current, total, message, task_id),
|
||||
)
|
||||
elif total is not None:
|
||||
conn.execute(
|
||||
"""
|
||||
UPDATE background_tasks
|
||||
SET status = 'running',
|
||||
progress_current = ?,
|
||||
progress_total = ?,
|
||||
updated_at = CURRENT_TIMESTAMP
|
||||
WHERE id = ? AND status != 'cancelled'
|
||||
""",
|
||||
(current, total, task_id),
|
||||
)
|
||||
elif message is not None:
|
||||
conn.execute(
|
||||
"""
|
||||
UPDATE background_tasks
|
||||
SET status = 'running',
|
||||
progress_current = ?,
|
||||
message = ?,
|
||||
updated_at = CURRENT_TIMESTAMP
|
||||
WHERE id = ? AND status != 'cancelled'
|
||||
""",
|
||||
(current, message, task_id),
|
||||
)
|
||||
else:
|
||||
conn.execute(
|
||||
"""
|
||||
UPDATE background_tasks
|
||||
SET status = 'running',
|
||||
progress_current = ?,
|
||||
updated_at = CURRENT_TIMESTAMP
|
||||
WHERE id = ? AND status != 'cancelled'
|
||||
""",
|
||||
(current, task_id),
|
||||
)
|
||||
set_clauses = ["status = 'running'", "progress_current = ?", "updated_at = CURRENT_TIMESTAMP"]
|
||||
params: List[Any] = [current]
|
||||
|
||||
if total is not None:
|
||||
set_clauses.append("progress_total = ?")
|
||||
params.append(total)
|
||||
|
||||
if message is not None:
|
||||
set_clauses.append("message = ?")
|
||||
params.append(message)
|
||||
|
||||
params.append(task_id)
|
||||
sql = f"""
|
||||
UPDATE background_tasks
|
||||
SET {', '.join(set_clauses)}
|
||||
WHERE id = ? AND status != 'cancelled'
|
||||
"""
|
||||
conn.execute(sql, params)
|
||||
|
||||
|
||||
def complete_task(
|
||||
|
||||
Reference in New Issue
Block a user