mirror of
https://github.com/ollama/ollama-python.git
synced 2026-05-01 11:48:17 +08:00
Fix incorrect 'in' operator used for role assertion in test_client.py
This commit is contained in:
parent
dbccf192ac
commit
a5c3257081
@ -134,7 +134,7 @@ def test_client_chat_stream(httpserver: HTTPServer):
|
||||
|
||||
it = iter(['I ', "don't ", 'know.'])
|
||||
for part in response:
|
||||
assert part['message']['role'] in 'assistant'
|
||||
assert part['message']['role'] == 'assistant'
|
||||
assert part['message']['content'] == next(it)
|
||||
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user