Added home.py file for /chat route

This commit is contained in:
Saurab-Shrestha 2024-01-25 17:37:00 +05:45
parent bcc3f03e25
commit 55565fd3a7
20 changed files with 295 additions and 179 deletions

View file

@ -14,7 +14,7 @@ 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 home_router
logger = logging.getLogger(__name__)
@ -25,28 +25,25 @@ def create_app(root_injector: Injector) -> FastAPI:
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(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(home_router)
settings = root_injector.get(Settings)
if settings.server.cors.enabled:
logger.debug("Setting up CORS middleware")
app.add_middleware(
CORSMiddleware,
allow_credentials=settings.server.cors.allow_credentials,
allow_origins=settings.server.cors.allow_origins,
allow_origin_regex=settings.server.cors.allow_origin_regex,
allow_methods=settings.server.cors.allow_methods,
allow_headers=settings.server.cors.allow_headers,
)
# if settings.server.cors.enabled/:
logger.debug("Setting up CORS middleware")
app.add_middleware(
CORSMiddleware,
allow_credentials=True,
allow_origins=["http://localhost:5173", "http://localhost:8001"],
allow_methods=["DELETE", "GET", "POST", "PUT", "OPTIONS"],
allow_headers=["*"],
)
# if settings.ui.enabled:
# logger.debug("Importing the UI module")