kimi-code/packages/kosong/test/regression.test.ts

import { generate } from '#/generate';
import type { Message, StreamedMessagePart } from '#/message';
import { createAssistantMessage, createUserMessage } from '#/message';
import type { ChatProvider, GenerateOptions, StreamedMessage, ThinkingEffort } from '#/provider';
import type { Tool } from '#/tool';
import type { TokenUsage } from '#/usage';
import { describe, expect, it } from 'vitest';

function createMockStream(
  parts: StreamedMessagePart[],
  opts?: { id?: string; usage?: TokenUsage },
): StreamedMessage {
  return {
    get id(): string | null {
      return opts?.id ?? null;
    },
    get usage(): TokenUsage | null {
      return opts?.usage ?? null;
    },
    finishReason: null,
    rawFinishReason: null,
    async *[Symbol.asyncIterator](): AsyncIterator<StreamedMessagePart> {
      for (const part of parts) {
        yield part;
      }
    },
  };
}

function createMockProvider(stream: StreamedMessage): ChatProvider {
  return {
    name: 'mock',
    modelName: 'mock-model',
    thinkingEffort: null,
    generate: async (
      _systemPrompt: string,
      _tools: Tool[],
      _history: Message[],
      _options?: GenerateOptions,
    ): Promise<StreamedMessage> => stream,
    withThinking(_effort: ThinkingEffort): ChatProvider {
      return this;
    },
  };
}

describe('regression', () => {
  describe('empty text parts', () => {
    it('standalone empty TextPart is kept in content', async () => {
      const stream = createMockStream([
        { type: 'text', text: 'before' },
        { type: 'image_url', imageUrl: { url: 'https://example.com/img.png' } },
        { type: 'text', text: '' },
      ]);
      const provider = createMockProvider(stream);

      const result = await generate(provider, '', [], []);

      expect(result.message.content).toEqual([
        { type: 'text', text: 'before' },
        { type: 'image_url', imageUrl: { url: 'https://example.com/img.png' } },
        { type: 'text', text: '' },
      ]);
    });
  });

  describe('toolCalls defaults', () => {
    it('createUserMessage has toolCalls as empty array (not undefined)', () => {
      const msg = createUserMessage('hello');
      expect(msg.toolCalls).toEqual([]);
    });

    it('createAssistantMessage without toolCalls has empty array', () => {
      const msg = createAssistantMessage([{ type: 'text', text: 'test' }]);
      expect(msg.toolCalls).toEqual([]);
    });

    it('createAssistantMessage with explicit toolCalls preserves them', () => {
      const msg = createAssistantMessage(
        [{ type: 'text', text: 'test' }],
        [
          {
            type: 'function',
            id: 'call-1',
            name: 'search', arguments: '{}',
          },
        ],
      );
      expect(msg.toolCalls).toEqual([
        {
          type: 'function',
          id: 'call-1',
          name: 'search', arguments: '{}',
        },
      ]);
    });

    it('generate() result message has toolCalls as empty array when no tools called', async () => {
      const stream = createMockStream([{ type: 'text', text: 'hi' }]);
      const provider = createMockProvider(stream);

      const result = await generate(provider, '', [], []);
      expect(result.message.toolCalls).toEqual([]);
    });
  });
});