code-lm 0.1.0__py3-none-any.whl

code_lm-0.1.0.dist-info/METADATA

@@ -0,0 +1,181 @@
+Metadata-Version: 2.2
+Name: code-lm
+Version: 0.1.0
+Summary: An AI coding assistant CLI using OpenRouter and various LLM models.
+Home-page: https://github.com/Panagiotis897/lm-code
+Author: Panagiotis897
+Author-email: Panagiotis897 <your.email@example.com>
+Project-URL: Homepage, https://github.com/Panagiotis897/lm-code
+Project-URL: Bug Tracker, https://github.com/Panagiotis897/lm-code/issues
+Classifier: Programming Language :: Python :: 3
+Classifier: Operating System :: OS Independent
+Classifier: Development Status :: 3 - Alpha
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Developers
+Classifier: Topic :: Software Development
+Classifier: Topic :: Utilities
+Requires-Python: >=3.7
+Description-Content-Type: text/markdown
+Requires-Dist: click>=8.0
+Requires-Dist: rich>=13.0
+Requires-Dist: requests>=2.25.0
+Requires-Dist: PyYAML>=6.0
+Requires-Dist: tiktoken>=0.6.0
+Requires-Dist: questionary>=2.0.0
+Dynamic: author
+Dynamic: home-page
+Dynamic: requires-python
+
+# Gemini Code
+
+A powerful AI coding assistant for your terminal, powered by Gemini 2.5 Pro with support for other LLM models.
+More information [here](https://blossom-tarsier-434.notion.site/Gemini-Code-1c6c13716ff180db86a0c7f4b2da13ab?pvs=4)
+
+## Features
+
+- Interactive chat sessions in your terminal
+- Multiple model support (Gemini 2.5 Pro, Gemini 1.5 Pro, and more)
+- Basic history management (prevents excessive length)
+- Markdown rendering in the terminal
+- Automatic tool usage by the assistant:
+  - File operations (view, edit, list, grep, glob)
+  - Directory operations (ls, tree, create_directory)
+  - System commands (bash)
+  - Quality checks (linting, formatting)
+  - Test running capabilities (pytest, etc.)
+
+## Installation
+
+### Method 1: Install from PyPI (Recommended)
+
+```bash
+# Install directly from PyPI
+pip install gemini-code
+```
+
+### Method 2: Install from Source
+
+```bash
+# Clone the repository
+git clone https://github.com/raizamartin/gemini-code.git
+cd gemini-code
+
+# Install the package
+pip install -e .
+```
+
+## Setup
+
+Before using Gemini CLI, you need to set up your API keys:
+
+```bash
+# Set up Google API key for Gemini models
+gemini setup YOUR_GOOGLE_API_KEY
+```
+
+## Usage
+
+```bash
+# Start an interactive session with the default model
+gemini
+
+# Start a session with a specific model
+gemini --model models/gemini-2.5-pro-exp-03-25
+
+# Set default model
+gemini set-default-model models/gemini-2.5-pro-exp-03-25
+
+# List all available models
+gemini list-models
+```
+
+## Interactive Commands
+
+During an interactive session, you can use these commands:
+
+- `/exit` - Exit the chat session
+- `/help` - Display help information
+
+## How It Works
+
+### Tool Usage
+
+Unlike direct command-line tools, the Gemini CLI's tools are used automatically by the assistant to help answer your questions. For example:
+
+1. You ask: "What files are in the current directory?"
+2. The assistant uses the `ls` tool behind the scenes
+3. The assistant provides you with a formatted response
+
+This approach makes the interaction more natural and similar to how Claude Code works.
+
+## Development
+
+This project is under active development. More models and features will be added soon!
+
+### Recent Changes in v0.1.69
+
+- Added test_runner tool to execute automated tests (e.g., pytest)
+- Fixed syntax issues in the tool definitions
+- Improved error handling in tool execution
+- Updated status displays during tool execution with more informative messages
+- Added additional utility tools (directory_tools, quality_tools, task_complete_tool, summarizer_tool)
+
+### Recent Changes in v0.1.21
+
+- Implemented native Gemini function calling for much more reliable tool usage
+- Rewritten the tool execution system to use Gemini's built-in function calling capability
+- Enhanced the edit tool to better handle file creation and content updating
+- Updated system prompt to encourage function calls instead of text-based tool usage
+- Fixed issues with Gemini not actively creating or modifying files
+- Simplified the BaseTool interface to support both legacy and function call modes
+
+### Recent Changes in v0.1.20
+
+- Fixed error with Flask version check in example code
+- Improved error handling in system prompt example code
+
+### Recent Changes in v0.1.19
+
+- Improved system prompt to encourage more active tool usage
+- Added thinking/planning phase to help Gemini reason about solutions
+- Enhanced response format to prioritize creating and modifying files over printing code
+- Filtered out thinking stages from final output to keep responses clean
+- Made Gemini more proactive as a coding partner, not just an advisor
+
+### Recent Changes in v0.1.18
+
+- Updated default model to Gemini 2.5 Pro Experimental (models/gemini-2.5-pro-exp-03-25)
+- Updated system prompts to reference Gemini 2.5 Pro
+- Improved model usage and documentation
+
+### Recent Changes in v0.1.17
+
+- Added `list-models` command to show all available Gemini models
+- Improved error handling for models that don't exist or require permission
+- Added model initialization test to verify model availability
+- Updated help documentation with new commands
+
+### Recent Changes in v0.1.16
+
+- Fixed file creation issues: The CLI now properly handles creating files with content
+- Enhanced tool pattern matching: Added support for more formats that Gemini might use
+- Improved edit tool handling: Better handling of missing arguments when creating files
+- Added special case for natural language edit commands (e.g., "edit filename with content: ...")
+
+### Recent Changes in v0.1.15
+
+- Fixed tool execution issues: The CLI now properly processes tool calls and executes Bash commands correctly
+- Fixed argument parsing for Bash tool: Commands are now passed as a single argument to avoid parsing issues
+- Improved error handling in tools: Better handling of failures and timeouts
+- Updated model name throughout the codebase to use `gemini-1.5-pro` consistently
+
+### Known Issues
+
+- If you created a config file with earlier versions, you may need to delete it to get the correct defaults:
+  ```bash
+  rm -rf ~/.config/gemini-code
+  ```
+
+## License
+
+MIT

code_lm-0.1.0.dist-info/RECORD

@@ -0,0 +1,22 @@
+gemini_cli/__init__.py,sha256=r8biTIkUHhaUtKuiP90Gyodr3Dh1Uk1ETMM9gIpt4UQ,107
+gemini_cli/config.py,sha256=ZzD3O-dIH3fXkucDdaeBFb0N5vODIf6vqOtT7d2kmhA,2557
+gemini_cli/main.py,sha256=x-F-DXFWjCQEziBg7edGz3bsE21hBwcndg82XTgbCl4,9429
+gemini_cli/utils.py,sha256=fPmaS8Hi6trX8lhuqu52as-ljf3rJhXY0P6QZBmu52A,538
+gemini_cli/models/__init__.py,sha256=Ql8zY71YLQjt528UkZbakpueMzQcdsiXJXGfP21n-IY,157
+gemini_cli/models/gemini.py,sha256=aA_aMsay6XipgaH5kCNyC9uUWanzBWQWyYyz5MsDGsk,37507
+gemini_cli/models/openrouter.py,sha256=BfzF524GI7mvjeu_AqvSZXNOUQHiTUMyfRqk4URE8B8,31954
+gemini_cli/tools/__init__.py,sha256=gONeoSuMKLFLcLcIQPwQX_VbMqNLSwbAynXNDx95Sm4,3641
+gemini_cli/tools/base.py,sha256=urYf9rQPaDhM-kQAGcSgDtJ4QyEefWftBtdjN3ixmf4,3293
+gemini_cli/tools/directory_tools.py,sha256=xzmgaYromC252FIFiHV11Du5NpMsi-8pRuFmp4hvMPk,5704
+gemini_cli/tools/file_tools.py,sha256=4jQElvVsMXSN7LDiTtbJSCLdDtp9dW9azPwyCr4SKGs,11819
+gemini_cli/tools/quality_tools.py,sha256=qSL5Ie0ZCNdeDD7VoybGI50KXCNp5ulBVKckpuyekW0,4238
+gemini_cli/tools/summarizer_tool.py,sha256=js0UVyP9M_VtURuT6ZnUIoogWvaQiSH_dAq12-fnKw0,7265
+gemini_cli/tools/system_tools.py,sha256=-a1aesREYo-pIpIa5mMUMLailoswfHRjX8Q-8tSLmHI,2038
+gemini_cli/tools/task_complete_tool.py,sha256=B3qhfwZGLgrxEIbDaWJ8Xrj-G2Ut5xUFi239o0KDk-8,2487
+gemini_cli/tools/test_runner.py,sha256=p6y18BmdHmCNzlqk9V3zk-7auYfitpyLrCrcOwcyiqs,4100
+gemini_cli/tools/tree_tool.py,sha256=fX60z6dw2q-SHdJJCbVLWnxta8Lq7cxwi94A2qWTn4Q,4581
+code_lm-0.1.0.dist-info/METADATA,sha256=4ncXd05mBqbW9LIDfSS4-Rrx5CL4WLmZgSCEiGFitLA,6261
+code_lm-0.1.0.dist-info/WHEEL,sha256=nn6H5-ilmfVryoAQl3ZQ2l8SH5imPWFpm1A5FgEuFV4,91
+code_lm-0.1.0.dist-info/entry_points.txt,sha256=X0PFugPAiZDO8YsKuHnGrQJJ2ZSxRtKcNsivc96C9k8,47
+code_lm-0.1.0.dist-info/top_level.txt,sha256=SRsqiKyD15llWPipzbeaivAzk2CXqZJTf1QDwaxFOgg,11
+code_lm-0.1.0.dist-info/RECORD,,

code_lm-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (75.8.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

code_lm-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+lmcode = gemini_cli.main:cli

code_lm-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+gemini_cli

gemini_cli/__init__.py

@@ -0,0 +1,5 @@
+"""
+LM CLI - A command-line interface for interacting with various LLM models.
+"""
+
+__version__ = "0.1.70"

gemini_cli/config.py

@@ -0,0 +1,78 @@
+"""
+Configuration management for LM CLI.
+"""
+
+import os
+import yaml
+from pathlib import Path
+
+class Config:
+    """Manages configuration for the LM CLI application."""
+    
+    def __init__(self):
+        self.config_dir = Path.home() / ".config" / "gemini-code"
+        self.config_file = self.config_dir / "config.yaml"
+        self._ensure_config_exists()
+        self.config = self._load_config()
+    
+    def _ensure_config_exists(self):
+        """Create config directory and file if they don't exist."""
+        self.config_dir.mkdir(parents=True, exist_ok=True)
+        
+        if not self.config_file.exists():
+            default_config = {
+                "api_keys": {},
+                "default_model": "qwen/qwen-2.5-coder-32b-instruct:free",
+                "settings": {
+                    "max_tokens": 1000000,
+                    "temperature": 0.7,
+                    "token_warning_threshold": 800000,
+                    "auto_compact_threshold": 950000,
+                }
+            }
+            
+            with open(self.config_file, 'w') as f:
+                yaml.dump(default_config, f)
+    
+    def _load_config(self):
+        """Load configuration from file."""
+        with open(self.config_file, 'r') as f:
+            return yaml.safe_load(f)
+    
+    def _save_config(self):
+        """Save configuration to file."""
+        with open(self.config_file, 'w') as f:
+            yaml.dump(self.config, f)
+    
+    def get_api_key(self, provider):
+        """Get API key for a specific provider."""
+        return self.config.get("api_keys", {}).get(provider)
+    
+    def set_api_key(self, provider, key):
+        """Set API key for a specific provider."""
+        if "api_keys" not in self.config:
+            self.config["api_keys"] = {}
+        
+        self.config["api_keys"][provider] = key
+        self._save_config()
+    
+    def get_default_model(self):
+        """Get the default model."""
+        return self.config.get("default_model")
+    
+    def set_default_model(self, model):
+        """Set the default model."""
+        self.config["default_model"] = model
+        self._save_config()
+    
+    def get_setting(self, setting, default=None):
+        """Get a specific setting."""
+        return self.config.get("settings", {}).get(setting, default)
+    
+    def set_setting(self, setting, value):
+        """Set a specific setting."""
+        if "settings" not in self.config:
+            self.config["settings"] = {}
+        
+        self.config["settings"][setting] = value
+        self._save_config()

gemini_cli/main.py

@@ -0,0 +1,225 @@
+"""
+Main entry point for the LM Code application.
+Targets OpenRouter and other supported models. Includes ASCII Art welcome.
+"""
+
+import os
+import sys
+import click
+from rich.console import Console
+from rich.markdown import Markdown
+from rich.panel import Panel
+from pathlib import Path
+import yaml
+import logging
+import time
+
+from .models.openrouter import OpenRouterModel, list_available_models
+from .config import Config
+from .utils import count_tokens
+from .tools import AVAILABLE_TOOLS
+
+# Setup console and config
+console = Console()  # Create console instance HERE
+try:
+    config = Config()
+except Exception as e:
+    console.print(f"[bold red]Error loading configuration:[/bold red] {e}")
+    config = None
+
+# Setup logging - MORE EXPLICIT CONFIGURATION
+log_level = os.environ.get("LOG_LEVEL", "WARNING").upper()  # <-- Default back to WARNING
+log_format = '%(asctime)s - %(levelname)s - [%(filename)s:%(lineno)d] - %(message)s'
+
+# Get root logger and set level
+root_logger = logging.getLogger()
+root_logger.setLevel(log_level)
+
+# Remove existing handlers to avoid duplicates if basicConfig was called elsewhere
+for handler in root_logger.handlers[:]:
+    root_logger.removeHandler(handler)
+
+# Add a stream handler to output to console
+stream_handler = logging.StreamHandler(sys.stdout)
+stream_handler.setLevel(log_level)
+formatter = logging.Formatter(log_format)
+stream_handler.setFormatter(formatter)
+root_logger.addHandler(stream_handler)
+
+log = logging.getLogger(__name__)  # Get logger for this module
+log.info(f"Logging initialized with level: {log_level}")  # Confirm level
+
+# --- Default Model ---
+DEFAULT_MODEL = "qwen/qwen-2.5-coder-32b-instruct:free"
+# --- ---
+
+# --- Supported Models ---
+SUPPORTED_MODELS = [
+    {"id": "qwen/qwen-2.5-coder-32b-instruct:free", "description": "Qwen 2.5 Coder 32B"},
+    {"id": "qwen/qwq-32b:free", "description": "Qwen QWQ 32B"},
+    {"id": "deepseek/deepseek-r1:free", "description": "DeepSeek R1"},
+    {"id": "google/gemma-3-27b-it:free", "description": "Gemma 3 27B Italian"},
+    {"id": "google/gemini-2.5-pro-exp-03-25:free", "description": "Gemini 2.5 Pro Experimental"},
+]
+# --- ---
+
+# --- ASCII Art Definition ---
+LM_CODE_ART = r"""
+
+[green]
+  ██╗     ███╗   ███╗     ██████╗ ██████╗ ██████╗ ███████╗
+  ██║     ████╗ ████║    ██╔════╝██╔═══██╗██╔══██╗██╔════╝
+  ██║     ██╔████╔██║    ██║     ██║   ██║██║  ██║█████╗  
+  ██║     ██║╚██╔╝██║    ██║     ██║   ██║██║  ██║██╔══╝  
+  ███████╗██║ ╚═╝ ██║    ╚██████╗╚██████╔╝██████╔╝███████╗
+  ╚══════╝╚═╝     ╚═╝     ╚═════╝ ╚═════╝ ╚═════╝ ╚══════╝
+[/green]
+"""
+# --- End ASCII Art ---
+
+
+CONTEXT_SETTINGS = dict(help_option_names=['-h', '--help'])
+
+@click.group(invoke_without_command=True, context_settings=CONTEXT_SETTINGS)
+@click.option(
+    '--model', '-m',
+    help=f'Model ID to use (e.g., qwen/qwen-2.5-coder-32b-instruct:free). Default: {DEFAULT_MODEL}',
+    default=None
+)
+@click.pass_context
+def cli(ctx, model):
+    """Interactive CLI for LM models with coding assistance tools."""
+    if not config:
+        console.print("[bold red]Configuration could not be loaded. Cannot proceed.[/bold red]")
+        sys.exit(1)
+
+    if ctx.invoked_subcommand is None:
+        model_name_to_use = model or config.get_default_model() or DEFAULT_MODEL
+        log.info(f"Attempting to start interactive session with model: {model_name_to_use}")
+        # Pass the console object to start_interactive_session
+        start_interactive_session(model_name_to_use, console)
+
+@cli.command(name="setup")
+@click.argument('key', required=True)
+def setup(key):
+    if not config: console.print("[bold red]Config error.[/bold red]"); return
+    try: config.set_api_key("openrouter", key); console.print("[green]✓[/green] OpenRouter API key saved.")
+    except Exception as e: console.print(f"[bold red]Error saving API key:[/bold red] {e}")
+
+@cli.command(name="set-default-model")
+@click.argument('model_name', required=True)
+def set_default_model(model_name):
+    if not config: console.print("[bold red]Config error.[/bold red]"); return
+    try: config.set_default_model(model_name); console.print(f"[green]✓[/green] Default model set to [bold]{model_name}[/bold].")
+    except Exception as e: console.print(f"[bold red]Error setting default model:[/bold red] {e}")
+
+@cli.command(name="list-models")
+def list_models():
+    if not config: console.print("[bold red]Config error.[/bold red]"); return
+    api_key = config.get_api_key("openrouter")
+    if not api_key: console.print("[bold red]Error:[/bold red] API key not found. Run 'lmcode setup YOUR_OPENROUTER_API_KEY'."); return
+    console.print("[yellow]Fetching models...[/yellow]")
+    try:
+        models_list = list_available_models(api_key)
+        if not models_list:
+            console.print("[red]No models found or fetch error.[/red]")
+        else:
+            console.print("\n[bold cyan]Supported Models:[/bold cyan]")
+            for model in SUPPORTED_MODELS:
+                console.print(f"- [bold green]{model['id']}[/bold green]: {model['description']}")
+            console.print("\nUse 'lmcode --model MODEL' or 'lmcode set-default-model MODEL'.")
+    except Exception as e:
+        console.print(f"[bold red]Error listing models:[/bold red] {e}")
+        log.error("List models failed", exc_info=True)
+
+
+# --- Modified start_interactive_session to use OpenRouter ---
+def start_interactive_session(model_name: str, console: Console):
+    """Start an interactive chat session with the selected OpenRouter model."""
+    if not config: console.print("[bold red]Config error.[/bold red]"); return
+
+    # --- Display Welcome Art ---
+    console.clear()
+    console.print(LM_CODE_ART)
+    console.print(Panel("[b]Welcome to LM Code AI Assistant![/b]", border_style="green", expand=False))
+    time.sleep(0.1)
+    # --- End Welcome Art ---
+
+    api_key = config.get_api_key("openrouter")
+    if not api_key:
+        console.print("\n[bold red]Error:[/bold red] OpenRouter API key not found.")
+        console.print("Please run [bold]'lmcode setup YOUR_OPENROUTER_API_KEY'[/bold] first.")
+        return
+
+    try:
+        console.print(f"\nInitializing model [bold]{model_name}[/bold]...")
+        # Pass the console object to OpenRouterModel constructor
+        model = OpenRouterModel(api_key=api_key, console=console, model_name=model_name)
+        console.print("[green]Model initialized successfully.[/green]\n")
+
+    except Exception as e:
+        console.print(f"\n[bold red]Error initializing model '{model_name}':[/bold red] {e}")
+        log.error(f"Failed to initialize model {model_name}", exc_info=True)
+        console.print("Please check model name, API key permissions, network. Use 'lmcode list-models'.")
+        return
+
+    # --- Session Start Message ---
+    console.print("Type '/help' for commands, '/exit' or Ctrl+C to quit.")
+
+    while True:
+        try:
+            user_input = console.input("[bold green]You:[/bold green] ")
+
+            if user_input.lower() == '/exit': break
+            elif user_input.lower() == '/help': show_help(); continue
+
+            # Display initial "thinking" status - generate handles intermediate ones
+            response_text = model.generate(user_input)
+
+            if response_text is None and user_input.startswith('/'): console.print(f"[yellow]Unknown command:[/yellow] {user_input}"); continue
+            elif response_text is None: console.print("[red]Received an empty response from the model.[/red]"); log.warning("generate() returned None unexpectedly."); continue
+
+            console.print("[bold green]Assistant:[/bold green]")
+            console.print(Markdown(response_text), highlight=True)
+
+        except KeyboardInterrupt:
+            console.print("\n[yellow]Session interrupted. Exiting.[/yellow]")
+            break
+        except Exception as e:
+            console.print(f"\n[bold red]An error occurred during the session:[/bold red] {e}")
+            log.error("Error during interactive loop", exc_info=True)
+
+
+def show_help():
+    """Show help information for interactive mode."""
+    tool_list_formatted = ""
+    if AVAILABLE_TOOLS:
+        # Add indentation for the bullet points
+        tool_list_formatted = "\n".join([f"  • [white]`{name}`[/white]" for name in sorted(AVAILABLE_TOOLS.keys())])
+    else:
+        tool_list_formatted = "  (No tools available)"
+        
+    # Use direct rich markup and ensure newlines are preserved
+    help_text = f""" [bold]Help[/bold]
+
+ [cyan]Interactive Commands:[/cyan]
+  /exit
+  /help
+
+ [cyan]CLI Commands:[/cyan]
+  lmcode setup KEY
+  lmcode list-models
+  lmcode set-default-model NAME
+  lmcode --model NAME
+
+ [cyan]Workflow Hint:[/cyan] Analyze -> Plan -> Execute -> Verify -> Summarize
+
+ [cyan]Available Tools:[/cyan]
+{tool_list_formatted}
+"""
+    # Print directly to Panel without Markdown wrapper
+    console.print(Panel(help_text, title="Help", border_style="green", expand=False))
+
+
+if __name__ == "__main__":
+    cli()

gemini_cli/models/__init__.py

@@ -0,0 +1,6 @@
+"""
+Model interfaces for different LLM providers.
+"""
+
+# Import model classes for easy access
+from .openrouter import OpenRouterModel, list_available_models