feat(community): Adds graph_document to export list (#7555)

Co-authored-by: quantropi-minh <[email protected]>
Co-authored-by: jacoblee93 <[email protected]>

Author: 3 people

libs/langchain-community/.gitignore

@@ -722,14 +722,18 @@ caches/vercel_kv.cjs
 caches/vercel_kv.js
 caches/vercel_kv.d.ts
 caches/vercel_kv.d.cts
-graphs/neo4j_graph.cjs
-graphs/neo4j_graph.js
-graphs/neo4j_graph.d.ts
-graphs/neo4j_graph.d.cts
+graphs/document.cjs
+graphs/document.js
+graphs/document.d.ts
+graphs/document.d.cts
 graphs/memgraph_graph.cjs
 graphs/memgraph_graph.js
 graphs/memgraph_graph.d.ts
 graphs/memgraph_graph.d.cts
+graphs/neo4j_graph.cjs
+graphs/neo4j_graph.js
+graphs/neo4j_graph.d.ts
+graphs/neo4j_graph.d.cts
 document_compressors/ibm.cjs
 document_compressors/ibm.js
 document_compressors/ibm.d.ts

libs/langchain-community/langchain.config.js

@@ -225,8 +225,9 @@ export const config = {
     "caches/upstash_redis": "caches/upstash_redis",
     "caches/vercel_kv": "caches/vercel_kv",
     // graphs
-    "graphs/neo4j_graph": "graphs/neo4j_graph",
+    "graphs/document": "graphs/document",
     "graphs/memgraph_graph": "graphs/memgraph_graph",
+    "graphs/neo4j_graph": "graphs/neo4j_graph",
     // document_compressors
     "document_compressors/ibm": "document_compressors/ibm",
     // document transformers
@@ -461,8 +462,8 @@ export const config = {
     "cache/upstash_redis",
     "cache/vercel_kv",
     //graphs
-    "graphs/neo4j_graph",
     "graphs/memgraph_graph",
+    "graphs/neo4j_graph",
     // document_compressors
     "document_compressors/ibm",
     // document_transformers

libs/langchain-community/package.json

@@ -2347,14 +2347,14 @@
       "import": "./caches/vercel_kv.js",
       "require": "./caches/vercel_kv.cjs"
     },
-    "./graphs/neo4j_graph": {
+    "./graphs/document": {
       "types": {
-        "import": "./graphs/neo4j_graph.d.ts",
-        "require": "./graphs/neo4j_graph.d.cts",
-        "default": "./graphs/neo4j_graph.d.ts"
+        "import": "./graphs/document.d.ts",
+        "require": "./graphs/document.d.cts",
+        "default": "./graphs/document.d.ts"
       },
-      "import": "./graphs/neo4j_graph.js",
-      "require": "./graphs/neo4j_graph.cjs"
+      "import": "./graphs/document.js",
+      "require": "./graphs/document.cjs"
     },
     "./graphs/memgraph_graph": {
       "types": {
@@ -2365,6 +2365,15 @@
       "import": "./graphs/memgraph_graph.js",
       "require": "./graphs/memgraph_graph.cjs"
     },
+    "./graphs/neo4j_graph": {
+      "types": {
+        "import": "./graphs/neo4j_graph.d.ts",
+        "require": "./graphs/neo4j_graph.d.cts",
+        "default": "./graphs/neo4j_graph.d.ts"
+      },
+      "import": "./graphs/neo4j_graph.js",
+      "require": "./graphs/neo4j_graph.cjs"
+    },
     "./document_compressors/ibm": {
       "types": {
         "import": "./document_compressors/ibm.d.ts",
@@ -3921,14 +3930,18 @@
     "caches/vercel_kv.js",
     "caches/vercel_kv.d.ts",
     "caches/vercel_kv.d.cts",
-    "graphs/neo4j_graph.cjs",
-    "graphs/neo4j_graph.js",
-    "graphs/neo4j_graph.d.ts",
-    "graphs/neo4j_graph.d.cts",
+    "graphs/document.cjs",
+    "graphs/document.js",
+    "graphs/document.d.ts",
+    "graphs/document.d.cts",
     "graphs/memgraph_graph.cjs",
     "graphs/memgraph_graph.js",
     "graphs/memgraph_graph.d.ts",
     "graphs/memgraph_graph.d.cts",
+    "graphs/neo4j_graph.cjs",
+    "graphs/neo4j_graph.js",
+    "graphs/neo4j_graph.d.ts",
+    "graphs/neo4j_graph.d.cts",
     "document_compressors/ibm.cjs",
     "document_compressors/ibm.js",
     "document_compressors/ibm.d.ts",

libs/langchain-community/src/experimental/graph_transformers/llm.int.test.ts

@@ -1,11 +1,7 @@
 import { ChatOpenAI } from "@langchain/openai";
 import { Document } from "@langchain/core/documents";
 import { LLMGraphTransformer } from "./llm.js";
-import {
-  GraphDocument,
-  Node,
-  Relationship,
-} from "../../graphs/graph_document.js";
+import { GraphDocument, Node, Relationship } from "../../graphs/document.js";
 
 test.skip("convertToGraphDocuments", async () => {
   const model = new ChatOpenAI({

libs/langchain-community/src/experimental/graph_transformers/llm.ts

@@ -3,11 +3,7 @@ import { zodToJsonSchema } from "zod-to-json-schema";
 import { BaseLanguageModel } from "@langchain/core/language_models/base";
 import { ChatPromptTemplate } from "@langchain/core/prompts";
 import { Document } from "@langchain/core/documents";
-import {
-  Node,
-  Relationship,
-  GraphDocument,
-} from "../../graphs/graph_document.js";
+import { Node, Relationship, GraphDocument } from "../../graphs/document.js";
 
 export const SYSTEM_PROMPT = `
 # Knowledge Graph Instructions for GPT-4\n

libs/langchain-community/src/graphs/graph_document.ts libs/langchain-community/src/graphs/document.ts

@@ -1,6 +1,6 @@
 import neo4j, { RoutingControl } from "neo4j-driver";
 import { insecureHash } from "@langchain/core/utils/hash";
-import { GraphDocument } from "./graph_document.js";
+import { GraphDocument } from "./document.js";
 
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 type Any = any;

libs/langchain-community/src/graphs/neo4j_graph.ts

@@ -3,7 +3,7 @@
 import { test } from "@jest/globals";
 import { Document } from "@langchain/core/documents";
 import { BASE_ENTITY_LABEL, Neo4jGraph } from "../neo4j_graph.js";
-import { GraphDocument, Relationship, Node } from "../graph_document.js";
+import { GraphDocument, Relationship, Node } from "../document.js";
 
 const TEST_DATA = [
   new GraphDocument({

libs/langchain-community/src/graphs/tests/neo4j_graph.int.test.ts

@@ -112,8 +112,8 @@ export const optionalImportEntrypoints: string[] = [
   "langchain_community/structured_query/supabase",
   "langchain_community/structured_query/vectara",
   "langchain_community/retrievers/zep_cloud",
-  "langchain_community/graphs/neo4j_graph",
   "langchain_community/graphs/memgraph_graph",
+  "langchain_community/graphs/neo4j_graph",
   "langchain_community/document_compressors/ibm",
   "langchain_community/document_transformers/html_to_text",
   "langchain_community/document_transformers/mozilla_readability",

libs/langchain-community/src/load/import_constants.ts

@@ -69,6 +69,7 @@ export * as caches__ioredis from "../caches/ioredis.js";
 export * as caches__momento from "../caches/momento.js";
 export * as caches__upstash_redis from "../caches/upstash_redis.js";
 export * as caches__vercel_kv from "../caches/vercel_kv.js";
+export * as graphs__document from "../graphs/document.js";
 export * as stores__doc__base from "../stores/doc/base.js";
 export * as stores__doc__gcs from "../stores/doc/gcs.js";
 export * as stores__doc__in_memory from "../stores/doc/in_memory.js";