Function createSimpleMockToolset

packages/kosong/test/step.test.ts:48–72

* Create a mock toolset that returns a simple success result for any tool call.

(
  handler?: (toolCall: ToolCall) => Promise<ToolResult> | ToolResult,
)

 * Create a mock toolset that returns a simple success result for any tool call.
 */
function createSimpleMockToolset(
  handler?: (toolCall: ToolCall) => Promise<ToolResult> | ToolResult,
): Toolset {
  return {
    tools: [
      {
        name: 'plus',
        description: 'Add two numbers',
        parameters: {
          type: 'object',
          properties: {
            a: { type: 'integer' },
            b: { type: 'integer' },
          },
        },
      },
    ],
    handle:
      handler ??
      ((toolCall: ToolCall): ToolResult => ({
        toolCallId: toolCall.id,
        returnValue: toolOk({ output: 'mock-result' }),
      })),
  };
}
describe('step()', () => {
  it('returns a StepResult with message, toolCalls, and toolResults', async () => {
    const plusToolCall: ToolCall = {