langchain 0.0.108 → 0.0.110

package/chat_models/googlepalm.cjs

@@ -0,0 +1 @@
+module.exports = require('../dist/chat_models/googlepalm.cjs');

package/chat_models/googlepalm.d.ts

@@ -0,0 +1 @@
+export * from '../dist/chat_models/googlepalm.js'

package/chat_models/googlepalm.js

@@ -0,0 +1 @@
+export * from '../dist/chat_models/googlepalm.js'

package/dist/callbacks/handlers/tracer.d.ts

@@ -1,4 +1,4 @@
-import { KVMap, BaseRun } from "langchainplus-sdk/schemas";
+import { KVMap, BaseRun } from "langsmith/schemas";
 import { AgentAction, AgentFinish, BaseMessage, ChainValues, LLMResult } from "../../schema/index.js";
 import { Serialized } from "../../load/serializable.js";
 import { BaseCallbackHandler, BaseCallbackHandlerInput, NewTokenIndices } from "../base.js";

package/dist/callbacks/handlers/tracer_langchain.cjs

@@ -1,7 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.LangChainTracer = void 0;
-const langchainplus_sdk_1 = require("langchainplus-sdk");
+const langsmith_1 = require("langsmith");
 const env_js_1 = require("../../util/env.cjs");
 const tracer_js_1 = require("./tracer.cjs");
 class LangChainTracer extends tracer_js_1.BaseTracer {
@@ -37,7 +37,7 @@ class LangChainTracer extends tracer_js_1.BaseTracer {
                 (0, env_js_1.getEnvironmentVariable)("LANGCHAIN_PROJECT") ??
                 (0, env_js_1.getEnvironmentVariable)("LANGCHAIN_SESSION");
         this.exampleId = exampleId;
-        this.client = client ?? new langchainplus_sdk_1.Client({});
+        this.client = client ?? new langsmith_1.Client({});
     }
     async _convertToCreate(run, example_id = undefined) {
         return {

package/dist/callbacks/handlers/tracer_langchain.d.ts

@@ -1,5 +1,5 @@
-import { Client } from "langchainplus-sdk";
-import { BaseRun, RunUpdate as BaseRunUpdate } from "langchainplus-sdk/schemas";
+import { Client } from "langsmith";
+import { BaseRun, RunUpdate as BaseRunUpdate } from "langsmith/schemas";
 import { BaseTracer } from "./tracer.js";
 import { BaseCallbackHandlerInput } from "../base.js";
 export interface Run extends BaseRun {

package/dist/callbacks/handlers/tracer_langchain.js

@@ -1,4 +1,4 @@
-import { Client } from "langchainplus-sdk";
+import { Client } from "langsmith";
 import { getEnvironmentVariable, getRuntimeEnvironment, } from "../../util/env.js";
 import { BaseTracer } from "./tracer.js";
 export class LangChainTracer extends BaseTracer {

package/dist/chat_models/anthropic.cjs

@@ -1,23 +1,49 @@
 "use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ChatAnthropic = void 0;
-const sdk_1 = require("@anthropic-ai/sdk");
+const AnthropicApi = __importStar(require("@anthropic-ai/sdk"));
 const index_js_1 = require("../schema/index.cjs");
 const env_js_1 = require("../util/env.cjs");
 const base_js_1 = require("./base.cjs");
+// Anthropic's 0.5.3 SDK currently has a collision with the default exported class
+// and an exported namespace that causes issues when transpiling to CommonJS
+const AnthropicClientConstructor = AnthropicApi.Anthropic ?? AnthropicApi.default;
 function getAnthropicPromptFromMessage(type) {
     switch (type) {
         case "ai":
-            return sdk_1.AI_PROMPT;
+            return AnthropicApi.AI_PROMPT;
         case "human":
-            return sdk_1.HUMAN_PROMPT;
+            return AnthropicApi.HUMAN_PROMPT;
         case "system":
             return "";
         default:
             throw new Error(`Unknown message type: ${type}`);
     }
 }
-const DEFAULT_STOP_SEQUENCES = [sdk_1.HUMAN_PROMPT];
+const DEFAULT_STOP_SEQUENCES = [AnthropicApi.HUMAN_PROMPT];
 /**
  * Wrapper around Anthropic large language models.
  *
@@ -180,7 +206,7 @@ class ChatAnthropic extends base_js_1.BaseChatModel {
             const messagePrompt = getAnthropicPromptFromMessage(message._getType());
             return `${messagePrompt} ${message.content}`;
         })
-            .join("") + sdk_1.AI_PROMPT);
+            .join("") + AnthropicApi.AI_PROMPT);
     }
     /** @ignore */
     async _generate(messages, options, runManager) {
@@ -193,7 +219,7 @@ class ChatAnthropic extends base_js_1.BaseChatModel {
             prompt: this.formatMessagesAsPrompt(messages),
         }, { signal: options.signal }, runManager);
         const generations = response.completion
-            .split(sdk_1.AI_PROMPT)
+            .split(AnthropicApi.AI_PROMPT)
             .map((message) => ({
             text: message,
             message: new index_js_1.AIMessage(message),
@@ -212,7 +238,7 @@ class ChatAnthropic extends base_js_1.BaseChatModel {
         if (request.stream) {
             if (!this.streamingClient) {
                 const options = this.apiUrl ? { apiUrl: this.apiUrl } : undefined;
-                this.streamingClient = new sdk_1.Anthropic({
+                this.streamingClient = new AnthropicClientConstructor({
                     ...options,
                     apiKey: this.anthropicApiKey,
                 });
@@ -249,7 +275,7 @@ class ChatAnthropic extends base_js_1.BaseChatModel {
         else {
             if (!this.batchClient) {
                 const options = this.apiUrl ? { apiUrl: this.apiUrl } : undefined;
-                this.batchClient = new sdk_1.Anthropic({
+                this.batchClient = new AnthropicClientConstructor({
                     ...options,
                     apiKey: this.anthropicApiKey,
                 });

package/dist/chat_models/anthropic.d.ts

@@ -1,4 +1,3 @@
-import { Anthropic as AnthropicApi } from "@anthropic-ai/sdk";
 import type { CompletionCreateParams } from "@anthropic-ai/sdk/resources/completions";
 import { BaseLanguageModelCallOptions } from "../base_language/index.js";
 import { CallbackManagerForLLMRun } from "../callbacks/manager.js";
@@ -89,7 +88,7 @@ export declare class ChatAnthropic extends BaseChatModel implements AnthropicInp
     invocationParams(options?: this["ParsedCallOptions"]): Omit<CompletionCreateParams, "prompt"> & Kwargs;
     /** @ignore */
     _identifyingParams(): {
-        metadata?: AnthropicApi.Completions.CompletionCreateParams.CompletionRequestNonStreaming.Metadata | AnthropicApi.Completions.CompletionCreateParams.CompletionRequestStreaming.Metadata | undefined;
+        metadata?: CompletionCreateParams.CompletionRequestNonStreaming.Metadata | CompletionCreateParams.CompletionRequestStreaming.Metadata | undefined;
         stream?: boolean | undefined;
         model: string;
         temperature?: number | undefined;
@@ -103,7 +102,7 @@ export declare class ChatAnthropic extends BaseChatModel implements AnthropicInp
      * Get the identifying parameters for the model
      */
     identifyingParams(): {
-        metadata?: AnthropicApi.Completions.CompletionCreateParams.CompletionRequestNonStreaming.Metadata | AnthropicApi.Completions.CompletionCreateParams.CompletionRequestStreaming.Metadata | undefined;
+        metadata?: CompletionCreateParams.CompletionRequestNonStreaming.Metadata | CompletionCreateParams.CompletionRequestStreaming.Metadata | undefined;
         stream?: boolean | undefined;
         model: string;
         temperature?: number | undefined;

package/dist/chat_models/anthropic.js

@@ -1,20 +1,23 @@
-import { AI_PROMPT, Anthropic as AnthropicApi, HUMAN_PROMPT, } from "@anthropic-ai/sdk";
+import * as AnthropicApi from "@anthropic-ai/sdk";
 import { AIMessage, } from "../schema/index.js";
 import { getEnvironmentVariable } from "../util/env.js";
 import { BaseChatModel } from "./base.js";
+// Anthropic's 0.5.3 SDK currently has a collision with the default exported class
+// and an exported namespace that causes issues when transpiling to CommonJS
+const AnthropicClientConstructor = AnthropicApi.Anthropic ?? AnthropicApi.default;
 function getAnthropicPromptFromMessage(type) {
     switch (type) {
         case "ai":
-            return AI_PROMPT;
+            return AnthropicApi.AI_PROMPT;
         case "human":
-            return HUMAN_PROMPT;
+            return AnthropicApi.HUMAN_PROMPT;
         case "system":
             return "";
         default:
             throw new Error(`Unknown message type: ${type}`);
     }
 }
-const DEFAULT_STOP_SEQUENCES = [HUMAN_PROMPT];
+const DEFAULT_STOP_SEQUENCES = [AnthropicApi.HUMAN_PROMPT];
 /**
  * Wrapper around Anthropic large language models.
  *
@@ -177,7 +180,7 @@ export class ChatAnthropic extends BaseChatModel {
             const messagePrompt = getAnthropicPromptFromMessage(message._getType());
             return `${messagePrompt} ${message.content}`;
         })
-            .join("") + AI_PROMPT);
+            .join("") + AnthropicApi.AI_PROMPT);
     }
     /** @ignore */
     async _generate(messages, options, runManager) {
@@ -190,7 +193,7 @@ export class ChatAnthropic extends BaseChatModel {
             prompt: this.formatMessagesAsPrompt(messages),
         }, { signal: options.signal }, runManager);
         const generations = response.completion
-            .split(AI_PROMPT)
+            .split(AnthropicApi.AI_PROMPT)
             .map((message) => ({
             text: message,
             message: new AIMessage(message),
@@ -209,7 +212,7 @@ export class ChatAnthropic extends BaseChatModel {
         if (request.stream) {
             if (!this.streamingClient) {
                 const options = this.apiUrl ? { apiUrl: this.apiUrl } : undefined;
-                this.streamingClient = new AnthropicApi({
+                this.streamingClient = new AnthropicClientConstructor({
                     ...options,
                     apiKey: this.anthropicApiKey,
                 });
@@ -246,7 +249,7 @@ export class ChatAnthropic extends BaseChatModel {
         else {
             if (!this.batchClient) {
                 const options = this.apiUrl ? { apiUrl: this.apiUrl } : undefined;
-                this.batchClient = new AnthropicApi({
+                this.batchClient = new AnthropicClientConstructor({
                     ...options,
                     apiKey: this.anthropicApiKey,
                 });

package/dist/chat_models/googlepalm.cjs

@@ -0,0 +1,167 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ChatGooglePaLM = void 0;
+const generativelanguage_1 = require("@google-ai/generativelanguage");
+const google_auth_library_1 = require("google-auth-library");
+const index_js_1 = require("../schema/index.cjs");
+const env_js_1 = require("../util/env.cjs");
+const base_js_1 = require("./base.cjs");
+class ChatGooglePaLM extends base_js_1.BaseChatModel {
+    get lc_secrets() {
+        return {
+            apiKey: "GOOGLE_PALM_API_KEY",
+        };
+    }
+    constructor(fields) {
+        super(fields ?? {});
+        Object.defineProperty(this, "modelName", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "models/chat-bison-001"
+        });
+        Object.defineProperty(this, "temperature", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        }); // default value chosen based on model
+        Object.defineProperty(this, "topP", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        }); // default value chosen based on model
+        Object.defineProperty(this, "topK", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        }); // default value chosen based on model
+        Object.defineProperty(this, "examples", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: []
+        });
+        Object.defineProperty(this, "apiKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "client", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.modelName = fields?.modelName ?? this.modelName;
+        this.temperature = fields?.temperature ?? this.temperature;
+        if (this.temperature && (this.temperature < 0 || this.temperature > 1)) {
+            throw new Error("`temperature` must be in the range of [0.0,1.0]");
+        }
+        this.topP = fields?.topP ?? this.topP;
+        if (this.topP && this.topP < 0) {
+            throw new Error("`topP` must be a positive integer");
+        }
+        this.topK = fields?.topK ?? this.topK;
+        if (this.topK && this.topK < 0) {
+            throw new Error("`topK` must be a positive integer");
+        }
+        this.examples = fields?.examples ?? this.examples;
+        this.apiKey =
+            fields?.apiKey ?? (0, env_js_1.getEnvironmentVariable)("GOOGLE_PALM_API_KEY");
+        if (!this.apiKey) {
+            throw new Error("Please set an API key for Google Palm 2 in the environment variable GOOGLE_PALM_API_KEY or in the `apiKey` field of the GooglePalm constructor");
+        }
+        this.client = new generativelanguage_1.DiscussServiceClient({
+            authClient: new google_auth_library_1.GoogleAuth().fromAPIKey(this.apiKey),
+        });
+    }
+    _combineLLMOutput() {
+        return [];
+    }
+    _llmType() {
+        return "googlepalm";
+    }
+    async _generate(messages, options, runManager) {
+        const palmMessages = await this.caller.callWithOptions({ signal: options.signal }, this._generateMessage.bind(this), this._mapBaseMessagesToPalmMessages(messages), this._getPalmContextInstruction(messages), this.examples);
+        const chatResult = this._mapPalmMessagesToChatResult(palmMessages);
+        // Google Palm doesn't provide streaming as of now. But to support streaming handlers
+        // we call the handler with entire response text
+        void runManager?.handleLLMNewToken(chatResult.generations.length > 0 ? chatResult.generations[0].text : "");
+        return chatResult;
+    }
+    async _generateMessage(messages, context, examples) {
+        const [palmMessages] = await this.client.generateMessage({
+            candidateCount: 1,
+            model: this.modelName,
+            temperature: this.temperature,
+            topK: this.topK,
+            topP: this.topP,
+            prompt: {
+                context,
+                examples,
+                messages,
+            },
+        });
+        return palmMessages;
+    }
+    _getPalmContextInstruction(messages) {
+        // get the first message and checks if it's a system 'system' messages
+        const systemMessage = messages.length > 0 && messages[0]._getType() === "system"
+            ? messages[0]
+            : undefined;
+        return systemMessage?.content;
+    }
+    _mapBaseMessagesToPalmMessages(messages) {
+        // remove all 'system' messages
+        const nonSystemMessages = messages.filter((m) => m._getType() !== "system");
+        // requires alternate human & ai messages. Throw error if two messages are consecutive
+        nonSystemMessages.forEach((msg, index) => {
+            if (index < 1)
+                return;
+            if (msg._getType() === nonSystemMessages[index - 1]._getType()) {
+                throw new Error(`Google PaLM requires alternate messages between authors`);
+            }
+        });
+        return nonSystemMessages.map((m) => ({
+            author: m.name ?? m._getType(),
+            content: m.content,
+            citationMetadata: {
+                citationSources: m.additional_kwargs.citationSources,
+            },
+        }));
+    }
+    _mapPalmMessagesToChatResult(msgRes) {
+        if (msgRes.candidates &&
+            msgRes.candidates.length > 0 &&
+            msgRes.candidates[0]) {
+            const message = msgRes.candidates[0];
+            return {
+                generations: [
+                    {
+                        text: message.content ?? "",
+                        message: new index_js_1.AIMessage({
+                            content: message.content ?? "",
+                            name: message.author === null ? undefined : message.author,
+                            additional_kwargs: {
+                                citationSources: message.citationMetadata?.citationSources,
+                                filters: msgRes.filters, // content filters applied
+                            },
+                        }),
+                    },
+                ],
+            };
+        }
+        // if rejected or error, return empty generations with reason in filters
+        return {
+            generations: [],
+            llmOutput: {
+                filters: msgRes.filters,
+            },
+        };
+    }
+}
+exports.ChatGooglePaLM = ChatGooglePaLM;

package/dist/chat_models/googlepalm.d.ts

@@ -0,0 +1,72 @@
+import type { protos } from "@google-ai/generativelanguage";
+import { CallbackManagerForLLMRun } from "../callbacks/manager.js";
+import { BaseMessage, ChatResult } from "../schema/index.js";
+import { BaseChatModel, BaseChatModelParams } from "./base.js";
+export interface GooglePaLMChatInput extends BaseChatModelParams {
+    /**
+     * Model Name to use
+     *
+     * Note: The format must follow the pattern - `models/{model}`
+     */
+    modelName?: string;
+    /**
+     * Controls the randomness of the output.
+     *
+     * Values can range from [0.0,1.0], inclusive. A value closer to 1.0
+     * will produce responses that are more varied and creative, while
+     * a value closer to 0.0 will typically result in less surprising
+     * responses from the model.
+     *
+     * Note: The default value varies by model
+     */
+    temperature?: number;
+    /**
+     * Top-p changes how the model selects tokens for output.
+     *
+     * Tokens are selected from most probable to least until the sum
+     * of their probabilities equals the top-p value.
+     *
+     * For example, if tokens A, B, and C have a probability of
+     * .3, .2, and .1 and the top-p value is .5, then the model will
+     * select either A or B as the next token (using temperature).
+     *
+     * Note: The default value varies by model
+     */
+    topP?: number;
+    /**
+     * Top-k changes how the model selects tokens for output.
+     *
+     * A top-k of 1 means the selected token is the most probable among
+     * all tokens in the model’s vocabulary (also called greedy decoding),
+     * while a top-k of 3 means that the next token is selected from
+     * among the 3 most probable tokens (using temperature).
+     *
+     * Note: The default value varies by model
+     */
+    topK?: number;
+    examples?: protos.google.ai.generativelanguage.v1beta2.IExample[];
+    /**
+     * Google Palm API key to use
+     */
+    apiKey?: string;
+}
+export declare class ChatGooglePaLM extends BaseChatModel implements GooglePaLMChatInput {
+    get lc_secrets(): {
+        [key: string]: string;
+    } | undefined;
+    modelName: string;
+    temperature?: number;
+    topP?: number;
+    topK?: number;
+    examples: protos.google.ai.generativelanguage.v1beta2.IExample[];
+    apiKey?: string;
+    private client;
+    constructor(fields?: GooglePaLMChatInput);
+    _combineLLMOutput(): never[];
+    _llmType(): string;
+    _generate(messages: BaseMessage[], options: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): Promise<ChatResult>;
+    protected _generateMessage(messages: protos.google.ai.generativelanguage.v1beta2.IMessage[], context?: string, examples?: protos.google.ai.generativelanguage.v1beta2.IExample[]): Promise<protos.google.ai.generativelanguage.v1beta2.IGenerateMessageResponse>;
+    protected _getPalmContextInstruction(messages: BaseMessage[]): string | undefined;
+    protected _mapBaseMessagesToPalmMessages(messages: BaseMessage[]): protos.google.ai.generativelanguage.v1beta2.IMessage[];
+    protected _mapPalmMessagesToChatResult(msgRes: protos.google.ai.generativelanguage.v1beta2.IGenerateMessageResponse): ChatResult;
+}

package/dist/chat_models/googlepalm.js

@@ -0,0 +1,163 @@
+import { DiscussServiceClient } from "@google-ai/generativelanguage";
+import { GoogleAuth } from "google-auth-library";
+import { AIMessage } from "../schema/index.js";
+import { getEnvironmentVariable } from "../util/env.js";
+import { BaseChatModel } from "./base.js";
+export class ChatGooglePaLM extends BaseChatModel {
+    get lc_secrets() {
+        return {
+            apiKey: "GOOGLE_PALM_API_KEY",
+        };
+    }
+    constructor(fields) {
+        super(fields ?? {});
+        Object.defineProperty(this, "modelName", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "models/chat-bison-001"
+        });
+        Object.defineProperty(this, "temperature", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        }); // default value chosen based on model
+        Object.defineProperty(this, "topP", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        }); // default value chosen based on model
+        Object.defineProperty(this, "topK", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        }); // default value chosen based on model
+        Object.defineProperty(this, "examples", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: []
+        });
+        Object.defineProperty(this, "apiKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "client", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.modelName = fields?.modelName ?? this.modelName;
+        this.temperature = fields?.temperature ?? this.temperature;
+        if (this.temperature && (this.temperature < 0 || this.temperature > 1)) {
+            throw new Error("`temperature` must be in the range of [0.0,1.0]");
+        }
+        this.topP = fields?.topP ?? this.topP;
+        if (this.topP && this.topP < 0) {
+            throw new Error("`topP` must be a positive integer");
+        }
+        this.topK = fields?.topK ?? this.topK;
+        if (this.topK && this.topK < 0) {
+            throw new Error("`topK` must be a positive integer");
+        }
+        this.examples = fields?.examples ?? this.examples;
+        this.apiKey =
+            fields?.apiKey ?? getEnvironmentVariable("GOOGLE_PALM_API_KEY");
+        if (!this.apiKey) {
+            throw new Error("Please set an API key for Google Palm 2 in the environment variable GOOGLE_PALM_API_KEY or in the `apiKey` field of the GooglePalm constructor");
+        }
+        this.client = new DiscussServiceClient({
+            authClient: new GoogleAuth().fromAPIKey(this.apiKey),
+        });
+    }
+    _combineLLMOutput() {
+        return [];
+    }
+    _llmType() {
+        return "googlepalm";
+    }
+    async _generate(messages, options, runManager) {
+        const palmMessages = await this.caller.callWithOptions({ signal: options.signal }, this._generateMessage.bind(this), this._mapBaseMessagesToPalmMessages(messages), this._getPalmContextInstruction(messages), this.examples);
+        const chatResult = this._mapPalmMessagesToChatResult(palmMessages);
+        // Google Palm doesn't provide streaming as of now. But to support streaming handlers
+        // we call the handler with entire response text
+        void runManager?.handleLLMNewToken(chatResult.generations.length > 0 ? chatResult.generations[0].text : "");
+        return chatResult;
+    }
+    async _generateMessage(messages, context, examples) {
+        const [palmMessages] = await this.client.generateMessage({
+            candidateCount: 1,
+            model: this.modelName,
+            temperature: this.temperature,
+            topK: this.topK,
+            topP: this.topP,
+            prompt: {
+                context,
+                examples,
+                messages,
+            },
+        });
+        return palmMessages;
+    }
+    _getPalmContextInstruction(messages) {
+        // get the first message and checks if it's a system 'system' messages
+        const systemMessage = messages.length > 0 && messages[0]._getType() === "system"
+            ? messages[0]
+            : undefined;
+        return systemMessage?.content;
+    }
+    _mapBaseMessagesToPalmMessages(messages) {
+        // remove all 'system' messages
+        const nonSystemMessages = messages.filter((m) => m._getType() !== "system");
+        // requires alternate human & ai messages. Throw error if two messages are consecutive
+        nonSystemMessages.forEach((msg, index) => {
+            if (index < 1)
+                return;
+            if (msg._getType() === nonSystemMessages[index - 1]._getType()) {
+                throw new Error(`Google PaLM requires alternate messages between authors`);
+            }
+        });
+        return nonSystemMessages.map((m) => ({
+            author: m.name ?? m._getType(),
+            content: m.content,
+            citationMetadata: {
+                citationSources: m.additional_kwargs.citationSources,
+            },
+        }));
+    }
+    _mapPalmMessagesToChatResult(msgRes) {
+        if (msgRes.candidates &&
+            msgRes.candidates.length > 0 &&
+            msgRes.candidates[0]) {
+            const message = msgRes.candidates[0];
+            return {
+                generations: [
+                    {
+                        text: message.content ?? "",
+                        message: new AIMessage({
+                            content: message.content ?? "",
+                            name: message.author === null ? undefined : message.author,
+                            additional_kwargs: {
+                                citationSources: message.citationMetadata?.citationSources,
+                                filters: msgRes.filters, // content filters applied
+                            },
+                        }),
+                    },
+                ],
+            };
+        }
+        // if rejected or error, return empty generations with reason in filters
+        return {
+            generations: [],
+            llmOutput: {
+                filters: msgRes.filters,
+            },
+        };
+    }
+}

package/dist/chat_models/openai.cjs

@@ -589,6 +589,9 @@ class PromptLayerChatOpenAI extends ChatOpenAI {
             else if (message._getType() === "ai") {
                 messageDict = { role: "assistant", content: message.content };
             }
+            else if (message._getType() === "function") {
+                messageDict = { role: "assistant", content: message.content };
+            }
             else if (message._getType() === "system") {
                 messageDict = { role: "system", content: message.content };
             }

package/dist/chat_models/openai.js

@@ -582,6 +582,9 @@ export class PromptLayerChatOpenAI extends ChatOpenAI {
             else if (message._getType() === "ai") {
                 messageDict = { role: "assistant", content: message.content };
             }
+            else if (message._getType() === "function") {
+                messageDict = { role: "assistant", content: message.content };
+            }
             else if (message._getType() === "system") {
                 messageDict = { role: "system", content: message.content };
             }

package/dist/client/index.cjs

@@ -1,5 +1,5 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.runOnDataset = void 0;
-var langchainplus_js_1 = require("./langchainplus.cjs");
-Object.defineProperty(exports, "runOnDataset", { enumerable: true, get: function () { return langchainplus_js_1.runOnDataset; } });
+var langsmith_js_1 = require("./langsmith.cjs");
+Object.defineProperty(exports, "runOnDataset", { enumerable: true, get: function () { return langsmith_js_1.runOnDataset; } });

package/dist/client/index.d.ts

@@ -1 +1 @@
-export { DatasetRunResults, runOnDataset } from "./langchainplus.js";
+export { DatasetRunResults, runOnDataset } from "./langsmith.js";

package/dist/client/index.js

@@ -1 +1 @@
-export { runOnDataset } from "./langchainplus.js";
+export { runOnDataset } from "./langsmith.js";