google-adk 0.0.3__py3-none-any.whl → 0.0.4__py3-none-any.whl

google/adk/tests/unittests/flows/llm_flows/test_agent_transfer.py

@@ -1,311 +0,0 @@
-# Copyright 2025 Google LLC
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-from google.adk.agents.llm_agent import Agent
-from google.adk.agents.loop_agent import LoopAgent
-from google.adk.agents.sequential_agent import SequentialAgent
-from google.adk.tools import exit_loop
-from google.genai.types import Part
-
-from ... import utils
-
-
-def transfer_call_part(agent_name: str) -> Part:
-  return Part.from_function_call(
-      name='transfer_to_agent', args={'agent_name': agent_name}
-  )
-
-
-TRANSFER_RESPONSE_PART = Part.from_function_response(
-    name='transfer_to_agent', response={}
-)
-
-
-def test_auto_to_auto():
-  response = [
-      transfer_call_part('sub_agent_1'),
-      'response1',
-      'response2',
-  ]
-  mockModel = utils.MockModel.create(responses=response)
-  # root (auto) - sub_agent_1 (auto)
-  sub_agent_1 = Agent(name='sub_agent_1', model=mockModel)
-  root_agent = Agent(
-      name='root_agent',
-      model=mockModel,
-      sub_agents=[sub_agent_1],
-  )
-
-  runner = utils.InMemoryRunner(root_agent)
-
-  # Asserts the transfer.
-  assert utils.simplify_events(runner.run('test1')) == [
-      ('root_agent', transfer_call_part('sub_agent_1')),
-      ('root_agent', TRANSFER_RESPONSE_PART),
-      ('sub_agent_1', 'response1'),
-  ]
-
-  # sub_agent_1 should still be the current agent.
-  assert utils.simplify_events(runner.run('test2')) == [
-      ('sub_agent_1', 'response2'),
-  ]
-
-
-def test_auto_to_single():
-  response = [
-      transfer_call_part('sub_agent_1'),
-      'response1',
-      'response2',
-  ]
-  mockModel = utils.MockModel.create(responses=response)
-  # root (auto) - sub_agent_1 (single)
-  sub_agent_1 = Agent(
-      name='sub_agent_1',
-      model=mockModel,
-      disallow_transfer_to_parent=True,
-      disallow_transfer_to_peers=True,
-  )
-  root_agent = Agent(
-      name='root_agent', model=mockModel, sub_agents=[sub_agent_1]
-  )
-
-  runner = utils.InMemoryRunner(root_agent)
-
-  # Asserts the responses.
-  assert utils.simplify_events(runner.run('test1')) == [
-      ('root_agent', transfer_call_part('sub_agent_1')),
-      ('root_agent', TRANSFER_RESPONSE_PART),
-      ('sub_agent_1', 'response1'),
-  ]
-
-  # root_agent should still be the current agent, becaues sub_agent_1 is single.
-  assert utils.simplify_events(runner.run('test2')) == [
-      ('root_agent', 'response2'),
-  ]
-
-
-def test_auto_to_auto_to_single():
-  response = [
-      transfer_call_part('sub_agent_1'),
-      # sub_agent_1 transfers to sub_agent_1_1.
-      transfer_call_part('sub_agent_1_1'),
-      'response1',
-      'response2',
-  ]
-  mockModel = utils.MockModel.create(responses=response)
-  # root (auto) - sub_agent_1 (auto) - sub_agent_1_1 (single)
-  sub_agent_1_1 = Agent(
-      name='sub_agent_1_1',
-      model=mockModel,
-      disallow_transfer_to_parent=True,
-      disallow_transfer_to_peers=True,
-  )
-  sub_agent_1 = Agent(
-      name='sub_agent_1', model=mockModel, sub_agents=[sub_agent_1_1]
-  )
-  root_agent = Agent(
-      name='root_agent', model=mockModel, sub_agents=[sub_agent_1]
-  )
-
-  runner = utils.InMemoryRunner(root_agent)
-
-  # Asserts the responses.
-  assert utils.simplify_events(runner.run('test1')) == [
-      ('root_agent', transfer_call_part('sub_agent_1')),
-      ('root_agent', TRANSFER_RESPONSE_PART),
-      ('sub_agent_1', transfer_call_part('sub_agent_1_1')),
-      ('sub_agent_1', TRANSFER_RESPONSE_PART),
-      ('sub_agent_1_1', 'response1'),
-  ]
-
-  # sub_agent_1 should still be the current agent. sub_agent_1_1 is single so it should
-  # not be the current agent, otherwise the conversation will be tied to
-  # sub_agent_1_1 forever.
-  assert utils.simplify_events(runner.run('test2')) == [
-      ('sub_agent_1', 'response2'),
-  ]
-
-
-def test_auto_to_sequential():
-  response = [
-      transfer_call_part('sub_agent_1'),
-      # sub_agent_1 responds directly instead of transfering.
-      'response1',
-      'response2',
-      'response3',
-  ]
-  mockModel = utils.MockModel.create(responses=response)
-  # root (auto) - sub_agent_1 (sequential) - sub_agent_1_1 (single)
-  #                                   \ sub_agent_1_2 (single)
-  sub_agent_1_1 = Agent(
-      name='sub_agent_1_1',
-      model=mockModel,
-      disallow_transfer_to_parent=True,
-      disallow_transfer_to_peers=True,
-  )
-  sub_agent_1_2 = Agent(
-      name='sub_agent_1_2',
-      model=mockModel,
-      disallow_transfer_to_parent=True,
-      disallow_transfer_to_peers=True,
-  )
-  sub_agent_1 = SequentialAgent(
-      name='sub_agent_1',
-      sub_agents=[sub_agent_1_1, sub_agent_1_2],
-  )
-  root_agent = Agent(
-      name='root_agent',
-      model=mockModel,
-      sub_agents=[sub_agent_1],
-  )
-
-  runner = utils.InMemoryRunner(root_agent)
-
-  # Asserts the transfer.
-  assert utils.simplify_events(runner.run('test1')) == [
-      ('root_agent', transfer_call_part('sub_agent_1')),
-      ('root_agent', TRANSFER_RESPONSE_PART),
-      ('sub_agent_1_1', 'response1'),
-      ('sub_agent_1_2', 'response2'),
-  ]
-
-  # root_agent should still be the current agent because sub_agent_1 is sequential.
-  assert utils.simplify_events(runner.run('test2')) == [
-      ('root_agent', 'response3'),
-  ]
-
-
-def test_auto_to_sequential_to_auto():
-  response = [
-      transfer_call_part('sub_agent_1'),
-      # sub_agent_1 responds directly instead of transfering.
-      'response1',
-      transfer_call_part('sub_agent_1_2_1'),
-      'response2',
-      'response3',
-      'response4',
-  ]
-  mockModel = utils.MockModel.create(responses=response)
-  # root (auto) - sub_agent_1 (seq) - sub_agent_1_1 (single)
-  #                            \ sub_agent_1_2 (auto) - sub_agent_1_2_1 (auto)
-  #                            \ sub_agent_1_3 (single)
-  sub_agent_1_1 = Agent(
-      name='sub_agent_1_1',
-      model=mockModel,
-      disallow_transfer_to_parent=True,
-      disallow_transfer_to_peers=True,
-  )
-  sub_agent_1_2_1 = Agent(name='sub_agent_1_2_1', model=mockModel)
-  sub_agent_1_2 = Agent(
-      name='sub_agent_1_2',
-      model=mockModel,
-      sub_agents=[sub_agent_1_2_1],
-  )
-  sub_agent_1_3 = Agent(
-      name='sub_agent_1_3',
-      model=mockModel,
-      disallow_transfer_to_parent=True,
-      disallow_transfer_to_peers=True,
-  )
-  sub_agent_1 = SequentialAgent(
-      name='sub_agent_1',
-      sub_agents=[sub_agent_1_1, sub_agent_1_2, sub_agent_1_3],
-  )
-  root_agent = Agent(
-      name='root_agent',
-      model=mockModel,
-      sub_agents=[sub_agent_1],
-  )
-
-  runner = utils.InMemoryRunner(root_agent)
-
-  # Asserts the transfer.
-  assert utils.simplify_events(runner.run('test1')) == [
-      ('root_agent', transfer_call_part('sub_agent_1')),
-      ('root_agent', TRANSFER_RESPONSE_PART),
-      ('sub_agent_1_1', 'response1'),
-      ('sub_agent_1_2', transfer_call_part('sub_agent_1_2_1')),
-      ('sub_agent_1_2', TRANSFER_RESPONSE_PART),
-      ('sub_agent_1_2_1', 'response2'),
-      ('sub_agent_1_3', 'response3'),
-  ]
-
-  # root_agent should still be the current agent because sub_agent_1 is sequential.
-  assert utils.simplify_events(runner.run('test2')) == [
-      ('root_agent', 'response4'),
-  ]
-
-
-def test_auto_to_loop():
-  response = [
-      transfer_call_part('sub_agent_1'),
-      # sub_agent_1 responds directly instead of transfering.
-      'response1',
-      'response2',
-      'response3',
-      Part.from_function_call(name='exit_loop', args={}),
-      'response4',
-      'response5',
-  ]
-  mockModel = utils.MockModel.create(responses=response)
-  # root (auto) - sub_agent_1 (loop) - sub_agent_1_1 (single)
-  #                             \ sub_agent_1_2 (single)
-  sub_agent_1_1 = Agent(
-      name='sub_agent_1_1',
-      model=mockModel,
-      disallow_transfer_to_parent=True,
-      disallow_transfer_to_peers=True,
-  )
-  sub_agent_1_2 = Agent(
-      name='sub_agent_1_2',
-      model=mockModel,
-      disallow_transfer_to_parent=True,
-      disallow_transfer_to_peers=True,
-      tools=[exit_loop],
-  )
-  sub_agent_1 = LoopAgent(
-      name='sub_agent_1',
-      sub_agents=[sub_agent_1_1, sub_agent_1_2],
-  )
-  root_agent = Agent(
-      name='root_agent',
-      model=mockModel,
-      sub_agents=[sub_agent_1],
-  )
-
-  runner = utils.InMemoryRunner(root_agent)
-
-  # Asserts the transfer.
-  assert utils.simplify_events(runner.run('test1')) == [
-      # Transfers to sub_agent_1.
-      ('root_agent', transfer_call_part('sub_agent_1')),
-      ('root_agent', TRANSFER_RESPONSE_PART),
-      # Loops.
-      ('sub_agent_1_1', 'response1'),
-      ('sub_agent_1_2', 'response2'),
-      ('sub_agent_1_1', 'response3'),
-      # Exits.
-      ('sub_agent_1_2', Part.from_function_call(name='exit_loop', args={})),
-      (
-          'sub_agent_1_2',
-          Part.from_function_response(name='exit_loop', response={}),
-      ),
-      # root_agent summarizes.
-      ('root_agent', 'response4'),
-  ]
-
-  # root_agent should still be the current agent because sub_agent_1 is loop.
-  assert utils.simplify_events(runner.run('test2')) == [
-      ('root_agent', 'response5'),
-  ]

google/adk/tests/unittests/flows/llm_flows/test_functions_long_running.py

@@ -1,244 +0,0 @@
-# Copyright 2025 Google LLC
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-from google.adk.agents import Agent
-from google.adk.tools import ToolContext
-from google.adk.tools.long_running_tool import LongRunningFunctionTool
-from google.genai.types import Part
-
-from ... import utils
-
-
-def test_async_function():
-  responses = [
-      Part.from_function_call(name='increase_by_one', args={'x': 1}),
-      'response1',
-      'response2',
-      'response3',
-      'response4',
-  ]
-  mockModel = utils.MockModel.create(responses=responses)
-  function_called = 0
-
-  def increase_by_one(x: int, tool_context: ToolContext) -> int:
-    nonlocal function_called
-
-    function_called += 1
-    return {'status': 'pending'}
-
-  # Calls the first time.
-  agent = Agent(
-      name='root_agent',
-      model=mockModel,
-      tools=[LongRunningFunctionTool(func=increase_by_one)],
-  )
-  runner = utils.InMemoryRunner(agent)
-  events = runner.run('test1')
-
-  # Asserts the requests.
-  assert len(mockModel.requests) == 2
-  # 1 item: user content
-  assert mockModel.requests[0].contents == [
-      utils.UserContent('test1'),
-  ]
-  increase_by_one_call = Part.from_function_call(
-      name='increase_by_one', args={'x': 1}
-  )
-  pending_response = Part.from_function_response(
-      name='increase_by_one', response={'status': 'pending'}
-  )
-
-  assert utils.simplify_contents(mockModel.requests[1].contents) == [
-      ('user', 'test1'),
-      ('model', increase_by_one_call),
-      ('user', pending_response),
-  ]
-
-  # Asserts the function calls.
-  assert function_called == 1
-
-  # Asserts the responses.
-  assert utils.simplify_events(events) == [
-      (
-          'root_agent',
-          Part.from_function_call(name='increase_by_one', args={'x': 1}),
-      ),
-      (
-          'root_agent',
-          Part.from_function_response(
-              name='increase_by_one', response={'status': 'pending'}
-          ),
-      ),
-      ('root_agent', 'response1'),
-  ]
-  assert events[0].long_running_tool_ids
-
-  # Updates with another pending progress.
-  still_waiting_response = Part.from_function_response(
-      name='increase_by_one', response={'status': 'still waiting'}
-  )
-  events = runner.run(utils.UserContent(still_waiting_response))
-  # We have one new request.
-  assert len(mockModel.requests) == 3
-  assert utils.simplify_contents(mockModel.requests[2].contents) == [
-      ('user', 'test1'),
-      ('model', increase_by_one_call),
-      ('user', still_waiting_response),
-  ]
-
-  assert utils.simplify_events(events) == [('root_agent', 'response2')]
-
-  # Calls when the result is ready.
-  result_response = Part.from_function_response(
-      name='increase_by_one', response={'result': 2}
-  )
-  events = runner.run(utils.UserContent(result_response))
-  # We have one new request.
-  assert len(mockModel.requests) == 4
-  assert utils.simplify_contents(mockModel.requests[3].contents) == [
-      ('user', 'test1'),
-      ('model', increase_by_one_call),
-      ('user', result_response),
-  ]
-  assert utils.simplify_events(events) == [('root_agent', 'response3')]
-
-  # Calls when the result is ready. Here we still accept the result and do
-  # another summarization. Whether this is the right behavior is TBD.
-  another_result_response = Part.from_function_response(
-      name='increase_by_one', response={'result': 3}
-  )
-  events = runner.run(utils.UserContent(another_result_response))
-  # We have one new request.
-  assert len(mockModel.requests) == 5
-  assert utils.simplify_contents(mockModel.requests[4].contents) == [
-      ('user', 'test1'),
-      ('model', increase_by_one_call),
-      ('user', another_result_response),
-  ]
-  assert utils.simplify_events(events) == [('root_agent', 'response4')]
-
-  # At the end, function_called should still be 1.
-  assert function_called == 1
-
-
-def test_async_function_with_none_response():
-  responses = [
-      Part.from_function_call(name='increase_by_one', args={'x': 1}),
-      'response1',
-      'response2',
-      'response3',
-      'response4',
-  ]
-  mockModel = utils.MockModel.create(responses=responses)
-  function_called = 0
-
-  def increase_by_one(x: int, tool_context: ToolContext) -> int:
-    nonlocal function_called
-    function_called += 1
-    return 'pending'
-
-  # Calls the first time.
-  agent = Agent(
-      name='root_agent',
-      model=mockModel,
-      tools=[LongRunningFunctionTool(func=increase_by_one)],
-  )
-  runner = utils.InMemoryRunner(agent)
-  events = runner.run('test1')
-
-  # Asserts the requests.
-  assert len(mockModel.requests) == 2
-  # 1 item: user content
-  assert mockModel.requests[0].contents == [
-      utils.UserContent('test1'),
-  ]
-  increase_by_one_call = Part.from_function_call(
-      name='increase_by_one', args={'x': 1}
-  )
-
-  assert utils.simplify_contents(mockModel.requests[1].contents) == [
-      ('user', 'test1'),
-      ('model', increase_by_one_call),
-      (
-          'user',
-          Part.from_function_response(
-              name='increase_by_one', response={'result': 'pending'}
-          ),
-      ),
-  ]
-
-  # Asserts the function calls.
-  assert function_called == 1
-
-  # Asserts the responses.
-  assert utils.simplify_events(events) == [
-      (
-          'root_agent',
-          Part.from_function_call(name='increase_by_one', args={'x': 1}),
-      ),
-      (
-          'root_agent',
-          Part.from_function_response(
-              name='increase_by_one', response={'result': 'pending'}
-          ),
-      ),
-      ('root_agent', 'response1'),
-  ]
-
-  # Updates with another pending progress.
-  still_waiting_response = Part.from_function_response(
-      name='increase_by_one', response={'status': 'still waiting'}
-  )
-  events = runner.run(utils.UserContent(still_waiting_response))
-  # We have one new request.
-  assert len(mockModel.requests) == 3
-  assert utils.simplify_contents(mockModel.requests[2].contents) == [
-      ('user', 'test1'),
-      ('model', increase_by_one_call),
-      ('user', still_waiting_response),
-  ]
-
-  assert utils.simplify_events(events) == [('root_agent', 'response2')]
-
-  # Calls when the result is ready.
-  result_response = Part.from_function_response(
-      name='increase_by_one', response={'result': 2}
-  )
-  events = runner.run(utils.UserContent(result_response))
-  # We have one new request.
-  assert len(mockModel.requests) == 4
-  assert utils.simplify_contents(mockModel.requests[3].contents) == [
-      ('user', 'test1'),
-      ('model', increase_by_one_call),
-      ('user', result_response),
-  ]
-  assert utils.simplify_events(events) == [('root_agent', 'response3')]
-
-  # Calls when the result is ready. Here we still accept the result and do
-  # another summarization. Whether this is the right behavior is TBD.
-  another_result_response = Part.from_function_response(
-      name='increase_by_one', response={'result': 3}
-  )
-  events = runner.run(utils.UserContent(another_result_response))
-  # We have one new request.
-  assert len(mockModel.requests) == 5
-  assert utils.simplify_contents(mockModel.requests[4].contents) == [
-      ('user', 'test1'),
-      ('model', increase_by_one_call),
-      ('user', another_result_response),
-  ]
-  assert utils.simplify_events(events) == [('root_agent', 'response4')]
-
-  # At the end, function_called should still be 1.
-  assert function_called == 1