fix: relax openapi spec to gemini schema conversion to tolerate more cases

PiperOrigin-RevId: 766985694

.github/release-please.yml

@@ -0,0 +1,5 @@
+releaseType: python
+handleGHRelease: true
+bumpMinorPreMajor: false
+extraFiles:
+  - src/google/adk/version.py

.github/release-trigger.yml

@@ -0,0 +1 @@
+enabled: true

.github/workflows/isort.yml

@@ -0,0 +1,69 @@
+# 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.
+
+name: Check sorting of imports
+
+on:
+  pull_request:
+    paths:
+      - '**.py'
+      - 'pyproject.toml'
+
+jobs:
+  isort-check:
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v4
+        with:
+          fetch-depth: 2
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: '3.x'
+
+      - name: Install isort
+        run: |
+          pip install isort
+
+      - name: Run isort on changed files
+        id: run_isort
+        run: |
+          git fetch origin ${{ github.base_ref }}
+          CHANGED_FILES=$(git diff --diff-filter=ACMR --name-only origin/${{ github.base_ref }}...HEAD | grep -E '\.py$' || true)
+          if [ -n "$CHANGED_FILES" ]; then
+            echo "Changed Python files:"
+            echo "$CHANGED_FILES"
+            echo ""
+            FORMATTED_FILES=$(echo "$CHANGED_FILES" | tr '\n' ' ')
+
+            # Run isort --check
+            set +e
+            isort --check $CHANGED_FILES
+            RESULT=$?
+            set -e
+            if [ $RESULT -ne 0 ]; then
+              echo ""
+              echo "❌ isort check failed!"
+              echo "👉 To fix import order, run locally:"
+              echo ""
+              echo "    isort $FORMATTED_FILES"
+              echo ""
+              exit $RESULT
+            fi
+          else
+            echo "No Python files changed. Skipping isort check."
+          fi

.github/workflows/pyink.yml

@@ -17,8 +17,7 @@ name: Check Pyink Formatting
 on:
   pull_request:
     paths:
-      - 'src/**/*.py'
-      - 'tests/**/*.py'
+      - '**.py'
       - 'pyproject.toml'
 
 jobs:
@@ -28,6 +27,8 @@ jobs:
     steps:
       - name: Checkout code
         uses: actions/checkout@v4
+        with:
+          fetch-depth: 2
 
       - name: Set up Python
         uses: actions/setup-python@v5
@@ -38,36 +39,31 @@ jobs:
         run: |
           pip install pyink
 
-      - name: Detect changed Python files
-        id: detect_changes
+      - name: Run pyink on changed files
+        id: run_pyink
         run: |
           git fetch origin ${{ github.base_ref }}
           CHANGED_FILES=$(git diff --diff-filter=ACMR --name-only origin/${{ github.base_ref }}...HEAD | grep -E '\.py$' || true)
-          echo "CHANGED_FILES=${CHANGED_FILES}" >> $GITHUB_ENV
-
-      - name: Run pyink on changed files
-        if: env.CHANGED_FILES != ''
-        run: |
-          echo "Changed Python files:"
-          echo "$CHANGED_FILES"
-
-          # Run pyink --check
-          set +e
-          pyink --check --config pyproject.toml $CHANGED_FILES
-          RESULT=$?
-          set -e
-
-          if [ $RESULT -ne 0 ]; then
+          if [ -n "$CHANGED_FILES" ]; then
+            echo "Changed Python files:"
+            echo "$CHANGED_FILES"
             echo ""
-            echo "❌ Pyink formatting check failed!"
-            echo "👉 To fix formatting, run locally:"
-            echo ""
-            echo "    pyink --config pyproject.toml $CHANGED_FILES"
-            echo ""
-            exit $RESULT
+            FORMATTED_FILES=$(echo "$CHANGED_FILES" | tr '\n' ' ')
+
+            # Run pyink --check
+            set +e
+            pyink --check --diff --config pyproject.toml $CHANGED_FILES
+            RESULT=$?
+            set -e
+            if [ $RESULT -ne 0 ]; then
+              echo ""
+              echo "❌ Pyink formatting check failed!"
+              echo "👉 To fix formatting, run locally:"
+              echo ""
+              echo "    pyink --config pyproject.toml $FORMATTED_FILES"
+              echo ""
+              exit $RESULT
+            fi
+          else
+            echo "No Python files changed. Skipping pyink check."
           fi
-
-      - name: No changed Python files detected
-        if: env.CHANGED_FILES == ''
-        run: |
-          echo "No Python files changed. Skipping pyink check."

.github/workflows/python-unit-tests.yml

@@ -50,5 +50,4 @@ jobs:
         source .venv/bin/activate
         pytest tests/unittests \
           --ignore=tests/unittests/artifacts/test_artifact_service.py \
-          --ignore=tests/unittests/tools/google_api_tool/test_googleapi_to_openapi_converter.py        
-          
+          --ignore=tests/unittests/tools/google_api_tool/test_googleapi_to_openapi_converter.py

.gitignore

@@ -96,4 +96,4 @@ site/
 Thumbs.db
 *.bak
 *.tmp
-*.temp 
+*.temp

CONTRIBUTING.md

@@ -2,8 +2,7 @@
 
 We'd love to accept your patches and contributions to this project.
 
-## Table of Contents
-
+- [How to contribute](#how-to-contribute)
 - [Before you begin](#before-you-begin)
   - [Sign our Contributor License Agreement](#sign-our-contributor-license-agreement)
   - [Review our community guidelines](#review-our-community-guidelines)
@@ -12,16 +11,16 @@ We'd love to accept your patches and contributions to this project.
   - [Requirement for PRs](#requirement-for-prs)
   - [Large or Complex Changes](#large-or-complex-changes)
   - [Testing Requirements](#testing-requirements)
-  - [Unit Tests](#unit-tests)
-  - [End-to-End (E2E) Tests](#manual-end-to-end-e2e-tests)
+    - [Unit Tests](#unit-tests)
+    - [Manual End-to-End (E2E) Tests](#manual-end-to-end-e2e-tests)
   - [Documentation](#documentation)
   - [Development Setup](#development-setup)
-- [Code reviews](#code-reviews)
+  - [Code reviews](#code-reviews)
 
 
-## Before you begin
+# Before you begin
 
-### Sign our Contributor License Agreement
+## Sign our Contributor License Agreement
 
 Contributions to this project must be accompanied by a
 [Contributor License Agreement](https://cla.developers.google.com/about) (CLA).
@@ -35,38 +34,38 @@ was for a different project), you probably don't need to do it again.
 Visit <https://cla.developers.google.com/> to see your current agreements or to
 sign a new one.
 
-### Review our community guidelines
+## Review our community guidelines
 
 This project follows
 [Google's Open Source Community Guidelines](https://opensource.google/conduct/).
 
-## Contribution workflow
+# Contribution workflow
 
-### Finding Issues to Work On
+## Finding Issues to Work On
 
 - Browse issues labeled **`good first issue`** (newcomer-friendly) or **`help wanted`** (general contributions).
 - For other issues, please kindly ask before contributing to avoid duplication.
 
 
-### Requirement for PRs
+## Requirement for PRs
 
 - All PRs, other than small documentation or typo fixes, should have a Issue assoicated. If not, please create one.
 - Small, focused PRs. Keep changes minimal—one concern per PR.
 - For bug fixes or features, please provide logs or screenshot after the fix is applied to help reviewers better understand the fix.
 - Please include a `testing plan` section in your PR to talk about how you will test. This will save time for PR review. See `Testing Requirements` section for more details.
 
-### Large or Complex Changes
+## Large or Complex Changes
 For substantial features or architectural revisions:
 
 - Open an Issue First: Outline your proposal, including design considerations and impact.
 - Gather Feedback: Discuss with maintainers and the community to ensure alignment and avoid duplicate work
 
-### Testing Requirements
+## Testing Requirements
 
 To maintain code quality and prevent regressions, all code changes must include comprehensive tests and verifiable end-to-end (E2E) evidence.
 
 
-#### Unit Tests
+### Unit Tests
 
 Please add or update unit tests for your change. Please include a summary of passed `pytest` results.
 
@@ -80,7 +79,7 @@ Requirements for unit tests:
   - Free of external dependencies (use mocks or fixtures as needed).
 - **Quality:** Aim for high readability and maintainability; include docstrings or comments for complex scenarios.
 
-#### Manual End-to-End (E2E) Tests
+### Manual End-to-End (E2E) Tests
 
 Manual E2E tests ensure integrated flows work as intended. Your tests should cover all scenarios. Sometimes, it's also good to ensure relevant functionality is not impacted.
 
@@ -97,23 +96,34 @@ Depending on your change:
   - Include the command used and console output showing test results.
   - Highlight sections of the log that directly relate to your change.
 
-### Documentation
+## Documentation
 
 For any changes that impact user-facing documentation (guides, API reference, tutorials), please open a PR in the [adk-docs](https://github.com/google/adk-docs) repository to update relevant part before or alongside your code PR.
 
-### Development Setup
+## Development Setup
 1.  **Clone the repository:**
 
     ```shell
-    git clone git@github.com:google/adk-python.git
+    gh repo clone google/adk-python
     cd adk-python
     ```
-2.  **Create and activate a virtual environment:**
+
+2.  **Install uv:**
+
+    Check out [uv installation guide](https://docs.astral.sh/uv/getting-started/installation/).
+
+3.  **Create and activate a virtual environment:**
+
+    **NOTE**: ADK supports Python 3.9+. Python 3.11 and above is strongly recommended.
+
+    Create a workspace venv using uv.
 
     ```shell
-    python -m venv .venv
+    uv venv --python "python3.11" ".venv"
     ```
 
+    Activate the workspace venv.
+
     ```shell
     source .venv/bin/activate
     ```
@@ -123,39 +133,77 @@ For any changes that impact user-facing documentation (guides, API reference, tu
     source .\.venv\Scripts\activate
     ```
 
-3.  **Install dependencies:**
+4.  **Install dependencies:**
 
     ```shell
-    pip install uv
     uv sync --all-extras
     ```
-4.  **Run unit tests:**
+
+    **NOTE**: for convenience, installing all extra deps as a starting point.
+
+5.  **Run unit tests:**
 
     ```shell
-    uv run pytest ./tests/unittests
+    pytest ./tests/unittests
     ```
-5.  **Run pyink to format codebase:**
+
+    NOTE: for accurately repro test failure, only include `test` and `eval` as
+    extra dependencies.
 
     ```shell
-    uv run pyink  --config pyproject.toml ./src
+    uv sync --extra test --extra eval
+    pytest ./tests/unittests
     ```
 
-6. **Build the package**
+6.  **Auto-format the code:**
+
+    **NOTE**: We use `isort` and `pyink` for styles. Use the included
+    autoformat.sh to auto-format.
+
+    ```shell
+    ./autoformat.sh
+    ```
+
+7. **Build the wheel file:**
+
     ```shell
     uv build
     ```
 
-7. **Local Testing**
-    Have a simple testing folder setup as mentioned in the [quickstart](https://google.github.io/adk-docs/get-started/quickstart/)
-    then install the local package with changes after building it using the below command to test the changes.
+8.  **Test the locally built wheel file:**
+    Have a simple testing folder setup as mentioned in the
+    [quickstart](https://google.github.io/adk-docs/get-started/quickstart/).
+
+    Then following below steps to test your changes:
+
+    Create a clean venv and activate it:
 
     ```shell
-    uv pip install <YOUR_WHL_FILE_PATH>
-
-    [eg]: uv pip install <ADK_PROJECT_PATH>/dist/google_adk-0.4.0-py3-none-any.whl
+    VENV_PATH=~/venvs/adk-quickstart
     ```
 
-### Code reviews
+    ```shell
+    command -v deactivate >/dev/null 2>&1 && deactivate
+    ```
+
+    ```shell
+    rm -rf $VENV_PATH \
+      && python3 -m venv $VENV_PATH \
+      && source $VENV_PATH/bin/activate
+    ```
+
+    Install the locally built wheel file:
+
+    ```shell
+    pip install dist/google_adk-<version>-py3-none-any.whl
+    ```
+
+## Contributing Resources
+
+[Contributing folder](https://github.com/google/adk-python/tree/main/contributing/samples) has resources that is helpful for contributors.
+
+
+## Code reviews
 
 All submissions, including submissions by project members, require review. We
 use GitHub pull requests for this purpose. Consult

README.md

@@ -43,8 +43,8 @@ Agent Development Kit (ADK) is a flexible and modular framework for developing a
 ## 🤖 Agent2Agent (A2A) Protocol and ADK Integration
 
 For remote agent-to-agent communication, ADK integrates with the
-[A2A protocol](https://github.com/google/A2A/).
-See this [example](https://github.com/google/A2A/tree/main/samples/python/agents/google_adk)
+[A2A protocol](https://github.com/google-a2a/A2A/).
+See this [example](https://github.com/google-a2a/a2a-samples/tree/main/samples/python/agents/google_adk)
 for how they can work together.
 
 ## 🚀 Installation

autoformat.sh

@@ -41,6 +41,13 @@ echo '---------------------------------------'
 isort tests/
 echo 'All done! ✨ 🍰 ✨'
 
+echo '---------------------------------------'
+echo '|  Organizing imports for contributing/...'
+echo '---------------------------------------'
+
+isort contributing/
+echo 'All done! ✨ 🍰 ✨'
+
 echo '---------------------------------------'
 echo '|  Auto-formatting src/...'
 echo '---------------------------------------'
@@ -52,3 +59,9 @@ echo '|  Auto-formatting tests/...'
 echo '---------------------------------------'
 
 find -L tests/ -type f -name "*.py" -exec pyink --config pyproject.toml {} +
+
+echo '---------------------------------------'
+echo '|  Auto-formatting contributing/...'
+echo '---------------------------------------'
+
+find -L contributing/ -type f -name "*.py" -exec pyink --config pyproject.toml {} +

contributing/samples/README.md

@@ -0,0 +1,9 @@
+# Contributing Resources
+
+This folder host resources for ADK contributors, for example, testing samples etc.
+
+# Samples
+
+Samples folder host samples to test different features. The samples are usually minimal and simplistic to test one or a few scenarios.
+
+**Note**: This is different from the [google/adk-samples](https://github.com/google/adk-samples) repo, which hosts more complex e2e samples for customers to use or modify directly.

contributing/samples/artifact_save_text/agent.py

@@ -19,8 +19,12 @@ from google.genai import types
 
 
 async def log_query(tool_context: ToolContext, query: str):
-  """Roll a die with the specified number of sides."""
-  await tool_context.save_artifact('query', types.Part(text=query))
+  """Saves the provided query string as a 'text/plain' artifact named 'query'."""
+  query_bytes = query.encode('utf-8')
+  artifact_part = types.Part(
+      inline_data=types.Blob(mime_type='text/plain', data=query_bytes)
+  )
+  await tool_context.save_artifact('query', artifact_part)
 
 
 root_agent = Agent(

contributing/samples/bigquery/agent.py

@@ -19,7 +19,6 @@ from google.adk.tools.bigquery import BigQueryCredentialsConfig
 from google.adk.tools.bigquery import BigQueryToolset
 import google.auth
 
-
 RUN_WITH_ADC = False
 
 
@@ -36,7 +35,6 @@ else:
   credentials_config = BigQueryCredentialsConfig(
       client_id=os.getenv("OAUTH_CLIENT_ID"),
       client_secret=os.getenv("OAUTH_CLIENT_SECRET"),
-      scopes=["https://www.googleapis.com/auth/bigquery"],
   )
 
 bigquery_toolset = BigQueryToolset(credentials_config=credentials_config)

contributing/samples/bigquery_agent/README.md

@@ -1,8 +1,8 @@
-# OAuth Sample
+# BigQuery Sample
 
 ## Introduction
 
-This sample tests and demos the OAuth support in ADK via two tools:
+This sample tests and demos the BigQuery support in ADK via two tools:
 
 * 1. bigquery_datasets_list:
 
@@ -28,14 +28,14 @@ This sample tests and demos the OAuth support in ADK via two tools:
 * 1. Follow https://developers.google.com/identity/protocols/oauth2#1.-obtain-oauth-2.0-credentials-from-the-dynamic_data.setvar.console_name. to get your client id and client secret.
   Be sure to choose "web" as your client type.
 
-* 2. Configure your .env file to add two variables:
+* 2. Configure your `.env` file to add two variables:
 
-  * GOOGLE_CLIENT_ID={your client id}
-  * GOOGLE_CLIENT_SECRET={your client secret}
+  * OAUTH_CLIENT_ID={your client id}
+  * OAUTH_CLIENT_SECRET={your client secret}
 
-  Note: done't create a separate .env , instead put it to the same .env file that stores your Vertex AI or Dev ML credentials
+  Note: don't create a separate `.env` file , instead put it to the same `.env` file that stores your Vertex AI or Dev ML credentials
 
-* 3. Follow https://developers.google.com/identity/protocols/oauth2/web-server#creatingcred to add http://localhost/dev-ui to "Authorized redirect URIs".
+* 3. Follow https://developers.google.com/identity/protocols/oauth2/web-server#creatingcred to add http://localhost/dev-ui/ to "Authorized redirect URIs".
 
   Note: localhost here is just a hostname that you use to access the dev ui, replace it with the actual hostname you use to access the dev ui.
 

contributing/samples/generate_image/agent.py

@@ -12,12 +12,11 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from google.genai import Client
-from google.genai import types
-
 from google.adk import Agent
 from google.adk.tools import load_artifacts
 from google.adk.tools import ToolContext
+from google.genai import Client
+from google.genai import types
 
 # Only Vertex AI supports image generation for now.
 client = Client()

contributing/samples/google_search_agent/agent.py

@@ -12,10 +12,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from google.genai import Client
-
 from google.adk import Agent
 from google.adk.tools import google_search
+from google.genai import Client
 
 # Only Vertex AI supports image generation for now.
 client = Client()

contributing/samples/hello_world/agent.py

@@ -65,9 +65,10 @@ async def check_prime(nums: list[int]) -> str:
       else f"{', '.join(str(num) for num in primes)} are prime numbers."
   )
 
+
 root_agent = Agent(
     model='gemini-2.0-flash',
-    name='data_processing_agent',
+    name='hello_world_agent',
     description=(
         'hello world agent that can roll a dice of 8 sides and check prime'
         ' numbers.'

contributing/samples/hello_world_anthropic/__init__.py

@@ -0,0 +1,16 @@
+# 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

contributing/samples/hello_world_anthropic/agent.py

@@ -0,0 +1,90 @@
+# 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 random
+
+from google.adk import Agent
+from google.adk.models.anthropic_llm import Claude
+
+
+def roll_die(sides: int) -> int:
+  """Roll a die and return the rolled result.
+
+  Args:
+    sides: The integer number of sides the die has.
+
+  Returns:
+    An integer of the result of rolling the die.
+  """
+  return random.randint(1, sides)
+
+
+async def check_prime(nums: list[int]) -> str:
+  """Check if a given list of numbers are prime.
+
+  Args:
+    nums: The list of numbers to check.
+
+  Returns:
+    A str indicating which number is prime.
+  """
+  primes = set()
+  for number in nums:
+    number = int(number)
+    if number <= 1:
+      continue
+    is_prime = True
+    for i in range(2, int(number**0.5) + 1):
+      if number % i == 0:
+        is_prime = False
+        break
+    if is_prime:
+      primes.add(number)
+  return (
+      "No prime numbers found."
+      if not primes
+      else f"{', '.join(str(num) for num in primes)} are prime numbers."
+  )
+
+
+root_agent = Agent(
+    model=Claude(model="claude-3-5-sonnet-v2@20241022"),
+    name="data_processing_agent",
+    description=(
+        "hello world agent that can roll a dice of 8 sides and check prime"
+        " numbers."
+    ),
+    instruction="""
+      You roll dice and answer questions about the outcome of the dice rolls.
+      You can roll dice of different sizes.
+      You can use multiple tools in parallel by calling functions in parallel(in one request and in one round).
+      It is ok to discuss previous dice roles, and comment on the dice rolls.
+      When you are asked to roll a die, you must call the roll_die tool with the number of sides. Be sure to pass in an integer. Do not pass in a string.
+      You should never roll a die on your own.
+      When checking prime numbers, call the check_prime tool with a list of integers. Be sure to pass in a list of integers. You should never pass in a string.
+      You should not check prime numbers before calling the tool.
+      When you are asked to roll a die and check prime numbers, you should always make the following two function calls:
+      1. You should first call the roll_die tool to get a roll. Wait for the function response before calling the check_prime tool.
+      2. After you get the function response from roll_die tool, you should call the check_prime tool with the roll_die result.
+        2.1 If user asks you to check primes based on previous rolls, make sure you include the previous rolls in the list.
+      3. When you respond, you must include the roll_die result from step 1.
+      You should always perform the previous 3 steps when asking for a roll and checking prime numbers.
+      You should not rely on the previous history on prime results.
+    """,
+    tools=[
+        roll_die,
+        check_prime,
+    ],
+)

contributing/samples/hello_world_anthropic/main.py

@@ -0,0 +1,76 @@
+# 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 asyncio
+import time
+
+import agent
+from dotenv import load_dotenv
+from google.adk import Runner
+from google.adk.artifacts import InMemoryArtifactService
+from google.adk.cli.utils import logs
+from google.adk.sessions import InMemorySessionService
+from google.adk.sessions import Session
+from google.genai import types
+
+load_dotenv(override=True)
+logs.log_to_tmp_folder()
+
+
+async def main():
+  app_name = 'my_app'
+  user_id_1 = 'user1'
+  session_service = InMemorySessionService()
+  artifact_service = InMemoryArtifactService()
+  runner = Runner(
+      app_name=app_name,
+      agent=agent.root_agent,
+      artifact_service=artifact_service,
+      session_service=session_service,
+  )
+  session_11 = await session_service.create_session(
+      app_name=app_name, user_id=user_id_1
+  )
+
+  async def run_prompt(session: Session, new_message: str):
+    content = types.Content(
+        role='user', parts=[types.Part.from_text(text=new_message)]
+    )
+    print('** User says:', content.model_dump(exclude_none=True))
+    async for event in runner.run_async(
+        user_id=user_id_1,
+        session_id=session.id,
+        new_message=content,
+    ):
+      if event.content.parts and event.content.parts[0].text:
+        print(f'** {event.author}: {event.content.parts[0].text}')
+
+  start_time = time.time()
+  print('Start time:', start_time)
+  print('------------------------------------')
+  await run_prompt(session_11, 'Hi, introduce yourself.')
+  await run_prompt(
+      session_11,
+      'Run the following request 10 times: roll a die with 100 sides and check'
+      ' if it is prime',
+  )
+  end_time = time.time()
+  print('------------------------------------')
+  print('End time:', end_time)
+  print('Total time:', end_time - start_time)
+
+
+if __name__ == '__main__':
+  asyncio.run(main())

contributing/samples/human_in_loop/main.py

@@ -13,25 +13,23 @@
 # limitations under the License.
 
 import asyncio
+import os
+from typing import Any
+from typing import Union
 
 import agent
 from dotenv import load_dotenv
-from typing import Any
-from typing import Union
 from google.adk.agents import Agent
 from google.adk.events import Event
 from google.adk.runners import Runner
-from google.adk.tools import LongRunningFunctionTool
 from google.adk.sessions import InMemorySessionService
+from google.adk.tools import LongRunningFunctionTool
 from google.genai import types
-
-import os
 from opentelemetry import trace
 from opentelemetry.exporter.cloud_trace import CloudTraceSpanExporter
 from opentelemetry.sdk.trace import export
 from opentelemetry.sdk.trace import TracerProvider
 
-
 load_dotenv(override=True)
 
 APP_NAME = "human_in_the_loop"