2b4d182b24
모듈 레벨 변수를 직접 import하면 init_queue() 이후에도 None 참조가 유지됨. 모듈 자체를 import하여 접근. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
74 lines
2.0 KiB
Python
74 lines
2.0 KiB
Python
"""NanoClaude — 비동기 job 기반 AI Gateway (Phase 2: EXAONE → Gemma 파이프라인)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
from contextlib import asynccontextmanager
|
|
|
|
from fastapi import FastAPI, Request
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from fastapi.responses import JSONResponse
|
|
|
|
from config import settings
|
|
from db.database import init_db
|
|
from routers import chat
|
|
from services.backend_registry import backend_registry
|
|
from services import job_queue as jq_module
|
|
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format="%(asctime)s %(levelname)s %(name)s — %(message)s",
|
|
)
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
await init_db()
|
|
backend_registry.init_from_settings(settings)
|
|
backend_registry.start_health_loop(settings.health_check_interval)
|
|
jq_module.init_queue(settings.max_concurrent_jobs)
|
|
yield
|
|
backend_registry.stop_health_loop()
|
|
|
|
|
|
app = FastAPI(
|
|
title="NanoClaude",
|
|
version="0.2.0",
|
|
description="비동기 job 기반 AI Gateway — Phase 2 (EXAONE → Gemma 파이프라인)",
|
|
lifespan=lifespan,
|
|
)
|
|
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["*"],
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
|
|
@app.middleware("http")
|
|
async def check_api_key(request: Request, call_next):
|
|
"""Optional API key check. 설정이 비어있으면 통과."""
|
|
if settings.api_key:
|
|
auth = request.headers.get("Authorization", "")
|
|
if request.url.path not in ("/", "/health") and auth != f"Bearer {settings.api_key}":
|
|
return JSONResponse(status_code=401, content={"detail": "Invalid API key"})
|
|
return await call_next(request)
|
|
|
|
|
|
app.include_router(chat.router)
|
|
|
|
|
|
@app.get("/")
|
|
async def root():
|
|
return {"service": "NanoClaude", "version": "0.2.0", "phase": 2}
|
|
|
|
|
|
@app.get("/health")
|
|
async def health():
|
|
return {
|
|
"status": "ok",
|
|
"backends": backend_registry.health_summary(),
|
|
"queue": jq_module.job_queue.stats if jq_module.job_queue else {},
|
|
}
|