github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-17 23:41:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(anthropic): update test model names and adjust token count assertions in integration tests (#32422)

Browse Source
This commit is contained in:
Mason Daugherty
2025-08-12 15:39:35 -04:00
committed by GitHub
parent d5fd0bca35
commit 1167e7458e
1 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
libs/partners/anthropic/tests/integration_tests/test_chat_models.py
View File

@@ -31,8 +31,8 @@ from pydantic import BaseModel, Field
from langchain_anthropic import ChatAnthropic, ChatAnthropicMessages
from tests.unit_tests._utils import FakeCallbackHandler
MODEL_NAME = "claude-3-5-haiku-20241022"
IMAGE_MODEL_NAME = "claude-3-5-sonnet-20241022"
MODEL_NAME = "claude-opus-4-1-20250805"
IMAGE_MODEL_NAME = "claude-opus-4-1-20250805"
def test_stream() -> None:
@@ -128,12 +128,11 @@ async def test_astream() -> None:
async for event in stream:
if event.type == "message_start":
assert event.message.usage.input_tokens > 1
# Note: this single output token included in message start event
# does not appear to contribute to overall output token counts. It
# is excluded from the total token count.
assert event.message.usage.output_tokens == 1
# Different models may report different initial output token counts
# in the message_start event. Ensure it's a positive value.
assert event.message.usage.output_tokens >= 1
elif event.type == "message_delta":
assert event.usage.output_tokens > 1
assert event.usage.output_tokens >= 1
else:
pass