mirror of
https://github.com/hwchase17/langchain
synced 2024-10-31 15:20:26 +00:00
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",
|
||
|
|
}
|
||
|
|
)
|
||
|
|
)
|