Skip to content

Commit 20230bc

Browse files
committed
fix: images_to_keep bruh
1 parent 0743dce commit 20230bc

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

src/scrapybara/client.py

+5-5
Original file line numberDiff line numberDiff line change
@@ -1746,7 +1746,7 @@ def act(
17461746
total_tokens=total_tokens,
17471747
)
17481748

1749-
_filter_images(result_messages, images_to_keep)
1749+
_filter_images(result_messages, images_to_keep or 4)
17501750

17511751
return ActResponse(
17521752
messages=result_messages, steps=steps, text=text, output=output, usage=usage
@@ -1821,7 +1821,7 @@ def act_stream(
18211821
# Convert tools to ApiTools
18221822
api_tools = [ApiTool.from_tool(tool) for tool in current_tools]
18231823

1824-
_filter_images(current_messages, images_to_keep)
1824+
_filter_images(current_messages, images_to_keep or 4)
18251825

18261826
request = SingleActRequest(
18271827
model=model,
@@ -2164,7 +2164,7 @@ async def act(
21642164
total_tokens=total_tokens,
21652165
)
21662166

2167-
_filter_images(result_messages, images_to_keep)
2167+
_filter_images(result_messages, images_to_keep or 4)
21682168

21692169
return ActResponse(
21702170
messages=result_messages, steps=steps, text=text, output=output, usage=usage
@@ -2239,7 +2239,7 @@ async def act_stream(
22392239
# Convert tools to ApiTools
22402240
api_tools = [ApiTool.from_tool(tool) for tool in current_tools]
22412241

2242-
_filter_images(current_messages, images_to_keep)
2242+
_filter_images(current_messages, images_to_keep or 4)
22432243

22442244
request = SingleActRequest(
22452245
model=model,
@@ -2387,7 +2387,7 @@ def _create_request_from_action(action):
23872387
else:
23882388
return None
23892389

2390-
def _filter_images(messages: List[Message], images_to_keep: Optional[int] = 4):
2390+
def _filter_images(messages: List[Message], images_to_keep: int):
23912391
"""
23922392
Helper function to filter base64 images in messages, keeping only the latest ones up to specified limit.
23932393

0 commit comments

Comments
 (0)