types: allow single or multiple types tool property (#499)

Author: ParthSareen

ollama/_types.py

@@ -313,7 +313,7 @@ class Parameters(SubscriptableBaseModel):
       class Property(SubscriptableBaseModel):
         model_config = ConfigDict(arbitrary_types_allowed=True)
 
-        type: Optional[str] = None
+        type: Optional[Union[str, Sequence[str]]] = None
         description: Optional[str] = None
         enum: Optional[Sequence] = None
 

tests/test_client.py

@@ -1139,7 +1139,7 @@ def func2(y: str) -> int:
       'description': 'Test function',
       'parameters': {
         'type': 'object',
-        'properties': {'x': {'type': 'string', 'description': 'A string'}},
+        'properties': {'x': {'type': 'string', 'description': 'A string', 'enum': ['a', 'b', 'c']}, 'y': {'type': ['integer', 'number'], 'description': 'An integer'}},
         'required': ['x'],
       },
     },