Spaces:
Build error
Build error
File size: 1,003 Bytes
51ff9e5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
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() -> str:
return os.getenv('OPENAI_API_KEY', os.getenv('SANDBOX_ENV_OPENAI_API_KEY', ''))
def _get_openai_base_url() -> str:
return os.getenv('OPENAI_BASE_URL', 'https://api.openai.com/v1')
def _get_openai_model() -> str:
return os.getenv('OPENAI_MODEL', 'gpt-4o')
def _get_max_token() -> int:
return int(os.getenv('MAX_TOKEN', '500'))
def _get_openai_client() -> OpenAI:
client = OpenAI(api_key=_get_openai_api_key(), base_url=_get_openai_base_url())
return client
|