mirror of
https://github.com/Sea-Haven-Industries/open-swe.git
synced 2026-09-30 11:33:14 +00:00
36 lines
1.2 KiB
Python
36 lines
1.2 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
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
class SchedulerState(TypedDict, total=False):
|
||
|
|
schedule_id: str
|
||
|
|
result: dict[str, Any]
|
||
|
|
|
||
|
|
|
||
|
|
async def _launch(state: SchedulerState, config: RunnableConfig) -> dict[str, Any]:
|
||
|
|
configurable = config.get("configurable") or {}
|
||
|
|
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 {})
|