client.py 484 B

1234567891011121314151617
  1. from agents import Agent, Runner, OpenAIChatCompletionsModel
  2. from openai import AsyncOpenAI
  3. import os
  4. BASE_URL = os.getenv("EXAMPLE_BASE_URL") or "https://openrouter.ai/api/v1"
  5. API_KEY = os.getenv("OPENROUTER_API_KEY")
  6. MODEL_NAME = "google/gemini-2.5-flash"
  7. client = AsyncOpenAI(
  8. base_url=BASE_URL,
  9. api_key=API_KEY,
  10. max_retries=5,
  11. )
  12. def get_model(model_name=MODEL_NAME):
  13. return OpenAIChatCompletionsModel(
  14. openai_client=client,
  15. model=model_name,
  16. )