code-lm 0.3.0__tar.gz → 0.3.2__tar.gz

{code_lm-0.3.0/src/code_lm.egg-info → code_lm-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-lm
-Version: 0.3.0
+Version: 0.3.2
 Summary: An AI coding assistant using various LLM models.
 Home-page: https://github.com/Panagiotis897/lm-code
 Author: Panagiotis897
@@ -39,6 +39,10 @@ LM Code is a powerful AI coding assistant for your terminal supporting 17 free m
   - Markdown rendering for improved readability.
 - **17 Free Models via OpenRouter**:
   - NVIDIA Nemotron, Qwen, OpenAI, Meta Llama, Mistral, and more.
+- **Session Persistence**:
+  - Conversations are saved per project directory.
+  - Resume previous sessions automatically or manually with `/load`.
+  - Auto-saves after each exchange.
 - **Automated Tool Usage**:
   - File operations: `view`, `edit`, `grep`, `glob`.
   - Directory operations: `ls`, `tree`, `create_directory`.
@@ -132,8 +136,12 @@ lmcode list-models
 
 During an interactive session:
 
-- **`/exit`**: Exit the session.
+- **`/exit`**: Save session and exit.
 - **`/help`**: Display help information.
+- **`/sessions`**: List all sessions for the current project.
+- **`/load <id>`**: Load a session by ID.
+- **`/new`**: Start a new session (saves current first).
+- **`/save`**: Manually save the current session.
 
 ---
 
@@ -153,10 +161,16 @@ LM Code is under active development. Contributions, feature requests, and feedba
 
 ### Changelog
 
+#### v0.3.2
+- Added session persistence: conversations are saved per project directory.
+- Sessions auto-save after each exchange and on exit.
+- New commands: `/sessions`, `/load <id>`, `/new`, `/save`.
+- On startup, offers to resume previous sessions for the current project.
+
 #### v0.3.0
 - Updated default model to NVIDIA Nemotron 3 Super 120B.
 - Added 17 free models from OpenRouter (previously 6).
-- Fixed `ModuleNotFoundError: No module named 'gemini_cli'` from stale entry point.
+- Fixed `ModuleNotFoundError: No module named 'gemini_cli'` from stale entry point after package rename.
 - Fixed `UnicodeDecodeError` on Windows (cp1253) for all subprocess commands.
 - Fixed API URL (`/chat/completions` was missing) causing HTML response errors.
 - Improved API error handling for empty/invalid responses.
@@ -168,20 +182,18 @@ LM Code is under active development. Contributions, feature requests, and feedba
 #### v0.2.5
 - Added more models to the model list.
 - Fixed crucial bugs from previous versions.
-- Removed Gemini models.
+- Removed legacy Gemini module.
 - Updated models to latest versions.
 
 #### v0.1.0
-- Rebranded from Gemini to LM Code.
+- Rebranded to LM Code.
 - Integrated OpenRouter as the default provider.
 - Added multi-model support.
-- Overhauled CLI commands (`gemini` -> `lmcode`).
+- Overhauled CLI commands.
 
 ---
 
 ## Future Plans
-
-- Pricing with appropriate rate limits.
 - Non-free model support.
 - MCP Server integration.
 - Additional providers.

{code_lm-0.3.0 → code_lm-0.3.2}/README.md

@@ -11,6 +11,10 @@ LM Code is a powerful AI coding assistant for your terminal supporting 17 free m
   - Markdown rendering for improved readability.
 - **17 Free Models via OpenRouter**:
   - NVIDIA Nemotron, Qwen, OpenAI, Meta Llama, Mistral, and more.
+- **Session Persistence**:
+  - Conversations are saved per project directory.
+  - Resume previous sessions automatically or manually with `/load`.
+  - Auto-saves after each exchange.
 - **Automated Tool Usage**:
   - File operations: `view`, `edit`, `grep`, `glob`.
   - Directory operations: `ls`, `tree`, `create_directory`.
@@ -104,8 +108,12 @@ lmcode list-models
 
 During an interactive session:
 
-- **`/exit`**: Exit the session.
+- **`/exit`**: Save session and exit.
 - **`/help`**: Display help information.
+- **`/sessions`**: List all sessions for the current project.
+- **`/load <id>`**: Load a session by ID.
+- **`/new`**: Start a new session (saves current first).
+- **`/save`**: Manually save the current session.
 
 ---
 
@@ -125,10 +133,16 @@ LM Code is under active development. Contributions, feature requests, and feedba
 
 ### Changelog
 
+#### v0.3.2
+- Added session persistence: conversations are saved per project directory.
+- Sessions auto-save after each exchange and on exit.
+- New commands: `/sessions`, `/load <id>`, `/new`, `/save`.
+- On startup, offers to resume previous sessions for the current project.
+
 #### v0.3.0
 - Updated default model to NVIDIA Nemotron 3 Super 120B.
 - Added 17 free models from OpenRouter (previously 6).
-- Fixed `ModuleNotFoundError: No module named 'gemini_cli'` from stale entry point.
+- Fixed `ModuleNotFoundError: No module named 'gemini_cli'` from stale entry point after package rename.
 - Fixed `UnicodeDecodeError` on Windows (cp1253) for all subprocess commands.
 - Fixed API URL (`/chat/completions` was missing) causing HTML response errors.
 - Improved API error handling for empty/invalid responses.
@@ -140,20 +154,18 @@ LM Code is under active development. Contributions, feature requests, and feedba
 #### v0.2.5
 - Added more models to the model list.
 - Fixed crucial bugs from previous versions.
-- Removed Gemini models.
+- Removed legacy Gemini module.
 - Updated models to latest versions.
 
 #### v0.1.0
-- Rebranded from Gemini to LM Code.
+- Rebranded to LM Code.
 - Integrated OpenRouter as the default provider.
 - Added multi-model support.
-- Overhauled CLI commands (`gemini` -> `lmcode`).
+- Overhauled CLI commands.
 
 ---
 
 ## Future Plans
-
-- Pricing with appropriate rate limits.
 - Non-free model support.
 - MCP Server integration.
 - Additional providers.

{code_lm-0.3.0 → code_lm-0.3.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "code-lm"
-version = "0.3.0"
+version = "0.3.2"
 authors = [
   { name="Panagiotis897", email="orion256business@gmail.com" }
 ]

{code_lm-0.3.0 → code_lm-0.3.2/src/code_lm.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-lm
-Version: 0.3.0
+Version: 0.3.2
 Summary: An AI coding assistant using various LLM models.
 Home-page: https://github.com/Panagiotis897/lm-code
 Author: Panagiotis897
@@ -39,6 +39,10 @@ LM Code is a powerful AI coding assistant for your terminal supporting 17 free m
   - Markdown rendering for improved readability.
 - **17 Free Models via OpenRouter**:
   - NVIDIA Nemotron, Qwen, OpenAI, Meta Llama, Mistral, and more.
+- **Session Persistence**:
+  - Conversations are saved per project directory.
+  - Resume previous sessions automatically or manually with `/load`.
+  - Auto-saves after each exchange.
 - **Automated Tool Usage**:
   - File operations: `view`, `edit`, `grep`, `glob`.
   - Directory operations: `ls`, `tree`, `create_directory`.
@@ -132,8 +136,12 @@ lmcode list-models
 
 During an interactive session:
 
-- **`/exit`**: Exit the session.
+- **`/exit`**: Save session and exit.
 - **`/help`**: Display help information.
+- **`/sessions`**: List all sessions for the current project.
+- **`/load <id>`**: Load a session by ID.
+- **`/new`**: Start a new session (saves current first).
+- **`/save`**: Manually save the current session.
 
 ---
 
@@ -153,10 +161,16 @@ LM Code is under active development. Contributions, feature requests, and feedba
 
 ### Changelog
 
+#### v0.3.2
+- Added session persistence: conversations are saved per project directory.
+- Sessions auto-save after each exchange and on exit.
+- New commands: `/sessions`, `/load <id>`, `/new`, `/save`.
+- On startup, offers to resume previous sessions for the current project.
+
 #### v0.3.0
 - Updated default model to NVIDIA Nemotron 3 Super 120B.
 - Added 17 free models from OpenRouter (previously 6).
-- Fixed `ModuleNotFoundError: No module named 'gemini_cli'` from stale entry point.
+- Fixed `ModuleNotFoundError: No module named 'gemini_cli'` from stale entry point after package rename.
 - Fixed `UnicodeDecodeError` on Windows (cp1253) for all subprocess commands.
 - Fixed API URL (`/chat/completions` was missing) causing HTML response errors.
 - Improved API error handling for empty/invalid responses.
@@ -168,20 +182,18 @@ LM Code is under active development. Contributions, feature requests, and feedba
 #### v0.2.5
 - Added more models to the model list.
 - Fixed crucial bugs from previous versions.
-- Removed Gemini models.
+- Removed legacy Gemini module.
 - Updated models to latest versions.
 
 #### v0.1.0
-- Rebranded from Gemini to LM Code.
+- Rebranded to LM Code.
 - Integrated OpenRouter as the default provider.
 - Added multi-model support.
-- Overhauled CLI commands (`gemini` -> `lmcode`).
+- Overhauled CLI commands.
 
 ---
 
 ## Future Plans
-
-- Pricing with appropriate rate limits.
 - Non-free model support.
 - MCP Server integration.
 - Additional providers.

{code_lm-0.3.0 → code_lm-0.3.2}/src/code_lm.egg-info/SOURCES.txt

@@ -11,9 +11,9 @@ src/code_lm.egg-info/top_level.txt
 src/lm_code/__init__.py
 src/lm_code/config.py
 src/lm_code/main.py
+src/lm_code/session.py
 src/lm_code/utils.py
 src/lm_code/models/__init__.py
-src/lm_code/models/gemini.py
 src/lm_code/models/openrouter.py
 src/lm_code/tools/__init__.py
 src/lm_code/tools/base.py

{code_lm-0.3.0 → code_lm-0.3.2}/src/lm_code/main.py

@@ -10,14 +10,17 @@ from rich.console import Console
 from rich.markdown import Markdown
 from rich.panel import Panel
 from pathlib import Path
+from typing import Optional, Dict, Any
 import yaml
 import logging
 import time
+import questionary
 
 from .models.openrouter import OpenRouterModel, list_available_models
 from .config import Config
 from .utils import count_tokens
 from .tools import AVAILABLE_TOOLS
+from .session import SessionManager
 
 # Setup console and config
 console = Console(
@@ -259,6 +262,13 @@ def start_interactive_session(model_name: str, console: Console):
         )
         return
 
+    # --- Session Persistence Setup ---
+    session_manager = SessionManager()
+    project_dir = os.getcwd()
+    current_session_id = None
+    current_model_name = model_name
+    # ---
+
     try:
         console.print(f"\nInitializing model [bold]{model_name}[/bold]...")
         # Pass the console object to OpenRouterModel constructor
@@ -275,6 +285,63 @@ def start_interactive_session(model_name: str, console: Console):
         )
         return
 
+    # --- Check for existing sessions and offer to resume ---
+    existing_sessions = session_manager.list_sessions(project_dir)
+    if existing_sessions:
+        console.print(
+            f"\n[yellow]Found {len(existing_sessions)} previous session(s) for this project:[/yellow]"
+        )
+        for i, session in enumerate(existing_sessions[:5], 1):
+            updated = session.get("updated_at", "")[:19].replace("T", " ")
+            messages = session.get("message_count", 0)
+            model = session.get("model_name", "unknown")
+            console.print(f"  {i}. [{updated}] {messages} messages (model: {model})")
+
+        resume_choice = questionary.select(
+            "Resume a previous session?",
+            choices=["Start new session", "Resume latest", "Choose session"],
+            default="Start new session",
+        ).ask()
+
+        if resume_choice == "Resume latest":
+            latest_session = session_manager.get_latest_session(project_dir)
+            if latest_session:
+                model.chat_history = latest_session["chat_history"]
+                current_session_id = latest_session["id"]
+                current_model_name = latest_session.get("model_name", model_name)
+                console.print(
+                    f"[green]Resumed session {current_session_id} ({len(model.chat_history)} messages)[/green]\n"
+                )
+            else:
+                console.print("[yellow]No session to resume. Starting new.[/yellow]\n")
+        elif resume_choice == "Choose session":
+            session_choices = [
+                f"{s['id']} ({s['message_count']} msgs, {s['updated_at'][:16]})"
+                for s in existing_sessions[:10]
+            ]
+            session_choices.append("Cancel - start new session")
+            chosen = questionary.select(
+                "Select a session to resume:",
+                choices=session_choices,
+            ).ask()
+            if chosen and chosen != "Cancel - start new session":
+                chosen_id = chosen.split(" (")[0]
+                loaded_session = session_manager.load_session(project_dir, chosen_id)
+                if loaded_session:
+                    model.chat_history = loaded_session["chat_history"]
+                    current_session_id = loaded_session["id"]
+                    current_model_name = loaded_session.get("model_name", model_name)
+                    console.print(
+                        f"[green]Resumed session {current_session_id} ({len(model.chat_history)} messages)[/green]\n"
+                    )
+                else:
+                    console.print("[red]Failed to load session. Starting new.[/red]\n")
+            else:
+                console.print("[yellow]Starting new session.[/yellow]\n")
+        else:
+            console.print("[yellow]Starting new session.[/yellow]\n")
+    # --- End Session Resume Check ---
+
     # --- Session Start Message ---
     console.print("Type '/help' for commands, '/exit' or Ctrl+C to quit.")
 
@@ -283,10 +350,64 @@ def start_interactive_session(model_name: str, console: Console):
             user_input = console.input("[bold green]You:[/bold green] ")
 
             if user_input.lower() == "/exit":
+                # Save session before exiting
+                _save_current_session(
+                    session_manager,
+                    project_dir,
+                    model,
+                    current_model_name,
+                    current_session_id,
+                    console,
+                )
                 break
             elif user_input.lower() == "/help":
                 show_help()
                 continue
+            elif user_input.lower() == "/sessions":
+                _show_sessions(session_manager, project_dir, console)
+                continue
+            elif user_input.lower().startswith("/load "):
+                session_id = user_input[6:].strip()
+                loaded = _load_session_by_id(
+                    session_manager, project_dir, session_id, model, console
+                )
+                if loaded:
+                    current_session_id = loaded["id"]
+                    current_model_name = loaded.get("model_name", model_name)
+                continue
+            elif user_input.lower() == "/new":
+                # Save current session first
+                _save_current_session(
+                    session_manager,
+                    project_dir,
+                    model,
+                    current_model_name,
+                    current_session_id,
+                    console,
+                )
+                # Reset chat history
+                model.chat_history = [
+                    {"role": "system", "content": model.system_instruction},
+                    {
+                        "role": "assistant",
+                        "content": "Okay, I'm ready. Provide the directory context and your request.",
+                    },
+                ]
+                current_session_id = None
+                console.print("[green]Started new session.[/green]\n")
+                continue
+            elif user_input.lower() == "/save":
+                session_id = _save_current_session(
+                    session_manager,
+                    project_dir,
+                    model,
+                    current_model_name,
+                    current_session_id,
+                    console,
+                )
+                if session_id:
+                    current_session_id = session_id
+                continue
 
             # Display initial "thinking" status - generate handles intermediate ones
             response_text = model.generate(user_input)
@@ -302,7 +423,26 @@ def start_interactive_session(model_name: str, console: Console):
             console.print("[bold green]Assistant:[/bold green]")
             console.print(Markdown(response_text), highlight=True)
 
+            # Auto-save after each exchange
+            current_session_id = _auto_save_session(
+                session_manager,
+                project_dir,
+                model,
+                current_model_name,
+                current_session_id,
+                console,
+            )
+
         except KeyboardInterrupt:
+            # Save session before exiting
+            _save_current_session(
+                session_manager,
+                project_dir,
+                model,
+                current_model_name,
+                current_session_id,
+                console,
+            )
             console.print("\n[yellow]Session interrupted. Exiting.[/yellow]")
             break
         except Exception as e:
@@ -312,6 +452,96 @@ def start_interactive_session(model_name: str, console: Console):
             log.error("Error during interactive loop", exc_info=True)
 
 
+def _save_current_session(
+    session_manager: SessionManager,
+    project_dir: str,
+    model: OpenRouterModel,
+    model_name: str,
+    session_id: Optional[str],
+    console: Console,
+) -> Optional[str]:
+    """Save the current session. Returns the session ID if saved."""
+    # Don't save if there are no user messages
+    user_messages = [msg for msg in model.chat_history if msg.get("role") == "user"]
+    if not user_messages:
+        return None
+
+    try:
+        saved_id = session_manager.save_session(
+            project_dir=project_dir,
+            chat_history=model.chat_history,
+            model_name=model_name,
+            session_id=session_id,
+        )
+        console.print(f"[dim]Session saved: {saved_id}[/dim]")
+        return saved_id
+    except Exception as e:
+        log.error(f"Failed to save session: {e}")
+        console.print(f"[red]Warning: Failed to save session: {e}[/red]")
+        return None
+
+
+def _auto_save_session(
+    session_manager: SessionManager,
+    project_dir: str,
+    model: OpenRouterModel,
+    model_name: str,
+    session_id: Optional[str],
+    console: Console,
+) -> Optional[str]:
+    """Auto-save session silently. Returns the session ID."""
+    try:
+        saved_id = session_manager.save_session(
+            project_dir=project_dir,
+            chat_history=model.chat_history,
+            model_name=model_name,
+            session_id=session_id,
+        )
+        return saved_id
+    except Exception as e:
+        log.error(f"Auto-save failed: {e}")
+        return session_id
+
+
+def _show_sessions(session_manager: SessionManager, project_dir: str, console: Console):
+    """Display sessions for the current project."""
+    sessions = session_manager.list_sessions(project_dir)
+    if not sessions:
+        console.print("[yellow]No sessions found for this project.[/yellow]")
+        return
+
+    console.print(f"\n[bold cyan]Sessions for current project:[/bold cyan]")
+    for i, session in enumerate(sessions, 1):
+        updated = session.get("updated_at", "")[:19].replace("T", " ")
+        messages = session.get("message_count", 0)
+        model = session.get("model_name", "unknown")
+        console.print(
+            f"  {i}. ID: [green]{session['id']}[/green] | {updated} | {messages} msgs | {model}"
+        )
+    console.print("\nUse '/load <session_id>' to resume a session.\n")
+
+
+def _load_session_by_id(
+    session_manager: SessionManager,
+    project_dir: str,
+    session_id: str,
+    model: OpenRouterModel,
+    console: Console,
+) -> Optional[Dict[str, Any]]:
+    """Load a session by ID. Returns session data or None."""
+    loaded_session = session_manager.load_session(project_dir, session_id)
+    if loaded_session:
+        model.chat_history = loaded_session["chat_history"]
+        console.print(
+            f"[green]Loaded session {session_id} ({len(model.chat_history)} messages)[/green]\n"
+        )
+        return loaded_session
+    else:
+        console.print(f"[red]Session '{session_id}' not found.[/red]")
+        console.print("[yellow]Use '/sessions' to list available sessions.[/yellow]\n")
+        return None
+
+
 def show_help():
     """Show help information for interactive mode."""
     tool_list_formatted = ""
@@ -327,8 +557,12 @@ def show_help():
     help_text = f""" [bold]Help[/bold]
 
  [cyan]Interactive Commands:[/cyan]
-  /exit
-  /help
+  /exit          - Save session and exit
+  /help          - Show this help message
+  /sessions      - List sessions for current project
+  /load <id>     - Load a session by ID
+  /new           - Start a new session (saves current first)
+  /save          - Manually save current session
 
  [cyan]CLI Commands:[/cyan]
   lmcode setup KEY
@@ -338,6 +572,10 @@ def show_help():
 
  [cyan]Workflow Hint:[/cyan] Analyze -> Plan -> Execute -> Verify -> Summarize
 
+ [cyan]Session Persistence:[/cyan]
+  Sessions are automatically saved per project directory.
+  Resume previous sessions with /load or on startup.
+
  [cyan]Available Tools:[/cyan]
 {tool_list_formatted}
 """

code_lm-0.3.2/src/lm_code/session.py

@@ -0,0 +1,283 @@
+"""
+Session persistence for LM Code CLI.
+Saves and loads conversation history per project directory.
+"""
+
+import json
+import hashlib
+import os
+import logging
+from datetime import datetime
+from pathlib import Path
+from typing import List, Dict, Optional, Any
+
+log = logging.getLogger(__name__)
+
+
+class SessionManager:
+    """Manages session persistence for conversation history."""
+
+    def __init__(self, sessions_dir: Optional[str] = None):
+        self.sessions_dir = (
+            Path(sessions_dir)
+            if sessions_dir
+            else Path.home() / ".config" / "lm-code" / "sessions"
+        )
+        self.sessions_dir.mkdir(parents=True, exist_ok=True)
+        self.current_session_id = None
+        self.current_project_dir = None
+
+    def _get_project_hash(self, project_dir: str) -> str:
+        """Generate a unique hash for a project directory path."""
+        normalized = os.path.normpath(os.path.abspath(project_dir))
+        return hashlib.sha256(normalized.encode()).hexdigest()[:16]
+
+    def _get_project_sessions_dir(self, project_dir: str) -> Path:
+        """Get the sessions directory for a specific project."""
+        project_hash = self._get_project_hash(project_dir)
+        project_sessions_dir = self.sessions_dir / project_hash
+        project_sessions_dir.mkdir(parents=True, exist_ok=True)
+        return project_sessions_dir
+
+    def save_session(
+        self,
+        project_dir: str,
+        chat_history: List[Dict[str, Any]],
+        model_name: str,
+        session_id: Optional[str] = None,
+    ) -> str:
+        """
+        Save a session to disk.
+
+        Args:
+            project_dir: The project directory path
+            chat_history: The chat history list from OpenRouterModel
+            model_name: The model being used
+            session_id: Optional existing session ID to overwrite
+
+        Returns:
+            The session ID of the saved session
+        """
+        if session_id is None:
+            session_id = datetime.now().strftime("%Y%m%d_%H%M%S")
+
+        project_sessions_dir = self._get_project_sessions_dir(project_dir)
+
+        session_data = {
+            "id": session_id,
+            "project_dir": os.path.abspath(project_dir),
+            "model_name": model_name,
+            "created_at": datetime.now().isoformat(),
+            "updated_at": datetime.now().isoformat(),
+            "chat_history": chat_history,
+            "message_count": len(
+                [
+                    msg
+                    for msg in chat_history
+                    if msg.get("role") in ("user", "assistant")
+                ]
+            ),
+        }
+
+        session_file = project_sessions_dir / f"{session_id}.json"
+
+        try:
+            with open(session_file, "w", encoding="utf-8") as f:
+                json.dump(session_data, f, indent=2, ensure_ascii=False)
+            log.info(f"Session saved: {session_id} for project {project_dir}")
+            self.current_session_id = session_id
+            self.current_project_dir = project_dir
+            return session_id
+        except Exception as e:
+            log.error(f"Failed to save session: {e}", exc_info=True)
+            raise
+
+    def load_session(
+        self, project_dir: str, session_id: str
+    ) -> Optional[Dict[str, Any]]:
+        """
+        Load a session from disk.
+
+        Args:
+            project_dir: The project directory path
+            session_id: The session ID to load
+
+        Returns:
+            Session data dict or None if not found
+        """
+        project_sessions_dir = self._get_project_sessions_dir(project_dir)
+        session_file = project_sessions_dir / f"{session_id}.json"
+
+        if not session_file.exists():
+            log.warning(f"Session file not found: {session_file}")
+            return None
+
+        try:
+            with open(session_file, "r", encoding="utf-8") as f:
+                session_data = json.load(f)
+            log.info(f"Session loaded: {session_id}")
+            self.current_session_id = session_id
+            self.current_project_dir = project_dir
+            return session_data
+        except Exception as e:
+            log.error(f"Failed to load session: {e}", exc_info=True)
+            return None
+
+    def list_sessions(self, project_dir: str) -> List[Dict[str, Any]]:
+        """
+        List all sessions for a project directory.
+
+        Args:
+            project_dir: The project directory path
+
+        Returns:
+            List of session metadata dicts, sorted by most recent first
+        """
+        project_sessions_dir = self._get_project_sessions_dir(project_dir)
+        sessions = []
+
+        for session_file in project_sessions_dir.glob("*.json"):
+            try:
+                with open(session_file, "r", encoding="utf-8") as f:
+                    session_data = json.load(f)
+                sessions.append(
+                    {
+                        "id": session_data.get("id", session_file.stem),
+                        "project_dir": session_data.get("project_dir", project_dir),
+                        "model_name": session_data.get("model_name", "unknown"),
+                        "created_at": session_data.get("created_at", ""),
+                        "updated_at": session_data.get("updated_at", ""),
+                        "message_count": session_data.get("message_count", 0),
+                    }
+                )
+            except Exception as e:
+                log.warning(f"Failed to read session file {session_file}: {e}")
+                continue
+
+        # Sort by most recent first
+        sessions.sort(key=lambda s: s.get("updated_at", ""), reverse=True)
+        return sessions
+
+    def list_all_sessions(self) -> List[Dict[str, Any]]:
+        """
+        List all sessions across all projects.
+
+        Returns:
+            List of session metadata dicts, sorted by most recent first
+        """
+        all_sessions = []
+
+        for project_hash_dir in self.sessions_dir.iterdir():
+            if not project_hash_dir.is_dir():
+                continue
+
+            for session_file in project_hash_dir.glob("*.json"):
+                try:
+                    with open(session_file, "r", encoding="utf-8") as f:
+                        session_data = json.load(f)
+                    all_sessions.append(
+                        {
+                            "id": session_data.get("id", session_file.stem),
+                            "project_dir": session_data.get("project_dir", "unknown"),
+                            "model_name": session_data.get("model_name", "unknown"),
+                            "created_at": session_data.get("created_at", ""),
+                            "updated_at": session_data.get("updated_at", ""),
+                            "message_count": session_data.get("message_count", 0),
+                        }
+                    )
+                except Exception as e:
+                    log.warning(f"Failed to read session file {session_file}: {e}")
+                    continue
+
+        # Sort by most recent first
+        all_sessions.sort(key=lambda s: s.get("updated_at", ""), reverse=True)
+        return all_sessions
+
+    def get_latest_session(self, project_dir: str) -> Optional[Dict[str, Any]]:
+        """
+        Get the most recent session for a project directory.
+
+        Args:
+            project_dir: The project directory path
+
+        Returns:
+            Session data dict or None if no sessions exist
+        """
+        sessions = self.list_sessions(project_dir)
+        if not sessions:
+            return None
+
+        latest_id = sessions[0]["id"]
+        return self.load_session(project_dir, latest_id)
+
+    def delete_session(self, project_dir: str, session_id: str) -> bool:
+        """
+        Delete a session.
+
+        Args:
+            project_dir: The project directory path
+            session_id: The session ID to delete
+
+        Returns:
+            True if deleted successfully, False otherwise
+        """
+        project_sessions_dir = self._get_project_sessions_dir(project_dir)
+        session_file = project_sessions_dir / f"{session_id}.json"
+
+        if not session_file.exists():
+            log.warning(f"Session file not found for deletion: {session_file}")
+            return False
+
+        try:
+            session_file.unlink()
+            log.info(f"Session deleted: {session_id}")
+            if self.current_session_id == session_id:
+                self.current_session_id = None
+            return True
+        except Exception as e:
+            log.error(f"Failed to delete session: {e}", exc_info=True)
+            return False
+
+    def update_session(
+        self, project_dir: str, session_id: str, chat_history: List[Dict[str, Any]]
+    ) -> bool:
+        """
+        Update an existing session with new chat history.
+
+        Args:
+            project_dir: The project directory path
+            session_id: The session ID to update
+            chat_history: The updated chat history
+
+        Returns:
+            True if updated successfully, False otherwise
+        """
+        project_sessions_dir = self._get_project_sessions_dir(project_dir)
+        session_file = project_sessions_dir / f"{session_id}.json"
+
+        if not session_file.exists():
+            log.warning(f"Session file not found for update: {session_file}")
+            return False
+
+        try:
+            with open(session_file, "r", encoding="utf-8") as f:
+                session_data = json.load(f)
+
+            session_data["chat_history"] = chat_history
+            session_data["updated_at"] = datetime.now().isoformat()
+            session_data["message_count"] = len(
+                [
+                    msg
+                    for msg in chat_history
+                    if msg.get("role") in ("user", "assistant")
+                ]
+            )
+
+            with open(session_file, "w", encoding="utf-8") as f:
+                json.dump(session_data, f, indent=2, ensure_ascii=False)
+
+            log.info(f"Session updated: {session_id}")
+            return True
+        except Exception as e:
+            log.error(f"Failed to update session: {e}", exc_info=True)
+            return False

{code_lm-0.3.0 → code_lm-0.3.2}/src/lm_code/utils.py

@@ -9,9 +9,7 @@ import json
 def count_tokens(text):
     """
     Count the number of tokens in a text string.
-
-    This is a rough estimate for Gemini 2.5 Pro, using GPT-4 tokenizer as a proxy.
-    For production, you'd want to use model-specific token counting.
+    Uses GPT-4 tokenizer as a proxy.
     """
     try:
         encoding = tiktoken.encoding_for_model("gpt-4")

code_lm-0.3.0/src/lm_code/models/gemini.py

@@ -1,43 +0,0 @@
-"""
-Gemini model integration for the CLI tool.
-"""
-
-import logging
-import time
-from rich.console import Console
-from rich.panel import Panel
-import questionary
-
-from ..utils import count_tokens
-from ..tools import get_tool, AVAILABLE_TOOLS
-
-# Setup logging (basic config, consider moving to main.py)
-logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - [%(filename)s:%(lineno)d] - %(message)s')
-log = logging.getLogger(__name__)
-
-MAX_AGENT_ITERATIONS = 10
-CONTEXT_TRUNCATION_THRESHOLD_TOKENS = 800000  # Example token limit
-
-
-class GeminiModel:
-    """Interface for Gemini models using native function calling agentic loop."""
-
-    def __init__(self, console: Console):
-        """Initialize the Gemini model interface."""
-        self.console = console
-
-        # --- Tool Definition ---
-        self.function_declarations = None  # Tools have been removed
-        # ---
-
-        # --- System Prompt (Native Functions & Planning) ---
-        self.system_instruction = "Initialize system prompt."
-        # ---
-
-        # --- Initialize Persistent History ---
-        self.chat_history = [
-            {'role': 'user', 'parts': [self.system_instruction]},
-            {'role': 'model', 'parts': ["Okay, I'm ready. Provide the directory context and your request."]}
-        ]
-        log.info("Initialized persistent chat history.")
-        # ---