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({