101 lines
1.9 KiB
Python
101 lines
1.9 KiB
Python
from __future__ import annotations
|
|
|
|
import asyncio
|
|
import logging
|
|
import sys
|
|
from contextlib import asynccontextmanager
|
|
from pathlib import Path
|
|
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
|
|
|
|
PROJECT_ROOT = Path(__file__).resolve().parents[1]
|
|
|
|
if str(PROJECT_ROOT) not in sys.path:
|
|
sys.path.insert(
|
|
0,
|
|
str(PROJECT_ROOT),
|
|
)
|
|
|
|
|
|
from app.routes import router
|
|
from app.security import (
|
|
validate_security_configuration,
|
|
)
|
|
from scripts.embedding_utils import embed_query
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
OPENWEBUI_ORIGIN = (
|
|
"https://ui.tukekemt.xyz"
|
|
)
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(
|
|
_: FastAPI,
|
|
):
|
|
# Bez správne nakonfigurovanej security
|
|
# aplikáciu nespustíme.
|
|
validate_security_configuration()
|
|
|
|
# Predhriatie embedding modelu je
|
|
# optimalizácia, nie podmienka samotného
|
|
# spustenia HTTP API.
|
|
#
|
|
# Ak warm-up zlyhá, API zostane dostupné.
|
|
# Retrieval endpoint následne vráti
|
|
# kontrolovanú chybu 503, ak embeddingový
|
|
# model skutočne nebude dostupný.
|
|
try:
|
|
await asyncio.to_thread(
|
|
embed_query,
|
|
"warmup",
|
|
)
|
|
|
|
except Exception:
|
|
logger.exception(
|
|
"Predhriatie embedding modelu "
|
|
"zlyhalo. API pokračuje bez "
|
|
"úspešného warm-upu."
|
|
)
|
|
|
|
yield
|
|
|
|
|
|
app = FastAPI(
|
|
title="ZP Agent API",
|
|
description=(
|
|
"RAG API pre vyhľadávanie "
|
|
"a získavanie informácií "
|
|
"z repozitára záverečných prác "
|
|
"ZP Wiki."
|
|
),
|
|
version="0.9.0",
|
|
lifespan=lifespan,
|
|
)
|
|
|
|
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=[
|
|
OPENWEBUI_ORIGIN,
|
|
],
|
|
allow_credentials=True,
|
|
allow_methods=[
|
|
"GET",
|
|
"POST",
|
|
"OPTIONS",
|
|
],
|
|
allow_headers=["*"],
|
|
allow_private_network=True,
|
|
)
|
|
|
|
|
|
app.include_router(
|
|
router
|
|
)
|