askimo 1.6.0 → 1.7.1

package/README.md

@@ -36,6 +36,7 @@ OPENAI_MODEL=gpt-4o
 ANTHROPIC_MODEL=claude-sonnet-4-20250514
 XAI_MODEL=grok-4
 GEMINI_MODEL=gemini-3-pro-preview
+IMAGE_MODEL=gemini-3-pro-image-preview
 ```
 
 ---
@@ -65,6 +66,16 @@ askimo "what's happening today?" -x      # Use xAI Grok
 askimo "summarize this topic" -g         # Use Google Gemini
 ```
 
+### Generate an image
+
+```bash
+askimo --image "a cat wearing a top hat"          # Save to current directory
+askimo --image -d ./images "a sunset over ocean"  # Save to ./images/
+askimo --image --json "a dog on a skateboard"     # JSON output with image paths
+```
+
+Uses Google Gemini's image generation model. Requires `GOOGLE_GENERATIVE_AI_API_KEY` in your config.
+
 ### Continue a conversation
 
 ```bash
@@ -123,14 +134,15 @@ askimo conversations       # Opens browser with all conversations
 
 ## ✨ Features
 
-| Feature        | Description                                       |
-|----------------|---------------------------------------------------|
-| Streaming      | Real-time response output                         |
-| Piping         | Pipe content via stdin                            |
-| File input     | Read content from files with `-f`                 |
-| Citations      | Source links with Perplexity                      |
-| History        | Conversations saved to `~/.askimo/conversations/` |
-| Multi-provider | Switch between AI providers easily                |
+| Feature          | Description                                       |
+|------------------|---------------------------------------------------|
+| Streaming        | Real-time response output                         |
+| Image generation | Generate images with `-i` (uses Gemini)           |
+| Piping           | Pipe content via stdin                            |
+| File input       | Read content from files with `-f`                 |
+| Citations        | Source links with Perplexity                      |
+| History          | Conversations saved to `~/.askimo/conversations/` |
+| Multi-provider   | Switch between AI providers easily                |
 
 ---
 

package/index.mjs

@@ -5,9 +5,17 @@ import { startChat } from './lib/chat.mjs'
 import { ensureDirectories, loadConfig } from './lib/config.mjs'
 import { createConversation, loadConversation, loadConversationById, saveConversation } from './lib/conversation.mjs'
 import { showConversationsInBrowser } from './lib/conversationsUi.mjs'
+import { saveImages } from './lib/image.mjs'
 import { buildMessage, readFile, readStdin } from './lib/input.mjs'
-import { DEFAULT_MODELS, determineProvider, getProvider, listModels } from './lib/providers.mjs'
-import { generateResponse, outputJson, printResponse, streamResponse } from './lib/stream.mjs'
+import { DEFAULT_MODELS, determineProvider, getImageProvider, getProvider, listModels } from './lib/providers.mjs'
+import {
+  generateImageResponse,
+  generateResponse,
+  outputJson,
+  printImageResponse,
+  printResponse,
+  streamResponse
+} from './lib/stream.mjs'
 import pkg from './package.json' with { type: 'json' }
 
 const program = new Command()
@@ -28,6 +36,8 @@ program
   .option('-c, --continue <n>', 'Continue conversation N (1=last, 2=second-to-last)', Number.parseInt)
   .option('--cid <id>', 'Continue conversation by ID')
   .option('-f, --file <path>', 'Read content from file')
+  .option('-i, --image', 'Generate an image (uses Gemini)')
+  .option('-d, --output-dir <path>', 'Directory to save generated images (default: current directory)')
   .action(async (question, options) => {
     try {
       const stdinContent = await readStdin()
@@ -49,8 +59,19 @@ program
       const config = await loadConfig()
       await ensureDirectories()
 
-      const providerName = determineProvider(options, config)
-      const { model, name, modelName } = getProvider(providerName, config)
+      let model
+      let name
+      let modelName
+
+      if (options.image) {
+        if (options.perplexity || options.openai || options.anthropic || options.xai) {
+          console.error('Note: --image uses Gemini regardless of other provider flags')
+        }
+        ;({ model, name, modelName } = getImageProvider(config))
+      } else {
+        const providerName = determineProvider(options, config)
+        ;({ model, name, modelName } = getProvider(providerName, config))
+      }
 
       let conversation
       let existingPath = null
@@ -87,7 +108,23 @@ program
 
       let responseText
 
-      if (options.json) {
+      if (options.image) {
+        const { text, files, duration } = await generateImageResponse(model, conversation.messages)
+        const imagePaths = await saveImages(files, options.outputDir)
+        responseText = text || ''
+        conversation.messages.push({
+          role: 'assistant',
+          content: responseText,
+          images: imagePaths
+        })
+        await saveConversation(conversation, existingPath)
+
+        if (options.json) {
+          outputJson(conversation, responseText, undefined, duration, imagePaths)
+        } else {
+          printImageResponse(responseText, imagePaths, duration, modelName)
+        }
+      } else if (options.json) {
         const { text, sources, duration } = await generateResponse(model, conversation.messages)
         responseText = text
         conversation.messages.push({

package/lib/config.mjs

@@ -51,4 +51,4 @@ async function ensureDirectories() {
   await mkdir(CONVERSATIONS_DIR, { recursive: true })
 }
 
-export { loadConfig, ensureDirectories, parseConfig, ASKIMO_DIR, CONVERSATIONS_DIR }
+export { ASKIMO_DIR, CONVERSATIONS_DIR, ensureDirectories, loadConfig, parseConfig }

package/lib/conversation.mjs

@@ -127,4 +127,4 @@ async function saveConversation(conversation, existingPath = null) {
   return filePath
 }
 
-export { createConversation, loadConversation, loadConversationById, getAllConversations, saveConversation }
+export { createConversation, getAllConversations, loadConversation, loadConversationById, saveConversation }

package/lib/conversationsUi.mjs

@@ -386,4 +386,4 @@ async function showConversationsInBrowser() {
   hcat(html, { port: 0 })
 }
 
-export { showConversationsInBrowser, generateHtml }
+export { generateHtml, showConversationsInBrowser }

package/lib/image.mjs

@@ -0,0 +1,36 @@
+import fs from 'node:fs/promises'
+import path from 'node:path'
+
+const MEDIA_TYPE_EXTENSIONS = {
+  'image/png': '.png',
+  'image/jpeg': '.jpg',
+  'image/webp': '.webp'
+}
+
+function buildImagePath(outputDir, timestamp, index, mediaType) {
+  const ext = MEDIA_TYPE_EXTENSIONS[mediaType] || '.png'
+  const safestamp = timestamp.replace(/:/g, '-')
+  const filename = `askimo-${safestamp}-${index}${ext}`
+  return path.join(outputDir, filename)
+}
+
+async function saveImages(files, outputDir) {
+  if (!files?.length) return []
+
+  const dir = outputDir || process.cwd()
+  await fs.mkdir(dir, { recursive: true })
+
+  const timestamp = new Date().toISOString()
+  const savedPaths = []
+
+  for (let i = 0; i < files.length; i++) {
+    const file = files[i]
+    const filePath = buildImagePath(dir, timestamp, i + 1, file.mediaType)
+    await fs.writeFile(filePath, file.uint8Array)
+    savedPaths.push(filePath)
+  }
+
+  return savedPaths
+}
+
+export { buildImagePath, saveImages }

package/lib/input.mjs

@@ -52,4 +52,4 @@ function buildMessage(prompt, content) {
   return content || prompt || null
 }
 
-export { readStdin, readFile, buildMessage }
+export { buildMessage, readFile, readStdin }

package/lib/providers.mjs

@@ -12,6 +12,8 @@ const DEFAULT_MODELS = {
   gemini: 'gemini-3-pro-preview'
 }
 
+const DEFAULT_IMAGE_MODEL = 'gemini-3-pro-image-preview'
+
 // Provider name mapping (askimo name → models.dev name)
 const PROVIDER_MAP = {
   gemini: 'google'
@@ -112,6 +114,20 @@ function getProvider(providerName, config) {
   }
 }
 
+function getImageProvider(config) {
+  const apiKey = config.GOOGLE_GENERATIVE_AI_API_KEY
+  if (!apiKey) {
+    throw new Error('GOOGLE_GENERATIVE_AI_API_KEY not found in config')
+  }
+  const modelName = config.IMAGE_MODEL || DEFAULT_IMAGE_MODEL
+  const google = createGoogleGenerativeAI({ apiKey })
+  return {
+    model: google(modelName),
+    name: 'gemini',
+    modelName
+  }
+}
+
 function determineProvider(options, config = {}) {
   if (options.openai) return 'openai'
   if (options.anthropic) return 'anthropic'
@@ -127,4 +143,4 @@ function determineProvider(options, config = {}) {
   return 'perplexity'
 }
 
-export { getProvider, determineProvider, DEFAULT_MODELS, listModels }
+export { DEFAULT_MODELS, determineProvider, getImageProvider, getProvider, listModels }

package/lib/stream.mjs

@@ -55,6 +55,22 @@ async function generateResponse(model, messages) {
   return { text, sources, duration }
 }
 
+async function generateImageResponse(model, messages) {
+  const startTime = Date.now()
+  process.stdout.write('\x1b[2mGenerating image...\x1b[0m')
+
+  const { text, files } = await generateText({
+    model,
+    messages
+  })
+
+  process.stdout.write('\r\x1b[K')
+
+  const duration = Date.now() - startTime
+  const imageFiles = files?.filter((f) => f.mediaType?.startsWith('image/'))
+  return { text, files: imageFiles, duration }
+}
+
 function printResponse(text, sources, duration, modelName) {
   process.stdout.write(text)
   process.stdout.write('\n')
@@ -74,7 +90,7 @@ function printResponse(text, sources, duration, modelName) {
   process.stdout.write(`\n\x1b[2m${modelName} · ${formatDuration(duration)}\x1b[0m\n`)
 }
 
-function buildJsonOutput(conversation, response, sources, duration) {
+function buildJsonOutput(conversation, response, sources, duration, images) {
   const lastUserMessage = conversation.messages.findLast((m) => m.role === 'user')
   const output = {
     provider: conversation.provider,
@@ -90,12 +106,43 @@ function buildJsonOutput(conversation, response, sources, duration) {
     output.sources = sources
   }
 
+  if (images?.length > 0) {
+    output.images = images
+  }
+
   return output
 }
 
-function outputJson(conversation, response, sources, duration) {
-  const output = buildJsonOutput(conversation, response, sources, duration)
+function printImageResponse(text, imagePaths, duration, modelName) {
+  if (text) {
+    process.stdout.write(text)
+    process.stdout.write('\n')
+  }
+
+  if (imagePaths?.length > 0) {
+    for (const imgPath of imagePaths) {
+      process.stdout.write(`\x1b[2m[image]\x1b[0m ${imgPath}\n`)
+    }
+  }
+
+  if (!text && !imagePaths?.length) {
+    process.stdout.write('No image or text was generated.\n')
+  }
+
+  process.stdout.write(`\n\x1b[2m${modelName} · ${formatDuration(duration)}\x1b[0m\n`)
+}
+
+function outputJson(conversation, response, sources, duration, images) {
+  const output = buildJsonOutput(conversation, response, sources, duration, images)
   console.log(JSON.stringify(output, null, 2))
 }
 
-export { streamResponse, generateResponse, printResponse, outputJson, buildJsonOutput }
+export {
+  buildJsonOutput,
+  generateImageResponse,
+  generateResponse,
+  outputJson,
+  printImageResponse,
+  printResponse,
+  streamResponse
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "askimo",
-  "version": "1.6.0",
+  "version": "1.7.1",
   "description": "A CLI tool for communicating with AI providers (Perplexity, OpenAI, Anthropic)",
   "license": "Apache-2.0",
   "author": "Amit Tal",
@@ -28,17 +28,17 @@
   "repository": "https://github.com/amitosdev/askimo.git",
   "homepage": "https://github.com/amitosdev/askimo#readme",
   "devDependencies": {
-    "@biomejs/biome": "^2.4.4",
+    "@biomejs/biome": "^2.4.8",
     "ava": "^6.4.1"
   },
   "dependencies": {
-    "@ai-sdk/anthropic": "^3.0.46",
-    "@ai-sdk/google": "^3.0.30",
-    "@ai-sdk/openai": "^3.0.30",
-    "@ai-sdk/perplexity": "^3.0.19",
-    "@ai-sdk/xai": "^3.0.57",
-    "@inquirer/input": "^5.0.7",
-    "ai": "^6.0.97",
+    "@ai-sdk/anthropic": "^3.0.64",
+    "@ai-sdk/google": "^3.0.53",
+    "@ai-sdk/openai": "^3.0.48",
+    "@ai-sdk/perplexity": "^3.0.26",
+    "@ai-sdk/xai": "^3.0.74",
+    "@inquirer/input": "^5.0.10",
+    "ai": "^6.0.138",
     "commander": "^14.0.3",
     "hcat": "^2.2.1"
   },

package/test/image.mjs

@@ -0,0 +1,63 @@
+import fs from 'node:fs/promises'
+import path from 'node:path'
+import test from 'ava'
+import { buildImagePath, saveImages } from '../lib/image.mjs'
+
+test('saveImages returns empty array when files is undefined', async (t) => {
+  const result = await saveImages(undefined)
+  t.deepEqual(result, [])
+})
+
+test('saveImages returns empty array when files is empty', async (t) => {
+  const result = await saveImages([])
+  t.deepEqual(result, [])
+})
+
+test('saveImages writes files and returns paths', async (t) => {
+  const absDir = path.join(process.cwd(), `test-saveImages-${Date.now()}`)
+
+  const files = [
+    { mediaType: 'image/png', uint8Array: Buffer.from('fake-png') },
+    { mediaType: 'image/jpeg', uint8Array: Buffer.from('fake-jpg') }
+  ]
+
+  const paths = await saveImages(files, absDir)
+
+  t.is(paths.length, 2)
+  t.true(paths[0].endsWith('.png'))
+  t.true(paths[1].endsWith('.jpg'))
+
+  const content0 = await fs.readFile(paths[0])
+  t.deepEqual(content0, Buffer.from('fake-png'))
+
+  const content1 = await fs.readFile(paths[1])
+  t.deepEqual(content1, Buffer.from('fake-jpg'))
+
+  // cleanup
+  await fs.rm(absDir, { recursive: true })
+})
+
+test('buildImagePath uses correct extension for png', (t) => {
+  const result = buildImagePath('/tmp', '2026-01-01T00:00:00.000Z', 1, 'image/png')
+  t.true(result.endsWith('-1.png'))
+})
+
+test('buildImagePath uses correct extension for jpeg', (t) => {
+  const result = buildImagePath('/tmp', '2026-01-01T00:00:00.000Z', 1, 'image/jpeg')
+  t.true(result.endsWith('-1.jpg'))
+})
+
+test('buildImagePath uses correct extension for webp', (t) => {
+  const result = buildImagePath('/tmp', '2026-01-01T00:00:00.000Z', 1, 'image/webp')
+  t.true(result.endsWith('-1.webp'))
+})
+
+test('buildImagePath defaults to .png for unknown media type', (t) => {
+  const result = buildImagePath('/tmp', '2026-01-01T00:00:00.000Z', 1, 'image/bmp')
+  t.true(result.endsWith('-1.png'))
+})
+
+test('buildImagePath replaces colons in timestamp', (t) => {
+  const result = buildImagePath('/tmp', '2026-01-01T12:30:45.000Z', 1, 'image/png')
+  t.false(result.includes(':'))
+})

package/test/stream.mjs

@@ -101,3 +101,22 @@ test('buildJsonOutput excludes sources when null', (t) => {
   const output = buildJsonOutput(conversation, 'response', null)
   t.false('sources' in output)
 })
+
+test('buildJsonOutput includes images when provided', (t) => {
+  const conversation = createMockConversation()
+  const images = ['/tmp/askimo-1.png', '/tmp/askimo-2.png']
+  const output = buildJsonOutput(conversation, 'response', undefined, undefined, images)
+  t.deepEqual(output.images, images)
+})
+
+test('buildJsonOutput excludes images when empty array', (t) => {
+  const conversation = createMockConversation()
+  const output = buildJsonOutput(conversation, 'response', undefined, undefined, [])
+  t.false('images' in output)
+})
+
+test('buildJsonOutput excludes images when undefined', (t) => {
+  const conversation = createMockConversation()
+  const output = buildJsonOutput(conversation, 'response', undefined, undefined, undefined)
+  t.false('images' in output)
+})