NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-07-16 04:02:55 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: fix base_llm_flow ut

Browse Source 
PiperOrigin-RevId: 764585592
This commit is contained in:
Xiang (Sean) Zhou 2025-05-28 23:29:12 -07:00 committed by Copybara-Service
parent 0250d9e3ac
commit 96b36b70dc
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/unittests/flows/llm_flows/test_agent_transfer.py
View File

@ -28,7 +28,7 @@ def transfer_call_part(agent_name: str) -> Part:
TRANSFER_RESPONSE_PART = Part.from_function_response( TRANSFER_RESPONSE_PART = Part.from_function_response(
name='transfer_to_agent', response={} name='transfer_to_agent', response={'result': None}
) )
@ -299,7 +299,9 @@ def test_auto_to_loop():
('sub_agent_1_2', Part.from_function_call(name='exit_loop', args={})), ('sub_agent_1_2', Part.from_function_call(name='exit_loop', args={})),
( (
'sub_agent_1_2', 'sub_agent_1_2',
Part.from_function_response(name='exit_loop', response={}), Part.from_function_response(
name='exit_loop', response={'result': None}
),
), ),
# root_agent summarizes. # root_agent summarizes.
('root_agent', 'response4'), ('root_agent', 'response4'),