Spaces:
Sleeping
Sleeping
File size: 641 Bytes
570f7bd c1bc4eb 1fa9a31 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
import os
from dotenv import load_dotenv
ROOT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
ENV_PATH = os.path.join(ROOT_DIR, ".env")
load_dotenv(dotenv_path=ENV_PATH)
# --- add: make tests independent of real OPENAI_* in CI ---
def _ensure_openai_env_for_tests():
# map PROXY_* -> OPENAI_* or set a harmless dummy
if not os.getenv("OPENAI_API_KEY"):
os.environ["OPENAI_API_KEY"] = os.getenv("PROXY_API_KEY", "DUMMY_TEST_KEY")
if not os.getenv("OPENAI_BASE_URL") and os.getenv("PROXY_BASE_URL"):
os.environ["OPENAI_BASE_URL"] = os.environ["PROXY_BASE_URL"]
_ensure_openai_env_for_tests()
|