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 188

test: update model

Browse source
This commit is contained in:
Krrish Dholakia 2025-07-03 14:30:22 -07:00
parent 7ea0f939f7
commit f6a0eef58a
2 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/llm_translation/test_anthropic_completion.py
View file

@ -1328,6 +1328,8 @@ def test_anthropic_mcp_server_responses_api(model: str):
)
assert response is not None
except litellm.exceptions.InternalServerError:
pass
except Exception as e:
if "Connection to MCP server 'deepwiki' timed out." in str(e):
pass

3
tests/llm_translation/test_gemini.py
View file

@ -210,7 +210,7 @@ def test_gemini_url_context():
{url}
"""
response = completion(
model="gemini/gemini-2.0-flash",
model="gemini/gemini-2.5-flash",
messages=[{"role": "user", "content": prompt}],
tools=[{"urlContext": {}}],
)
@ -219,6 +219,7 @@ def test_gemini_url_context():
assert message is not None
url_context_metadata = response.model_extra["vertex_ai_url_context_metadata"]
assert url_context_metadata is not None
print(f"url_context_metadata: {url_context_metadata}")
urlMetadata = url_context_metadata[0]["urlMetadata"][0]
assert urlMetadata["retrievedUrl"] == url
assert urlMetadata["urlRetrievalStatus"] == "URL_RETRIEVAL_STATUS_SUCCESS"