mirror of
https://github.com/zylon-ai/private-gpt.git
synced 2025-12-22 23:22:57 +01:00
56 lines
No EOL
2.3 KiB
Python
56 lines
No EOL
2.3 KiB
Python
"""FastAPI app creation, logger configuration and main API routes."""
|
|
import logging
|
|
|
|
from fastapi import Depends, FastAPI, Request
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from injector import Injector
|
|
|
|
from private_gpt.server.chat.chat_router import chat_router
|
|
from private_gpt.server.chunks.chunks_router import chunks_router
|
|
from private_gpt.server.completions.completions_router import completions_router
|
|
from private_gpt.server.embeddings.embeddings_router import embeddings_router
|
|
from private_gpt.server.health.health_router import health_router
|
|
from private_gpt.server.ingest.ingest_router import ingest_router
|
|
from private_gpt.users.api.v1.api import api_router
|
|
|
|
from private_gpt.settings.settings import Settings
|
|
from private_gpt.home import chat_router
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def create_app(root_injector: Injector) -> FastAPI:
|
|
|
|
# Start the API
|
|
async def bind_injector_to_request(request: Request) -> None:
|
|
request.state.injector = root_injector
|
|
|
|
app = FastAPI(dependencies=[Depends(bind_injector_to_request)])
|
|
app.include_router(completions_router)
|
|
app.include_router(chat_router)
|
|
app.include_router(chunks_router)
|
|
app.include_router(ingest_router)
|
|
app.include_router(embeddings_router)
|
|
app.include_router(health_router)
|
|
|
|
app.include_router(api_router)
|
|
app.include_router(chat_router)
|
|
settings = root_injector.get(Settings)
|
|
if settings.server.cors.enabled:
|
|
logger.debug("Setting up CORS middleware")
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_credentials=True,
|
|
allow_origins=["http://localhost:80/", "http://10.1.101.125:80", "http://quickgpt.gibl.com.np:80",
|
|
"http://10.1.101.125", "http://quickgpt.gibl.com.np", "http://localhost:8001", "http://192.168.1.93", "http://192.168.1.93:88", "http://192.168.1.98", "http://192.168.1.98:5173", "http://localhost:5173"],
|
|
allow_methods=["DELETE", "GET", "POST", "PUT", "OPTIONS", "PATCH"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
# if settings.ui.enabled:
|
|
# logger.debug("Importing the UI module")
|
|
# from private_gpt.ui.admin_ui import PrivateAdminGptUi
|
|
# admin_ui = root_injector.get(PrivateAdminGptUi)
|
|
# admin_ui.mount_in_admin_app(app, '/admin')
|
|
|
|
|
|
return app |