fix: update agent event logger

src/llama_stack_client/lib/agents/event_logger.py

@@ -4,7 +4,7 @@
 # This source code is licensed under the terms described in the LICENSE file in
 # the root directory of this source tree.
 
-from typing import Any, Iterator, Optional, Tuple
+from typing import Any, Iterator, Optional
 
 from termcolor import cprint
 
@@ -55,17 +55,10 @@ def print(self, flush: bool = True) -> None:
 
 
 class TurnStreamEventPrinter:
-    def __init__(self) -> None:
-        self.previous_event_type: Optional[str] = None
-        self.previous_step_type: Optional[str] = None
-
     def yield_printable_events(self, chunk: Any) -> Iterator[TurnStreamPrintableEvent]:
-        for printable_event in self._yield_printable_events(chunk, self.previous_event_type, self.previous_step_type):
+        for printable_event in self._yield_printable_events(chunk):
             yield printable_event
 
-        if not hasattr(chunk, "error"):
-            self.previous_event_type, self.previous_step_type = self._get_event_type_step_type(chunk)
-
     def _yield_printable_events(self, chunk: Any) -> Iterator[TurnStreamPrintableEvent]:
         if hasattr(chunk, "error"):
             yield TurnStreamPrintableEvent(role=None, content=chunk.error["message"], color="red")
@@ -144,17 +137,6 @@ def _yield_printable_events(self, chunk: Any) -> Iterator[TurnStreamPrintableEve
                         color="green",
                     )
 
-    def _get_event_type_step_type(self, chunk: Any) -> Tuple[Optional[str], Optional[str]]:
-        if hasattr(chunk, "event"):
-            previous_event_type = chunk.event.payload.event_type if hasattr(chunk, "event") else None
-            previous_step_type = (
-                chunk.event.payload.step_type
-                if previous_event_type not in {"turn_start", "turn_complete", "turn_awaiting_input"}
-                else None
-            )
-            return previous_event_type, previous_step_type
-        return None, None
-
 
 class EventLogger:
     def log(self, event_generator: Iterator[Any]) -> Iterator[TurnStreamPrintableEvent]: