Skip to content

Commit 80d1881

Browse files
Jacksunweicopybara-github
authored andcommitted
chore: Fixes a few lint error for tools
PiperOrigin-RevId: 797945021
1 parent 29bb75f commit 80d1881

3 files changed

Lines changed: 3 additions & 5 deletions

File tree

src/google/adk/tools/agent_tool.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,6 @@
1818
from typing import TYPE_CHECKING
1919

2020
from google.genai import types
21-
from pydantic import BaseModel
22-
from pydantic import ConfigDict
2321
from pydantic import model_validator
2422
from typing_extensions import override
2523

@@ -164,8 +162,8 @@ async def run_async(
164162
tool_result = merged_text
165163
return tool_result
166164

167-
@classmethod
168165
@override
166+
@classmethod
169167
def from_config(
170168
cls, config: ToolArgsConfig, config_abs_path: str
171169
) -> AgentTool:

src/google/adk/tools/base_tool.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,6 @@
3030

3131
from google.genai import types
3232
from pydantic import BaseModel
33-
from pydantic import ConfigDict
3433

3534
from ..utils.variant_utils import get_google_llm_variant
3635
from ..utils.variant_utils import GoogleLLMVariant
@@ -40,6 +39,7 @@
4039

4140
if TYPE_CHECKING:
4241
from ..models.llm_request import LlmRequest
42+
from .tool_configs import ToolArgsConfig
4343

4444
SelfTool = TypeVar("SelfTool", bound="BaseTool")
4545

src/google/adk/utils/variant_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ class GoogleLLMVariant(Enum):
3939
"""For using API Key from Google AI Studio"""
4040

4141

42-
def get_google_llm_variant() -> str:
42+
def get_google_llm_variant() -> GoogleLLMVariant:
4343
return (
4444
GoogleLLMVariant.VERTEX_AI
4545
if os.environ.get('GOOGLE_GENAI_USE_VERTEXAI', '0').lower()

0 commit comments

Comments
 (0)