obscura/ollama-python
1
0
Fork 0
mirror of https://github.com/ollama/ollama-python.git synced 2026-09-03 18:19:54 +00:00
Code Issues 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
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ollama/_types.py
+1
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
tests/test_utils.py
+2 -2
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'],
} }