added RAG, Multiuser, TG bot

2026-06-13 20:20:56 +00:00
parent 66e1b0e29e
commit c8a9429bed
142 changed files with 19901 additions and 8790 deletions
@@ -1,54 +1,65 @@
-import asyncio
-from contextlib import asynccontextmanager, suppress
-
-from fastapi import FastAPI
-from fastapi.middleware.cors import CORSMiddleware
-
-from app.api.routes import api_router
-from app.config import get_settings
-from app.db.base import init_db
-from app.fitness.watcher import fitness_watcher_loop
-from app.homelab.watcher import homelab_watcher_loop
-from app.pomodoro.watcher import pomodoro_watcher_loop
-from app.reminders.watcher import reminders_watcher_loop
-
-
-@asynccontextmanager
-async def lifespan(_: FastAPI):
-    init_db()
-    pomodoro_task = asyncio.create_task(pomodoro_watcher_loop())
-    fitness_task = asyncio.create_task(fitness_watcher_loop())
-    homelab_task = asyncio.create_task(homelab_watcher_loop())
-    reminders_task = asyncio.create_task(reminders_watcher_loop())
-    yield
-    pomodoro_task.cancel()
-    fitness_task.cancel()
-    homelab_task.cancel()
-    reminders_task.cancel()
-    with suppress(asyncio.CancelledError):
-        await pomodoro_task
-    with suppress(asyncio.CancelledError):
-        await fitness_task
-    with suppress(asyncio.CancelledError):
-        await homelab_task
-    with suppress(asyncio.CancelledError):
-        await reminders_task
-
-
-def create_app() -> FastAPI:
-    settings = get_settings()
-    app = FastAPI(title="Home AI Assistant", lifespan=lifespan)
-
-    app.add_middleware(
-        CORSMiddleware,
-        allow_origins=settings.cors_origins_list,
-        allow_credentials=True,
-        allow_methods=["*"],
-        allow_headers=["*"],
-    )
-
-    app.include_router(api_router)
-    return app
-
-
-app = create_app()
+import asyncio
+from contextlib import asynccontextmanager, suppress
+
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
+
+from app.api.routes import api_router
+from app.config import get_settings
+from app.db.base import init_db
+from app.fitness.watcher import fitness_watcher_loop
+from app.homelab_scoped.watcher import homelab_watcher_loop
+from app.pomodoro.watcher import pomodoro_watcher_loop
+from app.reminders_scoped.watcher import reminders_watcher_loop
+
+
+@asynccontextmanager
+async def lifespan(_: FastAPI):
+    init_db()
+    from app.db.migrate_fitness import run_fitness_migrations
+
+    run_fitness_migrations()
+    from app.db.migrate_multi_user import run_multi_user_migrations
+
+    run_multi_user_migrations()
+    settings = get_settings()
+    if settings.rag_enabled:
+        from app.rag.store import ensure_collections
+
+        ensure_collections()
+    pomodoro_task = asyncio.create_task(pomodoro_watcher_loop())
+    fitness_task = asyncio.create_task(fitness_watcher_loop())
+    homelab_task = asyncio.create_task(homelab_watcher_loop())
+    reminders_task = asyncio.create_task(reminders_watcher_loop())
+    yield
+    pomodoro_task.cancel()
+    fitness_task.cancel()
+    homelab_task.cancel()
+    reminders_task.cancel()
+    with suppress(asyncio.CancelledError):
+        await pomodoro_task
+    with suppress(asyncio.CancelledError):
+        await fitness_task
+    with suppress(asyncio.CancelledError):
+        await homelab_task
+    with suppress(asyncio.CancelledError):
+        await reminders_task
+
+
+def create_app() -> FastAPI:
+    settings = get_settings()
+    app = FastAPI(title="Home AI Assistant", lifespan=lifespan)
+
+    app.add_middleware(
+        CORSMiddleware,
+        allow_origins=settings.cors_origins_list,
+        allow_credentials=True,
+        allow_methods=["*"],
+        allow_headers=["*"],
+    )
+
+    app.include_router(api_router)
+    return app
+
+
+app = create_app()