chore: reformat the codes using autoformat.sh

PiperOrigin-RevId: 762004002
This commit is contained in:
Xiang (Sean) Zhou
2025-05-22 09:43:03 -07:00
committed by Copybara-Service
parent a2263b1808
commit ff8a3c9b43
23 changed files with 496 additions and 447 deletions

View File

@@ -21,6 +21,7 @@ from typing import List
from typing import Optional
from typing import Union
from unittest import mock
from google.adk.agents.base_agent import BaseAgent
from google.adk.agents.callback_context import CallbackContext
from google.adk.agents.invocation_context import InvocationContext
@@ -30,6 +31,7 @@ from google.genai import types
import pytest
import pytest_mock
from typing_extensions import override
from .. import testing_utils

View File

@@ -1,7 +1,11 @@
import pytest
from unittest.mock import MagicMock, AsyncMock, patch
from google.adk.agents.live_request_queue import LiveRequest, LiveRequestQueue
from unittest.mock import AsyncMock
from unittest.mock import MagicMock
from unittest.mock import patch
from google.adk.agents.live_request_queue import LiveRequest
from google.adk.agents.live_request_queue import LiveRequestQueue
from google.genai import types
import pytest
@pytest.mark.asyncio

View File

@@ -15,7 +15,8 @@
"""Unit tests for canonical_xxx fields in LlmAgent."""
from typing import Any
from typing import Optional, cast
from typing import cast
from typing import Optional
from google.adk.agents.callback_context import CallbackContext
from google.adk.agents.invocation_context import InvocationContext
@@ -146,6 +147,7 @@ async def test_canonical_global_instruction():
assert canonical_global_instruction == 'global instruction: state_value'
assert bypass_state_injection
async def test_async_canonical_global_instruction():
async def _global_instruction_provider(ctx: ReadonlyContext) -> str:
return f'global instruction: {ctx.state["state_var"]}'

View File

@@ -1,7 +1,8 @@
import pytest
from unittest.mock import MagicMock
from types import MappingProxyType
from unittest.mock import MagicMock
from google.adk.agents.readonly_context import ReadonlyContext
import pytest
@pytest.fixture

View File

@@ -1,8 +1,10 @@
import pytest
import sys
import logging
from unittest.mock import patch, ANY
import sys
from unittest.mock import ANY
from unittest.mock import patch
from google.adk.agents.run_config import RunConfig
import pytest
def test_validate_max_llm_calls_valid():