langchain/templates/neo4j-cypher-memory/main.py
2024-05-22 15:21:08 -07:00

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",
}
)
)