from openai import AsyncOpenAI
from dotenv import load_dotenv
import os
from agents import OpenAIChatCompletionsModel

load_dotenv(override=True)

google_api_key = os.getenv('GOOGLE_API_KEY')
GEMINI_BASE_URL = "https://generativelanguage.googleapis.com/v1beta/openai/"
gemini_client = AsyncOpenAI(base_url=GEMINI_BASE_URL, api_key=google_api_key)
gemini_model = OpenAIChatCompletionsModel(model="gemini-2.0-flash", openai_client=gemini_client)

qwen_api_key = "blah"
qwen_base_url = "http://localhost:11434/v1"
qwen_client = AsyncOpenAI(base_url=qwen_base_url, api_key=qwen_api_key)
qwen_model = OpenAIChatCompletionsModel(model="qwen3:14b", openai_client=qwen_client)