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 --- R2R/r2r/examples/scripts/run_web_rag.py | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) create mode 100755 R2R/r2r/examples/scripts/run_web_rag.py (limited to 'R2R/r2r/examples/scripts/run_web_rag.py') diff --git a/R2R/r2r/examples/scripts/run_web_rag.py b/R2R/r2r/examples/scripts/run_web_rag.py new file mode 100755 index 00000000..7535ae27 --- /dev/null +++ b/R2R/r2r/examples/scripts/run_web_rag.py @@ -0,0 +1,26 @@ +import fire + +from r2r import R2RBuilder, SerperClient, WebSearchPipe +from r2r.base.abstractions.llm import GenerationConfig + + +def run_rag_pipeline(query="Who was Aristotle?"): + # Create search pipe override and pipes + web_search_pipe = WebSearchPipe( + serper_client=SerperClient() # TODO - Develop a `WebSearchProvider` for configurability + ) + + app = R2RBuilder().with_vector_search_pipe(web_search_pipe).build() + + # Run the RAG pipeline through the R2R application + result = app.rag( + query, + rag_generation_config=GenerationConfig(model="gpt-4o"), + ) + + print(f"Search Results:\n\n{result.search_results}") + print(f"RAG Results:\n\n{result.completion}") + + +if __name__ == "__main__": + fire.Fire(run_rag_pipeline) -- cgit v1.2.3