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,17 @@
from neo4j_vector_memory.chain import chain
if __name__ == "__main__":
user_id = "user_id_1"
session_id = "session_id_1"
original_query = "What is the plot of the Dune?"
print(
chain.invoke(
{"question": original_query, "user_id": user_id, "session_id": session_id}
)
)
follow_up_query = "Tell me more about Leto"
print(
chain.invoke(
{"question": follow_up_query, "user_id": user_id, "session_id": session_id}
)
)