First commit

workspace/api.py

@@ -0,0 +1,65 @@
+import logging
+from contextlib import asynccontextmanager
+
+from fastapi import Depends, FastAPI, HTTPException
+from fastapi.staticfiles import StaticFiles
+from routers.costs_router import router as costs_router
+from services.api_key_service import load_api_keys_from_file
+from utils.checking_keys import get_admin_key, get_api_key
+from utils.redis_utils import get_redis_client
+
+logging.basicConfig(
+    level=logging.INFO, format="%(asctime)s [%(levelname)s] %(name)s - %(message)s"
+)
+
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    # 애플리케이션 시작 시 파일에서 API 키 로드
+    print("Loading API keys from file...")
+    load_api_keys_from_file()
+    yield
+
+
+app = FastAPI(
+    title="LLM GATEWAY",
+    description="LLM 모델이 업로드된 문서를 분석하여 구조화된 JSON으로 변환하는 API 서비스입니다.",
+    docs_url="/docs",
+    lifespan=lifespan,
+)
+
+
+# API 키 검증을 위한 의존성 설정
+api_key_dependency = Depends(get_api_key)
+admin_key_dependency = Depends(get_admin_key)
+
+
+# 커스텀 라벨 콜백 함수
+def custom_labels(info):
+    # info.request 는 Starlette의 Request 객체
+    return {"job_id": info.request.headers.get("X-Job-ID", "unknown")}
+
+
+app.mount(
+    "/static", StaticFiles(directory="/workspace/workspace/static"), name="static"
+)
+
+app.include_router(costs_router, dependencies=[api_key_dependency])  # ✅ 비용 계산 API
+
+
+@app.get("/health/API")
+async def health_check():
+    """애플리케이션 상태 확인"""
+    return {"status": "API ok"}
+
+
+@app.get("/health/Redis")
+def redis_health_check():
+    client = get_redis_client()
+    if client is None:
+        raise HTTPException(status_code=500, detail="Redis connection failed")
+    try:
+        client.ping()
+        return {"status": "Redis ok"}
+    except Exception:
+        raise HTTPException(status_code=500, detail="Redis ping failed")