import os
from smolagents import CodeAgent, OpenAIModel

# Get API key from Windows Environment Variable
gemini_api_key = os.getenv("GEMINI_API_KEY")

# Initialize Gemini Model via smolagents
model = OpenAIModel(
    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,
)

# Initialize Agent
agent = CodeAgent(tools=[], model=model)

# Run Agent
agent.run("Explain how AI agents work.")