Updated streaming tests

This commit is contained in:
Cyber MacGeddon 2026-01-06 19:47:06 +00:00
parent 10f9489f97
commit b34c29ad15

View file

@ -51,7 +51,8 @@ class PromptClient(RequestResponse):
end_stream = getattr(resp, 'end_of_stream', False)
if resp.text:
# Always call callback if there's text OR if it's the final message
if resp.text is not None:
last_text = resp.text
# Call chunk callback if provided with both chunk and end_of_stream flag
if chunk_callback: