@ai-agent-tools/picgen 0.1.0-alpha.2 → 0.1.0-alpha.3

package/README.md

@@ -101,6 +101,12 @@ Provider `base_url` values should be host-only. Do not include `/v1` or `/v1beta
 
 Providers may optionally set `test_model` in `~/.picgen/config.yaml` when health checks should use a lightweight model instead of the first generation model.
 
+Generation requests use adaptive provider timeouts: fast draft requests allow 120s, balanced requests allow 180s, and high quality or large requests allow 300s. If a third-party channel is slower, override it with `PICGEN_PROVIDER_TIMEOUT_MS`:
+
+```bash
+PICGEN_PROVIDER_TIMEOUT_MS=450000 picgen create --yes --preset poster "一张产品发布会主视觉"
+```
+
 Providers expose capabilities such as `text-to-image` and `reference-image`. Old configs that omit capabilities are upgraded in memory from the provider protocol: Gemini supports both text and reference-image generation, while OpenAI-compatible `/v1/images/generations` supports text-to-image only.
 
 Generated image data and provider-only fields such as base64 image payloads and Gemini thought signatures are redacted from metadata. PicGen keeps the generated assets as local image files and keeps stdout compact for agent workflows.

package/dist/cli.js

@@ -459,6 +459,53 @@ async function ensureConfig() {
 // src/providers/gemini.ts
 import { readFile as readFile2 } from "fs/promises";
 
+// src/providers/timeout.ts
+var FAST_PROVIDER_TIMEOUT_MS = 12e4;
+var DEFAULT_PROVIDER_TIMEOUT_MS = 18e4;
+var SLOW_PROVIDER_TIMEOUT_MS = 3e5;
+function resolveProviderTimeoutMs(plan) {
+  const override = parseTimeoutOverride(process.env.PICGEN_PROVIDER_TIMEOUT_MS);
+  if (override !== void 0) return override;
+  if (plan.presetName === "fast-draft" || plan.modeName === "fast" || plan.preset.quality === "low") {
+    return FAST_PROVIDER_TIMEOUT_MS;
+  }
+  if (plan.modeName === "premium" || plan.preset.size === "large" || plan.preset.quality === "high") {
+    return SLOW_PROVIDER_TIMEOUT_MS;
+  }
+  return DEFAULT_PROVIDER_TIMEOUT_MS;
+}
+async function fetchWithProviderTimeout(input3, init, timeoutMs) {
+  const controller = new AbortController();
+  const timeout = setTimeout(() => controller.abort(), timeoutMs);
+  try {
+    return await fetch(input3, {
+      ...init,
+      signal: controller.signal
+    });
+  } catch (error) {
+    if (isAbortError(error)) {
+      throw new Error(formatProviderTimeoutError(timeoutMs));
+    }
+    throw error;
+  } finally {
+    clearTimeout(timeout);
+  }
+}
+function formatProviderTimeoutError(timeoutMs) {
+  return `Provider request timed out after ${Math.ceil(
+    timeoutMs / 1e3
+  )}s. The provider may still be processing or temporarily unavailable. Try again, use a faster preset, or increase PICGEN_PROVIDER_TIMEOUT_MS.`;
+}
+function parseTimeoutOverride(value) {
+  if (!value) return void 0;
+  const timeoutMs = Number(value);
+  if (!Number.isInteger(timeoutMs) || timeoutMs <= 0) return void 0;
+  return timeoutMs;
+}
+function isAbortError(error) {
+  return error instanceof Error && error.name === "AbortError";
+}
+
 // src/providers/urls.ts
 function normalizeProviderBaseUrl(baseUrl) {
   const url = new URL(baseUrl);
@@ -495,15 +542,20 @@ var GeminiAdapter = class {
     const providerImages = [];
     const requestCount = Math.max(1, plan.preset.n);
     const referenceParts = await readReferenceImageParts(plan);
+    const timeoutMs = resolveProviderTimeoutMs(plan);
     for (let index = 0; index < requestCount; index += 1) {
-      const response = await fetch(buildGeminiGenerateContentUrl(plan.provider.base_url, plan.model), {
-        method: "POST",
-        headers: {
-          "x-goog-api-key": apiKey,
-          "Content-Type": "application/json"
+      const response = await fetchWithProviderTimeout(
+        buildGeminiGenerateContentUrl(plan.provider.base_url, plan.model),
+        {
+          method: "POST",
+          headers: {
+            "x-goog-api-key": apiKey,
+            "Content-Type": "application/json"
+          },
+          body: JSON.stringify(buildGeminiGenerateContentRequest(plan, referenceParts))
         },
-        body: JSON.stringify(buildGeminiGenerateContentRequest(plan, referenceParts))
-      });
+        timeoutMs
+      );
       const raw = await readJsonResponse(response);
       if (!response.ok) {
         throw new Error(formatGeminiError(response.status, response.statusText, raw));
@@ -624,14 +676,18 @@ var OpenAIImagesAdapter = class {
     if (!apiKey) {
       throw new Error(`Missing API key environment variable: ${plan.provider.api_key_env}`);
     }
-    const response = await fetch(buildOpenAIImagesUrl(plan.provider.base_url), {
-      method: "POST",
-      headers: {
-        Authorization: `Bearer ${apiKey}`,
-        "Content-Type": "application/json"
+    const response = await fetchWithProviderTimeout(
+      buildOpenAIImagesUrl(plan.provider.base_url),
+      {
+        method: "POST",
+        headers: {
+          Authorization: `Bearer ${apiKey}`,
+          "Content-Type": "application/json"
+        },
+        body: JSON.stringify(buildOpenAIImagesRequest(plan))
       },
-      body: JSON.stringify(buildOpenAIImagesRequest(plan))
-    });
+      resolveProviderTimeoutMs(plan)
+    );
     const raw = await readJsonResponse2(response);
     if (!response.ok) {
       throw new Error(formatOpenAIImagesError(response.status, response.statusText, raw));
@@ -999,7 +1055,7 @@ import { homedir as homedir2 } from "os";
 
 // src/version.ts
 var PACKAGE_NAME = "@ai-agent-tools/picgen";
-var VERSION = "0.1.0-alpha.2";
+var VERSION = "0.1.0-alpha.3";
 
 // src/commands/update.ts
 var UPDATE_CACHE_TTL_MS = 24 * 60 * 60 * 1e3;

package/docs/release-alpha.md

@@ -143,6 +143,14 @@ picgen provider test <provider-name> --json
 
 Check `base_url`, API key, model name, and provider availability.
 
+`Provider request timed out`
+
+High quality, large, or slow third-party image channels can take longer. Try again, use a faster preset, or raise the request timeout:
+
+```bash
+PICGEN_PROVIDER_TIMEOUT_MS=450000 picgen create --yes --preset poster "<prompt>"
+```
+
 `No enabled provider can satisfy...`
 
 Run `picgen provider list`, enable a provider, add a fallback provider, or adjust the selected mode/model.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-agent-tools/picgen",
-  "version": "0.1.0-alpha.2",
+  "version": "0.1.0-alpha.3",
   "description": "A lightweight image generation connector for AI agents.",
   "type": "module",
   "bin": {