# Define a Function To Run the Chain with Our Query

def query(input: str) -> str:
    response = db_chain.run(input)
    return response