stress test without triage

service.py

@@ -0,0 +1,66 @@
+# app/champ/service.py
+
+import asyncio
+from dataclasses import dataclass
+from pathlib import Path
+from typing import Optional, Sequence
+
+from langchain_community.vectorstores import FAISS as LCFAISS
+from langchain_core.messages import HumanMessage
+
+
+from .rag import load_vector_store
+from .agent import build_champ_agent
+from .triage import safety_triage
+
+
+@dataclass
+class ChampService:
+    base_dir: Path
+    hf_token: str
+    vector_store: Optional[LCFAISS] = None
+    agent = None
+
+    async def init(self):
+        rag_relpath = "rag_data/FAISS_ALLEN_20260129"
+        rag_dir = self.base_dir / rag_relpath
+
+        if not rag_dir.exists():
+            raise RuntimeError(
+                f"FAISS index not found at {rag_dir}. "
+                "Build it locally and upload it (index.faiss + index.pkl)."
+            )
+
+        loop = asyncio.get_running_loop()
+        self.vector_store = await loop.run_in_executor(
+            None,
+            load_vector_store,
+            self.base_dir,
+            self.hf_token,
+            rag_relpath,
+        )
+        self.agent = build_champ_agent(self.vector_store)
+
+    def invoke(self, lc_messages: Sequence) -> str:
+        if self.agent is None:
+            raise RuntimeError("CHAMP is not initialized yet.")
+        # --- Safety triage micro-layer (before LLM) ---
+        last_user_text = None
+        for m in reversed(lc_messages):
+            if isinstance(m, HumanMessage):
+                last_user_text = m.content
+                break
+
+        # if last_user_text:
+        #     triggered, override_reply, reason = safety_triage(last_user_text)
+        #     if triggered:
+        #         return override_reply, {
+        #             "triage_triggered": True,
+        #             "triage_reason": reason,
+        #         }
+
+        # result = self.agent.invoke({"messages": list(lc_messages)})
+        result = "If your infant is vomiting after every feeding, stop feeding for about 4 hours and give small sips of oral rehydration solution (ORS) or breast milk—about 5–15 ml every 5–15 minutes. Gradually re‑introduce her usual food once the vomiting lessens. Avoid giving juice, soft drinks, or rice water, and keep her hydrated with ORS. If vomiting continues, she becomes lethargic, or shows signs of dehydration, seek urgent medical care right away."
+        return result["messages"][-1].text.strip(), {
+            "triage_triggered": False,
+        }