|
- # /api/runs routes
- from typing import Dict
- from uuid import UUID
-
- from fastapi import APIRouter, Body, Depends, HTTPException
- from pydantic import BaseModel
-
- from ...datamodel import Message, MessageConfig, Run, RunStatus, Session, Team
- from ..deps import get_db, get_team_manager, get_websocket_manager
-
- router = APIRouter()
-
-
- class CreateRunRequest(BaseModel):
- session_id: int
- user_id: str
-
-
- @router.post("/")
- async def create_run(
- request: CreateRunRequest,
- db=Depends(get_db),
- ) -> Dict:
- """Create a new run with initial state"""
- session_response = db.get(
- Session, filters={"id": request.session_id, "user_id": request.user_id}, return_json=False
- )
- if not session_response.status or not session_response.data:
- raise HTTPException(status_code=404, detail="Session not found")
-
- try:
- # Create run with default state
- run = db.upsert(
- Run(
- session_id=request.session_id,
- status=RunStatus.CREATED,
- task=None, # Will be set when run starts
- team_result=None,
- ),
- return_json=False,
- )
- return {"status": run.status, "data": {"run_id": str(run.data.id)}}
- except Exception as e:
- raise HTTPException(status_code=500, detail=str(e)) from e
-
-
- # We might want to add these endpoints:
-
-
- @router.get("/{run_id}")
- async def get_run(run_id: UUID, db=Depends(get_db)) -> Dict:
- """Get run details including task and result"""
- run = db.get(Run, filters={"id": run_id}, return_json=False)
- if not run.status or not run.data:
- raise HTTPException(status_code=404, detail="Run not found")
-
- return {"status": True, "data": run.data[0]}
-
-
- @router.get("/{run_id}/messages")
- async def get_run_messages(run_id: UUID, db=Depends(get_db)) -> Dict:
- """Get all messages for a run"""
- messages = db.get(Message, filters={"run_id": run_id}, order="created_at asc", return_json=False)
-
- return {"status": True, "data": messages.data}
|
