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

google/adk/docs/source/google-adk.rst

@@ -1,98 +0,0 @@
-Submodules
-----------
-
-google.adk.agents module
-------------------------
-
-.. automodule:: google.adk.agents
-    :members:
-    :undoc-members:
-    :show-inheritance:
-
-google.adk.artifacts module
----------------------------
-
-.. automodule:: google.adk.artifacts
-    :members:
-    :undoc-members:
-    :show-inheritance:
-
-google.adk.code_executors module
---------------------------------
-
-.. automodule:: google.adk.code_executors
-    :members:
-    :undoc-members:
-    :show-inheritance:
-
-google.adk.evaluation module
-----------------------------
-
-.. automodule:: google.adk.evaluation
-    :members:
-    :undoc-members:
-    :show-inheritance:
-
-google.adk.events module
-------------------------
-
-.. automodule:: google.adk.events
-    :members:
-    :undoc-members:
-    :show-inheritance:
-
-google.adk.examples module
---------------------------
-
-.. automodule:: google.adk.examples
-    :members:
-    :undoc-members:
-    :show-inheritance:
-
-google.adk.memory module
-------------------------
-
-.. automodule:: google.adk.memory
-    :members:
-    :undoc-members:
-    :show-inheritance:
-
-google.adk.models module
-------------------------
-
-.. automodule:: google.adk.models
-    :members:
-    :undoc-members:
-    :show-inheritance:
-
-google.adk.planners module
---------------------------
-
-.. automodule:: google.adk.planners
-    :members:
-    :undoc-members:
-    :show-inheritance:
-
-google.adk.runners module
---------------------------
-
-.. automodule:: google.adk.runners
-    :members:
-    :undoc-members:
-    :show-inheritance:
-
-google.adk.sessions module
---------------------------
-
-.. automodule:: google.adk.sessions
-    :members:
-    :undoc-members:
-    :show-inheritance:
-
-google.adk.tools module
------------------------
-
-.. automodule:: google.adk.tools
-    :members:
-    :undoc-members:
-    :show-inheritance:

google/adk/docs/source/index.rst

@@ -1,7 +0,0 @@
-google
-======
-
-.. toctree::
-   :maxdepth: 4
-
-   google-adk

google/adk/tests/__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.
-

google/adk/tests/integration/.env.example

@@ -1,10 +0,0 @@
-# Copy as .env file and fill your values below to run integration tests.
-
-# Choose Backend: GOOGLE_AI_ONLY | VERTEX_ONLY | BOTH (default)
-TEST_BACKEND=BOTH
-
-# ML Dev backend config
-GOOGLE_API_KEY=YOUR_VALUE_HERE
-# Vertex backend config
-GOOGLE_CLOUD_PROJECT=YOUR_VALUE_HERE
-GOOGLE_CLOUD_LOCATION=YOUR_VALUE_HERE

google/adk/tests/integration/__init__.py

@@ -1,18 +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.
-
-import pytest
-
-# This allows pytest to show the values of the asserts.
-pytest.register_assert_rewrite('tests.integration.utils')

google/adk/tests/integration/conftest.py

@@ -1,119 +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.
-
-import logging
-import os
-from typing import Literal
-import warnings
-
-from dotenv import load_dotenv
-from google.adk import Agent
-from pytest import fixture
-from pytest import FixtureRequest
-from pytest import hookimpl
-from pytest import Metafunc
-
-from .utils import TestRunner
-
-logger = logging.getLogger(__name__)
-
-
-def load_env_for_tests():
-  dotenv_path = os.path.join(os.path.dirname(__file__), '.env')
-  if not os.path.exists(dotenv_path):
-    warnings.warn(
-        f'Missing .env file at {dotenv_path}. See dotenv.sample for an example.'
-    )
-  else:
-    load_dotenv(dotenv_path, override=True, verbose=True)
-  if 'GOOGLE_API_KEY' not in os.environ:
-    warnings.warn(
-        'Missing GOOGLE_API_KEY in the environment variables. GOOGLE_AI backend'
-        ' integration tests will fail.'
-    )
-  for env_var in [
-      'GOOGLE_CLOUD_PROJECT',
-      'GOOGLE_CLOUD_LOCATION',
-  ]:
-    if env_var not in os.environ:
-      warnings.warn(
-          f'Missing {env_var} in the environment variables. Vertex backend'
-          ' integration tests will fail.'
-      )
-
-
-load_env_for_tests()
-
-BackendType = Literal['GOOGLE_AI', 'VERTEX']
-
-
-@fixture
-def agent_runner(request: FixtureRequest) -> TestRunner:
-  assert isinstance(request.param, dict)
-
-  if 'agent' in request.param:
-    assert isinstance(request.param['agent'], Agent)
-    return TestRunner(request.param['agent'])
-  elif 'agent_name' in request.param:
-    assert isinstance(request.param['agent_name'], str)
-    return TestRunner.from_agent_name(request.param['agent_name'])
-
-  raise NotImplementedError('Must provide agent or agent_name.')
-
-
-@fixture(autouse=True)
-def llm_backend(request: FixtureRequest):
-  # Set backend environment value.
-  original_val = os.environ.get('GOOGLE_GENAI_USE_VERTEXAI')
-  backend_type = request.param
-  if backend_type == 'GOOGLE_AI':
-    os.environ['GOOGLE_GENAI_USE_VERTEXAI'] = '0'
-  else:
-    os.environ['GOOGLE_GENAI_USE_VERTEXAI'] = '1'
-
-  yield  # Run the test
-
-  # Restore the environment
-  if original_val is None:
-    os.environ.pop('GOOGLE_GENAI_USE_VERTEXAI', None)
-  else:
-    os.environ['GOOGLE_GENAI_USE_VERTEXAI'] = original_val
-
-
-@hookimpl(tryfirst=True)
-def pytest_generate_tests(metafunc: Metafunc):
-  if llm_backend.__name__ in metafunc.fixturenames:
-    if not _is_explicitly_marked(llm_backend.__name__, metafunc):
-      test_backend = os.environ.get('TEST_BACKEND', 'BOTH')
-      if test_backend == 'GOOGLE_AI_ONLY':
-        metafunc.parametrize(llm_backend.__name__, ['GOOGLE_AI'], indirect=True)
-      elif test_backend == 'VERTEX_ONLY':
-        metafunc.parametrize(llm_backend.__name__, ['VERTEX'], indirect=True)
-      elif test_backend == 'BOTH':
-        metafunc.parametrize(
-            llm_backend.__name__, ['GOOGLE_AI', 'VERTEX'], indirect=True
-        )
-      else:
-        raise ValueError(
-            f'Invalid TEST_BACKEND value: {test_backend}, should be one of'
-            ' [GOOGLE_AI_ONLY, VERTEX_ONLY, BOTH]'
-        )
-
-
-def _is_explicitly_marked(mark_name: str, metafunc: Metafunc) -> bool:
-  if hasattr(metafunc.function, 'pytestmark'):
-    for mark in metafunc.function.pytestmark:
-      if mark.name == 'parametrize' and mark.args[0] == mark_name:
-        return True
-  return False

google/adk/tests/integration/fixture/__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.
-

google/adk/tests/integration/fixture/agent_with_config/__init__.py

@@ -1,15 +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 . import agent

google/adk/tests/integration/fixture/agent_with_config/agent.py

@@ -1,88 +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 import Agent
-from google.genai import types
-
-new_message = types.Content(
-    role="user",
-    parts=[types.Part.from_text(text="Count a number")],
-)
-
-google_agent_1 = Agent(
-    model="gemini-1.5-flash",
-    name="agent_1",
-    description="The first agent in the team.",
-    instruction="Just say 1",
-    generate_content_config=types.GenerateContentConfig(
-        temperature=0.1,
-    ),
-)
-
-google_agent_2 = Agent(
-    model="gemini-1.5-flash",
-    name="agent_2",
-    description="The second agent in the team.",
-    instruction="Just say 2",
-    generate_content_config=types.GenerateContentConfig(
-        temperature=0.2,
-        safety_settings=[{
-            "category": "HARM_CATEGORY_HATE_SPEECH",
-            "threshold": "BLOCK_ONLY_HIGH",
-        }],
-    ),
-)
-
-google_agent_3 = Agent(
-    model="gemini-1.5-flash",
-    name="agent_3",
-    description="The third agent in the team.",
-    instruction="Just say 3",
-    generate_content_config=types.GenerateContentConfig(
-        temperature=0.5,
-        safety_settings=[{
-            "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
-            "threshold": "BLOCK_NONE",
-        }],
-    ),
-)
-
-google_agent_with_instruction_in_config = Agent(
-    model="gemini-1.5-flash",
-    name="agent",
-    generate_content_config=types.GenerateContentConfig(
-        temperature=0.5, system_instruction="Count 1"
-    ),
-)
-
-
-def function():
-  pass
-
-
-google_agent_with_tools_in_config = Agent(
-    model="gemini-1.5-flash",
-    name="agent",
-    generate_content_config=types.GenerateContentConfig(
-        temperature=0.5, tools=[function]
-    ),
-)
-
-google_agent_with_response_schema_in_config = Agent(
-    model="gemini-1.5-flash",
-    name="agent",
-    generate_content_config=types.GenerateContentConfig(
-        temperature=0.5, response_schema={"key": "value"}
-    ),
-)

google/adk/tests/integration/fixture/callback_agent/__init__.py

@@ -1,15 +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 . import agent

google/adk/tests/integration/fixture/callback_agent/agent.py

@@ -1,105 +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 Optional
-
-from google.adk import Agent
-from google.adk.agents.callback_context import CallbackContext
-from google.adk.agents.invocation_context import InvocationContext
-from google.adk.models import LlmRequest
-from google.adk.models import LlmResponse
-from google.genai import types
-
-
-def before_agent_call_end_invocation(
-    callback_context: CallbackContext,
-) -> types.Content:
-  return types.Content(
-      role='model',
-      parts=[types.Part(text='End invocation event before agent call.')],
-  )
-
-
-def before_agent_call(
-    invocation_context: InvocationContext,
-) -> types.Content:
-  return types.Content(
-      role='model',
-      parts=[types.Part.from_text(text='Plain text event before agent call.')],
-  )
-
-
-def before_model_call_end_invocation(
-    callback_context: CallbackContext, llm_request: LlmRequest
-) -> LlmResponse:
-  return LlmResponse(
-      content=types.Content(
-          role='model',
-          parts=[
-              types.Part.from_text(
-                  text='End invocation event before model call.'
-              )
-          ],
-      )
-  )
-
-
-def before_model_call(
-    invocation_context: InvocationContext, request: LlmRequest
-) -> LlmResponse:
-  request.config.system_instruction = 'Just return 999 as response.'
-  return LlmResponse(
-      content=types.Content(
-          role='model',
-          parts=[
-              types.Part.from_text(
-                  text='Update request event before model call.'
-              )
-          ],
-      )
-  )
-
-
-def after_model_call(
-    callback_context: CallbackContext,
-    llm_response: LlmResponse,
-) -> Optional[LlmResponse]:
-  content = llm_response.content
-  if not content or not content.parts or not content.parts[0].text:
-    return
-
-  content.parts[0].text += 'Update response event after model call.'
-  return llm_response
-
-
-before_agent_callback_agent = Agent(
-    model='gemini-1.5-flash',
-    name='before_agent_callback_agent',
-    instruction='echo 1',
-    before_agent_callback=before_agent_call_end_invocation,
-)
-
-before_model_callback_agent = Agent(
-    model='gemini-1.5-flash',
-    name='before_model_callback_agent',
-    instruction='echo 2',
-    before_model_callback=before_model_call_end_invocation,
-)
-
-after_model_callback_agent = Agent(
-    model='gemini-1.5-flash',
-    name='after_model_callback_agent',
-    instruction='Say hello',
-    after_model_callback=after_model_call,
-)

google/adk/tests/integration/fixture/context_update_test/OWNERS

@@ -1 +0,0 @@
-gkcng

google/adk/tests/integration/fixture/context_update_test/__init__.py

@@ -1,15 +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 . import agent

google/adk/tests/integration/fixture/context_update_test/agent.py

@@ -1,43 +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 List
-from typing import Union
-
-from google.adk import Agent
-from google.adk.tools import ToolContext
-from pydantic import BaseModel
-
-
-def update_fc(
-    data_one: str,
-    data_two: Union[int, float, str],
-    data_three: list[str],
-    data_four: List[Union[int, float, str]],
-    tool_context: ToolContext,
-):
-  """Simply ask to update these variables in the context"""
-  tool_context.actions.update_state("data_one", data_one)
-  tool_context.actions.update_state("data_two", data_two)
-  tool_context.actions.update_state("data_three", data_three)
-  tool_context.actions.update_state("data_four", data_four)
-
-
-root_agent = Agent(
-    model="gemini-1.5-flash",
-    name="root_agent",
-    instruction="Call tools",
-    flow="auto",
-    tools=[update_fc],
-)