Add neo4j vector memory template (#12993)

This commit is contained in:
Tomaz Bratanic
2023-11-07 22:00:49 +01:00
committed by GitHub
parent 5ac2fc5bb2
commit 13bd83bd61
10 changed files with 2136 additions and 0 deletions

View File

@@ -0,0 +1,23 @@
from pathlib import Path
from langchain.document_loaders import TextLoader
from langchain.embeddings.openai import OpenAIEmbeddings
from langchain.text_splitter import TokenTextSplitter
from langchain.vectorstores import Neo4jVector
txt_path = Path(__file__).parent / "dune.txt"
# Load the text file
loader = TextLoader(str(txt_path))
raw_documents = loader.load()
# Define chunking strategy
splitter = TokenTextSplitter(chunk_size=512, chunk_overlap=24)
documents = splitter.split_documents(raw_documents)
# Calculate embedding values and store them in the graph
Neo4jVector.from_documents(
documents,
OpenAIEmbeddings(),
index_name="dune",
)