agent-pulse 1.4.4 → 1.5.0

package/dist/providers/google.js

@@ -72,12 +72,17 @@ class GoogleProvider {
                 }
                 if (m.tool_calls) {
                     for (const tc of m.tool_calls) {
-                        parts.push({
+                        const fcPart = {
                             functionCall: {
                                 name: tc.name,
                                 args: tc.arguments
                             }
-                        });
+                        };
+                        // Preserve thought_signature for Gemini 3+ models
+                        if (tc.thought_signature) {
+                            fcPart.thoughtSignature = tc.thought_signature;
+                        }
+                        parts.push(fcPart);
                     }
                 }
                 if (m.content) {
@@ -225,20 +230,18 @@ class GoogleProvider {
             // Tool Calls
             // New SDK usually aggregates or provides them in chunks.
             // We need to check `candidates[0].content.parts` for `functionCall`.
-            // Streaming tool calls might be tricky as they come in parts? 
-            // Google usually sends the full function call in one chunk or at the end? 
+            // Streaming tool calls might be tricky as they come in parts?
+            // Google usually sends the full function call in one chunk or at the end?
             // Let's inspect the chunk structure via candidates.
             const parts = chunk.candidates?.[0]?.content?.parts;
             if (parts) {
                 for (const part of parts) {
                     if (part.functionCall) {
-                        // It seems Google sends full function call or we need to accumulate?
-                        // Usually it's complete in the response object if stream is done, but in stream?
-                        // Let's collect them.
                         toolCalls.push({
                             name: part.functionCall.name,
-                            arguments: part.functionCall.args, // Already parsed JSON usually
-                            id: 'call_' + Date.now() // Google doesn't always send ID, need to check
+                            arguments: part.functionCall.args,
+                            id: 'call_' + Date.now(),
+                            thought_signature: part.thoughtSignature || undefined
                         });
                     }
                 }

package/dist/providers/grok.js

@@ -23,17 +23,51 @@ class GrokProvider {
             const promptText = Array.isArray(prompt)
                 ? prompt.filter(m => m.role === 'user').map(m => m.content).join('\n')
                 : String(prompt);
-            // Both generation and editing use /v1/images/generations
-            // For editing, just add image_url to the same request
             const imageUrl = config?.reference_image || config?.image_url;
-            const response = await this.client.images.generate({
-                model: this.model,
-                prompt: promptText,
-                n: config?.n || 1,
-                response_format: config?.response_format || 'b64_json',
-                ...(config?.aspect_ratio && { aspect_ratio: config.aspect_ratio }),
-                ...(imageUrl && { image_url: imageUrl }),
-            });
+            let response;
+            if (imageUrl) {
+                // Image EDITING: use /v1/images/edits with JSON body (not multipart/form-data)
+                // The OpenAI SDK's images.edit() uses multipart/form-data which xAI doesn't support,
+                // so we make a direct fetch call with application/json
+                const editBody = {
+                    model: this.model,
+                    prompt: promptText,
+                    n: config?.n || 1,
+                    response_format: config?.response_format || 'b64_json',
+                    image: {
+                        url: imageUrl,
+                        type: 'image_url',
+                    },
+                };
+                // For multi-image editing, aspect_ratio can override input ratio
+                if (config?.aspect_ratio) {
+                    editBody.aspect_ratio = config.aspect_ratio;
+                }
+                const res = await fetch('https://api.x.ai/v1/images/edits', {
+                    method: 'POST',
+                    headers: {
+                        'Content-Type': 'application/json',
+                        'Authorization': `Bearer ${this.client.apiKey}`,
+                    },
+                    body: JSON.stringify(editBody),
+                });
+                if (!res.ok) {
+                    const errorText = await res.text();
+                    throw new Error(`xAI image edit failed (${res.status}): ${errorText}`);
+                }
+                response = await res.json();
+            }
+            else {
+                // Image GENERATION: text-to-image via /v1/images/generations
+                const genResponse = await this.client.images.generate({
+                    model: this.model,
+                    prompt: promptText,
+                    n: config?.n || 1,
+                    response_format: config?.response_format || 'b64_json',
+                    ...(config?.aspect_ratio && { aspect_ratio: config.aspect_ratio }),
+                });
+                response = genResponse;
+            }
             const images = response.data;
             const markdownParts = images.map((img, i) => {
                 if (img.b64_json) {

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "agent-pulse",
-    "version": "1.4.4",
+    "version": "1.5.0",
     "description": "A lightweight, agentic AI framework for JavaScript/TypeScript",
     "main": "dist/index.js",
     "types": "dist/index.d.ts",