Skip to content

Commit defb3f6

Browse files
committed
Move 128k to constant.
1 parent 894afd2 commit defb3f6

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

python/packages/autogen-ext/src/autogen_ext/agents/web_surfer/_multimodal_web_surfer.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,7 @@
6565
from ._types import InteractiveRegion, UserContent
6666
from .playwright_controller import PlaywrightController
6767

68+
DEFAULT_CONTEXT_SIZE = 128000
6869

6970
class MultimodalWebSurferConfig(BaseModel):
7071
name: str
@@ -863,8 +864,8 @@ async def _summarize_page(
863864
try:
864865
remaining = self._model_client.remaining_tokens(messages + [trial_message])
865866
except KeyError:
866-
# Default to 128k if the model isn't found
867-
remaining = 128000 - self._model_client.count_tokens(messages + [trial_message])
867+
# Use the default if the model isn't found
868+
remaining = DEFAULT_CONTEXT_SIZE - self._model_client.count_tokens(messages + [trial_message])
868869

869870
if self._model_client.model_info["vision"] and remaining <= 0:
870871
break

0 commit comments

Comments
 (0)