wehub-resource-sync
28 lines
722 B
Python
28 lines
722 B
Python
import importlib
|
|
|
|
import openai
|
|
import pytest
|
|
|
|
from tests.helper_functions import start_mock_openai_server
|
|
from tests.tracing.helper import reset_autolog_state # noqa: F401
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def set_envs(monkeypatch, mock_openai):
|
|
monkeypatch.setenv("OPENAI_API_KEY", "test")
|
|
monkeypatch.setenv("OPENAI_API_BASE", mock_openai)
|
|
monkeypatch.setenv("SERPAPI_API_KEY", "test")
|
|
importlib.reload(openai)
|
|
|
|
|
|
@pytest.fixture(scope="module", autouse=True)
|
|
def mock_openai():
|
|
with start_mock_openai_server() as base_url:
|
|
yield base_url
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def reset_autolog(reset_autolog_state):
|
|
# Apply the reset_autolog_state fixture to all tests for LangChain
|
|
return
|