diff --git a/tests/integration/test_reasoning.py b/tests/integration/test_reasoning.py index 318b4a777..6d34950e0 100644 --- a/tests/integration/test_reasoning.py +++ b/tests/integration/test_reasoning.py @@ -7,6 +7,7 @@ from any_llm import AnyLLM, LLMProvider from any_llm.exceptions import MissingApiKeyError +from any_llm.providers.anthropic.utils import DEFAULT_MAX_TOKENS from any_llm.types.completion import ChatCompletion, ChatCompletionChunk from tests.constants import EXPECTED_PROVIDERS, LOCAL_PROVIDERS @@ -43,7 +44,7 @@ async def test_completion_reasoning( LLMProvider.PORTKEY, ) else "auto", - max_tokens=100, # Portkey with anthropic needed a max tokens value to be set (because it's an anthropic model) + max_tokens=DEFAULT_MAX_TOKENS, # Portkey with anthropic needed a max tokens value to be set (because it's an anthropic model) ) except MissingApiKeyError: if provider in EXPECTED_PROVIDERS: @@ -96,7 +97,7 @@ async def test_completion_reasoning_streaming( LLMProvider.TOGETHER, ) else "auto", - max_tokens=100, # Portkey with anthropic needed a max tokens value to be set (because it's an anthropic model) + max_tokens=DEFAULT_MAX_TOKENS, # Portkey with anthropic needed a max tokens value to be set (because it's an anthropic model) ) assert isinstance(results, AsyncIterable) async for result in results: