whale-code 6.5.10 → 6.6.0

package/dist/cli/services/__tests__/telemetry.part2.test.js

@@ -1,195 +0,0 @@
-import { describe, it, expect, beforeEach, vi } from "vitest";
-
-// ============================================================================
-// Mocks
-// ============================================================================
-
-const mockCreateAuthenticatedClient = vi.fn().mockReturnValue({
-  from: vi.fn()
-});
-const mockGetValidToken = vi.fn().mockResolvedValue(null);
-const mockResolveConfig = vi.fn().mockReturnValue({
-  supabaseUrl: "https://test.supabase.co",
-  supabaseKey: "test-key",
-  serverUrl: "https://whale-agent.fly.dev",
-  storeId: "store-default"
-});
-const mockLoadConfig = vi.fn().mockReturnValue({
-  user_id: "uid-1",
-  email: "u@test.com"
-});
-const mockCaptureError = vi.fn();
-
-// Mock fetch globally
-const mockFetch = vi.fn().mockResolvedValue({
-  ok: true,
-  json: async () => ({
-    success: true,
-    queued: 1
-  })
-});
-vi.stubGlobal("fetch", mockFetch);
-vi.mock("@supabase/supabase-js", () => ({
-  createClient: vi.fn()
-}));
-vi.mock("../config-store.js", () => ({
-  resolveConfig: () => mockResolveConfig(),
-  loadConfig: () => mockLoadConfig()
-}));
-vi.mock("../auth-service.js", () => ({
-  getValidToken: () => mockGetValidToken(),
-  createAuthenticatedClient: (...args) => mockCreateAuthenticatedClient(...args)
-}));
-vi.mock("../error-logger.js", () => ({
-  captureError: (...args) => mockCaptureError(...args)
-}));
-vi.mock("module", () => ({
-  createRequire: () => () => ({
-    version: "2.0.0-test"
-  })
-}));
-import { createTurnContext, logSpan, flushCliSpans } from "../telemetry.js";
-
-// Helper: log a span and flush immediately, returning the parsed spans from the fetch body
-async function logAndFlush(opts) {
-  logSpan(opts);
-  flushCliSpans();
-  // Wait for the async _sendSpans to complete
-  await vi.waitFor(() => {
-    expect(mockFetch).toHaveBeenCalled();
-  });
-  const lastCall = mockFetch.mock.calls[mockFetch.mock.calls.length - 1];
-  const body = JSON.parse(lastCall[1].body);
-  // Return the last span in the batch
-  return body.spans[body.spans.length - 1];
-}
-
-// ============================================================================
-// logSpan — spans are buffered and flushed to Fly.io server
-// ============================================================================
-
-describe("logSpan", () => {
-  beforeEach(() => {
-    vi.clearAllMocks();
-    mockResolveConfig.mockReturnValue({
-      supabaseUrl: "https://test.supabase.co",
-      supabaseKey: "test-key",
-      serverUrl: "https://whale-agent.fly.dev",
-      storeId: "store-default"
-    });
-  });
-  it("does not throw (fire-and-forget)", () => {
-    const ctx = createTurnContext();
-    expect(() => logSpan({
-      action: "test.action",
-      durationMs: 100,
-      context: ctx
-    })).not.toThrow();
-  });
-  it("sends spans to server via HTTP POST on flush", async () => {
-    const ctx = createTurnContext();
-    logSpan({
-      action: "test.insert",
-      durationMs: 50,
-      context: ctx
-    });
-    flushCliSpans();
-    await vi.waitFor(() => {
-      expect(mockFetch).toHaveBeenCalled();
-    });
-    const [url, options] = mockFetch.mock.calls[mockFetch.mock.calls.length - 1];
-    expect(url).toBe("https://whale-agent.fly.dev");
-    expect(options.method).toBe("POST");
-    expect(options.headers["Authorization"]).toBe("Bearer test-key");
-    const body = JSON.parse(options.body);
-    expect(body.mode).toBe("telemetry_ingest");
-    expect(body.spans.length).toBeGreaterThanOrEqual(1);
-    expect(body.spans.find(s => s.action === "test.insert")).toBeTruthy();
-  });
-  it("sets severity to error and status_code to ERROR when error present", async () => {
-    const ctx = createTurnContext();
-    const span = await logAndFlush({
-      action: "test.error",
-      durationMs: 10,
-      context: ctx,
-      error: "something failed"
-    });
-    expect(span.severity).toBe("error");
-    expect(span.status_code).toBe("ERROR");
-    expect(span.error_message).toBe("something failed");
-  });
-  it("bridges errors to captureError", async () => {
-    const ctx = createTurnContext();
-    logSpan({
-      action: "test.bridge",
-      durationMs: 10,
-      context: ctx,
-      error: "bridge test"
-    });
-    await vi.waitFor(() => {
-      expect(mockCaptureError).toHaveBeenCalledWith(expect.objectContaining({
-        errorType: "test.bridge",
-        errorMessage: "bridge test",
-        severity: "error"
-      }));
-    });
-  });
-  it("does not call captureError when no error", async () => {
-    const ctx = createTurnContext();
-    logSpan({
-      action: "test.ok",
-      durationMs: 10,
-      context: ctx
-    });
-    flushCliSpans();
-    await vi.waitFor(() => {
-      expect(mockFetch).toHaveBeenCalled();
-    });
-    expect(mockCaptureError).not.toHaveBeenCalled();
-  });
-  it("includes storeId from opts", async () => {
-    const ctx = createTurnContext();
-    const span = await logAndFlush({
-      action: "test.store",
-      durationMs: 10,
-      context: ctx,
-      storeId: "store-42"
-    });
-    expect(span.store_id).toBe("store-42");
-  });
-  it("passes details into the details field", async () => {
-    const ctx = createTurnContext();
-    const span = await logAndFlush({
-      action: "test.details",
-      durationMs: 5,
-      context: ctx,
-      details: {
-        foo: "bar"
-      }
-    });
-    expect(span.details).toMatchObject({
-      foo: "bar"
-    });
-  });
-  it("uses explicit severity override", async () => {
-    const ctx = createTurnContext();
-    const span = await logAndFlush({
-      action: "test.warn",
-      durationMs: 1,
-      context: ctx,
-      severity: "warn"
-    });
-    expect(span.severity).toBe("warn");
-  });
-  it("does not throw on fetch error", async () => {
-    mockFetch.mockRejectedValueOnce(new Error("network fail"));
-    const ctx = createTurnContext();
-    expect(() => logSpan({
-      action: "test.fetcherr",
-      durationMs: 1,
-      context: ctx
-    })).not.toThrow();
-    flushCliSpans();
-    // Should not throw even though fetch failed
-  });
-});

package/dist/cli/services/__tests__/telemetry.test.js

@@ -1,176 +0,0 @@
-import { describe, it, expect, vi } from "vitest";
-
-// ============================================================================
-// Mocks
-// ============================================================================
-
-const mockCreateAuthenticatedClient = vi.fn().mockReturnValue({
-  from: vi.fn()
-});
-const mockGetValidToken = vi.fn().mockResolvedValue(null);
-const mockResolveConfig = vi.fn().mockReturnValue({
-  supabaseUrl: "https://test.supabase.co",
-  supabaseKey: "test-key",
-  serverUrl: "https://whale-agent.fly.dev",
-  storeId: "store-default"
-});
-const mockLoadConfig = vi.fn().mockReturnValue({
-  user_id: "uid-1",
-  email: "u@test.com"
-});
-const mockCaptureError = vi.fn();
-
-// Mock fetch globally
-const mockFetch = vi.fn().mockResolvedValue({
-  ok: true,
-  json: async () => ({
-    success: true,
-    queued: 1
-  })
-});
-vi.stubGlobal("fetch", mockFetch);
-vi.mock("@supabase/supabase-js", () => ({
-  createClient: vi.fn()
-}));
-vi.mock("../config-store.js", () => ({
-  resolveConfig: () => mockResolveConfig(),
-  loadConfig: () => mockLoadConfig()
-}));
-vi.mock("../auth-service.js", () => ({
-  getValidToken: () => mockGetValidToken(),
-  createAuthenticatedClient: (...args) => mockCreateAuthenticatedClient(...args)
-}));
-vi.mock("../error-logger.js", () => ({
-  captureError: (...args) => mockCaptureError(...args)
-}));
-vi.mock("module", () => ({
-  createRequire: () => () => ({
-    version: "2.0.0-test"
-  })
-}));
-import { setConversationId, getConversationId, initializeTelemetryClient, generateTraceId, generateSpanId, nextTurn, createTurnContext, getTurnNumber, logSpan, flushCliSpans } from "../telemetry.js";
-
-// Helper: log a span and flush immediately, returning the parsed spans from the fetch body
-async function logAndFlush(opts) {
-  logSpan(opts);
-  flushCliSpans();
-  // Wait for the async _sendSpans to complete
-  await vi.waitFor(() => {
-    expect(mockFetch).toHaveBeenCalled();
-  });
-  const lastCall = mockFetch.mock.calls[mockFetch.mock.calls.length - 1];
-  const body = JSON.parse(lastCall[1].body);
-  // Return the last span in the batch
-  return body.spans[body.spans.length - 1];
-}
-
-// ============================================================================
-// Session state
-// ============================================================================
-
-describe("session state", () => {
-  it("getConversationId returns a UUID initially", () => {
-    const id = getConversationId();
-    expect(id).toMatch(/^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/);
-  });
-  it("setConversationId changes the conversation id", () => {
-    const custom = "custom-conv-123";
-    setConversationId(custom);
-    expect(getConversationId()).toBe(custom);
-  });
-});
-
-// ============================================================================
-// Turn management
-// ============================================================================
-
-describe("turn management", () => {
-  it("nextTurn increments the turn number", () => {
-    const before = getTurnNumber();
-    const n = nextTurn();
-    expect(n).toBe(before + 1);
-    expect(getTurnNumber()).toBe(n);
-  });
-  it("nextTurn increments on each call", () => {
-    const a = nextTurn();
-    const b = nextTurn();
-    const c = nextTurn();
-    expect(b).toBe(a + 1);
-    expect(c).toBe(b + 1);
-  });
-});
-
-// ============================================================================
-// Trace/Span ID generators
-// ============================================================================
-
-describe("generateTraceId", () => {
-  it("returns a UUID", () => {
-    const id = generateTraceId();
-    expect(id).toMatch(/^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/);
-  });
-  it("generates unique IDs", () => {
-    const a = generateTraceId();
-    const b = generateTraceId();
-    expect(a).not.toBe(b);
-  });
-});
-describe("generateSpanId", () => {
-  it("returns a 16-character hex string", () => {
-    const id = generateSpanId();
-    expect(id).toMatch(/^[0-9a-f]{16}$/);
-  });
-  it("generates unique IDs", () => {
-    const a = generateSpanId();
-    const b = generateSpanId();
-    expect(a).not.toBe(b);
-  });
-});
-
-// ============================================================================
-// createTurnContext
-// ============================================================================
-
-describe("createTurnContext", () => {
-  it("returns an ExecutionContext with defaults", () => {
-    const ctx = createTurnContext();
-    expect(ctx.source).toBe("whale_cli");
-    expect(ctx.serviceName).toBe("whale-code");
-    expect(ctx.traceId).toBeTruthy();
-    expect(ctx.spanId).toBeTruthy();
-    expect(ctx.traceFlags).toBe(1);
-    expect(ctx.conversationId).toBeTruthy();
-    expect(ctx.turnNumber).toBeDefined();
-  });
-  it("loads user info from config", () => {
-    const ctx = createTurnContext();
-    expect(ctx.userId).toBe("uid-1");
-    expect(ctx.userEmail).toBe("u@test.com");
-  });
-  it("applies overrides", () => {
-    const ctx = createTurnContext({
-      model: "sonnet",
-      inputTokens: 100,
-      outputTokens: 200
-    });
-    expect(ctx.model).toBe("sonnet");
-    expect(ctx.inputTokens).toBe(100);
-    expect(ctx.outputTokens).toBe(200);
-  });
-  it("overrides can replace source", () => {
-    const ctx = createTurnContext({
-      source: "custom_source"
-    });
-    expect(ctx.source).toBe("custom_source");
-  });
-});
-
-// ============================================================================
-// initializeTelemetryClient
-// ============================================================================
-
-describe("initializeTelemetryClient", () => {
-  it("does not throw", () => {
-    expect(() => initializeTelemetryClient("some-token")).not.toThrow();
-  });
-});

package/dist/cli/services/agent-loop-iteration.d.ts

@@ -1,13 +0,0 @@
-/**
- * Agent Loop Iteration — context trimming, compaction handling, provider compaction
- * Extracted from agent-loop.ts for file size limits.
- */
-import type Anthropic from "@anthropic-ai/sdk";
-import type { AgentLoopCallbacks } from "./agent-loop-types.js";
-import type { AgentEventEmitter } from "./agent-loop-types.js";
-export declare function trimContextForProvider(currentModel: string, cachedMessages: Anthropic.MessageParam[], prevIterationInputTokens: number, callbacks: AgentLoopCallbacks, emitter?: AgentEventEmitter): Anthropic.MessageParam[];
-export declare function handleProviderCompaction(currentModel: string, messages: Anthropic.MessageParam[], systemPrompt: string, inputTokens: number, compactionCount: number, token: string, callbacks: AgentLoopCallbacks, emitter?: AgentEventEmitter, turnCtx?: any, storeId?: string): Promise<{
-    compacted: boolean;
-    compactionCount: number;
-    decrementIteration: boolean;
-}>;

package/dist/cli/services/agent-loop-setup.d.ts

@@ -1,32 +0,0 @@
-/**
- * Agent Loop Setup — initialization and configuration for runAgentLoop.
- * Extracted from agent-loop.ts.
- */
-import type Anthropic from "@anthropic-ai/sdk";
-import { createTurnContext } from "./telemetry.js";
-import { LoopDetector, resolveAgentLoopConfig } from "../../shared/agent-core.js";
-import type { AgentLoopOptions } from "./agent-loop-types.js";
-import { buildToolExecutor } from "./agent-loop-tools.js";
-export interface AgentLoopSetupResult {
-    tools: any[];
-    systemPrompt: string;
-    messages: Anthropic.MessageParam[];
-    effectiveMaxTurns: number;
-    shellSummarization: boolean;
-    sessionStart: number;
-    storeId: string;
-    turnNum: number;
-    turnCtx: ReturnType<typeof createTurnContext>;
-    activeModel: string;
-    effectiveBudget: number;
-    ctxOverrides: any;
-    ctxMgmt: any;
-    effectiveCompactionTrigger: number;
-    effectiveCompactionBudget: number;
-    maxContinuations: number;
-    loopDetector: LoopDetector;
-    toolExecutor: ReturnType<typeof buildToolExecutor>;
-    hooks: any[];
-    resolved: ReturnType<typeof resolveAgentLoopConfig> | null;
-}
-export declare function setupAgentLoop(opts: AgentLoopOptions): Promise<AgentLoopSetupResult>;

package/dist/cli/services/agent-worker-base-api.d.ts

@@ -1,19 +0,0 @@
-/**
- * Agent Worker Base — API Client
- *
- * Unified proxy caller: auth, request building, caching setup, and stream collection.
- */
-import type { StreamResult } from "../../shared/types.js";
-import type { AgentAPIResponse, AgentAPIOptions } from "./agent-worker-base-types.js";
-/**
- * Convert a StreamResult (from SSE collector) to the AgentAPIResponse format
- * that both subagent and teammate expect.
- *
- * Subagent includes thinking blocks; teammate does not (thinkingEnabled=false).
- */
-export declare function streamResultToResponse(result: StreamResult, includeThinking: boolean): AgentAPIResponse;
-/**
- * Call the server proxy API, collect the SSE stream, and return a typed response.
- * Handles auth, request building, caching setup, and stream collection.
- */
-export declare function callAgentAPI(opts: AgentAPIOptions): Promise<AgentAPIResponse>;

package/dist/cli/services/agent-worker-base-helpers.d.ts

@@ -1,27 +0,0 @@
-/**
- * Agent Worker Base — Content Block Helpers, Tool Deduplication, Yield Helpers
- *
- * Small utility functions shared by subagent.ts and teammate.ts.
- */
-import type Anthropic from "@anthropic-ai/sdk";
-import type { AgentContentBlock } from "./agent-worker-base-types.js";
-/** Extract text blocks from a response */
-export declare function extractTextBlocks(content: AgentContentBlock[]): Anthropic.TextBlock[];
-/** Extract tool use blocks from a response */
-export declare function extractToolUseBlocks(content: AgentContentBlock[]): Anthropic.ToolUseBlock[];
-/** Get combined text from all text blocks */
-export declare function getResponseText(content: AgentContentBlock[]): string;
-/**
- * Deduplicate tools by name — local tools take priority over server tools.
- * Ensures the Anthropic API never receives duplicate tool names (400 error).
- */
-export declare function deduplicateTools(tools: Anthropic.Tool[]): Anthropic.Tool[];
-/**
- * Assert that all tool names are unique. Call right before sending to the API
- * to fail fast with a clear error instead of a cryptic Anthropic 400.
- */
-export declare function assertUniqueToolNames(tools: Anthropic.Tool[]): void;
-/** Yield to event loop to prevent blocking (microtask resolution) */
-export declare function yieldToEventLoop(): Promise<void>;
-/** Longer yield for UI-critical moments (before long API calls, ~1 frame at 60fps) */
-export declare function yieldForRender(): Promise<void>;

package/dist/cli/services/agent-worker-base-tools.d.ts

@@ -1,16 +0,0 @@
-/**
- * Agent Worker Base — Tool Execution
- *
- * Shared tool execution loop with detection, routing, and truncation.
- */
-import { type ExecuteToolBlocksOptions, type ToolBlocksResult } from "./agent-worker-base-types.js";
-/**
- * Execute an array of tool use blocks, handling:
- * - Loop detection (via LoopDetector)
- * - Routing to local, server, or custom executor
- * - Result truncation to maxToolResultChars
- * - Optional per-tool timeout
- *
- * Returns tool results ready to append to messages, plus metadata.
- */
-export declare function executeToolBlocks(opts: ExecuteToolBlocksOptions): Promise<ToolBlocksResult>;

package/dist/cli/services/agent-worker-base-types.d.ts

@@ -1,81 +0,0 @@
-/**
- * Agent Worker Base — Types and interfaces
- *
- * Shared type definitions for the agent worker base module.
- */
-import type Anthropic from "@anthropic-ai/sdk";
-import { LoopDetector } from "../../shared/agent-core.js";
-/** Tools that spawn workers/subprocesses and need much longer than 2 min */
-export declare const LONG_RUNNING_TOOL_TIMEOUT_MS: number;
-export declare const LONG_RUNNING_TOOLS: Set<string>;
-/** Thinking blocks aren't part of Anthropic.ContentBlock but are returned by the API */
-interface ThinkingBlock {
-    type: "thinking";
-    thinking: string;
-    signature: string;
-}
-/** Content block including thinking (extends SDK's ContentBlock union) */
-export type AgentContentBlock = Anthropic.ContentBlock | ThinkingBlock;
-/** Unified response from an API call — used by both subagent and teammate */
-export interface AgentAPIResponse {
-    content: AgentContentBlock[];
-    usage: {
-        input_tokens: number;
-        output_tokens: number;
-        cache_read_tokens: number;
-        cache_creation_tokens: number;
-    };
-    stop_reason: string;
-}
-/** Options for callAgentAPI */
-export interface AgentAPIOptions {
-    modelId: string;
-    contextProfile: import("../../shared/types.js").ContextProfile;
-    systemPrompt: string;
-    messages: Anthropic.MessageParam[];
-    tools: Anthropic.Tool[];
-    thinkingEnabled?: boolean;
-    maxOutputTokens?: number;
-    timeoutMs?: number;
-    /** If true, add cache_control to last tool */
-    cacheLastTool?: boolean;
-    /** Called periodically during long API calls (streaming/thinking) to signal liveness */
-    onHeartbeat?: () => void;
-    /** AbortSignal for cancellation */
-    abortSignal?: AbortSignal;
-}
-/** Result of executing a single tool */
-export interface ToolExecResult {
-    success: boolean;
-    output: string;
-}
-/** Callbacks for tool execution progress */
-export interface ToolExecCallbacks {
-    onToolStart?: (toolName: string, input: Record<string, unknown>) => void;
-    onToolEnd?: (toolName: string, success: boolean, durationMs: number) => void;
-}
-/** Options for executeToolBlocks */
-export interface ExecuteToolBlocksOptions {
-    toolBlocks: Anthropic.ToolUseBlock[];
-    loopDetector: LoopDetector;
-    callbacks?: ToolExecCallbacks;
-    /** Additional tool executor for domain-specific tools (e.g., team tools) */
-    customExecutor?: (name: string, input: Record<string, unknown>) => Promise<ToolExecResult | null>;
-    /** Max chars per tool result before truncation */
-    maxToolResultChars?: number;
-    /** Timeout per tool call in ms (no timeout if not set) */
-    toolTimeoutMs?: number;
-    /** Max errors from a single parallel batch before aborting remaining calls.
-     *  Prevents Promise.all() from cascading N failures into the LoopDetector. */
-    batchErrorLimit?: number;
-    /** AbortSignal for cancellation */
-    signal?: AbortSignal;
-}
-/** Result of executing all tool blocks in a turn */
-export interface ToolBlocksResult {
-    toolResults: Anthropic.ToolResultBlockParam[];
-    toolsUsed: string[];
-    /** Set by custom executor if a domain event occurred (e.g., team task completed) */
-    customSignals: Record<string, unknown>;
-}
-export {};

package/dist/cli/services/background-agents.d.ts

@@ -1,26 +0,0 @@
-/**
- * Background Agent Tracking
- *
- * Tracks async agent tasks that run in the background,
- * with output file-based communication.
- */
-export interface BackgroundAgent {
-    id: string;
-    type: string;
-    outputFile: string;
-    startTime: number;
-    status: "running" | "completed" | "failed";
-    abortController: AbortController;
-}
-export declare function registerBackgroundAgent(id: string, type: string, outputFile: string): AbortController;
-export declare function getAgentStatus(id: string): BackgroundAgent | null;
-export declare function markAgentDone(id: string, success: boolean): void;
-export declare function readAgentOutput(id: string): {
-    status: string;
-    output: string;
-} | null;
-export declare function listBackgroundAgents(): BackgroundAgent[];
-export declare function stopBackgroundAgent(id: string): {
-    success: boolean;
-    message: string;
-};

package/dist/cli/services/background-processes-ops.d.ts

@@ -1,24 +0,0 @@
-/**
- * Background Process Operations — read, kill, list
- * Extracted from background-processes.ts for file size limits.
- */
-import type { BackgroundProcess, ProcessOutput } from "./background-processes.js";
-export declare function readProcessOutput(processes: Map<string, BackgroundProcess>, id: string, options?: {
-    filter?: string;
-}): ProcessOutput | {
-    error: string;
-};
-export declare function killProcess(processes: Map<string, BackgroundProcess>, id: string): {
-    success: boolean;
-    message: string;
-};
-export declare function listProcessesFromMap(processes: Map<string, BackgroundProcess>): Array<{
-    id: string;
-    pid: number | undefined;
-    command: string;
-    status: BackgroundProcess["status"];
-    startedAt: string;
-    runtime: string;
-    outputLines: number;
-    errorLines: number;
-}>;

package/dist/cli/services/background-tool-defs.d.ts

@@ -1,50 +0,0 @@
-/**
- * Tool definitions for background process management.
- * Used by the agent tool system to expose bash_output, kill_shell, list_shells.
- */
-export declare const BACKGROUND_TOOL_DEFINITIONS: ({
-    name: string;
-    description: string;
-    input_schema: {
-        type: string;
-        properties: {
-            bash_id: {
-                type: string;
-                description: string;
-            };
-            filter: {
-                type: string;
-                description: string;
-            };
-            shell_id?: undefined;
-        };
-        required: string[];
-    };
-} | {
-    name: string;
-    description: string;
-    input_schema: {
-        type: string;
-        properties: {
-            shell_id: {
-                type: string;
-                description: string;
-            };
-            bash_id?: undefined;
-            filter?: undefined;
-        };
-        required: string[];
-    };
-} | {
-    name: string;
-    description: string;
-    input_schema: {
-        type: string;
-        properties: {
-            bash_id?: undefined;
-            filter?: undefined;
-            shell_id?: undefined;
-        };
-        required: never[];
-    };
-})[];