File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1818from typing import TYPE_CHECKING
1919
2020from google .genai import types
21- from pydantic import BaseModel
22- from pydantic import ConfigDict
2321from pydantic import model_validator
2422from 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 :
Original file line number Diff line number Diff line change 3030
3131from google .genai import types
3232from pydantic import BaseModel
33- from pydantic import ConfigDict
3433
3534from ..utils .variant_utils import get_google_llm_variant
3635from ..utils .variant_utils import GoogleLLMVariant
4039
4140if TYPE_CHECKING :
4241 from ..models .llm_request import LlmRequest
42+ from .tool_configs import ToolArgsConfig
4343
4444SelfTool = TypeVar ("SelfTool" , bound = "BaseTool" )
4545
Original file line number Diff line number Diff 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 ()
You can’t perform that action at this time.
0 commit comments