from neo4j_cypher.chain import chain

if __name__ == "__main__":
    original_query = "Who played in Top Gun?"
    print(chain.invoke({"question": original_query}))