from r2r import R2RClient

client = R2RClient()

#with open("test.txt", "w") as file:

#    file.write("John is a person that works at Google.")

#client.documents.create(file_path="test.txt")

# Call RAG directly

rag_response = client.retrieval.rag(

    query="What is the role of obesity in diabetes",

    rag_generation_config={"model": "openai/gpt-4o-mini", "temperature": 0.0},

)

print(f"Search Results:\n{rag_response.results.search_results}")


print(f"Completion:\n{rag_response.results.generated_answer}")


print(f"Citations:\n{rag_response.results.citations}")