import OpenAI from 'openai';

export type WebSearchTool = Omit<OpenAI.Responses.WebSearchTool, 'type'> & {
  type: 'web_search';
  name: 'web_search_preview' | string;
};

export type FileSearchTool = Omit<OpenAI.Responses.FileSearchTool, 'type'> & {
  type: 'file_search';
  name: 'file_search' | string;
  include_search_results?: boolean;
};

export type CodeInterpreterTool = Omit<
  OpenAI.Responses.Tool.CodeInterpreter,
  'type'
> & {
  type: 'code_interpreter';
  name: 'code_interpreter' | string;
};

export type ImageGenerationTool = Omit<
  OpenAI.Responses.Tool.ImageGeneration,
  | 'type'
  | 'background'
  | 'model'
  | 'moderation'
  | 'output_format'
  | 'quality'
  | 'size'
> & {
  type: 'image_generation';
  name: 'image_generation' | string;
  background?: 'transparent' | 'opaque' | 'auto' | string;
  model?: 'gpt-image-1' | string;
  moderation?: 'auto' | 'low' | string;
  output_format?: 'png' | 'webp' | 'jpeg' | string;
  quality?: 'low' | 'medium' | 'high' | 'auto' | string;
  size?: '1024x1024' | '1024x1536' | '1536x1024' | 'auto' | string;
};