import { describe, expectTypeOf, it } from 'vitest'; import { z } from 'zod/v4'; import { tool, type Tool } from '@ai-sdk/provider-utils'; import type { ActiveTools } from './active-tools'; import { filterActiveTools } from './filter-active-tools'; const mockTools = { tool1: tool({ description: 'Tool 1 description', inputSchema: z.object({}), }), tool2: tool({ description: 'Tool 2 description', inputSchema: z.object({ city: z.string() }), }), }; const mockProviderDefinedTool: Tool = { type: 'provider', id: 'provider.tool-id', isProviderExecuted: false, args: { key: 'value' }, inputSchema: z.object({}), }; const mockToolsWithProviderDefined = { ...mockTools, providerTool: mockProviderDefinedTool, }; declare const maybeToolsWithProviderDefined: | typeof mockToolsWithProviderDefined | undefined; describe('filterActiveTools types', () => { it('should infer the active tool subset for literal activeTools', () => { const result = filterActiveTools({ tools: mockToolsWithProviderDefined, activeTools: ['tool1', 'providerTool'] as const, }); expectTypeOf().toEqualTypeOf< Pick >(); }); it('should preserve the full tool set when activeTools is undefined', () => { const result = filterActiveTools({ tools: mockToolsWithProviderDefined, activeTools: undefined, }); expectTypeOf().toEqualTypeOf< typeof mockToolsWithProviderDefined >(); }); it('should preserve the full tool set for non-literal activeTools arrays', () => { const activeTools: Array = [ 'tool1', ]; const result = filterActiveTools({ tools: mockToolsWithProviderDefined, activeTools, }); expectTypeOf().toEqualTypeOf< typeof mockToolsWithProviderDefined >(); }); it('should infer undefined from possibly undefined tools', () => { const result = filterActiveTools({ tools: maybeToolsWithProviderDefined, activeTools: ['tool1', 'providerTool'] as const, }); expectTypeOf().toEqualTypeOf< | Pick | undefined >(); }); it('should infer undefined when tools are undefined', () => { const result = filterActiveTools({ tools: undefined, activeTools: ['tool1'] as const, }); expectTypeOf().toEqualTypeOf(); }); it('should preserve the full tool set for maybe undefined activeTools', () => { const activeTools: ActiveTools = [ 'tool1', ]; const result = filterActiveTools({ tools: mockToolsWithProviderDefined, activeTools, }); expectTypeOf().toEqualTypeOf< typeof mockToolsWithProviderDefined >(); }); it('should reject invalid active tool names', () => { filterActiveTools({ tools: mockToolsWithProviderDefined, // @ts-expect-error activeTools only accepts keys of the tool set activeTools: ['unknownTool'] as const, }); }); });