kanshan/ollama-python
1
0
Fork 0
mirror of https://github.com/ollama/ollama-python.git synced 2026-05-01 11:48:17 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

types: relax enum type for tool (#498)

Browse Source
This commit is contained in:
Parth Sareen 2025-04-08 15:05:46 -07:00 committed by GitHub
parent 3eaa83781d
commit 1b3406887d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ollama/_types.py
View File

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

4
tests/test_utils.py
View File

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