@n8n/n8n-nodes-langchain 1.85.2 → 1.86.0

package/dist/nodes/llms/N8nLlmTracing.js

@@ -1,137 +1,171 @@
 "use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.N8nLlmTracing = void 0;
-const base_1 = require("@langchain/core/callbacks/base");
-const base_2 = require("@langchain/core/language_models/base");
-const tiktoken_1 = require("@langchain/core/utils/tiktoken");
-const lodash_1 = require("lodash");
-const n8n_workflow_1 = require("n8n-workflow");
-const helpers_1 = require("../../utils/helpers");
-const TIKTOKEN_ESTIMATE_MODEL = 'gpt-4o';
-class N8nLlmTracing extends base_1.BaseCallbackHandler {
-    constructor(executionFunctions, options) {
-        super();
-        this.executionFunctions = executionFunctions;
-        this.name = 'N8nLlmTracing';
-        this.awaitHandlers = true;
-        this.connectionType = n8n_workflow_1.NodeConnectionTypes.AiLanguageModel;
-        this.promptTokensEstimate = 0;
-        this.completionTokensEstimate = 0;
-        this.runsMap = {};
-        this.options = {
-            tokensUsageParser: (llmOutput) => {
-                const completionTokens = llmOutput?.tokenUsage?.completionTokens ?? 0;
-                const promptTokens = llmOutput?.tokenUsage?.promptTokens ?? 0;
-                return {
-                    completionTokens,
-                    promptTokens,
-                    totalTokens: completionTokens + promptTokens,
-                };
-            },
-            errorDescriptionMapper: (error) => error.description,
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var N8nLlmTracing_exports = {};
+__export(N8nLlmTracing_exports, {
+  N8nLlmTracing: () => N8nLlmTracing
+});
+module.exports = __toCommonJS(N8nLlmTracing_exports);
+var import_base = require("@langchain/core/callbacks/base");
+var import_base2 = require("@langchain/core/language_models/base");
+var import_tiktoken = require("@langchain/core/utils/tiktoken");
+var import_lodash = require("lodash");
+var import_n8n_workflow = require("n8n-workflow");
+var import_helpers = require("../../utils/helpers");
+const TIKTOKEN_ESTIMATE_MODEL = "gpt-4o";
+class N8nLlmTracing extends import_base.BaseCallbackHandler {
+  constructor(executionFunctions, options) {
+    super();
+    this.executionFunctions = executionFunctions;
+    this.name = "N8nLlmTracing";
+    // This flag makes sure that LangChain will wait for the handlers to finish before continuing
+    // This is crucial for the handleLLMError handler to work correctly (it should be called before the error is propagated to the root node)
+    this.awaitHandlers = true;
+    this.connectionType = import_n8n_workflow.NodeConnectionTypes.AiLanguageModel;
+    this.promptTokensEstimate = 0;
+    this.completionTokensEstimate = 0;
+    /**
+     * A map to associate LLM run IDs to run details.
+     * Key: Unique identifier for each LLM run (run ID)
+     * Value: RunDetails object
+     *
+     */
+    this.runsMap = {};
+    this.options = {
+      // Default(OpenAI format) parser
+      tokensUsageParser: (llmOutput) => {
+        const completionTokens = llmOutput?.tokenUsage?.completionTokens ?? 0;
+        const promptTokens = llmOutput?.tokenUsage?.promptTokens ?? 0;
+        return {
+          completionTokens,
+          promptTokens,
+          totalTokens: completionTokens + promptTokens
         };
-        this.options = { ...this.options, ...options };
+      },
+      errorDescriptionMapper: (error) => error.description
+    };
+    this.options = { ...this.options, ...options };
+  }
+  async estimateTokensFromGeneration(generations) {
+    const messages = generations.flatMap((gen) => gen.map((g) => g.text));
+    return await this.estimateTokensFromStringList(messages);
+  }
+  async estimateTokensFromStringList(list) {
+    const embeddingModel = (0, import_base2.getModelNameForTiktoken)(TIKTOKEN_ESTIMATE_MODEL);
+    const encoder = await (0, import_tiktoken.encodingForModel)(embeddingModel);
+    const encodedListLength = await Promise.all(
+      list.map(async (text) => encoder.encode(text).length)
+    );
+    return encodedListLength.reduce((acc, curr) => acc + curr, 0);
+  }
+  async handleLLMEnd(output, runId) {
+    const runDetails = this.runsMap[runId] ?? { index: Object.keys(this.runsMap).length };
+    output.generations = output.generations.map(
+      (gen) => gen.map((g) => (0, import_lodash.pick)(g, ["text", "generationInfo"]))
+    );
+    const tokenUsageEstimate = {
+      completionTokens: 0,
+      promptTokens: 0,
+      totalTokens: 0
+    };
+    const tokenUsage = this.options.tokensUsageParser(output.llmOutput);
+    if (output.generations.length > 0) {
+      tokenUsageEstimate.completionTokens = await this.estimateTokensFromGeneration(
+        output.generations
+      );
+      tokenUsageEstimate.promptTokens = this.promptTokensEstimate;
+      tokenUsageEstimate.totalTokens = tokenUsageEstimate.completionTokens + this.promptTokensEstimate;
     }
-    async estimateTokensFromGeneration(generations) {
-        const messages = generations.flatMap((gen) => gen.map((g) => g.text));
-        return await this.estimateTokensFromStringList(messages);
+    const response = {
+      response: { generations: output.generations }
+    };
+    if (tokenUsage.completionTokens > 0) {
+      response.tokenUsage = tokenUsage;
+    } else {
+      response.tokenUsageEstimate = tokenUsageEstimate;
     }
-    async estimateTokensFromStringList(list) {
-        const embeddingModel = (0, base_2.getModelNameForTiktoken)(TIKTOKEN_ESTIMATE_MODEL);
-        const encoder = await (0, tiktoken_1.encodingForModel)(embeddingModel);
-        const encodedListLength = await Promise.all(list.map(async (text) => encoder.encode(text).length));
-        return encodedListLength.reduce((acc, curr) => acc + curr, 0);
-    }
-    async handleLLMEnd(output, runId) {
-        const runDetails = this.runsMap[runId] ?? { index: Object.keys(this.runsMap).length };
-        output.generations = output.generations.map((gen) => gen.map((g) => (0, lodash_1.pick)(g, ['text', 'generationInfo'])));
-        const tokenUsageEstimate = {
-            completionTokens: 0,
-            promptTokens: 0,
-            totalTokens: 0,
-        };
-        const tokenUsage = this.options.tokensUsageParser(output.llmOutput);
-        if (output.generations.length > 0) {
-            tokenUsageEstimate.completionTokens = await this.estimateTokensFromGeneration(output.generations);
-            tokenUsageEstimate.promptTokens = this.promptTokensEstimate;
-            tokenUsageEstimate.totalTokens =
-                tokenUsageEstimate.completionTokens + this.promptTokensEstimate;
-        }
-        const response = {
-            response: { generations: output.generations },
-        };
-        if (tokenUsage.completionTokens > 0) {
-            response.tokenUsage = tokenUsage;
-        }
-        else {
-            response.tokenUsageEstimate = tokenUsageEstimate;
-        }
-        const parsedMessages = typeof runDetails.messages === 'string'
-            ? runDetails.messages
-            : runDetails.messages.map((message) => {
-                if (typeof message === 'string')
-                    return message;
-                if (typeof message?.toJSON === 'function')
-                    return message.toJSON();
-                return message;
-            });
-        this.executionFunctions.addOutputData(this.connectionType, runDetails.index, [
-            [{ json: { ...response } }],
-        ]);
-        (0, helpers_1.logAiEvent)(this.executionFunctions, 'ai-llm-generated-output', {
-            messages: parsedMessages,
-            options: runDetails.options,
-            response,
-        });
-    }
-    async handleLLMStart(llm, prompts, runId) {
-        const estimatedTokens = await this.estimateTokensFromStringList(prompts);
-        const options = llm.type === 'constructor' ? llm.kwargs : llm;
-        const { index } = this.executionFunctions.addInputData(this.connectionType, [
-            [
-                {
-                    json: {
-                        messages: prompts,
-                        estimatedTokens,
-                        options,
-                    },
-                },
-            ],
-        ]);
-        this.runsMap[runId] = {
-            index,
-            options,
+    const parsedMessages = typeof runDetails.messages === "string" ? runDetails.messages : runDetails.messages.map((message) => {
+      if (typeof message === "string") return message;
+      if (typeof message?.toJSON === "function") return message.toJSON();
+      return message;
+    });
+    this.executionFunctions.addOutputData(this.connectionType, runDetails.index, [
+      [{ json: { ...response } }]
+    ]);
+    (0, import_helpers.logAiEvent)(this.executionFunctions, "ai-llm-generated-output", {
+      messages: parsedMessages,
+      options: runDetails.options,
+      response
+    });
+  }
+  async handleLLMStart(llm, prompts, runId) {
+    const estimatedTokens = await this.estimateTokensFromStringList(prompts);
+    const options = llm.type === "constructor" ? llm.kwargs : llm;
+    const { index } = this.executionFunctions.addInputData(this.connectionType, [
+      [
+        {
+          json: {
             messages: prompts,
-        };
-        this.promptTokensEstimate = estimatedTokens;
-    }
-    async handleLLMError(error, runId, parentRunId) {
-        const runDetails = this.runsMap[runId] ?? { index: Object.keys(this.runsMap).length };
-        if (typeof error === 'object' && error?.hasOwnProperty('headers')) {
-            const errorWithHeaders = error;
-            Object.keys(errorWithHeaders.headers).forEach((key) => {
-                if (!key.startsWith('x-')) {
-                    delete errorWithHeaders.headers[key];
-                }
-            });
+            estimatedTokens,
+            options
+          }
         }
-        if (error instanceof n8n_workflow_1.NodeError) {
-            if (this.options.errorDescriptionMapper) {
-                error.description = this.options.errorDescriptionMapper(error);
-            }
-            this.executionFunctions.addOutputData(this.connectionType, runDetails.index, error);
+      ]
+    ]);
+    this.runsMap[runId] = {
+      index,
+      options,
+      messages: prompts
+    };
+    this.promptTokensEstimate = estimatedTokens;
+  }
+  async handleLLMError(error, runId, parentRunId) {
+    const runDetails = this.runsMap[runId] ?? { index: Object.keys(this.runsMap).length };
+    if (typeof error === "object" && error?.hasOwnProperty("headers")) {
+      const errorWithHeaders = error;
+      Object.keys(errorWithHeaders.headers).forEach((key) => {
+        if (!key.startsWith("x-")) {
+          delete errorWithHeaders.headers[key];
         }
-        else {
-            this.executionFunctions.addOutputData(this.connectionType, runDetails.index, new n8n_workflow_1.NodeOperationError(this.executionFunctions.getNode(), error, {
-                functionality: 'configuration-node',
-            }));
-        }
-        (0, helpers_1.logAiEvent)(this.executionFunctions, 'ai-llm-errored', {
-            error: Object.keys(error).length === 0 ? error.toString() : error,
-            runId,
-            parentRunId,
-        });
+      });
+    }
+    if (error instanceof import_n8n_workflow.NodeError) {
+      if (this.options.errorDescriptionMapper) {
+        error.description = this.options.errorDescriptionMapper(error);
+      }
+      this.executionFunctions.addOutputData(this.connectionType, runDetails.index, error);
+    } else {
+      this.executionFunctions.addOutputData(
+        this.connectionType,
+        runDetails.index,
+        new import_n8n_workflow.NodeOperationError(this.executionFunctions.getNode(), error, {
+          functionality: "configuration-node"
+        })
+      );
     }
+    (0, import_helpers.logAiEvent)(this.executionFunctions, "ai-llm-errored", {
+      error: Object.keys(error).length === 0 ? error.toString() : error,
+      runId,
+      parentRunId
+    });
+  }
 }
-exports.N8nLlmTracing = N8nLlmTracing;
-//# sourceMappingURL=N8nLlmTracing.js.map
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  N8nLlmTracing
+});

package/dist/nodes/llms/gemini-common/additional-options.js

@@ -1,83 +1,107 @@
 "use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.additionalOptions = void 0;
-const safety_options_1 = require("./safety-options");
-exports.additionalOptions = {
-    displayName: 'Options',
-    name: 'options',
-    placeholder: 'Add Option',
-    description: 'Additional options to add',
-    type: 'collection',
-    default: {},
-    options: [
-        {
-            displayName: 'Maximum Number of Tokens',
-            name: 'maxOutputTokens',
-            default: 2048,
-            description: 'The maximum number of tokens to generate in the completion',
-            type: 'number',
-        },
-        {
-            displayName: 'Sampling Temperature',
-            name: 'temperature',
-            default: 0.4,
-            typeOptions: { maxValue: 1, minValue: 0, numberPrecision: 1 },
-            description: 'Controls randomness: Lowering results in less random completions. As the temperature approaches zero, the model will become deterministic and repetitive.',
-            type: 'number',
-        },
-        {
-            displayName: 'Top K',
-            name: 'topK',
-            default: 32,
-            typeOptions: { maxValue: 40, minValue: -1, numberPrecision: 1 },
-            description: 'Used to remove "long tail" low probability responses. Defaults to -1, which disables it.',
-            type: 'number',
-        },
-        {
-            displayName: 'Top P',
-            name: 'topP',
-            default: 1,
-            typeOptions: { maxValue: 1, minValue: 0, numberPrecision: 1 },
-            description: 'Controls diversity via nucleus sampling: 0.5 means half of all likelihood-weighted options are considered. We generally recommend altering this or temperature but not both.',
-            type: 'number',
-        },
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var additional_options_exports = {};
+__export(additional_options_exports, {
+  additionalOptions: () => additionalOptions
+});
+module.exports = __toCommonJS(additional_options_exports);
+var import_safety_options = require("./safety-options");
+const additionalOptions = {
+  displayName: "Options",
+  name: "options",
+  placeholder: "Add Option",
+  description: "Additional options to add",
+  type: "collection",
+  default: {},
+  options: [
+    {
+      displayName: "Maximum Number of Tokens",
+      name: "maxOutputTokens",
+      default: 2048,
+      description: "The maximum number of tokens to generate in the completion",
+      type: "number"
+    },
+    {
+      displayName: "Sampling Temperature",
+      name: "temperature",
+      default: 0.4,
+      typeOptions: { maxValue: 1, minValue: 0, numberPrecision: 1 },
+      description: "Controls randomness: Lowering results in less random completions. As the temperature approaches zero, the model will become deterministic and repetitive.",
+      type: "number"
+    },
+    {
+      displayName: "Top K",
+      name: "topK",
+      default: 32,
+      typeOptions: { maxValue: 40, minValue: -1, numberPrecision: 1 },
+      description: 'Used to remove "long tail" low probability responses. Defaults to -1, which disables it.',
+      type: "number"
+    },
+    {
+      displayName: "Top P",
+      name: "topP",
+      default: 1,
+      typeOptions: { maxValue: 1, minValue: 0, numberPrecision: 1 },
+      description: "Controls diversity via nucleus sampling: 0.5 means half of all likelihood-weighted options are considered. We generally recommend altering this or temperature but not both.",
+      type: "number"
+    },
+    // Safety Settings
+    {
+      displayName: "Safety Settings",
+      name: "safetySettings",
+      type: "fixedCollection",
+      typeOptions: { multipleValues: true },
+      default: {
+        values: {
+          category: import_safety_options.harmCategories[0].name,
+          threshold: import_safety_options.harmThresholds[0].name
+        }
+      },
+      placeholder: "Add Option",
+      options: [
         {
-            displayName: 'Safety Settings',
-            name: 'safetySettings',
-            type: 'fixedCollection',
-            typeOptions: { multipleValues: true },
-            default: {
-                values: {
-                    category: safety_options_1.harmCategories[0].name,
-                    threshold: safety_options_1.harmThresholds[0].name,
-                },
+          name: "values",
+          displayName: "Values",
+          values: [
+            {
+              displayName: "Safety Category",
+              name: "category",
+              type: "options",
+              description: "The category of harmful content to block",
+              default: "HARM_CATEGORY_UNSPECIFIED",
+              options: import_safety_options.harmCategories
             },
-            placeholder: 'Add Option',
-            options: [
-                {
-                    name: 'values',
-                    displayName: 'Values',
-                    values: [
-                        {
-                            displayName: 'Safety Category',
-                            name: 'category',
-                            type: 'options',
-                            description: 'The category of harmful content to block',
-                            default: 'HARM_CATEGORY_UNSPECIFIED',
-                            options: safety_options_1.harmCategories,
-                        },
-                        {
-                            displayName: 'Safety Threshold',
-                            name: 'threshold',
-                            type: 'options',
-                            description: 'The threshold of harmful content to block',
-                            default: 'HARM_BLOCK_THRESHOLD_UNSPECIFIED',
-                            options: safety_options_1.harmThresholds,
-                        },
-                    ],
-                },
-            ],
-        },
-    ],
+            {
+              displayName: "Safety Threshold",
+              name: "threshold",
+              type: "options",
+              description: "The threshold of harmful content to block",
+              default: "HARM_BLOCK_THRESHOLD_UNSPECIFIED",
+              options: import_safety_options.harmThresholds
+            }
+          ]
+        }
+      ]
+    }
+  ]
 };
-//# sourceMappingURL=additional-options.js.map
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  additionalOptions
+});

package/dist/nodes/llms/gemini-common/safety-options.js

@@ -1,53 +1,78 @@
 "use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.harmThresholds = exports.harmCategories = void 0;
-exports.harmCategories = [
-    {
-        value: 'HARM_CATEGORY_HARASSMENT',
-        name: 'HARM_CATEGORY_HARASSMENT',
-        description: 'Harassment content',
-    },
-    {
-        value: 'HARM_CATEGORY_HATE_SPEECH',
-        name: 'HARM_CATEGORY_HATE_SPEECH',
-        description: 'Hate speech and content',
-    },
-    {
-        value: 'HARM_CATEGORY_SEXUALLY_EXPLICIT',
-        name: 'HARM_CATEGORY_SEXUALLY_EXPLICIT',
-        description: 'Sexually explicit content',
-    },
-    {
-        value: 'HARM_CATEGORY_DANGEROUS_CONTENT',
-        name: 'HARM_CATEGORY_DANGEROUS_CONTENT',
-        description: 'Dangerous content',
-    },
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var safety_options_exports = {};
+__export(safety_options_exports, {
+  harmCategories: () => harmCategories,
+  harmThresholds: () => harmThresholds
+});
+module.exports = __toCommonJS(safety_options_exports);
+const harmCategories = [
+  {
+    value: "HARM_CATEGORY_HARASSMENT",
+    name: "HARM_CATEGORY_HARASSMENT",
+    description: "Harassment content"
+  },
+  {
+    value: "HARM_CATEGORY_HATE_SPEECH",
+    name: "HARM_CATEGORY_HATE_SPEECH",
+    description: "Hate speech and content"
+  },
+  {
+    value: "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+    name: "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+    description: "Sexually explicit content"
+  },
+  {
+    value: "HARM_CATEGORY_DANGEROUS_CONTENT",
+    name: "HARM_CATEGORY_DANGEROUS_CONTENT",
+    description: "Dangerous content"
+  }
 ];
-exports.harmThresholds = [
-    {
-        value: 'HARM_BLOCK_THRESHOLD_UNSPECIFIED',
-        name: 'HARM_BLOCK_THRESHOLD_UNSPECIFIED',
-        description: 'Threshold is unspecified',
-    },
-    {
-        value: 'BLOCK_LOW_AND_ABOVE',
-        name: 'BLOCK_LOW_AND_ABOVE',
-        description: 'Content with NEGLIGIBLE will be allowed',
-    },
-    {
-        value: 'BLOCK_MEDIUM_AND_ABOVE',
-        name: 'BLOCK_MEDIUM_AND_ABOVE',
-        description: 'Content with NEGLIGIBLE and LOW will be allowed',
-    },
-    {
-        value: 'BLOCK_ONLY_HIGH',
-        name: 'BLOCK_ONLY_HIGH',
-        description: 'Content with NEGLIGIBLE, LOW, and MEDIUM will be allowed',
-    },
-    {
-        value: 'BLOCK_NONE',
-        name: 'BLOCK_NONE',
-        description: 'All content will be allowed',
-    },
+const harmThresholds = [
+  {
+    value: "HARM_BLOCK_THRESHOLD_UNSPECIFIED",
+    name: "HARM_BLOCK_THRESHOLD_UNSPECIFIED",
+    description: "Threshold is unspecified"
+  },
+  {
+    value: "BLOCK_LOW_AND_ABOVE",
+    name: "BLOCK_LOW_AND_ABOVE",
+    description: "Content with NEGLIGIBLE will be allowed"
+  },
+  {
+    value: "BLOCK_MEDIUM_AND_ABOVE",
+    name: "BLOCK_MEDIUM_AND_ABOVE",
+    description: "Content with NEGLIGIBLE and LOW will be allowed"
+  },
+  {
+    value: "BLOCK_ONLY_HIGH",
+    name: "BLOCK_ONLY_HIGH",
+    description: "Content with NEGLIGIBLE, LOW, and MEDIUM will be allowed"
+  },
+  {
+    value: "BLOCK_NONE",
+    name: "BLOCK_NONE",
+    description: "All content will be allowed"
+  }
 ];
-//# sourceMappingURL=safety-options.js.map
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  harmCategories,
+  harmThresholds
+});