Refine Stage 2: Add MemoryService, JSON logging middleware, and fuzzed stress tests

middleware/logging_middleware.py

@@ -1,3 +1,4 @@
+import json
 import logging
 from models.event import Event, EventType
 
@@ -12,14 +13,20 @@ class LoggingMiddleware:
         if event.type in (EventType.LLM_TOKEN, EventType.TTS_CHUNK):
             return
 
-        payload_summary = ""
+        payload_summary = None
         if event.payload is not None:
             # Truncate strings to prevent log flooding
             payload_str = str(event.payload)
             if len(payload_str) > 80:
-                payload_str = payload_str[:80] + "..."
-            payload_summary = f" | Payload: {payload_str}"
+                payload_summary = payload_str[:80] + "..."
+            else:
+                payload_summary = payload_str
 
-        logger.info(
-            f"[{self.session_id}] [TraceEvent] {event.type.name} (EventId: {event.event_id[:8]}){payload_summary}"
-        )
+        trace_record = {
+            "session_id": self.session_id,
+            "event_type": event.type.name,
+            "event_id": event.event_id,
+            "timestamp": event.timestamp,
+            "payload": payload_summary,
+        }
+        logger.info(f"JSON_TRACE: {json.dumps(trace_record)}")

runtime/cleanup_manager.py

@@ -40,5 +40,9 @@ class CleanupManager:
         # 6. Reset TTS service state
         if self.context.tts_service:
             self.context.tts_service.reset()
+
+        # 7. Reset Memory service state
+        if self.context.memory_service:
+            self.context.memory_service.reset()
             
         self.context.logger.debug(f"[{self.context.session_id}] CleanupManager: Conversation reset completed.")

runtime/runtime_context.py

@@ -36,3 +36,4 @@ class RuntimeContext:
         self.tts_service: Any = None
         self.sentence_accumulator: Any = None
         self.error_handler: Any = None
+        self.memory_service: Any = None

runtime/session.py

@@ -19,6 +19,7 @@ from services.asr_service import ASRService
 from services.llm_service import LLMService
 from services.tts_service import TTSService
 from services.tool_service import ToolService
+from services.memory_service import MemoryService
 from middleware.logging_middleware import LoggingMiddleware
 from middleware.metrics_middleware import MetricsMiddleware
 
@@ -75,12 +76,14 @@ class ClientSession:
         self.llm = LLMService(self.context)
         self.tts = TTSService(self.context)
         self.tools = ToolService(self.context)
+        self.memory = MemoryService(self.context)
 
         # Link services to context
         self.context.vad_service = self.vad
         self.context.asr_service = self.asr
         self.context.llm_service = self.llm
         self.context.tts_service = self.tts
+        self.context.memory_service = self.memory
 
         # Turn Orchestrator
         self.turn_manager = TurnManager(self.context)
@@ -111,6 +114,7 @@ class ClientSession:
         await self.llm.start()
         await self.tts.start()
         await self.tools.start()
+        await self.memory.start()
         self.context.logger.info(f"[{self.session_id}] ClientSession: Services running.")
 
     async def stop(self):
@@ -120,6 +124,7 @@ class ClientSession:
         await self.llm.stop()
         await self.tts.stop()
         await self.tools.stop()
+        await self.memory.stop()
         self.task_supervisor.cancel_all()
 
     async def cleanup(self):

services/__init__.py

@@ -6,6 +6,7 @@ from services.asr_service import ASRService
 from services.llm_service import LLMService
 from services.tts_service import TTSService
 from services.tool_service import ToolService
+from services.memory_service import MemoryService
 
 __all__ = [
     "BaseVAD",
@@ -16,4 +17,5 @@ __all__ = [
     "LLMService",
     "TTSService",
     "ToolService",
+    "MemoryService",
 ]

services/memory_service.py

@@ -0,0 +1,21 @@
+from typing import Any, List
+
+class MemoryService:
+    def __init__(self, context: Any):
+        self.context = context
+
+    async def start(self):
+        # Stub service setup
+        pass
+
+    async def stop(self):
+        # Stub service teardown
+        pass
+
+    def reset(self):
+        # Stub state reset
+        pass
+
+    async def retrieve(self) -> List[Any]:
+        # Stub for context retrieval, long-term memory, etc.
+        return []