code-puppy 0.0.193__py3-none-any.whl → 0.0.194__py3-none-any.whl

code_puppy/command_line/command_handler.py

@@ -76,10 +76,22 @@ def get_commands_help():
         Text("/load_context", style="cyan")
         + Text(" <name>  Load message history from file")
     )
+    help_lines.append(
+        Text("", style="cyan")
+        + Text("Session Management:", style="bold yellow")
+    )
+    help_lines.append(
+        Text("auto_save_session", style="cyan")
+        + Text("    Auto-save session after each response (true/false)")
+    )
+    help_lines.append(
+        Text("max_saved_sessions", style="cyan")
+        + Text("    Maximum number of sessions to keep (default: 20, 0 = unlimited)")
+    )
     help_lines.append(
         Text("/set", style="cyan")
         + Text(
-            "                  Set puppy config key-values (e.g., /set yolo_mode true, /set compaction_strategy truncation)"
+            "                  Set puppy config key-values (e.g., /set yolo_mode true, /set auto_save_session true, /set max_saved_sessions 20)"
         )
     )
     help_lines.append(

code_puppy/config.py

@@ -121,6 +121,8 @@ def get_config_keys():
         "message_limit",
         "allow_recursion",
         "openai_reasoning_effort",
+        "auto_save_session",
+        "max_saved_sessions",
     ]
     config = configparser.ConfigParser()
     config.read(CONFIG_FILE)
@@ -645,3 +647,166 @@ def clear_agent_pinned_model(agent_name: str):
     # We can't easily delete keys from configparser, so set to empty string
     # which will be treated as None by get_agent_pinned_model
     set_config_value(f"agent_model_{agent_name}", "")
+
+
+def get_auto_save_session() -> bool:
+    """
+    Checks puppy.cfg for 'auto_save_session' (case-insensitive in value only).
+    Defaults to True if not set.
+    Allowed values for ON: 1, '1', 'true', 'yes', 'on' (all case-insensitive for value).
+    """
+    true_vals = {"1", "true", "yes", "on"}
+    cfg_val = get_value("auto_save_session")
+    if cfg_val is not None:
+        if str(cfg_val).strip().lower() in true_vals:
+            return True
+        return False
+    return True
+
+
+def set_auto_save_session(enabled: bool):
+    """Sets the auto_save_session configuration value.
+    
+    Args:
+        enabled: Whether to enable auto-saving of sessions
+    """
+    set_config_value("auto_save_session", "true" if enabled else "false")
+
+
+def get_max_saved_sessions() -> int:
+    """
+    Gets the maximum number of sessions to keep.
+    Defaults to 20 if not set.
+    """
+    cfg_val = get_value("max_saved_sessions")
+    if cfg_val is not None:
+        try:
+            val = int(cfg_val)
+            return max(0, val)  # Ensure non-negative
+        except (ValueError, TypeError):
+            pass
+    return 20
+
+
+def set_max_saved_sessions(max_sessions: int):
+    """Sets the max_saved_sessions configuration value.
+    
+    Args:
+        max_sessions: Maximum number of sessions to keep (0 for unlimited)
+    """
+    set_config_value("max_saved_sessions", str(max_sessions))
+
+
+def _cleanup_old_sessions():
+    """Remove oldest sessions if we exceed the max_saved_sessions limit."""
+    max_sessions = get_max_saved_sessions()
+    if max_sessions <= 0:  # 0 means unlimited
+        return
+        
+    from pathlib import Path
+    
+    contexts_dir = Path(CONFIG_DIR) / "contexts"
+    if not contexts_dir.exists():
+        return
+        
+    # Get all .pkl files (session files) and sort by modification time
+    session_files = []
+    for pkl_file in contexts_dir.glob("*.pkl"):
+        try:
+            session_files.append((pkl_file.stat().st_mtime, pkl_file))
+        except OSError:
+            continue
+            
+    # Sort by modification time (oldest first)
+    session_files.sort(key=lambda x: x[0])
+    
+    # If we have more than max_sessions, remove the oldest ones
+    if len(session_files) > max_sessions:
+        files_to_remove = session_files[:-max_sessions]  # All except the last max_sessions
+        
+        from rich.console import Console
+        console = Console()
+        
+        for _, old_file in files_to_remove:
+            try:
+                # Remove the .pkl file
+                old_file.unlink()
+                
+                # Also remove the corresponding _meta.json file if it exists
+                meta_file = contexts_dir / f"{old_file.stem}_meta.json"
+                if meta_file.exists():
+                    meta_file.unlink()
+                    
+                console.print(f"[dim]🗑️  Removed old session: {old_file.name}[/dim]")
+                
+            except OSError as e:
+                console.print(f"[dim]❌ Failed to remove {old_file.name}: {e}[/dim]")
+
+
+def auto_save_session_if_enabled() -> bool:
+    """Automatically save the current session if auto_save_session is enabled.
+    
+    Returns:
+        True if session was saved, False otherwise
+    """
+    if not get_auto_save_session():
+        return False
+        
+    try:
+        import datetime
+        import json
+        import pickle
+        from pathlib import Path
+        from code_puppy.agents.agent_manager import get_current_agent
+        
+        # Get current agent and message history
+        current_agent = get_current_agent()
+        history = current_agent.get_message_history()
+        
+        if not history:
+            return False  # No history to save
+            
+        # Create timestamp-based session name
+        timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
+        session_name = f"auto_session_{timestamp}"
+        
+        # Create contexts directory if it doesn't exist
+        contexts_dir = Path(CONFIG_DIR) / "contexts"
+        contexts_dir.mkdir(parents=True, exist_ok=True)
+        
+        # Save as pickle for exact preservation
+        pickle_file = contexts_dir / f"{session_name}.pkl"
+        with open(pickle_file, "wb") as f:
+            pickle.dump(history, f)
+            
+        # Also save metadata as JSON for readability
+        meta_file = contexts_dir / f"{session_name}_meta.json"
+        metadata = {
+            "session_name": session_name,
+            "timestamp": datetime.datetime.now().isoformat(),
+            "message_count": len(history),
+            "total_tokens": sum(
+                current_agent.estimate_tokens_for_message(m) for m in history
+            ),
+            "file_path": str(pickle_file),
+            "auto_saved": True,
+        }
+        
+        with open(meta_file, "w") as f:
+            json.dump(metadata, f, indent=2)
+            
+        from rich.console import Console
+        console = Console()
+        console.print(
+            f"🐾 [dim]Auto-saved session: {len(history)} messages ({metadata['total_tokens']} tokens)[/dim]"
+        )
+        
+        # Cleanup old sessions if limit is set
+        _cleanup_old_sessions()
+        return True
+        
+    except Exception as e:
+        from rich.console import Console
+        console = Console()
+        console.print(f"[dim]❌ Failed to auto-save session: {str(e)}[/dim]")
+        return False

code_puppy/main.py

@@ -456,11 +456,15 @@ async def interactive_mode(message_renderer, initial_command: str = None) -> Non
                     f"\n[bold purple]AGENT RESPONSE: [/bold purple]\n{agent_response}"
                 )
 
+                # Auto-save session if enabled
+                from code_puppy.config import auto_save_session_if_enabled
+                auto_save_session_if_enabled()
+
                 # Ensure console output is flushed before next prompt
                 # This fixes the issue where prompt doesn't appear after agent response
                 display_console.file.flush() if hasattr(
                     display_console.file, "flush"
-                ) else None
+        ) else None
                 import time
 
                 time.sleep(0.1)  # Brief pause to ensure all messages are rendered
@@ -592,4 +596,4 @@ def main_entry():
 
 
 if __name__ == "__main__":
-    main_entry()
+    main_entry()

{code_puppy-0.0.193.dist-info → code_puppy-0.0.194.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-puppy
-Version: 0.0.193
+Version: 0.0.194
 Summary: Code generation agent
 Project-URL: repository, https://github.com/mpfaffenberger/code_puppy
 Project-URL: HomePage, https://github.com/mpfaffenberger/code_puppy

{code_puppy-0.0.193.dist-info → code_puppy-0.0.194.dist-info}/RECORD

@@ -1,9 +1,9 @@
 code_puppy/__init__.py,sha256=ehbM1-wMjNmOXk_DBhhJECFyBv2dRHwwo7ucjHeM68E,107
 code_puppy/__main__.py,sha256=pDVssJOWP8A83iFkxMLY9YteHYat0EyWDQqMkKHpWp4,203
 code_puppy/callbacks.py,sha256=ukSgVFaEO68o6J09qFwDrnmNanrVv3toTLQhS504Meo,6162
-code_puppy/config.py,sha256=xAFt2_8z_YrDrq8j-hATgj0ZCEZ9QPu206uJ5-F8FP0,21522
+code_puppy/config.py,sha256=5axUM2QHUF9kEBRP7yBzdLhaUNjKzSJhXYdRb5uyxiY,26991
 code_puppy/http_utils.py,sha256=YLd8Y16idbI32JGeBXG8n5rT4o4X_zxk9FgUvK9XFo8,8248
-code_puppy/main.py,sha256=PATGAZLxigQ1kp6UIsWgzPTmYHO6EbgaJa6f7OXgWrk,21667
+code_puppy/main.py,sha256=k91xsgVXOTUFR_njg5k9kbQ7HfprfGYV6u189WoZju0,21828
 code_puppy/model_factory.py,sha256=ZbIAJWMNKNdTCEMQK8Ig6TDDZlVNyGO9hOLHoLLPMYw,15397
 code_puppy/models.json,sha256=dClUciCo2RlVDs0ZAQCIur8MOavZUEAXHEecn0uPa-4,1629
 code_puppy/reopenable_async_client.py,sha256=4UJRaMp5np8cbef9F0zKQ7TPKOfyf5U-Kv-0zYUWDho,8274
@@ -29,7 +29,7 @@ code_puppy/agents/agent_typescript_reviewer.py,sha256=EDY1mFkVpuJ1BPXsJFu2wQ2pfA
 code_puppy/agents/base_agent.py,sha256=rJm0xA9kLT_NU9MSZIrN-Z_T5O4Q-QuUmQM8paGZKHQ,39066
 code_puppy/agents/json_agent.py,sha256=KPS1q-Rr3b5ekem4i3wtu8eLJRDd5nSPiZ8duJ_tn0U,4630
 code_puppy/command_line/__init__.py,sha256=y7WeRemfYppk8KVbCGeAIiTuiOszIURCDjOMZv_YRmU,45
-code_puppy/command_line/command_handler.py,sha256=i5VR_d8OErBLV8IFvkFkS0m34Xy2Pe0BH5G1wBrkOSQ,31016
+code_puppy/command_line/command_handler.py,sha256=kTZ6ES35b2xnYxrVR7HZCPKp8S9vomgUKR9QIjPbmgQ,31469
 code_puppy/command_line/file_path_completion.py,sha256=gw8NpIxa6GOpczUJRyh7VNZwoXKKn-yvCqit7h2y6Gg,2931
 code_puppy/command_line/load_context_completion.py,sha256=6eZxV6Bs-EFwZjN93V8ZDZUC-6RaWxvtZk-04Wtikyw,2240
 code_puppy/command_line/model_picker_completion.py,sha256=vYNCZS1QWu6fxF__hTwpc7jwH7h_48wUxrnITawc83E,4140
@@ -120,9 +120,9 @@ code_puppy/tui/screens/help.py,sha256=eJuPaOOCp7ZSUlecearqsuX6caxWv7NQszUh0tZJjB
 code_puppy/tui/screens/mcp_install_wizard.py,sha256=vObpQwLbXjQsxmSg-WCasoev1usEi0pollKnL0SHu9U,27693
 code_puppy/tui/screens/settings.py,sha256=-WLldnKyWVKUYVPJcfOn1UU6eP9t8lLPUAVI317SOOM,10685
 code_puppy/tui/screens/tools.py,sha256=3pr2Xkpa9Js6Yhf1A3_wQVRzFOui-KDB82LwrsdBtyk,1715
-code_puppy-0.0.193.data/data/code_puppy/models.json,sha256=dClUciCo2RlVDs0ZAQCIur8MOavZUEAXHEecn0uPa-4,1629
-code_puppy-0.0.193.dist-info/METADATA,sha256=uKQ_g5EjsbSu7vZ9B9haZPieh9t7IDN0-ESzvE_cNaM,19987
-code_puppy-0.0.193.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-code_puppy-0.0.193.dist-info/entry_points.txt,sha256=Tp4eQC99WY3HOKd3sdvb22vZODRq0XkZVNpXOag_KdI,91
-code_puppy-0.0.193.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
-code_puppy-0.0.193.dist-info/RECORD,,
+code_puppy-0.0.194.data/data/code_puppy/models.json,sha256=dClUciCo2RlVDs0ZAQCIur8MOavZUEAXHEecn0uPa-4,1629
+code_puppy-0.0.194.dist-info/METADATA,sha256=s2UdXeovRnMviRqdtBhkDuE1Eq_k4g6DClEgoXQ2etA,19987
+code_puppy-0.0.194.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+code_puppy-0.0.194.dist-info/entry_points.txt,sha256=Tp4eQC99WY3HOKd3sdvb22vZODRq0XkZVNpXOag_KdI,91
+code_puppy-0.0.194.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
+code_puppy-0.0.194.dist-info/RECORD,,