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

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

@@ -1,66 +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.flows.llm_flows import identity
-from google.adk.models import LlmRequest
-from google.genai import types
-import pytest
-
-from ... import utils
-
-
-@pytest.mark.asyncio
-async def test_no_description():
-  request = LlmRequest(
-      model="gemini-1.5-flash",
-      config=types.GenerateContentConfig(system_instruction=""),
-  )
-  agent = Agent(model="gemini-1.5-flash", name="agent")
-  invocation_context = utils.create_invocation_context(agent=agent)
-
-  async for _ in identity.request_processor.run_async(
-      invocation_context,
-      request,
-  ):
-    pass
-
-  assert request.config.system_instruction == (
-      """You are an agent. Your internal name is "agent"."""
-  )
-
-
-@pytest.mark.asyncio
-async def test_with_description():
-  request = LlmRequest(
-      model="gemini-1.5-flash",
-      config=types.GenerateContentConfig(system_instruction=""),
-  )
-  agent = Agent(
-      model="gemini-1.5-flash",
-      name="agent",
-      description="test description",
-  )
-  invocation_context = utils.create_invocation_context(agent=agent)
-
-  async for _ in identity.request_processor.run_async(
-      invocation_context,
-      request,
-  ):
-    pass
-
-  assert request.config.system_instruction == "\n\n".join([
-      'You are an agent. Your internal name is "agent".',
-      ' The description about you is "test description"',
-  ])

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

@@ -1,164 +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.agents.readonly_context import ReadonlyContext
-from google.adk.flows.llm_flows import instructions
-from google.adk.models import LlmRequest
-from google.adk.sessions import Session
-from google.genai import types
-import pytest
-
-from ... import utils
-
-
-@pytest.mark.asyncio
-async def test_build_system_instruction():
-  request = LlmRequest(
-      model="gemini-1.5-flash",
-      config=types.GenerateContentConfig(system_instruction=""),
-  )
-  agent = Agent(
-      model="gemini-1.5-flash",
-      name="agent",
-      instruction=("""Use the echo_info tool to echo { customerId }, \
-{{customer_int  }, {  non-identifier-float}}, \
-{'key1': 'value1'} and {{'key2': 'value2'}}."""),
-  )
-  invocation_context = utils.create_invocation_context(agent=agent)
-  invocation_context.session = Session(
-      app_name="test_app",
-      user_id="test_user",
-      id="test_id",
-      state={"customerId": "1234567890", "customer_int": 30},
-  )
-
-  async for _ in instructions.request_processor.run_async(
-      invocation_context,
-      request,
-  ):
-    pass
-
-  assert request.config.system_instruction == (
-      """Use the echo_info tool to echo 1234567890, 30, \
-{  non-identifier-float}}, {'key1': 'value1'} and {{'key2': 'value2'}}."""
-  )
-
-
-@pytest.mark.asyncio
-async def test_function_system_instruction():
-  def build_function_instruction(readonly_context: ReadonlyContext) -> str:
-    return (
-        "This is the function agent instruction for invocation:"
-        f" {readonly_context.invocation_id}."
-    )
-
-  request = LlmRequest(
-      model="gemini-1.5-flash",
-      config=types.GenerateContentConfig(system_instruction=""),
-  )
-  agent = Agent(
-      model="gemini-1.5-flash",
-      name="agent",
-      instruction=build_function_instruction,
-  )
-  invocation_context = utils.create_invocation_context(agent=agent)
-  invocation_context.session = Session(
-      app_name="test_app",
-      user_id="test_user",
-      id="test_id",
-      state={"customerId": "1234567890", "customer_int": 30},
-  )
-
-  async for _ in instructions.request_processor.run_async(
-      invocation_context,
-      request,
-  ):
-    pass
-
-  assert request.config.system_instruction == (
-      "This is the function agent instruction for invocation: test_id."
-  )
-
-
-@pytest.mark.asyncio
-async def test_global_system_instruction():
-  sub_agent = Agent(
-      model="gemini-1.5-flash",
-      name="sub_agent",
-      instruction="This is the sub agent instruction.",
-  )
-  root_agent = Agent(
-      model="gemini-1.5-flash",
-      name="root_agent",
-      global_instruction="This is the global instruction.",
-      sub_agents=[sub_agent],
-  )
-  request = LlmRequest(
-      model="gemini-1.5-flash",
-      config=types.GenerateContentConfig(system_instruction=""),
-  )
-  invocation_context = utils.create_invocation_context(agent=sub_agent)
-  invocation_context.session = Session(
-      app_name="test_app",
-      user_id="test_user",
-      id="test_id",
-      state={"customerId": "1234567890", "customer_int": 30},
-  )
-
-  async for _ in instructions.request_processor.run_async(
-      invocation_context,
-      request,
-  ):
-    pass
-
-  assert request.config.system_instruction == (
-      "This is the global instruction.\n\nThis is the sub agent instruction."
-  )
-
-
-@pytest.mark.asyncio
-async def test_build_system_instruction_with_namespace():
-  request = LlmRequest(
-      model="gemini-1.5-flash",
-      config=types.GenerateContentConfig(system_instruction=""),
-  )
-  agent = Agent(
-      model="gemini-1.5-flash",
-      name="agent",
-      instruction=(
-          """Use the echo_info tool to echo { customerId }, {app:key}, {user:key}, {a:key}."""
-      ),
-  )
-  invocation_context = utils.create_invocation_context(agent=agent)
-  invocation_context.session = Session(
-      app_name="test_app",
-      user_id="test_user",
-      id="test_id",
-      state={
-          "customerId": "1234567890",
-          "app:key": "app_value",
-          "user:key": "user_value",
-      },
-  )
-
-  async for _ in instructions.request_processor.run_async(
-      invocation_context,
-      request,
-  ):
-    pass
-
-  assert request.config.system_instruction == (
-      """Use the echo_info tool to echo 1234567890, app_value, user_value, {a:key}."""
-  )

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

@@ -1,142 +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 google.adk.agents import Agent
-from google.adk.agents.callback_context import CallbackContext
-from google.adk.models import LlmRequest
-from google.adk.models import LlmResponse
-from google.genai import types
-from pydantic import BaseModel
-import pytest
-
-from ... import utils
-
-
-class MockBeforeModelCallback(BaseModel):
-  mock_response: str
-
-  def __call__(
-      self,
-      callback_context: CallbackContext,
-      llm_request: LlmRequest,
-  ) -> LlmResponse:
-    return LlmResponse(
-        content=utils.ModelContent(
-            [types.Part.from_text(text=self.mock_response)]
-        )
-    )
-
-
-class MockAfterModelCallback(BaseModel):
-  mock_response: str
-
-  def __call__(
-      self,
-      callback_context: CallbackContext,
-      llm_response: LlmResponse,
-  ) -> LlmResponse:
-    return LlmResponse(
-        content=utils.ModelContent(
-            [types.Part.from_text(text=self.mock_response)]
-        )
-    )
-
-
-def noop_callback(**kwargs) -> Optional[LlmResponse]:
-  pass
-
-
-def test_before_model_callback():
-  responses = ['model_response']
-  mock_model = utils.MockModel.create(responses=responses)
-  agent = Agent(
-      name='root_agent',
-      model=mock_model,
-      before_model_callback=MockBeforeModelCallback(
-          mock_response='before_model_callback'
-      ),
-  )
-
-  runner = utils.InMemoryRunner(agent)
-  assert utils.simplify_events(runner.run('test')) == [
-      ('root_agent', 'before_model_callback'),
-  ]
-
-
-def test_before_model_callback_noop():
-  responses = ['model_response']
-  mock_model = utils.MockModel.create(responses=responses)
-  agent = Agent(
-      name='root_agent',
-      model=mock_model,
-      before_model_callback=noop_callback,
-  )
-
-  runner = utils.InMemoryRunner(agent)
-  assert utils.simplify_events(runner.run('test')) == [
-      ('root_agent', 'model_response'),
-  ]
-
-
-def test_before_model_callback_end():
-  responses = ['model_response']
-  mock_model = utils.MockModel.create(responses=responses)
-  agent = Agent(
-      name='root_agent',
-      model=mock_model,
-      before_model_callback=MockBeforeModelCallback(
-          mock_response='before_model_callback',
-      ),
-  )
-
-  runner = utils.InMemoryRunner(agent)
-  assert utils.simplify_events(runner.run('test')) == [
-      ('root_agent', 'before_model_callback'),
-  ]
-
-
-def test_after_model_callback():
-  responses = ['model_response']
-  mock_model = utils.MockModel.create(responses=responses)
-  agent = Agent(
-      name='root_agent',
-      model=mock_model,
-      after_model_callback=MockAfterModelCallback(
-          mock_response='after_model_callback'
-      ),
-  )
-
-  runner = utils.InMemoryRunner(agent)
-  assert utils.simplify_events(runner.run('test')) == [
-      ('root_agent', 'after_model_callback'),
-  ]
-
-
-@pytest.mark.asyncio
-async def test_after_model_callback_noop():
-  responses = ['model_response']
-  mock_model = utils.MockModel.create(responses=responses)
-  agent = Agent(
-      name='root_agent',
-      model=mock_model,
-      after_model_callback=noop_callback,
-  )
-
-  runner = utils.TestInMemoryRunner(agent)
-  assert utils.simplify_events(
-      await runner.run_async_with_new_session('test')
-  ) == [('root_agent', 'model_response')]

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

@@ -1,46 +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.genai.types import Part
-from pydantic import BaseModel
-
-from ... import utils
-
-
-def test_output_schema():
-  class CustomOutput(BaseModel):
-    custom_field: str
-
-  response = [
-      'response1',
-  ]
-  mockModel = utils.MockModel.create(responses=response)
-  root_agent = Agent(
-      name='root_agent',
-      model=mockModel,
-      output_schema=CustomOutput,
-      disallow_transfer_to_parent=True,
-      disallow_transfer_to_peers=True,
-  )
-
-  runner = utils.InMemoryRunner(root_agent)
-
-  assert utils.simplify_events(runner.run('test1')) == [
-      ('root_agent', 'response1'),
-  ]
-  assert len(mockModel.requests) == 1
-  assert mockModel.requests[0].config.response_schema == CustomOutput
-  assert mockModel.requests[0].config.response_mime_type == 'application/json'

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

@@ -1,269 +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.adk.tools import BaseTool
-from google.adk.tools import ToolContext
-from google.genai import types
-from google.genai.types import Part
-from pydantic import BaseModel
-
-from ... import utils
-
-
-def simple_function(input_str: str) -> str:
-  return {'result': input_str}
-
-
-class MockBeforeToolCallback(BaseModel):
-  mock_response: dict[str, object]
-  modify_tool_request: bool = False
-
-  def __call__(
-      self,
-      tool: BaseTool,
-      args: dict[str, Any],
-      tool_context: ToolContext,
-  ) -> dict[str, object]:
-    if self.modify_tool_request:
-      args['input_str'] = 'modified_input'
-      return None
-    return self.mock_response
-
-
-class MockAfterToolCallback(BaseModel):
-  mock_response: dict[str, object]
-  modify_tool_request: bool = False
-  modify_tool_response: bool = False
-
-  def __call__(
-      self,
-      tool: BaseTool,
-      args: dict[str, Any],
-      tool_context: ToolContext,
-      tool_response: dict[str, Any] = None,
-  ) -> dict[str, object]:
-    if self.modify_tool_request:
-      args['input_str'] = 'modified_input'
-      return None
-    if self.modify_tool_response:
-      tool_response['result'] = 'modified_output'
-      return tool_response
-    return self.mock_response
-
-
-def noop_callback(
-    **kwargs,
-) -> dict[str, object]:
-  pass
-
-
-def test_before_tool_callback():
-  responses = [
-      types.Part.from_function_call(name='simple_function', args={}),
-      'response1',
-  ]
-  mock_model = utils.MockModel.create(responses=responses)
-  agent = Agent(
-      name='root_agent',
-      model=mock_model,
-      before_tool_callback=MockBeforeToolCallback(
-          mock_response={'test': 'before_tool_callback'}
-      ),
-      tools=[simple_function],
-  )
-
-  runner = utils.InMemoryRunner(agent)
-  assert utils.simplify_events(runner.run('test')) == [
-      ('root_agent', Part.from_function_call(name='simple_function', args={})),
-      (
-          'root_agent',
-          Part.from_function_response(
-              name='simple_function', response={'test': 'before_tool_callback'}
-          ),
-      ),
-      ('root_agent', 'response1'),
-  ]
-
-
-def test_before_tool_callback_noop():
-  responses = [
-      types.Part.from_function_call(
-          name='simple_function', args={'input_str': 'simple_function_call'}
-      ),
-      'response1',
-  ]
-  mock_model = utils.MockModel.create(responses=responses)
-  agent = Agent(
-      name='root_agent',
-      model=mock_model,
-      before_tool_callback=noop_callback,
-      tools=[simple_function],
-  )
-
-  runner = utils.InMemoryRunner(agent)
-  assert utils.simplify_events(runner.run('test')) == [
-      (
-          'root_agent',
-          Part.from_function_call(
-              name='simple_function', args={'input_str': 'simple_function_call'}
-          ),
-      ),
-      (
-          'root_agent',
-          Part.from_function_response(
-              name='simple_function',
-              response={'result': 'simple_function_call'},
-          ),
-      ),
-      ('root_agent', 'response1'),
-  ]
-
-
-def test_before_tool_callback_modify_tool_request():
-  responses = [
-      types.Part.from_function_call(name='simple_function', args={}),
-      'response1',
-  ]
-  mock_model = utils.MockModel.create(responses=responses)
-  agent = Agent(
-      name='root_agent',
-      model=mock_model,
-      before_tool_callback=MockBeforeToolCallback(
-          mock_response={'test': 'before_tool_callback'},
-          modify_tool_request=True,
-      ),
-      tools=[simple_function],
-  )
-
-  runner = utils.InMemoryRunner(agent)
-  assert utils.simplify_events(runner.run('test')) == [
-      ('root_agent', Part.from_function_call(name='simple_function', args={})),
-      (
-          'root_agent',
-          Part.from_function_response(
-              name='simple_function',
-              response={'result': 'modified_input'},
-          ),
-      ),
-      ('root_agent', 'response1'),
-  ]
-
-
-def test_after_tool_callback():
-  responses = [
-      types.Part.from_function_call(
-          name='simple_function', args={'input_str': 'simple_function_call'}
-      ),
-      'response1',
-  ]
-  mock_model = utils.MockModel.create(responses=responses)
-  agent = Agent(
-      name='root_agent',
-      model=mock_model,
-      after_tool_callback=MockAfterToolCallback(
-          mock_response={'test': 'after_tool_callback'}
-      ),
-      tools=[simple_function],
-  )
-
-  runner = utils.InMemoryRunner(agent)
-  assert utils.simplify_events(runner.run('test')) == [
-      (
-          'root_agent',
-          Part.from_function_call(
-              name='simple_function', args={'input_str': 'simple_function_call'}
-          ),
-      ),
-      (
-          'root_agent',
-          Part.from_function_response(
-              name='simple_function', response={'test': 'after_tool_callback'}
-          ),
-      ),
-      ('root_agent', 'response1'),
-  ]
-
-
-def test_after_tool_callback_noop():
-  responses = [
-      types.Part.from_function_call(
-          name='simple_function', args={'input_str': 'simple_function_call'}
-      ),
-      'response1',
-  ]
-  mock_model = utils.MockModel.create(responses=responses)
-  agent = Agent(
-      name='root_agent',
-      model=mock_model,
-      after_tool_callback=noop_callback,
-      tools=[simple_function],
-  )
-
-  runner = utils.InMemoryRunner(agent)
-  assert utils.simplify_events(runner.run('test')) == [
-      (
-          'root_agent',
-          Part.from_function_call(
-              name='simple_function', args={'input_str': 'simple_function_call'}
-          ),
-      ),
-      (
-          'root_agent',
-          Part.from_function_response(
-              name='simple_function',
-              response={'result': 'simple_function_call'},
-          ),
-      ),
-      ('root_agent', 'response1'),
-  ]
-
-
-def test_after_tool_callback_modify_tool_response():
-  responses = [
-      types.Part.from_function_call(
-          name='simple_function', args={'input_str': 'simple_function_call'}
-      ),
-      'response1',
-  ]
-  mock_model = utils.MockModel.create(responses=responses)
-  agent = Agent(
-      name='root_agent',
-      model=mock_model,
-      after_tool_callback=MockAfterToolCallback(
-          mock_response={'result': 'after_tool_callback'},
-          modify_tool_response=True,
-      ),
-      tools=[simple_function],
-  )
-
-  runner = utils.InMemoryRunner(agent)
-  assert utils.simplify_events(runner.run('test')) == [
-      (
-          'root_agent',
-          Part.from_function_call(
-              name='simple_function', args={'input_str': 'simple_function_call'}
-          ),
-      ),
-      (
-          'root_agent',
-          Part.from_function_response(
-              name='simple_function',
-              response={'result': 'modified_output'},
-          ),
-      ),
-      ('root_agent', 'response1'),
-  ]

google/adk/tests/unittests/models/__init__.py

@@ -1,14 +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.
-