fix: add missing async change in code execution.

PiperOrigin-RevId: 755429199
This commit is contained in:
Shangjie Chen 2025-05-06 10:36:51 -07:00 committed by Copybara-Service
parent 28abc306c3
commit e2ba484b43

View File

@ -152,7 +152,7 @@ class _CodeExecutionResponseProcessor(BaseLlmResponseProcessor):
if llm_response.partial: if llm_response.partial:
return return
for event in _run_post_processor(invocation_context, llm_response): async for event in _run_post_processor(invocation_context, llm_response):
yield event yield event
@ -249,10 +249,10 @@ async def _run_pre_processor(
llm_request.contents.append(copy.deepcopy(execution_result_event.content)) llm_request.contents.append(copy.deepcopy(execution_result_event.content))
def _run_post_processor( async def _run_post_processor(
invocation_context: InvocationContext, invocation_context: InvocationContext,
llm_response, llm_response,
) -> Generator[Event, None, None]: ) -> AsyncGenerator[Event, None]:
"""Post-process the model response by extracting and executing the first code block.""" """Post-process the model response by extracting and executing the first code block."""
agent = invocation_context.agent agent = invocation_context.agent
code_executor = agent.code_executor code_executor = agent.code_executor
@ -305,7 +305,7 @@ def _run_post_processor(
code_execution_result.stdout, code_execution_result.stdout,
code_execution_result.stderr, code_execution_result.stderr,
) )
yield _post_process_code_execution_result( yield await _post_process_code_execution_result(
invocation_context, code_executor_context, code_execution_result invocation_context, code_executor_context, code_execution_result
) )