feat: add backend application foundation

2026-07-13 11:24:05 +08:00
parent 81b0faa67a
commit 44e01e2f93
10 changed files with 314 additions and 0 deletions
@@ -0,0 +1,38 @@
+from collections.abc import AsyncIterator
+from contextlib import asynccontextmanager
+from pathlib import Path
+
+from fastapi import FastAPI
+from pymilvus import MilvusClient
+
+from kbqa.config import get_settings
+from kbqa.database import Database
+
+
+def _ensure_runtime_directories(database_url: str, milvus_uri: str, raw_data_dir: Path) -> None:
+    raw_data_dir.mkdir(parents=True, exist_ok=True)
+    if database_url.startswith("sqlite+aiosqlite:///./"):
+        Path(database_url.removeprefix("sqlite+aiosqlite:///./")).parent.mkdir(
+            parents=True, exist_ok=True
+        )
+    Path(milvus_uri).parent.mkdir(parents=True, exist_ok=True)
+
+
+@asynccontextmanager
+async def lifespan(app: FastAPI) -> AsyncIterator[None]:
+    settings = get_settings()
+    _ensure_runtime_directories(
+        settings.database_url,
+        settings.milvus_uri,
+        settings.raw_data_dir,
+    )
+    database = Database(settings.database_url)
+    milvus_client = MilvusClient(uri=settings.milvus_uri)
+    app.state.settings = settings
+    app.state.database = database
+    app.state.milvus_client = milvus_client
+    try:
+        yield
+    finally:
+        milvus_client.close()
+        await database.dispose()