mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-25 01:02:15 +00:00
test(e2e): accept common cat descriptions
This commit is contained in:
parent
3af44daf6d
commit
c508df64fe
1 changed files with 1 additions and 1 deletions
|
|
@ -111,7 +111,7 @@ def _assert_describes_cat(response: ChatResponse) -> None:
|
|||
assert response.choices, f"vision returned no choices: {response}"
|
||||
message = response.choices[0].message
|
||||
content = (message.content if message else None) or ""
|
||||
assert "cat" in content.lower() or "feline" in content.lower(), (
|
||||
assert any(term in content.lower() for term in ("cat", "feline", "kitten", "kitty")), (
|
||||
f"vision response did not describe the image: {content[:200]}"
|
||||
)
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue