| 1234567891011121314151617 |
- from agents import Agent, Runner, OpenAIChatCompletionsModel
- from openai import AsyncOpenAI
- import os
- BASE_URL = os.getenv("EXAMPLE_BASE_URL") or "https://openrouter.ai/api/v1"
- API_KEY = os.getenv("OPENROUTER_API_KEY")
- MODEL_NAME = "google/gemini-2.5-flash"
- client = AsyncOpenAI(
- base_url=BASE_URL,
- api_key=API_KEY,
- max_retries=5,
- )
- def get_model(model_name=MODEL_NAME):
- return OpenAIChatCompletionsModel(
- openai_client=client,
- model=model_name,
- )
|