Update api/app.py

api/app.py

@@ -1,41 +1,50 @@
-from fastapi import FastAPI, Request
-from starlette.middleware.cors import CORSMiddleware
-from fastapi.responses import JSONResponse
-from api.logger import setup_logger
-from api.routes import router
-
-
-logger = setup_logger(__name__)
-
-def create_app():
-    app = FastAPI(
-        title="NiansuhAI API Gateway",
-        docs_url=None,          # Disable Swagger UI
-        redoc_url=None,         # Disable ReDoc
-        openapi_url=None,       # Disable OpenAPI schema
-    )
-
-    # CORS settings
-    app.add_middleware(
-        CORSMiddleware,
-        allow_origins=["*"],                   # Adjust as needed for security
-        allow_credentials=True,
-        allow_methods=["*"],
-        allow_headers=["*"],
-    )
-
-    # Include routes
-    app.include_router(router)
-
-    # Global exception handler for better error reporting
-    @app.exception_handler(Exception)
-    async def global_exception_handler(request: Request, exc: Exception):
-        logger.error(f"An error occurred: {str(exc)}")
-        return JSONResponse(
-            status_code=500,
-            content={"message": "An internal server error occurred."},
-        )
-
-    return app
-
-app = create_app()
+from fastapi import FastAPI, Depends, HTTPException, Request, Response
+from fastapi.responses import JSONResponse, StreamingResponse
+from api.auth import verify_app_secret
+from api.config import ALLOWED_MODELS
+from api.models import ChatRequest
+from api.utils import process_response
+from api.logger import setup_logger
+
+logger = setup_logger(__name__)
+
+app = FastAPI(
+    title="Your API",
+    docs_url=None,          # Disable Swagger UI
+    redoc_url=None,         # Disable ReDoc
+    openapi_url=None,       # Disable OpenAPI schema
+)
+
+# Include your authentication if needed
+# from api.auth import verify_app_secret
+
+@app.post("/v1/chat/completions")
+async def chat_completions(
+    request: ChatRequest, app_secret: str = Depends(verify_app_secret)
+):
+    logger.info("Processing chat completion request")
+    if request.model not in [model["id"] for model in ALLOWED_MODELS]:
+        raise HTTPException(
+            status_code=400,
+            detail=f"Model {request.model} is not allowed."
+        )
+
+    # Process the response using Editee API
+    if request.stream:
+        # If streaming is required
+        generator = process_response(request, stream=True)
+        return StreamingResponse(generator, media_type="text/event-stream")
+    else:
+        # Non-streaming response
+        response_data = await process_response(request)
+        return JSONResponse(content=response_data)
+
+# Health check endpoints
+@app.get("/")
+@app.get("/healthz")
+@app.get("/ready")
+@app.get("/alive")
+@app.get("/status")
+@app.get("/health")
+def health_check(request: Request):
+    return Response(content='{"status": "ok"}', media_type="application/json")