@vellumai/assistant 0.6.2 → 0.6.4

package/src/cli/commands/email.ts

@@ -1,7 +1,12 @@
-import { readFileSync, writeFileSync } from "node:fs";
+import { createWriteStream, mkdirSync, readFileSync, writeFileSync } from "node:fs";
+import { basename, join } from "node:path";
+import type { Readable } from "node:stream";
+import { pipeline } from "node:stream/promises";
 
 import type { Command } from "commander";
 
+import { getAssistantDomain } from "../../config/env.js";
+import { markdownToEmailHtml } from "../../email/html-renderer.js";
 import { VellumPlatformClient } from "../../platform/client.js";
 import { getCliLogger } from "../logger.js";
 import { shouldOutputJson, writeOutput } from "../output.js";
@@ -9,15 +14,19 @@ import { shouldOutputJson, writeOutput } from "../output.js";
 const log = getCliLogger("email");
 
 export function registerEmailCommand(program: Command): void {
+  const domain = getAssistantDomain();
   const email = program
     .command("email")
-    .description("Email channel operations")
+    .description(
+      `Get your own email address (@${domain}) — register, send, receive, and manage email natively`,
+    )
     .option("--json", "Machine-readable compact JSON output");
 
   email.addHelpText(
     "after",
     `
-Manage the assistant's email channel on the Vellum platform.
+Set up and manage this assistant's native email address on the Vellum
+platform. No third-party email provider or browser sign-up needed.
 
 Examples:
   $ assistant email register mybot
@@ -25,17 +34,19 @@ Examples:
   $ assistant email send user@example.com -s "Hello" -b "Hi there"
   $ assistant email status
   $ assistant email list
+  $ assistant email attachment msg_abc1 --list
+  $ assistant email attachment msg_abc1 att_xyz1
   $ assistant email register mybot --json`,
   );
 
   email
     .command("register <username>")
-    .description("Register an @vellum.me email address for this assistant")
+    .description(`Register an @${domain} email address for this assistant`)
     .addHelpText(
       "after",
       `
 Arguments:
-  username   The local part of the email address (e.g. "mybot" → mybot@vellum.me)
+  username   The local part of the email address (e.g. "mybot" → mybot@${domain})
 
 Registers a new email address on the Vellum platform for the current
 assistant. Each assistant can have one email address. The address is
@@ -43,10 +54,10 @@ immediately active for receiving inbound email.
 
 Examples:
   $ assistant email register mybot
-  ✓ Registered mybot@vellum.me
+  ✓ Registered mybot@${domain}
 
   $ assistant email register support --json
-  {"address":"support@vellum.me","id":"...","created_at":"..."}`,
+  {"address":"support@${domain}","id":"...","created_at":"..."}`,
     )
     .action(async (username: string, _opts: unknown, cmd: Command) => {
       try {
@@ -122,14 +133,14 @@ immediately available for reuse.
 
 Examples:
   $ assistant email unregister
-  Remove mybot@vellum.me? (y/N) y
-  ✓ Unregistered mybot@vellum.me
+  Remove mybot@${domain}? (y/N) y
+  ✓ Unregistered mybot@${domain}
 
   $ assistant email unregister --confirm
-  ✓ Unregistered mybot@vellum.me
+  ✓ Unregistered mybot@${domain}
 
   $ assistant email unregister --json
-  {"unregistered":"mybot@vellum.me"}`,
+  {"unregistered":"mybot@${domain}"}`,
     )
     .action(async (_opts: { confirm?: boolean }, cmd: Command) => {
       try {
@@ -222,12 +233,12 @@ current usage and quota information from the platform.
 
 Examples:
   $ assistant email status
-  Address: mybot@vellum.me
+  Address: mybot@${domain}
   Status:  active
   Sent:    12 / 100 (daily)
 
   $ assistant email status --json
-  {"address":"mybot@vellum.me","status":"active","usage":{"sent_today":12,"daily_limit":100}}`,
+  {"address":"mybot@${domain}","status":"active","usage":{"sent_today":12,"daily_limit":100}}`,
     )
     .action(async (_opts: unknown, cmd: Command) => {
       try {
@@ -446,7 +457,7 @@ or --format json for the full message object.
 Examples:
   $ assistant email download msg_abc123
   From:    user@example.com
-  To:      mybot@vellum.me
+  To:      mybot@${domain}
   Subject: Hello
   Date:    2026-04-05 12:00:00
 
@@ -564,6 +575,10 @@ Examples:
     .option("-b, --body <text>", "Email body (plain text)")
     .option("-f, --file <path>", "Read body from file")
     .option("--html <path>", "HTML body file (optional)")
+    .option(
+      "--reply-to <email_id>",
+      "Reply to an email by its ID (auto-resolves threading headers and subject)",
+    )
     .addHelpText(
       "after",
       `
@@ -576,13 +591,20 @@ from the assistant's registered email address.
 
 Body source priority: --body flag > --file flag > stdin (if not a TTY).
 
+When --reply-to is provided, the platform auto-resolves In-Reply-To,
+References, and Subject headers from the referenced email. You can
+still override subject with -s.
+
 Examples:
   $ assistant email send user@example.com -s "Hello" -b "Hi there"
   ✓ Sent to user@example.com (delivery_id: abc123)
 
-  $ echo "Body text" | assistant email send user@example.com -s "Hello"
+  $ assistant email send user@example.com -b "Thanks!" --reply-to 019d96e4-e5d2-7201-890e-04a21e8f95bb
   ✓ Sent to user@example.com (delivery_id: def456)
 
+  $ echo "Body text" | assistant email send user@example.com -s "Hello"
+  ✓ Sent to user@example.com (delivery_id: ghi789)
+
   $ assistant email send user@example.com -s "Hello" -b "Hi" --json
   {"delivery_id":"abc123","status":"accepted"}`,
     )
@@ -594,6 +616,7 @@ Examples:
           body?: string;
           file?: string;
           html?: string;
+          replyTo?: string;
         },
         cmd: Command,
       ) => {
@@ -648,10 +671,13 @@ Examples:
             );
           }
 
-          // 3. Resolve optional HTML body from file
+          // 3. Resolve HTML body: explicit file > auto-generate from text
           let html: string | undefined;
           if (opts.html) {
             html = readFileSync(opts.html, "utf-8");
+          } else {
+            // Auto-generate HTML from the text body (markdown → email HTML).
+            html = markdownToEmailHtml(text);
           }
 
           // 4. Build payload
@@ -662,6 +688,7 @@ Examples:
           };
           if (opts.subject) payload.subject = opts.subject;
           if (html) payload.html = html;
+          if (opts.replyTo) payload.reply_to = opts.replyTo;
 
           // 5. Send via runtime proxy
           const response = await client.fetch("/v1/runtime-proxy/email/send/", {
@@ -700,4 +727,234 @@ Examples:
         }
       },
     );
+
+email
+  .command("attachment <message-id> [attachment-id]")
+  .description("Download email attachments")
+  .option("--all", "Download all attachments for the message")
+  .option(
+    "-o, --output <dir>",
+    "Output directory (default: current directory)",
+    ".",
+  )
+  .option("--list", "List attachments without downloading")
+  .addHelpText(
+    "after",
+    `
+Arguments:
+message-id      Email message ID (from \`assistant email list --json\`)
+attachment-id   Attachment ID (optional — required unless --all or --list)
+
+Download one or all attachments from a specific email message. Use
+--list to see available attachments without downloading.
+
+Examples:
+$ assistant email attachment msg_abc1 --list
+$ assistant email attachment msg_abc1 att_xyz1
+$ assistant email attachment msg_abc1 att_xyz1 -o ./downloads/
+$ assistant email attachment msg_abc1 --all
+$ assistant email attachment msg_abc1 --all -o ./attachments/
+$ assistant email attachment msg_abc1 --list --json`,
+  )
+  .action(
+    async (
+      messageId: string,
+      attachmentId: string | undefined,
+      opts: {
+        all?: boolean;
+        output?: string;
+        list?: boolean;
+      },
+      cmd: Command,
+    ) => {
+      try {
+        const client = await VellumPlatformClient.create();
+        if (!client) {
+          throw new Error(
+            "Platform credentials not configured. Run: assistant platform connect",
+          );
+        }
+        if (!client.platformAssistantId) {
+          throw new Error(
+            "Assistant ID not configured. Set PLATFORM_ASSISTANT_ID or run: assistant platform connect",
+          );
+        }
+
+        const assistantId = client.platformAssistantId;
+        const basePath = `/v1/assistants/${assistantId}/emails/${messageId}/attachments`;
+
+        if (opts.list) {
+          // List mode — show attachment metadata without downloading
+          const response = await client.fetch(`${basePath}/`);
+          if (!response.ok) {
+            const body = (await response
+              .json()
+              .catch(() => ({}))) as Record<string, unknown>;
+            const detail = body.detail ?? `HTTP ${response.status}`;
+            throw new Error(String(detail));
+          }
+
+          const data = (await response.json()) as {
+            results: AttachmentMeta[];
+          };
+
+          if (shouldOutputJson(cmd)) {
+            writeOutput(cmd, data);
+          } else {
+            const attachments = data.results ?? [];
+            if (attachments.length === 0) {
+              log.info("No attachments for this message.");
+            } else {
+              for (const att of attachments) {
+                log.info(
+                  `  ${att.id}  ${att.filename}  (${att.content_type}, ${formatBytes(att.size_bytes)})`,
+                );
+              }
+              log.info(`\n${attachments.length} attachment(s)`);
+            }
+          }
+          return;
+        }
+
+        if (!opts.all && !attachmentId) {
+          throw new Error(
+            "Specify an attachment ID, or use --all to download all attachments. Use --list to see available attachments.",
+          );
+        }
+
+        // Ensure output directory exists
+        const outDir = opts.output ?? ".";
+        mkdirSync(outDir, { recursive: true });
+
+        if (opts.all) {
+          // Download all attachments
+          const listResponse = await client.fetch(`${basePath}/`);
+          if (!listResponse.ok) {
+            const body = (await listResponse
+              .json()
+              .catch(() => ({}))) as Record<string, unknown>;
+            const detail = body.detail ?? `HTTP ${listResponse.status}`;
+            throw new Error(String(detail));
+          }
+
+          const listData = (await listResponse.json()) as {
+            results: AttachmentMeta[];
+          };
+
+          const attachments = listData.results ?? [];
+          if (attachments.length === 0) {
+            throw new Error("No attachments for this message.");
+          }
+
+          const downloaded: { filename: string; size_bytes: number }[] = [];
+          for (const att of attachments) {
+            const dest = join(outDir, safeFilename(att.filename));
+            await downloadAttachment(client, basePath, att.id, dest);
+            downloaded.push({
+              filename: att.filename,
+              size_bytes: att.size_bytes,
+            });
+          }
+
+          if (shouldOutputJson(cmd)) {
+            writeOutput(cmd, {
+              downloaded: downloaded.length,
+              directory: outDir,
+              files: downloaded,
+            });
+          } else {
+            log.info(
+              `✓ Downloaded ${downloaded.length} attachment(s) to ${outDir}`,
+            );
+            for (const f of downloaded) {
+              log.info(`  - ${f.filename} (${formatBytes(f.size_bytes)})`);
+            }
+          }
+        } else {
+          // Download single attachment — first get metadata for the filename
+          const metaResponse = await client.fetch(
+            `${basePath}/${attachmentId}/`,
+          );
+          if (!metaResponse.ok) {
+            const body = (await metaResponse
+              .json()
+              .catch(() => ({}))) as Record<string, unknown>;
+            const detail = body.detail ?? `HTTP ${metaResponse.status}`;
+            throw new Error(String(detail));
+          }
+
+          const meta = (await metaResponse.json()) as AttachmentMeta;
+          const dest = join(outDir, safeFilename(meta.filename));
+          await downloadAttachment(client, basePath, meta.id, dest);
+
+          if (shouldOutputJson(cmd)) {
+            writeOutput(cmd, {
+              filename: meta.filename,
+              size_bytes: meta.size_bytes,
+              saved: dest,
+            });
+          } else {
+            log.info(
+              `✓ Downloaded ${meta.filename} (${formatBytes(meta.size_bytes)})`,
+            );
+          }
+        }
+      } catch (err) {
+        const message = err instanceof Error ? err.message : String(err);
+        if (shouldOutputJson(cmd)) {
+          writeOutput(cmd, { error: message });
+        } else {
+          log.error(`Error: ${message}`);
+        }
+        process.exitCode = 1;
+      }
+    },
+  );
+}
+
+interface AttachmentMeta {
+id: string;
+filename: string;
+content_type: string;
+size_bytes: number;
+content_id: string;
+created_at: string;
+}
+
+function formatBytes(bytes: number): string {
+if (bytes < 1024) return `${bytes} B`;
+if (bytes < 1024 * 1024) return `${(bytes / 1024).toFixed(1)} KB`;
+return `${(bytes / (1024 * 1024)).toFixed(1)} MB`;
+}
+
+function safeFilename(name: string): string {
+// Strip path separators and null bytes — keep the basename only
+return basename(name).replace(/[\x00/\\]/g, "_") || "attachment";
+}
+
+async function downloadAttachment(
+client: VellumPlatformClient,
+basePath: string,
+attachmentId: string,
+dest: string,
+): Promise<void> {
+const response = await client.fetch(
+  `${basePath}/${attachmentId}/download/`,
+);
+
+if (!response.ok) {
+  const body = (await response.json().catch(() => ({}))) as Record<
+    string,
+    unknown
+  >;
+  const detail = body.detail ?? `HTTP ${response.status}`;
+  throw new Error(`Failed to download attachment: ${detail}`);
+}
+
+if (!response.body) {
+  throw new Error("Empty response body from download endpoint.");
+}
+
+const fileStream = createWriteStream(dest);
+await pipeline(response.body as unknown as Readable, fileStream);
 }

package/src/cli/commands/mcp.ts

@@ -34,13 +34,25 @@ export async function checkServerHealth(
       }),
     ]);
 
-    if (!client.isConnected) {
-      return "! Needs authentication";
+    if (client.isConnected) {
+      await client.disconnect();
+      return "\u2713 Connected";
+    }
+
+    // connect() swallows errors — check lastError to distinguish auth from
+    // transport failures (DNS, TLS, 500, stdio crash, etc.).
+    const err = client.lastError;
+    if (err) {
+      const message = err.message;
+      if (message.includes("timeout")) {
+        return "\u2717 Timed out";
+      }
+      return `\u2717 Error: ${message}`;
     }
 
-    await client.disconnect();
-    return "\u2713 Connected";
+    return "! Needs authentication";
   } catch (err) {
+    // Only the external timeout Promise can throw here (connect() never does).
     try {
       await client.disconnect();
     } catch {

package/src/cli/commands/oauth/__tests__/connect.test.ts

@@ -264,9 +264,10 @@ describe("assistant oauth connect", () => {
 
   test("managed mode with --no-browser: prints connect URL", async () => {
     mockGetProvider = () => ({
-      providerKey: "google",
-      authUrl: "https://accounts.google.com/o/oauth2/v2/auth",
-      tokenUrl: "https://oauth2.googleapis.com/token",
+      provider: "google",
+      authorizeUrl: "https://accounts.google.com/o/oauth2/v2/auth",
+      tokenExchangeUrl: "https://oauth2.googleapis.com/token",
+      tokenExchangeBodyFormat: "form",
       managedServiceConfigKey: "google-oauth",
     });
     mockIsManagedMode = () => true;
@@ -301,9 +302,10 @@ describe("assistant oauth connect", () => {
 
   test("managed mode default: opens browser and polls for new connection", async () => {
     mockGetProvider = () => ({
-      providerKey: "google",
-      authUrl: "https://accounts.google.com/o/oauth2/v2/auth",
-      tokenUrl: "https://oauth2.googleapis.com/token",
+      provider: "google",
+      authorizeUrl: "https://accounts.google.com/o/oauth2/v2/auth",
+      tokenExchangeUrl: "https://oauth2.googleapis.com/token",
+      tokenExchangeBodyFormat: "form",
       managedServiceConfigKey: "google-oauth",
     });
     mockIsManagedMode = () => true;
@@ -357,9 +359,10 @@ describe("assistant oauth connect", () => {
 
   test("BYO mode with --no-browser: prints auth URL", async () => {
     mockGetProvider = () => ({
-      providerKey: "google",
-      authUrl: "https://accounts.google.com/o/oauth2/v2/auth",
-      tokenUrl: "https://oauth2.googleapis.com/token",
+      provider: "google",
+      authorizeUrl: "https://accounts.google.com/o/oauth2/v2/auth",
+      tokenExchangeUrl: "https://oauth2.googleapis.com/token",
+      tokenExchangeBodyFormat: "form",
       managedServiceConfigKey: null,
     });
     mockIsManagedMode = () => false;
@@ -368,7 +371,7 @@ describe("assistant oauth connect", () => {
       id: "app-1",
       clientId: "byo-client-id",
       clientSecretCredentialPath: "oauth_app/app-1/client_secret",
-      providerKey: "google",
+      provider: "google",
       createdAt: 0,
       updatedAt: 0,
     });
@@ -376,7 +379,7 @@ describe("assistant oauth connect", () => {
     mockOrchestrateOAuthConnect = async () => ({
       success: true,
       deferred: true,
-      authUrl: "https://accounts.google.com/o/oauth2/v2/auth?state=abc",
+      authorizeUrl: "https://accounts.google.com/o/oauth2/v2/auth?state=abc",
       state: "abc",
       service: "google",
     });
@@ -403,9 +406,10 @@ describe("assistant oauth connect", () => {
 
   test("BYO mode default calls orchestrator with isInteractive: true", async () => {
     mockGetProvider = () => ({
-      providerKey: "google",
-      authUrl: "https://accounts.google.com/o/oauth2/v2/auth",
-      tokenUrl: "https://oauth2.googleapis.com/token",
+      provider: "google",
+      authorizeUrl: "https://accounts.google.com/o/oauth2/v2/auth",
+      tokenExchangeUrl: "https://oauth2.googleapis.com/token",
+      tokenExchangeBodyFormat: "form",
       managedServiceConfigKey: null,
     });
     mockIsManagedMode = () => false;
@@ -414,7 +418,7 @@ describe("assistant oauth connect", () => {
       id: "app-1",
       clientId: "test-id",
       clientSecretCredentialPath: "oauth_app/app-1/client_secret",
-      providerKey: "google",
+      provider: "google",
       createdAt: 0,
       updatedAt: 0,
     });
@@ -455,9 +459,10 @@ describe("assistant oauth connect", () => {
 
   test("BYO mode: missing app with --client-id returns error with hint", async () => {
     mockGetProvider = () => ({
-      providerKey: "google",
-      authUrl: "https://accounts.google.com/o/oauth2/v2/auth",
-      tokenUrl: "https://oauth2.googleapis.com/token",
+      provider: "google",
+      authorizeUrl: "https://accounts.google.com/o/oauth2/v2/auth",
+      tokenExchangeUrl: "https://oauth2.googleapis.com/token",
+      tokenExchangeBodyFormat: "form",
       managedServiceConfigKey: null,
     });
     mockIsManagedMode = () => false;
@@ -483,9 +488,10 @@ describe("assistant oauth connect", () => {
 
   test("BYO mode: no client_id found returns error with hint", async () => {
     mockGetProvider = () => ({
-      providerKey: "google",
-      authUrl: "https://accounts.google.com/o/oauth2/v2/auth",
-      tokenUrl: "https://oauth2.googleapis.com/token",
+      provider: "google",
+      authorizeUrl: "https://accounts.google.com/o/oauth2/v2/auth",
+      tokenExchangeUrl: "https://oauth2.googleapis.com/token",
+      tokenExchangeBodyFormat: "form",
       managedServiceConfigKey: null,
     });
     mockIsManagedMode = () => false;
@@ -509,9 +515,10 @@ describe("assistant oauth connect", () => {
 
   test("--client-id is silently ignored in managed mode", async () => {
     mockGetProvider = () => ({
-      providerKey: "google",
-      authUrl: "https://accounts.google.com/o/oauth2/v2/auth",
-      tokenUrl: "https://oauth2.googleapis.com/token",
+      provider: "google",
+      authorizeUrl: "https://accounts.google.com/o/oauth2/v2/auth",
+      tokenExchangeUrl: "https://oauth2.googleapis.com/token",
+      tokenExchangeBodyFormat: "form",
       managedServiceConfigKey: "google-oauth",
     });
     mockIsManagedMode = () => true;
@@ -547,9 +554,10 @@ describe("assistant oauth connect", () => {
 
   test("JSON output for deferred case includes ok, deferred, authUrl, service", async () => {
     mockGetProvider = () => ({
-      providerKey: "slack",
-      authUrl: "https://slack.com/oauth/v2/authorize",
-      tokenUrl: "https://slack.com/api/oauth.v2.access",
+      provider: "slack",
+      authorizeUrl: "https://slack.com/oauth/v2/authorize",
+      tokenExchangeUrl: "https://slack.com/api/oauth.v2.access",
+      tokenExchangeBodyFormat: "form",
       managedServiceConfigKey: null,
     });
     mockIsManagedMode = () => false;
@@ -558,7 +566,7 @@ describe("assistant oauth connect", () => {
       id: "app-slack",
       clientId: "slack-client-id",
       clientSecretCredentialPath: "oauth_app/app-slack/client_secret",
-      providerKey: "slack",
+      provider: "slack",
       createdAt: 0,
       updatedAt: 0,
     });
@@ -566,7 +574,7 @@ describe("assistant oauth connect", () => {
     mockOrchestrateOAuthConnect = async () => ({
       success: true,
       deferred: true,
-      authUrl: "https://slack.com/oauth/v2/authorize?state=xyz",
+      authorizeUrl: "https://slack.com/oauth/v2/authorize?state=xyz",
       state: "xyz",
       service: "slack",
     });
@@ -591,9 +599,10 @@ describe("assistant oauth connect", () => {
 
   test("JSON output for completed case includes ok, grantedScopes, accountInfo", async () => {
     mockGetProvider = () => ({
-      providerKey: "google",
-      authUrl: "https://accounts.google.com/o/oauth2/v2/auth",
-      tokenUrl: "https://oauth2.googleapis.com/token",
+      provider: "google",
+      authorizeUrl: "https://accounts.google.com/o/oauth2/v2/auth",
+      tokenExchangeUrl: "https://oauth2.googleapis.com/token",
+      tokenExchangeBodyFormat: "form",
       managedServiceConfigKey: null,
     });
     mockIsManagedMode = () => false;
@@ -602,7 +611,7 @@ describe("assistant oauth connect", () => {
       id: "app-1",
       clientId: "completed-client-id",
       clientSecretCredentialPath: "oauth_app/app-1/client_secret",
-      providerKey: "google",
+      provider: "google",
       createdAt: 0,
       updatedAt: 0,
     });
@@ -635,9 +644,10 @@ describe("assistant oauth connect", () => {
 
   test("BYO mode: client_secret required but missing returns error with hint", async () => {
     mockGetProvider = () => ({
-      providerKey: "google",
-      authUrl: "https://accounts.google.com/o/oauth2/v2/auth",
-      tokenUrl: "https://oauth2.googleapis.com/token",
+      provider: "google",
+      authorizeUrl: "https://accounts.google.com/o/oauth2/v2/auth",
+      tokenExchangeUrl: "https://oauth2.googleapis.com/token",
+      tokenExchangeBodyFormat: "form",
       tokenEndpointAuthMethod: "client_secret_post",
       managedServiceConfigKey: null,
       requiresClientSecret: 1,
@@ -648,7 +658,7 @@ describe("assistant oauth connect", () => {
       id: "app-1",
       clientId: "test-id",
       clientSecretCredentialPath: "oauth_app/app-1/client_secret",
-      providerKey: "google",
+      provider: "google",
       createdAt: 0,
       updatedAt: 0,
     });
@@ -674,9 +684,10 @@ describe("assistant oauth connect", () => {
 
   test("manual-token provider returns error directing to credentials command", async () => {
     mockGetProvider = () => ({
-      providerKey: "slack_channel",
-      authUrl: "urn:manual-token",
-      tokenUrl: "urn:manual-token",
+      provider: "slack_channel",
+      authorizeUrl: "urn:manual-token",
+      tokenExchangeUrl: "urn:manual-token",
+      tokenExchangeBodyFormat: "form",
       managedServiceConfigKey: null,
     });
     mockIsManagedMode = () => false;
@@ -701,9 +712,10 @@ describe("assistant oauth connect", () => {
 
   test("BYO mode: orchestrator error propagates correctly", async () => {
     mockGetProvider = () => ({
-      providerKey: "google",
-      authUrl: "https://accounts.google.com/o/oauth2/v2/auth",
-      tokenUrl: "https://oauth2.googleapis.com/token",
+      provider: "google",
+      authorizeUrl: "https://accounts.google.com/o/oauth2/v2/auth",
+      tokenExchangeUrl: "https://oauth2.googleapis.com/token",
+      tokenExchangeBodyFormat: "form",
       managedServiceConfigKey: null,
     });
     mockIsManagedMode = () => false;
@@ -712,7 +724,7 @@ describe("assistant oauth connect", () => {
       id: "app-1",
       clientId: "client-id",
       clientSecretCredentialPath: "oauth_app/app-1/client_secret",
-      providerKey: "google",
+      provider: "google",
       createdAt: 0,
       updatedAt: 0,
     });