gemini/gemini_litellm.py

import os
from smolagents import LiteLLMModel, CodeAgent

gemini_api_key = os.getenv("GEMINI_API_KEY")

# LiteLLM uses the 'gemini/' prefix for Gemini models
model = LiteLLMModel(
    model_id="gemini-2.0-flash", # Use a valid Gemini model ID
    api_base="https://generativelanguage.googleapis.com/v1beta/openai/",
    api_key=gemini_api_key,
)

agent = CodeAgent(tools=[], model=model)
agent.run("Provide a brief history of the Eiffel Tower.")