langchain 0.0.203 → 0.0.204-rc.1

package/dist/graphs/neo4j_graph.cjs

@@ -10,259 +10,8 @@ var __createBinding = (this && this.__createBinding) || (Object.create ? (functi
     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;
+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.Neo4jGraph = void 0;
-const neo4j_driver_1 = __importStar(require("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.
- */
-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_1.default.driver(url, neo4j_driver_1.default.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 neo4j_driver_1.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();
-    }
-}
-exports.Neo4jGraph = Neo4jGraph;
-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_driver_1.default.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_driver_1.default.types.Node ||
-        // eslint-disable-next-line
-        obj instanceof neo4j_driver_1.default.types.Relationship) {
-        return obj.properties;
-        // eslint-disable-next-line
-    }
-    else if (obj instanceof neo4j_driver_1.default.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));
-};
+__exportStar(require("@langchain/community/graphs/neo4j_graph"), exports);

package/dist/graphs/neo4j_graph.d.ts

@@ -1,63 +1 @@
-interface Neo4jGraphConfig {
-    url: string;
-    username: string;
-    password: string;
-    database?: string;
-}
-interface StructuredSchema {
-    nodeProps: {
-        [key: NodeType["labels"]]: NodeType["properties"];
-    };
-    relProps: {
-        [key: RelType["type"]]: RelType["properties"];
-    };
-    relationships: PathType[];
-}
-type NodeType = {
-    labels: string;
-    properties: {
-        property: string;
-        type: string;
-    }[];
-};
-type RelType = {
-    type: string;
-    properties: {
-        property: string;
-        type: string;
-    }[];
-};
-type PathType = {
-    start: string;
-    type: string;
-    end: string;
-};
-/**
- * @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 declare class Neo4jGraph {
-    private driver;
-    private database;
-    private schema;
-    private structuredSchema;
-    constructor({ url, username, password, database, }: Neo4jGraphConfig);
-    static initialize(config: Neo4jGraphConfig): Promise<Neo4jGraph>;
-    getSchema(): string;
-    getStructuredSchema(): StructuredSchema;
-    query(query: string, params?: any): Promise<any[] | undefined>;
-    verifyConnectivity(): Promise<void>;
-    refreshSchema(): Promise<void>;
-    close(): Promise<void>;
-}
-export {};
+export * from "@langchain/community/graphs/neo4j_graph";

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";