as_chat_history adapted

Author: HendrikStrobelt

mellea/stdlib/chat.py

@@ -7,8 +7,8 @@
 from mellea.stdlib.base import (
     CBlock,
     Component,
+    Context,
     ImageBlock,
-    LegacyContext,
     ModelOutputThunk,
     ModelToolCall,
     TemplateRepresentation,
@@ -113,7 +113,7 @@ def __str__(self):
         return f'mellea.Message(role="{self.role}", content="{self.content}", name="{self.name}")'
 
 
-def as_chat_history(ctx: LegacyContext) -> list[Message]:
+def as_chat_history(ctx: Context) -> list[Message]:
     """Returns a list of Messages corresponding to a Context."""
 
     def _to_msg(c: CBlock | Component | ModelOutputThunk) -> Message | None:
@@ -129,7 +129,7 @@ def _to_msg(c: CBlock | Component | ModelOutputThunk) -> Message | None:
             case _:
                 return None
 
-    all_ctx_events = ctx.full_event_log()
+    all_ctx_events = ctx.full_data_as_list()
     if all_ctx_events is None:
         raise Exception("Trying to cast a non-linear history into a chat history.")
     else: