1  import importlib
 2  
 3  import openai
 4  import pytest
 5  
 6  from tests.helper_functions import start_mock_openai_server
 7  
 8  
 9  @pytest.fixture(autouse=True)
10  def set_envs(monkeypatch, mock_openai):
11      monkeypatch.setenv("OPENAI_API_KEY", "test")
12      monkeypatch.setenv("OPENAI_API_BASE", mock_openai)
13      importlib.reload(openai)
14  
15  
16  @pytest.fixture(scope="module", autouse=True)
17  def mock_openai():
18      with start_mock_openai_server() as base_url:
19          yield base_url