fix 'content' check on messages in chat()

ollama · Apr 15, 2024 · dc9543b · dc9543b
1 parent 2fde317
commit dc9543b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ollama/_client.py b/ollama/_client.py
@@ -169,7 +169,7 @@ def chat(
         raise TypeError('messages must be a list of Message or dict-like objects')
       if not (role := message.get('role')) or role not in ['system', 'user', 'assistant']:
         raise RequestError('messages must contain a role and it must be one of "system", "user", or "assistant"')
-      if not message.get('content'):
+      if 'content' not in message:
         raise RequestError('messages must contain content')
       if images := message.get('images'):
         message['images'] = [_encode_image(image) for image in images]
@@ -449,7 +449,7 @@ async def chat(
         raise TypeError('messages must be a list of strings')
       if not (role := message.get('role')) or role not in ['system', 'user', 'assistant']:
         raise RequestError('messages must contain a role and it must be one of "system", "user", or "assistant"')
-      if not message.get('content'):
+      if 'content' not in message:
         raise RequestError('messages must contain content')
       if images := message.get('images'):
         message['images'] = [_encode_image(image) for image in images]