import os

from openai import OpenAI


# ==================================================================================================
# OPENAI
# TODO: Move this to EventStream Actions when DockerRuntime is fully implemented
# NOTE: we need to get env vars inside functions because they will be set in IPython
# AFTER the agentskills is imported (the case for DockerRuntime)
# ==================================================================================================
def _get_openai_api_key():
    return os.getenv('OPENAI_API_KEY', os.getenv('SANDBOX_ENV_OPENAI_API_KEY', ''))


def _get_openai_base_url():
    return os.getenv('OPENAI_BASE_URL', 'https://api.openai.com/v1')


def _get_openai_model():
    return os.getenv('OPENAI_MODEL', 'gpt-4o')


def _get_max_token():
    return os.getenv('MAX_TOKEN', 500)


def _get_openai_client():
    client = OpenAI(api_key=_get_openai_api_key(), base_url=_get_openai_base_url())
    return client