kanshan/ollama-python
1
0
Fork 0
mirror of https://github.com/ollama/ollama-python.git synced 2026-01-13 21:57:16 +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
description: Optional[str] = None
enum: Optional[Sequence] = 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'] == {
'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'],
}