Revert "Merge pull request #11 from madebygps/fix/agent-summarization-source-id"

This reverts commit dc92a9e, reversing
changes made to 51bde7b.

Author: pamelafox

examples/agent_summarization.py

@@ -183,7 +183,7 @@ async def process(
 
         # Before the agent runs: check if we should compact the history
         if session and self.context_tokens > self.token_threshold:
-            history = session.state.get("memory", {}).get("messages", [])
+            history = session.state.get(InMemoryHistoryProvider.DEFAULT_SOURCE_ID, {}).get("messages", [])
             if len(history) > 2:
                 logger.info(
                     "[📝 Summarization] Token usage (%d) exceeds threshold (%d). "
@@ -201,7 +201,7 @@ async def process(
                 )
 
                 # Replace session history with a single summary message
-                session.state["memory"]["messages"] = [
+                session.state[InMemoryHistoryProvider.DEFAULT_SOURCE_ID]["messages"] = [
                     Message(role="assistant", text=f"[Summary of earlier conversation]\n{summary_text}"),
                 ]
 

examples/spanish/agent_summarization.py

@@ -182,7 +182,7 @@ async def process(
 
         # Antes de ejecutar el agente: revisar si hay que compactar el historial
         if session and self.context_tokens > self.token_threshold:
-            history = session.state.get("memory", {}).get("messages", [])
+            history = session.state.get(InMemoryHistoryProvider.DEFAULT_SOURCE_ID, {}).get("messages", [])
             if len(history) > 2:
                 logger.info(
                     "[📝 Resumen] Uso de tokens (%d) excede el umbral (%d). "
@@ -200,7 +200,7 @@ async def process(
                 )
 
                 # Reemplazar el historial de la sesión con un único mensaje de resumen
-                session.state["memory"]["messages"] = [
+                session.state[InMemoryHistoryProvider.DEFAULT_SOURCE_ID]["messages"] = [
                     Message(role="assistant", text=f"[Resumen de la conversación anterior]\n{summary_text}"),
                 ]