/packages/runtime/tests/service-adapters/anthropic/anthropic-adapter.test.ts

import { AnthropicAdapter } from "../../../src/service-adapters/anthropic/anthropic-adapter";
import {
  TextMessage,
  ActionExecutionMessage,
  ResultMessage,
  Role,
} from "../../../src/graphql/types/converted";

// Mock only the Anthropic SDK, not our adapter
vi.mock("@anthropic-ai/sdk", () => {
  return {
    default: vi.fn().mockImplementation(() => ({
      messages: {
        create: vi.fn(),
      },
    })),
  };
});

// Mock the message classes
vi.mock("../../../src/graphql/types/converted", () => {
  class MockTextMessage {
    content: string;
    role: string;
    id: string;

    constructor(options: { role: string; content: string }) {
      this.role = options.role;
      this.content = options.content;
      this.id = "mock-text-" + Math.random().toString(36).slice(7);
    }

    isTextMessage() {
      return true;
    }
    isImageMessage() {
      return false;
    }
    isActionExecutionMessage() {
      return false;
    }
    isResultMessage() {
      return false;
    }
  }

  class MockActionExecutionMessage {
    id: string;
    name: string;
    arguments: string;

    constructor(params: { id: string; name: string; arguments: string }) {
      this.id = params.id;
      this.name = params.name;
      this.arguments = params.arguments;
    }

    isTextMessage() {
      return false;
    }
    isImageMessage() {
      return false;
    }
    isActionExecutionMessage() {
      return true;
    }
    isResultMessage() {
      return false;
    }
  }

  class MockResultMessage {
    actionExecutionId: string;
    result: string;
    id: string;

    constructor(params: { actionExecutionId: string; result: string }) {
      this.actionExecutionId = params.actionExecutionId;
      this.result = params.result;
      this.id = "mock-result-" + Math.random().toString(36).slice(7);
    }

    isTextMessage() {
      return false;
    }
    isImageMessage() {
      return false;
    }
    isActionExecutionMessage() {
      return false;
    }
    isResultMessage() {
      return true;
    }
  }

  return {
    TextMessage: MockTextMessage,
    ActionExecutionMessage: MockActionExecutionMessage,
    ResultMessage: MockResultMessage,
    Role: {
      assistant: "assistant",
      developer: "developer",
      system: "system",
      tool: "tool",
      user: "user",
    },
  };
});

describe("AnthropicAdapter", () => {
  let adapter: AnthropicAdapter;
  let mockEventSource: any;
  let mockAnthropicCreate: any;

  beforeEach(() => {
    vi.clearAllMocks();

    // Create a mock Anthropic instance
    const mockAnthropic = {
      messages: {
        create: vi.fn(),
      },
    };

    // Create adapter with the mocked instance
    adapter = new AnthropicAdapter({ anthropic: mockAnthropic as any });

    // Mock the create method to capture what's being sent
    mockAnthropicCreate = mockAnthropic.messages.create;

    mockEventSource = {
      stream: vi.fn((callback) => {
        const mockStream = {
          sendTextMessageStart: vi.fn(),
          sendTextMessageContent: vi.fn(),
          sendTextMessageEnd: vi.fn(),
          sendActionExecutionStart: vi.fn(),
          sendActionExecutionArgs: vi.fn(),
          sendActionExecutionEnd: vi.fn(),
          complete: vi.fn(),
        };
        callback(mockStream);
        return Promise.resolve();
      }),
    };
  });

  describe("Deduplication Logic", () => {
    it("should filter out duplicate result messages", async () => {
      const systemMessage = new TextMessage({
        role: Role.system,
        content: "System message",
      });
      const userMessage = new TextMessage({
        role: Role.user,
        content: "Set theme to orange",
      });

      // Tool execution
      const toolExecution = new ActionExecutionMessage({
        id: "tool-123",
        name: "setThemeColor",
        arguments: '{"themeColor": "orange"}',
      });

      // Multiple duplicate results (this was causing the infinite loop)
      const result1 = new ResultMessage({
        actionExecutionId: "tool-123",
        result: "Theme color set to orange",
      });
      const result2 = new ResultMessage({
        actionExecutionId: "tool-123",
        result: "Theme color set to orange",
      });
      const result3 = new ResultMessage({
        actionExecutionId: "tool-123",
        result: "Theme color set to orange",
      });

      // Mock Anthropic to return empty stream (simulating the original problem)
      mockAnthropicCreate.mockResolvedValue({
        [Symbol.asyncIterator]: async function* () {
          // Empty stream - no content from Anthropic
        },
      });

      await adapter.process({
        threadId: "test-thread",
        model: "claude-3-5-sonnet-latest",
        messages: [
          systemMessage,
          userMessage,
          toolExecution,
          result1,
          result2,
          result3,
        ],
        actions: [],
        eventSource: mockEventSource,
        forwardedParameters: {},
      });

      // Check that only one result message was sent to Anthropic
      expect(mockAnthropicCreate).toHaveBeenCalledWith(
        expect.objectContaining({
          messages: expect.arrayContaining([
            expect.objectContaining({
              role: "user",
              content: [{ type: "text", text: "Set theme to orange" }],
            }),
            expect.objectContaining({
              role: "assistant",
              content: [
                expect.objectContaining({
                  type: "tool_use",
                  id: "tool-123",
                  name: "setThemeColor",
                }),
              ],
            }),
            expect.objectContaining({
              role: "user",
              content: [
                expect.objectContaining({
                  type: "tool_result",
                  content: "Theme color set to orange",
                  tool_use_id: "tool-123",
                }),
              ],
            }),
          ]),
        }),
      );

      // Verify only 3 messages sent (user, assistant tool_use, user tool_result) - no duplicates
      const sentMessages = mockAnthropicCreate.mock.calls[0][0].messages;
      expect(sentMessages).toHaveLength(3);

      // Count tool_result messages - should be only 1
      const toolResults = sentMessages.filter(
        (msg: any) =>
          msg.role === "user" &&
          msg.content.some((c: any) => c.type === "tool_result"),
      );
      expect(toolResults).toHaveLength(1);
    });

    it("should filter out invalid result messages without corresponding tool_use", async () => {
      const systemMessage = new TextMessage({
        role: Role.system,
        content: "System message",
      });

      // Valid tool execution
      const validTool = new ActionExecutionMessage({
        id: "valid-tool",
        name: "validAction",
        arguments: "{}",
      });
      const validResult = new ResultMessage({
        actionExecutionId: "valid-tool",
        result: "Valid result",
      });

      // Invalid result with no corresponding tool_use
      const invalidResult = new ResultMessage({
        actionExecutionId: "nonexistent-tool",
        result: "Invalid result",
      });

      mockAnthropicCreate.mockResolvedValue({
        [Symbol.asyncIterator]: async function* () {},
      });

      await adapter.process({
        threadId: "test-thread",
        messages: [systemMessage, validTool, validResult, invalidResult],
        actions: [],
        eventSource: mockEventSource,
        forwardedParameters: {},
      });

      const sentMessages = mockAnthropicCreate.mock.calls[0][0].messages;

      // Should only include the valid tool result
      const toolResults = sentMessages.filter(
        (msg: any) =>
          msg.role === "user" &&
          msg.content.some((c: any) => c.type === "tool_result"),
      );
      expect(toolResults).toHaveLength(1);
      expect(toolResults[0].content[0].tool_use_id).toBe("valid-tool");
    });
  });

  describe("Fallback Response Logic", () => {
    it("should generate contextual fallback when Anthropic returns no content", async () => {
      const systemMessage = new TextMessage({
        role: Role.system,
        content: "System message",
      });
      const userMessage = new TextMessage({
        role: Role.user,
        content: "Set theme to orange",
      });

      const toolExecution = new ActionExecutionMessage({
        id: "tool-123",
        name: "setThemeColor",
        arguments: '{"themeColor": "orange"}',
      });

      const toolResult = new ResultMessage({
        actionExecutionId: "tool-123",
        result: "Theme color set to orange",
      });

      // Mock Anthropic to return empty stream
      mockAnthropicCreate.mockResolvedValue({
        [Symbol.asyncIterator]: async function* () {
          // No content blocks - simulates Anthropic not responding
        },
      });

      const mockStream = {
        sendTextMessageStart: vi.fn(),
        sendTextMessageContent: vi.fn(),
        sendTextMessageEnd: vi.fn(),
        complete: vi.fn(),
      };

      mockEventSource.stream.mockImplementation((callback) => {
        callback(mockStream);
        return Promise.resolve();
      });

      await adapter.process({
        threadId: "test-thread",
        messages: [systemMessage, userMessage, toolExecution, toolResult],
        actions: [],
        eventSource: mockEventSource,
        forwardedParameters: {},
      });

      // Should generate fallback response with the tool result content
      expect(mockStream.sendTextMessageStart).toHaveBeenCalled();
      expect(mockStream.sendTextMessageContent).toHaveBeenCalledWith({
        messageId: expect.any(String),
        content: "Theme color set to orange", // Uses the actual result content
      });
      expect(mockStream.sendTextMessageEnd).toHaveBeenCalled();
    });

    it("should use generic fallback when no tool result content available", async () => {
      const systemMessage = new TextMessage({
        role: Role.system,
        content: "System message",
      });

      const toolExecution = new ActionExecutionMessage({
        id: "tool-123",
        name: "someAction",
        arguments: "{}",
      });

      const toolResult = new ResultMessage({
        actionExecutionId: "tool-123",
        result: "", // Empty result
      });

      mockAnthropicCreate.mockResolvedValue({
        [Symbol.asyncIterator]: async function* () {},
      });

      const mockStream = {
        sendTextMessageStart: vi.fn(),
        sendTextMessageContent: vi.fn(),
        sendTextMessageEnd: vi.fn(),
        complete: vi.fn(),
      };

      mockEventSource.stream.mockImplementation((callback) => {
        callback(mockStream);
        return Promise.resolve();
      });

      await adapter.process({
        threadId: "test-thread",
        messages: [systemMessage, toolExecution, toolResult],
        actions: [],
        eventSource: mockEventSource,
        forwardedParameters: {},
      });

      // Should use generic fallback
      expect(mockStream.sendTextMessageContent).toHaveBeenCalledWith({
        messageId: expect.any(String),
        content: "Task completed successfully.",
      });
    });
  });

  describe("Unknown Tool Use Handling", () => {
    it("should skip unknown tool_use blocks without crashing", async () => {
      const systemMessage = new TextMessage({
        role: Role.system,
        content: "System message",
      });
      const userMessage = new TextMessage({
        role: Role.user,
        content: "Do something",
      });

      // Mock Anthropic to return a stream with an unknown tool_use block
      mockAnthropicCreate.mockResolvedValue({
        [Symbol.asyncIterator]: async function* () {
          yield { type: "message_start", message: { id: "msg-1" } };
          // Unknown tool_use block — tool name not in the actions list
          yield {
            type: "content_block_start",
            content_block: {
              type: "tool_use",
              id: "tool-unknown",
              name: "nonexistent_tool",
            },
          };
          yield {
            type: "content_block_delta",
            delta: { type: "input_json_delta", partial_json: '{"query":' },
          };
          yield {
            type: "content_block_delta",
            delta: { type: "input_json_delta", partial_json: '"test"}' },
          };
          yield { type: "content_block_stop" };
          // Then a normal text block
          yield {
            type: "content_block_start",
            content_block: { type: "text" },
          };
          yield {
            type: "content_block_delta",
            delta: { type: "text_delta", text: "Here is the result." },
          };
          yield { type: "content_block_stop" };
        },
      });

      const mockStream = {
        sendTextMessageStart: vi.fn(),
        sendTextMessageContent: vi.fn(),
        sendTextMessageEnd: vi.fn(),
        sendActionExecutionStart: vi.fn(),
        sendActionExecutionArgs: vi.fn(),
        sendActionExecutionEnd: vi.fn(),
        complete: vi.fn(),
      };

      let streamCallbackDone: Promise<void>;
      mockEventSource.stream.mockImplementation((callback: any) => {
        streamCallbackDone = callback(mockStream);
      });

      await adapter.process({
        threadId: "test-thread",
        messages: [systemMessage, userMessage],
        actions: [
          {
            name: "known_tool",
            description: "A known tool",
            jsonSchema: '{"type":"object","properties":{}}',
          },
        ],
        eventSource: mockEventSource,
        forwardedParameters: {},
      });

      // Wait for async stream processing to complete
      await streamCallbackDone!;

      // Should NOT have sent action execution events for the unknown tool
      expect(mockStream.sendActionExecutionStart).not.toHaveBeenCalled();
      expect(mockStream.sendActionExecutionArgs).not.toHaveBeenCalled();
      expect(mockStream.sendActionExecutionEnd).not.toHaveBeenCalled();

      // Should still process the text block normally
      expect(mockStream.sendTextMessageStart).toHaveBeenCalled();
      expect(mockStream.sendTextMessageContent).toHaveBeenCalledWith({
        messageId: "msg-1",
        content: "Here is the result.",
      });
      expect(mockStream.sendTextMessageEnd).toHaveBeenCalled();
      expect(mockStream.complete).toHaveBeenCalled();
    });

    it("should trigger fallback when only unknown tool_use blocks are returned", async () => {
      const systemMessage = new TextMessage({
        role: Role.system,
        content: "System message",
      });
      const userMessage = new TextMessage({
        role: Role.user,
        content: "Do something",
      });

      const toolExecution = new ActionExecutionMessage({
        id: "tool-prev",
        name: "someAction",
        arguments: "{}",
      });

      const toolResult = new ResultMessage({
        actionExecutionId: "tool-prev",
        result: "Previous result",
      });

      // Mock Anthropic to return ONLY an unknown tool_use block
      mockAnthropicCreate.mockResolvedValue({
        [Symbol.asyncIterator]: async function* () {
          yield { type: "message_start", message: { id: "msg-1" } };
          yield {
            type: "content_block_start",
            content_block: {
              type: "tool_use",
              id: "tool-unknown",
              name: "nonexistent_tool",
            },
          };
          yield {
            type: "content_block_delta",
            delta: { type: "input_json_delta", partial_json: "{}" },
          };
          yield { type: "content_block_stop" };
        },
      });

      const mockStream = {
        sendTextMessageStart: vi.fn(),
        sendTextMessageContent: vi.fn(),
        sendTextMessageEnd: vi.fn(),
        sendActionExecutionStart: vi.fn(),
        sendActionExecutionArgs: vi.fn(),
        sendActionExecutionEnd: vi.fn(),
        complete: vi.fn(),
      };

      let streamCallbackDone: Promise<void>;
      mockEventSource.stream.mockImplementation((callback: any) => {
        streamCallbackDone = callback(mockStream);
      });

      await adapter.process({
        threadId: "test-thread",
        messages: [systemMessage, userMessage, toolExecution, toolResult],
        actions: [
          {
            name: "known_tool",
            description: "A known tool",
            jsonSchema: '{"type":"object","properties":{}}',
          },
        ],
        eventSource: mockEventSource,
        forwardedParameters: {},
      });

      // Wait for async stream processing to complete
      await streamCallbackDone!;

      // Should NOT have sent action execution events
      expect(mockStream.sendActionExecutionStart).not.toHaveBeenCalled();

      // Should trigger fallback since hasReceivedContent should be false
      expect(mockStream.sendTextMessageStart).toHaveBeenCalled();
      expect(mockStream.sendTextMessageContent).toHaveBeenCalledWith({
        messageId: expect.any(String),
        content: "Previous result",
      });
      expect(mockStream.sendTextMessageEnd).toHaveBeenCalled();
    });
  });
});

describe("AnthropicAdapter max_tokens default", () => {
  let mockAnthropicCreate: any;
  let mockEventSource: any;

  beforeEach(() => {
    vi.clearAllMocks();
  });

  it("should default max_tokens to 4096 when not specified", async () => {
    const mockAnthropic = {
      messages: {
        create: vi.fn(),
      },
    };

    const adapter = new AnthropicAdapter({ anthropic: mockAnthropic as any });
    mockAnthropicCreate = mockAnthropic.messages.create;

    mockAnthropicCreate.mockResolvedValue({
      [Symbol.asyncIterator]: async function* () {},
    });

    mockEventSource = {
      stream: vi.fn((callback) => {
        const mockStream = {
          sendTextMessageStart: vi.fn(),
          sendTextMessageContent: vi.fn(),
          sendTextMessageEnd: vi.fn(),
          sendActionExecutionStart: vi.fn(),
          sendActionExecutionArgs: vi.fn(),
          sendActionExecutionEnd: vi.fn(),
          complete: vi.fn(),
        };
        callback(mockStream);
        return Promise.resolve();
      }),
    };

    const systemMessage = new TextMessage({
      role: Role.system,
      content: "System message",
    });
    const userMessage = new TextMessage({ role: Role.user, content: "Hello" });

    await adapter.process({
      threadId: "test-thread",
      messages: [systemMessage, userMessage],
      actions: [],
      eventSource: mockEventSource,
      forwardedParameters: {},
    });

    const createCallArgs = mockAnthropicCreate.mock.calls[0][0];
    expect(createCallArgs.max_tokens).toBe(4096);
  });

  it("should use provided maxTokens when specified", async () => {
    const mockAnthropic = {
      messages: {
        create: vi.fn(),
      },
    };

    const adapter = new AnthropicAdapter({ anthropic: mockAnthropic as any });
    mockAnthropicCreate = mockAnthropic.messages.create;

    mockAnthropicCreate.mockResolvedValue({
      [Symbol.asyncIterator]: async function* () {},
    });

    mockEventSource = {
      stream: vi.fn((callback) => {
        const mockStream = {
          sendTextMessageStart: vi.fn(),
          sendTextMessageContent: vi.fn(),
          sendTextMessageEnd: vi.fn(),
          sendActionExecutionStart: vi.fn(),
          sendActionExecutionArgs: vi.fn(),
          sendActionExecutionEnd: vi.fn(),
          complete: vi.fn(),
        };
        callback(mockStream);
        return Promise.resolve();
      }),
    };

    const systemMessage = new TextMessage({
      role: Role.system,
      content: "System message",
    });
    const userMessage = new TextMessage({ role: Role.user, content: "Hello" });

    await adapter.process({
      threadId: "test-thread",
      messages: [systemMessage, userMessage],
      actions: [],
      eventSource: mockEventSource,
      forwardedParameters: { maxTokens: 8192 },
    });

    const createCallArgs = mockAnthropicCreate.mock.calls[0][0];
    expect(createCallArgs.max_tokens).toBe(8192);
  });
});