blob: 65052378e7ab30e002ebd6c3bbcd637c9e1aa10c (
about) (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
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}")
|
