codrninja 0.3.2__tar.gz → 0.4.0__tar.gz

{codrninja-0.3.2/src/codrninja.egg-info → codrninja-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codrninja
-Version: 0.3.2
+Version: 0.4.0
 Summary: AI-first coding assistant for automation
 Author: 20ZollCoder
 License: MIT

{codrninja-0.3.2 → codrninja-0.4.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "codrninja"
-version = "0.3.2"
+version = "0.4.0"
 description = "AI-first coding assistant for automation"
 readme = "README.md"
 license = {text = "MIT"}

{codrninja-0.3.2 → codrninja-0.4.0}/src/codrninja/agent.py

@@ -121,7 +121,7 @@ class Agent:
                 
                 # Progress callback
                 if self.on_progress:
-                    self.on_progress(f"✅ {tool_name}: {result.output[:100]}...")
+                    self.on_progress(f"[OK] {tool_name}: {result.output[:100]}...")
         
         return {
             'success': True,
@@ -186,7 +186,7 @@ class Agent:
     
     def _ask_permission(self, tool_name: str, params: Dict) -> bool:
         """Ask user for permission (interactive mode)."""
-        print(f"\n🔒 Permission required for: {tool_name}")
+        print(f"\n[LOCK] Permission required for: {tool_name}")
         print(f"Params: {json.dumps(params, indent=2)}")
         
         try:

{codrninja-0.3.2 → codrninja-0.4.0}/src/codrninja/cli.py

@@ -7,7 +7,7 @@ from typing import Optional
 
 from .core import AICode
 from .config import Config
-from .interactive import InteractiveSession
+from .tui import TUI
 
 
 def main():
@@ -108,18 +108,18 @@ def main():
             output(result, True)
         else:
             if result['success']:
-                print(f"\n✅ Task completed in {result['iterations']} iterations")
+                print(f"\n[OK] Task completed in {result['iterations']} iterations")
                 print(f"Tools used: {result['tool_calls']}")
                 print(f"\nResponse:\n{result['response'][:500]}...")
             else:
-                print(f"\n❌ Error: {result.get('error', 'Unknown error')}")
+                print(f"\n[FAIL] Error: {result.get('error', 'Unknown error')}")
     
     # Interactive human mode commands
     elif command == "chat" or command == "interactive" or command == "i":
         session_name = sys.argv[2] if len(sys.argv) > 2 else "default"
         session = ai.create_session(session_name)
         
-        interactive = InteractiveSession(ai, session_name)
+        interactive = TUI(ai, session_name)
         interactive.start()
     
     elif command == "new":
@@ -131,7 +131,7 @@ def main():
             output({"session": session_name, "status": "created"}, True)
         else:
             print(f"Created session: {session_name}")
-            interactive = InteractiveSession(ai, session_name)
+            interactive = TUI(ai, session_name)
             interactive.start()
     
     elif command == "help" or command == "--help" or command == "-h":
@@ -242,11 +242,11 @@ def start_tui(session_name: str = "default"):
         tui = TUI(ai, session_name)
         tui.start()
     except ImportError:
-        print("❌ Rich not installed. Install with: pip3 install rich")
+        print("[FAIL] Rich not installed. Install with: pip3 install rich")
         print("   Or use: codrninja chat my-session")
         sys.exit(1)
     except Exception as e:
-        print(f"❌ Error starting TUI: {e}")
+        print(f"[FAIL] Error starting TUI: {e}")
         sys.exit(1)
 
 

{codrninja-0.3.2 → codrninja-0.4.0}/src/codrninja/tui.py

@@ -1,18 +1,18 @@
 #!/usr/bin/env python3
 """
-codrninja TUI — Interactive coding assistant using prompt_toolkit.
+codrninja TUI -- Interactive coding assistant using prompt_toolkit.
+Features: Auto-onboarding, provider/model selection, slash commands.
 """
 
 import json
 import os
 import sys
-from typing import Optional
+from typing import Optional, List
 
 try:
     from prompt_toolkit import PromptSession
     from prompt_toolkit.completion import Completer, Completion
     from prompt_toolkit.styles import Style
-    from prompt_toolkit.key_binding import KeyBindings
     HAS_PROMPT = True
 except ImportError:
     HAS_PROMPT = False
@@ -35,18 +35,18 @@ from codrninja.tools import ToolRegistry
 
 
 CODRNINJA_LOGO = """
-╔══════════════════════════════════════════════════════════════════════════════════════╗
-║                                                                                      ║
-║   ██████╗ ██████╗ ██████╗ ██████╗ ███╗   ██╗██╗███╗   ██╗     ██╗ █████╗            ║
-║  ██╔════╝██╔═══██╗██╔══██╗██╔══██╗████╗  ██║██║████╗  ██║     ██║██╔══██╗           ║
-║  ██║     ██║   ██║██║  ██║██████╔╝██╔██╗ ██║██║██╔██╗ ██║     ██║███████║           ║
-║  ██║     ██║   ██║██║  ██║██╔══██╗██║╚██╗██║██║██║╚██╗██║██   ██║██╔══██║           ║
-║  ╚██████╗╚██████╔╝██████╔╝██║  ██║██║ ╚████║██║██║ ╚████║╚█████╔╝██║  ██║           ║
-║   ╚═════╝ ╚═════╝ ╚═════╝ ╚═╝  ╚═╝╚═╝  ╚═══╝╚═╝╚═╝  ╚═══╝ ╚════╝ ╚═╝  ╚═╝           ║
-║                                                                                      ║
-║              AI-first coding assistant for automation                                  ║
-║                                                                                      ║
-╚══════════════════════════════════════════════════════════════════════════════════════╝
++======================================================================================+
+|                                                                                      |
+|   #######  #######  #######  #######  #    #  #   #    #       #    #######          |
+|   #       #     #  #     #  #     #  ##   #  #   ##   #       #   #                  |
+|   #       #     #  #     #  #     #  # #  #  #   # #  #       #   #                  |
+|   #       #     #  #     #  #     #  #  # #  #   #  # #  ###  #   #                  |
+|   #       #     #  #     #  #     #  #   ##  #   #   ##       #   #                  |
+|   #######  #######  #######  #     #  #    #  #   #    #       #    #######          |
+|                                                                                      |
+|              AI-first coding assistant for automation                                  |
+|                                                                                      |
++======================================================================================+
 """
 
 SLASH_COMMANDS = {
@@ -68,6 +68,37 @@ SLASH_COMMANDS = {
     "/write": "Write to file",
 }
 
+# Provider configurations
+PROVIDERS = {
+    "ollama": {
+        "name": "Ollama (local, free)",
+        "models": ["codellama", "llama2", "mistral", "phi", "deepseek-coder"],
+        "needs_key": False,
+        "default_url": "http://localhost:11434",
+    },
+    "openai": {
+        "name": "OpenAI (GPT-4, GPT-3.5)",
+        "models": ["gpt-4", "gpt-4-turbo", "gpt-3.5-turbo"],
+        "needs_key": True,
+        "env_var": "OPENAI_API_KEY",
+    },
+    "anthropic": {
+        "name": "Anthropic (Claude)",
+        "models": ["claude-3-opus-20240229", "claude-3-sonnet-20240229", "claude-3-haiku-20240307"],
+        "needs_key": True,
+        "env_var": "ANTHROPIC_API_KEY",
+    },
+    "openrouter": {
+        "name": "OpenRouter (all models)",
+        "models": ["openai/gpt-4", "anthropic/claude-3-opus", "google/gemini-pro"],
+        "needs_key": True,
+        "env_var": "OPENROUTER_API_KEY",
+    },
+}
+
+CONFIG_DIR = os.path.expanduser("~/.config/codrninja")
+CONFIG_FILE = os.path.join(CONFIG_DIR, "config.json")
+
 
 class SlashCompleter(Completer):
     """Custom completer for slash commands."""
@@ -105,6 +136,10 @@ class TUI:
             print("Run: pip3 install prompt_toolkit")
             return
         
+        # Check for config -- auto-onboarding if missing
+        if not self._has_config():
+            self._onboarding()
+        
         self._show_welcome()
         
         # Create prompt session with completion
@@ -144,6 +179,140 @@ class TUI:
             except EOFError:
                 break
     
+    def _has_config(self) -> bool:
+        """Check if configuration exists."""
+        return os.path.exists(CONFIG_FILE)
+    
+    def _load_config(self) -> dict:
+        """Load configuration from file."""
+        if not self._has_config():
+            return {}
+        with open(CONFIG_FILE, 'r') as f:
+            return json.load(f)
+    
+    def _save_config(self, config: dict):
+        """Save configuration to file."""
+        os.makedirs(CONFIG_DIR, exist_ok=True)
+        with open(CONFIG_FILE, 'w') as f:
+            json.dump(config, f, indent=2)
+    
+    def _onboarding(self):
+        """Interactive onboarding for first-time users."""
+        print("\n" + "=" * 80)
+        print("  Welcome to codrninja!")
+        print("  Let's set up your AI provider.")
+        print("=" * 80 + "\n")
+        
+        # Step 1: Provider selection
+        print("Step 1: Choose your AI provider\n")
+        providers_list = list(PROVIDERS.items())
+        for i, (key, info) in enumerate(providers_list, 1):
+            print(f"  {i}. {info['name']}")
+        print()
+        
+        while True:
+            choice = input("Enter choice [1-4]: ").strip()
+            try:
+                idx = int(choice) - 1
+                if 0 <= idx < len(providers_list):
+                    provider_key, provider_info = providers_list[idx]
+                    break
+                else:
+                    print("Invalid choice. Please enter 1-4.")
+            except ValueError:
+                print("Please enter a number.")
+        
+        print(f"\n  Selected: {provider_info['name']}\n")
+        
+        # Step 2: API key (if needed)
+        api_key = None
+        if provider_info.get('needs_key'):
+            env_var = provider_info['env_var']
+            existing = os.environ.get(env_var, "")
+            
+            if existing:
+                print(f"  Found {env_var} in environment.")
+                use_existing = input("  Use existing key? [Y/n]: ").strip().lower()
+                if use_existing in ['', 'y', 'yes']:
+                    api_key = existing
+            
+            if not api_key:
+                print(f"\n  Please enter your {provider_info['name'].split('(')[0].strip()} API key:")
+                api_key = input("  > ").strip()
+                
+                if api_key:
+                    # Save to shell config
+                    shell_config = self._get_shell_config()
+                    if shell_config:
+                        with open(shell_config, "a") as f:
+                            f.write(f"\n# codrninja config\n")
+                            f.write(f"export {env_var}=\"{api_key}\"\n")
+                        print(f"\n  Saved to {shell_config}")
+        
+        # Step 3: Model selection
+        print(f"\nStep 2: Choose a model for {provider_info['name']}\n")
+        models = provider_info['models']
+        for i, model in enumerate(models, 1):
+            print(f"  {i}. {model}")
+        print()
+        
+        while True:
+            choice = input("Enter choice [1-{}]: ".format(len(models))).strip()
+            try:
+                idx = int(choice) - 1
+                if 0 <= idx < len(models):
+                    model = models[idx]
+                    break
+                else:
+                    print(f"Invalid choice. Please enter 1-{len(models)}.")
+            except ValueError:
+                print("Please enter a number.")
+        
+        print(f"\n  Selected model: {model}\n")
+        
+        # Step 4: Ollama URL (if Ollama)
+        ollama_url = None
+        if provider_key == "ollama":
+            default_url = provider_info['default_url']
+            url = input(f"Ollama URL [{default_url}]: ").strip()
+            ollama_url = url if url else default_url
+            print(f"  Using: {ollama_url}\n")
+        
+        # Save config
+        config = {
+            "provider": provider_key,
+            "model": model,
+        }
+        if ollama_url:
+            config["ollama_url"] = ollama_url
+        
+        self._save_config(config)
+        
+        # Update AI config
+        self.ai.config.default_provider = provider_key
+        self.ai.config.default_model = model
+        if ollama_url:
+            self.ai.config.ollama_url = ollama_url
+        
+        print("=" * 80)
+        print("  Setup complete!")
+        print(f"  Provider: {provider_info['name']}")
+        print(f"  Model: {model}")
+        print("=" * 80 + "\n")
+        print("  You can change this later with /model")
+        print("  Press Enter to continue...\n")
+        input()
+    
+    def _get_shell_config(self) -> Optional[str]:
+        """Get the user's shell config file path."""
+        if os.path.exists(os.path.expanduser("~/.zshrc")):
+            return os.path.expanduser("~/.zshrc")
+        elif os.path.exists(os.path.expanduser("~/.bashrc")):
+            return os.path.expanduser("~/.bashrc")
+        elif os.path.exists(os.path.expanduser("~/.bash_profile")):
+            return os.path.expanduser("~/.bash_profile")
+        return None
+    
     def _show_welcome(self):
         """Show welcome screen."""
         if self.console:
@@ -199,23 +368,13 @@ class TUI:
             result = self.tools.list_files(path=".", depth=2)
             if result.success:
                 if self.console:
-                    self.console.print(Panel(result.output, title="📁 Files", border_style="blue"))
+                    self.console.print(Panel(result.output, title="Files", border_style="blue"))
                 else:
                     print(f"\n{result.output}")
             else:
                 print(f"Error: {result.error}")
         elif cmd == '/model':
-            if self.console:
-                table = Table(title="AI Configuration", box=box.ROUNDED)
-                table.add_column("Setting", style="cyan")
-                table.add_column("Value", style="white")
-                table.add_row("Provider", self.ai.config.default_provider)
-                table.add_row("Model", self.ai.config.default_model)
-                table.add_row("Ollama URL", self.ai.config.ollama_url)
-                self.console.print(table)
-            else:
-                print(f"\nProvider: {self.ai.config.default_provider}")
-                print(f"Model: {self.ai.config.default_model}")
+            self._show_model_config()
         elif cmd == '/session':
             history = self.ai.get_history(self.session_name)
             if history:
@@ -233,7 +392,7 @@ class TUI:
                     result = self.tools.execute_command(cmd_str)
                 if result.success:
                     if self.console:
-                        self.console.print(Panel(result.output, title=f"⚡ {cmd_str}", border_style="green"))
+                        self.console.print(Panel(result.output, title=f"{cmd_str}", border_style="green"))
                     else:
                         print(f"\n{result.output}")
                 else:
@@ -247,7 +406,7 @@ class TUI:
                         lang_map = {'.py': 'python', '.js': 'javascript', '.ts': 'typescript', '.jsx': 'javascript', '.tsx': 'typescript', '.json': 'json', '.css': 'css', '.html': 'html', '.md': 'markdown'}
                         lang = lang_map.get(ext, "text")
                         syntax = Syntax(result.output, lang, theme="monokai", line_numbers=True)
-                        self.console.print(Panel(syntax, title=f"📄 {parts[1]}", border_style="blue"))
+                        self.console.print(Panel(syntax, title=f"{parts[1]}", border_style="blue"))
                     else:
                         print(f"\n{result.output}")
                 else:
@@ -273,7 +432,7 @@ class TUI:
             result = self.tools.execute_command(target)
             if result.success:
                 if self.console:
-                    self.console.print(Panel(result.output, title=f"🧪 {target}", border_style="green"))
+                    self.console.print(Panel(result.output, title=f"{target}", border_style="green"))
                 else:
                     print(f"\n{result.output}")
             else:
@@ -283,7 +442,7 @@ class TUI:
             self.tools.execute_command('git add -A')
             result = self.tools.execute_command(f'git commit -m "{msg}"')
             if result.success:
-                print(f"✅ Committed: {msg}")
+                print(f"[OK] Committed: {msg}")
             else:
                 print(f"Error: {result.error}")
         else:
@@ -291,13 +450,35 @@ class TUI:
         
         return True
     
+    def _show_model_config(self):
+        """Show and allow changing model configuration."""
+        config = self._load_config()
+        current_provider = config.get('provider', self.ai.config.default_provider)
+        current_model = config.get('model', self.ai.config.default_model)
+        
+        print("\n" + "=" * 60)
+        print("  Current Configuration")
+        print("=" * 60)
+        print(f"  Provider: {current_provider}")
+        print(f"  Model: {current_model}")
+        if 'ollama_url' in config:
+            print(f"  Ollama URL: {config['ollama_url']}")
+        print("=" * 60)
+        
+        change = input("\nChange configuration? [y/N]: ").strip().lower()
+        if change not in ['y', 'yes']:
+            return
+        
+        # Re-run onboarding
+        self._onboarding()
+    
     def _display_response(self, response: str):
         """Display AI response with Rich."""
         parts = response.split('```')
         for i, part in enumerate(parts):
             if i % 2 == 0:
                 if part.strip():
-                    self.console.print(f"\n[bold blue]🤖 AI[/bold blue]:")
+                    self.console.print(f"\n[bold blue]AI:[/bold blue]")
                     self.console.print(Markdown(part.strip()))
             else:
                 lines = part.split('\n')
@@ -314,7 +495,7 @@ class TUI:
         table.add_column("Status", style="green")
         table.add_column("Result", style="white")
         for tool in tools_used:
-            status = "✅" if tool['success'] else "❌"
+            status = "OK" if tool['success'] else "FAIL"
             result = tool['output'][:60] + "..." if len(tool['output']) > 60 else tool['output']
             table.add_row(tool['tool'], status, result)
         self.console.print(table)

{codrninja-0.3.2 → codrninja-0.4.0/src/codrninja.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codrninja
-Version: 0.3.2
+Version: 0.4.0
 Summary: AI-first coding assistant for automation
 Author: 20ZollCoder
 License: MIT