@blaxel/langgraph 0.2.49 → 0.2.50-preview.115

package/dist/esm/model/cohere.js

@@ -11,7 +11,9 @@ export const createCohereFetcher = () => {
         // Extract all fields from args
         const { url, method, headers: argsHeaders, body, contentType, queryParameters, timeoutMs, withCredentials, abortSignal, requestType, responseType, duplex } = args;
         // Build URL with query parameters
-        let requestUrl = url;
+        // Rewrite /v1/chat to /v2/chat for Cohere API v2 compatibility
+        let requestUrl = url.replace('/v1/chat', '/v2/chat');
+        const isV2Endpoint = requestUrl.includes('/v2/chat');
         if (queryParameters) {
             const params = new URLSearchParams();
             Object.entries(queryParameters).forEach(([key, value]) => {
@@ -57,7 +59,77 @@ export const createCohereFetcher = () => {
         let requestBody;
         if (body !== undefined) {
             if (requestType === 'json' || !requestType) {
-                requestBody = JSON.stringify(body);
+                // Transform body for Cohere v2 API compatibility (only if using v2 endpoint)
+                let transformedBody = body;
+                if (isV2Endpoint && typeof body === 'object' && body !== null && !Array.isArray(body)) {
+                    const bodyObj = body;
+                    transformedBody = { ...bodyObj };
+                    const transformedObj = transformedBody;
+                    // Remove v1-only fields that are not supported in v2
+                    const fieldsToRemove = ['chat_history'];
+                    for (const field of fieldsToRemove) {
+                        if (field in transformedObj) {
+                            delete transformedObj[field];
+                        }
+                    }
+                    // Convert 'message' to 'messages' format if message exists and messages doesn't
+                    if ('message' in transformedObj && !('messages' in transformedObj)) {
+                        const message = transformedObj.message;
+                        if (typeof message === 'string' && message.trim().length > 0) {
+                            // Convert single message string to messages array format
+                            transformedObj.messages = [
+                                {
+                                    role: 'user',
+                                    content: message,
+                                },
+                            ];
+                        }
+                        // Remove the old message field
+                        delete transformedObj.message;
+                    }
+                    // Handle tool_results - v2 might use a different format, remove for now
+                    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);
             }
             else if (requestType === 'bytes' && body instanceof Uint8Array) {
                 // Create a new ArrayBuffer from the Uint8Array to avoid SharedArrayBuffer issues
@@ -70,7 +142,82 @@ export const createCohereFetcher = () => {
                 requestBody = body;
             }
             else if (typeof body === 'string') {
-                requestBody = body;
+                // Parse and transform JSON strings (only if using v2 endpoint)
+                if (isV2Endpoint) {
+                    try {
+                        const parsed = JSON.parse(body);
+                        if (typeof parsed === 'object' && parsed !== null && !Array.isArray(parsed)) {
+                            const transformed = { ...parsed };
+                            // Remove v1-only fields
+                            if ('chat_history' in transformed) {
+                                delete transformed.chat_history;
+                            }
+                            if ('tool_results' in transformed) {
+                                delete transformed.tool_results;
+                            }
+                            // Convert 'message' to 'messages' format if message exists and messages doesn't
+                            if ('message' in transformed && !('messages' in transformed)) {
+                                const message = transformed.message;
+                                if (typeof message === 'string' && message.trim().length > 0) {
+                                    transformed.messages = [
+                                        {
+                                            role: 'user',
+                                            content: message,
+                                        },
+                                    ];
+                                }
+                                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 {
+                            requestBody = body;
+                        }
+                    }
+                    catch {
+                        requestBody = body;
+                    }
+                }
+                else {
+                    requestBody = body;
+                }
             }
             else {
                 requestBody = JSON.stringify(body);
@@ -123,6 +270,41 @@ export const createCohereFetcher = () => {
                 else {
                     // Default to JSON
                     responseBody = await response.json();
+                    // Transform v2 response format to v1 format for ChatCohere compatibility
+                    if (isV2Endpoint && typeof responseBody === 'object' && responseBody !== null) {
+                        const responseObj = responseBody;
+                        if ('message' in responseObj && typeof responseObj.message === 'object' && responseObj.message !== null) {
+                            const v2Message = responseObj.message;
+                            // Extract text from content array
+                            let text = '';
+                            if (Array.isArray(v2Message.content)) {
+                                const contentArray = v2Message.content;
+                                // Find the text content block
+                                const textBlock = contentArray.find((item) => item.type === 'text' && item.text);
+                                if (textBlock && textBlock.text) {
+                                    text = textBlock.text;
+                                }
+                                else {
+                                    // Fallback: join all text-like content
+                                    text = contentArray
+                                        .map((item) => item.text || item.thinking || '')
+                                        .filter(Boolean)
+                                        .join('\n');
+                                }
+                            }
+                            else if (typeof v2Message.content === 'string') {
+                                text = v2Message.content;
+                            }
+                            // Transform to v1-like format that ChatCohere expects
+                            const transformedResponse = {
+                                ...responseObj,
+                                text: text,
+                                // Keep the original message structure in case ChatCohere needs it
+                                message: responseObj.message,
+                            };
+                            responseBody = transformedResponse;
+                        }
+                    }
                 }
                 // Return success response in the format CohereClient expects
                 return {

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

@@ -1,26 +1,89 @@
 import { authenticate, settings } from "@blaxel/core";
-import { ChatGoogleGenerativeAI } from "./google-genai/index.js";
+import { ChatGoogleGenerativeAI } from "@langchain/google-genai";
+import { GoogleGenerativeAI as GenerativeAI } from "@google/generative-ai";
 /**
  * Custom ChatGoogleGenerativeAI that ensures authentication before each request
+ * and supports custom headers without modifying the library code
  */
 export class AuthenticatedChatGoogleGenerativeAI extends 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 GenerativeAI(apiKey).getGenerativeModel(modelParams, requestOptions);
+    }
     async _generate(messages, options, runManager) {
         // Authenticate before making the request
         await authenticate();
-        this.customHeaders = {};
-        for (const header in settings.headers) {
-            this.customHeaders[header] = settings.headers[header];
-        }
-        this.client = this.initClient();
+        // Update custom headers from settings
+        this.customHeaders = { ...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 authenticate();
-        this.customHeaders = {};
-        for (const header in settings.headers) {
-            this.customHeaders[header] = settings.headers[header];
-        }
-        yield* super._streamResponseChunks(messages, options || {}, runManager);
+        // Update custom headers from settings
+        this.customHeaders = { ...settings.headers };
+        // Recreate client with updated headers
+        this.recreateClient();
+        yield* super._streamResponseChunks(messages, options, runManager);
+    }
+    bindTools(tools, kwargs) {
+        return super.bindTools(tools, kwargs);
     }
 }

package/dist/esm/model.js

@@ -3,9 +3,9 @@ import { ChatAnthropic } from "@langchain/anthropic";
 import { ChatCohere } from "@langchain/cohere";
 import { ChatDeepSeek } from "@langchain/deepseek";
 import { ChatOpenAI } from "@langchain/openai";
+import { AuthenticatedChatGoogleGenerativeAI } from "./model/google-genai.js";
 import { CohereClient } from "cohere-ai";
 import { createCohereFetcher } from "./model/cohere.js";
-import { AuthenticatedChatGoogleGenerativeAI } from "./model/google-genai.js";
 import { ChatXAI } from "./model/xai.js";
 /**
  * Creates a custom fetch function that adds dynamic headers to each request
@@ -20,6 +20,24 @@ const authenticatedFetch = () => {
             ...dynamicHeaders,
             ...(init?.headers || {}),
         };
+        // Ensure Content-Type is set for JSON requests if body exists and Content-Type is not already set
+        if (init?.body && !headers['Content-Type'] && !headers['content-type']) {
+            // If body is an object, it will be serialized to JSON by fetch
+            // If body is a string, check if it looks like JSON
+            if (typeof init.body === 'string') {
+                const trimmed = init.body.trim();
+                if (trimmed.startsWith('{') || trimmed.startsWith('[')) {
+                    headers['Content-Type'] = 'application/json';
+                }
+            }
+            else {
+                // For non-string bodies (FormData, Blob, etc.), let fetch handle it
+                // For objects, assume JSON
+                if (typeof init.body === 'object' && !(init.body instanceof FormData) && !(init.body instanceof Blob)) {
+                    headers['Content-Type'] = 'application/json';
+                }
+            }
+        }
         // Make the request with merged headers
         return await fetch(input, {
             ...init,
@@ -60,6 +78,11 @@ export const blModel = async (model, options) => {
             });
         }
         else if (type === "cohere") {
+            // ChatCohere requires a custom client with fetcher for:
+            // 1. Dynamic authentication headers (settings.headers)
+            // 2. Custom environment URL (url)
+            // 3. URL rewriting (v1 -> v2) and body transformation for v2 compatibility
+            // @ts-ignore Error in langgraph
             return new ChatCohere({
                 apiKey: "replaced",
                 model: modelData?.spec?.runtime?.model,
@@ -72,6 +95,7 @@ export const blModel = async (model, options) => {
             });
         }
         else if (type === "deepseek") {
+            // @ts-ignore Error in langgraph
             return new ChatDeepSeek({
                 apiKey: "replaced",
                 model: modelData?.spec?.runtime?.model,
@@ -84,9 +108,11 @@ export const blModel = async (model, options) => {
             });
         }
         else if (type === "anthropic") {
+            // @ts-ignore Error in langgraph
             return new ChatAnthropic({
                 anthropicApiUrl: url,
                 model: modelData?.spec?.runtime?.model,
+                apiKey: "replaced",
                 clientOptions: {
                     // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment
                     fetch: authenticatedFetch(),

package/dist/esm/telemetry.js

@@ -1,15 +1,14 @@
 import * as RunnableModule from "@langchain/core/runnables";
 import * as ToolsModule from "@langchain/core/tools";
+import * as AgentsModule from "@langchain/core/agents";
 import * as VectorStoresModule from "@langchain/core/vectorstores";
 import { registerInstrumentations } from "@opentelemetry/instrumentation";
 import { LangChainInstrumentation } from "@traceloop/instrumentation-langchain";
-import * as AgentsModule from "langchain/agents";
-import * as ChainsModule from "langchain/chains";
 const langchain = new LangChainInstrumentation();
 langchain.manuallyInstrument({
+    // @ts-ignore - Type definitions may be incorrect, but the method accepts these parameters at runtime
     runnablesModule: RunnableModule,
     toolsModule: ToolsModule,
-    chainsModule: ChainsModule,
     agentsModule: AgentsModule,
     vectorStoreModule: VectorStoresModule,
 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@blaxel/langgraph",
-  "version": "0.2.49",
+  "version": "0.2.50-preview.115",
   "description": "Blaxel SDK for TypeScript",
   "license": "MIT",
   "author": "Blaxel, INC (https://blaxel.ai)",
@@ -41,18 +41,19 @@
   ],
   "dependencies": {
     "@google/generative-ai": "^0.24.1",
-    "@langchain/anthropic": "^0.3.20",
-    "@langchain/cohere": "^0.3.3",
-    "@langchain/core": "^0.3.51",
-    "@langchain/deepseek": "^0.0.1",
-    "@langchain/openai": "^0.5.10",
+    "@langchain/anthropic": "^1.1.1",
+    "@langchain/cohere": "^1.0.0",
+    "@langchain/core": "^1.0.6",
+    "@langchain/deepseek": "^1.0.1",
+    "@langchain/google-genai": "^1.0.3",
+    "@langchain/openai": "^1.1.2",
     "@opentelemetry/instrumentation": "^0.200.0",
-    "@traceloop/instrumentation-langchain": "^0.13.0",
-    "cohere-ai": "^7.17.1",
-    "langchain": "^0.3.24",
+    "@traceloop/instrumentation-langchain": "^0.20.0",
+    "cohere-ai": "^7.19.0",
+    "langchain": "^1.0.6",
     "zod": "^3.24.3",
     "zod-to-json-schema": "^3.24.5",
-    "@blaxel/core": "0.2.49"
+    "@blaxel/core": "0.2.50-preview.115"
   },
   "devDependencies": {
     "@eslint/js": "^9.26.0",