Skip to content

Commit e8aab58

Browse files
committed
I made MessageDeltaTextContent available in StreamEventData, now use it
1 parent 23df393 commit e8aab58

File tree

1 file changed

+5
-7
lines changed

1 file changed

+5
-7
lines changed

src/quartapp/chat.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -83,13 +83,11 @@ async def stop_server():
8383

8484
async def yield_callback(event_type: str, event_obj: StreamEventData, **kwargs) -> Optional[str]:
8585
accumulated_text = kwargs['accumulated_text']
86-
if (isinstance(event_obj, MessageDeltaChunk)):
87-
for content_part in event_obj.delta.content:
88-
if isinstance(content_part, MessageDeltaTextContent):
89-
text_value = content_part.text.value if content_part.text else "No text"
90-
stream_data = json.dumps({'content': text_value, 'type': "message"})
91-
accumulated_text[0] += text_value
92-
return f"data: {stream_data}\n\n"
86+
if (isinstance(event_obj, MessageDeltaTextContent)):
87+
text_value = event_obj.text.value if event_obj.text else "No text"
88+
stream_data = json.dumps({'content': text_value, 'type': "message"})
89+
accumulated_text[0] += text_value
90+
return f"data: {stream_data}\n\n"
9391
elif isinstance(event_obj, ThreadMessage):
9492
if (event_obj.status == "completed"):
9593
stream_data = json.dumps({'content': accumulated_text[0], 'type': "completed_message"})

0 commit comments

Comments
 (0)