fix: format code with pyink

Author: SOORAJTS2001

src/google/adk/tools/_function_parameter_parse_util.py

@@ -15,6 +15,7 @@
 
 from __future__ import annotations
 
+from enum import Enum
 import inspect
 import logging
 import types as typing_types
@@ -27,7 +28,6 @@
 
 from google.genai import types
 import pydantic
-from enum import Enum
 
 from ..utils.variant_utils import GoogleLLMVariant
 
@@ -147,15 +147,19 @@ def _parse_schema_from_parameter(
     _raise_if_schema_unsupported(variant, schema)
     return schema
   if isinstance(param.annotation, type) and issubclass(param.annotation, Enum):
-      schema.type = types.Type.STRING
-      schema.enum = [e.value for e in param.annotation]
-      if param.default is not inspect.Parameter.empty:
-          default_value = param.default.value if isinstance(param.default, Enum) else param.default
-          if default_value not in schema.enum:
-              raise ValueError(default_value_error_msg)
-          schema.default = default_value
-      _raise_if_schema_unsupported(variant, schema)
-      return schema
+    schema.type = types.Type.STRING
+    schema.enum = [e.value for e in param.annotation]
+    if param.default is not inspect.Parameter.empty:
+      default_value = (
+          param.default.value
+          if isinstance(param.default, Enum)
+          else param.default
+      )
+      if default_value not in schema.enum:
+        raise ValueError(default_value_error_msg)
+      schema.default = default_value
+    _raise_if_schema_unsupported(variant, schema)
+    return schema
   if (
       get_origin(param.annotation) is Union
       # only parse simple UnionType, example int | str | float | bool

tests/unittests/tools/test_build_function_declaration.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from enum import Enum
 from typing import Dict
 from typing import List
 
@@ -22,9 +23,9 @@
 # TODO: crewai requires python 3.10 as minimum
 # from crewai_tools import FileReadTool
 from pydantic import BaseModel
-from enum import Enum
 import pytest
 
+
 def test_string_input():
   def simple_function(input_str: str) -> str:
     return {'result': input_str}
@@ -220,17 +221,18 @@ def simple_function(
   assert function_decl.parameters.properties['input_dir'].type == 'ARRAY'
   assert function_decl.parameters.properties['input_dir'].items.type == 'OBJECT'
 
+
 def test_enums():
 
   class InputEnum(Enum):
-    AGENT = "agent"
-    TOOL = "tool"
+    AGENT = 'agent'
+    TOOL = 'tool'
 
-  def simple_function(input:InputEnum=InputEnum.AGENT):
+  def simple_function(input: InputEnum = InputEnum.AGENT):
     return input.value
 
   function_decl = _automatic_function_calling_util.build_function_declaration(
-    func=simple_function
+      func=simple_function
   )
 
   assert function_decl.name == 'simple_function'
@@ -239,14 +241,15 @@ def simple_function(input:InputEnum=InputEnum.AGENT):
   assert function_decl.parameters.properties['input'].default == 'agent'
   assert function_decl.parameters.properties['input'].enum == ['agent', 'tool']
 
-  def simple_function_with_wrong_enum(input:InputEnum="WRONG_ENUM"):
+  def simple_function_with_wrong_enum(input: InputEnum = 'WRONG_ENUM'):
     return input.value
 
   with pytest.raises(ValueError):
     _automatic_function_calling_util.build_function_declaration(
-      func=simple_function_with_wrong_enum
+        func=simple_function_with_wrong_enum
     )
 
+
 def test_basemodel_list():
   class ChildInput(BaseModel):
     input_str: str