fix(webview): dismiss welcome after Zoo Gateway sign-in (#961) (#962)

checkExistKey ignored Zoo Gateway session auth, so Finish saved config but never left the setup screen.

Co-authored-by: Cursor <cursoragent@cursor.com>

Author: JamesRobert20

src/shared/__tests__/checkExistApiConfig.spec.ts

@@ -86,4 +86,36 @@ describe("checkExistKey", () => {
 		}
 		expect(checkExistKey(config)).toBe(false)
 	})
+
+	it("should return false for zoo-gateway without session token or auth", () => {
+		const config: ProviderSettings = {
+			apiProvider: "zoo-gateway",
+			zooGatewayModelId: "alibaba/qwen-3.6-max-preview",
+		}
+		expect(checkExistKey(config)).toBe(false)
+		expect(checkExistKey(config, false)).toBe(false)
+	})
+
+	it("should return true for zoo-gateway when profile has zooSessionToken", () => {
+		const config: ProviderSettings = {
+			apiProvider: "zoo-gateway",
+			zooSessionToken: "zoo_ext_test_token",
+		}
+		expect(checkExistKey(config)).toBe(true)
+	})
+
+	it("should return true for zoo-gateway when Zoo Code session auth is active", () => {
+		const config: ProviderSettings = {
+			apiProvider: "zoo-gateway",
+			zooGatewayModelId: "alibaba/qwen-3.6-max-preview",
+		}
+		expect(checkExistKey(config, true)).toBe(true)
+	})
+
+	it("should ignore zooCodeIsAuthenticated for non-zoo-gateway providers", () => {
+		const config: ProviderSettings = {
+			apiProvider: "openrouter",
+		}
+		expect(checkExistKey(config, true)).toBe(false)
+	})
 })

src/shared/checkExistApiConfig.ts

@@ -1,6 +1,14 @@
 import { SECRET_STATE_KEYS, GLOBAL_SECRET_KEYS, ProviderSettings } from "@roo-code/types"
 
-export function checkExistKey(config: ProviderSettings | undefined) {
+/**
+ * Returns whether a provider profile is sufficiently configured to leave the
+ * welcome/setup gate.
+ *
+ * `zooCodeIsAuthenticated` is needed for Zoo Gateway: auth lives in global
+ * secret storage (`zoo-code-auth`), and `zooSessionToken` is not part of
+ * `SECRET_STATE_KEYS`, so session-auth alone would otherwise look unconfigured.
+ */
+export function checkExistKey(config: ProviderSettings | undefined, zooCodeIsAuthenticated?: boolean) {
 	if (!config) {
 		return false
 	}
@@ -10,6 +18,12 @@ export function checkExistKey(config: ProviderSettings | undefined) {
 		return true
 	}
 
+	// Zoo Gateway uses session auth (profile token and/or global Zoo Code login),
+	// not a traditional API key listed in SECRET_STATE_KEYS.
+	if (config.apiProvider === "zoo-gateway") {
+		return Boolean(config.zooSessionToken) || Boolean(zooCodeIsAuthenticated)
+	}
+
 	// Check all secret keys from the centralized SECRET_STATE_KEYS array.
 	// Filter out keys that are not part of ProviderSettings (global secrets are stored separately)
 	const providerSecretKeys = SECRET_STATE_KEYS.filter((key) => !GLOBAL_SECRET_KEYS.includes(key as any))

webview-ui/src/context/ExtensionStateContext.tsx

@@ -317,7 +317,7 @@ export const ExtensionStateContextProvider: React.FC<{ children: React.ReactNode
 				case "state": {
 					const newState = message.state ?? {}
 					setState((prevState) => mergeExtensionState(prevState, newState))
-					setShowWelcome(!checkExistKey(newState.apiConfiguration))
+					setShowWelcome(!checkExistKey(newState.apiConfiguration, newState.zooCodeIsAuthenticated))
 					setDidHydrateState(true)
 					// Update alwaysAllowFollowupQuestions if present in state message
 					if ((newState as any).alwaysAllowFollowupQuestions !== undefined) {