From 91ca6c5b8f050ceeebd4360982923b7c17093566 Mon Sep 17 00:00:00 2001 From: keenborder786 <21110290@lums.edu.pk> Date: Tue, 29 Jul 2025 05:46:13 +0500 Subject: [PATCH] tests: updated tests --- .../anthropic/tests/unit_tests/test_chat_models.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/libs/partners/anthropic/tests/unit_tests/test_chat_models.py b/libs/partners/anthropic/tests/unit_tests/test_chat_models.py index 3ffc1bf02d6..a626e8a81bd 100644 --- a/libs/partners/anthropic/tests/unit_tests/test_chat_models.py +++ b/libs/partners/anthropic/tests/unit_tests/test_chat_models.py @@ -84,13 +84,9 @@ def test_validate_max_tokens() -> None: llm = ChatAnthropic(model="claude-3-5-haiku-latest", anthropic_api_key="test") assert llm.max_tokens == 8192 - # Test claude-3-5-opus models (should default to 1024) - llm = ChatAnthropic(model="claude-3-5-opus-latest", anthropic_api_key="test") - assert llm.max_tokens == 1024 - - # Test claude-3-haiku models (should default to 1024) + # Test claude-3-haiku models (should default to 4096) llm = ChatAnthropic(model="claude-3-haiku-latest", anthropic_api_key="test") - assert llm.max_tokens == 1024 + assert llm.max_tokens == 4096 # Test that existing max_tokens values are preserved llm = ChatAnthropic(