kimi-code/packages/node-sdk/test/session-set-thinking.test.ts

import { afterEach, describe, expect, it } from 'vitest';

import { KimiHarness, type KimiError } from '#/index';

import { makeTempDir, removeTempDirs, waitForAgentWireEvent } from './session-runtime-helpers';
import { TEST_IDENTITY } from './test-identity';

const tempDirs: string[] = [];

afterEach(async () => {
  await removeTempDirs(tempDirs);
});

describe('Session.setThinking', () => {
  it('sends config.update with the new thinking level', async () => {
    const homeDir = await makeTempDir(tempDirs, 'kimi-sdk-thinking-home-');
    const workDir = await makeTempDir(tempDirs, 'kimi-sdk-thinking-work-');
    const harness = new KimiHarness({ homeDir, identity: TEST_IDENTITY });

    try {
      const session = await harness.createSession({ id: 'ses_thinking_wire', workDir });

      await session.setThinking('low');

      await expect(
        waitForAgentWireEvent(
          homeDir,
          session.id,
          'config.update',
          (event) => event['thinkingLevel'] === 'low',
        ),
      ).resolves.toMatchObject({
        type: 'config.update',
        thinkingLevel: 'low',
      });
    } finally {
      await harness.close();
    }
  });

  it('rejects empty thinking levels', async () => {
    const homeDir = await makeTempDir(tempDirs, 'kimi-sdk-thinking-home-');
    const workDir = await makeTempDir(tempDirs, 'kimi-sdk-thinking-work-');
    const harness = new KimiHarness({ homeDir, identity: TEST_IDENTITY });

    try {
      const session = await harness.createSession({ id: 'ses_thinking_empty', workDir });

      await expect(session.setThinking('   ')).rejects.toMatchObject({
        name: 'KimiError',
        code: 'session.thinking_empty',
      } satisfies Partial<KimiError>);
    } finally {
      await harness.close();
    }
  });

  it('rejects after the session is closed', async () => {
    const homeDir = await makeTempDir(tempDirs, 'kimi-sdk-thinking-home-');
    const workDir = await makeTempDir(tempDirs, 'kimi-sdk-thinking-work-');
    const harness = new KimiHarness({ homeDir, identity: TEST_IDENTITY });

    try {
      const session = await harness.createSession({ id: 'ses_thinking_closed', workDir });
      await session.close();

      await expect(session.setThinking('high')).rejects.toMatchObject({
        name: 'KimiError',
        code: 'session.closed',
      } satisfies Partial<KimiError>);
    } finally {
      await harness.close();
    }
  });
});