github-sync-test / gemini /gemini_openai.py
billyaungmyint's picture
Sync from GitHub via hub-sync
135756a verified
raw
history blame contribute delete
494 Bytes
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.")