langchain 0.0.203 → 0.0.204-rc.0

package/dist/llms/cloudflare_workersai.js

@@ -1,166 +1 @@
-import { LLM } from "./base.js";
-import { getEnvironmentVariable } from "../util/env.js";
-import { GenerationChunk } from "../schema/index.js";
-import { convertEventStreamToIterableReadableDataStream } from "../util/event-source-parse.js";
-/**
- * Class representing the CloudflareWorkersAI language model. It extends the LLM (Large
- * Language Model) class, providing a standard interface for interacting
- * with the CloudflareWorkersAI language model.
- */
-export class CloudflareWorkersAI extends LLM {
-    static lc_name() {
-        return "CloudflareWorkersAI";
-    }
-    constructor(fields) {
-        super(fields ?? {});
-        Object.defineProperty(this, "model", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: "@cf/meta/llama-2-7b-chat-int8"
-        });
-        Object.defineProperty(this, "cloudflareAccountId", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "cloudflareApiToken", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "baseUrl", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "streaming", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: false
-        });
-        Object.defineProperty(this, "lc_serializable", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: true
-        });
-        this.model = fields?.model ?? this.model;
-        this.streaming = fields?.streaming ?? this.streaming;
-        this.cloudflareAccountId =
-            fields?.cloudflareAccountId ??
-                getEnvironmentVariable("CLOUDFLARE_ACCOUNT_ID");
-        this.cloudflareApiToken =
-            fields?.cloudflareApiToken ??
-                getEnvironmentVariable("CLOUDFLARE_API_TOKEN");
-        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);
-        }
-    }
-    /**
-     * Method to validate the environment.
-     */
-    validateEnvironment() {
-        if (this.baseUrl === undefined) {
-            if (!this.cloudflareAccountId) {
-                throw new Error(`No Cloudflare account ID found. Please provide it when instantiating the CloudflareWorkersAI class, or set it as "CLOUDFLARE_ACCOUNT_ID" in your environment variables.`);
-            }
-            if (!this.cloudflareApiToken) {
-                throw new Error(`No Cloudflare API key found. Please provide it when instantiating the CloudflareWorkersAI class, or set it as "CLOUDFLARE_API_KEY" in your environment variables.`);
-            }
-        }
-    }
-    /** Get the identifying parameters for this LLM. */
-    get identifyingParams() {
-        return { model: this.model };
-    }
-    /**
-     * Get the parameters used to invoke the model
-     */
-    invocationParams() {
-        return {
-            model: this.model,
-        };
-    }
-    /** Get the type of LLM. */
-    _llmType() {
-        return "cloudflare";
-    }
-    async _request(prompt, options, stream) {
-        this.validateEnvironment();
-        const url = `${this.baseUrl}/${this.model}`;
-        const headers = {
-            Authorization: `Bearer ${this.cloudflareApiToken}`,
-            "Content-Type": "application/json",
-        };
-        const data = { prompt, stream };
-        return this.caller.call(async () => {
-            const response = await fetch(url, {
-                method: "POST",
-                headers,
-                body: JSON.stringify(data),
-                signal: options.signal,
-            });
-            if (!response.ok) {
-                const error = new Error(`Cloudflare LLM call failed with status code ${response.status}`);
-                // eslint-disable-next-line @typescript-eslint/no-explicit-any
-                error.response = response;
-                throw error;
-            }
-            return response;
-        });
-    }
-    async *_streamResponseChunks(prompt, options, runManager) {
-        const response = await this._request(prompt, options, true);
-        if (!response.body) {
-            throw new Error("Empty response from Cloudflare. Please try again.");
-        }
-        const stream = convertEventStreamToIterableReadableDataStream(response.body);
-        for await (const chunk of stream) {
-            if (chunk !== "[DONE]") {
-                const parsedChunk = JSON.parse(chunk);
-                const generationChunk = new GenerationChunk({
-                    text: parsedChunk.response,
-                });
-                yield generationChunk;
-                // eslint-disable-next-line no-void
-                void runManager?.handleLLMNewToken(generationChunk.text ?? "");
-            }
-        }
-    }
-    /** Call out to CloudflareWorkersAI's complete endpoint.
-     Args:
-         prompt: The prompt to pass into the model.
-     Returns:
-         The string generated by the model.
-     Example:
-     let response = CloudflareWorkersAI.call("Tell me a joke.");
-     */
-    async _call(prompt, options, runManager) {
-        if (!this.streaming) {
-            const response = await this._request(prompt, options);
-            const responseData = await response.json();
-            return responseData.result.response;
-        }
-        else {
-            const stream = this._streamResponseChunks(prompt, options, runManager);
-            let finalResult;
-            for await (const chunk of stream) {
-                if (finalResult === undefined) {
-                    finalResult = chunk;
-                }
-                else {
-                    finalResult = finalResult.concat(chunk);
-                }
-            }
-            return finalResult?.text ?? "";
-        }
-    }
-}
+export * from "@langchain/community/llms/cloudflare_workersai";

package/dist/llms/cohere.cjs

@@ -1,112 +1,17 @@
 "use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.Cohere = void 0;
-const env_js_1 = require("../util/env.cjs");
-const base_js_1 = require("./base.cjs");
-/**
- * Class representing a Cohere Large Language Model (LLM). It interacts
- * with the Cohere API to generate text completions.
- * @example
- * ```typescript
- * const model = new Cohere({
- *   temperature: 0.7,
- *   maxTokens: 20,
- *   maxRetries: 5,
- * });
- *
- * const res = await model.call(
- *   "Question: What would be a good company name for a company that makes colorful socks?\nAnswer:"
- * );
- * console.log({ res });
- * ```
- */
-class Cohere extends base_js_1.LLM {
-    static lc_name() {
-        return "Cohere";
-    }
-    get lc_secrets() {
-        return {
-            apiKey: "COHERE_API_KEY",
-        };
-    }
-    get lc_aliases() {
-        return {
-            apiKey: "cohere_api_key",
-        };
-    }
-    constructor(fields) {
-        super(fields ?? {});
-        Object.defineProperty(this, "lc_serializable", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: true
-        });
-        Object.defineProperty(this, "temperature", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: 0
-        });
-        Object.defineProperty(this, "maxTokens", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: 250
-        });
-        Object.defineProperty(this, "model", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "apiKey", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        const apiKey = fields?.apiKey ?? (0, env_js_1.getEnvironmentVariable)("COHERE_API_KEY");
-        if (!apiKey) {
-            throw new Error("Please set the COHERE_API_KEY environment variable or pass it to the constructor as the apiKey field.");
-        }
-        this.apiKey = apiKey;
-        this.maxTokens = fields?.maxTokens ?? this.maxTokens;
-        this.temperature = fields?.temperature ?? this.temperature;
-        this.model = fields?.model ?? this.model;
+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]; } };
     }
-    _llmType() {
-        return "cohere";
-    }
-    /** @ignore */
-    async _call(prompt, options) {
-        const { cohere } = await Cohere.imports();
-        cohere.init(this.apiKey);
-        // Hit the `generate` endpoint on the `large` model
-        const generateResponse = await this.caller.callWithOptions({ signal: options.signal }, cohere.generate.bind(cohere), {
-            prompt,
-            model: this.model,
-            max_tokens: this.maxTokens,
-            temperature: this.temperature,
-            end_sequences: options.stop,
-        });
-        try {
-            return generateResponse.body.generations[0].text;
-        }
-        catch {
-            console.log(generateResponse);
-            throw new Error("Could not parse response.");
-        }
-    }
-    /** @ignore */
-    static async imports() {
-        try {
-            const { default: cohere } = await import("cohere-ai");
-            return { cohere };
-        }
-        catch (e) {
-            throw new Error("Please install cohere-ai as a dependency with, e.g. `yarn add cohere-ai`");
-        }
-    }
-}
-exports.Cohere = Cohere;
+    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/llms/cohere"), exports);

package/dist/llms/cohere.d.ts

@@ -1,54 +1 @@
-import { LLM, BaseLLMParams } from "./base.js";
-/**
- * Interface for the input parameters specific to the Cohere model.
- */
-export interface CohereInput extends BaseLLMParams {
-    /** Sampling temperature to use */
-    temperature?: number;
-    /**
-     * Maximum number of tokens to generate in the completion.
-     */
-    maxTokens?: number;
-    /** Model to use */
-    model?: string;
-    apiKey?: string;
-}
-/**
- * Class representing a Cohere Large Language Model (LLM). It interacts
- * with the Cohere API to generate text completions.
- * @example
- * ```typescript
- * const model = new Cohere({
- *   temperature: 0.7,
- *   maxTokens: 20,
- *   maxRetries: 5,
- * });
- *
- * const res = await model.call(
- *   "Question: What would be a good company name for a company that makes colorful socks?\nAnswer:"
- * );
- * console.log({ res });
- * ```
- */
-export declare class Cohere extends LLM implements CohereInput {
-    static lc_name(): string;
-    get lc_secrets(): {
-        [key: string]: string;
-    } | undefined;
-    get lc_aliases(): {
-        [key: string]: string;
-    } | undefined;
-    lc_serializable: boolean;
-    temperature: number;
-    maxTokens: number;
-    model: string;
-    apiKey: string;
-    constructor(fields?: CohereInput);
-    _llmType(): string;
-    /** @ignore */
-    _call(prompt: string, options: this["ParsedCallOptions"]): Promise<string>;
-    /** @ignore */
-    static imports(): Promise<{
-        cohere: typeof import("cohere-ai");
-    }>;
-}
+export * from "@langchain/community/llms/cohere";

package/dist/llms/cohere.js

@@ -1,108 +1 @@
-import { getEnvironmentVariable } from "../util/env.js";
-import { LLM } from "./base.js";
-/**
- * Class representing a Cohere Large Language Model (LLM). It interacts
- * with the Cohere API to generate text completions.
- * @example
- * ```typescript
- * const model = new Cohere({
- *   temperature: 0.7,
- *   maxTokens: 20,
- *   maxRetries: 5,
- * });
- *
- * const res = await model.call(
- *   "Question: What would be a good company name for a company that makes colorful socks?\nAnswer:"
- * );
- * console.log({ res });
- * ```
- */
-export class Cohere extends LLM {
-    static lc_name() {
-        return "Cohere";
-    }
-    get lc_secrets() {
-        return {
-            apiKey: "COHERE_API_KEY",
-        };
-    }
-    get lc_aliases() {
-        return {
-            apiKey: "cohere_api_key",
-        };
-    }
-    constructor(fields) {
-        super(fields ?? {});
-        Object.defineProperty(this, "lc_serializable", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: true
-        });
-        Object.defineProperty(this, "temperature", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: 0
-        });
-        Object.defineProperty(this, "maxTokens", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: 250
-        });
-        Object.defineProperty(this, "model", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "apiKey", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        const apiKey = fields?.apiKey ?? getEnvironmentVariable("COHERE_API_KEY");
-        if (!apiKey) {
-            throw new Error("Please set the COHERE_API_KEY environment variable or pass it to the constructor as the apiKey field.");
-        }
-        this.apiKey = apiKey;
-        this.maxTokens = fields?.maxTokens ?? this.maxTokens;
-        this.temperature = fields?.temperature ?? this.temperature;
-        this.model = fields?.model ?? this.model;
-    }
-    _llmType() {
-        return "cohere";
-    }
-    /** @ignore */
-    async _call(prompt, options) {
-        const { cohere } = await Cohere.imports();
-        cohere.init(this.apiKey);
-        // Hit the `generate` endpoint on the `large` model
-        const generateResponse = await this.caller.callWithOptions({ signal: options.signal }, cohere.generate.bind(cohere), {
-            prompt,
-            model: this.model,
-            max_tokens: this.maxTokens,
-            temperature: this.temperature,
-            end_sequences: options.stop,
-        });
-        try {
-            return generateResponse.body.generations[0].text;
-        }
-        catch {
-            console.log(generateResponse);
-            throw new Error("Could not parse response.");
-        }
-    }
-    /** @ignore */
-    static async imports() {
-        try {
-            const { default: cohere } = await import("cohere-ai");
-            return { cohere };
-        }
-        catch (e) {
-            throw new Error("Please install cohere-ai as a dependency with, e.g. `yarn add cohere-ai`");
-        }
-    }
-}
+export * from "@langchain/community/llms/cohere";

package/dist/llms/fireworks.cjs

@@ -1,92 +1,17 @@
 "use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.Fireworks = void 0;
-const env_js_1 = require("../util/env.cjs");
-const openai_js_1 = require("./openai.cjs");
-/**
- * Wrapper around Fireworks API for large language models
- *
- * Fireworks API is compatible to the OpenAI API with some limitations described in
- * https://readme.fireworks.ai/docs/openai-compatibility.
- *
- * To use, you should have the `openai` package installed and
- * the `FIREWORKS_API_KEY` environment variable set.
- */
-class Fireworks extends openai_js_1.OpenAI {
-    static lc_name() {
-        return "Fireworks";
-    }
-    _llmType() {
-        return "fireworks";
-    }
-    get lc_secrets() {
-        return {
-            fireworksApiKey: "FIREWORKS_API_KEY",
-        };
-    }
-    constructor(fields) {
-        const fireworksApiKey = fields?.fireworksApiKey || (0, env_js_1.getEnvironmentVariable)("FIREWORKS_API_KEY");
-        if (!fireworksApiKey) {
-            throw new Error(`Fireworks API key not found. Please set the FIREWORKS_API_KEY environment variable or provide the key into "fireworksApiKey"`);
-        }
-        super({
-            ...fields,
-            openAIApiKey: fireworksApiKey,
-            modelName: fields?.modelName || "accounts/fireworks/models/llama-v2-13b",
-            configuration: {
-                baseURL: "https://api.fireworks.ai/inference/v1",
-            },
-        });
-        Object.defineProperty(this, "lc_serializable", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: true
-        });
-        Object.defineProperty(this, "fireworksApiKey", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.fireworksApiKey = fireworksApiKey;
+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]; } };
     }
-    toJSON() {
-        const result = super.toJSON();
-        if ("kwargs" in result &&
-            typeof result.kwargs === "object" &&
-            result.kwargs != null) {
-            delete result.kwargs.openai_api_key;
-            delete result.kwargs.configuration;
-        }
-        return result;
-    }
-    /**
-     * Calls the Fireworks API with retry logic in case of failures.
-     * @param request The request to send to the Fireworks API.
-     * @param options Optional configuration for the API call.
-     * @returns The response from the Fireworks API.
-     */
-    async completionWithRetry(request, options) {
-        // https://readme.fireworks.ai/docs/openai-compatibility#api-compatibility
-        if (Array.isArray(request.prompt)) {
-            if (request.prompt.length > 1) {
-                throw new Error("Multiple prompts are not supported by Fireworks");
-            }
-            const prompt = request.prompt[0];
-            if (typeof prompt !== "string") {
-                throw new Error("Only string prompts are supported by Fireworks");
-            }
-            request.prompt = prompt;
-        }
-        delete request.frequency_penalty;
-        delete request.presence_penalty;
-        delete request.best_of;
-        delete request.logit_bias;
-        if (request.stream === true) {
-            return super.completionWithRetry(request, options);
-        }
-        return super.completionWithRetry(request, options);
-    }
-}
-exports.Fireworks = Fireworks;
+    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/llms/fireworks"), exports);

package/dist/llms/fireworks.d.ts

@@ -1,33 +1 @@
-import type { OpenAI as OpenAIClient } from "openai";
-import type { BaseLLMParams } from "./base.js";
-import type { OpenAICallOptions, OpenAIInput } from "./openai.js";
-import type { OpenAICoreRequestOptions } from "../types/openai-types.js";
-import { OpenAI } from "./openai.js";
-type FireworksUnsupportedArgs = "frequencyPenalty" | "presencePenalty" | "bestOf" | "logitBias";
-type FireworksUnsupportedCallOptions = "functions" | "function_call" | "tools";
-export type FireworksCallOptions = Partial<Omit<OpenAICallOptions, FireworksUnsupportedCallOptions>>;
-/**
- * Wrapper around Fireworks API for large language models
- *
- * Fireworks API is compatible to the OpenAI API with some limitations described in
- * https://readme.fireworks.ai/docs/openai-compatibility.
- *
- * To use, you should have the `openai` package installed and
- * the `FIREWORKS_API_KEY` environment variable set.
- */
-export declare class Fireworks extends OpenAI<FireworksCallOptions> {
-    static lc_name(): string;
-    _llmType(): string;
-    get lc_secrets(): {
-        [key: string]: string;
-    } | undefined;
-    lc_serializable: boolean;
-    fireworksApiKey?: string;
-    constructor(fields?: Partial<Omit<OpenAIInput, "openAIApiKey" | FireworksUnsupportedArgs>> & BaseLLMParams & {
-        fireworksApiKey?: string;
-    });
-    toJSON(): import("@langchain/core/load/serializable").Serialized;
-    completionWithRetry(request: OpenAIClient.CompletionCreateParamsStreaming, options?: OpenAICoreRequestOptions): Promise<AsyncIterable<OpenAIClient.Completion>>;
-    completionWithRetry(request: OpenAIClient.CompletionCreateParamsNonStreaming, options?: OpenAICoreRequestOptions): Promise<OpenAIClient.Completions.Completion>;
-}
-export {};
+export * from "@langchain/community/llms/fireworks";