@blaxel/langgraph 0.2.50-dev.215 → 0.2.50-preview.115

package/dist/cjs/model/cohere.js

@@ -94,6 +94,43 @@ const createCohereFetcher = () => {
                     if ('tool_results' in transformedObj) {
                         delete transformedObj.tool_results;
                     }
+                    // Transform tools array to ensure each tool has a 'type' field (required by Cohere v2)
+                    if ('tools' in transformedObj && Array.isArray(transformedObj.tools)) {
+                        const tools = transformedObj.tools;
+                        transformedObj.tools = tools.map((tool) => {
+                            if (typeof tool === 'object' && tool !== null) {
+                                const toolObj = tool;
+                                // If tool already has 'type' field, keep it
+                                if ('type' in toolObj) {
+                                    return toolObj;
+                                }
+                                // If tool has 'function' field (OpenAI format), wrap it with type
+                                if ('function' in toolObj) {
+                                    return {
+                                        type: 'function',
+                                        function: toolObj.function,
+                                    };
+                                }
+                                // If tool has name/description/parameters (direct format), wrap it
+                                if ('name' in toolObj && ('description' in toolObj || 'parameters' in toolObj)) {
+                                    return {
+                                        type: 'function',
+                                        function: {
+                                            name: toolObj.name,
+                                            description: toolObj.description || '',
+                                            parameters: toolObj.parameters || toolObj.inputSchema || {},
+                                        },
+                                    };
+                                }
+                                // Default: add type field
+                                return {
+                                    type: 'function',
+                                    ...toolObj,
+                                };
+                            }
+                            return tool;
+                        });
+                    }
                 }
                 requestBody = JSON.stringify(transformedBody);
             }
@@ -134,6 +171,43 @@ const createCohereFetcher = () => {
                                 }
                                 delete transformed.message;
                             }
+                            // Transform tools array to ensure each tool has a 'type' field (required by Cohere v2)
+                            if ('tools' in transformed && Array.isArray(transformed.tools)) {
+                                const tools = transformed.tools;
+                                transformed.tools = tools.map((tool) => {
+                                    if (typeof tool === 'object' && tool !== null) {
+                                        const toolObj = tool;
+                                        // If tool already has 'type' field, keep it
+                                        if ('type' in toolObj) {
+                                            return toolObj;
+                                        }
+                                        // If tool has 'function' field (OpenAI format), wrap it with type
+                                        if ('function' in toolObj) {
+                                            return {
+                                                type: 'function',
+                                                function: toolObj.function,
+                                            };
+                                        }
+                                        // If tool has name/description/parameters (direct format), wrap it
+                                        if ('name' in toolObj && ('description' in toolObj || 'parameters' in toolObj)) {
+                                            return {
+                                                type: 'function',
+                                                function: {
+                                                    name: toolObj.name,
+                                                    description: toolObj.description || '',
+                                                    parameters: toolObj.parameters || toolObj.inputSchema || {},
+                                                },
+                                            };
+                                        }
+                                        // Default: add type field
+                                        return {
+                                            type: 'function',
+                                            ...toolObj,
+                                        };
+                                    }
+                                    return tool;
+                                });
+                            }
                             requestBody = JSON.stringify(transformed);
                         }
                         else {

package/dist/cjs/model/google-genai.js

@@ -2,29 +2,92 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AuthenticatedChatGoogleGenerativeAI = void 0;
 const core_1 = require("@blaxel/core");
-const index_js_1 = require("./google-genai/index.js");
+const google_genai_1 = require("@langchain/google-genai");
+const generative_ai_1 = require("@google/generative-ai");
 /**
  * Custom ChatGoogleGenerativeAI that ensures authentication before each request
+ * and supports custom headers without modifying the library code
  */
-class AuthenticatedChatGoogleGenerativeAI extends index_js_1.ChatGoogleGenerativeAI {
+class AuthenticatedChatGoogleGenerativeAI extends google_genai_1.ChatGoogleGenerativeAI {
+    customHeaders;
+    constructorParams;
+    constructor(fields) {
+        super(fields);
+        // Store constructor parameters for recreating the client
+        this.constructorParams = {
+            apiKey: fields.apiKey,
+            apiVersion: fields.apiVersion,
+            baseUrl: fields.baseUrl,
+            model: fields.model,
+            safetySettings: fields.safetySettings,
+            stopSequences: fields.stopSequences,
+            maxOutputTokens: fields.maxOutputTokens,
+            temperature: fields.temperature,
+            topP: fields.topP,
+            topK: fields.topK,
+            json: fields.json,
+            thinkingConfig: fields.thinkingConfig,
+        };
+        this.customHeaders = fields.customHeaders;
+        // Initialize client with custom headers if provided
+        if (this.customHeaders) {
+            this.recreateClient();
+        }
+    }
+    /**
+     * Recreates the client with updated custom headers
+     * Uses type assertion to access the private client property
+     */
+    recreateClient() {
+        const apiKey = this.constructorParams.apiKey || this.apiKey;
+        if (!apiKey)
+            return;
+        // Get the processed model name from the base class (it removes "models/" prefix)
+        const model = this.model || this.constructorParams.model.replace(/^models\//, "");
+        const modelParams = {
+            model,
+            safetySettings: this.constructorParams.safetySettings,
+            generationConfig: {
+                candidateCount: 1,
+                stopSequences: this.constructorParams.stopSequences,
+                maxOutputTokens: this.constructorParams.maxOutputTokens,
+                temperature: this.constructorParams.temperature,
+                topP: this.constructorParams.topP,
+                topK: this.constructorParams.topK,
+                ...(this.constructorParams.json ? { responseMimeType: "application/json" } : {}),
+                ...(this.constructorParams.thinkingConfig
+                    ? { thinkingConfig: this.constructorParams.thinkingConfig }
+                    : {}),
+            },
+        };
+        const requestOptions = {
+            apiVersion: this.constructorParams.apiVersion,
+            baseUrl: this.constructorParams.baseUrl,
+            customHeaders: this.customHeaders,
+        };
+        // Use type assertion to access private client property
+        this.client = new generative_ai_1.GoogleGenerativeAI(apiKey).getGenerativeModel(modelParams, requestOptions);
+    }
     async _generate(messages, options, runManager) {
         // Authenticate before making the request
         await (0, core_1.authenticate)();
-        this.customHeaders = {};
-        for (const header in core_1.settings.headers) {
-            this.customHeaders[header] = core_1.settings.headers[header];
-        }
-        this.client = this.initClient();
+        // Update custom headers from settings
+        this.customHeaders = { ...core_1.settings.headers };
+        // Recreate client with updated headers
+        this.recreateClient();
         return await super._generate(messages, options || {}, runManager);
     }
     async *_streamResponseChunks(messages, options, runManager) {
         // Authenticate before making the request
         await (0, core_1.authenticate)();
-        this.customHeaders = {};
-        for (const header in core_1.settings.headers) {
-            this.customHeaders[header] = core_1.settings.headers[header];
-        }
-        yield* super._streamResponseChunks(messages, options || {}, runManager);
+        // Update custom headers from settings
+        this.customHeaders = { ...core_1.settings.headers };
+        // Recreate client with updated headers
+        this.recreateClient();
+        yield* super._streamResponseChunks(messages, options, runManager);
+    }
+    bindTools(tools, kwargs) {
+        return super.bindTools(tools, kwargs);
     }
 }
 exports.AuthenticatedChatGoogleGenerativeAI = AuthenticatedChatGoogleGenerativeAI;

package/dist/cjs/model.js

@@ -6,9 +6,9 @@ const anthropic_1 = require("@langchain/anthropic");
 const cohere_1 = require("@langchain/cohere");
 const deepseek_1 = require("@langchain/deepseek");
 const openai_1 = require("@langchain/openai");
+const google_genai_js_1 = require("./model/google-genai.js");
 const cohere_ai_1 = require("cohere-ai");
 const cohere_js_1 = require("./model/cohere.js");
-const google_genai_js_1 = require("./model/google-genai.js");
 const xai_js_1 = require("./model/xai.js");
 /**
  * Creates a custom fetch function that adds dynamic headers to each request

package/dist/cjs/types/model/google-genai.d.ts

@@ -1,11 +1,28 @@
 import { CallbackManagerForLLMRun } from "@langchain/core/callbacks/manager";
+import { BaseLanguageModelInput } from "@langchain/core/language_models/base";
 import { BaseMessage } from "@langchain/core/messages";
-import { ChatResult } from "@langchain/core/outputs";
-import { ChatGoogleGenerativeAI } from "./google-genai/index.js";
+import { AIMessageChunk } from "@langchain/core/messages";
+import { ChatResult, ChatGenerationChunk } from "@langchain/core/outputs";
+import { Runnable } from "@langchain/core/runnables";
+import { ChatGoogleGenerativeAI, GoogleGenerativeAIChatCallOptions, GoogleGenerativeAIChatInput } from "@langchain/google-genai";
+type GoogleGenerativeAIToolType = NonNullable<GoogleGenerativeAIChatCallOptions["tools"]>[number];
 /**
  * Custom ChatGoogleGenerativeAI that ensures authentication before each request
+ * and supports custom headers without modifying the library code
  */
 export declare class AuthenticatedChatGoogleGenerativeAI extends ChatGoogleGenerativeAI {
+    private customHeaders?;
+    private constructorParams;
+    constructor(fields: GoogleGenerativeAIChatInput & {
+        customHeaders?: Record<string, string>;
+    });
+    /**
+     * Recreates the client with updated custom headers
+     * Uses type assertion to access the private client property
+     */
+    private recreateClient;
     _generate(messages: BaseMessage[], options?: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): Promise<ChatResult>;
-    _streamResponseChunks(messages: BaseMessage[], options?: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): AsyncGenerator<any>;
+    _streamResponseChunks(messages: BaseMessage[], options: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): AsyncGenerator<ChatGenerationChunk>;
+    bindTools(tools: GoogleGenerativeAIToolType[], kwargs?: Partial<GoogleGenerativeAIChatCallOptions>): Runnable<BaseLanguageModelInput, AIMessageChunk, GoogleGenerativeAIChatCallOptions>;
 }
+export {};