Skip to content

Commit 1adf5df

Browse files
authored
fix minor issue in e2b (#436)
1 parent b453d63 commit 1adf5df

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/smolagents/e2b_executor.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,7 @@ def __call__(self, code_action: str, additional_args: dict) -> Tuple[Any, Any]:
137137
if getattr(result, attribute_name) is not None:
138138
image_output = getattr(result, attribute_name)
139139
decoded_bytes = base64.b64decode(image_output.encode("utf-8"))
140-
return Image.open(BytesIO(decoded_bytes)), execution_logs
140+
return Image.open(BytesIO(decoded_bytes)), execution_logs, self.final_answer
141141
for attribute_name in [
142142
"chart",
143143
"data",

0 commit comments

Comments
 (0)