feat: Added support for enums as arguments for function tools

Author: SOORAJTS2001

src/google/adk/tools/_function_parameter_parse_util.py

@@ -27,6 +27,7 @@
 
 from google.genai import types
 import pydantic
+from enum import Enum
 
 from ..utils.variant_utils import GoogleLLMVariant
 
@@ -145,6 +146,15 @@ def _parse_schema_from_parameter(
     schema.type = _py_builtin_type_to_schema_type[param.annotation]
     _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:
+          if not _is_default_value_compatible(param.default, param.annotation):
+              raise ValueError(default_value_error_msg)
+          schema.default = param.default
+      _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

@@ -22,7 +22,7 @@
 # TODO: crewai requires python 3.10 as minimum
 # from crewai_tools import FileReadTool
 from pydantic import BaseModel
-
+from enum import Enum
 
 def test_string_input():
   def simple_function(input_str: str) -> str:
@@ -219,6 +219,24 @@ 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"
+
+  def simple_function(input:InputEnum):
+    return input.value
+
+  function_decl = _automatic_function_calling_util.build_function_declaration(
+    func=simple_function
+  )
+
+  assert function_decl.name == 'simple_function'
+  assert function_decl.parameters.type == 'OBJECT'
+  assert function_decl.parameters.properties['input'].type == 'STRING'
+  assert function_decl.parameters.properties['input'].enum == ['agent', 'tool']
+
 
 def test_basemodel_list():
   class ChildInput(BaseModel):