@tanstack/ai-grok 0.1.0 → 0.4.0

package/dist/esm/adapters/summarize.js

@@ -11,7 +11,7 @@ class GrokSummarizeAdapter extends BaseSummarizeAdapter {
   async summarize(options) {
     const systemPrompt = this.buildSummarizationPrompt(options);
     let summary = "";
-    let id = "";
+    const id = "";
     let model = options.model;
     let usage = { promptTokens: 0, completionTokens: 0, totalTokens: 0 };
     for await (const chunk of this.textAdapter.chatStream({
@@ -21,13 +21,18 @@ class GrokSummarizeAdapter extends BaseSummarizeAdapter {
       maxTokens: options.maxLength,
       temperature: 0.3
     })) {
-      if (chunk.type === "content") {
-        summary = chunk.content;
-        id = chunk.id;
-        model = chunk.model;
+      if (chunk.type === "TEXT_MESSAGE_CONTENT") {
+        if (chunk.content) {
+          summary = chunk.content;
+        } else {
+          summary += chunk.delta;
+        }
+        model = chunk.model || model;
       }
-      if (chunk.type === "done" && chunk.usage) {
-        usage = chunk.usage;
+      if (chunk.type === "RUN_FINISHED") {
+        if (chunk.usage) {
+          usage = chunk.usage;
+        }
       }
     }
     return { id, model, summary, usage };

package/dist/esm/adapters/summarize.js.map

@@ -1 +1 @@
-{"version":3,"file":"summarize.js","sources":["../../../src/adapters/summarize.ts"],"sourcesContent":["import { BaseSummarizeAdapter } from '@tanstack/ai/adapters'\nimport { getGrokApiKeyFromEnv } from '../utils'\nimport { GrokTextAdapter } from './text'\nimport type { GROK_CHAT_MODELS } from '../model-meta'\nimport type {\n  StreamChunk,\n  SummarizationOptions,\n  SummarizationResult,\n} from '@tanstack/ai'\nimport type { GrokClientConfig } from '../utils'\n\n/**\n * Configuration for Grok summarize adapter\n */\nexport interface GrokSummarizeConfig extends GrokClientConfig {}\n\n/**\n * Grok-specific provider options for summarization\n */\nexport interface GrokSummarizeProviderOptions {\n  /** Temperature for response generation (0-2) */\n  temperature?: number\n  /** Maximum tokens in the response */\n  maxTokens?: number\n}\n\n/** Model type for Grok summarization */\nexport type GrokSummarizeModel = (typeof GROK_CHAT_MODELS)[number]\n\n/**\n * Grok Summarize Adapter\n *\n * A thin wrapper around the text adapter that adds summarization-specific prompting.\n * Delegates all API calls to the GrokTextAdapter.\n */\nexport class GrokSummarizeAdapter<\n  TModel extends GrokSummarizeModel,\n> extends BaseSummarizeAdapter<TModel, GrokSummarizeProviderOptions> {\n  readonly kind = 'summarize' as const\n  readonly name = 'grok' as const\n\n  private textAdapter: GrokTextAdapter<TModel>\n\n  constructor(config: GrokSummarizeConfig, model: TModel) {\n    super({}, model)\n    this.textAdapter = new GrokTextAdapter(config, model)\n  }\n\n  async summarize(options: SummarizationOptions): Promise<SummarizationResult> {\n    const systemPrompt = this.buildSummarizationPrompt(options)\n\n    // Use the text adapter's streaming and collect the result\n    let summary = ''\n    let id = ''\n    let model = options.model\n    let usage = { promptTokens: 0, completionTokens: 0, totalTokens: 0 }\n\n    for await (const chunk of this.textAdapter.chatStream({\n      model: options.model,\n      messages: [{ role: 'user', content: options.text }],\n      systemPrompts: [systemPrompt],\n      maxTokens: options.maxLength,\n      temperature: 0.3,\n    })) {\n      if (chunk.type === 'content') {\n        summary = chunk.content\n        id = chunk.id\n        model = chunk.model\n      }\n      if (chunk.type === 'done' && chunk.usage) {\n        usage = chunk.usage\n      }\n    }\n\n    return { id, model, summary, usage }\n  }\n\n  async *summarizeStream(\n    options: SummarizationOptions,\n  ): AsyncIterable<StreamChunk> {\n    const systemPrompt = this.buildSummarizationPrompt(options)\n\n    // Delegate directly to the text adapter's streaming\n    yield* this.textAdapter.chatStream({\n      model: options.model,\n      messages: [{ role: 'user', content: options.text }],\n      systemPrompts: [systemPrompt],\n      maxTokens: options.maxLength,\n      temperature: 0.3,\n    })\n  }\n\n  private buildSummarizationPrompt(options: SummarizationOptions): string {\n    let prompt = 'You are a professional summarizer. '\n\n    switch (options.style) {\n      case 'bullet-points':\n        prompt += 'Provide a summary in bullet point format. '\n        break\n      case 'paragraph':\n        prompt += 'Provide a summary in paragraph format. '\n        break\n      case 'concise':\n        prompt += 'Provide a very concise summary in 1-2 sentences. '\n        break\n      default:\n        prompt += 'Provide a clear and concise summary. '\n    }\n\n    if (options.focus && options.focus.length > 0) {\n      prompt += `Focus on the following aspects: ${options.focus.join(', ')}. `\n    }\n\n    if (options.maxLength) {\n      prompt += `Keep the summary under ${options.maxLength} tokens. `\n    }\n\n    return prompt\n  }\n}\n\n/**\n * Creates a Grok summarize adapter with explicit API key.\n * Type resolution happens here at the call site.\n *\n * @param model - The model name (e.g., 'grok-3', 'grok-4')\n * @param apiKey - Your xAI API key\n * @param config - Optional additional configuration\n * @returns Configured Grok summarize adapter instance with resolved types\n *\n * @example\n * ```typescript\n * const adapter = createGrokSummarize('grok-3', \"xai-...\");\n * ```\n */\nexport function createGrokSummarize<TModel extends GrokSummarizeModel>(\n  model: TModel,\n  apiKey: string,\n  config?: Omit<GrokSummarizeConfig, 'apiKey'>,\n): GrokSummarizeAdapter<TModel> {\n  return new GrokSummarizeAdapter({ apiKey, ...config }, model)\n}\n\n/**\n * Creates a Grok summarize adapter with automatic API key detection from environment variables.\n * Type resolution happens here at the call site.\n *\n * Looks for `XAI_API_KEY` in:\n * - `process.env` (Node.js)\n * - `window.env` (Browser with injected env)\n *\n * @param model - The model name (e.g., 'grok-3', 'grok-4')\n * @param config - Optional configuration (excluding apiKey which is auto-detected)\n * @returns Configured Grok summarize adapter instance with resolved types\n * @throws Error if XAI_API_KEY is not found in environment\n *\n * @example\n * ```typescript\n * // Automatically uses XAI_API_KEY from environment\n * const adapter = grokSummarize('grok-3');\n *\n * await summarize({\n *   adapter,\n *   text: \"Long article text...\"\n * });\n * ```\n */\nexport function grokSummarize<TModel extends GrokSummarizeModel>(\n  model: TModel,\n  config?: Omit<GrokSummarizeConfig, 'apiKey'>,\n): GrokSummarizeAdapter<TModel> {\n  const apiKey = getGrokApiKeyFromEnv()\n  return createGrokSummarize(model, apiKey, config)\n}\n"],"names":[],"mappings":";;;AAmCO,MAAM,6BAEH,qBAA2D;AAAA,EAMnE,YAAY,QAA6B,OAAe;AACtD,UAAM,CAAA,GAAI,KAAK;AANjB,SAAS,OAAO;AAChB,SAAS,OAAO;AAMd,SAAK,cAAc,IAAI,gBAAgB,QAAQ,KAAK;AAAA,EACtD;AAAA,EAEA,MAAM,UAAU,SAA6D;AAC3E,UAAM,eAAe,KAAK,yBAAyB,OAAO;AAG1D,QAAI,UAAU;AACd,QAAI,KAAK;AACT,QAAI,QAAQ,QAAQ;AACpB,QAAI,QAAQ,EAAE,cAAc,GAAG,kBAAkB,GAAG,aAAa,EAAA;AAEjE,qBAAiB,SAAS,KAAK,YAAY,WAAW;AAAA,MACpD,OAAO,QAAQ;AAAA,MACf,UAAU,CAAC,EAAE,MAAM,QAAQ,SAAS,QAAQ,MAAM;AAAA,MAClD,eAAe,CAAC,YAAY;AAAA,MAC5B,WAAW,QAAQ;AAAA,MACnB,aAAa;AAAA,IAAA,CACd,GAAG;AACF,UAAI,MAAM,SAAS,WAAW;AAC5B,kBAAU,MAAM;AAChB,aAAK,MAAM;AACX,gBAAQ,MAAM;AAAA,MAChB;AACA,UAAI,MAAM,SAAS,UAAU,MAAM,OAAO;AACxC,gBAAQ,MAAM;AAAA,MAChB;AAAA,IACF;AAEA,WAAO,EAAE,IAAI,OAAO,SAAS,MAAA;AAAA,EAC/B;AAAA,EAEA,OAAO,gBACL,SAC4B;AAC5B,UAAM,eAAe,KAAK,yBAAyB,OAAO;AAG1D,WAAO,KAAK,YAAY,WAAW;AAAA,MACjC,OAAO,QAAQ;AAAA,MACf,UAAU,CAAC,EAAE,MAAM,QAAQ,SAAS,QAAQ,MAAM;AAAA,MAClD,eAAe,CAAC,YAAY;AAAA,MAC5B,WAAW,QAAQ;AAAA,MACnB,aAAa;AAAA,IAAA,CACd;AAAA,EACH;AAAA,EAEQ,yBAAyB,SAAuC;AACtE,QAAI,SAAS;AAEb,YAAQ,QAAQ,OAAA;AAAA,MACd,KAAK;AACH,kBAAU;AACV;AAAA,MACF,KAAK;AACH,kBAAU;AACV;AAAA,MACF,KAAK;AACH,kBAAU;AACV;AAAA,MACF;AACE,kBAAU;AAAA,IAAA;AAGd,QAAI,QAAQ,SAAS,QAAQ,MAAM,SAAS,GAAG;AAC7C,gBAAU,mCAAmC,QAAQ,MAAM,KAAK,IAAI,CAAC;AAAA,IACvE;AAEA,QAAI,QAAQ,WAAW;AACrB,gBAAU,0BAA0B,QAAQ,SAAS;AAAA,IACvD;AAEA,WAAO;AAAA,EACT;AACF;AAgBO,SAAS,oBACd,OACA,QACA,QAC8B;AAC9B,SAAO,IAAI,qBAAqB,EAAE,QAAQ,GAAG,OAAA,GAAU,KAAK;AAC9D;AA0BO,SAAS,cACd,OACA,QAC8B;AAC9B,QAAM,SAAS,qBAAA;AACf,SAAO,oBAAoB,OAAO,QAAQ,MAAM;AAClD;"}
+{"version":3,"file":"summarize.js","sources":["../../../src/adapters/summarize.ts"],"sourcesContent":["import { BaseSummarizeAdapter } from '@tanstack/ai/adapters'\nimport { getGrokApiKeyFromEnv } from '../utils'\nimport { GrokTextAdapter } from './text'\nimport type { GROK_CHAT_MODELS } from '../model-meta'\nimport type {\n  StreamChunk,\n  SummarizationOptions,\n  SummarizationResult,\n} from '@tanstack/ai'\nimport type { GrokClientConfig } from '../utils'\n\n/**\n * Configuration for Grok summarize adapter\n */\nexport interface GrokSummarizeConfig extends GrokClientConfig {}\n\n/**\n * Grok-specific provider options for summarization\n */\nexport interface GrokSummarizeProviderOptions {\n  /** Temperature for response generation (0-2) */\n  temperature?: number\n  /** Maximum tokens in the response */\n  maxTokens?: number\n}\n\n/** Model type for Grok summarization */\nexport type GrokSummarizeModel = (typeof GROK_CHAT_MODELS)[number]\n\n/**\n * Grok Summarize Adapter\n *\n * A thin wrapper around the text adapter that adds summarization-specific prompting.\n * Delegates all API calls to the GrokTextAdapter.\n */\nexport class GrokSummarizeAdapter<\n  TModel extends GrokSummarizeModel,\n> extends BaseSummarizeAdapter<TModel, GrokSummarizeProviderOptions> {\n  readonly kind = 'summarize' as const\n  readonly name = 'grok' as const\n\n  private textAdapter: GrokTextAdapter<TModel>\n\n  constructor(config: GrokSummarizeConfig, model: TModel) {\n    super({}, model)\n    this.textAdapter = new GrokTextAdapter(config, model)\n  }\n\n  async summarize(options: SummarizationOptions): Promise<SummarizationResult> {\n    const systemPrompt = this.buildSummarizationPrompt(options)\n\n    // Use the text adapter's streaming and collect the result\n    let summary = ''\n    const id = ''\n    let model = options.model\n    let usage = { promptTokens: 0, completionTokens: 0, totalTokens: 0 }\n\n    for await (const chunk of this.textAdapter.chatStream({\n      model: options.model,\n      messages: [{ role: 'user', content: options.text }],\n      systemPrompts: [systemPrompt],\n      maxTokens: options.maxLength,\n      temperature: 0.3,\n    })) {\n      // AG-UI TEXT_MESSAGE_CONTENT event\n      if (chunk.type === 'TEXT_MESSAGE_CONTENT') {\n        if (chunk.content) {\n          summary = chunk.content\n        } else {\n          summary += chunk.delta\n        }\n        model = chunk.model || model\n      }\n      // AG-UI RUN_FINISHED event\n      if (chunk.type === 'RUN_FINISHED') {\n        if (chunk.usage) {\n          usage = chunk.usage\n        }\n      }\n    }\n\n    return { id, model, summary, usage }\n  }\n\n  async *summarizeStream(\n    options: SummarizationOptions,\n  ): AsyncIterable<StreamChunk> {\n    const systemPrompt = this.buildSummarizationPrompt(options)\n\n    // Delegate directly to the text adapter's streaming\n    yield* this.textAdapter.chatStream({\n      model: options.model,\n      messages: [{ role: 'user', content: options.text }],\n      systemPrompts: [systemPrompt],\n      maxTokens: options.maxLength,\n      temperature: 0.3,\n    })\n  }\n\n  private buildSummarizationPrompt(options: SummarizationOptions): string {\n    let prompt = 'You are a professional summarizer. '\n\n    switch (options.style) {\n      case 'bullet-points':\n        prompt += 'Provide a summary in bullet point format. '\n        break\n      case 'paragraph':\n        prompt += 'Provide a summary in paragraph format. '\n        break\n      case 'concise':\n        prompt += 'Provide a very concise summary in 1-2 sentences. '\n        break\n      default:\n        prompt += 'Provide a clear and concise summary. '\n    }\n\n    if (options.focus && options.focus.length > 0) {\n      prompt += `Focus on the following aspects: ${options.focus.join(', ')}. `\n    }\n\n    if (options.maxLength) {\n      prompt += `Keep the summary under ${options.maxLength} tokens. `\n    }\n\n    return prompt\n  }\n}\n\n/**\n * Creates a Grok summarize adapter with explicit API key.\n * Type resolution happens here at the call site.\n *\n * @param model - The model name (e.g., 'grok-3', 'grok-4')\n * @param apiKey - Your xAI API key\n * @param config - Optional additional configuration\n * @returns Configured Grok summarize adapter instance with resolved types\n *\n * @example\n * ```typescript\n * const adapter = createGrokSummarize('grok-3', \"xai-...\");\n * ```\n */\nexport function createGrokSummarize<TModel extends GrokSummarizeModel>(\n  model: TModel,\n  apiKey: string,\n  config?: Omit<GrokSummarizeConfig, 'apiKey'>,\n): GrokSummarizeAdapter<TModel> {\n  return new GrokSummarizeAdapter({ apiKey, ...config }, model)\n}\n\n/**\n * Creates a Grok summarize adapter with automatic API key detection from environment variables.\n * Type resolution happens here at the call site.\n *\n * Looks for `XAI_API_KEY` in:\n * - `process.env` (Node.js)\n * - `window.env` (Browser with injected env)\n *\n * @param model - The model name (e.g., 'grok-3', 'grok-4')\n * @param config - Optional configuration (excluding apiKey which is auto-detected)\n * @returns Configured Grok summarize adapter instance with resolved types\n * @throws Error if XAI_API_KEY is not found in environment\n *\n * @example\n * ```typescript\n * // Automatically uses XAI_API_KEY from environment\n * const adapter = grokSummarize('grok-3');\n *\n * await summarize({\n *   adapter,\n *   text: \"Long article text...\"\n * });\n * ```\n */\nexport function grokSummarize<TModel extends GrokSummarizeModel>(\n  model: TModel,\n  config?: Omit<GrokSummarizeConfig, 'apiKey'>,\n): GrokSummarizeAdapter<TModel> {\n  const apiKey = getGrokApiKeyFromEnv()\n  return createGrokSummarize(model, apiKey, config)\n}\n"],"names":[],"mappings":";;;AAmCO,MAAM,6BAEH,qBAA2D;AAAA,EAMnE,YAAY,QAA6B,OAAe;AACtD,UAAM,CAAA,GAAI,KAAK;AANjB,SAAS,OAAO;AAChB,SAAS,OAAO;AAMd,SAAK,cAAc,IAAI,gBAAgB,QAAQ,KAAK;AAAA,EACtD;AAAA,EAEA,MAAM,UAAU,SAA6D;AAC3E,UAAM,eAAe,KAAK,yBAAyB,OAAO;AAG1D,QAAI,UAAU;AACd,UAAM,KAAK;AACX,QAAI,QAAQ,QAAQ;AACpB,QAAI,QAAQ,EAAE,cAAc,GAAG,kBAAkB,GAAG,aAAa,EAAA;AAEjE,qBAAiB,SAAS,KAAK,YAAY,WAAW;AAAA,MACpD,OAAO,QAAQ;AAAA,MACf,UAAU,CAAC,EAAE,MAAM,QAAQ,SAAS,QAAQ,MAAM;AAAA,MAClD,eAAe,CAAC,YAAY;AAAA,MAC5B,WAAW,QAAQ;AAAA,MACnB,aAAa;AAAA,IAAA,CACd,GAAG;AAEF,UAAI,MAAM,SAAS,wBAAwB;AACzC,YAAI,MAAM,SAAS;AACjB,oBAAU,MAAM;AAAA,QAClB,OAAO;AACL,qBAAW,MAAM;AAAA,QACnB;AACA,gBAAQ,MAAM,SAAS;AAAA,MACzB;AAEA,UAAI,MAAM,SAAS,gBAAgB;AACjC,YAAI,MAAM,OAAO;AACf,kBAAQ,MAAM;AAAA,QAChB;AAAA,MACF;AAAA,IACF;AAEA,WAAO,EAAE,IAAI,OAAO,SAAS,MAAA;AAAA,EAC/B;AAAA,EAEA,OAAO,gBACL,SAC4B;AAC5B,UAAM,eAAe,KAAK,yBAAyB,OAAO;AAG1D,WAAO,KAAK,YAAY,WAAW;AAAA,MACjC,OAAO,QAAQ;AAAA,MACf,UAAU,CAAC,EAAE,MAAM,QAAQ,SAAS,QAAQ,MAAM;AAAA,MAClD,eAAe,CAAC,YAAY;AAAA,MAC5B,WAAW,QAAQ;AAAA,MACnB,aAAa;AAAA,IAAA,CACd;AAAA,EACH;AAAA,EAEQ,yBAAyB,SAAuC;AACtE,QAAI,SAAS;AAEb,YAAQ,QAAQ,OAAA;AAAA,MACd,KAAK;AACH,kBAAU;AACV;AAAA,MACF,KAAK;AACH,kBAAU;AACV;AAAA,MACF,KAAK;AACH,kBAAU;AACV;AAAA,MACF;AACE,kBAAU;AAAA,IAAA;AAGd,QAAI,QAAQ,SAAS,QAAQ,MAAM,SAAS,GAAG;AAC7C,gBAAU,mCAAmC,QAAQ,MAAM,KAAK,IAAI,CAAC;AAAA,IACvE;AAEA,QAAI,QAAQ,WAAW;AACrB,gBAAU,0BAA0B,QAAQ,SAAS;AAAA,IACvD;AAEA,WAAO;AAAA,EACT;AACF;AAgBO,SAAS,oBACd,OACA,QACA,QAC8B;AAC9B,SAAO,IAAI,qBAAqB,EAAE,QAAQ,GAAG,OAAA,GAAU,KAAK;AAC9D;AA0BO,SAAS,cACd,OACA,QAC8B;AAC9B,QAAM,SAAS,qBAAA;AACf,SAAO,oBAAoB,OAAO,QAAQ,MAAM;AAClD;"}

package/dist/esm/adapters/text.js

@@ -12,19 +12,44 @@ class GrokTextAdapter extends BaseTextAdapter {
   }
   async *chatStream(options) {
     const requestParams = this.mapTextOptionsToGrok(options);
+    const timestamp = Date.now();
+    const aguiState = {
+      runId: generateId(this.name),
+      messageId: generateId(this.name),
+      timestamp,
+      hasEmittedRunStarted: false
+    };
     try {
       const stream = await this.client.chat.completions.create({
         ...requestParams,
         stream: true
       });
-      yield* this.processGrokStreamChunks(stream, options);
+      yield* this.processGrokStreamChunks(stream, options, aguiState);
     } catch (error) {
       const err = error;
+      if (!aguiState.hasEmittedRunStarted) {
+        aguiState.hasEmittedRunStarted = true;
+        yield {
+          type: "RUN_STARTED",
+          runId: aguiState.runId,
+          model: options.model,
+          timestamp
+        };
+      }
+      yield {
+        type: "RUN_ERROR",
+        runId: aguiState.runId,
+        model: options.model,
+        timestamp,
+        error: {
+          message: err.message || "Unknown error",
+          code: err.code
+        }
+      };
       console.error(">>> chatStream: Fatal error during response creation <<<");
       console.error(">>> Error message:", err.message);
       console.error(">>> Error stack:", err.stack);
       console.error(">>> Full error:", err);
-      throw error;
     }
   }
   /**
@@ -80,29 +105,46 @@ class GrokTextAdapter extends BaseTextAdapter {
       throw error;
     }
   }
-  async *processGrokStreamChunks(stream, options) {
+  async *processGrokStreamChunks(stream, options, aguiState) {
     let accumulatedContent = "";
-    const timestamp = Date.now();
-    let responseId = generateId(this.name);
+    const timestamp = aguiState.timestamp;
+    let hasEmittedTextMessageStart = false;
     const toolCallsInProgress = /* @__PURE__ */ new Map();
     try {
       for await (const chunk of stream) {
-        responseId = chunk.id || responseId;
         const choice = chunk.choices[0];
         if (!choice) continue;
+        if (!aguiState.hasEmittedRunStarted) {
+          aguiState.hasEmittedRunStarted = true;
+          yield {
+            type: "RUN_STARTED",
+            runId: aguiState.runId,
+            model: chunk.model || options.model,
+            timestamp
+          };
+        }
         const delta = choice.delta;
         const deltaContent = delta.content;
         const deltaToolCalls = delta.tool_calls;
         if (deltaContent) {
+          if (!hasEmittedTextMessageStart) {
+            hasEmittedTextMessageStart = true;
+            yield {
+              type: "TEXT_MESSAGE_START",
+              messageId: aguiState.messageId,
+              model: chunk.model || options.model,
+              timestamp,
+              role: "assistant"
+            };
+          }
           accumulatedContent += deltaContent;
           yield {
-            type: "content",
-            id: responseId,
+            type: "TEXT_MESSAGE_CONTENT",
+            messageId: aguiState.messageId,
             model: chunk.model || options.model,
             timestamp,
             delta: deltaContent,
-            content: accumulatedContent,
-            role: "assistant"
+            content: accumulatedContent
           };
         }
         if (deltaToolCalls) {
@@ -112,7 +154,8 @@ class GrokTextAdapter extends BaseTextAdapter {
               toolCallsInProgress.set(index, {
                 id: toolCallDelta.id || "",
                 name: toolCallDelta.function?.name || "",
-                arguments: ""
+                arguments: "",
+                started: false
               });
             }
             const toolCall = toolCallsInProgress.get(index);
@@ -125,31 +168,59 @@ class GrokTextAdapter extends BaseTextAdapter {
             if (toolCallDelta.function?.arguments) {
               toolCall.arguments += toolCallDelta.function.arguments;
             }
+            if (toolCall.id && toolCall.name && !toolCall.started) {
+              toolCall.started = true;
+              yield {
+                type: "TOOL_CALL_START",
+                toolCallId: toolCall.id,
+                toolName: toolCall.name,
+                model: chunk.model || options.model,
+                timestamp,
+                index
+              };
+            }
+            if (toolCallDelta.function?.arguments && toolCall.started) {
+              yield {
+                type: "TOOL_CALL_ARGS",
+                toolCallId: toolCall.id,
+                model: chunk.model || options.model,
+                timestamp,
+                delta: toolCallDelta.function.arguments
+              };
+            }
           }
         }
         if (choice.finish_reason) {
           if (choice.finish_reason === "tool_calls" || toolCallsInProgress.size > 0) {
-            for (const [index, toolCall] of toolCallsInProgress) {
+            for (const [, toolCall] of toolCallsInProgress) {
+              let parsedInput = {};
+              try {
+                parsedInput = toolCall.arguments ? JSON.parse(toolCall.arguments) : {};
+              } catch {
+                parsedInput = {};
+              }
               yield {
-                type: "tool_call",
-                id: responseId,
+                type: "TOOL_CALL_END",
+                toolCallId: toolCall.id,
+                toolName: toolCall.name,
                 model: chunk.model || options.model,
                 timestamp,
-                index,
-                toolCall: {
-                  id: toolCall.id,
-                  type: "function",
-                  function: {
-                    name: toolCall.name,
-                    arguments: toolCall.arguments
-                  }
-                }
+                input: parsedInput
               };
             }
           }
+          const computedFinishReason = choice.finish_reason === "tool_calls" || toolCallsInProgress.size > 0 ? "tool_calls" : "stop";
+          if (hasEmittedTextMessageStart) {
+            yield {
+              type: "TEXT_MESSAGE_END",
+              messageId: aguiState.messageId,
+              model: chunk.model || options.model,
+              timestamp
+            };
+          }
           yield {
-            type: "done",
-            id: responseId,
+            type: "RUN_FINISHED",
+            runId: aguiState.runId,
             model: chunk.model || options.model,
             timestamp,
             usage: chunk.usage ? {
@@ -157,7 +228,7 @@ class GrokTextAdapter extends BaseTextAdapter {
               completionTokens: chunk.usage.completion_tokens || 0,
               totalTokens: chunk.usage.total_tokens || 0
             } : void 0,
-            finishReason: choice.finish_reason === "tool_calls" || toolCallsInProgress.size > 0 ? "tool_calls" : "stop"
+            finishReason: computedFinishReason
           };
         }
       }
@@ -165,8 +236,8 @@ class GrokTextAdapter extends BaseTextAdapter {
       const err = error;
       console.log("[Grok Adapter] Stream ended with error:", err.message);
       yield {
-        type: "error",
-        id: responseId,
+        type: "RUN_ERROR",
+        runId: aguiState.runId,
         model: options.model,
         timestamp,
         error: {
@@ -245,10 +316,12 @@ class GrokTextAdapter extends BaseTextAdapter {
         parts.push({ type: "text", text: part.content });
       } else if (part.type === "image") {
         const imageMetadata = part.metadata;
+        const imageValue = part.source.value;
+        const imageUrl = part.source.type === "data" && !imageValue.startsWith("data:") ? `data:${part.source.mimeType};base64,${imageValue}` : imageValue;
         parts.push({
           type: "image_url",
           image_url: {
-            url: part.source.value,
+            url: imageUrl,
             detail: imageMetadata?.detail || "auto"
           }
         });

package/dist/esm/adapters/text.js.map

@@ -1 +1 @@
-{"version":3,"file":"text.js","sources":["../../../src/adapters/text.ts"],"sourcesContent":["import { BaseTextAdapter } from '@tanstack/ai/adapters'\nimport { validateTextProviderOptions } from '../text/text-provider-options'\nimport { convertToolsToProviderFormat } from '../tools'\nimport {\n  createGrokClient,\n  generateId,\n  getGrokApiKeyFromEnv,\n  makeGrokStructuredOutputCompatible,\n  transformNullsToUndefined,\n} from '../utils'\nimport type {\n  GROK_CHAT_MODELS,\n  ResolveInputModalities,\n  ResolveProviderOptions,\n} from '../model-meta'\nimport type {\n  StructuredOutputOptions,\n  StructuredOutputResult,\n} from '@tanstack/ai/adapters'\nimport type OpenAI_SDK from 'openai'\nimport type {\n  ContentPart,\n  ModelMessage,\n  StreamChunk,\n  TextOptions,\n} from '@tanstack/ai'\nimport type { InternalTextProviderOptions } from '../text/text-provider-options'\nimport type {\n  GrokImageMetadata,\n  GrokMessageMetadataByModality,\n} from '../message-types'\nimport type { GrokClientConfig } from '../utils'\n\n/**\n * Configuration for Grok text adapter\n */\nexport interface GrokTextConfig extends GrokClientConfig {}\n\n/**\n * Alias for TextProviderOptions for external use\n */\nexport type { ExternalTextProviderOptions as GrokTextProviderOptions } from '../text/text-provider-options'\n\n/**\n * Grok Text (Chat) Adapter\n *\n * Tree-shakeable adapter for Grok chat/text completion functionality.\n * Uses OpenAI-compatible Chat Completions API (not Responses API).\n */\nexport class GrokTextAdapter<\n  TModel extends (typeof GROK_CHAT_MODELS)[number],\n> extends BaseTextAdapter<\n  TModel,\n  ResolveProviderOptions<TModel>,\n  ResolveInputModalities<TModel>,\n  GrokMessageMetadataByModality\n> {\n  readonly kind = 'text' as const\n  readonly name = 'grok' as const\n\n  private client: OpenAI_SDK\n\n  constructor(config: GrokTextConfig, model: TModel) {\n    super({}, model)\n    this.client = createGrokClient(config)\n  }\n\n  async *chatStream(\n    options: TextOptions<ResolveProviderOptions<TModel>>,\n  ): AsyncIterable<StreamChunk> {\n    const requestParams = this.mapTextOptionsToGrok(options)\n\n    try {\n      const stream = await this.client.chat.completions.create({\n        ...requestParams,\n        stream: true,\n      })\n\n      yield* this.processGrokStreamChunks(stream, options)\n    } catch (error: unknown) {\n      const err = error as Error\n      console.error('>>> chatStream: Fatal error during response creation <<<')\n      console.error('>>> Error message:', err.message)\n      console.error('>>> Error stack:', err.stack)\n      console.error('>>> Full error:', err)\n      throw error\n    }\n  }\n\n  /**\n   * Generate structured output using Grok's JSON Schema response format.\n   * Uses stream: false to get the complete response in one call.\n   *\n   * Grok has strict requirements for structured output (via OpenAI-compatible API):\n   * - All properties must be in the `required` array\n   * - Optional fields should have null added to their type union\n   * - additionalProperties must be false for all objects\n   *\n   * The outputSchema is already JSON Schema (converted in the ai layer).\n   * We apply Grok-specific transformations for structured output compatibility.\n   */\n  async structuredOutput(\n    options: StructuredOutputOptions<ResolveProviderOptions<TModel>>,\n  ): Promise<StructuredOutputResult<unknown>> {\n    const { chatOptions, outputSchema } = options\n    const requestParams = this.mapTextOptionsToGrok(chatOptions)\n\n    // Apply Grok-specific transformations for structured output compatibility\n    const jsonSchema = makeGrokStructuredOutputCompatible(\n      outputSchema,\n      outputSchema.required || [],\n    )\n\n    try {\n      const response = await this.client.chat.completions.create({\n        ...requestParams,\n        stream: false,\n        response_format: {\n          type: 'json_schema',\n          json_schema: {\n            name: 'structured_output',\n            schema: jsonSchema,\n            strict: true,\n          },\n        },\n      })\n\n      // Extract text content from the response\n      const rawText = response.choices[0]?.message.content || ''\n\n      // Parse the JSON response\n      let parsed: unknown\n      try {\n        parsed = JSON.parse(rawText)\n      } catch {\n        throw new Error(\n          `Failed to parse structured output as JSON. Content: ${rawText.slice(0, 200)}${rawText.length > 200 ? '...' : ''}`,\n        )\n      }\n\n      // Transform null values to undefined to match original Zod schema expectations\n      // Grok returns null for optional fields we made nullable in the schema\n      const transformed = transformNullsToUndefined(parsed)\n\n      return {\n        data: transformed,\n        rawText,\n      }\n    } catch (error: unknown) {\n      const err = error as Error\n      console.error('>>> structuredOutput: Error during response creation <<<')\n      console.error('>>> Error message:', err.message)\n      throw error\n    }\n  }\n\n  private async *processGrokStreamChunks(\n    stream: AsyncIterable<OpenAI_SDK.Chat.Completions.ChatCompletionChunk>,\n    options: TextOptions,\n  ): AsyncIterable<StreamChunk> {\n    let accumulatedContent = ''\n    const timestamp = Date.now()\n    let responseId = generateId(this.name)\n\n    // Track tool calls being streamed (arguments come in chunks)\n    const toolCallsInProgress = new Map<\n      number,\n      {\n        id: string\n        name: string\n        arguments: string\n      }\n    >()\n\n    try {\n      for await (const chunk of stream) {\n        responseId = chunk.id || responseId\n        const choice = chunk.choices[0]\n\n        if (!choice) continue\n\n        const delta = choice.delta\n        const deltaContent = delta.content\n        const deltaToolCalls = delta.tool_calls\n\n        // Handle content delta\n        if (deltaContent) {\n          accumulatedContent += deltaContent\n          yield {\n            type: 'content',\n            id: responseId,\n            model: chunk.model || options.model,\n            timestamp,\n            delta: deltaContent,\n            content: accumulatedContent,\n            role: 'assistant',\n          }\n        }\n\n        // Handle tool calls - they come in as deltas\n        if (deltaToolCalls) {\n          for (const toolCallDelta of deltaToolCalls) {\n            const index = toolCallDelta.index\n\n            // Initialize or update the tool call in progress\n            if (!toolCallsInProgress.has(index)) {\n              toolCallsInProgress.set(index, {\n                id: toolCallDelta.id || '',\n                name: toolCallDelta.function?.name || '',\n                arguments: '',\n              })\n            }\n\n            const toolCall = toolCallsInProgress.get(index)!\n\n            // Update with any new data from the delta\n            if (toolCallDelta.id) {\n              toolCall.id = toolCallDelta.id\n            }\n            if (toolCallDelta.function?.name) {\n              toolCall.name = toolCallDelta.function.name\n            }\n            if (toolCallDelta.function?.arguments) {\n              toolCall.arguments += toolCallDelta.function.arguments\n            }\n          }\n        }\n\n        // Handle finish reason\n        if (choice.finish_reason) {\n          // Emit all completed tool calls\n          if (\n            choice.finish_reason === 'tool_calls' ||\n            toolCallsInProgress.size > 0\n          ) {\n            for (const [index, toolCall] of toolCallsInProgress) {\n              yield {\n                type: 'tool_call',\n                id: responseId,\n                model: chunk.model || options.model,\n                timestamp,\n                index,\n                toolCall: {\n                  id: toolCall.id,\n                  type: 'function',\n                  function: {\n                    name: toolCall.name,\n                    arguments: toolCall.arguments,\n                  },\n                },\n              }\n            }\n          }\n\n          yield {\n            type: 'done',\n            id: responseId,\n            model: chunk.model || options.model,\n            timestamp,\n            usage: chunk.usage\n              ? {\n                  promptTokens: chunk.usage.prompt_tokens || 0,\n                  completionTokens: chunk.usage.completion_tokens || 0,\n                  totalTokens: chunk.usage.total_tokens || 0,\n                }\n              : undefined,\n            finishReason:\n              choice.finish_reason === 'tool_calls' ||\n              toolCallsInProgress.size > 0\n                ? 'tool_calls'\n                : 'stop',\n          }\n        }\n      }\n    } catch (error: unknown) {\n      const err = error as Error & { code?: string }\n      console.log('[Grok Adapter] Stream ended with error:', err.message)\n      yield {\n        type: 'error',\n        id: responseId,\n        model: options.model,\n        timestamp,\n        error: {\n          message: err.message || 'Unknown error occurred',\n          code: err.code,\n        },\n      }\n    }\n  }\n\n  /**\n   * Maps common options to Grok-specific Chat Completions format\n   */\n  private mapTextOptionsToGrok(\n    options: TextOptions,\n  ): OpenAI_SDK.Chat.Completions.ChatCompletionCreateParamsStreaming {\n    const modelOptions = options.modelOptions as\n      | Omit<\n          InternalTextProviderOptions,\n          'max_tokens' | 'tools' | 'temperature' | 'input' | 'top_p'\n        >\n      | undefined\n\n    if (modelOptions) {\n      validateTextProviderOptions({\n        ...modelOptions,\n        model: options.model,\n      })\n    }\n\n    const tools = options.tools\n      ? convertToolsToProviderFormat(options.tools)\n      : undefined\n\n    // Build messages array with system prompts\n    const messages: Array<OpenAI_SDK.Chat.Completions.ChatCompletionMessageParam> =\n      []\n\n    // Add system prompts first\n    if (options.systemPrompts && options.systemPrompts.length > 0) {\n      messages.push({\n        role: 'system',\n        content: options.systemPrompts.join('\\n'),\n      })\n    }\n\n    // Convert messages\n    for (const message of options.messages) {\n      messages.push(this.convertMessageToGrok(message))\n    }\n\n    return {\n      model: options.model,\n      messages,\n      temperature: options.temperature,\n      max_tokens: options.maxTokens,\n      top_p: options.topP,\n      tools: tools as Array<OpenAI_SDK.Chat.Completions.ChatCompletionTool>,\n      stream: true,\n      stream_options: { include_usage: true },\n    }\n  }\n\n  private convertMessageToGrok(\n    message: ModelMessage,\n  ): OpenAI_SDK.Chat.Completions.ChatCompletionMessageParam {\n    // Handle tool messages\n    if (message.role === 'tool') {\n      return {\n        role: 'tool',\n        tool_call_id: message.toolCallId || '',\n        content:\n          typeof message.content === 'string'\n            ? message.content\n            : JSON.stringify(message.content),\n      }\n    }\n\n    // Handle assistant messages\n    if (message.role === 'assistant') {\n      const toolCalls = message.toolCalls?.map((tc) => ({\n        id: tc.id,\n        type: 'function' as const,\n        function: {\n          name: tc.function.name,\n          arguments:\n            typeof tc.function.arguments === 'string'\n              ? tc.function.arguments\n              : JSON.stringify(tc.function.arguments),\n        },\n      }))\n\n      return {\n        role: 'assistant',\n        content: this.extractTextContent(message.content),\n        ...(toolCalls && toolCalls.length > 0 ? { tool_calls: toolCalls } : {}),\n      }\n    }\n\n    // Handle user messages - support multimodal content\n    const contentParts = this.normalizeContent(message.content)\n\n    // If only text, use simple string format\n    if (contentParts.length === 1 && contentParts[0]?.type === 'text') {\n      return {\n        role: 'user',\n        content: contentParts[0].content,\n      }\n    }\n\n    // Otherwise, use array format for multimodal\n    const parts: Array<OpenAI_SDK.Chat.Completions.ChatCompletionContentPart> =\n      []\n    for (const part of contentParts) {\n      if (part.type === 'text') {\n        parts.push({ type: 'text', text: part.content })\n      } else if (part.type === 'image') {\n        const imageMetadata = part.metadata as GrokImageMetadata | undefined\n        parts.push({\n          type: 'image_url',\n          image_url: {\n            url: part.source.value,\n            detail: imageMetadata?.detail || 'auto',\n          },\n        })\n      }\n    }\n\n    return {\n      role: 'user',\n      content: parts.length > 0 ? parts : '',\n    }\n  }\n\n  /**\n   * Normalizes message content to an array of ContentPart.\n   * Handles backward compatibility with string content.\n   */\n  private normalizeContent(\n    content: string | null | Array<ContentPart>,\n  ): Array<ContentPart> {\n    if (content === null) {\n      return []\n    }\n    if (typeof content === 'string') {\n      return [{ type: 'text', content: content }]\n    }\n    return content\n  }\n\n  /**\n   * Extracts text content from a content value that may be string, null, or ContentPart array.\n   */\n  private extractTextContent(\n    content: string | null | Array<ContentPart>,\n  ): string {\n    if (content === null) {\n      return ''\n    }\n    if (typeof content === 'string') {\n      return content\n    }\n    // It's an array of ContentPart\n    return content\n      .filter((p) => p.type === 'text')\n      .map((p) => p.content)\n      .join('')\n  }\n}\n\n/**\n * Creates a Grok text adapter with explicit API key.\n * Type resolution happens here at the call site.\n *\n * @param model - The model name (e.g., 'grok-3', 'grok-4')\n * @param apiKey - Your xAI API key\n * @param config - Optional additional configuration\n * @returns Configured Grok text adapter instance with resolved types\n *\n * @example\n * ```typescript\n * const adapter = createGrokText('grok-3', \"xai-...\");\n * // adapter has type-safe providerOptions for grok-3\n * ```\n */\nexport function createGrokText<\n  TModel extends (typeof GROK_CHAT_MODELS)[number],\n>(\n  model: TModel,\n  apiKey: string,\n  config?: Omit<GrokTextConfig, 'apiKey'>,\n): GrokTextAdapter<TModel> {\n  return new GrokTextAdapter({ apiKey, ...config }, model)\n}\n\n/**\n * Creates a Grok text adapter with automatic API key detection from environment variables.\n * Type resolution happens here at the call site.\n *\n * Looks for `XAI_API_KEY` in:\n * - `process.env` (Node.js)\n * - `window.env` (Browser with injected env)\n *\n * @param model - The model name (e.g., 'grok-3', 'grok-4')\n * @param config - Optional configuration (excluding apiKey which is auto-detected)\n * @returns Configured Grok text adapter instance with resolved types\n * @throws Error if XAI_API_KEY is not found in environment\n *\n * @example\n * ```typescript\n * // Automatically uses XAI_API_KEY from environment\n * const adapter = grokText('grok-3');\n *\n * const stream = chat({\n *   adapter,\n *   messages: [{ role: \"user\", content: \"Hello!\" }]\n * });\n * ```\n */\nexport function grokText<TModel extends (typeof GROK_CHAT_MODELS)[number]>(\n  model: TModel,\n  config?: Omit<GrokTextConfig, 'apiKey'>,\n): GrokTextAdapter<TModel> {\n  const apiKey = getGrokApiKeyFromEnv()\n  return createGrokText(model, apiKey, config)\n}\n"],"names":[],"mappings":";;;;;AAiDO,MAAM,wBAEH,gBAKR;AAAA,EAMA,YAAY,QAAwB,OAAe;AACjD,UAAM,CAAA,GAAI,KAAK;AANjB,SAAS,OAAO;AAChB,SAAS,OAAO;AAMd,SAAK,SAAS,iBAAiB,MAAM;AAAA,EACvC;AAAA,EAEA,OAAO,WACL,SAC4B;AAC5B,UAAM,gBAAgB,KAAK,qBAAqB,OAAO;AAEvD,QAAI;AACF,YAAM,SAAS,MAAM,KAAK,OAAO,KAAK,YAAY,OAAO;AAAA,QACvD,GAAG;AAAA,QACH,QAAQ;AAAA,MAAA,CACT;AAED,aAAO,KAAK,wBAAwB,QAAQ,OAAO;AAAA,IACrD,SAAS,OAAgB;AACvB,YAAM,MAAM;AACZ,cAAQ,MAAM,0DAA0D;AACxE,cAAQ,MAAM,sBAAsB,IAAI,OAAO;AAC/C,cAAQ,MAAM,oBAAoB,IAAI,KAAK;AAC3C,cAAQ,MAAM,mBAAmB,GAAG;AACpC,YAAM;AAAA,IACR;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAcA,MAAM,iBACJ,SAC0C;AAC1C,UAAM,EAAE,aAAa,aAAA,IAAiB;AACtC,UAAM,gBAAgB,KAAK,qBAAqB,WAAW;AAG3D,UAAM,aAAa;AAAA,MACjB;AAAA,MACA,aAAa,YAAY,CAAA;AAAA,IAAC;AAG5B,QAAI;AACF,YAAM,WAAW,MAAM,KAAK,OAAO,KAAK,YAAY,OAAO;AAAA,QACzD,GAAG;AAAA,QACH,QAAQ;AAAA,QACR,iBAAiB;AAAA,UACf,MAAM;AAAA,UACN,aAAa;AAAA,YACX,MAAM;AAAA,YACN,QAAQ;AAAA,YACR,QAAQ;AAAA,UAAA;AAAA,QACV;AAAA,MACF,CACD;AAGD,YAAM,UAAU,SAAS,QAAQ,CAAC,GAAG,QAAQ,WAAW;AAGxD,UAAI;AACJ,UAAI;AACF,iBAAS,KAAK,MAAM,OAAO;AAAA,MAC7B,QAAQ;AACN,cAAM,IAAI;AAAA,UACR,uDAAuD,QAAQ,MAAM,GAAG,GAAG,CAAC,GAAG,QAAQ,SAAS,MAAM,QAAQ,EAAE;AAAA,QAAA;AAAA,MAEpH;AAIA,YAAM,cAAc,0BAA0B,MAAM;AAEpD,aAAO;AAAA,QACL,MAAM;AAAA,QACN;AAAA,MAAA;AAAA,IAEJ,SAAS,OAAgB;AACvB,YAAM,MAAM;AACZ,cAAQ,MAAM,0DAA0D;AACxE,cAAQ,MAAM,sBAAsB,IAAI,OAAO;AAC/C,YAAM;AAAA,IACR;AAAA,EACF;AAAA,EAEA,OAAe,wBACb,QACA,SAC4B;AAC5B,QAAI,qBAAqB;AACzB,UAAM,YAAY,KAAK,IAAA;AACvB,QAAI,aAAa,WAAW,KAAK,IAAI;AAGrC,UAAM,0CAA0B,IAAA;AAShC,QAAI;AACF,uBAAiB,SAAS,QAAQ;AAChC,qBAAa,MAAM,MAAM;AACzB,cAAM,SAAS,MAAM,QAAQ,CAAC;AAE9B,YAAI,CAAC,OAAQ;AAEb,cAAM,QAAQ,OAAO;AACrB,cAAM,eAAe,MAAM;AAC3B,cAAM,iBAAiB,MAAM;AAG7B,YAAI,cAAc;AAChB,gCAAsB;AACtB,gBAAM;AAAA,YACJ,MAAM;AAAA,YACN,IAAI;AAAA,YACJ,OAAO,MAAM,SAAS,QAAQ;AAAA,YAC9B;AAAA,YACA,OAAO;AAAA,YACP,SAAS;AAAA,YACT,MAAM;AAAA,UAAA;AAAA,QAEV;AAGA,YAAI,gBAAgB;AAClB,qBAAW,iBAAiB,gBAAgB;AAC1C,kBAAM,QAAQ,cAAc;AAG5B,gBAAI,CAAC,oBAAoB,IAAI,KAAK,GAAG;AACnC,kCAAoB,IAAI,OAAO;AAAA,gBAC7B,IAAI,cAAc,MAAM;AAAA,gBACxB,MAAM,cAAc,UAAU,QAAQ;AAAA,gBACtC,WAAW;AAAA,cAAA,CACZ;AAAA,YACH;AAEA,kBAAM,WAAW,oBAAoB,IAAI,KAAK;AAG9C,gBAAI,cAAc,IAAI;AACpB,uBAAS,KAAK,cAAc;AAAA,YAC9B;AACA,gBAAI,cAAc,UAAU,MAAM;AAChC,uBAAS,OAAO,cAAc,SAAS;AAAA,YACzC;AACA,gBAAI,cAAc,UAAU,WAAW;AACrC,uBAAS,aAAa,cAAc,SAAS;AAAA,YAC/C;AAAA,UACF;AAAA,QACF;AAGA,YAAI,OAAO,eAAe;AAExB,cACE,OAAO,kBAAkB,gBACzB,oBAAoB,OAAO,GAC3B;AACA,uBAAW,CAAC,OAAO,QAAQ,KAAK,qBAAqB;AACnD,oBAAM;AAAA,gBACJ,MAAM;AAAA,gBACN,IAAI;AAAA,gBACJ,OAAO,MAAM,SAAS,QAAQ;AAAA,gBAC9B;AAAA,gBACA;AAAA,gBACA,UAAU;AAAA,kBACR,IAAI,SAAS;AAAA,kBACb,MAAM;AAAA,kBACN,UAAU;AAAA,oBACR,MAAM,SAAS;AAAA,oBACf,WAAW,SAAS;AAAA,kBAAA;AAAA,gBACtB;AAAA,cACF;AAAA,YAEJ;AAAA,UACF;AAEA,gBAAM;AAAA,YACJ,MAAM;AAAA,YACN,IAAI;AAAA,YACJ,OAAO,MAAM,SAAS,QAAQ;AAAA,YAC9B;AAAA,YACA,OAAO,MAAM,QACT;AAAA,cACE,cAAc,MAAM,MAAM,iBAAiB;AAAA,cAC3C,kBAAkB,MAAM,MAAM,qBAAqB;AAAA,cACnD,aAAa,MAAM,MAAM,gBAAgB;AAAA,YAAA,IAE3C;AAAA,YACJ,cACE,OAAO,kBAAkB,gBACzB,oBAAoB,OAAO,IACvB,eACA;AAAA,UAAA;AAAA,QAEV;AAAA,MACF;AAAA,IACF,SAAS,OAAgB;AACvB,YAAM,MAAM;AACZ,cAAQ,IAAI,2CAA2C,IAAI,OAAO;AAClE,YAAM;AAAA,QACJ,MAAM;AAAA,QACN,IAAI;AAAA,QACJ,OAAO,QAAQ;AAAA,QACf;AAAA,QACA,OAAO;AAAA,UACL,SAAS,IAAI,WAAW;AAAA,UACxB,MAAM,IAAI;AAAA,QAAA;AAAA,MACZ;AAAA,IAEJ;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKQ,qBACN,SACiE;AACjE,UAAM,eAAe,QAAQ;AAO7B,QAAI,cAAc;AAChB,kCAA4B;AAAA,QAC1B,GAAG;AAAA,QACH,OAAO,QAAQ;AAAA,MAAA,CAChB;AAAA,IACH;AAEA,UAAM,QAAQ,QAAQ,QAClB,6BAA6B,QAAQ,KAAK,IAC1C;AAGJ,UAAM,WACJ,CAAA;AAGF,QAAI,QAAQ,iBAAiB,QAAQ,cAAc,SAAS,GAAG;AAC7D,eAAS,KAAK;AAAA,QACZ,MAAM;AAAA,QACN,SAAS,QAAQ,cAAc,KAAK,IAAI;AAAA,MAAA,CACzC;AAAA,IACH;AAGA,eAAW,WAAW,QAAQ,UAAU;AACtC,eAAS,KAAK,KAAK,qBAAqB,OAAO,CAAC;AAAA,IAClD;AAEA,WAAO;AAAA,MACL,OAAO,QAAQ;AAAA,MACf;AAAA,MACA,aAAa,QAAQ;AAAA,MACrB,YAAY,QAAQ;AAAA,MACpB,OAAO,QAAQ;AAAA,MACf;AAAA,MACA,QAAQ;AAAA,MACR,gBAAgB,EAAE,eAAe,KAAA;AAAA,IAAK;AAAA,EAE1C;AAAA,EAEQ,qBACN,SACwD;AAExD,QAAI,QAAQ,SAAS,QAAQ;AAC3B,aAAO;AAAA,QACL,MAAM;AAAA,QACN,cAAc,QAAQ,cAAc;AAAA,QACpC,SACE,OAAO,QAAQ,YAAY,WACvB,QAAQ,UACR,KAAK,UAAU,QAAQ,OAAO;AAAA,MAAA;AAAA,IAExC;AAGA,QAAI,QAAQ,SAAS,aAAa;AAChC,YAAM,YAAY,QAAQ,WAAW,IAAI,CAAC,QAAQ;AAAA,QAChD,IAAI,GAAG;AAAA,QACP,MAAM;AAAA,QACN,UAAU;AAAA,UACR,MAAM,GAAG,SAAS;AAAA,UAClB,WACE,OAAO,GAAG,SAAS,cAAc,WAC7B,GAAG,SAAS,YACZ,KAAK,UAAU,GAAG,SAAS,SAAS;AAAA,QAAA;AAAA,MAC5C,EACA;AAEF,aAAO;AAAA,QACL,MAAM;AAAA,QACN,SAAS,KAAK,mBAAmB,QAAQ,OAAO;AAAA,QAChD,GAAI,aAAa,UAAU,SAAS,IAAI,EAAE,YAAY,cAAc,CAAA;AAAA,MAAC;AAAA,IAEzE;AAGA,UAAM,eAAe,KAAK,iBAAiB,QAAQ,OAAO;AAG1D,QAAI,aAAa,WAAW,KAAK,aAAa,CAAC,GAAG,SAAS,QAAQ;AACjE,aAAO;AAAA,QACL,MAAM;AAAA,QACN,SAAS,aAAa,CAAC,EAAE;AAAA,MAAA;AAAA,IAE7B;AAGA,UAAM,QACJ,CAAA;AACF,eAAW,QAAQ,cAAc;AAC/B,UAAI,KAAK,SAAS,QAAQ;AACxB,cAAM,KAAK,EAAE,MAAM,QAAQ,MAAM,KAAK,SAAS;AAAA,MACjD,WAAW,KAAK,SAAS,SAAS;AAChC,cAAM,gBAAgB,KAAK;AAC3B,cAAM,KAAK;AAAA,UACT,MAAM;AAAA,UACN,WAAW;AAAA,YACT,KAAK,KAAK,OAAO;AAAA,YACjB,QAAQ,eAAe,UAAU;AAAA,UAAA;AAAA,QACnC,CACD;AAAA,MACH;AAAA,IACF;AAEA,WAAO;AAAA,MACL,MAAM;AAAA,MACN,SAAS,MAAM,SAAS,IAAI,QAAQ;AAAA,IAAA;AAAA,EAExC;AAAA;AAAA;AAAA;AAAA;AAAA,EAMQ,iBACN,SACoB;AACpB,QAAI,YAAY,MAAM;AACpB,aAAO,CAAA;AAAA,IACT;AACA,QAAI,OAAO,YAAY,UAAU;AAC/B,aAAO,CAAC,EAAE,MAAM,QAAQ,SAAkB;AAAA,IAC5C;AACA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKQ,mBACN,SACQ;AACR,QAAI,YAAY,MAAM;AACpB,aAAO;AAAA,IACT;AACA,QAAI,OAAO,YAAY,UAAU;AAC/B,aAAO;AAAA,IACT;AAEA,WAAO,QACJ,OAAO,CAAC,MAAM,EAAE,SAAS,MAAM,EAC/B,IAAI,CAAC,MAAM,EAAE,OAAO,EACpB,KAAK,EAAE;AAAA,EACZ;AACF;AAiBO,SAAS,eAGd,OACA,QACA,QACyB;AACzB,SAAO,IAAI,gBAAgB,EAAE,QAAQ,GAAG,OAAA,GAAU,KAAK;AACzD;AA0BO,SAAS,SACd,OACA,QACyB;AACzB,QAAM,SAAS,qBAAA;AACf,SAAO,eAAe,OAAO,QAAQ,MAAM;AAC7C;"}
+{"version":3,"file":"text.js","sources":["../../../src/adapters/text.ts"],"sourcesContent":["import { BaseTextAdapter } from '@tanstack/ai/adapters'\nimport { validateTextProviderOptions } from '../text/text-provider-options'\nimport { convertToolsToProviderFormat } from '../tools'\nimport {\n  createGrokClient,\n  generateId,\n  getGrokApiKeyFromEnv,\n  makeGrokStructuredOutputCompatible,\n  transformNullsToUndefined,\n} from '../utils'\nimport type {\n  GROK_CHAT_MODELS,\n  ResolveInputModalities,\n  ResolveProviderOptions,\n} from '../model-meta'\nimport type {\n  StructuredOutputOptions,\n  StructuredOutputResult,\n} from '@tanstack/ai/adapters'\nimport type OpenAI_SDK from 'openai'\nimport type {\n  ContentPart,\n  ModelMessage,\n  StreamChunk,\n  TextOptions,\n} from '@tanstack/ai'\nimport type { InternalTextProviderOptions } from '../text/text-provider-options'\nimport type {\n  GrokImageMetadata,\n  GrokMessageMetadataByModality,\n} from '../message-types'\nimport type { GrokClientConfig } from '../utils'\n\n/**\n * Configuration for Grok text adapter\n */\nexport interface GrokTextConfig extends GrokClientConfig {}\n\n/**\n * Alias for TextProviderOptions for external use\n */\nexport type { ExternalTextProviderOptions as GrokTextProviderOptions } from '../text/text-provider-options'\n\n/**\n * Grok Text (Chat) Adapter\n *\n * Tree-shakeable adapter for Grok chat/text completion functionality.\n * Uses OpenAI-compatible Chat Completions API (not Responses API).\n */\nexport class GrokTextAdapter<\n  TModel extends (typeof GROK_CHAT_MODELS)[number],\n> extends BaseTextAdapter<\n  TModel,\n  ResolveProviderOptions<TModel>,\n  ResolveInputModalities<TModel>,\n  GrokMessageMetadataByModality\n> {\n  readonly kind = 'text' as const\n  readonly name = 'grok' as const\n\n  private client: OpenAI_SDK\n\n  constructor(config: GrokTextConfig, model: TModel) {\n    super({}, model)\n    this.client = createGrokClient(config)\n  }\n\n  async *chatStream(\n    options: TextOptions<ResolveProviderOptions<TModel>>,\n  ): AsyncIterable<StreamChunk> {\n    const requestParams = this.mapTextOptionsToGrok(options)\n    const timestamp = Date.now()\n\n    // AG-UI lifecycle tracking (mutable state object for ESLint compatibility)\n    const aguiState = {\n      runId: generateId(this.name),\n      messageId: generateId(this.name),\n      timestamp,\n      hasEmittedRunStarted: false,\n    }\n\n    try {\n      const stream = await this.client.chat.completions.create({\n        ...requestParams,\n        stream: true,\n      })\n\n      yield* this.processGrokStreamChunks(stream, options, aguiState)\n    } catch (error: unknown) {\n      const err = error as Error & { code?: string }\n\n      // Emit RUN_STARTED if not yet emitted\n      if (!aguiState.hasEmittedRunStarted) {\n        aguiState.hasEmittedRunStarted = true\n        yield {\n          type: 'RUN_STARTED',\n          runId: aguiState.runId,\n          model: options.model,\n          timestamp,\n        }\n      }\n\n      // Emit AG-UI RUN_ERROR\n      yield {\n        type: 'RUN_ERROR',\n        runId: aguiState.runId,\n        model: options.model,\n        timestamp,\n        error: {\n          message: err.message || 'Unknown error',\n          code: err.code,\n        },\n      }\n\n      console.error('>>> chatStream: Fatal error during response creation <<<')\n      console.error('>>> Error message:', err.message)\n      console.error('>>> Error stack:', err.stack)\n      console.error('>>> Full error:', err)\n    }\n  }\n\n  /**\n   * Generate structured output using Grok's JSON Schema response format.\n   * Uses stream: false to get the complete response in one call.\n   *\n   * Grok has strict requirements for structured output (via OpenAI-compatible API):\n   * - All properties must be in the `required` array\n   * - Optional fields should have null added to their type union\n   * - additionalProperties must be false for all objects\n   *\n   * The outputSchema is already JSON Schema (converted in the ai layer).\n   * We apply Grok-specific transformations for structured output compatibility.\n   */\n  async structuredOutput(\n    options: StructuredOutputOptions<ResolveProviderOptions<TModel>>,\n  ): Promise<StructuredOutputResult<unknown>> {\n    const { chatOptions, outputSchema } = options\n    const requestParams = this.mapTextOptionsToGrok(chatOptions)\n\n    // Apply Grok-specific transformations for structured output compatibility\n    const jsonSchema = makeGrokStructuredOutputCompatible(\n      outputSchema,\n      outputSchema.required || [],\n    )\n\n    try {\n      const response = await this.client.chat.completions.create({\n        ...requestParams,\n        stream: false,\n        response_format: {\n          type: 'json_schema',\n          json_schema: {\n            name: 'structured_output',\n            schema: jsonSchema,\n            strict: true,\n          },\n        },\n      })\n\n      // Extract text content from the response\n      const rawText = response.choices[0]?.message.content || ''\n\n      // Parse the JSON response\n      let parsed: unknown\n      try {\n        parsed = JSON.parse(rawText)\n      } catch {\n        throw new Error(\n          `Failed to parse structured output as JSON. Content: ${rawText.slice(0, 200)}${rawText.length > 200 ? '...' : ''}`,\n        )\n      }\n\n      // Transform null values to undefined to match original Zod schema expectations\n      // Grok returns null for optional fields we made nullable in the schema\n      const transformed = transformNullsToUndefined(parsed)\n\n      return {\n        data: transformed,\n        rawText,\n      }\n    } catch (error: unknown) {\n      const err = error as Error\n      console.error('>>> structuredOutput: Error during response creation <<<')\n      console.error('>>> Error message:', err.message)\n      throw error\n    }\n  }\n\n  private async *processGrokStreamChunks(\n    stream: AsyncIterable<OpenAI_SDK.Chat.Completions.ChatCompletionChunk>,\n    options: TextOptions,\n    aguiState: {\n      runId: string\n      messageId: string\n      timestamp: number\n      hasEmittedRunStarted: boolean\n    },\n  ): AsyncIterable<StreamChunk> {\n    let accumulatedContent = ''\n    const timestamp = aguiState.timestamp\n    let hasEmittedTextMessageStart = false\n\n    // Track tool calls being streamed (arguments come in chunks)\n    const toolCallsInProgress = new Map<\n      number,\n      {\n        id: string\n        name: string\n        arguments: string\n        started: boolean // Track if TOOL_CALL_START has been emitted\n      }\n    >()\n\n    try {\n      for await (const chunk of stream) {\n        const choice = chunk.choices[0]\n\n        if (!choice) continue\n\n        // Emit RUN_STARTED on first chunk\n        if (!aguiState.hasEmittedRunStarted) {\n          aguiState.hasEmittedRunStarted = true\n          yield {\n            type: 'RUN_STARTED',\n            runId: aguiState.runId,\n            model: chunk.model || options.model,\n            timestamp,\n          }\n        }\n\n        const delta = choice.delta\n        const deltaContent = delta.content\n        const deltaToolCalls = delta.tool_calls\n\n        // Handle content delta\n        if (deltaContent) {\n          // Emit TEXT_MESSAGE_START on first text content\n          if (!hasEmittedTextMessageStart) {\n            hasEmittedTextMessageStart = true\n            yield {\n              type: 'TEXT_MESSAGE_START',\n              messageId: aguiState.messageId,\n              model: chunk.model || options.model,\n              timestamp,\n              role: 'assistant',\n            }\n          }\n\n          accumulatedContent += deltaContent\n\n          // Emit AG-UI TEXT_MESSAGE_CONTENT\n          yield {\n            type: 'TEXT_MESSAGE_CONTENT',\n            messageId: aguiState.messageId,\n            model: chunk.model || options.model,\n            timestamp,\n            delta: deltaContent,\n            content: accumulatedContent,\n          }\n        }\n\n        // Handle tool calls - they come in as deltas\n        if (deltaToolCalls) {\n          for (const toolCallDelta of deltaToolCalls) {\n            const index = toolCallDelta.index\n\n            // Initialize or update the tool call in progress\n            if (!toolCallsInProgress.has(index)) {\n              toolCallsInProgress.set(index, {\n                id: toolCallDelta.id || '',\n                name: toolCallDelta.function?.name || '',\n                arguments: '',\n                started: false,\n              })\n            }\n\n            const toolCall = toolCallsInProgress.get(index)!\n\n            // Update with any new data from the delta\n            if (toolCallDelta.id) {\n              toolCall.id = toolCallDelta.id\n            }\n            if (toolCallDelta.function?.name) {\n              toolCall.name = toolCallDelta.function.name\n            }\n            if (toolCallDelta.function?.arguments) {\n              toolCall.arguments += toolCallDelta.function.arguments\n            }\n\n            // Emit TOOL_CALL_START when we have id and name\n            if (toolCall.id && toolCall.name && !toolCall.started) {\n              toolCall.started = true\n              yield {\n                type: 'TOOL_CALL_START',\n                toolCallId: toolCall.id,\n                toolName: toolCall.name,\n                model: chunk.model || options.model,\n                timestamp,\n                index,\n              }\n            }\n\n            // Emit TOOL_CALL_ARGS for argument deltas\n            if (toolCallDelta.function?.arguments && toolCall.started) {\n              yield {\n                type: 'TOOL_CALL_ARGS',\n                toolCallId: toolCall.id,\n                model: chunk.model || options.model,\n                timestamp,\n                delta: toolCallDelta.function.arguments,\n              }\n            }\n          }\n        }\n\n        // Handle finish reason\n        if (choice.finish_reason) {\n          // Emit all completed tool calls\n          if (\n            choice.finish_reason === 'tool_calls' ||\n            toolCallsInProgress.size > 0\n          ) {\n            for (const [, toolCall] of toolCallsInProgress) {\n              // Parse arguments for TOOL_CALL_END\n              let parsedInput: unknown = {}\n              try {\n                parsedInput = toolCall.arguments\n                  ? JSON.parse(toolCall.arguments)\n                  : {}\n              } catch {\n                parsedInput = {}\n              }\n\n              // Emit AG-UI TOOL_CALL_END\n              yield {\n                type: 'TOOL_CALL_END',\n                toolCallId: toolCall.id,\n                toolName: toolCall.name,\n                model: chunk.model || options.model,\n                timestamp,\n                input: parsedInput,\n              }\n            }\n          }\n\n          const computedFinishReason =\n            choice.finish_reason === 'tool_calls' ||\n            toolCallsInProgress.size > 0\n              ? 'tool_calls'\n              : 'stop'\n\n          // Emit TEXT_MESSAGE_END if we had text content\n          if (hasEmittedTextMessageStart) {\n            yield {\n              type: 'TEXT_MESSAGE_END',\n              messageId: aguiState.messageId,\n              model: chunk.model || options.model,\n              timestamp,\n            }\n          }\n\n          // Emit AG-UI RUN_FINISHED\n          yield {\n            type: 'RUN_FINISHED',\n            runId: aguiState.runId,\n            model: chunk.model || options.model,\n            timestamp,\n            usage: chunk.usage\n              ? {\n                  promptTokens: chunk.usage.prompt_tokens || 0,\n                  completionTokens: chunk.usage.completion_tokens || 0,\n                  totalTokens: chunk.usage.total_tokens || 0,\n                }\n              : undefined,\n            finishReason: computedFinishReason,\n          }\n        }\n      }\n    } catch (error: unknown) {\n      const err = error as Error & { code?: string }\n      console.log('[Grok Adapter] Stream ended with error:', err.message)\n\n      // Emit AG-UI RUN_ERROR\n      yield {\n        type: 'RUN_ERROR',\n        runId: aguiState.runId,\n        model: options.model,\n        timestamp,\n        error: {\n          message: err.message || 'Unknown error occurred',\n          code: err.code,\n        },\n      }\n    }\n  }\n\n  /**\n   * Maps common options to Grok-specific Chat Completions format\n   */\n  private mapTextOptionsToGrok(\n    options: TextOptions,\n  ): OpenAI_SDK.Chat.Completions.ChatCompletionCreateParamsStreaming {\n    const modelOptions = options.modelOptions as\n      | Omit<\n          InternalTextProviderOptions,\n          'max_tokens' | 'tools' | 'temperature' | 'input' | 'top_p'\n        >\n      | undefined\n\n    if (modelOptions) {\n      validateTextProviderOptions({\n        ...modelOptions,\n        model: options.model,\n      })\n    }\n\n    const tools = options.tools\n      ? convertToolsToProviderFormat(options.tools)\n      : undefined\n\n    // Build messages array with system prompts\n    const messages: Array<OpenAI_SDK.Chat.Completions.ChatCompletionMessageParam> =\n      []\n\n    // Add system prompts first\n    if (options.systemPrompts && options.systemPrompts.length > 0) {\n      messages.push({\n        role: 'system',\n        content: options.systemPrompts.join('\\n'),\n      })\n    }\n\n    // Convert messages\n    for (const message of options.messages) {\n      messages.push(this.convertMessageToGrok(message))\n    }\n\n    return {\n      model: options.model,\n      messages,\n      temperature: options.temperature,\n      max_tokens: options.maxTokens,\n      top_p: options.topP,\n      tools: tools as Array<OpenAI_SDK.Chat.Completions.ChatCompletionTool>,\n      stream: true,\n      stream_options: { include_usage: true },\n    }\n  }\n\n  private convertMessageToGrok(\n    message: ModelMessage,\n  ): OpenAI_SDK.Chat.Completions.ChatCompletionMessageParam {\n    // Handle tool messages\n    if (message.role === 'tool') {\n      return {\n        role: 'tool',\n        tool_call_id: message.toolCallId || '',\n        content:\n          typeof message.content === 'string'\n            ? message.content\n            : JSON.stringify(message.content),\n      }\n    }\n\n    // Handle assistant messages\n    if (message.role === 'assistant') {\n      const toolCalls = message.toolCalls?.map((tc) => ({\n        id: tc.id,\n        type: 'function' as const,\n        function: {\n          name: tc.function.name,\n          arguments:\n            typeof tc.function.arguments === 'string'\n              ? tc.function.arguments\n              : JSON.stringify(tc.function.arguments),\n        },\n      }))\n\n      return {\n        role: 'assistant',\n        content: this.extractTextContent(message.content),\n        ...(toolCalls && toolCalls.length > 0 ? { tool_calls: toolCalls } : {}),\n      }\n    }\n\n    // Handle user messages - support multimodal content\n    const contentParts = this.normalizeContent(message.content)\n\n    // If only text, use simple string format\n    if (contentParts.length === 1 && contentParts[0]?.type === 'text') {\n      return {\n        role: 'user',\n        content: contentParts[0].content,\n      }\n    }\n\n    // Otherwise, use array format for multimodal\n    const parts: Array<OpenAI_SDK.Chat.Completions.ChatCompletionContentPart> =\n      []\n    for (const part of contentParts) {\n      if (part.type === 'text') {\n        parts.push({ type: 'text', text: part.content })\n      } else if (part.type === 'image') {\n        const imageMetadata = part.metadata as GrokImageMetadata | undefined\n        // For base64 data, construct a data URI using the mimeType from source\n        const imageValue = part.source.value\n        const imageUrl =\n          part.source.type === 'data' && !imageValue.startsWith('data:')\n            ? `data:${part.source.mimeType};base64,${imageValue}`\n            : imageValue\n        parts.push({\n          type: 'image_url',\n          image_url: {\n            url: imageUrl,\n            detail: imageMetadata?.detail || 'auto',\n          },\n        })\n      }\n    }\n\n    return {\n      role: 'user',\n      content: parts.length > 0 ? parts : '',\n    }\n  }\n\n  /**\n   * Normalizes message content to an array of ContentPart.\n   * Handles backward compatibility with string content.\n   */\n  private normalizeContent(\n    content: string | null | Array<ContentPart>,\n  ): Array<ContentPart> {\n    if (content === null) {\n      return []\n    }\n    if (typeof content === 'string') {\n      return [{ type: 'text', content: content }]\n    }\n    return content\n  }\n\n  /**\n   * Extracts text content from a content value that may be string, null, or ContentPart array.\n   */\n  private extractTextContent(\n    content: string | null | Array<ContentPart>,\n  ): string {\n    if (content === null) {\n      return ''\n    }\n    if (typeof content === 'string') {\n      return content\n    }\n    // It's an array of ContentPart\n    return content\n      .filter((p) => p.type === 'text')\n      .map((p) => p.content)\n      .join('')\n  }\n}\n\n/**\n * Creates a Grok text adapter with explicit API key.\n * Type resolution happens here at the call site.\n *\n * @param model - The model name (e.g., 'grok-3', 'grok-4')\n * @param apiKey - Your xAI API key\n * @param config - Optional additional configuration\n * @returns Configured Grok text adapter instance with resolved types\n *\n * @example\n * ```typescript\n * const adapter = createGrokText('grok-3', \"xai-...\");\n * // adapter has type-safe providerOptions for grok-3\n * ```\n */\nexport function createGrokText<\n  TModel extends (typeof GROK_CHAT_MODELS)[number],\n>(\n  model: TModel,\n  apiKey: string,\n  config?: Omit<GrokTextConfig, 'apiKey'>,\n): GrokTextAdapter<TModel> {\n  return new GrokTextAdapter({ apiKey, ...config }, model)\n}\n\n/**\n * Creates a Grok text adapter with automatic API key detection from environment variables.\n * Type resolution happens here at the call site.\n *\n * Looks for `XAI_API_KEY` in:\n * - `process.env` (Node.js)\n * - `window.env` (Browser with injected env)\n *\n * @param model - The model name (e.g., 'grok-3', 'grok-4')\n * @param config - Optional configuration (excluding apiKey which is auto-detected)\n * @returns Configured Grok text adapter instance with resolved types\n * @throws Error if XAI_API_KEY is not found in environment\n *\n * @example\n * ```typescript\n * // Automatically uses XAI_API_KEY from environment\n * const adapter = grokText('grok-3');\n *\n * const stream = chat({\n *   adapter,\n *   messages: [{ role: \"user\", content: \"Hello!\" }]\n * });\n * ```\n */\nexport function grokText<TModel extends (typeof GROK_CHAT_MODELS)[number]>(\n  model: TModel,\n  config?: Omit<GrokTextConfig, 'apiKey'>,\n): GrokTextAdapter<TModel> {\n  const apiKey = getGrokApiKeyFromEnv()\n  return createGrokText(model, apiKey, config)\n}\n"],"names":[],"mappings":";;;;;AAiDO,MAAM,wBAEH,gBAKR;AAAA,EAMA,YAAY,QAAwB,OAAe;AACjD,UAAM,CAAA,GAAI,KAAK;AANjB,SAAS,OAAO;AAChB,SAAS,OAAO;AAMd,SAAK,SAAS,iBAAiB,MAAM;AAAA,EACvC;AAAA,EAEA,OAAO,WACL,SAC4B;AAC5B,UAAM,gBAAgB,KAAK,qBAAqB,OAAO;AACvD,UAAM,YAAY,KAAK,IAAA;AAGvB,UAAM,YAAY;AAAA,MAChB,OAAO,WAAW,KAAK,IAAI;AAAA,MAC3B,WAAW,WAAW,KAAK,IAAI;AAAA,MAC/B;AAAA,MACA,sBAAsB;AAAA,IAAA;AAGxB,QAAI;AACF,YAAM,SAAS,MAAM,KAAK,OAAO,KAAK,YAAY,OAAO;AAAA,QACvD,GAAG;AAAA,QACH,QAAQ;AAAA,MAAA,CACT;AAED,aAAO,KAAK,wBAAwB,QAAQ,SAAS,SAAS;AAAA,IAChE,SAAS,OAAgB;AACvB,YAAM,MAAM;AAGZ,UAAI,CAAC,UAAU,sBAAsB;AACnC,kBAAU,uBAAuB;AACjC,cAAM;AAAA,UACJ,MAAM;AAAA,UACN,OAAO,UAAU;AAAA,UACjB,OAAO,QAAQ;AAAA,UACf;AAAA,QAAA;AAAA,MAEJ;AAGA,YAAM;AAAA,QACJ,MAAM;AAAA,QACN,OAAO,UAAU;AAAA,QACjB,OAAO,QAAQ;AAAA,QACf;AAAA,QACA,OAAO;AAAA,UACL,SAAS,IAAI,WAAW;AAAA,UACxB,MAAM,IAAI;AAAA,QAAA;AAAA,MACZ;AAGF,cAAQ,MAAM,0DAA0D;AACxE,cAAQ,MAAM,sBAAsB,IAAI,OAAO;AAC/C,cAAQ,MAAM,oBAAoB,IAAI,KAAK;AAC3C,cAAQ,MAAM,mBAAmB,GAAG;AAAA,IACtC;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAcA,MAAM,iBACJ,SAC0C;AAC1C,UAAM,EAAE,aAAa,aAAA,IAAiB;AACtC,UAAM,gBAAgB,KAAK,qBAAqB,WAAW;AAG3D,UAAM,aAAa;AAAA,MACjB;AAAA,MACA,aAAa,YAAY,CAAA;AAAA,IAAC;AAG5B,QAAI;AACF,YAAM,WAAW,MAAM,KAAK,OAAO,KAAK,YAAY,OAAO;AAAA,QACzD,GAAG;AAAA,QACH,QAAQ;AAAA,QACR,iBAAiB;AAAA,UACf,MAAM;AAAA,UACN,aAAa;AAAA,YACX,MAAM;AAAA,YACN,QAAQ;AAAA,YACR,QAAQ;AAAA,UAAA;AAAA,QACV;AAAA,MACF,CACD;AAGD,YAAM,UAAU,SAAS,QAAQ,CAAC,GAAG,QAAQ,WAAW;AAGxD,UAAI;AACJ,UAAI;AACF,iBAAS,KAAK,MAAM,OAAO;AAAA,MAC7B,QAAQ;AACN,cAAM,IAAI;AAAA,UACR,uDAAuD,QAAQ,MAAM,GAAG,GAAG,CAAC,GAAG,QAAQ,SAAS,MAAM,QAAQ,EAAE;AAAA,QAAA;AAAA,MAEpH;AAIA,YAAM,cAAc,0BAA0B,MAAM;AAEpD,aAAO;AAAA,QACL,MAAM;AAAA,QACN;AAAA,MAAA;AAAA,IAEJ,SAAS,OAAgB;AACvB,YAAM,MAAM;AACZ,cAAQ,MAAM,0DAA0D;AACxE,cAAQ,MAAM,sBAAsB,IAAI,OAAO;AAC/C,YAAM;AAAA,IACR;AAAA,EACF;AAAA,EAEA,OAAe,wBACb,QACA,SACA,WAM4B;AAC5B,QAAI,qBAAqB;AACzB,UAAM,YAAY,UAAU;AAC5B,QAAI,6BAA6B;AAGjC,UAAM,0CAA0B,IAAA;AAUhC,QAAI;AACF,uBAAiB,SAAS,QAAQ;AAChC,cAAM,SAAS,MAAM,QAAQ,CAAC;AAE9B,YAAI,CAAC,OAAQ;AAGb,YAAI,CAAC,UAAU,sBAAsB;AACnC,oBAAU,uBAAuB;AACjC,gBAAM;AAAA,YACJ,MAAM;AAAA,YACN,OAAO,UAAU;AAAA,YACjB,OAAO,MAAM,SAAS,QAAQ;AAAA,YAC9B;AAAA,UAAA;AAAA,QAEJ;AAEA,cAAM,QAAQ,OAAO;AACrB,cAAM,eAAe,MAAM;AAC3B,cAAM,iBAAiB,MAAM;AAG7B,YAAI,cAAc;AAEhB,cAAI,CAAC,4BAA4B;AAC/B,yCAA6B;AAC7B,kBAAM;AAAA,cACJ,MAAM;AAAA,cACN,WAAW,UAAU;AAAA,cACrB,OAAO,MAAM,SAAS,QAAQ;AAAA,cAC9B;AAAA,cACA,MAAM;AAAA,YAAA;AAAA,UAEV;AAEA,gCAAsB;AAGtB,gBAAM;AAAA,YACJ,MAAM;AAAA,YACN,WAAW,UAAU;AAAA,YACrB,OAAO,MAAM,SAAS,QAAQ;AAAA,YAC9B;AAAA,YACA,OAAO;AAAA,YACP,SAAS;AAAA,UAAA;AAAA,QAEb;AAGA,YAAI,gBAAgB;AAClB,qBAAW,iBAAiB,gBAAgB;AAC1C,kBAAM,QAAQ,cAAc;AAG5B,gBAAI,CAAC,oBAAoB,IAAI,KAAK,GAAG;AACnC,kCAAoB,IAAI,OAAO;AAAA,gBAC7B,IAAI,cAAc,MAAM;AAAA,gBACxB,MAAM,cAAc,UAAU,QAAQ;AAAA,gBACtC,WAAW;AAAA,gBACX,SAAS;AAAA,cAAA,CACV;AAAA,YACH;AAEA,kBAAM,WAAW,oBAAoB,IAAI,KAAK;AAG9C,gBAAI,cAAc,IAAI;AACpB,uBAAS,KAAK,cAAc;AAAA,YAC9B;AACA,gBAAI,cAAc,UAAU,MAAM;AAChC,uBAAS,OAAO,cAAc,SAAS;AAAA,YACzC;AACA,gBAAI,cAAc,UAAU,WAAW;AACrC,uBAAS,aAAa,cAAc,SAAS;AAAA,YAC/C;AAGA,gBAAI,SAAS,MAAM,SAAS,QAAQ,CAAC,SAAS,SAAS;AACrD,uBAAS,UAAU;AACnB,oBAAM;AAAA,gBACJ,MAAM;AAAA,gBACN,YAAY,SAAS;AAAA,gBACrB,UAAU,SAAS;AAAA,gBACnB,OAAO,MAAM,SAAS,QAAQ;AAAA,gBAC9B;AAAA,gBACA;AAAA,cAAA;AAAA,YAEJ;AAGA,gBAAI,cAAc,UAAU,aAAa,SAAS,SAAS;AACzD,oBAAM;AAAA,gBACJ,MAAM;AAAA,gBACN,YAAY,SAAS;AAAA,gBACrB,OAAO,MAAM,SAAS,QAAQ;AAAA,gBAC9B;AAAA,gBACA,OAAO,cAAc,SAAS;AAAA,cAAA;AAAA,YAElC;AAAA,UACF;AAAA,QACF;AAGA,YAAI,OAAO,eAAe;AAExB,cACE,OAAO,kBAAkB,gBACzB,oBAAoB,OAAO,GAC3B;AACA,uBAAW,CAAA,EAAG,QAAQ,KAAK,qBAAqB;AAE9C,kBAAI,cAAuB,CAAA;AAC3B,kBAAI;AACF,8BAAc,SAAS,YACnB,KAAK,MAAM,SAAS,SAAS,IAC7B,CAAA;AAAA,cACN,QAAQ;AACN,8BAAc,CAAA;AAAA,cAChB;AAGA,oBAAM;AAAA,gBACJ,MAAM;AAAA,gBACN,YAAY,SAAS;AAAA,gBACrB,UAAU,SAAS;AAAA,gBACnB,OAAO,MAAM,SAAS,QAAQ;AAAA,gBAC9B;AAAA,gBACA,OAAO;AAAA,cAAA;AAAA,YAEX;AAAA,UACF;AAEA,gBAAM,uBACJ,OAAO,kBAAkB,gBACzB,oBAAoB,OAAO,IACvB,eACA;AAGN,cAAI,4BAA4B;AAC9B,kBAAM;AAAA,cACJ,MAAM;AAAA,cACN,WAAW,UAAU;AAAA,cACrB,OAAO,MAAM,SAAS,QAAQ;AAAA,cAC9B;AAAA,YAAA;AAAA,UAEJ;AAGA,gBAAM;AAAA,YACJ,MAAM;AAAA,YACN,OAAO,UAAU;AAAA,YACjB,OAAO,MAAM,SAAS,QAAQ;AAAA,YAC9B;AAAA,YACA,OAAO,MAAM,QACT;AAAA,cACE,cAAc,MAAM,MAAM,iBAAiB;AAAA,cAC3C,kBAAkB,MAAM,MAAM,qBAAqB;AAAA,cACnD,aAAa,MAAM,MAAM,gBAAgB;AAAA,YAAA,IAE3C;AAAA,YACJ,cAAc;AAAA,UAAA;AAAA,QAElB;AAAA,MACF;AAAA,IACF,SAAS,OAAgB;AACvB,YAAM,MAAM;AACZ,cAAQ,IAAI,2CAA2C,IAAI,OAAO;AAGlE,YAAM;AAAA,QACJ,MAAM;AAAA,QACN,OAAO,UAAU;AAAA,QACjB,OAAO,QAAQ;AAAA,QACf;AAAA,QACA,OAAO;AAAA,UACL,SAAS,IAAI,WAAW;AAAA,UACxB,MAAM,IAAI;AAAA,QAAA;AAAA,MACZ;AAAA,IAEJ;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKQ,qBACN,SACiE;AACjE,UAAM,eAAe,QAAQ;AAO7B,QAAI,cAAc;AAChB,kCAA4B;AAAA,QAC1B,GAAG;AAAA,QACH,OAAO,QAAQ;AAAA,MAAA,CAChB;AAAA,IACH;AAEA,UAAM,QAAQ,QAAQ,QAClB,6BAA6B,QAAQ,KAAK,IAC1C;AAGJ,UAAM,WACJ,CAAA;AAGF,QAAI,QAAQ,iBAAiB,QAAQ,cAAc,SAAS,GAAG;AAC7D,eAAS,KAAK;AAAA,QACZ,MAAM;AAAA,QACN,SAAS,QAAQ,cAAc,KAAK,IAAI;AAAA,MAAA,CACzC;AAAA,IACH;AAGA,eAAW,WAAW,QAAQ,UAAU;AACtC,eAAS,KAAK,KAAK,qBAAqB,OAAO,CAAC;AAAA,IAClD;AAEA,WAAO;AAAA,MACL,OAAO,QAAQ;AAAA,MACf;AAAA,MACA,aAAa,QAAQ;AAAA,MACrB,YAAY,QAAQ;AAAA,MACpB,OAAO,QAAQ;AAAA,MACf;AAAA,MACA,QAAQ;AAAA,MACR,gBAAgB,EAAE,eAAe,KAAA;AAAA,IAAK;AAAA,EAE1C;AAAA,EAEQ,qBACN,SACwD;AAExD,QAAI,QAAQ,SAAS,QAAQ;AAC3B,aAAO;AAAA,QACL,MAAM;AAAA,QACN,cAAc,QAAQ,cAAc;AAAA,QACpC,SACE,OAAO,QAAQ,YAAY,WACvB,QAAQ,UACR,KAAK,UAAU,QAAQ,OAAO;AAAA,MAAA;AAAA,IAExC;AAGA,QAAI,QAAQ,SAAS,aAAa;AAChC,YAAM,YAAY,QAAQ,WAAW,IAAI,CAAC,QAAQ;AAAA,QAChD,IAAI,GAAG;AAAA,QACP,MAAM;AAAA,QACN,UAAU;AAAA,UACR,MAAM,GAAG,SAAS;AAAA,UAClB,WACE,OAAO,GAAG,SAAS,cAAc,WAC7B,GAAG,SAAS,YACZ,KAAK,UAAU,GAAG,SAAS,SAAS;AAAA,QAAA;AAAA,MAC5C,EACA;AAEF,aAAO;AAAA,QACL,MAAM;AAAA,QACN,SAAS,KAAK,mBAAmB,QAAQ,OAAO;AAAA,QAChD,GAAI,aAAa,UAAU,SAAS,IAAI,EAAE,YAAY,cAAc,CAAA;AAAA,MAAC;AAAA,IAEzE;AAGA,UAAM,eAAe,KAAK,iBAAiB,QAAQ,OAAO;AAG1D,QAAI,aAAa,WAAW,KAAK,aAAa,CAAC,GAAG,SAAS,QAAQ;AACjE,aAAO;AAAA,QACL,MAAM;AAAA,QACN,SAAS,aAAa,CAAC,EAAE;AAAA,MAAA;AAAA,IAE7B;AAGA,UAAM,QACJ,CAAA;AACF,eAAW,QAAQ,cAAc;AAC/B,UAAI,KAAK,SAAS,QAAQ;AACxB,cAAM,KAAK,EAAE,MAAM,QAAQ,MAAM,KAAK,SAAS;AAAA,MACjD,WAAW,KAAK,SAAS,SAAS;AAChC,cAAM,gBAAgB,KAAK;AAE3B,cAAM,aAAa,KAAK,OAAO;AAC/B,cAAM,WACJ,KAAK,OAAO,SAAS,UAAU,CAAC,WAAW,WAAW,OAAO,IACzD,QAAQ,KAAK,OAAO,QAAQ,WAAW,UAAU,KACjD;AACN,cAAM,KAAK;AAAA,UACT,MAAM;AAAA,UACN,WAAW;AAAA,YACT,KAAK;AAAA,YACL,QAAQ,eAAe,UAAU;AAAA,UAAA;AAAA,QACnC,CACD;AAAA,MACH;AAAA,IACF;AAEA,WAAO;AAAA,MACL,MAAM;AAAA,MACN,SAAS,MAAM,SAAS,IAAI,QAAQ;AAAA,IAAA;AAAA,EAExC;AAAA;AAAA;AAAA;AAAA;AAAA,EAMQ,iBACN,SACoB;AACpB,QAAI,YAAY,MAAM;AACpB,aAAO,CAAA;AAAA,IACT;AACA,QAAI,OAAO,YAAY,UAAU;AAC/B,aAAO,CAAC,EAAE,MAAM,QAAQ,SAAkB;AAAA,IAC5C;AACA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKQ,mBACN,SACQ;AACR,QAAI,YAAY,MAAM;AACpB,aAAO;AAAA,IACT;AACA,QAAI,OAAO,YAAY,UAAU;AAC/B,aAAO;AAAA,IACT;AAEA,WAAO,QACJ,OAAO,CAAC,MAAM,EAAE,SAAS,MAAM,EAC/B,IAAI,CAAC,MAAM,EAAE,OAAO,EACpB,KAAK,EAAE;AAAA,EACZ;AACF;AAiBO,SAAS,eAGd,OACA,QACA,QACyB;AACzB,SAAO,IAAI,gBAAgB,EAAE,QAAQ,GAAG,OAAA,GAAU,KAAK;AACzD;AA0BO,SAAS,SACd,OACA,QACyB;AACzB,QAAM,SAAS,qBAAA;AACf,SAAO,eAAe,OAAO,QAAQ,MAAM;AAC7C;"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tanstack/ai-grok",
-  "version": "0.1.0",
+  "version": "0.4.0",
   "description": "Grok (xAI) adapter for TanStack AI",
   "author": "",
   "license": "MIT",
@@ -38,7 +38,7 @@
   },
   "peerDependencies": {
     "zod": "^4.0.0",
-    "@tanstack/ai": "^0.2.1"
+    "@tanstack/ai": "^0.4.0"
   },
   "scripts": {
     "build": "vite build",

package/src/adapters/summarize.ts

@@ -51,7 +51,7 @@ export class GrokSummarizeAdapter<
 
     // Use the text adapter's streaming and collect the result
     let summary = ''
-    let id = ''
+    const id = ''
     let model = options.model
     let usage = { promptTokens: 0, completionTokens: 0, totalTokens: 0 }
 
@@ -62,13 +62,20 @@ export class GrokSummarizeAdapter<
       maxTokens: options.maxLength,
       temperature: 0.3,
     })) {
-      if (chunk.type === 'content') {
-        summary = chunk.content
-        id = chunk.id
-        model = chunk.model
+      // AG-UI TEXT_MESSAGE_CONTENT event
+      if (chunk.type === 'TEXT_MESSAGE_CONTENT') {
+        if (chunk.content) {
+          summary = chunk.content
+        } else {
+          summary += chunk.delta
+        }
+        model = chunk.model || model
       }
-      if (chunk.type === 'done' && chunk.usage) {
-        usage = chunk.usage
+      // AG-UI RUN_FINISHED event
+      if (chunk.type === 'RUN_FINISHED') {
+        if (chunk.usage) {
+          usage = chunk.usage
+        }
       }
     }
 

package/src/adapters/text.ts

@@ -69,6 +69,15 @@ export class GrokTextAdapter<
     options: TextOptions<ResolveProviderOptions<TModel>>,
   ): AsyncIterable<StreamChunk> {
     const requestParams = this.mapTextOptionsToGrok(options)
+    const timestamp = Date.now()
+
+    // AG-UI lifecycle tracking (mutable state object for ESLint compatibility)
+    const aguiState = {
+      runId: generateId(this.name),
+      messageId: generateId(this.name),
+      timestamp,
+      hasEmittedRunStarted: false,
+    }
 
     try {
       const stream = await this.client.chat.completions.create({
@@ -76,14 +85,37 @@ export class GrokTextAdapter<
         stream: true,
       })
 
-      yield* this.processGrokStreamChunks(stream, options)
+      yield* this.processGrokStreamChunks(stream, options, aguiState)
     } catch (error: unknown) {
-      const err = error as Error
+      const err = error as Error & { code?: string }
+
+      // Emit RUN_STARTED if not yet emitted
+      if (!aguiState.hasEmittedRunStarted) {
+        aguiState.hasEmittedRunStarted = true
+        yield {
+          type: 'RUN_STARTED',
+          runId: aguiState.runId,
+          model: options.model,
+          timestamp,
+        }
+      }
+
+      // Emit AG-UI RUN_ERROR
+      yield {
+        type: 'RUN_ERROR',
+        runId: aguiState.runId,
+        model: options.model,
+        timestamp,
+        error: {
+          message: err.message || 'Unknown error',
+          code: err.code,
+        },
+      }
+
       console.error('>>> chatStream: Fatal error during response creation <<<')
       console.error('>>> Error message:', err.message)
       console.error('>>> Error stack:', err.stack)
       console.error('>>> Full error:', err)
-      throw error
     }
   }
 
@@ -157,10 +189,16 @@ export class GrokTextAdapter<
   private async *processGrokStreamChunks(
     stream: AsyncIterable<OpenAI_SDK.Chat.Completions.ChatCompletionChunk>,
     options: TextOptions,
+    aguiState: {
+      runId: string
+      messageId: string
+      timestamp: number
+      hasEmittedRunStarted: boolean
+    },
   ): AsyncIterable<StreamChunk> {
     let accumulatedContent = ''
-    const timestamp = Date.now()
-    let responseId = generateId(this.name)
+    const timestamp = aguiState.timestamp
+    let hasEmittedTextMessageStart = false
 
     // Track tool calls being streamed (arguments come in chunks)
     const toolCallsInProgress = new Map<
@@ -169,31 +207,55 @@ export class GrokTextAdapter<
         id: string
         name: string
         arguments: string
+        started: boolean // Track if TOOL_CALL_START has been emitted
       }
     >()
 
     try {
       for await (const chunk of stream) {
-        responseId = chunk.id || responseId
         const choice = chunk.choices[0]
 
         if (!choice) continue
 
+        // Emit RUN_STARTED on first chunk
+        if (!aguiState.hasEmittedRunStarted) {
+          aguiState.hasEmittedRunStarted = true
+          yield {
+            type: 'RUN_STARTED',
+            runId: aguiState.runId,
+            model: chunk.model || options.model,
+            timestamp,
+          }
+        }
+
         const delta = choice.delta
         const deltaContent = delta.content
         const deltaToolCalls = delta.tool_calls
 
         // Handle content delta
         if (deltaContent) {
+          // Emit TEXT_MESSAGE_START on first text content
+          if (!hasEmittedTextMessageStart) {
+            hasEmittedTextMessageStart = true
+            yield {
+              type: 'TEXT_MESSAGE_START',
+              messageId: aguiState.messageId,
+              model: chunk.model || options.model,
+              timestamp,
+              role: 'assistant',
+            }
+          }
+
           accumulatedContent += deltaContent
+
+          // Emit AG-UI TEXT_MESSAGE_CONTENT
           yield {
-            type: 'content',
-            id: responseId,
+            type: 'TEXT_MESSAGE_CONTENT',
+            messageId: aguiState.messageId,
             model: chunk.model || options.model,
             timestamp,
             delta: deltaContent,
             content: accumulatedContent,
-            role: 'assistant',
           }
         }
 
@@ -208,6 +270,7 @@ export class GrokTextAdapter<
                 id: toolCallDelta.id || '',
                 name: toolCallDelta.function?.name || '',
                 arguments: '',
+                started: false,
               })
             }
 
@@ -223,6 +286,30 @@ export class GrokTextAdapter<
             if (toolCallDelta.function?.arguments) {
               toolCall.arguments += toolCallDelta.function.arguments
             }
+
+            // Emit TOOL_CALL_START when we have id and name
+            if (toolCall.id && toolCall.name && !toolCall.started) {
+              toolCall.started = true
+              yield {
+                type: 'TOOL_CALL_START',
+                toolCallId: toolCall.id,
+                toolName: toolCall.name,
+                model: chunk.model || options.model,
+                timestamp,
+                index,
+              }
+            }
+
+            // Emit TOOL_CALL_ARGS for argument deltas
+            if (toolCallDelta.function?.arguments && toolCall.started) {
+              yield {
+                type: 'TOOL_CALL_ARGS',
+                toolCallId: toolCall.id,
+                model: chunk.model || options.model,
+                timestamp,
+                delta: toolCallDelta.function.arguments,
+              }
+            }
           }
         }
 
@@ -233,28 +320,49 @@ export class GrokTextAdapter<
             choice.finish_reason === 'tool_calls' ||
             toolCallsInProgress.size > 0
           ) {
-            for (const [index, toolCall] of toolCallsInProgress) {
+            for (const [, toolCall] of toolCallsInProgress) {
+              // Parse arguments for TOOL_CALL_END
+              let parsedInput: unknown = {}
+              try {
+                parsedInput = toolCall.arguments
+                  ? JSON.parse(toolCall.arguments)
+                  : {}
+              } catch {
+                parsedInput = {}
+              }
+
+              // Emit AG-UI TOOL_CALL_END
               yield {
-                type: 'tool_call',
-                id: responseId,
+                type: 'TOOL_CALL_END',
+                toolCallId: toolCall.id,
+                toolName: toolCall.name,
                 model: chunk.model || options.model,
                 timestamp,
-                index,
-                toolCall: {
-                  id: toolCall.id,
-                  type: 'function',
-                  function: {
-                    name: toolCall.name,
-                    arguments: toolCall.arguments,
-                  },
-                },
+                input: parsedInput,
               }
             }
           }
 
+          const computedFinishReason =
+            choice.finish_reason === 'tool_calls' ||
+            toolCallsInProgress.size > 0
+              ? 'tool_calls'
+              : 'stop'
+
+          // Emit TEXT_MESSAGE_END if we had text content
+          if (hasEmittedTextMessageStart) {
+            yield {
+              type: 'TEXT_MESSAGE_END',
+              messageId: aguiState.messageId,
+              model: chunk.model || options.model,
+              timestamp,
+            }
+          }
+
+          // Emit AG-UI RUN_FINISHED
           yield {
-            type: 'done',
-            id: responseId,
+            type: 'RUN_FINISHED',
+            runId: aguiState.runId,
             model: chunk.model || options.model,
             timestamp,
             usage: chunk.usage
@@ -264,20 +372,18 @@ export class GrokTextAdapter<
                   totalTokens: chunk.usage.total_tokens || 0,
                 }
               : undefined,
-            finishReason:
-              choice.finish_reason === 'tool_calls' ||
-              toolCallsInProgress.size > 0
-                ? 'tool_calls'
-                : 'stop',
+            finishReason: computedFinishReason,
           }
         }
       }
     } catch (error: unknown) {
       const err = error as Error & { code?: string }
       console.log('[Grok Adapter] Stream ended with error:', err.message)
+
+      // Emit AG-UI RUN_ERROR
       yield {
-        type: 'error',
-        id: responseId,
+        type: 'RUN_ERROR',
+        runId: aguiState.runId,
         model: options.model,
         timestamp,
         error: {
@@ -396,10 +502,16 @@ export class GrokTextAdapter<
         parts.push({ type: 'text', text: part.content })
       } else if (part.type === 'image') {
         const imageMetadata = part.metadata as GrokImageMetadata | undefined
+        // For base64 data, construct a data URI using the mimeType from source
+        const imageValue = part.source.value
+        const imageUrl =
+          part.source.type === 'data' && !imageValue.startsWith('data:')
+            ? `data:${part.source.mimeType};base64,${imageValue}`
+            : imageValue
         parts.push({
           type: 'image_url',
           image_url: {
-            url: part.source.value,
+            url: imageUrl,
             detail: imageMetadata?.detail || 'auto',
           },
         })