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

check defaults that aren't None too

Browse Source
This commit is contained in:
jmorganca 2024-11-23 18:15:05 -08:00
parent 986fb4c7b3
commit 00c64332cc
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
ollama/_types.py
View File

@ -45,8 +45,17 @@ class SubscriptableBaseModel(BaseModel):
>>> msg['tool_calls'] = [Message.ToolCall(function=Message.ToolCall.Function(name='foo', arguments={}))] >>> msg['tool_calls'] = [Message.ToolCall(function=Message.ToolCall.Function(name='foo', arguments={}))]
>>> 'tool_calls' in msg >>> 'tool_calls' in msg
True True
>>> tool = Tool()
>>> 'type' in tool
True
""" """
return key in self.model_fields_set if key in self.model_fields_set:
return True
if key in self.model_fields:
return self.model_fields[key].default is not None
return False
def get(self, key: str, default: Any = None) -> Any: def get(self, key: str, default: Any = None) -> Any:
return getattr(self, key, default) return getattr(self, key, default)