todo-agent 0.2.3__py3-none-any.whl → 0.2.4__py3-none-any.whl

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 = None

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/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/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.dist-info → todo_agent-0.2.4.dist-info}/METADATA

@@ -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.dist-info → todo_agent-0.2.4.dist-info}/RECORD

@@ -1,8 +1,8 @@
 todo_agent/__init__.py,sha256=RUowhd14r3tqB_7rl83unGV8oBjra3UOIl7jix-33fk,254
-todo_agent/_version.py,sha256=kBRz0P2plw1eVdIpt70W6m1LMbEIhLY3RyOfVGdubaI,704
+todo_agent/_version.py,sha256=NRw4Jle4n9v_DD2wtplRqflGCvX8OU5eAjycYY0vY3Y,704
 todo_agent/main.py,sha256=-ryhMm4c4sz4e4anXI8B-CYnpEh5HIkmnYcnGxcWHDk,1628
 todo_agent/core/__init__.py,sha256=QAZ4it63pXv5-DxtNcuSAmg7ZnCY5ackI5yycvKHr9I,365
-todo_agent/core/conversation_manager.py,sha256=Mxjxn3BTjSnju6xryQcpOWdE8wWWJqTdDFTC86D9xc8,11639
+todo_agent/core/conversation_manager.py,sha256=gSCcX356UJ0T3FCTS1q0fOud0ytFKXptf9RyKjzpTYI,11640
 todo_agent/core/exceptions.py,sha256=cPvvkIbKdI7l51wC7cE-ZxUi54P3nf2m7x2lMNMRFYM,399
 todo_agent/core/todo_manager.py,sha256=wV-J_E_aK7yRNT-WbILKJgqBFhbcXDZa6Pb4IjbgGkU,7974
 todo_agent/infrastructure/__init__.py,sha256=SGbHXgzq6U1DMgOfWPMsWEK99zjPSF-6gzy7xqc5fsI,284
@@ -17,12 +17,12 @@ todo_agent/infrastructure/todo_shell.py,sha256=z6kqUKDX-i4DfYJKoOLiPLCp8y6m1HdTD
 todo_agent/infrastructure/token_counter.py,sha256=PCKheOVJbp1s89yhh_i6iKgURMt9mVoYkwjQJCc2xCE,4958
 todo_agent/infrastructure/prompts/system_prompt.txt,sha256=uCb6yz3uDQdwcB8HJcF0y1_1b75oRtRnCMMHQLHI3NI,2415
 todo_agent/interface/__init__.py,sha256=vDD3rQu4qDkpvVwGVtkDzE1M4IiSHYzTif4GbYSFWaI,457
-todo_agent/interface/cli.py,sha256=audxpDafLw_o-pitqntUN0PpKUyWZap2wpHw4QUak_8,9926
-todo_agent/interface/formatters.py,sha256=aJc-a2aUzi3tId46jFK3uwZnsOJiXBTTMT8I_46Jjlg,13181
+todo_agent/interface/cli.py,sha256=FacwTDjLP60qAQ_BsM_7_jM9liYJpZpRdXpBNwNuO2Y,11060
+todo_agent/interface/formatters.py,sha256=QT6IgBVX6ghL8UWH54KFtJY9n9GbrPvOizjCN8emMGg,15503
 todo_agent/interface/tools.py,sha256=mlPPLVwECYrTtOX8ysIORRfErIOJl43qlTvfdpy2Vbs,28559
-todo_agent-0.2.3.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-todo_agent-0.2.3.dist-info/METADATA,sha256=VvqQLIc4ExA92Oa-nhukv96wszxQnYSf7nlG09hiZt0,10047
-todo_agent-0.2.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-todo_agent-0.2.3.dist-info/entry_points.txt,sha256=4W7LrCib6AXP5IZDwWRht8S5gutLu5oNfTJHGbt4oHs,52
-todo_agent-0.2.3.dist-info/top_level.txt,sha256=a65mlPIhPZHuq2bRIi_sCMAIJsUddvXt171OBF6r6co,11
-todo_agent-0.2.3.dist-info/RECORD,,
+todo_agent-0.2.4.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+todo_agent-0.2.4.dist-info/METADATA,sha256=fYnY58TYzNuQtiu2WBr3fRJwJEIh6bbt4gA2lrNC_9A,10047
+todo_agent-0.2.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+todo_agent-0.2.4.dist-info/entry_points.txt,sha256=4W7LrCib6AXP5IZDwWRht8S5gutLu5oNfTJHGbt4oHs,52
+todo_agent-0.2.4.dist-info/top_level.txt,sha256=a65mlPIhPZHuq2bRIi_sCMAIJsUddvXt171OBF6r6co,11
+todo_agent-0.2.4.dist-info/RECORD,,