Merge pull request #685 from ag2ai/polish-suppress-in-tests

Update Claude Sonnet to 3.5 in tests
ag2ai · Jan 28, 2025 · 3528612 · 3528612
2 parents bd0e329 + f516592
commit 3528612
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/test/conftest.py b/test/conftest.py
@@ -316,7 +316,7 @@ def credentials_gemini_flash_exp() -> Credentials:
 def credentials_anthropic_claude_sonnet() -> Credentials:
     return get_llm_credentials(
         "ANTHROPIC_API_KEY",
-        model="claude-3-sonnet-20240229",
+        model="claude-3-5-sonnet-latest",
         api_type="anthropic",
         filter_dict={"tags": ["anthropic-claude-sonnet"]},
     )

diff --git a/test/oai/test_anthropic.py b/test/oai/test_anthropic.py
@@ -134,14 +134,14 @@ def test_cost_calculation(mock_completion):
 @skip_on_missing_imports(["anthropic"], "anthropic")
 def test_load_config(anthropic_client):
     params = {
-        "model": "claude-3-sonnet-20240229",
+        "model": "claude-3-5-sonnet-latest",
         "stream": False,
         "temperature": 1,
         "top_p": 0.8,
         "max_tokens": 100,
     }
     expected_params = {
-        "model": "claude-3-sonnet-20240229",
+        "model": "claude-3-5-sonnet-latest",
         "stream": False,
         "temperature": 1,
         "top_p": 0.8,