Skip to content

Commit 933907f

Browse files
authored
[Hotfix] Fix ThinkingBlock import error (#794)
1 parent 77eefc2 commit 933907f

2 files changed

Lines changed: 2 additions & 4 deletions

File tree

src/agentscope/model/_model_response.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,12 +4,10 @@
44
from dataclasses import dataclass, field
55
from typing import Literal, Sequence
66

7-
from anthropic.types import ThinkingBlock
8-
97
from ._model_usage import ChatUsage
108
from .._utils._common import _get_timestamp
119
from .._utils._mixin import DictMixin
12-
from ..message import TextBlock, ToolUseBlock
10+
from ..message import TextBlock, ToolUseBlock, ThinkingBlock
1311
from ..types import JSONSerializableObject
1412

1513

tests/token_huggingface_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ async def test_huggingface_token_counter(self) -> None:
2424
"""Test the HuggingFace token counter."""
2525
counter = HuggingFaceTokenCounter(
2626
pretrained_model_name_or_path="Qwen/Qwen3-8B",
27-
use_mirror=True,
27+
use_mirror=False,
2828
)
2929

3030
res = await counter.count(

0 commit comments

Comments
 (0)