from rag_google_cloud_vertexai_search.chain import chain

if __name__ == "__main__":
    query = "Who is the CEO of Google Cloud?"
    print(chain.invoke(query))