langchain 0.0.203 → 0.0.204-rc.0

package/dist/chat_models/portkey.cjs

@@ -1,169 +1,17 @@
 "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 __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.PortkeyChat = void 0;
-const portkey_js_1 = require("../llms/portkey.cjs");
-const index_js_1 = require("../schema/index.cjs");
-const base_js_1 = require("./base.cjs");
-function portkeyResponseToChatMessage(message) {
-    switch (message.role) {
-        case "user":
-            return new index_js_1.HumanMessage(message.content || "");
-        case "assistant":
-            return new index_js_1.AIMessage(message.content || "");
-        case "system":
-            return new index_js_1.SystemMessage(message.content || "");
-        default:
-            return new index_js_1.ChatMessage(message.content || "", message.role ?? "unknown");
-    }
-}
-function _convertDeltaToMessageChunk(
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-delta) {
-    const { role } = delta;
-    const content = delta.content ?? "";
-    let additional_kwargs;
-    if (delta.function_call) {
-        additional_kwargs = {
-            function_call: delta.function_call,
-        };
-    }
-    else {
-        additional_kwargs = {};
-    }
-    if (role === "user") {
-        return new index_js_1.HumanMessageChunk({ content });
-    }
-    else if (role === "assistant") {
-        return new index_js_1.AIMessageChunk({ content, additional_kwargs });
-    }
-    else if (role === "system") {
-        return new index_js_1.SystemMessageChunk({ content });
-    }
-    else if (role === "function") {
-        return new index_js_1.FunctionMessageChunk({
-            content,
-            additional_kwargs,
-            name: delta.name,
-        });
-    }
-    else {
-        return new index_js_1.ChatMessageChunk({ content, role });
-    }
-}
-class PortkeyChat extends base_js_1.BaseChatModel {
-    constructor(init) {
-        super(init ?? {});
-        Object.defineProperty(this, "apiKey", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: undefined
-        });
-        Object.defineProperty(this, "baseURL", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: undefined
-        });
-        Object.defineProperty(this, "mode", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: undefined
-        });
-        Object.defineProperty(this, "llms", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: undefined
-        });
-        Object.defineProperty(this, "session", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.apiKey = init?.apiKey;
-        this.baseURL = init?.baseURL;
-        this.mode = init?.mode;
-        this.llms = init?.llms;
-        this.session = (0, portkey_js_1.getPortkeySession)({
-            apiKey: this.apiKey,
-            baseURL: this.baseURL,
-            llms: this.llms,
-            mode: this.mode,
-        });
-    }
-    _llmType() {
-        return "portkey";
-    }
-    async _generate(messages, options, _) {
-        const messagesList = messages.map((message) => {
-            if (typeof message.content !== "string") {
-                throw new Error("PortkeyChat does not support non-string message content.");
-            }
-            return {
-                role: message._getType(),
-                content: message.content,
-            };
-        });
-        const response = await this.session.portkey.chatCompletions.create({
-            messages: messagesList,
-            ...options,
-            stream: false,
-        });
-        const generations = [];
-        for (const data of response.choices ?? []) {
-            const text = data.message?.content ?? "";
-            const generation = {
-                text,
-                message: portkeyResponseToChatMessage(data.message ?? {}),
-            };
-            if (data.finish_reason) {
-                generation.generationInfo = { finish_reason: data.finish_reason };
-            }
-            generations.push(generation);
-        }
-        return {
-            generations,
-        };
-    }
-    async *_streamResponseChunks(messages, options, runManager) {
-        const messagesList = messages.map((message) => {
-            if (typeof message.content !== "string") {
-                throw new Error("PortkeyChat does not support non-string message content.");
-            }
-            return {
-                role: message._getType(),
-                content: message.content,
-            };
-        });
-        const response = await this.session.portkey.chatCompletions.create({
-            messages: messagesList,
-            ...options,
-            stream: true,
-        });
-        for await (const data of response) {
-            const choice = data?.choices[0];
-            if (!choice) {
-                continue;
-            }
-            const chunk = new index_js_1.ChatGenerationChunk({
-                message: _convertDeltaToMessageChunk(choice.delta ?? {}),
-                text: choice.message?.content ?? "",
-                generationInfo: {
-                    finishReason: choice.finish_reason,
-                },
-            });
-            yield chunk;
-            void runManager?.handleLLMNewToken(chunk.text ?? "");
-        }
-        if (options.signal?.aborted) {
-            throw new Error("AbortError");
-        }
-    }
-    _combineLLMOutput() {
-        return {};
-    }
-}
-exports.PortkeyChat = PortkeyChat;
+__exportStar(require("@langchain/community/chat_models/portkey"), exports);

package/dist/chat_models/portkey.d.ts

@@ -1,17 +1 @@
-import { LLMOptions } from "portkey-ai";
-import { CallbackManagerForLLMRun } from "../callbacks/manager.js";
-import { PortkeySession } from "../llms/portkey.js";
-import { BaseMessage, ChatGenerationChunk, ChatResult } from "../schema/index.js";
-import { BaseChatModel } from "./base.js";
-export declare class PortkeyChat extends BaseChatModel {
-    apiKey?: string;
-    baseURL?: string;
-    mode?: string;
-    llms?: [LLMOptions] | null;
-    session: PortkeySession;
-    constructor(init?: Partial<PortkeyChat>);
-    _llmType(): string;
-    _generate(messages: BaseMessage[], options: this["ParsedCallOptions"], _?: CallbackManagerForLLMRun): Promise<ChatResult>;
-    _streamResponseChunks(messages: BaseMessage[], options: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): AsyncGenerator<ChatGenerationChunk>;
-    _combineLLMOutput(): {};
-}
+export * from "@langchain/community/chat_models/portkey";

package/dist/chat_models/portkey.js

@@ -1,165 +1 @@
-import { getPortkeySession } from "../llms/portkey.js";
-import { AIMessage, AIMessageChunk, ChatGenerationChunk, ChatMessage, ChatMessageChunk, FunctionMessageChunk, HumanMessage, HumanMessageChunk, SystemMessage, SystemMessageChunk, } from "../schema/index.js";
-import { BaseChatModel } from "./base.js";
-function portkeyResponseToChatMessage(message) {
-    switch (message.role) {
-        case "user":
-            return new HumanMessage(message.content || "");
-        case "assistant":
-            return new AIMessage(message.content || "");
-        case "system":
-            return new SystemMessage(message.content || "");
-        default:
-            return new ChatMessage(message.content || "", message.role ?? "unknown");
-    }
-}
-function _convertDeltaToMessageChunk(
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-delta) {
-    const { role } = delta;
-    const content = delta.content ?? "";
-    let additional_kwargs;
-    if (delta.function_call) {
-        additional_kwargs = {
-            function_call: delta.function_call,
-        };
-    }
-    else {
-        additional_kwargs = {};
-    }
-    if (role === "user") {
-        return new HumanMessageChunk({ content });
-    }
-    else if (role === "assistant") {
-        return new AIMessageChunk({ content, additional_kwargs });
-    }
-    else if (role === "system") {
-        return new SystemMessageChunk({ content });
-    }
-    else if (role === "function") {
-        return new FunctionMessageChunk({
-            content,
-            additional_kwargs,
-            name: delta.name,
-        });
-    }
-    else {
-        return new ChatMessageChunk({ content, role });
-    }
-}
-export class PortkeyChat extends BaseChatModel {
-    constructor(init) {
-        super(init ?? {});
-        Object.defineProperty(this, "apiKey", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: undefined
-        });
-        Object.defineProperty(this, "baseURL", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: undefined
-        });
-        Object.defineProperty(this, "mode", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: undefined
-        });
-        Object.defineProperty(this, "llms", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: undefined
-        });
-        Object.defineProperty(this, "session", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.apiKey = init?.apiKey;
-        this.baseURL = init?.baseURL;
-        this.mode = init?.mode;
-        this.llms = init?.llms;
-        this.session = getPortkeySession({
-            apiKey: this.apiKey,
-            baseURL: this.baseURL,
-            llms: this.llms,
-            mode: this.mode,
-        });
-    }
-    _llmType() {
-        return "portkey";
-    }
-    async _generate(messages, options, _) {
-        const messagesList = messages.map((message) => {
-            if (typeof message.content !== "string") {
-                throw new Error("PortkeyChat does not support non-string message content.");
-            }
-            return {
-                role: message._getType(),
-                content: message.content,
-            };
-        });
-        const response = await this.session.portkey.chatCompletions.create({
-            messages: messagesList,
-            ...options,
-            stream: false,
-        });
-        const generations = [];
-        for (const data of response.choices ?? []) {
-            const text = data.message?.content ?? "";
-            const generation = {
-                text,
-                message: portkeyResponseToChatMessage(data.message ?? {}),
-            };
-            if (data.finish_reason) {
-                generation.generationInfo = { finish_reason: data.finish_reason };
-            }
-            generations.push(generation);
-        }
-        return {
-            generations,
-        };
-    }
-    async *_streamResponseChunks(messages, options, runManager) {
-        const messagesList = messages.map((message) => {
-            if (typeof message.content !== "string") {
-                throw new Error("PortkeyChat does not support non-string message content.");
-            }
-            return {
-                role: message._getType(),
-                content: message.content,
-            };
-        });
-        const response = await this.session.portkey.chatCompletions.create({
-            messages: messagesList,
-            ...options,
-            stream: true,
-        });
-        for await (const data of response) {
-            const choice = data?.choices[0];
-            if (!choice) {
-                continue;
-            }
-            const chunk = new ChatGenerationChunk({
-                message: _convertDeltaToMessageChunk(choice.delta ?? {}),
-                text: choice.message?.content ?? "",
-                generationInfo: {
-                    finishReason: choice.finish_reason,
-                },
-            });
-            yield chunk;
-            void runManager?.handleLLMNewToken(chunk.text ?? "");
-        }
-        if (options.signal?.aborted) {
-            throw new Error("AbortError");
-        }
-    }
-    _combineLLMOutput() {
-        return {};
-    }
-}
+export * from "@langchain/community/chat_models/portkey";

package/dist/chat_models/yandex.cjs

@@ -1,134 +1,17 @@
 "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 (typeof message.content !== "string") {
-            throw new Error("ChatYandexGPT does not support non-string message content.");
-        }
-        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];
-}
-/**
- * @example
- * ```typescript
- * const chat = new ChatYandexGPT({});
- * // The assistant is set to translate English to French.
- * const res = await chat.call([
- *   new SystemMessage(
- *     "You are a helpful assistant that translates English to French."
- *   ),
- *   new HumanMessage("I love programming."),
- * ]);
- * console.log(res);
- * ```
- */
-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";
+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]; } };
     }
-    _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;
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("@langchain/community/chat_models/yandex"), exports);

package/dist/chat_models/yandex.d.ts

@@ -1,30 +1 @@
-import { CallbackManagerForLLMRun } from "../callbacks/manager.js";
-import { YandexGPTInputs } from "../llms/yandex.js";
-import { BaseMessage, ChatResult } from "../schema/index.js";
-import { BaseChatModel } from "./base.js";
-/**
- * @example
- * ```typescript
- * const chat = new ChatYandexGPT({});
- * // The assistant is set to translate English to French.
- * const res = await chat.call([
- *   new SystemMessage(
- *     "You are a helpful assistant that translates English to French."
- *   ),
- *   new HumanMessage("I love programming."),
- * ]);
- * console.log(res);
- * ```
- */
-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>;
-}
+export * from "@langchain/community/chat_models/yandex";

package/dist/chat_models/yandex.js

@@ -1,130 +1 @@
-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 (typeof message.content !== "string") {
-            throw new Error("ChatYandexGPT does not support non-string message content.");
-        }
-        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];
-}
-/**
- * @example
- * ```typescript
- * const chat = new ChatYandexGPT({});
- * // The assistant is set to translate English to French.
- * const res = await chat.call([
- *   new SystemMessage(
- *     "You are a helpful assistant that translates English to French."
- *   ),
- *   new HumanMessage("I love programming."),
- * ]);
- * console.log(res);
- * ```
- */
-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 },
-        };
-    }
-}
+export * from "@langchain/community/chat_models/yandex";