Spaces:
Running
Running
File size: 3,798 Bytes
a63c61f d0668ca a63c61f 0cc6fd8 a63c61f d0668ca a63c61f 03c5a91 a63c61f 0cc6fd8 d0668ca 0cc6fd8 d0668ca 0cc6fd8 a63c61f 0cc6fd8 a63c61f 03c5a91 a63c61f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 | import os
import sys
import logging
from pathlib import Path
from contextlib import asynccontextmanager
# Add project root and local src to path to allow direct execution
current_dir = Path(__file__).resolve().parent
project_root = current_dir.parent.parent.parent.parent # d:\...pipeline
sys.path.append(str(project_root))
sys.path.append(str(current_dir.parent)) # d:\...\rag-api
# ── Configure logging BEFORE any imports that use loggers ────────────────────
logging.basicConfig(
level=logging.INFO,
format="%(asctime)s [%(levelname)s] %(name)s: %(message)s",
datefmt="%H:%M:%S",
force=True,
)
# Suppress noisy third-party loggers
logging.getLogger("httpx").setLevel(logging.WARNING)
logging.getLogger("httpcore").setLevel(logging.WARNING)
logging.getLogger("hf_xet").setLevel(logging.WARNING)
logging.getLogger("huggingface_hub").setLevel(logging.WARNING)
logging.getLogger("transformers").setLevel(logging.WARNING)
logging.getLogger("FlagEmbedding").setLevel(logging.WARNING)
logging.getLogger("sentence_transformers").setLevel(logging.WARNING)
logging.getLogger("urllib3").setLevel(logging.WARNING)
logger = logging.getLogger(__name__)
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from src.core.config import settings
from src.api.routes import rag, analytics, interactions, accounts, news, auth, top_stories
from src.infrastructure.database import init_db
from src.api.dependencies import prewarm_models
import threading
import uvicorn
@asynccontextmanager
async def lifespan(app: FastAPI):
# Startup
logger.info("=" * 60)
logger.info("ARKI AI RAG API — Pipeline v2.5 starting up")
logger.info(f" LLM provider : {settings.LLM_PROVIDER}")
logger.info(f" Reranker model : {settings.RERANKER_MODEL}")
logger.info(f" Hybrid search : {settings.ENABLE_HYBRID_SEARCH}")
logger.info(f" Jina Reader : {settings.ENABLE_JINA_READER}")
logger.info(f" Jina Reranker : {settings.JINA_RERANKER_ENABLED}")
logger.info(f" NewsAPI : {bool(settings.NEWSAPI_KEY and settings.NEWSAPI_KEY != 'your-newsapi-key-here')}")
logger.info("=" * 60)
init_db()
yield
# Shutdown
pass
app = FastAPI(
title=settings.PROJECT_NAME,
openapi_url=f"{settings.API_V1_STR}/openapi.json",
lifespan=lifespan
)
# Set all CORS enabled origins
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
app.include_router(rag.router, prefix=f"{settings.API_V1_STR}/rag", tags=["RAG"])
app.include_router(analytics.router, prefix=f"{settings.API_V1_STR}/analytics", tags=["Analytics"])
app.include_router(interactions.router, prefix=f"{settings.API_V1_STR}/interactions", tags=["Interactions"])
app.include_router(accounts.router, prefix=f"{settings.API_V1_STR}/accounts", tags=["Accounts"])
# Alias for registration convenience
app.include_router(accounts.router, prefix=f"{settings.API_V1_STR}/users", tags=["Users"], include_in_schema=False)
app.include_router(auth.router, prefix=f"{settings.API_V1_STR}/auth", tags=["Auth"])
app.include_router(news.router, prefix=f"{settings.API_V1_STR}/news", tags=["News"])
app.include_router(top_stories.router, prefix=f"{settings.API_V1_STR}", tags=["Top Stories"])
@app.get(f"{settings.API_V1_STR}/status")
def get_status():
return {"status": "online", "message": "RAG API Service is operational"}
@app.get("/")
def read_root():
return {"message": "Welcome to the RAG API Service", "status": "online"}
if __name__ == "__main__":
# Start the server directly from this script
# Use the app object directly for simplicity when running as main
uvicorn.run(app, host="0.0.0.0", port=8000)
|