@superatomai/sdk-node 0.0.74 → 0.0.75

package/dist/index.d.mts

@@ -1337,6 +1337,11 @@ declare class LLM {
     private static _groqStream;
     private static _geminiText;
     private static _geminiStream;
+    /**
+     * Recursively strip unsupported JSON Schema properties for Gemini
+     * Gemini doesn't support: additionalProperties, $schema, etc.
+     */
+    private static _cleanSchemaForGemini;
     private static _geminiStreamWithTools;
     private static _openaiText;
     private static _openaiStream;

package/dist/index.d.ts

@@ -1337,6 +1337,11 @@ declare class LLM {
     private static _groqStream;
     private static _geminiText;
     private static _geminiStream;
+    /**
+     * Recursively strip unsupported JSON Schema properties for Gemini
+     * Gemini doesn't support: additionalProperties, $schema, etc.
+     */
+    private static _cleanSchemaForGemini;
     private static _geminiStreamWithTools;
     private static _openaiText;
     private static _openaiStream;

package/dist/index.js

@@ -5400,14 +5400,97 @@ var LLM = class {
     const genAI = new import_generative_ai.GoogleGenerativeAI(apiKey);
     const systemPrompt = typeof messages.sys === "string" ? messages.sys : messages.sys.map((block) => block.text).join("\n");
     try {
+      if (json && options.partial) {
+        const model2 = genAI.getGenerativeModel({
+          model: modelName,
+          systemInstruction: systemPrompt,
+          generationConfig: {
+            maxOutputTokens: options.maxTokens || 1e3,
+            temperature: options.temperature,
+            topP: options.topP,
+            responseMimeType: "application/json"
+          }
+        });
+        const result2 = await model2.generateContentStream(messages.user);
+        let fullText2 = "";
+        let inputTokens2 = 0;
+        let outputTokens2 = 0;
+        for await (const chunk of result2.stream) {
+          try {
+            const text = chunk.text();
+            if (text) {
+              fullText2 += text;
+              options.partial(text);
+            }
+          } catch (chunkError) {
+          }
+          if (chunk.usageMetadata) {
+            inputTokens2 = chunk.usageMetadata.promptTokenCount || 0;
+            outputTokens2 = chunk.usageMetadata.candidatesTokenCount || 0;
+          }
+        }
+        const durationMs2 = Date.now() - startTime;
+        if (inputTokens2 === 0) {
+          inputTokens2 = Math.ceil((systemPrompt.length + messages.user.length) / 4);
+        }
+        if (outputTokens2 === 0) {
+          outputTokens2 = Math.ceil(fullText2.length / 4);
+        }
+        llmUsageLogger.log({
+          timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+          requestId,
+          provider: "gemini",
+          model: modelName,
+          method: "stream-json-partial",
+          inputTokens: inputTokens2,
+          outputTokens: outputTokens2,
+          totalTokens: inputTokens2 + outputTokens2,
+          costUSD: llmUsageLogger.calculateCost(modelName, inputTokens2, outputTokens2),
+          durationMs: durationMs2,
+          success: true
+        });
+        return this._parseJSON(fullText2);
+      }
+      if (json) {
+        const model2 = genAI.getGenerativeModel({
+          model: modelName,
+          systemInstruction: systemPrompt,
+          generationConfig: {
+            maxOutputTokens: options.maxTokens || 1e3,
+            temperature: options.temperature,
+            topP: options.topP,
+            responseMimeType: "application/json"
+          }
+        });
+        const result2 = await model2.generateContent(messages.user);
+        const response = result2.response;
+        const fullText2 = response.text();
+        const durationMs2 = Date.now() - startTime;
+        const usage = response.usageMetadata;
+        const inputTokens2 = usage?.promptTokenCount || Math.ceil((systemPrompt.length + messages.user.length) / 4);
+        const outputTokens2 = usage?.candidatesTokenCount || Math.ceil(fullText2.length / 4);
+        llmUsageLogger.log({
+          timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+          requestId,
+          provider: "gemini",
+          model: modelName,
+          method: "stream-json",
+          inputTokens: inputTokens2,
+          outputTokens: outputTokens2,
+          totalTokens: inputTokens2 + outputTokens2,
+          costUSD: llmUsageLogger.calculateCost(modelName, inputTokens2, outputTokens2),
+          durationMs: durationMs2,
+          success: true
+        });
+        return this._parseJSON(fullText2);
+      }
       const model = genAI.getGenerativeModel({
         model: modelName,
         systemInstruction: systemPrompt,
         generationConfig: {
           maxOutputTokens: options.maxTokens || 1e3,
           temperature: options.temperature,
-          topP: options.topP,
-          responseMimeType: json ? "application/json" : void 0
+          topP: options.topP
         }
       });
       const result = await model.generateContentStream(messages.user);
@@ -5415,12 +5498,15 @@ var LLM = class {
       let inputTokens = 0;
       let outputTokens = 0;
       for await (const chunk of result.stream) {
-        const text = chunk.text();
-        if (text) {
-          fullText += text;
-          if (options.partial) {
-            options.partial(text);
+        try {
+          const text = chunk.text();
+          if (text) {
+            fullText += text;
+            if (options.partial) {
+              options.partial(text);
+            }
           }
+        } catch (chunkError) {
         }
         if (chunk.usageMetadata) {
           inputTokens = chunk.usageMetadata.promptTokenCount || 0;
@@ -5447,9 +5533,6 @@ var LLM = class {
         durationMs,
         success: true
       });
-      if (json) {
-        return this._parseJSON(fullText);
-      }
       return fullText;
     } catch (error) {
       const durationMs = Date.now() - startTime;
@@ -5470,6 +5553,26 @@ var LLM = class {
       throw error;
     }
   }
+  /**
+   * Recursively strip unsupported JSON Schema properties for Gemini
+   * Gemini doesn't support: additionalProperties, $schema, etc.
+   */
+  static _cleanSchemaForGemini(obj) {
+    if (obj === null || typeof obj !== "object") {
+      return obj;
+    }
+    if (Array.isArray(obj)) {
+      return obj.map((item) => this._cleanSchemaForGemini(item));
+    }
+    const cleaned = {};
+    for (const [key, value] of Object.entries(obj)) {
+      if (key === "additionalProperties" || key === "$schema") {
+        continue;
+      }
+      cleaned[key] = this._cleanSchemaForGemini(value);
+    }
+    return cleaned;
+  }
   static async _geminiStreamWithTools(messages, tools, toolHandler, modelName, options, maxIterations) {
     const methodStartTime = Date.now();
     const apiKey = options.apiKey || process.env.GEMINI_API_KEY || "";
@@ -5480,7 +5583,7 @@ var LLM = class {
       description: tool.description,
       parameters: {
         type: import_generative_ai.SchemaType.OBJECT,
-        properties: tool.input_schema.properties,
+        properties: this._cleanSchemaForGemini(tool.input_schema.properties),
         required: tool.input_schema.required || []
       }
     }));
@@ -8422,10 +8525,10 @@ var GeminiLLM = class extends BaseLLM {
     super(config);
   }
   getDefaultModel() {
-    return "gemini/gemini-3-pro-preview";
+    return "gemini/gemini-2.5-flash";
   }
   getDefaultFastModel() {
-    return "gemini/gemini-3-flash-preview";
+    return "gemini/gemini-2.5-flash";
   }
   getDefaultApiKey() {
     return process.env.GEMINI_API_KEY;