langchain 0.0.168 → 0.0.170

package/dist/callbacks/handlers/llmonitor.js

@@ -65,13 +65,15 @@ const parseInput = (rawInput) => {
 const parseOutput = (rawOutput) => {
     if (!rawOutput)
         return null;
-    const { text, output, answer } = rawOutput;
+    const { text, output, answer, result } = rawOutput;
     if (text)
         return text;
     if (answer)
         return answer;
     if (output)
         return output;
+    if (result)
+        return result;
     return rawOutput;
 };
 export class LLMonitorHandler extends BaseCallbackHandler {
@@ -104,15 +106,14 @@ export class LLMonitorHandler extends BaseCallbackHandler {
             ...(extraParams?.invocation_params || {}),
             ...(metadata || {}),
         };
-        const name = params?.model || params?.name || params?.model_name || llm.id.at(-1);
-        const userId = params?.userId || undefined;
-        const userProps = params?.userProps || undefined;
+        const { model, model_name, modelName, userId, userProps, ...rest } = params;
+        const name = model || modelName || model_name || llm.id.at(-1);
         await this.monitor.trackEvent("llm", "start", {
             runId,
             parentRunId,
             name,
             input: convertToLLMonitorMessages(prompts),
-            extra: params,
+            extra: rest,
             userId,
             userProps,
             tags,
@@ -124,15 +125,15 @@ export class LLMonitorHandler extends BaseCallbackHandler {
             ...(extraParams?.invocation_params || {}),
             ...(metadata || {}),
         };
-        const name = params?.model || params?.name || params?.model_name || llm.id.at(-1);
-        const userId = params?.userId || undefined;
-        const userProps = params?.userProps || undefined;
+        // Expand them so they're excluded from the "extra" field
+        const { model, model_name, modelName, userId, userProps, ...rest } = params;
+        const name = model || modelName || model_name || llm.id.at(-1);
         await this.monitor.trackEvent("llm", "start", {
             runId,
             parentRunId,
             name,
             input: convertToLLMonitorMessages(messages),
-            extra: params,
+            extra: rest,
             userId,
             userProps,
             tags,
@@ -157,20 +158,19 @@ export class LLMonitorHandler extends BaseCallbackHandler {
         });
     }
     async handleChainStart(chain, inputs, runId, parentRunId, tags, metadata) {
+        const { agentName, userId, userProps, ...rest } = metadata || {};
         // allow the user to specify an agent name
-        const chainName = chain.id.at(-1);
-        const name = (metadata?.agentName ?? chainName);
+        const name = agentName || chain.id.at(-1);
         // Attempt to automatically detect if this is an agent or chain
-        const runType = metadata?.agentName ||
-            ["AgentExecutor", "PlanAndExecute"].includes(chainName)
+        const runType = agentName || ["AgentExecutor", "PlanAndExecute"].includes(name)
             ? "agent"
             : "chain";
-        // eslint-disable-next-line @typescript-eslint/no-unused-vars
-        const { agentName, ...rest } = metadata || {};
         await this.monitor.trackEvent(runType, "start", {
             runId,
             parentRunId,
             name,
+            userId,
+            userProps,
             input: parseInput(inputs),
             extra: rest,
             tags,
@@ -190,12 +190,16 @@ export class LLMonitorHandler extends BaseCallbackHandler {
         });
     }
     async handleToolStart(tool, input, runId, parentRunId, tags, metadata) {
+        const { toolName, userId, userProps, ...rest } = metadata || {};
+        const name = toolName || tool.id.at(-1);
         await this.monitor.trackEvent("tool", "start", {
             runId,
             parentRunId,
-            name: tool.id[tool.id.length - 1],
+            name,
+            userId,
+            userProps,
             input,
-            extra: metadata,
+            extra: rest,
             tags,
             runtime: "langchain-js",
         });

package/dist/chains/sql_db/sql_db_chain.cjs

@@ -10,6 +10,15 @@ const sql_utils_js_1 = require("../../util/sql_utils.cjs");
  * Class that represents a SQL database chain in the LangChain framework.
  * It extends the BaseChain class and implements the functionality
  * specific to a SQL database chain.
+ *
+ * @security **Security Notice**
+ * This chain generates SQL queries for the given database.
+ * The SQLDatabase class provides a getTableInfo method that can be used
+ * to get column information as well as sample data from the table.
+ * To mitigate risk of leaking sensitive data, limit permissions
+ * to read and scope to the tables that are needed.
+ * Optionally, use the includesTables or ignoreTables class parameters
+ * to limit which tables can/cannot be accessed.
  */
 class SqlDatabaseChain extends base_js_1.BaseChain {
     static lc_name() {

package/dist/chains/sql_db/sql_db_chain.d.ts

@@ -22,6 +22,15 @@ export interface SqlDatabaseChainInput extends ChainInputs {
  * Class that represents a SQL database chain in the LangChain framework.
  * It extends the BaseChain class and implements the functionality
  * specific to a SQL database chain.
+ *
+ * @security **Security Notice**
+ * This chain generates SQL queries for the given database.
+ * The SQLDatabase class provides a getTableInfo method that can be used
+ * to get column information as well as sample data from the table.
+ * To mitigate risk of leaking sensitive data, limit permissions
+ * to read and scope to the tables that are needed.
+ * Optionally, use the includesTables or ignoreTables class parameters
+ * to limit which tables can/cannot be accessed.
  */
 export declare class SqlDatabaseChain extends BaseChain {
     static lc_name(): string;

package/dist/chains/sql_db/sql_db_chain.js

@@ -7,6 +7,15 @@ import { getPromptTemplateFromDataSource } from "../../util/sql_utils.js";
  * Class that represents a SQL database chain in the LangChain framework.
  * It extends the BaseChain class and implements the functionality
  * specific to a SQL database chain.
+ *
+ * @security **Security Notice**
+ * This chain generates SQL queries for the given database.
+ * The SQLDatabase class provides a getTableInfo method that can be used
+ * to get column information as well as sample data from the table.
+ * To mitigate risk of leaking sensitive data, limit permissions
+ * to read and scope to the tables that are needed.
+ * Optionally, use the includesTables or ignoreTables class parameters
+ * to limit which tables can/cannot be accessed.
  */
 export class SqlDatabaseChain extends BaseChain {
     static lc_name() {

package/dist/chat_models/baiduwenxin.cjs

@@ -28,7 +28,7 @@ function messageToWenxinRole(message) {
         case "human":
             return "user";
         case "system":
-            throw new Error("System messages not supported");
+            throw new Error("System messages should not be here");
         case "function":
             throw new Error("Function messages not supported");
         case "generic": {
@@ -164,6 +164,10 @@ class ChatBaiduWenxin extends base_js_1.BaseChatModel {
             this.apiUrl =
                 "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/eb-instant";
         }
+        else if (this.modelName === "ERNIE-Bot-4") {
+            this.apiUrl =
+                "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions_pro";
+        }
         else {
             throw new Error(`Invalid model name: ${this.modelName}`);
         }
@@ -219,6 +223,13 @@ class ChatBaiduWenxin extends base_js_1.BaseChatModel {
     async _generate(messages, options, runManager) {
         const tokenUsage = {};
         const params = this.invocationParams();
+        // Wenxin requires the system message to be put in the params, not messages array
+        const systemMessage = messages.find((message) => message._getType() === "system");
+        if (systemMessage) {
+            // eslint-disable-next-line no-param-reassign
+            messages = messages.filter((message) => message !== systemMessage);
+            params.system = systemMessage.text;
+        }
         const messagesMapped = messages.map((message) => ({
             role: messageToWenxinRole(message),
             content: message.text,

package/dist/chat_models/baiduwenxin.d.ts

@@ -22,12 +22,13 @@ interface ChatCompletionRequest {
     temperature?: number;
     top_p?: number;
     penalty_score?: number;
+    system?: string;
 }
 /**
  * Interface defining the input to the ChatBaiduWenxin class.
  */
 declare interface BaiduWenxinChatInput {
-    /** Model name to use
+    /** Model name to use. Available options are: ERNIE-Bot, ERNIE-Bot-turbo, ERNIE-Bot-4
      * @default "ERNIE-Bot-turbo"
      */
     modelName: string;
@@ -110,6 +111,7 @@ export declare class ChatBaiduWenxin extends BaseChatModel implements BaiduWenxi
      * Get the identifying parameters for the model
      */
     identifyingParams(): {
+        system?: string | undefined;
         stream?: boolean | undefined;
         temperature?: number | undefined;
         top_p?: number | undefined;

package/dist/chat_models/baiduwenxin.js

@@ -25,7 +25,7 @@ function messageToWenxinRole(message) {
         case "human":
             return "user";
         case "system":
-            throw new Error("System messages not supported");
+            throw new Error("System messages should not be here");
         case "function":
             throw new Error("Function messages not supported");
         case "generic": {
@@ -161,6 +161,10 @@ export class ChatBaiduWenxin extends BaseChatModel {
             this.apiUrl =
                 "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/eb-instant";
         }
+        else if (this.modelName === "ERNIE-Bot-4") {
+            this.apiUrl =
+                "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions_pro";
+        }
         else {
             throw new Error(`Invalid model name: ${this.modelName}`);
         }
@@ -216,6 +220,13 @@ export class ChatBaiduWenxin extends BaseChatModel {
     async _generate(messages, options, runManager) {
         const tokenUsage = {};
         const params = this.invocationParams();
+        // Wenxin requires the system message to be put in the params, not messages array
+        const systemMessage = messages.find((message) => message._getType() === "system");
+        if (systemMessage) {
+            // eslint-disable-next-line no-param-reassign
+            messages = messages.filter((message) => message !== systemMessage);
+            params.system = systemMessage.text;
+        }
         const messagesMapped = messages.map((message) => ({
             role: messageToWenxinRole(message),
             content: message.text,

package/dist/chat_models/cloudflare_workersai.cjs

@@ -52,7 +52,12 @@ class ChatCloudflareWorkersAI extends base_js_1.SimpleChatModel {
         this.cloudflareApiToken =
             fields?.cloudflareApiToken ??
                 (0, env_js_1.getEnvironmentVariable)("CLOUDFLARE_API_TOKEN");
-        this.baseUrl = fields?.baseUrl;
+        this.baseUrl =
+            fields?.baseUrl ??
+                `https://api.cloudflare.com/client/v4/accounts/${this.cloudflareAccountId}/ai/run`;
+        if (this.baseUrl.endsWith("/")) {
+            this.baseUrl = this.baseUrl.slice(0, -1);
+        }
     }
     _llmType() {
         return "cloudflare";
@@ -112,7 +117,7 @@ class ChatCloudflareWorkersAI extends base_js_1.SimpleChatModel {
     /** @ignore */
     async _call(messages, options) {
         this.validateEnvironment();
-        const url = `https://api.cloudflare.com/client/v4/accounts/${this.cloudflareAccountId}/ai/run/${this.model}`;
+        const url = `${this.baseUrl}/${this.model}`;
         const headers = {
             Authorization: `Bearer ${this.cloudflareApiToken}`,
             "Content-Type": "application/json",

package/dist/chat_models/cloudflare_workersai.d.ts

@@ -19,7 +19,7 @@ export declare class ChatCloudflareWorkersAI extends SimpleChatModel implements
     model: string;
     cloudflareAccountId?: string;
     cloudflareApiToken?: string;
-    baseUrl?: string;
+    baseUrl: string;
     constructor(fields?: CloudflareWorkersAIInput & BaseChatModelParams);
     _llmType(): string;
     /** Get the identifying parameters for this LLM. */

package/dist/chat_models/cloudflare_workersai.js

@@ -49,7 +49,12 @@ export class ChatCloudflareWorkersAI extends SimpleChatModel {
         this.cloudflareApiToken =
             fields?.cloudflareApiToken ??
                 getEnvironmentVariable("CLOUDFLARE_API_TOKEN");
-        this.baseUrl = fields?.baseUrl;
+        this.baseUrl =
+            fields?.baseUrl ??
+                `https://api.cloudflare.com/client/v4/accounts/${this.cloudflareAccountId}/ai/run`;
+        if (this.baseUrl.endsWith("/")) {
+            this.baseUrl = this.baseUrl.slice(0, -1);
+        }
     }
     _llmType() {
         return "cloudflare";
@@ -109,7 +114,7 @@ export class ChatCloudflareWorkersAI extends SimpleChatModel {
     /** @ignore */
     async _call(messages, options) {
         this.validateEnvironment();
-        const url = `https://api.cloudflare.com/client/v4/accounts/${this.cloudflareAccountId}/ai/run/${this.model}`;
+        const url = `${this.baseUrl}/${this.model}`;
         const headers = {
             Authorization: `Bearer ${this.cloudflareApiToken}`,
             "Content-Type": "application/json",

package/dist/chat_models/yandex.cjs

@@ -0,0 +1,117 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ChatYandexGPT = void 0;
+const index_js_1 = require("../schema/index.cjs");
+const env_js_1 = require("../util/env.cjs");
+const base_js_1 = require("./base.cjs");
+const apiUrl = "https://llm.api.cloud.yandex.net/llm/v1alpha/chat";
+function _parseChatHistory(history) {
+    const chatHistory = [];
+    let instruction = "";
+    for (const message of history) {
+        if ("content" in message) {
+            if (message._getType() === "human") {
+                chatHistory.push({ role: "user", text: message.content });
+            }
+            else if (message._getType() === "ai") {
+                chatHistory.push({ role: "assistant", text: message.content });
+            }
+            else if (message._getType() === "system") {
+                instruction = message.content;
+            }
+        }
+    }
+    return [chatHistory, instruction];
+}
+class ChatYandexGPT extends base_js_1.BaseChatModel {
+    constructor(fields) {
+        super(fields ?? {});
+        Object.defineProperty(this, "apiKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "iamToken", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "temperature", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: 0.6
+        });
+        Object.defineProperty(this, "maxTokens", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: 1700
+        });
+        Object.defineProperty(this, "model", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "general"
+        });
+        const apiKey = fields?.apiKey ?? (0, env_js_1.getEnvironmentVariable)("YC_API_KEY");
+        const iamToken = fields?.iamToken ?? (0, env_js_1.getEnvironmentVariable)("YC_IAM_TOKEN");
+        if (apiKey === undefined && iamToken === undefined) {
+            throw new Error("Please set the YC_API_KEY or YC_IAM_TOKEN environment variable or pass it to the constructor as the apiKey or iamToken field.");
+        }
+        this.apiKey = apiKey;
+        this.iamToken = iamToken;
+        this.maxTokens = fields?.maxTokens ?? this.maxTokens;
+        this.temperature = fields?.temperature ?? this.temperature;
+        this.model = fields?.model ?? this.model;
+    }
+    _llmType() {
+        return "yandexgpt";
+    }
+    _combineLLMOutput() {
+        return {};
+    }
+    /** @ignore */
+    async _generate(messages, options, _) {
+        const [messageHistory, instruction] = _parseChatHistory(messages);
+        const headers = { "Content-Type": "application/json", Authorization: "" };
+        if (this.apiKey !== undefined) {
+            headers.Authorization = `Api-Key ${this.apiKey}`;
+        }
+        else {
+            headers.Authorization = `Bearer ${this.iamToken}`;
+        }
+        const bodyData = {
+            model: this.model,
+            generationOptions: {
+                temperature: this.temperature,
+                maxTokens: this.maxTokens,
+            },
+            messages: messageHistory,
+            instructionText: instruction,
+        };
+        const response = await fetch(apiUrl, {
+            method: "POST",
+            headers,
+            body: JSON.stringify(bodyData),
+            signal: options?.signal,
+        });
+        if (!response.ok) {
+            throw new Error(`Failed to fetch ${apiUrl} from YandexGPT: ${response.status}`);
+        }
+        const responseData = await response.json();
+        const { result } = responseData;
+        const { text } = result.message;
+        const totalTokens = result.num_tokens;
+        const generations = [
+            { text, message: new index_js_1.AIMessage(text) },
+        ];
+        return {
+            generations,
+            llmOutput: { totalTokens },
+        };
+    }
+}
+exports.ChatYandexGPT = ChatYandexGPT;

package/dist/chat_models/yandex.d.ts

@@ -0,0 +1,16 @@
+import { CallbackManagerForLLMRun } from "../callbacks/manager.js";
+import { YandexGPTInputs } from "../llms/yandex.js";
+import { BaseMessage, ChatResult } from "../schema/index.js";
+import { BaseChatModel } from "./base.js";
+export declare class ChatYandexGPT extends BaseChatModel {
+    apiKey?: string;
+    iamToken?: string;
+    temperature: number;
+    maxTokens: number;
+    model: string;
+    constructor(fields?: YandexGPTInputs);
+    _llmType(): string;
+    _combineLLMOutput?(): {};
+    /** @ignore */
+    _generate(messages: BaseMessage[], options: this["ParsedCallOptions"], _?: CallbackManagerForLLMRun | undefined): Promise<ChatResult>;
+}

package/dist/chat_models/yandex.js

@@ -0,0 +1,113 @@
+import { AIMessage, } from "../schema/index.js";
+import { getEnvironmentVariable } from "../util/env.js";
+import { BaseChatModel } from "./base.js";
+const apiUrl = "https://llm.api.cloud.yandex.net/llm/v1alpha/chat";
+function _parseChatHistory(history) {
+    const chatHistory = [];
+    let instruction = "";
+    for (const message of history) {
+        if ("content" in message) {
+            if (message._getType() === "human") {
+                chatHistory.push({ role: "user", text: message.content });
+            }
+            else if (message._getType() === "ai") {
+                chatHistory.push({ role: "assistant", text: message.content });
+            }
+            else if (message._getType() === "system") {
+                instruction = message.content;
+            }
+        }
+    }
+    return [chatHistory, instruction];
+}
+export class ChatYandexGPT extends BaseChatModel {
+    constructor(fields) {
+        super(fields ?? {});
+        Object.defineProperty(this, "apiKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "iamToken", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "temperature", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: 0.6
+        });
+        Object.defineProperty(this, "maxTokens", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: 1700
+        });
+        Object.defineProperty(this, "model", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "general"
+        });
+        const apiKey = fields?.apiKey ?? getEnvironmentVariable("YC_API_KEY");
+        const iamToken = fields?.iamToken ?? getEnvironmentVariable("YC_IAM_TOKEN");
+        if (apiKey === undefined && iamToken === undefined) {
+            throw new Error("Please set the YC_API_KEY or YC_IAM_TOKEN environment variable or pass it to the constructor as the apiKey or iamToken field.");
+        }
+        this.apiKey = apiKey;
+        this.iamToken = iamToken;
+        this.maxTokens = fields?.maxTokens ?? this.maxTokens;
+        this.temperature = fields?.temperature ?? this.temperature;
+        this.model = fields?.model ?? this.model;
+    }
+    _llmType() {
+        return "yandexgpt";
+    }
+    _combineLLMOutput() {
+        return {};
+    }
+    /** @ignore */
+    async _generate(messages, options, _) {
+        const [messageHistory, instruction] = _parseChatHistory(messages);
+        const headers = { "Content-Type": "application/json", Authorization: "" };
+        if (this.apiKey !== undefined) {
+            headers.Authorization = `Api-Key ${this.apiKey}`;
+        }
+        else {
+            headers.Authorization = `Bearer ${this.iamToken}`;
+        }
+        const bodyData = {
+            model: this.model,
+            generationOptions: {
+                temperature: this.temperature,
+                maxTokens: this.maxTokens,
+            },
+            messages: messageHistory,
+            instructionText: instruction,
+        };
+        const response = await fetch(apiUrl, {
+            method: "POST",
+            headers,
+            body: JSON.stringify(bodyData),
+            signal: options?.signal,
+        });
+        if (!response.ok) {
+            throw new Error(`Failed to fetch ${apiUrl} from YandexGPT: ${response.status}`);
+        }
+        const responseData = await response.json();
+        const { result } = responseData;
+        const { text } = result.message;
+        const totalTokens = result.num_tokens;
+        const generations = [
+            { text, message: new AIMessage(text) },
+        ];
+        return {
+            generations,
+            llmOutput: { totalTokens },
+        };
+    }
+}