@ai-sdk/openai/src/tool/custom.ts

The **[OpenAI provider](https://ai-sdk.dev/providers/ai-sdk-providers/openai)** for the [AI SDK](https://ai-sdk.dev/docs) contains language model support for the OpenAI chat and completion APIs and embedding model support for the OpenAI embeddings API.

import {
  createProviderDefinedToolFactory,
  lazySchema,
  zodSchema,
} from '@ai-sdk/provider-utils';
import { z } from 'zod/v4';

export const customArgsSchema = lazySchema(() =>
  zodSchema(
    z.object({
      description: z.string().optional(),
      format: z
        .union([
          z.object({
            type: z.literal('grammar'),
            syntax: z.enum(['regex', 'lark']),
            definition: z.string(),
          }),
          z.object({
            type: z.literal('text'),
          }),
        ])
        .optional(),
    }),
  ),
);

const customInputSchema = lazySchema(() => zodSchema(z.string()));

export const customToolFactory = createProviderDefinedToolFactory<
  string,
  {
    /**
     * An optional description of what the tool does.
     */
    description?: string;

    /**
     * The output format specification for the tool.
     * Omit for unconstrained text output.
     */
    format?:
      | {
          type: 'grammar';
          syntax: 'regex' | 'lark';
          definition: string;
        }
      | {
          type: 'text';
        };
  }
>({
  id: 'openai.custom',
  inputSchema: customInputSchema,
});

export const customTool = (args: Parameters<typeof customToolFactory>[0]) =>
  customToolFactory(args);