mirror of
https://github.com/EvolutionAPI/adk-python.git
synced 2025-07-14 09:51:25 -06:00
fix: add missing async change in code execution.
PiperOrigin-RevId: 755429199
This commit is contained in:
parent
28abc306c3
commit
e2ba484b43
@ -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
|
||||||
)
|
)
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user