--- a/runner/core.py +++ b/runner/core.py @@ -107,7 +107,7 @@ require(not event.get("error"), "Antigravity provider error") step = event.get("step_update") require(isinstance(step, dict), "Antigravity malformed step") - require(step.get("step_type") in ("tool", "agent_response", "checkpoint", "finish") + require(step.get("step_type") in ("user_input", "tool", "agent_response", "checkpoint", "finish") and step.get("state") in ("ACTIVE", "DONE", "ERROR"), "Antigravity unknown step") require(not step.get("error"), "Antigravity provider error") info = step.get("tool_info", {})