From 88d3d336bfdb5212f9eb3dcb2d89ec996733c262 Mon Sep 17 00:00:00 2001
From: Anish Sarkar <104695310+AnishSarkar22@users.noreply.github.com>
Date: Tue, 14 Jul 2026 00:14:34 +0530
Subject: [PATCH] feat(schemas): add stage field to LlmSetupStatusRead

---
 surfsense_backend/app/schemas/model_connections.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/surfsense_backend/app/schemas/model_connections.py b/surfsense_backend/app/schemas/model_connections.py
index f190ad745..9c61e9c38 100644
--- a/surfsense_backend/app/schemas/model_connections.py
+++ b/surfsense_backend/app/schemas/model_connections.py
@@ -153,9 +153,11 @@ class LlmSetupStatusRead(BaseModel):
 
     ``status`` is the only thing the frontend gate acts on; ``source`` is
     informational and ``can_configure`` selects the onboarding vs. blocked
-    screen for members who cannot manage models.
+    screen for members who cannot manage models. ``stage`` refines a
+    ``needs_setup`` verdict into first-run (``initial_setup``) vs. recovery.
     """
 
     status: Literal["ready", "needs_setup"]
     source: Literal["global_config", "models", "none"]
     can_configure: bool
+    stage: Literal["initial_setup", "recovery", "ready"]