@umituz/web-ai 1.0.0 → 1.0.1

package/dist/providers/index.cjs

@@ -8,25 +8,49 @@ function isMediaProvider(provider) {
   return provider.capabilities.includes("image") || provider.capabilities.includes("video");
 }
 
-// src/content/domain/errors/AIErrors.ts
-var AIError = class _AIError extends Error {
-  code;
-  cause;
-  constructor(message, code, cause) {
-    super(message);
-    this.name = "AIError";
-    this.code = code;
-    this.cause = cause;
-  }
-  static from(error, fallbackCode = "UNKNOWN") {
-    if (error instanceof _AIError) return error;
-    if (error instanceof Error) {
-      return new _AIError(error.message, mapMessageToCode(error.message, fallbackCode), error);
-    }
-    return new _AIError(String(error), fallbackCode, error);
+// src/core/errors/ai-error-types.ts
+var AIErrorType = /* @__PURE__ */ ((AIErrorType2) => {
+  AIErrorType2["INVALID_API_KEY"] = "INVALID_API_KEY";
+  AIErrorType2["MISSING_CONFIG"] = "MISSING_CONFIG";
+  AIErrorType2["NETWORK_ERROR"] = "NETWORK_ERROR";
+  AIErrorType2["ABORT_ERROR"] = "ABORT_ERROR";
+  AIErrorType2["RATE_LIMIT_ERROR"] = "RATE_LIMIT_ERROR";
+  AIErrorType2["SERVER_ERROR"] = "SERVER_ERROR";
+  AIErrorType2["CLIENT_ERROR"] = "CLIENT_ERROR";
+  AIErrorType2["UNKNOWN_ERROR"] = "UNKNOWN_ERROR";
+  return AIErrorType2;
+})(AIErrorType || {});
+
+// src/core/errors/ai-error.ts
+var KNOWN_ERROR_TYPES = new Set(Object.values(AIErrorType));
+function mapCodeToType(code) {
+  switch (code) {
+    case "VALIDATION_ERROR":
+    case "CLIENT_ERROR":
+      return "CLIENT_ERROR" /* CLIENT_ERROR */;
+    case "INVALID_API_KEY":
+      return "INVALID_API_KEY" /* INVALID_API_KEY */;
+    case "MISSING_CONFIG":
+    case "PROVIDER_UNAVAILABLE":
+      return "MISSING_CONFIG" /* MISSING_CONFIG */;
+    case "PROVIDER_QUOTA_EXHAUSTED":
+    case "RATE_LIMIT_ERROR":
+      return "RATE_LIMIT_ERROR" /* RATE_LIMIT_ERROR */;
+    case "NETWORK_ERROR":
+    case "TIMEOUT":
+      return "NETWORK_ERROR" /* NETWORK_ERROR */;
+    case "ABORT_ERROR":
+    case "ABORTED":
+      return "ABORT_ERROR" /* ABORT_ERROR */;
+    case "SERVER_ERROR":
+    case "PROVIDER_ERROR":
+    case "NO_RESULT":
+      return "SERVER_ERROR" /* SERVER_ERROR */;
+    default:
+      return "UNKNOWN_ERROR" /* UNKNOWN_ERROR */;
   }
-};
-function mapMessageToCode(message, fallback) {
+}
+function mapMessageToCode(message, fallback = "UNKNOWN") {
   const lower = message.toLowerCase();
   if (lower.includes("validation")) return "VALIDATION_ERROR";
   if (lower.includes("quota")) return "PROVIDER_QUOTA_EXHAUSTED";
@@ -39,6 +63,53 @@ function mapMessageToCode(message, fallback) {
   if (lower.includes("network") || lower.includes("fetch")) return "NETWORK_ERROR";
   return fallback;
 }
+var AIError = class _AIError extends Error {
+  type;
+  code;
+  cause;
+  metadata;
+  constructor(typeOrMessage, messageOrCode, cause, metadata = {}) {
+    let effectiveType;
+    let effectiveMessage;
+    let effectiveCode;
+    if (KNOWN_ERROR_TYPES.has(typeOrMessage)) {
+      effectiveType = typeOrMessage;
+      effectiveMessage = messageOrCode ?? "";
+      effectiveCode = effectiveType;
+    } else {
+      effectiveMessage = typeOrMessage;
+      effectiveCode = messageOrCode ?? "UNKNOWN";
+      effectiveType = mapCodeToType(effectiveCode);
+    }
+    super(effectiveMessage);
+    this.name = "AIError";
+    this.type = effectiveType;
+    this.code = effectiveCode;
+    this.cause = cause;
+    this.metadata = metadata;
+  }
+  static from(error, fallbackCode = "UNKNOWN") {
+    if (error instanceof _AIError) return error;
+    if (error instanceof Error) {
+      const code = mapMessageToCode(error.message, fallbackCode);
+      return new _AIError(error.message, code, error);
+    }
+    return new _AIError(String(error), fallbackCode, error);
+  }
+  toString() {
+    const statusPart = typeof this.metadata.status === "number" ? ` (HTTP ${this.metadata.status})` : "";
+    return `${this.name} [${this.type}]: ${this.message}${statusPart}`;
+  }
+  toJSON() {
+    return {
+      name: this.name,
+      type: this.type,
+      code: this.code,
+      message: this.message,
+      metadata: this.metadata
+    };
+  }
+};
 
 // src/content/domain/limits/ProviderTimingConfig.ts
 var ProviderTimingConfig = {
@@ -113,32 +184,6 @@ var ProviderError = class extends AIError {
   providerId;
 };
 
-// src/core/errors/ai-error.ts
-var AIError2 = class extends Error {
-  type;
-  cause;
-  metadata;
-  constructor(type, message, cause, metadata = {}) {
-    super(message);
-    this.name = "AIError";
-    this.type = type;
-    this.cause = cause;
-    this.metadata = metadata;
-  }
-  toString() {
-    const statusPart = typeof this.metadata.status === "number" ? ` (HTTP ${this.metadata.status})` : "";
-    return `${this.name} [${this.type}]: ${this.message}${statusPart}`;
-  }
-  toJSON() {
-    return {
-      name: this.name,
-      type: this.type,
-      message: this.message,
-      metadata: this.metadata
-    };
-  }
-};
-
 // src/core/errors/error-classifier.ts
 var DEFAULT_RETRYABLE_TYPES = /* @__PURE__ */ new Set([
   "NETWORK_ERROR" /* NETWORK_ERROR */,
@@ -264,7 +309,7 @@ var ProviderFactory = class {
         }
       }
     }
-    const failure = new AIError2(
+    const failure = new AIError(
       "SERVER_ERROR" /* SERVER_ERROR */,
       `All providers failed for capability 'text'`,
       lastError instanceof Error ? lastError : void 0
@@ -278,12 +323,16 @@ var ProviderFactory = class {
     let wake = null;
     let finished = false;
     let failure;
+    const controller = new AbortController();
+    const onAbort = () => controller.abort();
+    request.signal?.addEventListener("abort", onAbort);
+    const effectiveRequest = { ...request, signal: controller.signal };
     const notify = () => {
       const resolve = wake;
       wake = null;
       resolve?.();
     };
-    const completion = this.chatStream(request, {
+    const completion = this.chatStream(effectiveRequest, {
       onChunk: (chunk) => {
         queue.push(chunk);
         notify();
@@ -299,18 +348,23 @@ var ProviderFactory = class {
         notify();
       }
     );
-    while (!finished || queue.length > 0) {
-      if (queue.length === 0 && !finished) {
-        await new Promise((resolve) => {
-          wake = resolve;
-        });
-      }
-      while (queue.length > 0) {
-        yield queue.shift();
+    try {
+      while (!finished || queue.length > 0) {
+        if (queue.length === 0 && !finished) {
+          await new Promise((resolve) => {
+            wake = resolve;
+          });
+        }
+        while (queue.length > 0) {
+          yield queue.shift();
+        }
+        if (failure) throw failure;
       }
-      if (failure) throw failure;
+      await completion;
+    } finally {
+      controller.abort();
+      request.signal?.removeEventListener("abort", onAbort);
     }
-    await completion;
   }
   async generateJson(request) {
     const effective = request.model || request.tier ? request : { ...request, tier: this.defaultTier };
@@ -418,7 +472,7 @@ var ProviderFactory = class {
         }
       }
     }
-    throw new AIError2(
+    throw new AIError(
       "SERVER_ERROR" /* SERVER_ERROR */,
       `All providers failed for capability '${capability}'`,
       lastError instanceof Error ? lastError : void 0
@@ -447,7 +501,7 @@ var ProviderFactory = class {
     }
   }
   noProviderError(capability) {
-    return new AIError2(
+    return new AIError(
       "MISSING_CONFIG" /* MISSING_CONFIG */,
       `No provider registered for capability '${capability}' \u2014 create one with createAI() and include its config block`
     );
@@ -457,15 +511,22 @@ var ProviderFactory = class {
 // src/core/constants/protocol.constants.ts
 var SSE_PREFIX = "data: ";
 var SSE_DONE_MARKER = "[DONE]";
-var JSON_CODE_BLOCK_PATTERN = /```json\n([\s\S]*?)\n```/;
+var JSON_CODE_BLOCK_PATTERN = /```(?:json)?\s*\n?([\s\S]*?)\n?```/;
 var JSON_OBJECT_PATTERN = /\{[\s\S]*\}/;
+var JSON_ARRAY_PATTERN = /\[[\s\S]*\]/;
 
 // src/core/parsers/sse-lines.ts
 function isSseDataLine(line) {
-  return line.startsWith(SSE_PREFIX);
+  return line.startsWith(SSE_PREFIX) || line.startsWith("data:");
 }
 function extractSseDataPayload(line) {
-  return line.slice(SSE_PREFIX.length);
+  if (line.startsWith(SSE_PREFIX)) {
+    return line.slice(SSE_PREFIX.length);
+  }
+  if (line.startsWith("data:")) {
+    return line.slice(5);
+  }
+  return line;
 }
 function isSseDoneMarker(payload) {
   return payload === SSE_DONE_MARKER;
@@ -474,7 +535,7 @@ function isSseDoneMarker(payload) {
 // src/core/parsers/json-content-extractor.ts
 function extractJsonFromCodeFence(content) {
   const match = content.match(JSON_CODE_BLOCK_PATTERN);
-  if (match && match[1]) return match[1];
+  if (match && match[1]) return match[1].trim();
   return null;
 }
 function extractJsonObject(content) {
@@ -482,8 +543,27 @@ function extractJsonObject(content) {
   if (match && match[0]) return match[0];
   return null;
 }
+function extractJsonArray(content) {
+  const match = content.match(JSON_ARRAY_PATTERN);
+  if (match && match[0]) return match[0];
+  return null;
+}
+function repairCommonJsonDefects(text) {
+  return text.replace(/(:\s*-?\d+(?:\.\d+)?)\s*"/g, "$1").replace(/(:\s*"[^"]*")"/g, "$1");
+}
 function extractJsonCandidate(content) {
-  return extractJsonFromCodeFence(content) ?? extractJsonObject(content) ?? content;
+  const fromFence = extractJsonFromCodeFence(content);
+  if (fromFence) return fromFence;
+  const objectIndex = content.indexOf("{");
+  const arrayIndex = content.indexOf("[");
+  if (objectIndex !== -1 && (arrayIndex === -1 || objectIndex < arrayIndex)) {
+    const obj = extractJsonObject(content);
+    if (obj) return obj;
+  } else if (arrayIndex !== -1) {
+    const arr = extractJsonArray(content);
+    if (arr) return arr;
+  }
+  return content.trim();
 }
 
 // src/core/parsers/json-response.parser.ts
@@ -492,12 +572,17 @@ var JsonResponseParser = class {
     const candidate = extractJsonCandidate(rawContent);
     try {
       return JSON.parse(candidate);
-    } catch (error) {
-      throw new AIError2(
-        "UNKNOWN_ERROR" /* UNKNOWN_ERROR */,
-        "Failed to parse structured response as JSON",
-        error instanceof Error ? error : void 0
-      );
+    } catch {
+      try {
+        return JSON.parse(repairCommonJsonDefects(candidate));
+      } catch (error) {
+        throw new AIError(
+          "UNKNOWN_ERROR" /* UNKNOWN_ERROR */,
+          "Failed to parse structured response as JSON",
+          error instanceof Error ? error : void 0,
+          { rawContent }
+        );
+      }
     }
   }
 };
@@ -623,7 +708,7 @@ var AnthropicHttpTransport = class {
       });
       if (!response.ok) await this.throwFromErrorResponse(response);
       if (!response.body) {
-        throw new AIError2("NETWORK_ERROR" /* NETWORK_ERROR */, "Response body is null");
+        throw new AIError("NETWORK_ERROR" /* NETWORK_ERROR */, "Response body is null");
       }
       return response.body;
     } catch (error) {
@@ -656,7 +741,7 @@ var AnthropicHttpTransport = class {
       status: response.status,
       url: response.url
     };
-    throw new AIError2(
+    throw new AIError(
       classifyHttpStatus(response.status),
       errorMessage,
       void 0,
@@ -664,24 +749,24 @@ var AnthropicHttpTransport = class {
     );
   }
   normalizeError(error) {
-    if (error instanceof AIError2) return error;
+    if (error instanceof AIError) return error;
     if (error instanceof Error) {
       if (error.name === "AbortError") {
-        return new AIError2(
+        return new AIError(
           "ABORT_ERROR" /* ABORT_ERROR */,
           "Request was aborted by the client",
           error
         );
       }
       if (error.name === "TypeError" && error.message.toLowerCase().includes("network")) {
-        return new AIError2(
+        return new AIError(
           "NETWORK_ERROR" /* NETWORK_ERROR */,
           "Network error: Unable to connect to the AI service",
           error
         );
       }
     }
-    return new AIError2(
+    return new AIError(
       "UNKNOWN_ERROR" /* UNKNOWN_ERROR */,
       error instanceof Error ? error.message : "Unknown error occurred",
       error instanceof Error ? error : void 0
@@ -811,11 +896,14 @@ async function consumeAnthropicStream(stream, callbacks) {
         break;
       case "message_stop":
         break;
-      case "error":
-        throw new AIError2(
-          event.error.type === "rate_limit_error" ? "RATE_LIMIT_ERROR" /* RATE_LIMIT_ERROR */ : "SERVER_ERROR" /* SERVER_ERROR */,
+      case "error": {
+        const errType = event.error.type;
+        const mappedType = errType === "rate_limit_error" ? "RATE_LIMIT_ERROR" /* RATE_LIMIT_ERROR */ : errType === "authentication_error" ? "INVALID_API_KEY" /* INVALID_API_KEY */ : errType === "invalid_request_error" ? "CLIENT_ERROR" /* CLIENT_ERROR */ : "SERVER_ERROR" /* SERVER_ERROR */;
+        throw new AIError(
+          mappedType,
           event.error.message || "Anthropic stream failed mid-flight"
         );
+      }
     }
   };
   try {
@@ -830,8 +918,8 @@ async function consumeAnthropicStream(stream, callbacks) {
     buffer += decoder.decode();
     if (buffer.length > 0) consumeLine(buffer);
   } catch (error) {
-    if (error instanceof AIError2) throw error;
-    throw new AIError2(
+    if (error instanceof AIError) throw error;
+    throw new AIError(
       "NETWORK_ERROR" /* NETWORK_ERROR */,
       "Stream was interrupted",
       error instanceof Error ? error : void 0
@@ -893,7 +981,7 @@ var AnthropicProvider = class {
     const messages = [...request.messages];
     const lastUserIndex = findLastIndex(messages, (m) => m.role === "user");
     if (lastUserIndex === -1) {
-      throw new AIError2(
+      throw new AIError(
         "UNKNOWN_ERROR" /* UNKNOWN_ERROR */,
         "generateStructured requires at least one user message"
       );
@@ -948,7 +1036,7 @@ var AnthropicProvider = class {
           finishReason
         };
       }
-      throw new AIError2(
+      throw new AIError(
         "SERVER_ERROR" /* SERVER_ERROR */,
         "No content generated from Anthropic API",
         void 0,
@@ -1153,7 +1241,7 @@ var RetryManager = class {
   }
   shouldRetry(error, options) {
     if (options.shouldRetry) return options.shouldRetry(error);
-    if (error instanceof AIError2) {
+    if (error instanceof AIError) {
       return isRetryableErrorType(error.type, DEFAULT_ERROR_CLASSIFICATION_RULE);
     }
     const lower = error.message.toLowerCase();
@@ -1398,7 +1486,7 @@ var GroqHttpTransport = class {
       });
       if (!response.ok) await this.throwFromErrorResponse(response);
       if (!response.body) {
-        throw new AIError2("NETWORK_ERROR" /* NETWORK_ERROR */, "Response body is null");
+        throw new AIError("NETWORK_ERROR" /* NETWORK_ERROR */, "Response body is null");
       }
       return response.body;
     } catch (error) {
@@ -1446,7 +1534,7 @@ var GroqHttpTransport = class {
       status: response.status,
       url: response.url
     };
-    throw new AIError2(
+    throw new AIError(
       classifyHttpStatus(response.status),
       errorMessage,
       void 0,
@@ -1454,24 +1542,24 @@ var GroqHttpTransport = class {
     );
   }
   normalizeError(error) {
-    if (error instanceof AIError2) return error;
+    if (error instanceof AIError) return error;
     if (error instanceof Error) {
       if (error.name === "AbortError") {
-        return new AIError2(
+        return new AIError(
           "ABORT_ERROR" /* ABORT_ERROR */,
           "Request was aborted by the client",
           error
         );
       }
       if (error.name === "TypeError" && error.message.toLowerCase().includes("network")) {
-        return new AIError2(
+        return new AIError(
           "NETWORK_ERROR" /* NETWORK_ERROR */,
           "Network error: Unable to connect to the AI service",
           error
         );
       }
     }
-    return new AIError2(
+    return new AIError(
       "UNKNOWN_ERROR" /* UNKNOWN_ERROR */,
       error instanceof Error ? error.message : "Unknown error occurred",
       error instanceof Error ? error : void 0
@@ -1622,6 +1710,13 @@ async function consumeGroqStream(stream, callbacks) {
     } catch {
       return;
     }
+    const errorBody = chunk.error;
+    if (errorBody) {
+      throw new AIError(
+        errorBody.message?.toLowerCase().includes("rate limit") ? "RATE_LIMIT_ERROR" /* RATE_LIMIT_ERROR */ : "SERVER_ERROR" /* SERVER_ERROR */,
+        errorBody.message || "Groq stream failed mid-flight"
+      );
+    }
     if (chunk.model) model = chunk.model;
     const choice = chunk.choices?.[0];
     const delta = choice?.delta?.content;
@@ -1636,13 +1731,6 @@ async function consumeGroqStream(stream, callbacks) {
     if (tailUsage) {
       usage = mapGroqUsage(tailUsage);
     }
-    const errorBody = chunk.error;
-    if (errorBody) {
-      throw new AIError2(
-        "SERVER_ERROR" /* SERVER_ERROR */,
-        errorBody.message || "Groq stream failed mid-flight"
-      );
-    }
   };
   try {
     while (true) {
@@ -1656,8 +1744,8 @@ async function consumeGroqStream(stream, callbacks) {
     buffer += decoder.decode();
     if (buffer.length > 0) consumeLine(buffer);
   } catch (error) {
-    if (error instanceof AIError2) throw error;
-    throw new AIError2(
+    if (error instanceof AIError) throw error;
+    throw new AIError(
       "NETWORK_ERROR" /* NETWORK_ERROR */,
       "Stream was interrupted",
       error instanceof Error ? error : void 0
@@ -1715,7 +1803,7 @@ var GroqProvider = class {
     const choice = response.choices?.[0];
     const text = choice?.message?.content ?? "";
     if (text.length === 0) {
-      throw new AIError2(
+      throw new AIError(
         "SERVER_ERROR" /* SERVER_ERROR */,
         "No content generated from Groq API",
         void 0,
@@ -1737,7 +1825,7 @@ var GroqProvider = class {
     const messages = toGroqMessages(request.messages, request.systemPrompt);
     const lastUserIndex = findLastIndex2(messages, (m) => m.role === "user");
     if (lastUserIndex === -1) {
-      throw new AIError2(
+      throw new AIError(
         "UNKNOWN_ERROR" /* UNKNOWN_ERROR */,
         "generateStructured requires at least one user message"
       );
@@ -1751,7 +1839,7 @@ var GroqProvider = class {
     const choice = response.choices?.[0];
     const text = choice?.message?.content ?? "";
     if (text.length === 0) {
-      throw new AIError2(
+      throw new AIError(
         "SERVER_ERROR" /* SERVER_ERROR */,
         "No content generated from Groq API",
         void 0,
@@ -1778,7 +1866,7 @@ var GroqProvider = class {
     try {
       const outcome = await consumeGroqStream(streamBody, callbacks);
       if (outcome.text.length === 0) {
-        throw new AIError2(
+        throw new AIError(
           "SERVER_ERROR" /* SERVER_ERROR */,
           "No content generated from Groq API",
           void 0,
@@ -1909,7 +1997,9 @@ var GeminiProvider = class extends BaseProvider {
           headers: this.authHeaders()
         }
       );
-      return response.ok ? "healthy" : "unavailable";
+      if (response.ok) return "healthy";
+      if (response.status === 401 || response.status === 403) return "unavailable";
+      return "degraded";
     } catch {
       return "unavailable";
     }
@@ -1920,6 +2010,8 @@ var GeminiProvider = class extends BaseProvider {
   async generate(request) {
     try {
       const model = request.model ?? this.resolveModel(request.tier ?? "balanced");
+      const systemMessages = request.messages.filter((m) => m.role === "system");
+      const systemPrompt = request.systemPrompt && request.systemPrompt.length > 0 ? request.systemPrompt : systemMessages.length > 0 ? systemMessages.map((m) => m.content).join("\n") : void 0;
       const response = await this.withRetry(async () => {
         return await this.fetchWithTimeout(
           `${this.config.baseUrl}/models/${model}:generateContent`,
@@ -1927,24 +2019,30 @@ var GeminiProvider = class extends BaseProvider {
             method: "POST",
             headers: this.authHeaders(),
             body: JSON.stringify({
-              systemInstruction: request.systemPrompt && request.systemPrompt.length > 0 ? { parts: [{ text: request.systemPrompt }] } : void 0,
+              systemInstruction: systemPrompt && systemPrompt.length > 0 ? { parts: [{ text: systemPrompt }] } : void 0,
               contents: toGeminiContents(request.messages),
               generationConfig: {
-                temperature: request.temperature || 0.7,
-                maxOutputTokens: request.maxTokens || 2048,
-                topP: request.topP || 0.9
+                temperature: request.temperature ?? 0.7,
+                maxOutputTokens: request.maxTokens ?? 2048,
+                topP: request.topP ?? 0.9
               }
             })
           }
         );
       });
       if (!response.ok) {
-        throw new Error(`Gemini API error: ${response.statusText}`);
+        const message = await extractErrorMessageFromResponse(response);
+        throw new AIError(
+          classifyHttpStatus(response.status),
+          `Gemini API error (${response.status}): ${message}`,
+          void 0,
+          { status: response.status, model }
+        );
       }
       const data = await response.json();
-      const text = data.candidates[0]?.content?.parts?.map((part) => part.text ?? "").join("") ?? "";
+      const text = data.candidates?.[0]?.content?.parts?.map((part) => part.text ?? "").join("") ?? "";
       if (text.length === 0) {
-        throw new AIError2("SERVER_ERROR" /* SERVER_ERROR */, "No content generated from Gemini API", void 0, { model });
+        throw new AIError("SERVER_ERROR" /* SERVER_ERROR */, "No content generated from Gemini API", void 0, { model });
       }
       return {
         text,
@@ -1955,11 +2053,11 @@ var GeminiProvider = class extends BaseProvider {
         } : void 0,
         model,
         providerId: this.id,
-        finishReason: mapGeminiFinishReason(data.candidates[0]?.finishReason)
+        finishReason: mapGeminiFinishReason(data.candidates?.[0]?.finishReason)
       };
     } catch (error) {
-      if (error instanceof AIError2) throw error;
-      throw new AIError2(
+      if (error instanceof AIError) throw error;
+      throw new AIError(
         "UNKNOWN_ERROR" /* UNKNOWN_ERROR */,
         `Gemini text generation failed: ${error instanceof Error ? error.message : "Unknown error"}`,
         error instanceof Error ? error : void 0
@@ -1970,7 +2068,7 @@ var GeminiProvider = class extends BaseProvider {
     const messages = [...request.messages];
     const lastUserIndex = findLastIndex3(messages, (m) => m.role === "user");
     if (lastUserIndex === -1) {
-      throw new AIError2(
+      throw new AIError(
         "UNKNOWN_ERROR" /* UNKNOWN_ERROR */,
         "generateStructured requires at least one user message"
       );
@@ -2007,12 +2105,18 @@ var GeminiProvider = class extends BaseProvider {
         );
       });
       if (!response.ok) {
-        throw new Error(`Gemini API error: ${response.statusText}`);
+        const message = await extractErrorMessageFromResponse(response);
+        throw new AIError(
+          classifyHttpStatus(response.status),
+          `Gemini image generation error (${response.status}): ${message}`,
+          void 0,
+          { status: response.status }
+        );
       }
       const data = await response.json();
-      const imageData = data.candidates[0]?.content?.parts[0]?.inlineData;
+      const imageData = data.candidates?.[0]?.content?.parts?.[0]?.inlineData;
       if (!imageData) {
-        throw new Error("No image data in response");
+        throw new AIError("SERVER_ERROR" /* SERVER_ERROR */, "No image data in Gemini response");
       }
       const url = `data:${imageData.mimeType};base64,${imageData.data}`;
       return {
@@ -2331,17 +2435,70 @@ var PrunaProvider = class {
   async estimateCost(_request) {
     return 1;
   }
-  async generateImage(_request) {
-    throw new AIError2("UNKNOWN_ERROR" /* UNKNOWN_ERROR */, "Use generateWithPruna instead");
+  async generateImage(request) {
+    let finalUrl;
+    const model = this.config.models?.textToImage;
+    for await (const stage of generateWithPruna(
+      { prompt: request.prompt, model, quality: request.quality },
+      this.config.apiKey
+    )) {
+      if (stage.error) throw stage.error;
+      if (stage.result?.url) finalUrl = stage.result.url;
+    }
+    if (!finalUrl) {
+      throw new AIError("SERVER_ERROR" /* SERVER_ERROR */, "No image generated by Pruna AI");
+    }
+    return {
+      id: Date.now().toString(),
+      type: "image",
+      url: finalUrl,
+      metadata: { provider: this.id, model: model ?? "p-image" },
+      createdAt: (/* @__PURE__ */ new Date()).toISOString()
+    };
   }
-  async generateVideo(_request) {
-    throw new AIError2("UNKNOWN_ERROR" /* UNKNOWN_ERROR */, "Use generateWithPruna instead");
+  async generateVideo(request) {
+    let finalUrl;
+    const model = this.config.models?.textToVideo;
+    for await (const stage of generateWithPruna(
+      { prompt: request.prompt, model, duration: request.duration, aspectRatio: request.aspectRatio },
+      this.config.apiKey
+    )) {
+      if (stage.error) throw stage.error;
+      if (stage.result?.url) finalUrl = stage.result.url;
+    }
+    if (!finalUrl) {
+      throw new AIError("SERVER_ERROR" /* SERVER_ERROR */, "No video generated by Pruna AI");
+    }
+    return {
+      id: Date.now().toString(),
+      type: "video",
+      url: finalUrl,
+      metadata: { provider: this.id, model: model ?? "p-video" },
+      createdAt: (/* @__PURE__ */ new Date()).toISOString()
+    };
   }
-  async imageToVideo(_request) {
-    throw new AIError2("UNKNOWN_ERROR" /* UNKNOWN_ERROR */, "Use generateWithPruna instead");
+  async imageToVideo(request) {
+    let finalUrl;
+    for await (const stage of generateWithPruna(
+      { image: request.imageUrl, prompt: request.prompt, duration: request.duration, motion: request.motion },
+      this.config.apiKey
+    )) {
+      if (stage.error) throw stage.error;
+      if (stage.result?.url) finalUrl = stage.result.url;
+    }
+    if (!finalUrl) {
+      throw new AIError("SERVER_ERROR" /* SERVER_ERROR */, "No video generated from image by Pruna AI");
+    }
+    return {
+      id: Date.now().toString(),
+      type: "video",
+      url: finalUrl,
+      metadata: { provider: this.id, model: "p-image-to-video" },
+      createdAt: (/* @__PURE__ */ new Date()).toISOString()
+    };
   }
   async videoToVideo(_request) {
-    throw new AIError2("UNKNOWN_ERROR" /* UNKNOWN_ERROR */, "Use generateWithPruna instead");
+    throw new AIError("CLIENT_ERROR" /* CLIENT_ERROR */, "Pruna AI provider does not support video-to-video conversion");
   }
 };
 function createPrunaProvider(config) {