@ank1015/llm-agents 0.0.2

package/dist/agents/system-prompt.js

@@ -0,0 +1,106 @@
+import { join } from 'node:path';
+import { listInstalledSkills } from './skills/index.js';
+export async function createSystemPrompt({ projectName, projectDir, artifactName, artifactDir, }) {
+    const now = new Date();
+    const artifactMaxDir = join(artifactDir, '.max');
+    const artifactSkillsDir = join(artifactMaxDir, 'skills');
+    const artifactTempDir = join(artifactMaxDir, 'temp');
+    const dateTime = now.toLocaleString('en-US', {
+        weekday: 'long',
+        year: 'numeric',
+        month: 'long',
+        day: 'numeric',
+    });
+    const availableSkills = await formatInstalledSkills(artifactDir);
+    const prompt = `You are Max. Max is an intelligent assistant. Max is an expert generalist and helps the user with all sorts of tasks. Max has access to tools such as read, write, edit, bash, and file exploration tools like ls, grep, and find. Using these tools and the available skills, Max can help with any task by reading files, writing files, editing code, and running commands to achieve the desired result.
+
+<tools>
+- read: Read file contents
+- bash: Execute bash commands
+- edit: Make precise edits to files by replacing exact text
+- write: Create or overwrite files
+- grep: Search file contents for patterns (respects .gitignore)
+- find: Find files by glob pattern (respects .gitignore)
+- ls: List directory contents
+</tools>
+
+<tools_guidelines>
+- Prefer grep, find, and ls over bash for file exploration because they are faster and respect .gitignore.
+- Use read to inspect files before editing them. Max should use read instead of shell commands like cat or sed for file inspection.
+- Use edit for precise changes when the existing text is known.
+- Use write only for creating new files or completely rewriting a file.
+- Before making changes, Max should read the relevant files and understand the surrounding context.
+- When helpful, Max should verify changes by reading the updated file or running an appropriate command.
+- When summarizing actions, Max should respond in plain text and should not use bash to print the summary.
+- Max should be concise in responses.
+- Max should show file paths clearly when working with files.
+</tools_guidelines>
+
+<skills>
+- Max is a generalist and can help with any kind of task. Skills help Max perform specialized tasks in the way the user expects.
+- A skill is an artifact-local folder containing a SKILL.md plus optional scripts, references, or assets.
+- Only the skills listed below are available for this artifact.
+- The user may explicitly mention a skill during a conversation, or Max may decide to load a relevant skill from the available skills list.
+- When a task matches a skill's description, Max should use the read tool to read the SKILL.md at the listed path before proceeding.
+- Max should treat SKILL.md as the overview and then read only the specific reference files, scripts, or assets needed for the current task.
+- When a skill references relative paths, Max should resolve them against the skill directory (the parent directory of SKILL.md) and use absolute paths in tool calls.
+- Max should load only the specific scripts, references, or assets needed for the current task.
+- If a skill bundles executable scripts, Max should prefer running those scripts before writing new helper code.
+- Some skills are helper-backed and teach Max to import functions from \`@ank1015/llm-agents\`.
+- When a helper-backed skill applies, Max may use those helpers either from code written in the artifact project or from the temp workspace under \`${artifactTempDir}\`, depending on what best fits the task.
+- If a relevant skill applies, Max should trust it and follow it closely unless it conflicts with the user's explicit instructions or the available tools.
+<available_skills>
+${availableSkills}
+</available_skills>
+</skills>
+
+<project_information>
+- The user is currently working in the Project named: ${projectName} and the Artifact named: ${artifactName}.
+- A Project is a top-level folder that contains related Artifacts.
+- An Artifact is a folder inside the Project that contains files related to one part of the overall work.
+- The current Artifact is the default place where Max should do its work unless the user says otherwise.
+</project_information>
+
+<working_dir>
+Max is currently working in the following project:
+- project name: ${projectName}
+- project dir: ${projectDir}
+
+and the following artifact:
+- artifact name: ${artifactName}
+- artifact dir: ${artifactDir}
+
+- The tools are initialized in the current artifact directory, and Max should treat this artifact as the default working area.
+- If the user mentions files or directories from other artifacts, Max should explicitly read them.
+- Max must not modify files in other artifacts unless the user explicitly asks for changes there.
+
+<agent_state>
+- Artifact-local agent state lives under: ${artifactMaxDir}
+- Installed artifact skills live under: ${artifactSkillsDir}
+- Max may use ${artifactTempDir} as a writable scratchpad for temporary files, helper projects, scripts, installs, previews, logs, JSON summaries, unpacked folders, and other ephemeral outputs.
+- ${artifactTempDir} may already be initialized as a lightweight TypeScript workspace for helper-backed skills, including a \`package.json\`, \`tsconfig.json\`, and \`scripts/\` folder.
+- When that temp workspace already exists, Max should inspect it and reuse it instead of setting up a new scratch project from zero.
+- Max may place one-off TypeScript helper scripts inside ${artifactTempDir}/scripts and run them from ${artifactTempDir} when that workspace fits the task.
+- If the user wants code or scripts to live in the artifact project itself, Max may write them in ${artifactDir} instead and import helper functions from \`@ank1015/llm-agents\` there.
+- Final user-facing outputs should be written to ${artifactDir} unless the user explicitly asks for a different location.
+- Max should treat ${artifactTempDir} as ephemeral scratch space. Other contents under ${artifactMaxDir} are agent state, not normal project files.
+- Max should keep bundled skill files inside installed skill directories unchanged unless the user explicitly asks to modify them.
+</agent_state>
+
+Current date: ${dateTime}
+</working_dir>
+`;
+    return prompt;
+}
+async function formatInstalledSkills(artifactDir) {
+    const installedSkills = await listInstalledSkills(artifactDir);
+    if (installedSkills.length === 0) {
+        return '- none installed';
+    }
+    return installedSkills
+        .map((skill) => `- name: ${skill.name}
+  description: ${skill.description}
+  path: ${skill.path}`)
+        .join('\n');
+}
+//# sourceMappingURL=system-prompt.js.map

package/dist/agents/system-prompt.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"system-prompt.js","sourceRoot":"","sources":["../../src/agents/system-prompt.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAEjC,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AAExD,MAAM,CAAC,KAAK,UAAU,kBAAkB,CAAC,EACvC,WAAW,EACX,UAAU,EACV,YAAY,EACZ,WAAW,GAMZ;IACC,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;IACvB,MAAM,cAAc,GAAG,IAAI,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;IACjD,MAAM,iBAAiB,GAAG,IAAI,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;IACzD,MAAM,eAAe,GAAG,IAAI,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;IACrD,MAAM,QAAQ,GAAG,GAAG,CAAC,cAAc,CAAC,OAAO,EAAE;QAC3C,OAAO,EAAE,MAAM;QACf,IAAI,EAAE,SAAS;QACf,KAAK,EAAE,MAAM;QACb,GAAG,EAAE,SAAS;KACf,CAAC,CAAC;IACH,MAAM,eAAe,GAAG,MAAM,qBAAqB,CAAC,WAAW,CAAC,CAAC;IAEjE,MAAM,MAAM,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;sJAmCqI,eAAe;;;EAGnK,eAAe;;;;;wDAKuC,WAAW,4BAA4B,YAAY;;;;;;;;kBAQzF,WAAW;iBACZ,UAAU;;;mBAGR,YAAY;kBACb,WAAW;;;;;;;4CAOe,cAAc;0CAChB,iBAAiB;gBAC3C,eAAe;IAC3B,eAAe;;2DAEwC,eAAe,8BAA8B,eAAe;oGACnB,WAAW;mDAC5D,WAAW;qBACzC,eAAe,qDAAqD,cAAc;;;;gBAIvF,QAAQ;;CAEvB,CAAC;IAEA,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,KAAK,UAAU,qBAAqB,CAAC,WAAmB;IACtD,MAAM,eAAe,GAAG,MAAM,mBAAmB,CAAC,WAAW,CAAC,CAAC;IAE/D,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACjC,OAAO,kBAAkB,CAAC;IAC5B,CAAC;IAED,OAAO,eAAe;SACnB,GAAG,CACF,CAAC,KAAK,EAAE,EAAE,CACR,WAAW,KAAK,CAAC,IAAI;iBACZ,KAAK,CAAC,WAAW;UACxB,KAAK,CAAC,IAAI,EAAE,CACjB;SACA,IAAI,CAAC,IAAI,CAAC,CAAC;AAChB,CAAC"}

package/dist/agents/tools.d.ts

@@ -0,0 +1,3 @@
+import { createAllTools } from '../tools/index.js';
+export { createAllTools };
+//# sourceMappingURL=tools.d.ts.map

package/dist/agents/tools.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.d.ts","sourceRoot":"","sources":["../../src/agents/tools.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAEnD,OAAO,EAAE,cAAc,EAAE,CAAC"}

package/dist/agents/tools.js

@@ -0,0 +1,3 @@
+import { createAllTools } from '../tools/index.js';
+export { createAllTools };
+//# sourceMappingURL=tools.js.map

package/dist/agents/tools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.js","sourceRoot":"","sources":["../../src/agents/tools.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAEnD,OAAO,EAAE,cAAc,EAAE,CAAC"}

package/dist/helpers/ai-image/ai-image.d.ts

@@ -0,0 +1,75 @@
+export declare const IMAGE_MODEL_IDS: readonly ["gpt-5.4", "gemini-3.1-flash-image-preview", "gemini-3-pro-image-preview"];
+export type ImageModelId = (typeof IMAGE_MODEL_IDS)[number];
+export type ImageSource = string | readonly string[];
+export declare const OPENAI_IMAGE_SIZES: readonly ["1024x1024", "1024x1536", "1536x1024", "auto"];
+export declare const GOOGLE_IMAGE_SIZES: readonly ["1K", "2K", "4K"];
+export declare const GOOGLE_PRO_IMAGE_ASPECT_RATIOS: readonly ["1:1", "2:3", "3:2", "3:4", "4:3", "4:5", "5:4", "9:16", "16:9", "21:9"];
+export declare const GOOGLE_FLASH_IMAGE_ASPECT_RATIOS: readonly ["1:1", "1:4", "1:8", "2:3", "3:2", "3:4", "4:1", "4:3", "4:5", "5:4", "8:1", "9:16", "16:9", "21:9"];
+export type OpenAIImageSize = (typeof OPENAI_IMAGE_SIZES)[number];
+export type GoogleImageSize = (typeof GOOGLE_IMAGE_SIZES)[number];
+export type GoogleProImageAspectRatio = (typeof GOOGLE_PRO_IMAGE_ASPECT_RATIOS)[number];
+export type GoogleFlashImageAspectRatio = (typeof GOOGLE_FLASH_IMAGE_ASPECT_RATIOS)[number];
+export interface OpenAIImageOptions {
+    format?: 'png' | 'jpeg' | 'webp';
+    quality?: 'low' | 'medium' | 'high' | 'auto';
+    background?: 'transparent' | 'opaque' | 'auto';
+    size?: OpenAIImageSize;
+    partialImages?: 0 | 1 | 2 | 3;
+    inputFidelity?: 'low' | 'high';
+    moderation?: 'auto' | 'low';
+}
+export interface GoogleProImageOptions {
+    aspectRatio?: GoogleProImageAspectRatio;
+    imageSize?: GoogleImageSize;
+}
+export interface GoogleFlashImageOptions {
+    aspectRatio?: GoogleFlashImageAspectRatio;
+    imageSize?: GoogleImageSize;
+}
+export type GoogleImageOptions = GoogleFlashImageOptions | GoogleProImageOptions;
+export type ImageOptions = OpenAIImageOptions | GoogleImageOptions;
+export interface OpenAIImageProvider {
+    model: 'gpt-5.4';
+    apiKey?: string;
+    imageOptions?: OpenAIImageOptions;
+}
+export interface GoogleFlashImageProvider {
+    model: 'gemini-3.1-flash-image-preview';
+    apiKey?: string;
+    imageOptions?: GoogleFlashImageOptions;
+}
+export interface GoogleProImageProvider {
+    model: 'gemini-3-pro-image-preview';
+    apiKey?: string;
+    imageOptions?: GoogleProImageOptions;
+}
+export type GoogleImageProvider = GoogleFlashImageProvider | GoogleProImageProvider;
+export type ImageProvider = OpenAIImageProvider | GoogleImageProvider;
+export interface ImageUpdate {
+    stage: 'partial' | 'thought' | 'final';
+    path: string;
+    mimeType: string;
+    index: number;
+    model: ImageModelId;
+}
+interface BaseImageRequest {
+    provider: ImageProvider;
+    prompt: string;
+    outputDir: string;
+    outputName?: string;
+    systemPrompt?: string;
+    onUpdate?: (update: ImageUpdate) => void | Promise<void>;
+}
+export interface CreateImageRequest extends BaseImageRequest {
+    images?: ImageSource;
+}
+export interface EditImageRequest extends BaseImageRequest {
+    images: ImageSource;
+}
+export interface ImageResult {
+    path: string;
+}
+export declare function createImage(request: CreateImageRequest): Promise<ImageResult>;
+export declare function editImage(request: EditImageRequest): Promise<ImageResult>;
+export {};
+//# sourceMappingURL=ai-image.d.ts.map

package/dist/helpers/ai-image/ai-image.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai-image.d.ts","sourceRoot":"","sources":["../../../src/helpers/ai-image/ai-image.ts"],"names":[],"mappings":"AAgBA,eAAO,MAAM,eAAe,sFAIlB,CAAC;AAEX,MAAM,MAAM,YAAY,GAAG,CAAC,OAAO,eAAe,CAAC,CAAC,MAAM,CAAC,CAAC;AAC5D,MAAM,MAAM,WAAW,GAAG,MAAM,GAAG,SAAS,MAAM,EAAE,CAAC;AAErD,eAAO,MAAM,kBAAkB,0DAA2D,CAAC;AAC3F,eAAO,MAAM,kBAAkB,6BAA8B,CAAC;AAC9D,eAAO,MAAM,8BAA8B,oFAWjC,CAAC;AACX,eAAO,MAAM,gCAAgC,gHAenC,CAAC;AAEX,MAAM,MAAM,eAAe,GAAG,CAAC,OAAO,kBAAkB,CAAC,CAAC,MAAM,CAAC,CAAC;AAClE,MAAM,MAAM,eAAe,GAAG,CAAC,OAAO,kBAAkB,CAAC,CAAC,MAAM,CAAC,CAAC;AAClE,MAAM,MAAM,yBAAyB,GAAG,CAAC,OAAO,8BAA8B,CAAC,CAAC,MAAM,CAAC,CAAC;AACxF,MAAM,MAAM,2BAA2B,GAAG,CAAC,OAAO,gCAAgC,CAAC,CAAC,MAAM,CAAC,CAAC;AAE5F,MAAM,WAAW,kBAAkB;IACjC,MAAM,CAAC,EAAE,KAAK,GAAG,MAAM,GAAG,MAAM,CAAC;IACjC,OAAO,CAAC,EAAE,KAAK,GAAG,QAAQ,GAAG,MAAM,GAAG,MAAM,CAAC;IAC7C,UAAU,CAAC,EAAE,aAAa,GAAG,QAAQ,GAAG,MAAM,CAAC;IAC/C,IAAI,CAAC,EAAE,eAAe,CAAC;IACvB,aAAa,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IAC9B,aAAa,CAAC,EAAE,KAAK,GAAG,MAAM,CAAC;IAC/B,UAAU,CAAC,EAAE,MAAM,GAAG,KAAK,CAAC;CAC7B;AAED,MAAM,WAAW,qBAAqB;IACpC,WAAW,CAAC,EAAE,yBAAyB,CAAC;IACxC,SAAS,CAAC,EAAE,eAAe,CAAC;CAC7B;AAED,MAAM,WAAW,uBAAuB;IACtC,WAAW,CAAC,EAAE,2BAA2B,CAAC;IAC1C,SAAS,CAAC,EAAE,eAAe,CAAC;CAC7B;AAED,MAAM,MAAM,kBAAkB,GAAG,uBAAuB,GAAG,qBAAqB,CAAC;AACjF,MAAM,MAAM,YAAY,GAAG,kBAAkB,GAAG,kBAAkB,CAAC;AAEnE,MAAM,WAAW,mBAAmB;IAClC,KAAK,EAAE,SAAS,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,YAAY,CAAC,EAAE,kBAAkB,CAAC;CACnC;AAED,MAAM,WAAW,wBAAwB;IACvC,KAAK,EAAE,gCAAgC,CAAC;IACxC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,YAAY,CAAC,EAAE,uBAAuB,CAAC;CACxC;AAED,MAAM,WAAW,sBAAsB;IACrC,KAAK,EAAE,4BAA4B,CAAC;IACpC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,YAAY,CAAC,EAAE,qBAAqB,CAAC;CACtC;AAED,MAAM,MAAM,mBAAmB,GAAG,wBAAwB,GAAG,sBAAsB,CAAC;AACpF,MAAM,MAAM,aAAa,GAAG,mBAAmB,GAAG,mBAAmB,CAAC;AAEtE,MAAM,WAAW,WAAW;IAC1B,KAAK,EAAE,SAAS,GAAG,SAAS,GAAG,OAAO,CAAC;IACvC,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,YAAY,CAAC;CACrB;AAED,UAAU,gBAAgB;IACxB,QAAQ,EAAE,aAAa,CAAC;IACxB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,WAAW,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CAC1D;AAED,MAAM,WAAW,kBAAmB,SAAQ,gBAAgB;IAC1D,MAAM,CAAC,EAAE,WAAW,CAAC;CACtB;AAED,MAAM,WAAW,gBAAiB,SAAQ,gBAAgB;IACxD,MAAM,EAAE,WAAW,CAAC;CACrB;AAED,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,MAAM,CAAC;CACd;AAohBD,wBAAsB,WAAW,CAAC,OAAO,EAAE,kBAAkB,GAAG,OAAO,CAAC,WAAW,CAAC,CAEnF;AAED,wBAAsB,SAAS,CAAC,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,WAAW,CAAC,CAE/E"}

package/dist/helpers/ai-image/ai-image.js

@@ -0,0 +1,438 @@
+import { mkdir, readFile, writeFile } from 'node:fs/promises';
+import { basename, extname, join, parse, resolve } from 'node:path';
+import { buildUserMessage, getModel, stream as sdkStream } from '@ank1015/llm-sdk';
+import { createFileKeysAdapter } from '@ank1015/llm-sdk-adapters';
+const defaultKeysAdapter = createFileKeysAdapter();
+export const IMAGE_MODEL_IDS = [
+    'gpt-5.4',
+    'gemini-3.1-flash-image-preview',
+    'gemini-3-pro-image-preview',
+];
+export const OPENAI_IMAGE_SIZES = ['1024x1024', '1024x1536', '1536x1024', 'auto'];
+export const GOOGLE_IMAGE_SIZES = ['1K', '2K', '4K'];
+export const GOOGLE_PRO_IMAGE_ASPECT_RATIOS = [
+    '1:1',
+    '2:3',
+    '3:2',
+    '3:4',
+    '4:3',
+    '4:5',
+    '5:4',
+    '9:16',
+    '16:9',
+    '21:9',
+];
+export const GOOGLE_FLASH_IMAGE_ASPECT_RATIOS = [
+    '1:1',
+    '1:4',
+    '1:8',
+    '2:3',
+    '3:2',
+    '3:4',
+    '4:1',
+    '4:3',
+    '4:5',
+    '5:4',
+    '8:1',
+    '9:16',
+    '16:9',
+    '21:9',
+];
+const MIME_TYPE_TO_EXTENSION = {
+    'image/gif': 'gif',
+    'image/jpeg': 'jpg',
+    'image/png': 'png',
+    'image/webp': 'webp',
+};
+const EXTENSION_TO_MIME_TYPE = {
+    '.gif': 'image/gif',
+    '.jpeg': 'image/jpeg',
+    '.jpg': 'image/jpeg',
+    '.png': 'image/png',
+    '.webp': 'image/webp',
+};
+function isImageModelId(value) {
+    return IMAGE_MODEL_IDS.includes(value);
+}
+function assertImageModelId(model) {
+    if (!isImageModelId(model)) {
+        throw new Error(`Unsupported image model "${model}". Supported models: ${IMAGE_MODEL_IDS.join(', ')}.`);
+    }
+}
+function assertImageProvider(provider) {
+    if (typeof provider.model !== 'string') {
+        throw new Error('provider.model is required.');
+    }
+    assertImageModelId(provider.model);
+}
+function isHttpUrl(value) {
+    try {
+        const url = new URL(value);
+        return url.protocol === 'http:' || url.protocol === 'https:';
+    }
+    catch {
+        return false;
+    }
+}
+function inferMimeTypeFromExtension(pathLike) {
+    return EXTENSION_TO_MIME_TYPE[extname(pathLike).toLowerCase()];
+}
+function inferMimeTypeFromBuffer(buffer) {
+    if (buffer.length >= 8 &&
+        buffer[0] === 0x89 &&
+        buffer[1] === 0x50 &&
+        buffer[2] === 0x4e &&
+        buffer[3] === 0x47 &&
+        buffer[4] === 0x0d &&
+        buffer[5] === 0x0a &&
+        buffer[6] === 0x1a &&
+        buffer[7] === 0x0a) {
+        return 'image/png';
+    }
+    if (buffer.length >= 3 && buffer[0] === 0xff && buffer[1] === 0xd8 && buffer[2] === 0xff) {
+        return 'image/jpeg';
+    }
+    if (buffer.length >= 12 &&
+        buffer[0] === 0x52 &&
+        buffer[1] === 0x49 &&
+        buffer[2] === 0x46 &&
+        buffer[3] === 0x46 &&
+        buffer[8] === 0x57 &&
+        buffer[9] === 0x45 &&
+        buffer[10] === 0x42 &&
+        buffer[11] === 0x50) {
+        return 'image/webp';
+    }
+    if (buffer.length >= 6 &&
+        buffer[0] === 0x47 &&
+        buffer[1] === 0x49 &&
+        buffer[2] === 0x46 &&
+        buffer[3] === 0x38 &&
+        (buffer[4] === 0x37 || buffer[4] === 0x39) &&
+        buffer[5] === 0x61) {
+        return 'image/gif';
+    }
+    return undefined;
+}
+function inferMimeTypeFromContentType(contentType) {
+    if (!contentType)
+        return undefined;
+    const mimeType = contentType.split(';')[0]?.trim().toLowerCase();
+    if (!mimeType)
+        return undefined;
+    if (!mimeType.startsWith('image/'))
+        return undefined;
+    return MIME_TYPE_TO_EXTENSION[mimeType] ? mimeType : undefined;
+}
+function getExtensionForMimeType(mimeType) {
+    const extension = MIME_TYPE_TO_EXTENSION[mimeType];
+    if (!extension) {
+        throw new Error(`Unsupported image mime type "${mimeType}".`);
+    }
+    return extension;
+}
+function normalizeImageSources(images) {
+    if (images === undefined) {
+        return [];
+    }
+    return typeof images === 'string' ? [images] : [...images];
+}
+function sanitizeOutputStem(outputName) {
+    if (!outputName) {
+        return `image-${Date.now()}-${Math.random().toString(36).slice(2, 8)}`;
+    }
+    const trimmed = outputName.trim();
+    if (!trimmed) {
+        throw new Error('outputName must not be empty.');
+    }
+    if (trimmed.includes('/') || trimmed.includes('\\') || trimmed === '.' || trimmed === '..') {
+        throw new Error('outputName must be a filename stem, not a path.');
+    }
+    const stem = parse(trimmed).name.trim();
+    if (!stem) {
+        throw new Error('outputName must contain at least one non-extension character.');
+    }
+    return stem;
+}
+async function loadLocalImageSource(source) {
+    const buffer = await readFile(source);
+    const mimeType = inferMimeTypeFromBuffer(buffer) ?? inferMimeTypeFromExtension(source);
+    if (!mimeType) {
+        throw new Error(`Could not determine image type for local source "${source}".`);
+    }
+    return {
+        fileName: basename(source),
+        mimeType,
+        data: buffer.toString('base64'),
+        size: buffer.length,
+    };
+}
+async function loadRemoteImageSource(source, index) {
+    const response = await fetch(source);
+    if (!response.ok) {
+        throw new Error(`Failed to download image source "${source}": ${response.status} ${response.statusText}`);
+    }
+    const arrayBuffer = await response.arrayBuffer();
+    const buffer = Buffer.from(arrayBuffer);
+    const url = new URL(source);
+    const mimeType = inferMimeTypeFromContentType(response.headers.get('content-type')) ??
+        inferMimeTypeFromBuffer(buffer) ??
+        inferMimeTypeFromExtension(url.pathname);
+    if (!mimeType) {
+        throw new Error(`Could not determine image type for remote source "${source}".`);
+    }
+    const fallbackFileName = `source-image-${index}.${getExtensionForMimeType(mimeType)}`;
+    const nameFromUrl = basename(url.pathname);
+    return {
+        fileName: nameFromUrl || fallbackFileName,
+        mimeType,
+        data: buffer.toString('base64'),
+        size: buffer.length,
+    };
+}
+async function loadImageSource(source, index) {
+    return isHttpUrl(source) ? loadRemoteImageSource(source, index) : loadLocalImageSource(source);
+}
+function resolveImageModel(provider) {
+    if (provider.model === 'gpt-5.4') {
+        const model = getModel('openai', provider.model);
+        if (!model) {
+            throw new Error(`Could not resolve model "${provider.model}".`);
+        }
+        return model;
+    }
+    const model = getModel('google', provider.model);
+    if (!model) {
+        throw new Error(`Could not resolve model "${provider.model}".`);
+    }
+    return model;
+}
+function validateImageOptions(method, provider) {
+    if (provider.model === 'gpt-5.4') {
+        const imageOptions = provider.imageOptions;
+        if (!imageOptions)
+            return;
+        const openAIImageOptions = imageOptions;
+        if ('aspectRatio' in openAIImageOptions) {
+            throw new Error('provider.imageOptions.aspectRatio is only supported for Google image models.');
+        }
+        if ('imageSize' in openAIImageOptions) {
+            throw new Error('provider.imageOptions.imageSize is only supported for Google image models.');
+        }
+        if (imageOptions.size !== undefined && !OPENAI_IMAGE_SIZES.includes(imageOptions.size)) {
+            throw new Error(`provider.imageOptions.size must be one of: ${OPENAI_IMAGE_SIZES.join(', ')}.`);
+        }
+        if (method === 'create' && imageOptions.inputFidelity !== undefined) {
+            throw new Error('provider.imageOptions.inputFidelity is only supported for editImage() with OpenAI.');
+        }
+        return;
+    }
+    const imageOptions = provider.imageOptions;
+    if (!imageOptions)
+        return;
+    const googleImageOptions = imageOptions;
+    for (const openAIOnlyKey of [
+        'format',
+        'quality',
+        'background',
+        'size',
+        'partialImages',
+        'inputFidelity',
+        'moderation',
+    ]) {
+        if (openAIOnlyKey in googleImageOptions) {
+            throw new Error(`provider.imageOptions.${openAIOnlyKey} is only supported for OpenAI image models.`);
+        }
+    }
+    if (imageOptions.imageSize !== undefined &&
+        !GOOGLE_IMAGE_SIZES.includes(imageOptions.imageSize)) {
+        throw new Error(`provider.imageOptions.imageSize must be one of: ${GOOGLE_IMAGE_SIZES.join(', ')}.`);
+    }
+    const allowedAspectRatios = provider.model === 'gemini-3.1-flash-image-preview'
+        ? GOOGLE_FLASH_IMAGE_ASPECT_RATIOS
+        : GOOGLE_PRO_IMAGE_ASPECT_RATIOS;
+    if (imageOptions.aspectRatio !== undefined &&
+        !allowedAspectRatios.includes(imageOptions.aspectRatio)) {
+        throw new Error(`provider.imageOptions.aspectRatio must be one of: ${allowedAspectRatios.join(', ')}.`);
+    }
+}
+function buildOpenAIProviderOptions(method, provider, signal) {
+    const imageOptions = provider.imageOptions;
+    const tool = {
+        type: 'image_generation',
+        action: method === 'edit' ? 'edit' : 'generate',
+    };
+    if (imageOptions?.partialImages !== undefined) {
+        tool.partial_images = imageOptions.partialImages;
+    }
+    if (imageOptions?.inputFidelity !== undefined) {
+        tool.input_fidelity = imageOptions.inputFidelity;
+    }
+    if (imageOptions?.size !== undefined) {
+        tool.size = imageOptions.size;
+    }
+    if (imageOptions?.quality !== undefined) {
+        tool.quality = imageOptions.quality;
+    }
+    if (imageOptions?.background !== undefined) {
+        tool.background = imageOptions.background;
+    }
+    if (imageOptions?.format !== undefined) {
+        tool.output_format = imageOptions.format;
+    }
+    if (imageOptions?.moderation !== undefined) {
+        tool.moderation = imageOptions.moderation;
+    }
+    return {
+        ...(provider.apiKey ? { apiKey: provider.apiKey } : {}),
+        signal,
+        tools: [tool],
+    };
+}
+function buildGoogleProviderOptions(provider, signal) {
+    const imageConfig = {};
+    if (provider.imageOptions?.aspectRatio !== undefined) {
+        imageConfig.aspectRatio = provider.imageOptions.aspectRatio;
+    }
+    if (provider.imageOptions?.imageSize !== undefined) {
+        imageConfig.imageSize = provider.imageOptions.imageSize;
+    }
+    return {
+        ...(provider.apiKey ? { apiKey: provider.apiKey } : {}),
+        signal,
+        responseModalities: ['IMAGE'],
+        ...(Object.keys(imageConfig).length > 0 ? { imageConfig } : {}),
+    };
+}
+function getFirstFinalImage(message) {
+    for (const block of message.content) {
+        if (block.type !== 'response')
+            continue;
+        for (const content of block.content) {
+            if (content.type === 'image' && content.metadata?.generationStage === 'final') {
+                return content;
+            }
+        }
+    }
+    return undefined;
+}
+async function writeImage(filePath, image) {
+    await writeFile(filePath, Buffer.from(image.data, 'base64'));
+}
+function getEventImageStage(event) {
+    if (event.type === 'image_start') {
+        return event.metadata?.generationStage;
+    }
+    if (event.type === 'image_frame' || event.type === 'image_end') {
+        const stage = event.image.metadata?.generationStage;
+        return stage === 'partial' || stage === 'thought' || stage === 'final' ? stage : undefined;
+    }
+    return undefined;
+}
+async function runImageRequest(method, request) {
+    assertImageProvider(request.provider);
+    validateImageOptions(method, request.provider);
+    const imageSources = normalizeImageSources(request.images);
+    if (method === 'edit' && imageSources.length === 0) {
+        throw new Error('editImage() requires at least one image source.');
+    }
+    const outputDir = resolve(request.outputDir);
+    const baseName = sanitizeOutputStem(request.outputName);
+    const updatesDir = join(outputDir, `${baseName}__updates`);
+    await mkdir(outputDir, { recursive: true });
+    await mkdir(updatesDir, { recursive: true });
+    const loadedImages = await Promise.all(imageSources.map((source, index) => loadImageSource(source, index + 1)));
+    const attachments = loadedImages.map((image, index) => ({
+        id: `source-image-${index + 1}`,
+        type: 'image',
+        fileName: image.fileName,
+        mimeType: image.mimeType,
+        size: image.size,
+        content: image.data,
+    }));
+    const userMessage = buildUserMessage(request.prompt, attachments);
+    const model = resolveImageModel(request.provider);
+    const abortController = new AbortController();
+    const counters = {
+        partial: 0,
+        thought: 0,
+        final: 0,
+    };
+    let firstFinalUpdateSaved = false;
+    let firstFinalImage;
+    const saveUpdate = async (stage, image) => {
+        if (stage === 'final') {
+            if (firstFinalUpdateSaved) {
+                return undefined;
+            }
+            firstFinalUpdateSaved = true;
+            firstFinalImage = image;
+        }
+        const index = ++counters[stage];
+        const extension = getExtensionForMimeType(image.mimeType);
+        const artifactPath = join(updatesDir, `${stage}-${String(index).padStart(3, '0')}.${extension}`);
+        await writeImage(artifactPath, image);
+        if (request.onUpdate) {
+            await request.onUpdate({
+                stage,
+                path: artifactPath,
+                mimeType: image.mimeType,
+                index,
+                model: request.provider.model,
+            });
+        }
+        return artifactPath;
+    };
+    try {
+        const providerOptions = request.provider.model === 'gpt-5.4'
+            ? buildOpenAIProviderOptions(method, request.provider, abortController.signal)
+            : buildGoogleProviderOptions(request.provider, abortController.signal);
+        const eventStream = await sdkStream(model, {
+            messages: [userMessage],
+            ...(request.systemPrompt ? { systemPrompt: request.systemPrompt } : {}),
+        }, {
+            keysAdapter: defaultKeysAdapter,
+            providerOptions,
+        });
+        for await (const event of eventStream) {
+            const stage = getEventImageStage(event);
+            if (!stage)
+                continue;
+            if (stage === 'partial' && event.type === 'image_frame') {
+                await saveUpdate(stage, event.image);
+            }
+            if ((stage === 'thought' || stage === 'final') && event.type === 'image_end') {
+                await saveUpdate(stage, event.image);
+            }
+        }
+        const message = await eventStream.result();
+        if (message.stopReason === 'error' || message.stopReason === 'aborted') {
+            throw new Error(message.errorMessage ||
+                (message.stopReason === 'aborted'
+                    ? 'Image request was aborted.'
+                    : 'Image request failed.'));
+        }
+        const finalImage = firstFinalImage ?? getFirstFinalImage(message);
+        if (!finalImage) {
+            throw new Error(`Model "${request.provider.model}" did not return a final image.`);
+        }
+        if (!firstFinalUpdateSaved) {
+            await saveUpdate('final', finalImage);
+        }
+        const extension = getExtensionForMimeType(finalImage.mimeType);
+        const finalPath = join(outputDir, `${baseName}.${extension}`);
+        await writeImage(finalPath, finalImage);
+        return { path: finalPath };
+    }
+    catch (error) {
+        abortController.abort();
+        throw error;
+    }
+}
+export async function createImage(request) {
+    return runImageRequest('create', request);
+}
+export async function editImage(request) {
+    return runImageRequest('edit', request);
+}
+//# sourceMappingURL=ai-image.js.map