1 files changed, 27 insertions, 0 deletions
diff --git a/gnqa/test_r2r.py b/gnqa/test_r2r.py
new file mode 100644
index 00000000..65052378
--- /dev/null
+++ b/gnqa/test_r2r.py
@@ -0,0 +1,27 @@
+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}")