@providerprotocol/ai 0.0.18 → 0.0.20

package/dist/{stream-BjyVzBxV.d.ts → stream-DRHy6q1a.d.ts}

@@ -1,277 +1,4 @@
-/**
- * @fileoverview Content block types for multimodal messages.
- *
- * Defines the various content block types that can be included in
- * user and assistant messages, supporting text, images, audio, video,
- * and arbitrary binary data.
- *
- * @module types/content
- */
-/**
- * Image source variants for ImageBlock.
- *
- * Images can be provided as base64-encoded strings, URLs, or raw bytes.
- *
- * @example
- * ```typescript
- * // Base64 encoded image
- * const base64Source: ImageSource = {
- *   type: 'base64',
- *   data: 'iVBORw0KGgo...'
- * };
- *
- * // URL reference
- * const urlSource: ImageSource = {
- *   type: 'url',
- *   url: 'https://example.com/image.png'
- * };
- *
- * // Raw bytes
- * const bytesSource: ImageSource = {
- *   type: 'bytes',
- *   data: new Uint8Array([...])
- * };
- * ```
- */
-type ImageSource = {
-    type: 'base64';
-    data: string;
-} | {
-    type: 'url';
-    url: string;
-} | {
-    type: 'bytes';
-    data: Uint8Array;
-};
-/**
- * Text content block.
- *
- * The most common content block type, containing plain text content.
- *
- * @example
- * ```typescript
- * const textBlock: TextBlock = {
- *   type: 'text',
- *   text: 'Hello, world!'
- * };
- * ```
- */
-interface TextBlock {
-    /** Discriminator for text blocks */
-    type: 'text';
-    /** The text content */
-    text: string;
-}
-/**
- * Image content block.
- *
- * Contains an image with its source data and metadata.
- *
- * @example
- * ```typescript
- * const imageBlock: ImageBlock = {
- *   type: 'image',
- *   source: { type: 'url', url: 'https://example.com/photo.jpg' },
- *   mimeType: 'image/jpeg',
- *   width: 1920,
- *   height: 1080
- * };
- * ```
- */
-interface ImageBlock {
-    /** Discriminator for image blocks */
-    type: 'image';
-    /** The image data source */
-    source: ImageSource;
-    /** MIME type of the image (e.g., 'image/png', 'image/jpeg') */
-    mimeType: string;
-    /** Image width in pixels */
-    width?: number;
-    /** Image height in pixels */
-    height?: number;
-}
-/**
- * Audio content block.
- *
- * Contains audio data with its metadata.
- *
- * @example
- * ```typescript
- * const audioBlock: AudioBlock = {
- *   type: 'audio',
- *   data: audioBytes,
- *   mimeType: 'audio/mp3',
- *   duration: 120.5
- * };
- * ```
- */
-interface AudioBlock {
-    /** Discriminator for audio blocks */
-    type: 'audio';
-    /** Raw audio data */
-    data: Uint8Array;
-    /** MIME type of the audio (e.g., 'audio/mp3', 'audio/wav') */
-    mimeType: string;
-    /** Duration in seconds */
-    duration?: number;
-}
-/**
- * Video content block.
- *
- * Contains video data with its metadata.
- *
- * @example
- * ```typescript
- * const videoBlock: VideoBlock = {
- *   type: 'video',
- *   data: videoBytes,
- *   mimeType: 'video/mp4',
- *   duration: 30,
- *   width: 1920,
- *   height: 1080
- * };
- * ```
- */
-interface VideoBlock {
-    /** Discriminator for video blocks */
-    type: 'video';
-    /** Raw video data */
-    data: Uint8Array;
-    /** MIME type of the video (e.g., 'video/mp4', 'video/webm') */
-    mimeType: string;
-    /** Duration in seconds */
-    duration?: number;
-    /** Video width in pixels */
-    width?: number;
-    /** Video height in pixels */
-    height?: number;
-}
-/**
- * Binary content block for arbitrary data.
- *
- * A generic block type for data that doesn't fit other categories.
- *
- * @example
- * ```typescript
- * const binaryBlock: BinaryBlock = {
- *   type: 'binary',
- *   data: pdfBytes,
- *   mimeType: 'application/pdf',
- *   metadata: { filename: 'document.pdf', pages: 10 }
- * };
- * ```
- */
-interface BinaryBlock {
-    /** Discriminator for binary blocks */
-    type: 'binary';
-    /** Raw binary data */
-    data: Uint8Array;
-    /** MIME type of the data */
-    mimeType: string;
-    /** Additional metadata about the binary content */
-    metadata?: Record<string, unknown>;
-}
-/**
- * Union of all content block types.
- *
- * Used when a function or property can accept any type of content block.
- */
-type ContentBlock = TextBlock | ImageBlock | AudioBlock | VideoBlock | BinaryBlock;
-/**
- * Content types allowed in user messages.
- *
- * Users can send any type of content block including binary data.
- */
-type UserContent = TextBlock | ImageBlock | AudioBlock | VideoBlock | BinaryBlock;
-/**
- * Content types allowed in assistant messages.
- *
- * Assistants can generate text and media but not arbitrary binary data.
- */
-type AssistantContent = TextBlock | ImageBlock | AudioBlock | VideoBlock;
-/**
- * Creates a text content block from a string.
- *
- * @param content - The text content
- * @returns A TextBlock containing the provided text
- *
- * @example
- * ```typescript
- * const block = text('Hello, world!');
- * // { type: 'text', text: 'Hello, world!' }
- * ```
- */
-declare function text(content: string): TextBlock;
-/**
- * Type guard for TextBlock.
- *
- * @param block - The content block to check
- * @returns True if the block is a TextBlock
- *
- * @example
- * ```typescript
- * if (isTextBlock(block)) {
- *   console.log(block.text);
- * }
- * ```
- */
-declare function isTextBlock(block: ContentBlock): block is TextBlock;
-/**
- * Type guard for ImageBlock.
- *
- * @param block - The content block to check
- * @returns True if the block is an ImageBlock
- *
- * @example
- * ```typescript
- * if (isImageBlock(block)) {
- *   console.log(block.mimeType, block.width, block.height);
- * }
- * ```
- */
-declare function isImageBlock(block: ContentBlock): block is ImageBlock;
-/**
- * Type guard for AudioBlock.
- *
- * @param block - The content block to check
- * @returns True if the block is an AudioBlock
- *
- * @example
- * ```typescript
- * if (isAudioBlock(block)) {
- *   console.log(block.mimeType, block.duration);
- * }
- * ```
- */
-declare function isAudioBlock(block: ContentBlock): block is AudioBlock;
-/**
- * Type guard for VideoBlock.
- *
- * @param block - The content block to check
- * @returns True if the block is a VideoBlock
- *
- * @example
- * ```typescript
- * if (isVideoBlock(block)) {
- *   console.log(block.mimeType, block.duration);
- * }
- * ```
- */
-declare function isVideoBlock(block: ContentBlock): block is VideoBlock;
-/**
- * Type guard for BinaryBlock.
- *
- * @param block - The content block to check
- * @returns True if the block is a BinaryBlock
- *
- * @example
- * ```typescript
- * if (isBinaryBlock(block)) {
- *   console.log(block.mimeType, block.metadata);
- * }
- * ```
- */
-declare function isBinaryBlock(block: ContentBlock): block is BinaryBlock;
+import { C as ContentBlock, I as ImageBlock, a as AudioBlock, V as VideoBlock, A as AssistantContent, U as UserContent } from './content-DEl3z_W2.js';
 
 /**
  * @fileoverview JSON Schema types for tool parameters and structured outputs.
@@ -1283,4 +1010,4 @@ declare function contentBlockStart(index: number): StreamEvent;
  */
 declare function contentBlockStop(index: number): StreamEvent;
 
-export { type TurnJSON as $, type AssistantContent as A, type BinaryBlock as B, type ContentBlock as C, isUserMessage as D, isAssistantMessage as E, isToolResultMessage as F, type MessageMetadata as G, type MessageOptions as H, type ImageSource as I, type JSONSchema as J, createTurn as K, emptyUsage as L, Message as M, aggregateUsage as N, type StreamEventType as O, type EventDelta as P, createStreamResult as Q, textDelta as R, type StreamResult as S, type Turn as T, type UserContent as U, type VideoBlock as V, toolCallDelta as W, messageStart as X, messageStop as Y, contentBlockStart as Z, contentBlockStop as _, type MessageType as a, type MessageJSON as b, type Tool as c, type ToolUseStrategy as d, AssistantMessage as e, type TokenUsage as f, type StreamEvent as g, type ImageBlock as h, type JSONSchemaProperty as i, type JSONSchemaPropertyType as j, type TextBlock as k, type AudioBlock as l, isTextBlock as m, isImageBlock as n, isAudioBlock as o, isVideoBlock as p, isBinaryBlock as q, type ToolCall as r, type ToolResult as s, text as t, type ToolMetadata as u, type BeforeCallResult as v, type AfterCallResult as w, type ToolExecution as x, UserMessage as y, ToolResultMessage as z };
+export { AssistantMessage as A, type BeforeCallResult as B, messageStart as C, messageStop as D, type EventDelta as E, contentBlockStart as F, contentBlockStop as G, type TurnJSON as H, type JSONSchema as J, Message as M, type StreamResult as S, type Turn as T, UserMessage as U, type MessageType as a, type MessageJSON as b, type Tool as c, type ToolUseStrategy as d, type TokenUsage as e, type StreamEvent as f, type JSONSchemaProperty as g, type JSONSchemaPropertyType as h, type ToolCall as i, type ToolResult as j, type ToolMetadata as k, type AfterCallResult as l, type ToolExecution as m, ToolResultMessage as n, isUserMessage as o, isAssistantMessage as p, isToolResultMessage as q, type MessageMetadata as r, type MessageOptions as s, createTurn as t, emptyUsage as u, aggregateUsage as v, type StreamEventType as w, createStreamResult as x, textDelta as y, toolCallDelta as z };

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,3 +1,6 @@
+import {
+  Image
+} from "../chunk-WAKD3OO5.js";
 import {
   AssistantMessage,
   isAssistantMessage,
@@ -8,14 +11,14 @@ import {
   parseSSEStream
 } from "../chunk-Z7RBRCRN.js";
 import {
-  doFetch,
-  doStreamFetch,
-  normalizeHttpError,
   resolveApiKey
-} from "../chunk-5FEAOEXV.js";
+} from "../chunk-P5IRTEM5.js";
 import {
-  UPPError
-} from "../chunk-DZQHVGNV.js";
+  UPPError,
+  doFetch,
+  doStreamFetch,
+  normalizeHttpError
+} from "../chunk-U3FZWV4U.js";
 
 // src/providers/xai/transform.completions.ts
 function transformRequest(request, modelId) {
@@ -1728,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 {
@@ -1780,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;
@@ -1796,7 +1906,8 @@ function createXAIProvider() {
         default:
           return completionsHandler;
       }
-    }
+    },
+    image: imageHandler
   };
   Object.defineProperties(fn, {
     name: {
@@ -1819,6 +1930,7 @@ function createXAIProvider() {
   completionsHandler._setProvider?.(provider);
   responsesHandler._setProvider?.(provider);
   messagesHandler._setProvider?.(provider);
+  imageHandler._setProvider?.(provider);
   return provider;
 }
 var xai = createXAIProvider();