Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Merged
merged 3 commits into from
Jan 22, 2025
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 8 additions & 4 deletions libs/langchain-community/.gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
5 changes: 3 additions & 2 deletions libs/langchain-community/langchain.config.js
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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
Expand Down
33 changes: 23 additions & 10 deletions libs/langchain-community/package.json
Original file line number Diff line number Diff line change
Expand Up @@ -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": {
Expand All @@ -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",
Expand Down Expand Up @@ -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",
Expand Down
Original file line number Diff line number Diff line change
@@ -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({
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
2 changes: 1 addition & 1 deletion libs/langchain-community/src/graphs/neo4j_graph.ts
Original file line number Diff line number Diff line change
@@ -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;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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({
Expand Down
2 changes: 1 addition & 1 deletion libs/langchain-community/src/load/import_constants.ts
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down
1 change: 1 addition & 0 deletions libs/langchain-community/src/load/import_map.ts
Original file line number Diff line number Diff line change
Expand Up @@ -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";
Expand Down
Loading