mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-26 05:48:36 +08:00
87 lines
2.9 KiB
Python
87 lines
2.9 KiB
Python
import contextlib
|
|
import warnings
|
|
from contextlib import asynccontextmanager
|
|
from typing import AsyncIterator
|
|
|
|
from fastapi.routing import Mount
|
|
|
|
with warnings.catch_warnings():
|
|
warnings.simplefilter('ignore')
|
|
|
|
from fastapi import (
|
|
FastAPI,
|
|
Request,
|
|
)
|
|
from fastapi.responses import JSONResponse
|
|
|
|
import openhands.agenthub # noqa F401 (we import this to get the agents registered)
|
|
from openhands import __version__
|
|
from openhands.integrations.service_types import AuthenticationError
|
|
from openhands.server.routes.conversation import app as conversation_api_router
|
|
from openhands.server.routes.feedback import app as feedback_api_router
|
|
from openhands.server.routes.files import app as files_api_router
|
|
from openhands.server.routes.git import app as git_api_router
|
|
from openhands.server.routes.health import add_health_endpoints
|
|
from openhands.server.routes.manage_conversations import (
|
|
app as manage_conversation_api_router,
|
|
)
|
|
from openhands.server.routes.mcp import mcp_server
|
|
from openhands.server.routes.public import app as public_api_router
|
|
from openhands.server.routes.secrets import app as secrets_router
|
|
from openhands.server.routes.security import app as security_api_router
|
|
from openhands.server.routes.settings import app as settings_router
|
|
from openhands.server.routes.trajectory import app as trajectory_router
|
|
from openhands.server.shared import conversation_manager, server_config
|
|
from openhands.server.types import AppMode
|
|
|
|
mcp_app = mcp_server.http_app(path='/mcp')
|
|
|
|
|
|
def combine_lifespans(*lifespans):
|
|
# Create a combined lifespan to manage multiple session managers
|
|
@contextlib.asynccontextmanager
|
|
async def combined_lifespan(app):
|
|
async with contextlib.AsyncExitStack() as stack:
|
|
for lifespan in lifespans:
|
|
await stack.enter_async_context(lifespan(app))
|
|
yield
|
|
|
|
return combined_lifespan
|
|
|
|
|
|
@asynccontextmanager
|
|
async def _lifespan(app: FastAPI) -> AsyncIterator[None]:
|
|
async with conversation_manager:
|
|
yield
|
|
|
|
|
|
app = FastAPI(
|
|
title='OpenHands',
|
|
description='OpenHands: Code Less, Make More',
|
|
version=__version__,
|
|
lifespan=combine_lifespans(_lifespan, mcp_app.lifespan),
|
|
routes=[Mount(path='/mcp', app=mcp_app)],
|
|
)
|
|
|
|
|
|
@app.exception_handler(AuthenticationError)
|
|
async def authentication_error_handler(request: Request, exc: AuthenticationError):
|
|
return JSONResponse(
|
|
status_code=401,
|
|
content=str(exc),
|
|
)
|
|
|
|
|
|
app.include_router(public_api_router)
|
|
app.include_router(files_api_router)
|
|
app.include_router(security_api_router)
|
|
app.include_router(feedback_api_router)
|
|
app.include_router(conversation_api_router)
|
|
app.include_router(manage_conversation_api_router)
|
|
app.include_router(settings_router)
|
|
app.include_router(secrets_router)
|
|
if server_config.app_mode == AppMode.OSS:
|
|
app.include_router(git_api_router)
|
|
app.include_router(trajectory_router)
|
|
add_health_endpoints(app)
|