Wikipedia Agentic RAG

Simple example of a retrieval agent using TensorZero and Google Gemini API.

Requirements

• Google Gemini API
• Python 3

Setup

1. Get your API in Google AI Studio
2. Duplicate .env.example file and rename the copy to .env
3. Create a Python virtual environment and activate it
4. Add the API key in GOOGLE_AI_STUDIO_API_KEY on .env file
5. Install the dependencies pip install -r requirements.txt

Running

python main.py

Known issues

• This project is created from simple agentic-rag tutorial, so it's not realiable
• Sometimes the agent is not capable to answer the questions (changing the model or ai provider can be better)

Links

• TensorZero Documentation
• TensorZero Examples

License

This project is under MIT License, enjoy it!