From 4b4efc243aa6230d179c7445815319d3f170c4a2 Mon Sep 17 00:00:00 2001
From: Anish Sarkar <104695310+AnishSarkar22@users.noreply.github.com>
Date: Tue, 14 Jul 2026 00:15:06 +0530
Subject: [PATCH] feat(contracts): add stage to llmSetupStatus schema

---
 surfsense_web/contracts/types/model-connections.types.ts | 1 +
 1 file changed, 1 insertion(+)

diff --git a/surfsense_web/contracts/types/model-connections.types.ts b/surfsense_web/contracts/types/model-connections.types.ts
index 7771d2b93..778ade901 100644
--- a/surfsense_web/contracts/types/model-connections.types.ts
+++ b/surfsense_web/contracts/types/model-connections.types.ts
@@ -115,6 +115,7 @@ export const llmSetupStatus = z.object({
 	status: z.enum(["ready", "needs_setup"]),
 	source: z.enum(["global_config", "models", "none"]),
 	can_configure: z.boolean(),
+	stage: z.enum(["initial_setup", "recovery", "ready"]),
 });
 
 export const modelProviderRead = z.object({