#!/usr/bin/python3
# -*- coding: utf-8 -*-
#
# Copyright (C) 2025 - Present Sepine Tam, Inc. All Rights Reserved
#
# @Author : Sepine Tam (谭淞)
# @Email : sepinetam@gmail.com
# @File : set_model.py
import os
from agents import Model, OpenAIChatCompletionsModel
from openai import AsyncOpenAI
def set_model(model_name: str = None,
api_key: str = None,
base_url: str = "https://api.openai.com/v1",
openai_client: AsyncOpenAI = None) -> Model:
OPENAI_MODEL = os.getenv("OPENAI_MODEL", model_name)
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", api_key)
OPENAI_BASE_URL = os.getenv("OPENAI_BASE_URL", base_url)
OPENAI_CLIENT = openai_client or AsyncOpenAI(
api_key=OPENAI_API_KEY,
base_url=OPENAI_BASE_URL
)
return OpenAIChatCompletionsModel(
model=OPENAI_MODEL,
openai_client=OPENAI_CLIENT
)
