first commit

rag_api/app/main.py

@@ -0,0 +1,64 @@
+from __future__ import annotations
+
+from contextlib import asynccontextmanager
+
+from fastapi import FastAPI
+
+from .config import settings
+from .menu_catalog import MenuCatalog
+from .models import ChatRequest, ChatResponse, IndexResponse
+from .service import RagService
+
+
+rag_service = RagService()
+menu_catalog = MenuCatalog()
+
+
+@asynccontextmanager
+async def lifespan(_: FastAPI):
+    if settings.index_on_startup:
+        await rag_service.reindex()
+    yield
+
+
+app = FastAPI(title=settings.app_name, version="1.0.0", lifespan=lifespan)
+
+
+@app.get("/health")
+async def health() -> dict[str, str]:
+    return {"status": "ok"}
+
+
+@app.post("/chat", response_model=ChatResponse)
+async def chat(request: ChatRequest) -> ChatResponse:
+    return await rag_service.chat(request)
+
+
+@app.post("/admin/reindex", response_model=IndexResponse)
+async def reindex() -> IndexResponse:
+    return await rag_service.reindex()
+
+
+@app.get("/menu/search")
+async def search_menu(
+    query: str = "",
+    max_price: int | None = None,
+    category: str | None = None,
+    must_include: str | None = None,
+    must_not_include: str | None = None,
+    limit: int = 5,
+) -> dict[str, object]:
+    return {
+        "items": rag_service.search_menu(
+            query=query,
+            max_price=max_price,
+            category=category,
+            must_include=[value.strip() for value in must_include.split(",")]
+            if must_include
+            else None,
+            must_not_include=[value.strip() for value in must_not_include.split(",")]
+            if must_not_include
+            else None,
+            limit=limit,
+        )
+    }