whale-code 6.5.11 → 6.6.0

package/dist/server/lib/__tests__/server-agent-loop-features-caching.test.js

@@ -1,519 +0,0 @@
-import { describe, it, expect, vi, beforeEach } from "vitest";
-
-// ============================================================================
-// Mocks — must be declared before imports
-// ============================================================================
-
-vi.mock("../../../shared/agent-core.js", () => {
-  class MockLoopDetector {
-    resetTurn = vi.fn();
-    getSessionStats = vi.fn(() => ({
-      totalErrors: 0,
-      failedStrategies: 0,
-      consecutiveFailedTurns: 0
-    }));
-  }
-  return {
-    LoopDetector: MockLoopDetector,
-    getContextManagement: vi.fn(() => ({
-      betas: ["context-management-2025-06-27"],
-      config: {
-        edits: []
-      }
-    })),
-    getMaxOutputTokens: vi.fn(() => 16384),
-    getThinkingConfig: vi.fn(() => ({
-      thinking: {
-        type: "disabled"
-      },
-      beta: null
-    })),
-    addPromptCaching: vi.fn((tools, msgs) => ({
-      tools,
-      messages: msgs
-    })),
-    estimateCostUsd: vi.fn(() => 0.001),
-    isRetryableError: vi.fn(() => false),
-    sanitizeError: err => err instanceof Error ? err.message : String(err),
-    routeModel: vi.fn((text, model) => model),
-    resolveToolChoice: vi.fn(() => "auto"),
-    COMPACTION_TRIGGER_TOKENS: 167_000,
-    COMPACTION_TOTAL_BUDGET: 2_000_000,
-    DEFAULT_SESSION_COST_BUDGET_USD: 25,
-    emitCostWarningIfNeeded: vi.fn(),
-    demoteSubagentModel: vi.fn(() => "haiku"),
-    AGENT_DEFAULTS: {
-      maxTurns: 200,
-      temperature: 1.0,
-      maxConcurrentTools: 7,
-      enableDelegation: true,
-      maxDurationMs: 900_000,
-      maxDurationMsWorkflow: 120_000,
-      enableModelRouting: true,
-      subagentMaxTokens: 32000,
-      subagentMaxTurns: 25,
-      subagentTemperature: 1.0,
-      subagentMaxRetries: 3,
-      maxContinuations: 10,
-      subagentMaxContinuations: 5,
-      maxMessageChars: 100_000,
-      maxHistoryChars: 400_000,
-      maxToolResultChars: 2048,
-      compactionTriggerTokens: 167_000,
-      compactionTotalBudget: 2_000_000,
-      sessionCostBudgetUsd: 25.00,
-      teammateMaxTotalTurns: 50,
-      teamTimeoutMs: 600_000,
-      workerStallMs: 180_000,
-      teamReviewMaxTokens: 1024,
-      apiTimeoutMs: 90_000,
-      toolTimeoutMs: 60_000,
-      toolResultTruncateChars: 500_000
-    }
-  };
-});
-vi.mock("../../../shared/sse-parser.js", () => ({
-  processStreamWithCallbacks: vi.fn(async () => ({
-    text: "Test response",
-    toolUseBlocks: [],
-    thinkingBlocks: [],
-    thinkingTokens: 0,
-    citations: [],
-    usage: {
-      inputTokens: 100,
-      outputTokens: 50,
-      cacheCreationTokens: 0,
-      cacheReadTokens: 0
-    },
-    compactionContent: null,
-    contextManagementApplied: false,
-    stopReason: "end_turn"
-  }))
-}));
-vi.mock("../../../shared/constants.js", () => ({
-  MODELS: {
-    HAIKU: "claude-haiku-4-5-20251001",
-    SONNET: "claude-sonnet-4-6",
-    OPUS: "claude-opus-4-6"
-  },
-  getProvider: vi.fn(() => "anthropic")
-}));
-vi.mock("../../../shared/tool-dispatch.js", () => ({
-  dispatchTools: vi.fn(async (blocks, executor, _opts) => {
-    const results = [];
-    for (const block of blocks) {
-      const res = await executor(block.name, block.input);
-      results.push({
-        type: "tool_result",
-        tool_use_id: block.id,
-        content: res.output,
-        is_error: !res.success
-      });
-    }
-    return {
-      results,
-      bailOut: false,
-      bailMessage: undefined
-    };
-  }),
-  buildAssistantContent: vi.fn(({
-    text,
-    toolUseBlocks
-  }) => {
-    const blocks = [];
-    if (text) blocks.push({
-      type: "text",
-      text
-    });
-    if (toolUseBlocks) for (const t of toolUseBlocks) blocks.push({
-      type: "tool_use",
-      id: t.id,
-      name: t.name,
-      input: t.input
-    });
-    return blocks;
-  })
-}));
-vi.mock("../../tool-router.js", () => ({
-  getCachedToolDefs: vi.fn(tools => tools.map(t => ({
-    name: t.name,
-    description: t.description,
-    input_schema: t.input_schema
-  }))),
-  getFullToolSchemas: vi.fn(() => [])
-}));
-vi.mock("../server-subagent.js", () => ({
-  DELEGATE_TASK_TOOL_DEF: {
-    name: "delegate_task",
-    description: "Delegate",
-    input_schema: {
-      type: "object",
-      properties: {
-        prompt: {
-          type: "string"
-        }
-      },
-      required: ["prompt"]
-    }
-  },
-  runServerSubagent: vi.fn(async () => ({
-    success: true,
-    output: "ok",
-    turnCount: 1,
-    toolsUsed: [],
-    tokensUsed: {
-      input: 50,
-      output: 25
-    },
-    costUsd: 0.0005
-  }))
-}));
-vi.mock("../../handlers/transcription.js", () => ({
-  handleTranscribe: vi.fn(async () => ({
-    success: true,
-    transcript: "Transcribed text"
-  }))
-}));
-vi.mock("../compaction-service.js", () => ({
-  preCompact: vi.fn(msgs => ({
-    messages: msgs,
-    bytesRemoved: 0
-  }))
-}));
-vi.mock("../provider-failover.js", () => ({
-  providerFailover: {
-    getActiveProvider: vi.fn(model => ({
-      provider: "anthropic",
-      model,
-      failedOver: false,
-      originalProvider: "anthropic"
-    })),
-    recordSuccess: vi.fn(),
-    recordFailure: vi.fn()
-  }
-}));
-vi.mock("../session-checkpoint.js", () => ({
-  saveCheckpoint: vi.fn(async () => {})
-}));
-vi.mock("../logger.js", () => ({
-  createLogger: vi.fn(() => ({
-    info: vi.fn(),
-    warn: vi.fn(),
-    error: vi.fn(),
-    debug: vi.fn()
-  }))
-}));
-
-// ============================================================================
-// Imports
-// ============================================================================
-
-import { runServerAgentLoop } from "../server-agent-loop.js";
-import { processStreamWithCallbacks } from "../../../shared/sse-parser.js";
-import { estimateCostUsd, isRetryableError, routeModel, resolveToolChoice } from "../../../shared/agent-core.js";
-import { preCompact } from "../compaction-service.js";
-import { providerFailover } from "../provider-failover.js";
-import { saveCheckpoint } from "../session-checkpoint.js";
-
-// ============================================================================
-// Helpers
-// ============================================================================
-
-function makeTool(name) {
-  return {
-    name,
-    description: `Tool: ${name}`,
-    input_schema: {
-      type: "object",
-      properties: {}
-    }
-  };
-}
-function makeAnthropicMock() {
-  return {
-    beta: {
-      messages: {
-        create: vi.fn(async () => async function* () {}())
-      }
-    }
-  };
-}
-function makeSupabaseMock() {
-  return {
-    from: vi.fn(() => ({
-      insert: vi.fn(async () => ({
-        error: null
-      })),
-      select: vi.fn(() => ({
-        data: [],
-        error: null
-      }))
-    }))
-  };
-}
-function makeOpts(overrides = {}) {
-  return {
-    anthropic: makeAnthropicMock(),
-    supabase: makeSupabaseMock(),
-    model: "claude-sonnet-4-6",
-    systemPrompt: "You are a helpful assistant.",
-    messages: [{
-      role: "user",
-      content: "Hello"
-    }],
-    tools: [makeTool("test_tool")],
-    maxTurns: 10,
-    temperature: 0.7,
-    executeTool: vi.fn(async () => ({
-      success: true,
-      data: {
-        result: "ok"
-      }
-    })),
-    enableStreaming: true,
-    ...overrides
-  };
-}
-
-// ============================================================================
-// TESTS — pre-compaction, checkpoints, caching, callbacks, tool choice
-// ============================================================================
-
-describe("server-agent-loop — caching, checkpoints, callbacks", () => {
-  beforeEach(() => {
-    vi.clearAllMocks();
-    vi.mocked(processStreamWithCallbacks).mockReset();
-    vi.mocked(processStreamWithCallbacks).mockResolvedValue({
-      text: "Test response",
-      toolUseBlocks: [],
-      thinkingBlocks: [],
-      thinkingTokens: 0,
-      citations: [],
-      usage: {
-        inputTokens: 100,
-        outputTokens: 50,
-        cacheCreationTokens: 0,
-        cacheReadTokens: 0
-      },
-      compactionContent: null,
-      contextManagementApplied: false,
-      stopReason: "end_turn"
-    });
-    vi.mocked(estimateCostUsd).mockReturnValue(0.001);
-    vi.mocked(isRetryableError).mockReturnValue(false);
-    vi.mocked(routeModel).mockImplementation((_text, model) => model);
-    vi.mocked(preCompact).mockImplementation(msgs => ({
-      messages: msgs,
-      bytesRemoved: 0
-    }));
-    vi.mocked(providerFailover.getActiveProvider).mockImplementation(model => ({
-      provider: "anthropic",
-      model,
-      failedOver: false,
-      originalProvider: "anthropic"
-    }));
-  });
-  describe("pre-compaction", () => {
-    it("applies pre-compaction when significant bytes removed", async () => {
-      vi.mocked(preCompact).mockReturnValue({
-        messages: [{
-          role: "user",
-          content: "Hello"
-        }],
-        bytesRemoved: 50000
-      });
-      await runServerAgentLoop(makeOpts());
-      expect(preCompact).toHaveBeenCalled();
-    });
-    it("does not modify messages when bytes removed is small", async () => {
-      vi.mocked(preCompact).mockReturnValue({
-        messages: [{
-          role: "user",
-          content: "Hello"
-        }],
-        bytesRemoved: 100
-      });
-      await runServerAgentLoop(makeOpts());
-      expect(preCompact).toHaveBeenCalled();
-    });
-  });
-  describe("session checkpoint", () => {
-    it("saves checkpoint when conversationId is provided", async () => {
-      let callCount = 0;
-      vi.mocked(processStreamWithCallbacks).mockImplementation(async () => {
-        callCount++;
-        if (callCount === 1) {
-          return {
-            text: "",
-            toolUseBlocks: [{
-              id: "tu-1",
-              name: "test_tool",
-              input: {}
-            }],
-            thinkingBlocks: [],
-            thinkingTokens: 0,
-            citations: [],
-            usage: {
-              inputTokens: 100,
-              outputTokens: 50,
-              cacheCreationTokens: 0,
-              cacheReadTokens: 0
-            },
-            compactionContent: null,
-            contextManagementApplied: false,
-            stopReason: "tool_use"
-          };
-        }
-        return {
-          text: "Done",
-          toolUseBlocks: [],
-          thinkingBlocks: [],
-          thinkingTokens: 0,
-          citations: [],
-          usage: {
-            inputTokens: 50,
-            outputTokens: 20,
-            cacheCreationTokens: 0,
-            cacheReadTokens: 0
-          },
-          compactionContent: null,
-          contextManagementApplied: false,
-          stopReason: "end_turn"
-        };
-      });
-      await runServerAgentLoop(makeOpts({
-        conversationId: "conv-123"
-      }));
-      expect(saveCheckpoint).toHaveBeenCalled();
-    });
-    it("does not save checkpoint when conversationId is not provided", async () => {
-      let callCount = 0;
-      vi.mocked(processStreamWithCallbacks).mockImplementation(async () => {
-        callCount++;
-        if (callCount === 1) {
-          return {
-            text: "",
-            toolUseBlocks: [{
-              id: "tu-1",
-              name: "test_tool",
-              input: {}
-            }],
-            thinkingBlocks: [],
-            thinkingTokens: 0,
-            citations: [],
-            usage: {
-              inputTokens: 100,
-              outputTokens: 50,
-              cacheCreationTokens: 0,
-              cacheReadTokens: 0
-            },
-            compactionContent: null,
-            contextManagementApplied: false,
-            stopReason: "tool_use"
-          };
-        }
-        return {
-          text: "Done",
-          toolUseBlocks: [],
-          thinkingBlocks: [],
-          thinkingTokens: 0,
-          citations: [],
-          usage: {
-            inputTokens: 50,
-            outputTokens: 20,
-            cacheCreationTokens: 0,
-            cacheReadTokens: 0
-          },
-          compactionContent: null,
-          contextManagementApplied: false,
-          stopReason: "end_turn"
-        };
-      });
-      await runServerAgentLoop(makeOpts({
-        conversationId: undefined
-      }));
-      expect(saveCheckpoint).not.toHaveBeenCalled();
-    });
-  });
-  describe("prompt caching", () => {
-    it("applies prompt caching when enabled (default)", async () => {
-      const {
-        addPromptCaching
-      } = await import("../../../shared/agent-core.js");
-      await runServerAgentLoop(makeOpts({
-        enablePromptCaching: true
-      }));
-      expect(addPromptCaching).toHaveBeenCalled();
-    });
-    it("skips prompt caching when disabled", async () => {
-      const {
-        addPromptCaching
-      } = await import("../../../shared/agent-core.js");
-      vi.mocked(addPromptCaching).mockClear();
-      await runServerAgentLoop(makeOpts({
-        enablePromptCaching: false
-      }));
-      expect(addPromptCaching).not.toHaveBeenCalled();
-    });
-  });
-  describe("callbacks", () => {
-    it("calls onText callback during streaming", async () => {
-      vi.mocked(processStreamWithCallbacks).mockImplementation(async (_stream, callbacks) => {
-        callbacks?.onText?.("Streamed text");
-        return {
-          text: "Streamed text",
-          toolUseBlocks: [],
-          thinkingBlocks: [],
-          thinkingTokens: 0,
-          citations: [],
-          usage: {
-            inputTokens: 100,
-            outputTokens: 50,
-            cacheCreationTokens: 0,
-            cacheReadTokens: 0
-          },
-          compactionContent: null,
-          contextManagementApplied: false,
-          stopReason: "end_turn"
-        };
-      });
-      const onText = vi.fn();
-      await runServerAgentLoop(makeOpts({
-        onText
-      }));
-      expect(onText).toHaveBeenCalledWith("Streamed text");
-    });
-    it("calls onText for non-streaming text blocks", async () => {
-      const anthropic = {
-        beta: {
-          messages: {
-            create: vi.fn(async () => ({
-              content: [{
-                type: "text",
-                text: "Block text"
-              }],
-              usage: {
-                input_tokens: 50,
-                output_tokens: 20
-              },
-              stop_reason: "end_turn"
-            }))
-          }
-        }
-      };
-      const onText = vi.fn();
-      await runServerAgentLoop(makeOpts({
-        anthropic: anthropic,
-        enableStreaming: false,
-        onText
-      }));
-      expect(onText).toHaveBeenCalledWith("Block text");
-    });
-  });
-  describe("tool choice", () => {
-    it("resolves tool choice each turn", async () => {
-      await runServerAgentLoop(makeOpts());
-      expect(resolveToolChoice).toHaveBeenCalled();
-    });
-  });
-});