참고소스 수정본
This commit is contained in:
24
참고/instructor-main/tests/llm/test_openai/conftest.py
Normal file
24
참고/instructor-main/tests/llm/test_openai/conftest.py
Normal file
@@ -0,0 +1,24 @@
|
||||
# conftest.py
|
||||
import os
|
||||
import pytest
|
||||
|
||||
if not os.getenv("OPENAI_API_KEY"):
|
||||
pytest.skip(
|
||||
"OPENAI_API_KEY environment variable not set",
|
||||
allow_module_level=True,
|
||||
)
|
||||
|
||||
try:
|
||||
from openai import AsyncOpenAI, OpenAI
|
||||
except ImportError: # pragma: no cover - optional dependency
|
||||
pytest.skip("openai package is not installed", allow_module_level=True)
|
||||
|
||||
|
||||
@pytest.fixture(scope="function")
|
||||
def client():
|
||||
yield OpenAI()
|
||||
|
||||
|
||||
@pytest.fixture(scope="function")
|
||||
def aclient():
|
||||
yield AsyncOpenAI()
|
||||
Reference in New Issue
Block a user