diff --git a/tests/llm_translation/test_anthropic_completion.py b/tests/llm_translation/test_anthropic_completion.py index 72fe8f72ee6..eef66dbbed6 100644 --- a/tests/llm_translation/test_anthropic_completion.py +++ b/tests/llm_translation/test_anthropic_completion.py @@ -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 diff --git a/tests/llm_translation/test_gemini.py b/tests/llm_translation/test_gemini.py index 02c1d47ba97..62825e65a1b 100644 --- a/tests/llm_translation/test_gemini.py +++ b/tests/llm_translation/test_gemini.py @@ -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"