todo-agent 0.2.3__tar.gz → 0.2.4__tar.gz

{todo_agent-0.2.3 → todo_agent-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: todo-agent
-Version: 0.2.3
+Version: 0.2.4
 Summary: A natural language interface for todo.sh task management
 Author: codeprimate
 Maintainer: codeprimate

{todo_agent-0.2.3 → todo_agent-0.2.4}/tests/test_core/test_conversation_manager.py

@@ -14,7 +14,7 @@ class TestConversationManager:
         """Test ConversationManager initialization."""
         manager = ConversationManager()
         assert len(manager.history) == 0
-        assert manager.max_tokens == 4000
+        assert manager.max_tokens == 16000
         assert manager.max_messages == 50
         assert manager.system_prompt is None
 

{todo_agent-0.2.3 → todo_agent-0.2.4}/todo_agent/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.2.3'
-__version_tuple__ = version_tuple = (0, 2, 3)
+__version__ = version = '0.2.4'
+__version_tuple__ = version_tuple = (0, 2, 4)
 
-__commit_id__ = commit_id = 'ge1a08d166'
+__commit_id__ = commit_id = 'gc2b1a1313'

{todo_agent-0.2.3 → todo_agent-0.2.4}/todo_agent/core/conversation_manager.py

@@ -35,7 +35,7 @@ class ConversationManager:
     """Manages conversation state and memory for LLM interactions."""
 
     def __init__(
-        self, max_tokens: int = 4000, max_messages: int = 50, model: str = "gpt-4"
+        self, max_tokens: int = 16000, max_messages: int = 50, model: str = "gpt-4"
     ):
         self.history: List[ConversationMessage] = []
         self.max_tokens = max_tokens

{todo_agent-0.2.3 → todo_agent-0.2.4}/todo_agent/interface/cli.py

@@ -2,6 +2,8 @@
 Command-line interface for todo.sh LLM agent.
 """
 
+from typing import Optional
+
 try:
     import readline
 
@@ -130,6 +132,27 @@ class CLI:
         table = TableFormatter.create_stats_table(summary)
         self.console.print(table)
 
+    def _get_memory_usage(self) -> Optional[Text]:
+        """Get session memory usage as a progress bar."""
+        # Get conversation manager to access memory limits and current usage
+        conversation_manager = self.inference.get_conversation_manager()
+        
+        # Get current usage from conversation summary
+        summary = conversation_manager.get_conversation_summary()
+        current_tokens = summary.get("estimated_tokens", 0)
+        current_messages = summary.get("total_messages", 0)
+        
+        # Get limits from conversation manager
+        max_tokens = conversation_manager.max_tokens
+        max_messages = conversation_manager.max_messages
+        
+        # Create memory usage bar
+        memory_bar = PanelFormatter.create_memory_usage_bar(
+            current_tokens, max_tokens, current_messages, max_messages
+        )
+        
+        return memory_bar
+
     def run(self) -> None:
         """Main CLI interaction loop."""
         self.logger.info("Starting CLI interaction loop")
@@ -202,8 +225,13 @@ class CLI:
 
                 # Format the response and create a panel
                 formatted_response = ResponseFormatter.format_response(response)
+                
+                # Get memory usage
+                memory_usage = self._get_memory_usage()
+                
+                # Create response panel with memory usage
                 response_panel = PanelFormatter.create_response_panel(
-                    formatted_response
+                    formatted_response, memory_usage=memory_usage
                 )
                 self.console.print(response_panel)
 

{todo_agent-0.2.3 → todo_agent-0.2.4}/todo_agent/interface/formatters.py

@@ -2,11 +2,12 @@
 Formatters for CLI output with unicode characters and consistent styling.
 """
 
-from typing import Any, Dict
+from typing import Any, Dict, Optional
 
 from rich.align import Align
 from rich.box import ROUNDED
 from rich.panel import Panel
+from rich.progress import Progress, SpinnerColumn, TextColumn, BarColumn, TaskProgressColumn
 from rich.table import Table
 from rich.text import Text
 
@@ -346,11 +347,26 @@ class PanelFormatter:
         )
 
     @staticmethod
-    def create_response_panel(content: str, title: str = "🤖 Assistant") -> Panel:
+    def create_response_panel(content: str, title: str = "🤖 Assistant", memory_usage: Optional[Text] = None) -> Panel:
         """Create a panel for displaying LLM responses."""
-        return Panel(
-            content, title=title, border_style="dim", box=ROUNDED, width=PANEL_WIDTH
-        )
+        if memory_usage:
+            # Create the combined content with centered memory usage
+            return Panel(
+                Align.center(
+                    Text.assemble(
+                        content,
+                        "\n\n",
+                        "─" * (PANEL_WIDTH - 4),  # Separator line
+                        "\n",
+                        memory_usage
+                    )
+                ),
+                title=title, border_style="dim", box=ROUNDED, width=PANEL_WIDTH
+            )
+        else:
+            return Panel(
+                content, title=title, border_style="dim", box=ROUNDED, width=PANEL_WIDTH
+            )
 
     @staticmethod
     def create_error_panel(content: str, title: str = "❌ Error") -> Panel:
@@ -397,3 +413,45 @@ class PanelFormatter:
             box=ROUNDED,
             width=PANEL_WIDTH + 2,
         )
+
+    @staticmethod
+    def create_memory_usage_bar(current_tokens: int, max_tokens: int, current_messages: int, max_messages: int) -> Text:
+        """
+        Create a rich progress bar showing session memory usage.
+        
+        Args:
+            current_tokens: Current number of tokens in conversation
+            max_tokens: Maximum allowed tokens
+            current_messages: Current number of messages in conversation
+            max_messages: Maximum allowed messages
+            
+        Returns:
+            Rich Text object with memory usage progress bar
+        """
+        # Calculate percentage
+        token_percentage = min(100, (current_tokens / max_tokens) * 100)
+        
+        # Determine color based on usage
+        if token_percentage >= 90:
+            color = "red"
+        elif token_percentage >= 75:
+            color = "yellow"
+        else:
+            color = "green"
+        
+        # Create the progress bar text
+        memory_text = Text()
+        memory_text.append(f"{current_tokens:,}/{max_tokens:,} ", style="dim")
+        
+        # Create a simple text-based progress bar
+        bar_length = 25
+        token_filled = int((token_percentage / 100) * bar_length)
+        token_bar = "█" * token_filled + "░" * (bar_length - token_filled)
+        memory_text.append(f"[{token_bar}] ", style="dim")
+        memory_text.append(f"{token_percentage:.1f}%", style="dim")
+        
+        # Add message count without progress bar
+        memory_text.append(" | ", style="dim")
+        memory_text.append(f"{current_messages}/{max_messages}", style="dim")
+        
+        return memory_text

{todo_agent-0.2.3 → todo_agent-0.2.4}/todo_agent.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: todo-agent
-Version: 0.2.3
+Version: 0.2.4
 Summary: A natural language interface for todo.sh task management
 Author: codeprimate
 Maintainer: codeprimate