fix(xai): preserve runtime device auth choice alias

fuller-stack-dev · fuller-stack-dev · commit fef3cb24afb0 · 2026-06-27T08:48:25.000-07:00
diff --git a/extensions/xai/index.test.ts b/extensions/xai/index.test.ts
@@ -92,7 +92,8 @@ describe("xai provider plugin", () => {
     expect(oauth?.wizard?.choiceId).toBe("xai-oauth");
     const deviceCode = provider.auth?.find((method) => method.id === "device-code");
     expect(deviceCode?.kind).toBe("device_code");
-    expect(deviceCode?.wizard).toBeUndefined();
+    expect(deviceCode?.wizard?.choiceId).toBe("xai-device-code");
+    expect(deviceCode?.wizard?.assistantVisibility).toBe("manual-only");
     expect(manifest.providerAuthChoices).toContainEqual(
       expect.objectContaining({
         assistantVisibility: "manual-only",
diff --git a/extensions/xai/xai-oauth.test.ts b/extensions/xai/xai-oauth.test.ts
@@ -78,7 +78,9 @@ describe("xAI OAuth", () => {
 
     expect(method.id).toBe("device-code");
     expect(method.kind).toBe("device_code");
-    expect(method.wizard).toBeUndefined();
+    expect(method.wizard?.choiceId).toBe("xai-device-code");
+    expect(method.wizard?.methodId).toBe("device-code");
+    expect(method.wizard?.assistantVisibility).toBe("manual-only");
   });
 
   it("validates discovered endpoints before using them", async () => {
diff --git a/extensions/xai/xai-oauth.ts b/extensions/xai/xai-oauth.ts
@@ -19,6 +19,7 @@ const PROVIDER_ID = "xai";
 export const XAI_OAUTH_METHOD_ID = "oauth";
 export const XAI_OAUTH_CHOICE_ID = "xai-oauth";
 export const XAI_DEVICE_CODE_METHOD_ID = "device-code";
+export const XAI_DEVICE_CODE_CHOICE_ID = "xai-device-code";
 export const XAI_OAUTH_CLIENT_ID = "b1a00492-073a-47ea-816f-4c329264a828";
 export const XAI_OAUTH_SCOPE = "openid profile email offline_access grok-cli:access api:access";
 export const XAI_OAUTH_ISSUER = "https://auth.x.ai";
@@ -688,6 +689,16 @@ export function createXaiDeviceCodeAuthMethod(): ProviderAuthMethod {
     label: "xAI device code",
     hint: "Deprecated alias for xAI OAuth device-code login",
     kind: "device_code",
+    wizard: {
+      choiceId: XAI_DEVICE_CODE_CHOICE_ID,
+      choiceLabel: "xAI device code",
+      choiceHint: "Compatibility alias for xAI OAuth device-code sign-in",
+      assistantVisibility: "manual-only",
+      groupId: PROVIDER_ID,
+      groupLabel: "xAI (Grok)",
+      groupHint: "API key or OAuth",
+      methodId: XAI_DEVICE_CODE_METHOD_ID,
+    },
     run: async (ctx) => loginXaiDeviceCode(ctx),
   };
 }
