RandomZ / app /tests /test_openai_key_env_sync.py
StormShadow308's picture
Fix HF startup when OPENAI_API_KEY secret is missing or only in Space env.
970bad4
Raw
History Blame Contribute Delete
599 Bytes
"""OPENAI_API_KEY resolution from process environment (HF Space secrets)."""
from __future__ import annotations
import pytest
from app.config import Settings
def test_settings_sync_openai_key_from_env_on_hf(monkeypatch: pytest.MonkeyPatch) -> None:
monkeypatch.setenv("SPACE_ID", "user/test")
monkeypatch.setenv("OPENAI_API_KEY", "sk-from-env")
try:
s = Settings(_env_file=None, openai_api_key="")
assert s.openai_api_key == "sk-from-env"
finally:
monkeypatch.delenv("SPACE_ID", raising=False)
monkeypatch.delenv("OPENAI_API_KEY", raising=False)