@thegitai/cli 1.0.0-preview.25 → 1.0.0-preview.26

package/dist/src/api/chat.js

@@ -2,6 +2,7 @@ import { drainBackgroundJobNotifications } from '../background-jobs.js';
 import { createPromptCheckpoint, sanitizeSessionSafetyForServer, } from '../session-safety.js';
 import { applySessionSnapshot, saveSessionState, snapshotFromSession, } from '../session-store.js';
 import { executeLocalToolCall } from '../tool-executor.js';
+import { saveGeneratedImage } from '../tools/save-generated-image.js';
 import { isUserInputQuestionArray } from './contracts.js';
 import { createTraceContext, gatewayFailureCategory, normalizeServerUrl, readErrorResponse, } from './http.js';
 import { collectClientEnvironment } from '../client-environment.js';
@@ -165,6 +166,9 @@ function publicStatusMessage(data) {
     if (event.phase === 'analyzing_image') {
         return (event.imageCount ?? 1) > 1 ? 'Analyzing images...' : 'Analyzing image...';
     }
+    if (event.phase === 'generating_image') {
+        return 'Generating image...';
+    }
     if (event.phase === 'running_tool')
         return `Running ${toolName}...`;
     if (event.phase === 'waiting_for_tool')
@@ -312,7 +316,14 @@ async function executeAndPostToolResult({ config, projectIndex, session, event,
     }
     try {
         const call = normalizeShellJobToolCall(event.call);
-        const rawResult = await executeLocalToolCall({ projectIndex }, session, call);
+        const rawResult = event.generatedImage
+            ? saveGeneratedImage({
+                base64Data: event.generatedImage.base64Data,
+                mimeType: event.generatedImage.mimeType,
+                suggestedFilename: event.generatedImage.suggestedFilename ||
+                    String(call.args?.filename ?? call.args?.file_name ?? ''),
+            })
+            : await executeLocalToolCall({ projectIndex }, session, call);
         preserveCancelledTurnToolResult(session, input, { ...event, call }, rawResult);
         if (signal?.aborted) {
             throw new TurnCancelledError();
@@ -383,9 +394,15 @@ async function consumeTurnStream({ response, config, projectIndex, session, inpu
             const data = event.data;
             if (data?.phase === 'analyzing_image') {
                 session.onImageAnalysis?.(Math.max(1, Number(data.imageCount ?? 1) || 1));
+                session.onImageGeneration?.(false);
+            }
+            else if (data?.phase === 'generating_image') {
+                session.onImageGeneration?.(true);
+                session.onImageAnalysis?.(0);
             }
             else if (data?.phase) {
                 session.onImageAnalysis?.(0);
+                session.onImageGeneration?.(false);
             }
             const message = publicStatusMessage(event.data);
             if (message)

package/dist/src/session-store.js

@@ -4,6 +4,7 @@ import { existsSync, mkdirSync, readFileSync, readdirSync, renameSync, rmSync, w
 import path from 'node:path';
 import { getClientStateDir } from './client-state.js';
 import { findStoredImageByContent, pruneSessionImages, readSessionImage, sweepOrphanSessionImages, } from './core/session-image-store.js';
+import { sweepGeneratedImages } from './tools/save-generated-image.js';
 import { normalizeAssistantEditJournal } from './edit-journal.js';
 import { createSessionGrants } from './permissions.js';
 import { cloneSessionSafetyState, createSessionSafetyState, mergeLocalSessionSafetyState, normalizeSessionSafetyState, } from './session-safety.js';
@@ -224,6 +225,7 @@ export function pruneSavedSessions(rootDir, env = process.env) {
         activeSessionIds: listAllSessionIds(env),
         env,
     });
+    sweepGeneratedImages({ env });
 }
 export function readGitBranch(rootDir) {
     try {

package/dist/src/tools/index.js

@@ -23,6 +23,7 @@ import { strReplace } from './str-replace.js';
 import { undoEdit } from './undo-edit.js';
 import { updateTodos } from './update-todos.js';
 import { readImageFile } from './read-image-file.js';
+import { saveGeneratedImage } from './save-generated-image.js';
 import { writeFile } from './write-file.js';
 export const TOOL_MAP = {
     search_code: (context, args) => searchCode(context.projectIndex, args),
@@ -52,6 +53,16 @@ export const TOOL_MAP = {
     shell_job_kill: shellJobKill,
     update_todos: updateTodos,
     analyze_image: (context, args) => readImageFile(context, args),
+    generate_image: (_context, args) => {
+        if (!String(args.base64Data ?? '').trim()) {
+            return {
+                ok: false,
+                error: 'This CLI build cannot generate images without server-supplied image bytes. Update TheGitAI CLI.',
+                failureCategory: 'tool_exception',
+            };
+        }
+        return saveGeneratedImage(args);
+    },
 };
 function invalidToolCall(error) {
     return {

package/dist/src/tools/save-generated-image.js

@@ -0,0 +1,120 @@
+import { existsSync, mkdirSync, readdirSync, rmSync, statSync, writeFileSync, } from 'node:fs';
+import path from 'node:path';
+import { getClientStateDir } from '../client-state.js';
+import { MAX_IMAGE_SIZE_BYTES, sniffImageMimeType, } from '../core/image-limits.js';
+const IMAGE_FILE_MODE = 0o600;
+const IMAGE_DIR_MODE = 0o700;
+export const GENERATED_IMAGES_SUBDIR = 'generated_images';
+export const GENERATED_IMAGE_MAX_AGE_MS = 30 * 24 * 60 * 60 * 1000;
+export function getGeneratedImagesDir(env = process.env) {
+    return path.join(getClientStateDir(env), GENERATED_IMAGES_SUBDIR);
+}
+export function sanitizeGeneratedImageBasename(raw) {
+    const trimmed = String(raw ?? '').trim();
+    const base = trimmed
+        .replace(/\\/g, '/')
+        .split('/')
+        .pop()
+        ?.replace(/^\.+/, '')
+        .replace(/[^\w.\-]+/g, '-')
+        .replace(/-+/g, '-')
+        .replace(/^-+|-+$/g, '')
+        .replace(/-\.|\.-/g, '.')
+        .replace(/^-+|-+$/g, '')
+        .slice(0, 80);
+    if (!base || base === '.' || base === '..') {
+        return `generated-${Date.now()}.png`;
+    }
+    if (/\.png$/i.test(base)) {
+        return base.replace(/\.png$/i, '.png');
+    }
+    const withoutExt = base.replace(/\.[^.]+$/, '').replace(/-+$/g, '');
+    return `${withoutExt || 'generated'}.png`;
+}
+function uniquePath(dir, basename) {
+    const candidate = path.join(dir, basename);
+    if (!existsSync(candidate)) {
+        return candidate;
+    }
+    const ext = path.extname(basename) || '.png';
+    const stem = path.basename(basename, ext);
+    for (let i = 2; i < 10_000; i += 1) {
+        const next = path.join(dir, `${stem}-${i}${ext}`);
+        if (!existsSync(next)) {
+            return next;
+        }
+    }
+    return path.join(dir, `${stem}-${Date.now()}${ext}`);
+}
+export function sweepGeneratedImages({ maxAgeMs = GENERATED_IMAGE_MAX_AGE_MS, env = process.env, now = Date.now(), } = {}) {
+    const dir = getGeneratedImagesDir(env);
+    if (!existsSync(dir))
+        return 0;
+    let removed = 0;
+    let entries;
+    try {
+        entries = readdirSync(dir);
+    }
+    catch {
+        return 0;
+    }
+    for (const entry of entries) {
+        const filePath = path.join(dir, entry);
+        try {
+            if (now - statSync(filePath).mtimeMs < maxAgeMs)
+                continue;
+            rmSync(filePath, { force: true });
+            removed += 1;
+        }
+        catch {
+        }
+    }
+    return removed;
+}
+export function saveGeneratedImage(args) {
+    const base64Data = String(args.base64Data ?? '').trim();
+    if (!base64Data) {
+        return {
+            ok: false,
+            error: 'This CLI build cannot generate images without server-supplied image bytes. Update TheGitAI CLI.',
+            failureCategory: 'tool_exception',
+        };
+    }
+    let bytes;
+    try {
+        bytes = Buffer.from(base64Data, 'base64');
+    }
+    catch {
+        return {
+            ok: false,
+            error: 'Generated image bytes are invalid.',
+            failureCategory: 'invalid_argument',
+        };
+    }
+    if (bytes.length === 0 || bytes.length > MAX_IMAGE_SIZE_BYTES) {
+        return {
+            ok: false,
+            error: 'Generated image exceeded the allowed size limit.',
+            failureCategory: 'invalid_argument',
+        };
+    }
+    const sniffed = sniffImageMimeType(bytes);
+    if (sniffed !== 'image/png') {
+        return {
+            ok: false,
+            error: 'Generated image must be a PNG.',
+            failureCategory: 'invalid_argument',
+        };
+    }
+    const env = args.env ?? process.env;
+    const dir = getGeneratedImagesDir(env);
+    mkdirSync(dir, { recursive: true, mode: IMAGE_DIR_MODE });
+    const basename = sanitizeGeneratedImageBasename(args.suggestedFilename ?? args.filename);
+    const target = uniquePath(dir, basename);
+    writeFileSync(target, bytes, { mode: IMAGE_FILE_MODE });
+    return {
+        ok: true,
+        path: target,
+        message: `Image saved to ${target}`,
+    };
+}

package/dist/src/ui/repl.js

@@ -982,6 +982,7 @@ function createInitialShellState(session, serverModels, debugUi) {
         activeTurnInputPreformatted: false,
         agentMode: session.agentMode,
         analyzingImages: 0,
+        generatingImage: false,
         approvalCursor: 0,
         approvalPrompt: null,
         approvalScrollOffset: 0,
@@ -2623,6 +2624,7 @@ export async function runClientInteractive({ appendPromptHistory, authConfig, de
                 activeTurnInput: input,
                 activeTurnInputPreformatted: preformatted,
                 analyzingImages: 0,
+                generatingImage: false,
                 busy: true,
                 busyPausedAt: null,
                 busySince: turnStartedAt,
@@ -2777,6 +2779,14 @@ export async function runClientInteractive({ appendPromptHistory, authConfig, de
             store.update((current) => ({
                 ...current,
                 analyzingImages: Math.max(0, activeImageCount),
+                generatingImage: activeImageCount > 0 ? false : current.generatingImage,
+            }));
+        };
+        session.onImageGeneration = (active) => {
+            store.update((current) => ({
+                ...current,
+                generatingImage: Boolean(active),
+                analyzingImages: active ? 0 : current.analyzingImages,
             }));
         };
         session.onStatus = (message) => {

package/dist/src/ui/tui/build-frame.js

@@ -953,11 +953,13 @@ export function buildWorkingClockLine(state, elapsedSeconds) {
     if (state.busyPausedAt != null) {
         return `${WORKING_CLOCK_ICON} Paused · ${elapsed} · waiting for your response`;
     }
-    const label = state.analyzingImages > 0
-        ? state.analyzingImages > 1
-            ? 'Analyzing images'
-            : 'Analyzing image'
-        : 'Working';
+    const label = state.generatingImage
+        ? 'Generating image'
+        : state.analyzingImages > 0
+            ? state.analyzingImages > 1
+                ? 'Analyzing images'
+                : 'Analyzing image'
+            : 'Working';
     return `${WORKING_CLOCK_ICON} ${label} · ${elapsed}`;
 }
 export function approvalPanelInnerWidth(width) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@thegitai/cli",
-  "version": "1.0.0-preview.25",
+  "version": "1.0.0-preview.26",
   "description": "TheGitAI is an AI coding agent for your terminal. It indexes your repository, writes and edits files, runs commands, and builds features with you.",
   "keywords": [
     "ai",
@@ -37,10 +37,10 @@
     "@lydell/node-pty-linux-x64": "1.1.0",
     "@lydell/node-pty-win32-arm64": "1.1.0",
     "@lydell/node-pty-win32-x64": "1.1.0",
-    "@thegitai/tui-darwin-arm64": "1.0.0-preview.25",
-    "@thegitai/tui-darwin-x64": "1.0.0-preview.25",
-    "@thegitai/tui-linux-x64": "1.0.0-preview.25",
-    "@thegitai/tui-win32-x64": "1.0.0-preview.25",
+    "@thegitai/tui-darwin-arm64": "1.0.0-preview.26",
+    "@thegitai/tui-darwin-x64": "1.0.0-preview.26",
+    "@thegitai/tui-linux-x64": "1.0.0-preview.26",
+    "@thegitai/tui-win32-x64": "1.0.0-preview.26",
     "@vscode/ripgrep": "1.18.0"
   },
   "publishConfig": {