Skip to content

Commit d4c3897

Browse files
Update ollama/_types.py
Co-authored-by: Parth Sareen <[email protected]>
1 parent d8d98e1 commit d4c3897

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

ollama/_types.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -93,6 +93,9 @@ def get(self, key: str, default: Any = None) -> Any:
9393
>>> msg = Message(role='user')
9494
>>> msg.get('nonexistent', 'default')
9595
'default'
96+
>>> msg = Message(role='user', tool_calls=[ Message.ToolCall(function=Message.ToolCall.Function(name='foo', arguments={}))])
97+
>>> msg.get('tool_calls')[0]['function']['name']
98+
'foo'
9699
"""
97100
return self[key] if key in self else default
98101

0 commit comments

Comments
 (0)