File size: 504 Bytes
ed4d993
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
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}
        )
    )