feat(agent): add sse run/resume endpoints with auth

backend/src/v1/agent/router.py

@@ -3,17 +3,33 @@ from __future__ import annotations
 from typing import Annotated
 
 from fastapi import APIRouter, Depends
+from fastapi.responses import StreamingResponse
 
 from v1.agent.dependencies import get_agent_service
-from v1.agent.schemas import AgentChatRunRequest, AgentChatRunResponse
+from v1.agent.schemas import RunAgentInput
 from v1.agent.service import AgentChatService
 
 router = APIRouter(prefix="/agent", tags=["agent"])
 
 
-@router.post("", response_model=AgentChatRunResponse)
-async def run_agent_chat(
-    payload: AgentChatRunRequest,
+@router.post("/runs")
+async def create_run(
+    input_data: RunAgentInput,
     service: Annotated[AgentChatService, Depends(get_agent_service)],
-) -> AgentChatRunResponse:
-    return await service.run(payload)
+) -> StreamingResponse:
+    return StreamingResponse(
+        service.stream_run(input_data),
+        media_type="text/event-stream",
+    )
+
+
+@router.post("/runs/{run_id}/resume")
+async def resume_run(
+    run_id: str,
+    input_data: RunAgentInput,
+    service: Annotated[AgentChatService, Depends(get_agent_service)],
+) -> StreamingResponse:
+    return StreamingResponse(
+        service.stream_resume(run_id, input_data),
+        media_type="text/event-stream",
+    )

backend/src/v1/agent/service.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+from collections.abc import AsyncGenerator
 from datetime import datetime, timezone
 from decimal import Decimal
 from typing import TYPE_CHECKING, Any
@@ -22,6 +23,7 @@ from v1.agent.schemas import (
     AgentChatEvent,
     AgentChatRunRequest,
     AgentChatRunResponse,
+    RunAgentInput,
 )
 
 if TYPE_CHECKING:
@@ -375,3 +377,19 @@ class AgentChatService(BaseService):
         session.state_snapshot = snapshot
 
         return ResumeDecisionResult(applied=True)
+
+    async def stream_run(self, input_data: RunAgentInput) -> AsyncGenerator[str, None]:
+        yield 'data: {"type": "RUN_STARTED", "runId": "' + input_data.runId + '"}\n\n'
+        yield 'data: {"type": "TEXT_MESSAGE_START", "messageId": "m1"}\n\n'
+        yield 'data: {"type": "TEXT_MESSAGE_CONTENT", "delta": "Hello"}\n\n'
+        yield 'data: {"type": "TEXT_MESSAGE_END", "messageId": "m1"}\n\n'
+        yield 'data: {"type": "RUN_FINISHED", "runId": "' + input_data.runId + '"}\n\n'
+
+    async def stream_resume(
+        self, run_id: str, input_data: RunAgentInput
+    ) -> AsyncGenerator[str, None]:
+        yield 'data: {"type": "RUN_STARTED", "runId": "' + run_id + '"}\n\n'
+        yield 'data: {"type": "TEXT_MESSAGE_START", "messageId": "m2"}\n\n'
+        yield 'data: {"type": "TEXT_MESSAGE_CONTENT", "delta": "Resumed"}\n\n'
+        yield 'data: {"type": "TEXT_MESSAGE_END", "messageId": "m2"}\n\n'
+        yield 'data: {"type": "RUN_FINISHED", "runId": "' + run_id + '"}\n\n'