Skip to content

Commit 6951949

Browse files
committed
test(workflow): address formatting and released-version review feedback
1 parent c819410 commit 6951949

1 file changed

Lines changed: 3 additions & 7 deletions

File tree

‎tests/unittests/workflow/test_llm_agent_as_node.py‎

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -2158,7 +2158,7 @@ async def test_parallel_single_turn_nodes_with_sub_branch_see_own_node_input(
21582158
):
21592159
"""use_sub_branch=True keeps each parallel single_turn node's own node_input.
21602160
2161-
Main already isolates unbranched parallel single_turn nodes by node path
2161+
ADK 2.12.0 isolates unbranched parallel single_turn nodes by node path
21622162
(097ae1ea3). This covers the branched workaround: with use_sub_branch=True,
21632163
worker_b must still see INPUT_B under include_contents 'none' and 'default'.
21642164
On 2.11.0 without the sub-branch, worker_b lost INPUT_B and only saw
@@ -2210,17 +2210,13 @@ async def fanout(ctx: Context) -> dict[str, Any]:
22102210
await runner.run_async(_testing_utils.get_user_content('start'))
22112211

22122212
assert len(model_a.requests) == 1
2213-
texts_a = [
2214-
p.text for p in _request_parts(model_a.requests[0]) if p.text
2215-
]
2213+
texts_a = [p.text for p in _request_parts(model_a.requests[0]) if p.text]
22162214
assert any('INPUT_A' in t for t in texts_a)
22172215
assert not any('INPUT_B' in t for t in texts_a)
22182216
assert not any('out-b' in t for t in texts_a)
22192217

22202218
assert len(model_b.requests) == 1
2221-
texts_b = [
2222-
p.text for p in _request_parts(model_b.requests[0]) if p.text
2223-
]
2219+
texts_b = [p.text for p in _request_parts(model_b.requests[0]) if p.text]
22242220
# Primary regression: worker_b must keep its own node_input.
22252221
assert any('INPUT_B' in t for t in texts_b)
22262222
assert not any('INPUT_A' in t for t in texts_b)

0 commit comments

Comments
 (0)