remove DONE when getting error

Author: JC-386

sdk/agentserver/azure-ai-agentserver-agentframework/azure/ai/agentserver/agentframework/agent_framework.py

@@ -225,32 +225,6 @@ async def stream_updates():
                             update_count += 1
                             yield event
 
-                        # timeout_s = self._resolve_stream_timeout(context.request)
-                        # logger.info("Starting streaming with idle-timeout=%.2fs", timeout_s)
-                        # for ev in streaming_converter.initial_events():
-                        #     yield ev
-                        #
-                        # # Iterate with per-update timeout; terminate if idle too long
-                        # aiter = agent.run_stream(message).__aiter__()
-                        # while True:
-                        #     try:
-                        #         update = await asyncio.wait_for(aiter.__anext__(), timeout=timeout_s)
-                        #     except StopAsyncIteration:
-                        #         logger.debug("Agent streaming iterator finished (StopAsyncIteration)")
-                        #         break
-                        #     except asyncio.TimeoutError:
-                        #         logger.warning(
-                        #             "Streaming idle timeout reached (%.1fs); terminating stream.", timeout_s
-                        #         )
-                        #         for ev in streaming_converter.completion_events():
-                        #             yield ev
-                        #         return
-                        #     update_count += 1
-                        #     transformed = streaming_converter.transform_output_for_streaming(update)
-                        #     for event in transformed:
-                        #         yield event
-                        # for ev in streaming_converter.completion_events():
-                        #     yield ev
                         logger.info("Streaming completed with %d updates", update_count)
                     finally:
                         # Close tool_client if it was created for this request

sdk/agentserver/azure-ai-agentserver-agentframework/azure/ai/agentserver/agentframework/models/agent_framework_output_streaming_converter.py

@@ -299,13 +299,16 @@ async def convert(self, updates: AsyncIterable[AgentRunResponseUpdate]) -> Async
             elif isinstance(first, FunctionResultContent):
                 state = _FunctionCallOutputStreamingState(self)
             elif isinstance(first, ErrorContent):
-                yield ResponseErrorEvent(
-                    sequence_number=self.next_sequence(),
-                    code=getattr(first, "error_code", None) or "server_error",
-                    message=getattr(first, "message", None) or "An error occurred",
-                    param="",
-                )
-                continue
+                code=getattr(first, "error_code", None) or "server_error"
+                message=getattr(first, "message", None) or "An error occurred"
+                raise ValueError(f"ErrorContent received: code={code}, message={message}")
+                # yield ResponseErrorEvent(
+                #     sequence_number=self.next_sequence(),
+                #     code=getattr(first, "error_code", None) or "server_error",
+                #     message=getattr(first, "message", None) or "An error occurred",
+                #     param="",
+                # )
+                # continue
             if not state:
                 continue
 

sdk/agentserver/azure-ai-agentserver-core/azure/ai/agentserver/core/server/base.py

@@ -158,7 +158,6 @@ def gen():
                                 logger.error("Error in non-async generator: %s\n%s", e, traceback.format_exc())
                                 payload = {"error": err_msg}
                                 yield f"event: error\ndata: {json.dumps(payload)}\n\n"
-                                yield "data: [DONE]\n\n"
                                 error_sent = True
                             finally:
                                 logger.info("End of processing CreateResponse request.")