From 00c64332cc4ddd9e2ffd6f6839a1a55179e68f98 Mon Sep 17 00:00:00 2001 From: jmorganca Date: Sat, 23 Nov 2024 18:15:05 -0800 Subject: [PATCH] check defaults that aren't None too --- ollama/_types.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/ollama/_types.py b/ollama/_types.py index 28fb06c..89a74f0 100644 --- a/ollama/_types.py +++ b/ollama/_types.py @@ -45,8 +45,17 @@ class SubscriptableBaseModel(BaseModel): >>> msg['tool_calls'] = [Message.ToolCall(function=Message.ToolCall.Function(name='foo', arguments={}))] >>> 'tool_calls' in msg 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: return getattr(self, key, default)