open-swe/agent/scheduler.py
amoussa1229 0cac1ad363 Adopt upstream modular webhook skeleton (#1621)
Apply the durable-interrupt-dispatch refactor: split the monolithic
webapp.py into a thin routing layer plus per-source handlers in
webhooks/{github,slack,linear}.py, and add completion.py, dispatch.py,
and reconcile.py. Reconcile fork divergence by keeping the Bedrock/
Fireworks cross-provider fallback, the no-agent-attribution prompt
policy, the dashboard-handoff re-export, and the Slack channel-info
cache. ci_autofix is restored on the new dispatch model in a later
commit.

Refs: #80
2026-06-30 21:04:51 +00:00

40 lines
1.4 KiB
Python

"""LangGraph entrypoint that fans cron ticks into fresh agent threads."""
from __future__ import annotations
import logging
from typing import Any, TypedDict
from langgraph.graph import END, START, StateGraph
from langgraph.graph.state import RunnableConfig
from .dashboard.schedules import launch_scheduled_agent_run
from .reconcile import reconcile_stale_runs
logger = logging.getLogger(__name__)
class SchedulerState(TypedDict, total=False):
schedule_id: str
task: str
result: dict[str, Any]
async def _launch(state: SchedulerState, config: RunnableConfig) -> dict[str, Any]:
configurable = config.get("configurable") or {}
task = state.get("task") or configurable.get("task")
if task == "reconcile":
return {"result": await reconcile_stale_runs()}
schedule_id = state.get("schedule_id") or configurable.get("schedule_id")
if not isinstance(schedule_id, str) or not schedule_id:
logger.warning("Scheduled agent tick missing schedule_id")
return {"result": {"status": "missing_schedule_id"}}
return {"result": await launch_scheduled_agent_run(schedule_id)}
def get_scheduler(config: RunnableConfig | None = None):
builder = StateGraph(SchedulerState)
builder.add_node("launch", _launch)
builder.add_edge(START, "launch")
builder.add_edge("launch", END)
return builder.compile().with_config(config or {})