From 7bbf5e17f1ad406254696fbb4ce0dea10cb59991 Mon Sep 17 00:00:00 2001 From: Mai Development Date: Tue, 27 Jan 2026 12:12:36 -0500 Subject: [PATCH] fix(01-02): correct ConversationMetadata import and initialization - Add ConversationMetadata to imports - Fix metadata initialization in create_conversation() - Resolve type error for conversation metadata File: src/models/context_manager.py --- src/models/context_manager.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/src/models/context_manager.py b/src/models/context_manager.py index f1b458b..49eacc9 100644 --- a/src/models/context_manager.py +++ b/src/models/context_manager.py @@ -18,6 +18,7 @@ from .conversation import ( MessageRole, MessageType, MessageMetadata, + ConversationMetadata, calculate_importance_score, estimate_token_count, ) @@ -157,9 +158,12 @@ class ContextManager: """ context_size = model_context_size or self.default_context_size - metadata = {"session_id": conversation_id, "context_window_size": context_size} - - conversation = Conversation(id=conversation_id, metadata=metadata) + conversation = Conversation( + id=conversation_id, + metadata=ConversationMetadata( + session_id=conversation_id, context_window_size=context_size + ), + ) self.conversations[conversation_id] = conversation self.context_windows[conversation_id] = ContextWindow(