@vellumai/assistant 0.6.4 → 0.6.5

package/src/providers/__tests__/context-overflow-error.test.ts

@@ -0,0 +1,328 @@
+import { describe, expect, test } from "bun:test";
+
+// SDK error classes are constructed via their exports so we produce
+// real instances that pass the provider clients' `instanceof` checks.
+import Anthropic from "@anthropic-ai/sdk";
+import { ApiError as GeminiApiError } from "@google/genai";
+import OpenAI from "openai";
+
+import { detectAnthropicContextOverflow } from "../anthropic/client.js";
+import { detectGeminiContextOverflow } from "../gemini/client.js";
+import { detectOpenAICompatibleContextOverflow } from "../openai/chat-completions-provider.js";
+import { ContextOverflowError, isContextOverflowError } from "../types.js";
+
+describe("ContextOverflowError", () => {
+  test("constructs with provider, actualTokens, maxTokens", () => {
+    const err = new ContextOverflowError("msg", "anthropic", {
+      actualTokens: 242201,
+      maxTokens: 200000,
+    });
+    expect(err).toBeInstanceOf(Error);
+    expect(err).toBeInstanceOf(ContextOverflowError);
+    expect(err.name).toBe("ContextOverflowError");
+    expect(err.message).toBe("msg");
+    expect(err.provider).toBe("anthropic");
+    expect(err.actualTokens).toBe(242201);
+    expect(err.maxTokens).toBe(200000);
+  });
+
+  test("extends ProviderError so existing classifiers see it", async () => {
+    const { ProviderError } = await import("../../util/errors.js");
+    const err = new ContextOverflowError("m", "anthropic", {});
+    expect(err).toBeInstanceOf(ProviderError);
+  });
+
+  test("defaults statusCode to 400 and accepts override", () => {
+    const defaulted = new ContextOverflowError("m", "anthropic", {});
+    expect(defaulted.statusCode).toBe(400);
+    const overridden = new ContextOverflowError("m", "openai", {
+      statusCode: 413,
+    });
+    expect(overridden.statusCode).toBe(413);
+  });
+
+  test("carries cause when supplied", () => {
+    const cause = new Error("underlying");
+    const err = new ContextOverflowError("outer", "openai", { cause });
+    expect((err as Error & { cause?: unknown }).cause).toBe(cause);
+  });
+
+  test("omits optional fields when unset", () => {
+    const err = new ContextOverflowError("m", "gemini");
+    expect(err.actualTokens).toBeUndefined();
+    expect(err.maxTokens).toBeUndefined();
+    expect(err.provider).toBe("gemini");
+  });
+});
+
+describe("isContextOverflowError", () => {
+  test("returns true for a native ContextOverflowError", () => {
+    const err = new ContextOverflowError("m", "anthropic");
+    expect(isContextOverflowError(err)).toBe(true);
+  });
+
+  test("returns false for a plain Error", () => {
+    expect(isContextOverflowError(new Error("boom"))).toBe(false);
+  });
+
+  test("returns false for a ProviderError that is not ContextOverflowError", async () => {
+    const { ProviderError } = await import("../../util/errors.js");
+    const err = new ProviderError("generic provider failure", "anthropic", 500);
+    expect(isContextOverflowError(err)).toBe(false);
+  });
+
+  test("returns false for null / undefined / primitives", () => {
+    expect(isContextOverflowError(null)).toBe(false);
+    expect(isContextOverflowError(undefined)).toBe(false);
+    expect(isContextOverflowError("string")).toBe(false);
+    expect(isContextOverflowError(42)).toBe(false);
+  });
+});
+
+// ── Per-provider error-body parsing ────────────────────────────────────
+
+/**
+ * Construct a real Anthropic APIError so provider-side detection can match
+ * the production code path (which checks `instanceof Anthropic.APIError`).
+ */
+function buildAnthropicApiError(
+  status: number,
+  body: Record<string, unknown>,
+): InstanceType<typeof Anthropic.APIError> {
+  return new Anthropic.APIError(status, body, undefined, new Headers());
+}
+
+function buildOpenAIApiError(
+  status: number,
+  body: Record<string, unknown>,
+  topLevelMessage?: string,
+): InstanceType<typeof OpenAI.APIError> {
+  return new OpenAI.APIError(status, body, topLevelMessage, new Headers());
+}
+
+describe("detectAnthropicContextOverflow", () => {
+  test("matches the canonical nested body shape and extracts both counts", () => {
+    const err = buildAnthropicApiError(400, {
+      type: "error",
+      error: {
+        type: "invalid_request_error",
+        message: "prompt is too long: 242201 tokens > 200000 maximum",
+      },
+    });
+    const out = detectAnthropicContextOverflow(err);
+    expect(out).not.toBeNull();
+    expect(out?.actualTokens).toBe(242201);
+    expect(out?.maxTokens).toBe(200000);
+  });
+
+  test("handles comma-separated numbers in the message", () => {
+    const err = buildAnthropicApiError(400, {
+      type: "error",
+      error: {
+        type: "invalid_request_error",
+        message: "prompt is too long: 242,201 tokens > 200,000 maximum",
+      },
+    });
+    const out = detectAnthropicContextOverflow(err);
+    expect(out?.actualTokens).toBe(242201);
+    expect(out?.maxTokens).toBe(200000);
+  });
+
+  test("returns null for non-overflow 400 errors", () => {
+    const err = buildAnthropicApiError(400, {
+      type: "error",
+      error: {
+        type: "invalid_request_error",
+        message: "messages.0: role must be user or assistant",
+      },
+    });
+    expect(detectAnthropicContextOverflow(err)).toBeNull();
+  });
+
+  test("returns null for non-400 statuses", () => {
+    const err = buildAnthropicApiError(429, {
+      type: "error",
+      error: { type: "rate_limit_error", message: "prompt is too long" },
+    });
+    expect(detectAnthropicContextOverflow(err)).toBeNull();
+  });
+
+  test("matches even when only the top-level JSON message carries the text", () => {
+    // If the SDK falls back to JSON.stringify, the top-level message string
+    // will still contain the "prompt is too long" text.
+    const err = buildAnthropicApiError(400, {
+      type: "error",
+      error: {
+        type: "invalid_request_error",
+        message: "prompt is too long: 242201 tokens > 200000 maximum",
+      },
+    });
+    // The SDK constructor auto-builds err.message; just sanity-check it
+    // contains the canonical phrase.
+    expect(err.message).toContain("prompt is too long");
+    const out = detectAnthropicContextOverflow(err);
+    expect(out?.actualTokens).toBe(242201);
+  });
+
+  test("matches when no token counts are parseable (returns empty extraction)", () => {
+    const err = buildAnthropicApiError(400, {
+      type: "error",
+      error: {
+        type: "invalid_request_error",
+        message: "prompt is too long", // no numbers
+      },
+    });
+    const out = detectAnthropicContextOverflow(err);
+    expect(out).not.toBeNull();
+    expect(out?.actualTokens).toBeUndefined();
+    expect(out?.maxTokens).toBeUndefined();
+  });
+});
+
+describe("detectOpenAICompatibleContextOverflow", () => {
+  test("matches the canonical OpenAI context_length_exceeded body", () => {
+    const err = buildOpenAIApiError(400, {
+      message:
+        "This model's maximum context length is 128000 tokens. However, your messages resulted in 150000 tokens.",
+      type: "invalid_request_error",
+      code: "context_length_exceeded",
+    });
+    const out = detectOpenAICompatibleContextOverflow(err);
+    expect(out).not.toBeNull();
+  });
+
+  test("matches providers that omit `code` but include the phrase in message", () => {
+    // OpenRouter / Fireworks / Ollama often don't set `code` — rely on message.
+    const err = buildOpenAIApiError(400, {
+      message: "context_length_exceeded: the request has too many input tokens",
+      type: "invalid_request_error",
+    });
+    const out = detectOpenAICompatibleContextOverflow(err);
+    expect(out).not.toBeNull();
+  });
+
+  test("matches 413 status for request-too-large variants", () => {
+    const err = buildOpenAIApiError(413, {
+      message: "Request too large: input too long",
+      type: "invalid_request_error",
+    });
+    const out = detectOpenAICompatibleContextOverflow(err);
+    expect(out).not.toBeNull();
+  });
+
+  test("extracts actual/max tokens when provider surfaces them", () => {
+    const err = buildOpenAIApiError(400, {
+      message:
+        "maximum context length exceeded: 150000 > 128000 tokens in the request",
+      type: "invalid_request_error",
+      code: "context_length_exceeded",
+    });
+    const out = detectOpenAICompatibleContextOverflow(err);
+    expect(out?.actualTokens).toBe(150000);
+    expect(out?.maxTokens).toBe(128000);
+  });
+
+  test("returns null for non-overflow 400 errors", () => {
+    const err = buildOpenAIApiError(400, {
+      message: "Invalid model specified",
+      type: "invalid_request_error",
+      code: "invalid_model",
+    });
+    expect(detectOpenAICompatibleContextOverflow(err)).toBeNull();
+  });
+
+  test("returns null for non-4xx statuses", () => {
+    const err = buildOpenAIApiError(429, {
+      message: "Rate limit exceeded",
+      type: "rate_limit_error",
+    });
+    expect(detectOpenAICompatibleContextOverflow(err)).toBeNull();
+  });
+
+  test("matches 'too many input tokens' variant emitted by some OpenAI-compatible providers", () => {
+    const err = buildOpenAIApiError(400, {
+      message: "too many input tokens: 250000",
+      type: "invalid_request_error",
+    });
+    expect(detectOpenAICompatibleContextOverflow(err)).not.toBeNull();
+  });
+});
+
+describe("detectGeminiContextOverflow", () => {
+  // Smoke-check that the Gemini SDK's ApiError still surfaces status+message
+  // the way the detector expects.
+  test("Gemini ApiError exposes status + message", () => {
+    const err = new GeminiApiError({
+      status: 400,
+      message:
+        "RESOURCE_EXHAUSTED: The input token count exceeds the maximum number of tokens allowed",
+    });
+    expect(err.status).toBe(400);
+    expect(err.message).toContain("RESOURCE_EXHAUSTED");
+  });
+
+  test("429 quota-exceeded rate-limit returns null (NOT overflow)", () => {
+    // Vertex/Generative Language API returns 429 + RESOURCE_EXHAUSTED for
+    // rate-limit quota exhaustion. This must flow through ProviderError retry
+    // path, not be misclassified as context overflow.
+    const err = new GeminiApiError({
+      status: 429,
+      message:
+        "RESOURCE_EXHAUSTED: Quota exceeded for quota metric 'CustomClass' of service 'generativelanguage.googleapis.com'",
+    });
+    expect(detectGeminiContextOverflow(err)).toBeNull();
+  });
+
+  test("429 with token-count phrase returns overflow (Vertex path)", () => {
+    // Vertex surfaces context-overflow with 429 + RESOURCE_EXHAUSTED plus
+    // a token-count-specific phrase. This is the legitimate overflow path.
+    const err = new GeminiApiError({
+      status: 429,
+      message:
+        "RESOURCE_EXHAUSTED: The input token count 250000 exceeds the maximum number of tokens allowed",
+    });
+    const out = detectGeminiContextOverflow(err);
+    // Non-null signals that the overflow phrase matched; token counts are
+    // best-effort (this message uses "exceeds" prose, not "N > M" format).
+    expect(out).not.toBeNull();
+  });
+
+  test("400 with bare RESOURCE_EXHAUSTED returns overflow (preserves existing behavior)", () => {
+    // Generative Language API path: 400 INVALID_ARGUMENT with just the
+    // RESOURCE_EXHAUSTED phrase is an overflow signal on its own.
+    const err = new GeminiApiError({
+      status: 400,
+      message: "RESOURCE_EXHAUSTED",
+    });
+    expect(detectGeminiContextOverflow(err)).not.toBeNull();
+  });
+
+  test("429 with generic message (no token-specific phrase) returns null", () => {
+    // Any 429 without a token/context phrase should fall through to retry.
+    const err = new GeminiApiError({
+      status: 429,
+      message: "Too many requests — please try again later",
+    });
+    expect(detectGeminiContextOverflow(err)).toBeNull();
+  });
+
+  test("400 with token-count phrase extracts counts when using N > M format", () => {
+    const err = new GeminiApiError({
+      status: 400,
+      message:
+        "The input token count exceeds the maximum: 300000 tokens > 200000 maximum",
+    });
+    const out = detectGeminiContextOverflow(err);
+    expect(out).not.toBeNull();
+    expect(out?.actualTokens).toBe(300000);
+    expect(out?.maxTokens).toBe(200000);
+  });
+
+  test("non-4xx statuses return null", () => {
+    const err = new GeminiApiError({
+      status: 500,
+      message: "RESOURCE_EXHAUSTED: The input token count exceeds",
+    });
+    expect(detectGeminiContextOverflow(err)).toBeNull();
+  });
+});

package/src/providers/__tests__/provider-env-vars.test.ts

@@ -0,0 +1,102 @@
+import { readFileSync } from "node:fs";
+import { join } from "node:path";
+import { describe, expect, test } from "bun:test";
+
+import {
+  getAnyProviderEnvVar,
+  getLlmProviderEnvVar,
+  getSearchProviderEnvVar,
+  SEARCH_PROVIDER_ENV_VAR_NAMES,
+} from "../provider-env-vars.js";
+
+describe("getLlmProviderEnvVar", () => {
+  test("returns ANTHROPIC_API_KEY for anthropic", () => {
+    expect(getLlmProviderEnvVar("anthropic")).toBe("ANTHROPIC_API_KEY");
+  });
+
+  test("returns OPENAI_API_KEY for openai", () => {
+    expect(getLlmProviderEnvVar("openai")).toBe("OPENAI_API_KEY");
+  });
+
+  test("returns GEMINI_API_KEY for gemini", () => {
+    expect(getLlmProviderEnvVar("gemini")).toBe("GEMINI_API_KEY");
+  });
+
+  test("returns FIREWORKS_API_KEY for fireworks", () => {
+    expect(getLlmProviderEnvVar("fireworks")).toBe("FIREWORKS_API_KEY");
+  });
+
+  test("returns OPENROUTER_API_KEY for openrouter", () => {
+    expect(getLlmProviderEnvVar("openrouter")).toBe("OPENROUTER_API_KEY");
+  });
+
+  test("returns undefined for ollama (keyless provider)", () => {
+    expect(getLlmProviderEnvVar("ollama")).toBeUndefined();
+  });
+
+  test("returns undefined for search providers (out of scope)", () => {
+    expect(getLlmProviderEnvVar("brave")).toBeUndefined();
+    expect(getLlmProviderEnvVar("perplexity")).toBeUndefined();
+  });
+
+  test("returns undefined for unknown provider", () => {
+    expect(getLlmProviderEnvVar("unknown-provider")).toBeUndefined();
+  });
+});
+
+describe("getSearchProviderEnvVar", () => {
+  test("returns BRAVE_API_KEY for brave", () => {
+    expect(getSearchProviderEnvVar("brave")).toBe("BRAVE_API_KEY");
+  });
+
+  test("returns PERPLEXITY_API_KEY for perplexity", () => {
+    expect(getSearchProviderEnvVar("perplexity")).toBe("PERPLEXITY_API_KEY");
+  });
+
+  test("returns undefined for LLM providers (out of scope)", () => {
+    expect(getSearchProviderEnvVar("anthropic")).toBeUndefined();
+    expect(getSearchProviderEnvVar("openai")).toBeUndefined();
+  });
+
+  test("returns undefined for unknown provider", () => {
+    expect(getSearchProviderEnvVar("unknown-provider")).toBeUndefined();
+  });
+});
+
+describe("getAnyProviderEnvVar", () => {
+  test("returns LLM env var for LLM providers", () => {
+    expect(getAnyProviderEnvVar("anthropic")).toBe("ANTHROPIC_API_KEY");
+    expect(getAnyProviderEnvVar("openai")).toBe("OPENAI_API_KEY");
+  });
+
+  test("returns search env var for search providers", () => {
+    expect(getAnyProviderEnvVar("brave")).toBe("BRAVE_API_KEY");
+    expect(getAnyProviderEnvVar("perplexity")).toBe("PERPLEXITY_API_KEY");
+  });
+
+  test("returns undefined for ollama (keyless LLM provider)", () => {
+    expect(getAnyProviderEnvVar("ollama")).toBeUndefined();
+  });
+
+  test("returns undefined for unknown provider", () => {
+    expect(getAnyProviderEnvVar("unknown")).toBeUndefined();
+  });
+});
+
+describe("SEARCH_PROVIDER_ENV_VAR_NAMES parity with meta/provider-env-vars.json", () => {
+  // The daemon inlines the search-provider env-var map rather than reading
+  // meta/provider-env-vars.json at runtime (compiled binary, no reliable
+  // repo-relative path). This parity check prevents drift: the inline map
+  // must stay in sync with the canonical JSON file consumed by the macOS
+  // client bundle and the CLI cloud-infra flows.
+  test("inline map matches meta/provider-env-vars.json providers", () => {
+    const repoRoot = join(process.cwd(), "..");
+    const metaPath = join(repoRoot, "meta", "provider-env-vars.json");
+    const raw = readFileSync(metaPath, "utf-8");
+    const parsed = JSON.parse(raw) as {
+      version: number;
+      providers: Record<string, string>;
+    };
+    expect(SEARCH_PROVIDER_ENV_VAR_NAMES).toEqual(parsed.providers);
+  });
+});