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

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

@@ -1,346 +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 typing import Any
-from typing import Optional
-
-from fastapi.openapi.models import OAuth2
-from fastapi.openapi.models import OAuthFlowAuthorizationCode
-from fastapi.openapi.models import OAuthFlows
-from google.adk.agents import Agent
-from google.adk.auth import AuthConfig
-from google.adk.auth import AuthCredential
-from google.adk.auth import AuthCredentialTypes
-from google.adk.auth import OAuth2Auth
-from google.adk.flows.llm_flows import functions
-from google.adk.tools import AuthToolArguments
-from google.adk.tools import ToolContext
-from google.genai import types
-
-from ... import utils
-
-
-def function_call(function_call_id, name, args: dict[str, Any]) -> types.Part:
-  part = types.Part.from_function_call(name=name, args=args)
-  part.function_call.id = function_call_id
-  return part
-
-
-def test_function_request_euc():
-  responses = [
-      [
-          types.Part.from_function_call(name='call_external_api1', args={}),
-          types.Part.from_function_call(name='call_external_api2', args={}),
-      ],
-      [
-          types.Part.from_text(text='response1'),
-      ],
-  ]
-
-  auth_config1 = AuthConfig(
-      auth_scheme=OAuth2(
-          flows=OAuthFlows(
-              authorizationCode=OAuthFlowAuthorizationCode(
-                  authorizationUrl='https://accounts.google.com/o/oauth2/auth',
-                  tokenUrl='https://oauth2.googleapis.com/token',
-                  scopes={
-                      'https://www.googleapis.com/auth/calendar': (
-                          'See, edit, share, and permanently delete all the'
-                          ' calendars you can access using Google Calendar'
-                      )
-                  },
-              )
-          )
-      ),
-      raw_auth_credential=AuthCredential(
-          auth_type=AuthCredentialTypes.OAUTH2,
-          oauth2=OAuth2Auth(
-              client_id='oauth_client_id_1',
-              client_secret='oauth_client_secret1',
-          ),
-      ),
-  )
-  auth_config2 = AuthConfig(
-      auth_scheme=OAuth2(
-          flows=OAuthFlows(
-              authorizationCode=OAuthFlowAuthorizationCode(
-                  authorizationUrl='https://accounts.google.com/o/oauth2/auth',
-                  tokenUrl='https://oauth2.googleapis.com/token',
-                  scopes={
-                      'https://www.googleapis.com/auth/calendar': (
-                          'See, edit, share, and permanently delete all the'
-                          ' calendars you can access using Google Calendar'
-                      )
-                  },
-              )
-          )
-      ),
-      raw_auth_credential=AuthCredential(
-          auth_type=AuthCredentialTypes.OAUTH2,
-          oauth2=OAuth2Auth(
-              client_id='oauth_client_id_2',
-              client_secret='oauth_client_secret2',
-          ),
-      ),
-  )
-
-  mock_model = utils.MockModel.create(responses=responses)
-
-  def call_external_api1(tool_context: ToolContext) -> Optional[int]:
-    tool_context.request_credential(auth_config1)
-
-  def call_external_api2(tool_context: ToolContext) -> Optional[int]:
-    tool_context.request_credential(auth_config2)
-
-  agent = Agent(
-      name='root_agent',
-      model=mock_model,
-      tools=[call_external_api1, call_external_api2],
-  )
-  runner = utils.InMemoryRunner(agent)
-  events = runner.run('test')
-  assert events[0].content.parts[0].function_call is not None
-  assert events[0].content.parts[1].function_call is not None
-  auth_configs = list(events[2].actions.requested_auth_configs.values())
-  exchanged_auth_config1 = auth_configs[0]
-  exchanged_auth_config2 = auth_configs[1]
-  assert exchanged_auth_config1.auth_scheme == auth_config1.auth_scheme
-  assert (
-      exchanged_auth_config1.raw_auth_credential
-      == auth_config1.raw_auth_credential
-  )
-  assert (
-      exchanged_auth_config1.exchanged_auth_credential.oauth2.auth_uri
-      is not None
-  )
-  assert exchanged_auth_config2.auth_scheme == auth_config2.auth_scheme
-  assert (
-      exchanged_auth_config2.raw_auth_credential
-      == auth_config2.raw_auth_credential
-  )
-  assert (
-      exchanged_auth_config2.exchanged_auth_credential.oauth2.auth_uri
-      is not None
-  )
-  function_call_ids = list(events[2].actions.requested_auth_configs.keys())
-
-  for idx, part in enumerate(events[1].content.parts):
-    reqeust_euc_function_call = part.function_call
-    assert reqeust_euc_function_call is not None
-    assert (
-        reqeust_euc_function_call.name
-        == functions.REQUEST_EUC_FUNCTION_CALL_NAME
-    )
-    args = AuthToolArguments.model_validate(reqeust_euc_function_call.args)
-
-    assert args.function_call_id == function_call_ids[idx]
-    args.auth_config.auth_scheme.model_extra.clear()
-    assert args.auth_config.auth_scheme == auth_configs[idx].auth_scheme
-    assert (
-        args.auth_config.raw_auth_credential
-        == auth_configs[idx].raw_auth_credential
-    )
-
-
-def test_function_get_auth_response():
-  id_1 = 'id_1'
-  id_2 = 'id_2'
-  responses = [
-      [
-          function_call(id_1, 'call_external_api1', {}),
-          function_call(id_2, 'call_external_api2', {}),
-      ],
-      [
-          types.Part.from_text(text='response1'),
-      ],
-      [
-          types.Part.from_text(text='response2'),
-      ],
-  ]
-
-  mock_model = utils.MockModel.create(responses=responses)
-  function_invoked = 0
-
-  auth_config1 = AuthConfig(
-      auth_scheme=OAuth2(
-          flows=OAuthFlows(
-              authorizationCode=OAuthFlowAuthorizationCode(
-                  authorizationUrl='https://accounts.google.com/o/oauth2/auth',
-                  tokenUrl='https://oauth2.googleapis.com/token',
-                  scopes={
-                      'https://www.googleapis.com/auth/calendar': (
-                          'See, edit, share, and permanently delete all the'
-                          ' calendars you can access using Google Calendar'
-                      )
-                  },
-              )
-          )
-      ),
-      raw_auth_credential=AuthCredential(
-          auth_type=AuthCredentialTypes.OAUTH2,
-          oauth2=OAuth2Auth(
-              client_id='oauth_client_id_1',
-              client_secret='oauth_client_secret1',
-          ),
-      ),
-  )
-  auth_config2 = AuthConfig(
-      auth_scheme=OAuth2(
-          flows=OAuthFlows(
-              authorizationCode=OAuthFlowAuthorizationCode(
-                  authorizationUrl='https://accounts.google.com/o/oauth2/auth',
-                  tokenUrl='https://oauth2.googleapis.com/token',
-                  scopes={
-                      'https://www.googleapis.com/auth/calendar': (
-                          'See, edit, share, and permanently delete all the'
-                          ' calendars you can access using Google Calendar'
-                      )
-                  },
-              )
-          )
-      ),
-      raw_auth_credential=AuthCredential(
-          auth_type=AuthCredentialTypes.OAUTH2,
-          oauth2=OAuth2Auth(
-              client_id='oauth_client_id_2',
-              client_secret='oauth_client_secret2',
-          ),
-      ),
-  )
-
-  auth_response1 = AuthConfig(
-      auth_scheme=OAuth2(
-          flows=OAuthFlows(
-              authorizationCode=OAuthFlowAuthorizationCode(
-                  authorizationUrl='https://accounts.google.com/o/oauth2/auth',
-                  tokenUrl='https://oauth2.googleapis.com/token',
-                  scopes={
-                      'https://www.googleapis.com/auth/calendar': (
-                          'See, edit, share, and permanently delete all the'
-                          ' calendars you can access using Google Calendar'
-                      )
-                  },
-              )
-          )
-      ),
-      raw_auth_credential=AuthCredential(
-          auth_type=AuthCredentialTypes.OAUTH2,
-          oauth2=OAuth2Auth(
-              client_id='oauth_client_id_1',
-              client_secret='oauth_client_secret1',
-          ),
-      ),
-      exchanged_auth_credential=AuthCredential(
-          auth_type=AuthCredentialTypes.OAUTH2,
-          oauth2=OAuth2Auth(
-              client_id='oauth_client_id_1',
-              client_secret='oauth_client_secret1',
-              token={'access_token': 'token1'},
-          ),
-      ),
-  )
-  auth_response2 = AuthConfig(
-      auth_scheme=OAuth2(
-          flows=OAuthFlows(
-              authorizationCode=OAuthFlowAuthorizationCode(
-                  authorizationUrl='https://accounts.google.com/o/oauth2/auth',
-                  tokenUrl='https://oauth2.googleapis.com/token',
-                  scopes={
-                      'https://www.googleapis.com/auth/calendar': (
-                          'See, edit, share, and permanently delete all the'
-                          ' calendars you can access using Google Calendar'
-                      )
-                  },
-              )
-          )
-      ),
-      raw_auth_credential=AuthCredential(
-          auth_type=AuthCredentialTypes.OAUTH2,
-          oauth2=OAuth2Auth(
-              client_id='oauth_client_id_2',
-              client_secret='oauth_client_secret2',
-          ),
-      ),
-      exchanged_auth_credential=AuthCredential(
-          auth_type=AuthCredentialTypes.OAUTH2,
-          oauth2=OAuth2Auth(
-              client_id='oauth_client_id_2',
-              client_secret='oauth_client_secret2',
-              token={'access_token': 'token2'},
-          ),
-      ),
-  )
-
-  def call_external_api1(tool_context: ToolContext) -> int:
-    nonlocal function_invoked
-    function_invoked += 1
-    auth_response = tool_context.get_auth_response(auth_config1)
-    if not auth_response:
-      tool_context.request_credential(auth_config1)
-      return
-    assert auth_response == auth_response1.exchanged_auth_credential
-    return 1
-
-  def call_external_api2(tool_context: ToolContext) -> int:
-    nonlocal function_invoked
-    function_invoked += 1
-    auth_response = tool_context.get_auth_response(auth_config2)
-    if not auth_response:
-      tool_context.request_credential(auth_config2)
-      return
-    assert auth_response == auth_response2.exchanged_auth_credential
-    return 2
-
-  agent = Agent(
-      name='root_agent',
-      model=mock_model,
-      tools=[call_external_api1, call_external_api2],
-  )
-  runner = utils.InMemoryRunner(agent)
-  runner.run('test')
-  request_euc_function_call_event = runner.session.events[-3]
-  function_response1 = types.FunctionResponse(
-      name=request_euc_function_call_event.content.parts[0].function_call.name,
-      response=auth_response1.model_dump(),
-  )
-  function_response1.id = request_euc_function_call_event.content.parts[
-      0
-  ].function_call.id
-
-  function_response2 = types.FunctionResponse(
-      name=request_euc_function_call_event.content.parts[1].function_call.name,
-      response=auth_response2.model_dump(),
-  )
-  function_response2.id = request_euc_function_call_event.content.parts[
-      1
-  ].function_call.id
-  runner.run(
-      new_message=types.Content(
-          role='user',
-          parts=[
-              types.Part(function_response=function_response1),
-              types.Part(function_response=function_response2),
-          ],
-      ),
-  )
-
-  assert function_invoked == 4
-  reqeust = mock_model.requests[-1]
-  content = reqeust.contents[-1]
-  parts = content.parts
-  assert len(parts) == 2
-  assert parts[0].function_response.name == 'call_external_api1'
-  assert parts[0].function_response.response == {'result': 1}
-  assert parts[1].function_response.name == 'call_external_api2'
-  assert parts[1].function_response.response == {'result': 2}

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

@@ -1,93 +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 typing import Any
-
-from google.adk.agents import Agent
-from google.genai import types
-
-from ... import utils
-
-
-def function_call(args: dict[str, Any]) -> types.Part:
-  return types.Part.from_function_call(name='increase_by_one', args=args)
-
-
-def function_response(response: dict[str, Any]) -> types.Part:
-  return types.Part.from_function_response(
-      name='increase_by_one', response=response
-  )
-
-
-def test_sequential_calls():
-  responses = [
-      function_call({'x': 1}),
-      function_call({'x': 2}),
-      function_call({'x': 3}),
-      'response1',
-  ]
-  mockModel = utils.MockModel.create(responses=responses)
-  function_called = 0
-
-  def increase_by_one(x: int) -> int:
-    nonlocal function_called
-    function_called += 1
-    return x + 1
-
-  agent = Agent(name='root_agent', model=mockModel, tools=[increase_by_one])
-  runner = utils.InMemoryRunner(agent)
-  result = utils.simplify_events(runner.run('test'))
-  assert result == [
-      ('root_agent', function_call({'x': 1})),
-      ('root_agent', function_response({'result': 2})),
-      ('root_agent', function_call({'x': 2})),
-      ('root_agent', function_response({'result': 3})),
-      ('root_agent', function_call({'x': 3})),
-      ('root_agent', function_response({'result': 4})),
-      ('root_agent', 'response1'),
-  ]
-
-  # Asserts the requests.
-  assert len(mockModel.requests) == 4
-  # 1 item: user content
-  assert utils.simplify_contents(mockModel.requests[0].contents) == [
-      ('user', 'test')
-  ]
-  # 3 items: user content, functaion call / response for the 1st call
-  assert utils.simplify_contents(mockModel.requests[1].contents) == [
-      ('user', 'test'),
-      ('model', function_call({'x': 1})),
-      ('user', function_response({'result': 2})),
-  ]
-  # 5 items: user content, functaion call / response for two calls
-  assert utils.simplify_contents(mockModel.requests[2].contents) == [
-      ('user', 'test'),
-      ('model', function_call({'x': 1})),
-      ('user', function_response({'result': 2})),
-      ('model', function_call({'x': 2})),
-      ('user', function_response({'result': 3})),
-  ]
-  # 7 items: user content, functaion call / response for three calls
-  assert utils.simplify_contents(mockModel.requests[3].contents) == [
-      ('user', 'test'),
-      ('model', function_call({'x': 1})),
-      ('user', function_response({'result': 2})),
-      ('model', function_call({'x': 2})),
-      ('user', function_response({'result': 3})),
-      ('model', function_call({'x': 3})),
-      ('user', function_response({'result': 4})),
-  ]
-
-  # Asserts the function calls.
-  assert function_called == 3

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

@@ -1,258 +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 typing import Any
-from typing import AsyncGenerator
-from typing import Callable
-
-from google.adk.agents import Agent
-from google.adk.tools import ToolContext
-from google.adk.tools.function_tool import FunctionTool
-from google.genai import types
-import pytest
-
-from ... import utils
-
-
-def test_simple_function():
-  function_call_1 = types.Part.from_function_call(
-      name='increase_by_one', args={'x': 1}
-  )
-  function_respones_2 = types.Part.from_function_response(
-      name='increase_by_one', response={'result': 2}
-  )
-  responses: list[types.Content] = [
-      function_call_1,
-      'response1',
-      'response2',
-      'response3',
-      'response4',
-  ]
-  function_called = 0
-  mock_model = utils.MockModel.create(responses=responses)
-
-  def increase_by_one(x: int) -> int:
-    nonlocal function_called
-    function_called += 1
-    return x + 1
-
-  agent = Agent(name='root_agent', model=mock_model, tools=[increase_by_one])
-  runner = utils.InMemoryRunner(agent)
-  assert utils.simplify_events(runner.run('test')) == [
-      ('root_agent', function_call_1),
-      ('root_agent', function_respones_2),
-      ('root_agent', 'response1'),
-  ]
-
-  # Asserts the requests.
-  assert utils.simplify_contents(mock_model.requests[0].contents) == [
-      ('user', 'test')
-  ]
-  assert utils.simplify_contents(mock_model.requests[1].contents) == [
-      ('user', 'test'),
-      ('model', function_call_1),
-      ('user', function_respones_2),
-  ]
-
-  # Asserts the function calls.
-  assert function_called == 1
-
-
-@pytest.mark.asyncio
-async def test_async_function():
-  function_calls = [
-      types.Part.from_function_call(name='increase_by_one', args={'x': 1}),
-      types.Part.from_function_call(name='multiple_by_two', args={'x': 2}),
-      types.Part.from_function_call(name='multiple_by_two_sync', args={'x': 3}),
-  ]
-  function_responses = [
-      types.Part.from_function_response(
-          name='increase_by_one', response={'result': 2}
-      ),
-      types.Part.from_function_response(
-          name='multiple_by_two', response={'result': 4}
-      ),
-      types.Part.from_function_response(
-          name='multiple_by_two_sync', response={'result': 6}
-      ),
-  ]
-
-  responses: list[types.Content] = [
-      function_calls,
-      'response1',
-      'response2',
-      'response3',
-      'response4',
-  ]
-  function_called = 0
-  mock_model = utils.MockModel.create(responses=responses)
-
-  async def increase_by_one(x: int) -> int:
-    nonlocal function_called
-    function_called += 1
-    return x + 1
-
-  async def multiple_by_two(x: int) -> int:
-    nonlocal function_called
-    function_called += 1
-    return x * 2
-
-  def multiple_by_two_sync(x: int) -> int:
-    nonlocal function_called
-    function_called += 1
-    return x * 2
-
-  agent = Agent(
-      name='root_agent',
-      model=mock_model,
-      tools=[increase_by_one, multiple_by_two, multiple_by_two_sync],
-  )
-  runner = utils.TestInMemoryRunner(agent)
-  events = await runner.run_async_with_new_session('test')
-  assert utils.simplify_events(events) == [
-      ('root_agent', function_calls),
-      ('root_agent', function_responses),
-      ('root_agent', 'response1'),
-  ]
-
-  # Asserts the requests.
-  assert utils.simplify_contents(mock_model.requests[0].contents) == [
-      ('user', 'test')
-  ]
-  assert utils.simplify_contents(mock_model.requests[1].contents) == [
-      ('user', 'test'),
-      ('model', function_calls),
-      ('user', function_responses),
-  ]
-
-  # Asserts the function calls.
-  assert function_called == 3
-
-
-@pytest.mark.asyncio
-async def test_function_tool():
-  function_calls = [
-      types.Part.from_function_call(name='increase_by_one', args={'x': 1}),
-      types.Part.from_function_call(name='multiple_by_two', args={'x': 2}),
-      types.Part.from_function_call(name='multiple_by_two_sync', args={'x': 3}),
-  ]
-  function_responses = [
-      types.Part.from_function_response(
-          name='increase_by_one', response={'result': 2}
-      ),
-      types.Part.from_function_response(
-          name='multiple_by_two', response={'result': 4}
-      ),
-      types.Part.from_function_response(
-          name='multiple_by_two_sync', response={'result': 6}
-      ),
-  ]
-
-  responses: list[types.Content] = [
-      function_calls,
-      'response1',
-      'response2',
-      'response3',
-      'response4',
-  ]
-  function_called = 0
-  mock_model = utils.MockModel.create(responses=responses)
-
-  async def increase_by_one(x: int) -> int:
-    nonlocal function_called
-    function_called += 1
-    return x + 1
-
-  async def multiple_by_two(x: int) -> int:
-    nonlocal function_called
-    function_called += 1
-    return x * 2
-
-  def multiple_by_two_sync(x: int) -> int:
-    nonlocal function_called
-    function_called += 1
-    return x * 2
-
-  class TestTool(FunctionTool):
-
-    def __init__(self, func: Callable[..., Any]):
-      super().__init__(func=func)
-
-  wrapped_increase_by_one = TestTool(func=increase_by_one)
-  agent = Agent(
-      name='root_agent',
-      model=mock_model,
-      tools=[wrapped_increase_by_one, multiple_by_two, multiple_by_two_sync],
-  )
-  runner = utils.TestInMemoryRunner(agent)
-  events = await runner.run_async_with_new_session('test')
-  assert utils.simplify_events(events) == [
-      ('root_agent', function_calls),
-      ('root_agent', function_responses),
-      ('root_agent', 'response1'),
-  ]
-
-  # Asserts the requests.
-  assert utils.simplify_contents(mock_model.requests[0].contents) == [
-      ('user', 'test')
-  ]
-  assert utils.simplify_contents(mock_model.requests[1].contents) == [
-      ('user', 'test'),
-      ('model', function_calls),
-      ('user', function_responses),
-  ]
-
-  # Asserts the function calls.
-  assert function_called == 3
-
-
-def test_update_state():
-  mock_model = utils.MockModel.create(
-      responses=[
-          types.Part.from_function_call(name='update_state', args={}),
-          'response1',
-      ]
-  )
-
-  def update_state(tool_context: ToolContext):
-    tool_context.state['x'] = 1
-
-  agent = Agent(name='root_agent', model=mock_model, tools=[update_state])
-  runner = utils.InMemoryRunner(agent)
-  runner.run('test')
-  assert runner.session.state['x'] == 1
-
-
-def test_function_call_id():
-  responses = [
-      types.Part.from_function_call(name='increase_by_one', args={'x': 1}),
-      'response1',
-  ]
-  mock_model = utils.MockModel.create(responses=responses)
-
-  def increase_by_one(x: int) -> int:
-    return x + 1
-
-  agent = Agent(name='root_agent', model=mock_model, tools=[increase_by_one])
-  runner = utils.InMemoryRunner(agent)
-  events = runner.run('test')
-  for reqeust in mock_model.requests:
-    for content in reqeust.contents:
-      for part in content.parts:
-        if part.function_call:
-          assert part.function_call.id is None
-        if part.function_response:
-          assert part.function_response.id is None
-  assert events[0].content.parts[0].function_call.id.startswith('adk-')
-  assert events[1].content.parts[0].function_response.id.startswith('adk-')