77 lines
2.5 KiB
Python
77 lines
2.5 KiB
Python
from contextlib import asynccontextmanager
|
|
|
|
from fastapi import FastAPI
|
|
from fastapi.exceptions import RequestValidationError
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from starlette.exceptions import HTTPException as StarletteHttpException
|
|
|
|
from app.config import get_settings
|
|
from app.container import container
|
|
from app.errors import ApiError, api_error_handler, http_error_handler, validation_error_handler
|
|
from app.routes import router as api_router
|
|
from app.media_routes import router as media_router
|
|
from app.local_model_routes import router as local_model_router
|
|
from app.usage_routes import router as usage_router
|
|
from app.provider_preview_routes import router as provider_preview_router
|
|
from app.schemas import HealthResponse, ServiceStatusResponse
|
|
|
|
settings = get_settings()
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(_: FastAPI):
|
|
from app.services import transcription_service
|
|
transcription_service.recover_interrupted()
|
|
try:
|
|
yield
|
|
finally:
|
|
from app.services import index_service
|
|
await index_service.shutdown()
|
|
await transcription_service.shutdown()
|
|
from app.local_models import components
|
|
await components.shutdown()
|
|
from app.local_models import manager
|
|
for _, key in list(manager._downloads):
|
|
await manager.cancel_download(key)
|
|
container.plugins.shutdown()
|
|
container.mcp_servers.shutdown()
|
|
|
|
|
|
app = FastAPI(
|
|
title=settings.name,
|
|
version=settings.version,
|
|
description="AI 笔记软件的本地 AI Core 与 Agent Core 服务。",
|
|
lifespan=lifespan,
|
|
)
|
|
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["http://127.0.0.1:5173", "http://localhost:5173"],
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
app.add_exception_handler(ApiError, api_error_handler)
|
|
app.add_exception_handler(RequestValidationError, validation_error_handler)
|
|
app.add_exception_handler(StarletteHttpException, http_error_handler)
|
|
app.include_router(api_router)
|
|
app.include_router(media_router)
|
|
app.include_router(local_model_router)
|
|
app.include_router(usage_router)
|
|
app.include_router(provider_preview_router)
|
|
|
|
|
|
@app.get("/health", response_model=HealthResponse, tags=["System"])
|
|
async def health() -> HealthResponse:
|
|
return HealthResponse()
|
|
|
|
|
|
@app.get("/api/status", response_model=ServiceStatusResponse, tags=["System"])
|
|
async def service_status() -> ServiceStatusResponse:
|
|
return ServiceStatusResponse(
|
|
name=settings.name,
|
|
version=settings.version,
|
|
environment=settings.environment,
|
|
)
|