langchain/templates/neo4j-advanced-rag/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

11 lines
292 B
Python

from neo4j_advanced_rag.chain import chain
if __name__ == "__main__":
original_query = "What is the plot of the Dune?"
print( # noqa: T201
chain.invoke(
{"question": original_query},
{"configurable": {"strategy": "parent_document"}},
)
)