mirror of
https://github.com/hwchase17/langchain.git
synced 2025-09-02 11:39:18 +00:00
tests: updated tests
This commit is contained in:
@@ -84,13 +84,9 @@ def test_validate_max_tokens() -> None:
|
|||||||
llm = ChatAnthropic(model="claude-3-5-haiku-latest", anthropic_api_key="test")
|
llm = ChatAnthropic(model="claude-3-5-haiku-latest", anthropic_api_key="test")
|
||||||
assert llm.max_tokens == 8192
|
assert llm.max_tokens == 8192
|
||||||
|
|
||||||
# Test claude-3-5-opus models (should default to 1024)
|
# Test claude-3-haiku models (should default to 4096)
|
||||||
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")
|
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
|
# Test that existing max_tokens values are preserved
|
||||||
llm = ChatAnthropic(
|
llm = ChatAnthropic(
|
||||||
|
Reference in New Issue
Block a user