Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 3 additions & 2 deletions tests/integration/test_reasoning.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down Expand Up @@ -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:
Expand Down Expand Up @@ -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:
Expand Down