fix: remove EOL model name

2025-09-09 21:15:12 -07:00 · 2025-09-09 21:15:12 -07:00 · f0de7d1dfd
commit f0de7d1dfd
parent d05f58721e
1 changed files with 2 additions and 2 deletions
--- a/tests/llm_translation/test_bedrock_completion.py
+++ b/tests/llm_translation/test_bedrock_completion.py
@ -105,7 +105,7 @@ def test_completion_bedrock_guardrails(streaming):
    try:
        if streaming is False:
            response = completion(
-                model="anthropic.claude-v2",
+                model="anthropic.claude-3-5-sonnet-20240620-v1:0",
                messages=[
                    {
                        "content": "where do i buy coffee from? ",
@ -133,7 +133,7 @@ def test_completion_bedrock_guardrails(streaming):
        else:
            litellm.set_verbose = True
            response = completion(
-                model="anthropic.claude-v2",
+                model="anthropic.claude-3-5-sonnet-20240620-v1:0",
                messages=[
                    {
                        "content": "where do i buy coffee from? ",