Function getImageGenToolDefinitions

apps/webuiapps/src/lib/imageGenTools.ts:31–63

()

}

export function getImageGenToolDefinitions(): Array<{
  type: 'function';
  function: {
    name: string;
    description: string;
    parameters: {
      type: 'object';
      properties: Record<string, unknown>;
      required: string[];
    };
  };
}> {
  return [
    {
      type: 'function',
      function: {
        name: TOOL_NAME,
        description:
          'Generate an image from a text prompt. The image is displayed in chat and saved to disk. Returns a url.',
        parameters: {
          type: 'object',
          properties: {
            prompt: {
              type: 'string',
              description: 'Detailed description of the image to generate',
            },
          },
          required: ['prompt'],
        },
      },
    },
  ];
}

export function isImageGenTool(toolName: string): boolean {
  return toolName === TOOL_NAME;