from csv_agent.agent import agent_executor
if __name__ == "__main__":
question = "who was in cabin c28?"
print(agent_executor.invoke({"input": question}))