github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-27 13:31:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: added test_validate_max_tokens

Browse Source
This commit is contained in:
keenborder786 2025-07-11 06:01:31 +05:00
parent 11ac96c5da
commit c91fc04f03
1 changed files with 39 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
libs/partners/anthropic/tests/unit_tests/test_chat_models.py
View File

@ -62,6 +62,44 @@ def test_anthropic_client_caching() -> None:
assert llm1._client._client is not llm5._client._client
def test_validate_max_tokens() -> None:
"""Test the validate_max_tokens function through class initialization."""
# Test claude-opus-4 models
llm = ChatAnthropic(model="claude-opus-4-20250514", anthropic_api_key="test")
assert llm.max_tokens == 32000
# Test claude-sonnet-4 models
llm = ChatAnthropic(model="claude-sonnet-4-latest", anthropic_api_key="test")
assert llm.max_tokens == 64000
# Test claude-3-7-sonnet models
llm = ChatAnthropic(model="claude-3-7-sonnet-latest", anthropic_api_key="test")
assert llm.max_tokens == 64000
# Test claude-3-5-sonnet models
llm = ChatAnthropic(model="claude-3-5-sonnet-latest", anthropic_api_key="test")
assert llm.max_tokens == 8192
# Test claude-3-5-haiku models
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)
llm = ChatAnthropic(model="claude-3-haiku-latest", anthropic_api_key="test")
assert llm.max_tokens == 1024
# Test that existing max_tokens values are preserved
llm = ChatAnthropic(model="claude-3-5-sonnet-latest", max_tokens=2048, anthropic_api_key="test")
assert llm.max_tokens == 2048
# Test that explicitly set max_tokens values are preserved
llm = ChatAnthropic(model="claude-3-5-sonnet-latest", max_tokens=4096, anthropic_api_key="test")
assert llm.max_tokens == 4096
@pytest.mark.requires("anthropic")
def test_anthropic_model_name_param() -> None:
llm = ChatAnthropic(model_name="foo") # type: ignore[call-arg, call-arg]
@ -104,6 +142,7 @@ def test_anthropic_initialization() -> None:
ChatAnthropic(model="test", anthropic_api_key="test") # type: ignore[call-arg, call-arg]
def test__format_output() -> None:
anthropic_msg = Message(
id="foo",