neo4j_graphrag_vector_retriever_similarity_search_openai_embeddings.py

1import neo4j
2from neo4j_graphrag.retrievers import VectorRetriever
3from neo4j_graphrag.embeddings.openai import OpenAIEmbeddings
4
5# Connect to Neo4j database
6driver = neo4j.GraphDatabase.driver("neo4j://localhost:7687", auth=("neo4j", "password"))
7
8# Initialize the embedder
9embedder = OpenAIEmbeddings(model="text-embedding-3-small")
10
11# Initialize the retriever
12retriever = VectorRetriever(
13    driver,
14    index_name="typical_index_name",
15    embedder=embedder,
16    return_properties=["text", "description"],
17)
18
19# Perform a search
20query_text = "Find movie plots about space travel"
21search_result = retriever.search(query_text=query_text, top_k=5)
22
23print(search_result)
24
25driver.close()