27 lines
659 B
Python
27 lines
659 B
Python
|
import os
|
||
|
|
import pytest
|
||
|
|
import instructor
|
||
|
|
|
||
|
|
if not os.getenv("OPENAI_API_KEY"):
|
||
|
|
pytest.skip(
|
||
|
|
"OPENAI_API_KEY environment variable not set",
|
||
|
|
allow_module_level=True,
|
||
|
|
)
|
||
|
|
|
||
|
|
try:
|
||
|
|
from litellm import acompletion, completion
|
||
|
|
except ImportError: # pragma: no cover - optional dependency
|
||
|
|
pytest.skip("litellm package is not installed", allow_module_level=True)
|
||
|
|
|
||
|
|
|
||
|
|
def test_litellm_create():
|
||
|
|
client = instructor.from_litellm(completion)
|
||
|
|
|
||
|
|
assert isinstance(client, instructor.Instructor)
|
||
|
|
|
||
|
|
|
||
|
|
def test_async_litellm_create():
|
||
|
|
client = instructor.from_litellm(acompletion)
|
||
|
|
|
||
|
|
assert isinstance(client, instructor.AsyncInstructor)
|