Files
567-labs--instructor/tests/llm/test_litellm.py
T
wehub-resource-sync 97e91a83f3
Ruff / Ruff (push) Has been cancelled
Test / Core Tests (push) Has been cancelled
Test / Offline Coverage Tests (Python 3.10) (push) Has been cancelled
Test / Offline Coverage Tests (Python 3.11) (push) Has been cancelled
Test / Offline Coverage Tests (Python 3.12) (push) Has been cancelled
Test / Offline Coverage Tests (Python 3.13) (push) Has been cancelled
Test / Offline Coverage Tests (Python 3.9) (push) Has been cancelled
Test / Full Coverage (Python 3.11) (push) Has been cancelled
Test / Core Provider Tests (OpenAI) (push) Has been cancelled
Test / Core Provider Tests (Anthropic) (push) Has been cancelled
Test / Core Provider Tests (Google) (push) Has been cancelled
Test / Core Provider Tests (Other) (push) Has been cancelled
Test / Anthropic Tests (push) Has been cancelled
Test / Gemini Tests (push) Has been cancelled
Test / Google GenAI Tests (push) Has been cancelled
Test / Vertex AI Tests (push) Has been cancelled
Test / OpenAI Tests (push) Has been cancelled
Test / Writer Tests (push) Has been cancelled
Test / Auto Client Tests (push) Has been cancelled
ty / type-check (push) Has been cancelled
chore: import upstream snapshot with attribution
2026-07-13 13:36:38 +08:00

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)