@providerprotocol/ai 0.0.17 → 0.0.19

package/dist/xai/index.d.ts

@@ -1,4 +1,31 @@
-import { d as Provider, f as ModelReference, b as LLMHandler } from '../provider-D5MO3-pS.js';
+import { d as Provider, f as ModelReference, b as LLMHandler } from '../provider-BBMBZuGn.js';
+import { n as ImageHandler } from '../image-Dhq-Yuq4.js';
+import '../content-DEl3z_W2.js';
+
+/**
+ * @fileoverview xAI Image Generation API Handler
+ *
+ * This module implements the image handler for xAI's Image Generation API (Aurora).
+ * Supports the grok-2-image-1212 model.
+ *
+ * @see {@link https://docs.x.ai/docs/image-generation xAI Image Generation Reference}
+ * @module providers/xai/image
+ */
+
+/**
+ * xAI image generation parameters.
+ * Passed through unchanged to the API.
+ *
+ * Note: xAI does NOT support negative_prompt or seed parameters.
+ */
+interface XAIImageParams {
+    /** Number of images to generate (1-10) */
+    n?: number;
+    /** Response format */
+    response_format?: 'url' | 'b64_json';
+    /** User identifier */
+    user?: string;
+}
 
 /**
  * xAI Chat Completions API parameters (OpenAI-compatible).
@@ -660,6 +687,7 @@ interface XAIProvider extends Provider<XAIProviderOptions> {
     /** Supported modalities */
     readonly modalities: {
         llm: LLMHandler<XAILLMParamsUnion>;
+        image: ImageHandler<XAIImageParams>;
     };
 }
 /**

package/dist/xai/index.js

@@ -1,19 +1,24 @@
+import {
+  Image
+} from "../chunk-WAKD3OO5.js";
 import {
   AssistantMessage,
   isAssistantMessage,
   isToolResultMessage,
   isUserMessage
-} from "../chunk-SVYROCLD.js";
+} from "../chunk-UMKWXGO3.js";
 import {
   parseSSEStream
 } from "../chunk-Z7RBRCRN.js";
 import {
-  UPPError,
   doFetch,
   doStreamFetch,
   normalizeHttpError,
   resolveApiKey
-} from "../chunk-MOU4U3PO.js";
+} from "../chunk-5FEAOEXV.js";
+import {
+  UPPError
+} from "../chunk-DZQHVGNV.js";
 
 // src/providers/xai/transform.completions.ts
 function transformRequest(request, modelId) {
@@ -1726,6 +1731,112 @@ function createMessagesLLMHandler() {
   };
 }
 
+// src/providers/xai/image.ts
+var XAI_IMAGES_API_URL = "https://api.x.ai/v1/images/generations";
+function getCapabilities(modelId) {
+  return {
+    generate: true,
+    streaming: false,
+    edit: false,
+    maxImages: 10
+  };
+}
+function createImageHandler() {
+  let providerRef = null;
+  return {
+    _setProvider(provider) {
+      providerRef = provider;
+    },
+    bind(modelId) {
+      if (!providerRef) {
+        throw new UPPError(
+          "Provider reference not set. Handler must be used with createProvider().",
+          "INVALID_REQUEST",
+          "xai",
+          "image"
+        );
+      }
+      const capabilities = getCapabilities(modelId);
+      const model = {
+        modelId,
+        capabilities,
+        get provider() {
+          return providerRef;
+        },
+        async generate(request) {
+          return executeGenerate(modelId, request);
+        }
+      };
+      return model;
+    }
+  };
+}
+async function executeGenerate(modelId, request) {
+  const apiKey = await resolveApiKey(
+    request.config,
+    "XAI_API_KEY",
+    "xai",
+    "image"
+  );
+  const baseUrl = request.config.baseUrl ? `${request.config.baseUrl.replace(/\/$/, "")}/v1/images/generations` : XAI_IMAGES_API_URL;
+  const body = {
+    model: modelId,
+    prompt: request.prompt
+  };
+  if (request.params) {
+    const { n, response_format, user } = request.params;
+    if (n !== void 0) body.n = n;
+    if (response_format !== void 0) body.response_format = response_format;
+    if (user !== void 0) body.user = user;
+  }
+  const headers = {
+    "Content-Type": "application/json",
+    Authorization: `Bearer ${apiKey}`
+  };
+  if (request.config.headers) {
+    for (const [key, value] of Object.entries(request.config.headers)) {
+      if (value !== void 0) {
+        headers[key] = value;
+      }
+    }
+  }
+  const response = await doFetch(baseUrl, {
+    method: "POST",
+    headers,
+    body: JSON.stringify(body),
+    signal: request.signal
+  }, request.config, "xai", "image");
+  const data = await response.json();
+  return transformResponse4(data);
+}
+function transformResponse4(data) {
+  const images = data.data.map((item) => {
+    let image;
+    if (item.b64_json) {
+      image = Image.fromBase64(item.b64_json, "image/jpeg");
+    } else if (item.url) {
+      image = Image.fromUrl(item.url, "image/jpeg");
+    } else {
+      throw new UPPError(
+        "No image data in response",
+        "PROVIDER_ERROR",
+        "xai",
+        "image"
+      );
+    }
+    return {
+      image,
+      metadata: item.revised_prompt ? { revised_prompt: item.revised_prompt } : void 0
+    };
+  });
+  return {
+    images,
+    usage: {
+      imagesGenerated: images.length
+    }
+  };
+}
+
 // src/providers/xai/types.ts
 function webSearchTool(options) {
   return {
@@ -1778,6 +1889,7 @@ function createXAIProvider() {
   const completionsHandler = createCompletionsLLMHandler();
   const responsesHandler = createResponsesLLMHandler();
   const messagesHandler = createMessagesLLMHandler();
+  const imageHandler = createImageHandler();
   const fn = function(modelId, options) {
     const apiMode = options?.api ?? "completions";
     currentApiMode = apiMode;
@@ -1794,7 +1906,8 @@ function createXAIProvider() {
         default:
           return completionsHandler;
       }
-    }
+    },
+    image: imageHandler
   };
   Object.defineProperties(fn, {
     name: {
@@ -1817,6 +1930,7 @@ function createXAIProvider() {
   completionsHandler._setProvider?.(provider);
   responsesHandler._setProvider?.(provider);
   messagesHandler._setProvider?.(provider);
+  imageHandler._setProvider?.(provider);
   return provider;
 }
 var xai = createXAIProvider();