From 4a52a71956a8d46fcb7294ac71734504bb09bcc2 Mon Sep 17 00:00:00 2001
From: S. Solomon Darnell
Date: Fri, 28 Mar 2025 21:52:21 -0500
Subject: two version of R2R are here

---
 gnqa/test_r2r.py | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)
 create mode 100644 gnqa/test_r2r.py

(limited to 'gnqa/test_r2r.py')

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}")
-- 
cgit v1.2.3