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

Update ollama/_types.py

Browse Source 
Co-authored-by: Parth Sareen <parth.sareen@ollama.com>
This commit is contained in:
Jeffrey Morgan 2024-11-26 10:41:53 -08:00 committed by GitHub
parent d8d98e17b2
commit d4c38978d1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
ollama/_types.py
View File

@ -93,6 +93,9 @@ class SubscriptableBaseModel(BaseModel):
>>> msg = Message(role='user')
>>> msg.get('nonexistent', 'default')
'default'
>>> msg = Message(role='user', tool_calls=[ Message.ToolCall(function=Message.ToolCall.Function(name='foo', arguments={}))])
>>> msg.get('tool_calls')[0]['function']['name']
'foo'
"""
return self[key] if key in self else default