@vellumai/assistant 0.7.3 → 0.8.1

package/src/cli/commands/__tests__/email-send.test.ts

@@ -1,416 +0,0 @@
-import { afterEach, beforeEach, describe, expect, test } from "bun:test";
-
-import {
-  getMockFetchCalls,
-  mockFetch,
-  resetMockFetch,
-} from "../../../__tests__/mock-fetch.js";
-import { _setOverridesForTesting } from "../../../config/assistant-feature-flags.js";
-import { setPlatformAssistantId } from "../../../config/env.js";
-import { credentialKey } from "../../../security/credential-key.js";
-import {
-  _resetBackend,
-  deleteSecureKeyAsync,
-  setSecureKeyAsync,
-} from "../../../security/secure-keys.js";
-import { runAssistantCommand } from "../../__tests__/run-assistant-command.js";
-
-const ASSISTANT_ID = "aaaaaaaa-bbbb-cccc-dddd-eeeeeeeeeeee";
-const ADDRESS_ID = "550e8400-e29b-41d4-a716-446655440000";
-const ADDRESS = "mybot@vellum.me";
-const API_KEY_CREDENTIAL = credentialKey("vellum", "assistant_api_key");
-
-/**
- * Platform API calls made against the configured platform base URL. Filters
- * out auxiliary fetches (feature-flag sync to the gateway) that are performed
- * during CLI bootstrap and aren't relevant to these tests.
- */
-function getPlatformCalls(): ReturnType<typeof getMockFetchCalls> {
-  return getMockFetchCalls().filter((c) =>
-    c.path.startsWith("https://test-platform.vellum.ai/"),
-  );
-}
-
-function mockListAddresses(
-  addresses: { id: string; address: string }[] = [
-    { id: ADDRESS_ID, address: ADDRESS },
-  ],
-  status = 200,
-): void {
-  mockFetch("/email-addresses/", {}, { body: { results: addresses }, status });
-}
-
-function mockSendSuccess(deliveryId = "del_abc123", status = 202): void {
-  mockFetch(
-    "/runtime-proxy/email/send/",
-    { method: "POST" },
-    { body: { delivery_id: deliveryId, status: "accepted" }, status },
-  );
-}
-
-let savedCesUrl: string | undefined;
-let savedContainerized: string | undefined;
-
-beforeEach(async () => {
-  process.exitCode = 0;
-
-  // Force encrypted-store backend so setSecureKeyAsync works in sandbox.
-  // test-preload isolates the encrypted store to a per-file temp dir, so
-  // set/delete calls here do NOT touch the developer's real
-  // ~/.vellum/protected/keys.enc.
-  savedCesUrl = process.env.CES_CREDENTIAL_URL;
-  savedContainerized = process.env.IS_CONTAINERIZED;
-  delete process.env.CES_CREDENTIAL_URL;
-  delete process.env.IS_CONTAINERIZED;
-
-  _resetBackend();
-  resetMockFetch();
-  _setOverridesForTesting({ "email-channel": true });
-  setPlatformAssistantId(ASSISTANT_ID);
-  await setSecureKeyAsync(API_KEY_CREDENTIAL, "test-api-key");
-});
-
-afterEach(() => {
-  resetMockFetch();
-  _setOverridesForTesting({});
-  setPlatformAssistantId(undefined);
-  _resetBackend();
-
-  if (savedCesUrl !== undefined) process.env.CES_CREDENTIAL_URL = savedCesUrl;
-  else delete process.env.CES_CREDENTIAL_URL;
-  if (savedContainerized !== undefined)
-    process.env.IS_CONTAINERIZED = savedContainerized;
-  else delete process.env.IS_CONTAINERIZED;
-});
-
-describe("assistant email send", () => {
-  test("successful send with --body and --subject", async () => {
-    mockListAddresses();
-    mockSendSuccess();
-
-    await runAssistantCommand(
-      "email",
-      "send",
-      "user@example.com",
-      "-s",
-      "Hello",
-      "-b",
-      "Hi there",
-    );
-
-    const calls = getPlatformCalls();
-    expect(calls).toHaveLength(2);
-
-    // First call: list addresses to resolve "from"
-    expect(calls[0].path).toContain(
-      `/v1/assistants/${ASSISTANT_ID}/email-addresses/`,
-    );
-
-    // Second call: send via runtime proxy
-    expect(calls[1].path).toContain("/v1/runtime-proxy/email/send/");
-    expect(calls[1].init.method).toBe("POST");
-
-    const payload = JSON.parse(calls[1].init.body as string);
-    expect(payload.to).toEqual(["user@example.com"]);
-    expect(payload.from_address).toBe(ADDRESS);
-    expect(payload.subject).toBe("Hello");
-    expect(payload.text).toBe("Hi there");
-
-    expect(process.exitCode).toBe(0);
-  });
-
-  test("--json outputs structured response", async () => {
-    mockListAddresses();
-    mockSendSuccess("del_xyz789");
-
-    const output = await runAssistantCommand(
-      "email",
-      "--json",
-      "send",
-      "user@example.com",
-      "-s",
-      "Test",
-      "-b",
-      "Body",
-    );
-
-    const parsed = JSON.parse(output.trim());
-    expect(parsed.delivery_id).toBe("del_xyz789");
-    expect(parsed.status).toBe("accepted");
-    expect(process.exitCode).toBe(0);
-  });
-
-  test("no registered address returns error", async () => {
-    mockListAddresses([]);
-
-    const output = await runAssistantCommand(
-      "email",
-      "--json",
-      "send",
-      "user@example.com",
-      "-s",
-      "Hello",
-      "-b",
-      "Body",
-    );
-
-    expect(process.exitCode).toBe(1);
-    const parsed = JSON.parse(output.trim());
-    expect(parsed.error).toContain("No email address registered");
-  });
-
-  test("missing body returns error", async () => {
-    // Force isTTY to true so stdin fallback is skipped
-    const origIsTTY = process.stdin.isTTY;
-    process.stdin.isTTY = true as unknown as boolean;
-
-    mockListAddresses();
-
-    const output = await runAssistantCommand(
-      "email",
-      "--json",
-      "send",
-      "user@example.com",
-      "-s",
-      "Hello",
-    );
-
-    process.stdin.isTTY = origIsTTY;
-
-    expect(process.exitCode).toBe(1);
-    const parsed = JSON.parse(output.trim());
-    expect(parsed.error).toContain("Email body is required");
-  });
-
-  test("send endpoint failure surfaces error detail", async () => {
-    mockListAddresses();
-    mockFetch(
-      "/runtime-proxy/email/send/",
-      { method: "POST" },
-      {
-        body: { detail: "From address not owned by this assistant." },
-        status: 403,
-      },
-    );
-
-    const output = await runAssistantCommand(
-      "email",
-      "--json",
-      "send",
-      "user@example.com",
-      "-s",
-      "Hello",
-      "-b",
-      "Body",
-    );
-
-    expect(process.exitCode).toBe(1);
-    const parsed = JSON.parse(output.trim());
-    expect(parsed.error).toContain("not owned by this assistant");
-  });
-
-  test("list addresses failure returns error", async () => {
-    mockFetch(
-      "/email-addresses/",
-      {},
-      { body: { detail: "Internal server error" }, status: 500 },
-    );
-
-    const output = await runAssistantCommand(
-      "email",
-      "--json",
-      "send",
-      "user@example.com",
-      "-s",
-      "Hello",
-      "-b",
-      "Body",
-    );
-
-    expect(process.exitCode).toBe(1);
-    const parsed = JSON.parse(output.trim());
-    expect(parsed.error).toContain("Failed to list email addresses");
-  });
-
-  test("missing platform credentials returns error", async () => {
-    // Remove the API key so VellumPlatformClient.create() returns null.
-    await deleteSecureKeyAsync(API_KEY_CREDENTIAL);
-    _resetBackend();
-    setPlatformAssistantId(undefined);
-
-    const output = await runAssistantCommand(
-      "email",
-      "--json",
-      "send",
-      "user@example.com",
-      "-s",
-      "Hello",
-      "-b",
-      "Body",
-    );
-
-    expect(process.exitCode).toBe(1);
-    const parsed = JSON.parse(output.trim());
-    expect(parsed.error).toContain("Platform credentials not configured");
-  });
-
-  test("missing assistant ID returns error", async () => {
-    setPlatformAssistantId("");
-
-    const output = await runAssistantCommand(
-      "email",
-      "--json",
-      "send",
-      "user@example.com",
-      "-s",
-      "Hello",
-      "-b",
-      "Body",
-    );
-
-    expect(process.exitCode).toBe(1);
-    const parsed = JSON.parse(output.trim());
-    expect(parsed.error).toContain("Assistant ID");
-  });
-
-  test("send without subject omits subject from payload", async () => {
-    mockListAddresses();
-    mockSendSuccess();
-
-    await runAssistantCommand(
-      "email",
-      "send",
-      "user@example.com",
-      "-b",
-      "Body only, no subject",
-    );
-
-    const calls = getPlatformCalls();
-    const payload = JSON.parse(calls[1].init.body as string);
-    expect(payload.subject).toBeUndefined();
-    expect(payload.text).toBe("Body only, no subject");
-    expect(process.exitCode).toBe(0);
-  });
-
-  test("suppressed recipient error is surfaced", async () => {
-    mockListAddresses();
-    mockFetch(
-      "/runtime-proxy/email/send/",
-      { method: "POST" },
-      {
-        body: {
-          detail:
-            "Recipient user@example.com is suppressed due to prior bounce or spam complaint.",
-        },
-        status: 422,
-      },
-    );
-
-    const output = await runAssistantCommand(
-      "email",
-      "--json",
-      "send",
-      "user@example.com",
-      "-s",
-      "Hello",
-      "-b",
-      "Body",
-    );
-
-    expect(process.exitCode).toBe(1);
-    const parsed = JSON.parse(output.trim());
-    expect(parsed.error).toContain("suppressed");
-  });
-
-  test("multiple to recipients are sent as array", async () => {
-    mockListAddresses();
-    mockSendSuccess();
-
-    await runAssistantCommand(
-      "email",
-      "send",
-      "alice@example.com",
-      "bob@example.com",
-      "-s",
-      "Team",
-      "-b",
-      "Hi all",
-    );
-
-    const calls = getPlatformCalls();
-    const payload = JSON.parse(calls[1].init.body as string);
-    expect(payload.to).toEqual(["alice@example.com", "bob@example.com"]);
-    expect(process.exitCode).toBe(0);
-  });
-
-  test("--cc and --bcc are included in payload", async () => {
-    mockListAddresses();
-    mockSendSuccess();
-
-    await runAssistantCommand(
-      "email",
-      "send",
-      "alice@example.com",
-      "--cc",
-      "bob@example.com",
-      "--bcc",
-      "carol@example.com",
-      "-s",
-      "FYI",
-      "-b",
-      "See below",
-    );
-
-    const calls = getPlatformCalls();
-    const payload = JSON.parse(calls[1].init.body as string);
-    expect(payload.to).toEqual(["alice@example.com"]);
-    expect(payload.cc).toEqual(["bob@example.com"]);
-    expect(payload.bcc).toEqual(["carol@example.com"]);
-    expect(process.exitCode).toBe(0);
-  });
-
-  test("multiple --cc flags accumulate", async () => {
-    mockListAddresses();
-    mockSendSuccess();
-
-    await runAssistantCommand(
-      "email",
-      "send",
-      "alice@example.com",
-      "--cc",
-      "bob@example.com",
-      "--cc",
-      "dave@example.com",
-      "-s",
-      "Team",
-      "-b",
-      "Hi",
-    );
-
-    const calls = getPlatformCalls();
-    const payload = JSON.parse(calls[1].init.body as string);
-    expect(payload.cc).toEqual(["bob@example.com", "dave@example.com"]);
-    expect(process.exitCode).toBe(0);
-  });
-
-  test("cc and bcc omitted from payload when not provided", async () => {
-    mockListAddresses();
-    mockSendSuccess();
-
-    await runAssistantCommand(
-      "email",
-      "send",
-      "user@example.com",
-      "-s",
-      "Hi",
-      "-b",
-      "Hello",
-    );
-
-    const calls = getPlatformCalls();
-    const payload = JSON.parse(calls[1].init.body as string);
-    expect(payload.cc).toBeUndefined();
-    expect(payload.bcc).toBeUndefined();
-    expect(process.exitCode).toBe(0);
-  });
-});

package/src/cli/commands/__tests__/email-status.test.ts

@@ -1,185 +0,0 @@
-import { afterEach, beforeEach, describe, expect, test } from "bun:test";
-
-import {
-  getMockFetchCalls,
-  mockFetch,
-  resetMockFetch,
-} from "../../../__tests__/mock-fetch.js";
-import { _setOverridesForTesting } from "../../../config/assistant-feature-flags.js";
-import { setPlatformAssistantId } from "../../../config/env.js";
-import { credentialKey } from "../../../security/credential-key.js";
-import {
-  _resetBackend,
-  deleteSecureKeyAsync,
-  setSecureKeyAsync,
-} from "../../../security/secure-keys.js";
-import { runAssistantCommand } from "../../__tests__/run-assistant-command.js";
-
-const ASSISTANT_ID = "aaaaaaaa-bbbb-cccc-dddd-eeeeeeeeeeee";
-const ADDRESS_ID = "550e8400-e29b-41d4-a716-446655440000";
-const ADDRESS = "mybot@vellum.me";
-const API_KEY_CREDENTIAL = credentialKey("vellum", "assistant_api_key");
-
-function mockListAddresses(
-  addresses: { id: string; address: string }[] = [
-    { id: ADDRESS_ID, address: ADDRESS },
-  ],
-  status = 200,
-): void {
-  mockFetch("/email-addresses/", {}, { body: { results: addresses }, status });
-}
-
-function mockStatusSuccess(
-  overrides?: Partial<{
-    address: string;
-    status: string;
-    usage: { sent_today: number; daily_limit: number; received_today: number };
-  }>,
-): void {
-  mockFetch(
-    `/email-addresses/${ADDRESS_ID}/status/`,
-    {},
-    {
-      body: {
-        address: overrides?.address ?? ADDRESS,
-        status: overrides?.status ?? "active",
-        usage: overrides?.usage ?? {
-          sent_today: 12,
-          daily_limit: 100,
-          received_today: 5,
-        },
-      },
-      status: 200,
-    },
-  );
-}
-
-let savedCesUrl: string | undefined;
-let savedContainerized: string | undefined;
-
-beforeEach(async () => {
-  process.exitCode = 0;
-
-  // Force encrypted-store backend so setSecureKeyAsync works in sandbox
-  savedCesUrl = process.env.CES_CREDENTIAL_URL;
-  savedContainerized = process.env.IS_CONTAINERIZED;
-  delete process.env.CES_CREDENTIAL_URL;
-  delete process.env.IS_CONTAINERIZED;
-
-  _resetBackend();
-  resetMockFetch();
-  _setOverridesForTesting({ "email-channel": true });
-  setPlatformAssistantId(ASSISTANT_ID);
-  await setSecureKeyAsync(API_KEY_CREDENTIAL, "test-api-key");
-});
-
-afterEach(() => {
-  resetMockFetch();
-  _setOverridesForTesting({});
-  setPlatformAssistantId(undefined);
-  _resetBackend();
-
-  // Restore env
-  if (savedCesUrl !== undefined) process.env.CES_CREDENTIAL_URL = savedCesUrl;
-  else delete process.env.CES_CREDENTIAL_URL;
-  if (savedContainerized !== undefined)
-    process.env.IS_CONTAINERIZED = savedContainerized;
-  else delete process.env.IS_CONTAINERIZED;
-});
-
-describe("assistant email status", () => {
-  test("successful status shows address and usage", async () => {
-    mockListAddresses();
-    mockStatusSuccess();
-
-    const output = await runAssistantCommand("email", "--json", "status");
-
-    const parsed = JSON.parse(output.trim());
-    expect(parsed.address).toBe(ADDRESS);
-    expect(parsed.status).toBe("active");
-    expect(parsed.usage.sent_today).toBe(12);
-    expect(parsed.usage.daily_limit).toBe(100);
-    expect(parsed.usage.received_today).toBe(5);
-    expect(process.exitCode).toBe(0);
-  });
-
-  test("calls correct URLs in order", async () => {
-    mockListAddresses();
-    mockStatusSuccess();
-
-    await runAssistantCommand("email", "--json", "status");
-
-    // Filter out the feature-flags bootstrap call (initFeatureFlagOverrides
-    // fires on every CLI invocation to pre-populate the flag cache).
-    const calls = getMockFetchCalls().filter(
-      (c) => !c.path.includes("/v1/feature-flags"),
-    );
-    expect(calls).toHaveLength(2);
-    expect(calls[0].path).toContain(
-      `/v1/assistants/${ASSISTANT_ID}/email-addresses/`,
-    );
-    expect(calls[1].path).toContain(
-      `/v1/assistants/${ASSISTANT_ID}/email-addresses/${ADDRESS_ID}/status/`,
-    );
-  });
-
-  test("no registered address returns error", async () => {
-    mockListAddresses([]);
-
-    const output = await runAssistantCommand("email", "--json", "status");
-
-    expect(process.exitCode).toBe(1);
-    const parsed = JSON.parse(output.trim());
-    expect(parsed.error).toContain("No email address registered");
-  });
-
-  test("list endpoint failure returns error", async () => {
-    mockFetch(
-      "/email-addresses/",
-      {},
-      { body: { detail: "Internal server error" }, status: 500 },
-    );
-
-    const output = await runAssistantCommand("email", "--json", "status");
-
-    expect(process.exitCode).toBe(1);
-    const parsed = JSON.parse(output.trim());
-    expect(parsed.error).toContain("Failed to list email addresses");
-  });
-
-  test("status endpoint failure returns error", async () => {
-    mockListAddresses();
-    mockFetch(
-      `/email-addresses/${ADDRESS_ID}/status/`,
-      {},
-      { body: { detail: "Service unavailable" }, status: 503 },
-    );
-
-    const output = await runAssistantCommand("email", "--json", "status");
-
-    expect(process.exitCode).toBe(1);
-    const parsed = JSON.parse(output.trim());
-    expect(parsed.error).toContain("Service unavailable");
-  });
-
-  test("missing platform credentials returns error", async () => {
-    // Delete the API key so create() returns null
-    await deleteSecureKeyAsync(API_KEY_CREDENTIAL);
-
-    const output = await runAssistantCommand("email", "--json", "status");
-
-    expect(process.exitCode).toBe(1);
-    const parsed = JSON.parse(output.trim());
-    expect(parsed.error).toContain("Platform credentials not configured");
-  });
-
-  test("missing assistant ID returns error", async () => {
-    setPlatformAssistantId("");
-
-    const output = await runAssistantCommand("email", "--json", "status");
-
-    expect(process.exitCode).toBe(1);
-    const parsed = JSON.parse(output.trim());
-    expect(parsed.error).toContain("Assistant ID");
-  });
-});