From dc9543ba5f6120d8fa46fad30ba650414b7f4a35 Mon Sep 17 00:00:00 2001 From: DSLstandard Date: Mon, 15 Apr 2024 20:39:49 +0800 Subject: [PATCH] fix 'content' check on messages in chat() --- ollama/_client.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ollama/_client.py b/ollama/_client.py index 607f12c..b42d669 100644 --- a/ollama/_client.py +++ b/ollama/_client.py @@ -169,7 +169,7 @@ class Client(BaseClient): 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 @@ class AsyncClient(BaseClient): 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]