fix minor issue in e2b (#436)

This commit is contained in:
Parteek 2025-01-31 18:23:59 +05:30 committed by GitHub
parent b453d63f33
commit 1adf5df031
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions

View File

@ -137,7 +137,7 @@ locals().update({key: value for key, value in pickle_dict.items()})
if getattr(result, attribute_name) is not None:
image_output = getattr(result, attribute_name)
decoded_bytes = base64.b64decode(image_output.encode("utf-8"))
return Image.open(BytesIO(decoded_bytes)), execution_logs
return Image.open(BytesIO(decoded_bytes)), execution_logs, self.final_answer
for attribute_name in [
"chart",
"data",