langchain 0.0.202 → 0.0.204-rc.0

package/dist/graphs/neo4j_graph.js

@@ -1,241 +1 @@
-import neo4j, { Neo4jError } from "neo4j-driver";
-/**
- * @security *Security note*: Make sure that the database connection uses credentials
- * that are narrowly-scoped to only include necessary permissions.
- * Failure to do so may result in data corruption or loss, since the calling
- * code may attempt commands that would result in deletion, mutation
- * of data if appropriately prompted or reading sensitive data if such
- * data is present in the database.
- * The best way to guard against such negative outcomes is to (as appropriate)
- * limit the permissions granted to the credentials used with this tool.
- * For example, creating read only users for the database is a good way to
- * ensure that the calling code cannot mutate or delete data.
- *
- * @link See https://js.langchain.com/docs/security for more information.
- */
-export class Neo4jGraph {
-    constructor({ url, username, password, database = "neo4j", }) {
-        Object.defineProperty(this, "driver", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "database", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "schema", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: ""
-        });
-        Object.defineProperty(this, "structuredSchema", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: {
-                nodeProps: {},
-                relProps: {},
-                relationships: [],
-            }
-        });
-        try {
-            this.driver = neo4j.driver(url, neo4j.auth.basic(username, password));
-            this.database = database;
-        }
-        catch (error) {
-            throw new Error("Could not create a Neo4j driver instance. Please check the connection details.");
-        }
-    }
-    static async initialize(config) {
-        const graph = new Neo4jGraph(config);
-        try {
-            await graph.verifyConnectivity();
-            // eslint-disable-next-line @typescript-eslint/no-explicit-any
-        }
-        catch (error) {
-            console.log("Failed to verify connection.");
-        }
-        try {
-            await graph.refreshSchema();
-            // eslint-disable-next-line @typescript-eslint/no-explicit-any
-        }
-        catch (error) {
-            const message = [
-                "Could not use APOC procedures.",
-                "Please ensure the APOC plugin is installed in Neo4j and that",
-                "'apoc.meta.data()' is allowed in Neo4j configuration",
-            ].join("\n");
-            throw new Error(message);
-        }
-        finally {
-            console.log("Schema refreshed successfully.");
-        }
-        return graph;
-    }
-    getSchema() {
-        return this.schema;
-    }
-    getStructuredSchema() {
-        return this.structuredSchema;
-    }
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    async query(query, params = {}) {
-        try {
-            const result = await this.driver.executeQuery(query, params, {
-                database: this.database,
-            });
-            return toObjects(result.records);
-            // eslint-disable-next-line @typescript-eslint/no-explicit-any
-        }
-        catch (error) {
-            if (
-            // eslint-disable-next-line
-            error instanceof Neo4jError &&
-                error.code === "Neo.ClientError.Procedure.ProcedureNotFound") {
-                throw new Error("Procedure not found in Neo4j.");
-            }
-        }
-        return undefined;
-    }
-    async verifyConnectivity() {
-        await this.driver.verifyAuthentication();
-    }
-    async refreshSchema() {
-        const nodePropertiesQuery = `
-      CALL apoc.meta.data()
-      YIELD label, other, elementType, type, property
-      WHERE NOT type = "RELATIONSHIP" AND elementType = "node"
-      WITH label AS nodeLabels, collect({property:property, type:type}) AS properties
-      RETURN {labels: nodeLabels, properties: properties} AS output
-    `;
-        const relPropertiesQuery = `
-      CALL apoc.meta.data()
-      YIELD label, other, elementType, type, property
-      WHERE NOT type = "RELATIONSHIP" AND elementType = "relationship"
-      WITH label AS nodeLabels, collect({property:property, type:type}) AS properties
-      RETURN {type: nodeLabels, properties: properties} AS output
-    `;
-        const relQuery = `
-      CALL apoc.meta.data()
-      YIELD label, other, elementType, type, property
-      WHERE type = "RELATIONSHIP" AND elementType = "node"
-      UNWIND other AS other_node
-      RETURN {start: label, type: property, end: toString(other_node)} AS output
-    `;
-        // Assuming query method is defined and returns a Promise
-        const nodeProperties = (await this.query(nodePropertiesQuery))?.map((el) => el.output);
-        const relationshipsProperties = (await this.query(relPropertiesQuery))?.map((el) => el.output);
-        const relationships = (await this.query(relQuery))?.map((el) => el.output);
-        // Structured schema similar to Python's dictionary comprehension
-        this.structuredSchema = {
-            nodeProps: Object.fromEntries(nodeProperties?.map((el) => [el.labels, el.properties]) || []),
-            relProps: Object.fromEntries(relationshipsProperties?.map((el) => [el.type, el.properties]) || []),
-            relationships: relationships || [],
-        };
-        // Format node properties
-        const formattedNodeProps = nodeProperties?.map((el) => {
-            const propsStr = el.properties
-                .map((prop) => `${prop.property}: ${prop.type}`)
-                .join(", ");
-            return `${el.labels} {${propsStr}}`;
-        });
-        // Format relationship properties
-        const formattedRelProps = relationshipsProperties?.map((el) => {
-            const propsStr = el.properties
-                .map((prop) => `${prop.property}: ${prop.type}`)
-                .join(", ");
-            return `${el.type} {${propsStr}}`;
-        });
-        // Format relationships
-        const formattedRels = relationships?.map((el) => `(:${el.start})-[:${el.type}]->(:${el.end})`);
-        // Combine all formatted elements into a single string
-        this.schema = [
-            "Node properties are the following:",
-            formattedNodeProps?.join(", "),
-            "Relationship properties are the following:",
-            formattedRelProps?.join(", "),
-            "The relationships are the following:",
-            formattedRels?.join(", "),
-        ].join("\n");
-    }
-    async close() {
-        await this.driver.close();
-    }
-}
-function toObjects(records) {
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    const recordValues = records.map((record) => {
-        const rObj = record.toObject();
-        // eslint-disable-next-line @typescript-eslint/no-explicit-any
-        const out = {};
-        Object.keys(rObj).forEach((key) => {
-            out[key] = itemIntToString(rObj[key]);
-        });
-        return out;
-    });
-    return recordValues;
-}
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-function itemIntToString(item) {
-    if (neo4j.isInt(item))
-        return item.toString();
-    if (Array.isArray(item))
-        return item.map((ii) => itemIntToString(ii));
-    if (["number", "string", "boolean"].indexOf(typeof item) !== -1)
-        return item;
-    if (item === null)
-        return item;
-    if (typeof item === "object")
-        return objIntToString(item);
-}
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-function objIntToString(obj) {
-    const entry = extractFromNeoObjects(obj);
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    let newObj = null;
-    if (Array.isArray(entry)) {
-        newObj = entry.map((item) => itemIntToString(item));
-    }
-    else if (entry !== null && typeof entry === "object") {
-        newObj = {};
-        Object.keys(entry).forEach((key) => {
-            newObj[key] = itemIntToString(entry[key]);
-        });
-    }
-    return newObj;
-}
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-function extractFromNeoObjects(obj) {
-    if (
-    // eslint-disable-next-line
-    obj instanceof neo4j.types.Node ||
-        // eslint-disable-next-line
-        obj instanceof neo4j.types.Relationship) {
-        return obj.properties;
-        // eslint-disable-next-line
-    }
-    else if (obj instanceof neo4j.types.Path) {
-        // eslint-disable-next-line
-        return [].concat.apply([], extractPathForRows(obj));
-    }
-    return obj;
-}
-const extractPathForRows = (path) => {
-    let { segments } = path;
-    // Zero length path. No relationship, end === start
-    if (!Array.isArray(path.segments) || path.segments.length < 1) {
-        // eslint-disable-next-line @typescript-eslint/no-explicit-any
-        segments = [{ ...path, end: null }];
-    }
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    return segments.map((segment) => [
-        objIntToString(segment.start),
-        objIntToString(segment.relationship),
-        objIntToString(segment.end),
-    ].filter((part) => part !== null));
-};
+export * from "@langchain/community/graphs/neo4j_graph";

package/dist/llms/ai21.cjs

@@ -1,202 +1,17 @@
 "use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.AI21 = void 0;
-const base_js_1 = require("./base.cjs");
-const env_js_1 = require("../util/env.cjs");
-/**
- * Class representing the AI21 language model. It extends the LLM (Large
- * Language Model) class, providing a standard interface for interacting
- * with the AI21 language model.
- */
-class AI21 extends base_js_1.LLM {
-    constructor(fields) {
-        super(fields ?? {});
-        Object.defineProperty(this, "model", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: "j2-jumbo-instruct"
-        });
-        Object.defineProperty(this, "temperature", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: 0.7
-        });
-        Object.defineProperty(this, "maxTokens", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: 1024
-        });
-        Object.defineProperty(this, "minTokens", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: 0
-        });
-        Object.defineProperty(this, "topP", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: 1
-        });
-        Object.defineProperty(this, "presencePenalty", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: AI21.getDefaultAI21PenaltyData()
-        });
-        Object.defineProperty(this, "countPenalty", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: AI21.getDefaultAI21PenaltyData()
-        });
-        Object.defineProperty(this, "frequencyPenalty", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: AI21.getDefaultAI21PenaltyData()
-        });
-        Object.defineProperty(this, "numResults", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: 1
-        });
-        Object.defineProperty(this, "logitBias", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "ai21ApiKey", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "stop", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "baseUrl", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.model = fields?.model ?? this.model;
-        this.temperature = fields?.temperature ?? this.temperature;
-        this.maxTokens = fields?.maxTokens ?? this.maxTokens;
-        this.minTokens = fields?.minTokens ?? this.minTokens;
-        this.topP = fields?.topP ?? this.topP;
-        this.presencePenalty = fields?.presencePenalty ?? this.presencePenalty;
-        this.countPenalty = fields?.countPenalty ?? this.countPenalty;
-        this.frequencyPenalty = fields?.frequencyPenalty ?? this.frequencyPenalty;
-        this.numResults = fields?.numResults ?? this.numResults;
-        this.logitBias = fields?.logitBias;
-        this.ai21ApiKey =
-            fields?.ai21ApiKey ?? (0, env_js_1.getEnvironmentVariable)("AI21_API_KEY");
-        this.stop = fields?.stop;
-        this.baseUrl = fields?.baseUrl;
-    }
-    /**
-     * Method to validate the environment. It checks if the AI21 API key is
-     * set. If not, it throws an error.
-     */
-    validateEnvironment() {
-        if (!this.ai21ApiKey) {
-            throw new Error(`No AI21 API key found. Please set it as "AI21_API_KEY" in your environment variables.`);
-        }
-    }
-    /**
-     * Static method to get the default penalty data for AI21.
-     * @returns AI21PenaltyData
-     */
-    static getDefaultAI21PenaltyData() {
-        return {
-            scale: 0,
-            applyToWhitespaces: true,
-            applyToPunctuations: true,
-            applyToNumbers: true,
-            applyToStopwords: true,
-            applyToEmojis: true,
-        };
-    }
-    /** Get the type of LLM. */
-    _llmType() {
-        return "ai21";
+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]; } };
     }
-    /** Get the default parameters for calling AI21 API. */
-    get defaultParams() {
-        return {
-            temperature: this.temperature,
-            maxTokens: this.maxTokens,
-            minTokens: this.minTokens,
-            topP: this.topP,
-            presencePenalty: this.presencePenalty,
-            countPenalty: this.countPenalty,
-            frequencyPenalty: this.frequencyPenalty,
-            numResults: this.numResults,
-            logitBias: this.logitBias,
-        };
-    }
-    /** Get the identifying parameters for this LLM. */
-    get identifyingParams() {
-        return { ...this.defaultParams, model: this.model };
-    }
-    /** Call out to AI21's complete endpoint.
-     Args:
-         prompt: The prompt to pass into the model.
-     stop: Optional list of stop words to use when generating.
-  
-         Returns:
-     The string generated by the model.
-  
-     Example:
-     let response = ai21._call("Tell me a joke.");
-     */
-    async _call(prompt, options) {
-        let stop = options?.stop;
-        this.validateEnvironment();
-        if (this.stop && stop && this.stop.length > 0 && stop.length > 0) {
-            throw new Error("`stop` found in both the input and default params.");
-        }
-        stop = this.stop ?? stop ?? [];
-        const baseUrl = this.baseUrl ?? this.model === "j1-grande-instruct"
-            ? "https://api.ai21.com/studio/v1/experimental"
-            : "https://api.ai21.com/studio/v1";
-        const url = `${baseUrl}/${this.model}/complete`;
-        const headers = {
-            Authorization: `Bearer ${this.ai21ApiKey}`,
-            "Content-Type": "application/json",
-        };
-        const data = { prompt, stopSequences: stop, ...this.defaultParams };
-        const responseData = await this.caller.callWithOptions({}, async () => {
-            const response = await fetch(url, {
-                method: "POST",
-                headers,
-                body: JSON.stringify(data),
-                signal: options.signal,
-            });
-            if (!response.ok) {
-                const error = new Error(`AI21 call failed with status code ${response.status}`);
-                // eslint-disable-next-line @typescript-eslint/no-explicit-any
-                error.response = response;
-                throw error;
-            }
-            return response.json();
-        });
-        if (!responseData.completions ||
-            responseData.completions.length === 0 ||
-            !responseData.completions[0].data) {
-            throw new Error("No completions found in response");
-        }
-        return responseData.completions[0].data.text ?? "";
-    }
-}
-exports.AI21 = AI21;
+    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/ai21"), exports);

package/dist/llms/ai21.d.ts

@@ -1,100 +1 @@
-import { LLM, BaseLLMParams } from "./base.js";
-/**
- * Type definition for AI21 penalty data.
- */
-export type AI21PenaltyData = {
-    scale: number;
-    applyToWhitespaces: boolean;
-    applyToPunctuations: boolean;
-    applyToNumbers: boolean;
-    applyToStopwords: boolean;
-    applyToEmojis: boolean;
-};
-/**
- * Interface for AI21 input parameters.
- */
-export interface AI21Input extends BaseLLMParams {
-    ai21ApiKey?: string;
-    model?: string;
-    temperature?: number;
-    minTokens?: number;
-    maxTokens?: number;
-    topP?: number;
-    presencePenalty?: AI21PenaltyData;
-    countPenalty?: AI21PenaltyData;
-    frequencyPenalty?: AI21PenaltyData;
-    numResults?: number;
-    logitBias?: Record<string, number>;
-    stop?: string[];
-    baseUrl?: string;
-}
-/**
- * Class representing the AI21 language model. It extends the LLM (Large
- * Language Model) class, providing a standard interface for interacting
- * with the AI21 language model.
- */
-export declare class AI21 extends LLM implements AI21Input {
-    model: string;
-    temperature: number;
-    maxTokens: number;
-    minTokens: number;
-    topP: number;
-    presencePenalty: AI21PenaltyData;
-    countPenalty: AI21PenaltyData;
-    frequencyPenalty: AI21PenaltyData;
-    numResults: number;
-    logitBias?: Record<string, number>;
-    ai21ApiKey?: string;
-    stop?: string[];
-    baseUrl?: string;
-    constructor(fields?: AI21Input);
-    /**
-     * Method to validate the environment. It checks if the AI21 API key is
-     * set. If not, it throws an error.
-     */
-    validateEnvironment(): void;
-    /**
-     * Static method to get the default penalty data for AI21.
-     * @returns AI21PenaltyData
-     */
-    static getDefaultAI21PenaltyData(): AI21PenaltyData;
-    /** Get the type of LLM. */
-    _llmType(): string;
-    /** Get the default parameters for calling AI21 API. */
-    get defaultParams(): {
-        temperature: number;
-        maxTokens: number;
-        minTokens: number;
-        topP: number;
-        presencePenalty: AI21PenaltyData;
-        countPenalty: AI21PenaltyData;
-        frequencyPenalty: AI21PenaltyData;
-        numResults: number;
-        logitBias: Record<string, number> | undefined;
-    };
-    /** Get the identifying parameters for this LLM. */
-    get identifyingParams(): {
-        model: string;
-        temperature: number;
-        maxTokens: number;
-        minTokens: number;
-        topP: number;
-        presencePenalty: AI21PenaltyData;
-        countPenalty: AI21PenaltyData;
-        frequencyPenalty: AI21PenaltyData;
-        numResults: number;
-        logitBias: Record<string, number> | undefined;
-    };
-    /** Call out to AI21's complete endpoint.
-     Args:
-         prompt: The prompt to pass into the model.
-     stop: Optional list of stop words to use when generating.
-  
-         Returns:
-     The string generated by the model.
-  
-     Example:
-     let response = ai21._call("Tell me a joke.");
-     */
-    _call(prompt: string, options: this["ParsedCallOptions"]): Promise<string>;
-}
+export * from "@langchain/community/llms/ai21";