from server.http import Starlette, Route, Request, Response, JSONResponse, WebSocketRoute
from server.inference import list_conversations, connect_to_conversation

async def status(request: Request) -> Response:
  return JSONResponse({"status": "ok"})

app = Starlette(debug=True, routes=[
  Route("/api/", status),
  Route("/api/conversation", list_conversations, methods=["GET"]),
  WebSocketRoute("/api/conversation/:conversation/connect", connect_to_conversation)
])