fix: Fix cases where execution_result_delimiters have None type element due to code_execution_result.output is None

PiperOrigin-RevId: 861933883
This commit is contained in:
Google Team Member
2026-01-27 16:29:51 -08:00
committed by Copybara-Service
parent 8f7d9659cf
commit a16e3cc67e
@@ -251,9 +251,13 @@ class CodeExecutionUtils:
# Skip if the Content has multiple parts, which means the Content is
# likely generated by the model.
elif len(content.parts) == 1 and content.parts[-1].code_execution_result:
content.parts[-1] = types.Part(
text=execution_result_delimiters[0]
+ content.parts[-1].code_execution_result.output
+ execution_result_delimiters[1]
)
output = content.parts[-1].code_execution_result.output
if output is not None:
content.parts[-1] = types.Part(
text=execution_result_delimiters[0]
+ output
+ execution_result_delimiters[1]
)
else:
content.parts[-1] = types.Part(text='')
content.role = 'user'