langchain/templates/neo4j-cypher-memory/main.py
Erick Friis 3a2eb6e12b
infra: add print rule to ruff (#16221)
Added noqa for existing prints. Can slowly remove / will prevent more
being intro'd
2024-02-09 16:13:30 -08:00

24 lines
612 B
Python

from neo4j_cypher_memory.chain import chain
if __name__ == "__main__":
original_query = "Who played in Top Gun?"
print( # noqa: T201
chain.invoke(
{
"question": original_query,
"user_id": "user_123",
"session_id": "session_1",
}
)
)
follow_up_query = "Did they play in any other movies?"
print( # noqa: T201
chain.invoke(
{
"question": follow_up_query,
"user_id": "user_123",
"session_id": "session_1",
}
)
)