DemonGatanjieuandClaude Opus 5.5
049b265126
- A stop button on a running run (timeline, and beside the run-status button), and stop_run for the AI: only this run's prompts leave the queue or are interrupted; the step ends "cancelled" and a user's stop reaches the AI as a "stop" message (ai_run_control.py). - The run's progress carries ComfyUI's console since it began (the bar's note, models loading), time left and per step, and VRAM; once queued, each sampler's working size is estimated from the prompt, with warnings for an upscaled picture sampled again, a size past the card and a huge decode, for the user and in the AI's run news (ai_run_watch.py, ui_ai_run_live.js). - The materials picker opens as a side panel that leaves the page usable (the full-screen browser makes room), expands to the large window and back, and remembers the choice. - Esc closes the run-status card only (data-anomalous-escape). Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
56 lines
2.7 KiB
Python
56 lines
2.7 KiB
Python
"""Stopping a run an AI started (the workbench's stop button, the AI's stop_run): its prompts still
|
|
waiting leave ComfyUI's queue and the one running is interrupted, only while it is still that one
|
|
(ComfyUI's own cancel; never anyone else's prompt). The run's watcher (ai_task_runs.py) then ends
|
|
the step as "cancelled" with who stopped it; when the user did, the AI is told with its next answer."""
|
|
|
|
import time
|
|
import uuid
|
|
|
|
from . import ai_task_runs, ai_tasks
|
|
|
|
|
|
class RunControlError(ValueError):
|
|
pass
|
|
|
|
|
|
def _interrupt(queue, prompt_id):
|
|
"""Interrupts `prompt_id` if it is the prompt running now; True when it was."""
|
|
if hasattr(queue, "interrupt_if_running"):
|
|
return bool(queue.interrupt_if_running(prompt_id))
|
|
running, _ = queue.get_current_queue() # an older ComfyUI: check, then the global interrupt
|
|
if not any(item[1] == prompt_id for item in running):
|
|
return False
|
|
import nodes
|
|
nodes.interrupt_processing()
|
|
return True
|
|
|
|
|
|
def stop(step_id, by="user", task_id=None, now=None):
|
|
"""Stops run step `step_id` (in task `task_id` when given). Returns {"task", "removed", "interrupted"}."""
|
|
now = time.time() if now is None else now
|
|
with ai_tasks._lock:
|
|
tasks = ai_tasks._load()
|
|
task = next((item for item in tasks.values() if (not task_id or item["id"] == task_id)
|
|
and any(step["id"] == step_id for step in item.get("steps") or [])), None)
|
|
step = next((item for item in (task or {}).get("steps", []) if item["id"] == step_id), None)
|
|
if not step or step.get("tool") != "run_workflow":
|
|
raise RunControlError("No such run in this task")
|
|
if step.get("state") != "running" or not step.get("prompt_ids"):
|
|
raise RunControlError("This run is not going any more")
|
|
prompts = list(step["prompt_ids"])
|
|
step["stop"] = by
|
|
if by == "user":
|
|
task["steps"].append({
|
|
"id": uuid.uuid4().hex[:10], "at": now, "ms": 0, "tool": "user_stop", "kind": "user", "ok": True,
|
|
"state": "done", "summary": "user_stop", "stopped": step_id, "delivered": False, "results": [],
|
|
"text": "The user stopped your run in Anomalous. Do not start it again unless they ask; ask_user what to change.",
|
|
})
|
|
del task["steps"][:-ai_tasks.MAX_STEPS]
|
|
task["updated"] = max(task.get("updated", 0), now)
|
|
ai_tasks._save(task)
|
|
queue = ai_task_runs._queue()
|
|
removed = sum(1 for prompt_id in prompts if queue.delete_queue_item(lambda item, pid=prompt_id: item[1] == pid))
|
|
interrupted = sum(1 for prompt_id in prompts if _interrupt(queue, prompt_id))
|
|
ai_tasks._notify(task["id"])
|
|
return {"task": task["id"], "removed": removed, "interrupted": interrupted}
|