File tree Expand file tree Collapse file tree 1 file changed +5
-7
lines changed Expand file tree Collapse file tree 1 file changed +5
-7
lines changed Original file line number Diff line number Diff line change @@ -83,13 +83,11 @@ async def stop_server():
83
83
84
84
async def yield_callback (event_type : str , event_obj : StreamEventData , ** kwargs ) -> Optional [str ]:
85
85
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 "
93
91
elif isinstance (event_obj , ThreadMessage ):
94
92
if (event_obj .status == "completed" ):
95
93
stream_data = json .dumps ({'content' : accumulated_text [0 ], 'type' : "completed_message" })
You can’t perform that action at this time.
0 commit comments