ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test: mark flaky test

Browse Source
This commit is contained in:
Krrish Dholakia 2025-04-29 06:40:03 -07:00
parent f984089b01
commit ae7eecaa40
1 changed files with 28 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
tests/llm_translation/test_anthropic_completion.py
View File

@ -895,31 +895,35 @@ def test_anthropic_citations_api():
"""
from litellm import completion
resp = completion(
model="claude-3-5-sonnet-20241022",
messages=[
{
"role": "user",
"content": [
{
"type": "document",
"source": {
"type": "text",
"media_type": "text/plain",
"data": "The grass is green. The sky is blue.",
try:
resp = completion(
model="claude-3-5-sonnet-20241022",
messages=[
{
"role": "user",
"content": [
{
"type": "document",
"source": {
"type": "text",
"media_type": "text/plain",
"data": "The grass is green. The sky is blue.",
},
"title": "My Document",
"context": "This is a trustworthy document.",
"citations": {"enabled": True},
},
"title": "My Document",
"context": "This is a trustworthy document.",
"citations": {"enabled": True},
},
{
"type": "text",
"text": "What color is the grass and sky?",
},
],
}
],
)
{
"type": "text",
"text": "What color is the grass and sky?",
},
],
}
],
)
except litellm.InternalServerError:
pytest.skip("Anthropic overloaded")
citations = resp.choices[0].message.provider_specific_fields["citations"]