| from pathlib import Path |
|
|
| import httpx |
| from fastapi import FastAPI |
| from fastapi.middleware.cors import CORSMiddleware |
| from fastapi.staticfiles import StaticFiles |
|
|
| from config import BASE, RAG_SYSTEM_PROMPT |
| from helpers import auth_headers |
| from rag import rag_state, router as rag_router |
| from routers import clinical, frontend, llm, local_rag as local_rag_router, personas, sessions, telemetry as telemetry_router |
|
|
| app = FastAPI(title="Anam AI Demo with RAG") |
| app.add_middleware(CORSMiddleware, allow_origins=["*"], allow_methods=["*"], allow_headers=["*"]) |
| app.mount("/static", StaticFiles(directory=Path(__file__).parent / "static"), name="static") |
|
|
| for router in (frontend.router, personas.router, sessions.router, llm.router, clinical.router, telemetry_router.router): |
| app.include_router(router) |
| app.include_router(rag_router) |
| app.include_router(local_rag_router.router) |
|
|
|
|
|
|
| @app.on_event("startup") |
| async def sync_persona_prompt(): |
| """Push the current RAG_SYSTEM_PROMPT to the existing persona on every server start.""" |
| if not rag_state.get("persona_id"): |
| return |
| try: |
| async with httpx.AsyncClient(timeout=15) as c: |
| await c.put( |
| f"{BASE}/personas/{rag_state['persona_id']}", |
| headers=auth_headers(), |
| json={"systemPrompt": RAG_SYSTEM_PROMPT}, |
| ) |
| print("RAG persona system prompt synced on startup.") |
| except Exception as e: |
| print(f"Could not sync persona prompt on startup: {e}") |
|
|
