scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 108

test(edenai): move live integration tests out of the unit suite

Browse source 
Per review: keep only unit/mocked tests here; live API tests move to tests/e2e/test_edenai_e2e.py.
This commit is contained in:
Victor M. SMITH 2026-07-10 18:32:22 +02:00
parent 7637013587
commit d35bc330eb
1 changed files with 0 additions and 67 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

67
tests/test_litellm/llms/openai_like/test_edenai.py
View file

@ -105,70 +105,3 @@ class TestEdenAICompletionMocked:
assert hasattr(response, "choices")
assert len(response.choices) > 0
assert response.choices[0].message.content == "PING"
class TestEdenAILiveIntegration:
"""Live integration tests against the real Eden AI API.
Skipped automatically when EDENAI_API_KEY is not set, so CI without the
secret stays green. When the key is present (local dev, or a dedicated
integration runner) these validate that each endpoint flagged `true` in
provider_endpoints_support.json actually routes a request through to
api.edenai.run/v3 end-to-end. Endpoints flagged `false` (`/responses`,
`/embeddings`, `/image/generations`, `/audio/*`, `/moderations`,
`/batches`, `/rerank`) intentionally have no live test here; their
handlers ship in follow-up PRs and the tests come with them.
"""
@pytest.mark.skipif(
not os.environ.get("EDENAI_API_KEY"), reason="EDENAI_API_KEY not set"
)
def test_edenai_live_chat_completions(self):
"""`/v1/chat/completions` via litellm.completion() — the openai_like
loader's primary surface."""
response = litellm.completion(
model="edenai/openai/gpt-4o-mini",
messages=[
{
"role": "user",
"content": "Reply with 'PING' and nothing else.",
}
],
max_tokens=8,
)
assert response is not None
assert response.choices[0].message.content
assert response.usage is not None
assert response.usage.total_tokens > 0
@pytest.mark.skipif(
not os.environ.get("EDENAI_API_KEY"), reason="EDENAI_API_KEY not set"
)
@pytest.mark.asyncio
async def test_edenai_live_anthropic_messages(self):
"""`/v1/messages` (Anthropic-format) via litellm.anthropic_messages().
LiteLLM translates the Anthropic-shape request to OpenAI shape
internally and routes through Eden AI's openai-compatible endpoint;
the response is translated back to Anthropic shape before being
returned to the caller."""
response = await litellm.anthropic_messages(
model="edenai/anthropic/claude-opus-4-6",
messages=[
{
"role": "user",
"content": "Reply with 'MSG' and nothing else.",
}
],
max_tokens=8,
)
assert response is not None
# anthropic_messages returns a dict (Anthropic-shape response)
assert response.get("type") == "message"
assert response.get("role") == "assistant"
content = response.get("content", [])
assert content and content[0].get("type") == "text"
assert content[0].get("text")
usage = response.get("usage", {})
assert usage.get("total_tokens", 0) > 0