mirror of
https://github.com/hwchase17/langchain
synced 2024-11-18 09:25:54 +00:00
d9abcf1aae
Adding custom graph memory to Cypher chain --------- Co-authored-by: Erick Friis <erick@langchain.dev>
24 lines
584 B
Python
24 lines
584 B
Python
from neo4j_cypher_memory.chain import chain
|
|
|
|
if __name__ == "__main__":
|
|
original_query = "Who played in Top Gun?"
|
|
print(
|
|
chain.invoke(
|
|
{
|
|
"question": original_query,
|
|
"user_id": "user_123",
|
|
"session_id": "session_1",
|
|
}
|
|
)
|
|
)
|
|
follow_up_query = "Did they play in any other movies?"
|
|
print(
|
|
chain.invoke(
|
|
{
|
|
"question": follow_up_query,
|
|
"user_id": "user_123",
|
|
"session_id": "session_1",
|
|
}
|
|
)
|
|
)
|