types: relax enum type for tool (#498)

Author: ParthSareen

ollama/_types.py

@@ -315,6 +315,7 @@ class Property(SubscriptableBaseModel):
 
         type: Optional[str] = None
         description: Optional[str] = None
+        enum: Optional[Sequence] = None
 
       properties: Optional[Mapping[str, Property]] = None
 

tests/test_utils.py

@@ -200,8 +200,8 @@ def only_description_with_args(x: int, y: int):
   assert tool['function']['parameters'] == {
     'type': 'object',
     'properties': {
-      'x': {'type': 'integer', 'description': ''},
-      'y': {'type': 'integer', 'description': ''},
+      'x': {'type': 'integer', 'description': '', 'enum': None},
+      'y': {'type': 'integer', 'description': '', 'enum': None},
     },
     'required': ['x', 'y'],
   }