@arizeai/phoenix-client/src/schemas/llm/openai/toolChoiceSchemas.ts

A client for the Phoenix API

import z from "zod";

/**
 * OpenAI's tool choice schema
 *
 * @see https://platform.openAI.com/docs/api-reference/chat/create#chat-create-tool_choice
 */
export const openAIToolChoiceSchema = z.union([
  z.literal("auto"),
  z.literal("none"),
  z.literal("required"),
  z.object({
    type: z.literal("function"),
    function: z.object({ name: z.string() }),
  }),
]);

export type OpenaiToolChoice = z.infer<typeof openAIToolChoiceSchema>;