aizen-ai-cli 2.2.2__tar.gz → 2.2.4__tar.gz

{aizen_ai_cli-2.2.2/aizen_ai_cli.egg-info → aizen_ai_cli-2.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aizen-ai-cli
-Version: 2.2.2
+Version: 2.2.4
 Summary: Aizen AI Agent — A professional-grade AI coding assistant for your terminal.
 Author: Irtaza Malik
 License: MIT

{aizen_ai_cli-2.2.2 → aizen_ai_cli-2.2.4}/aizen/config.py

@@ -3,12 +3,12 @@ import json
 import logging
 import os
 import shutil
+import ssl
 import sys
 import threading
 import time
-import urllib.request
 import urllib.error
-import ssl
+import urllib.request
 from importlib.metadata import PackageNotFoundError
 from importlib.metadata import version as _pkg_version
 
@@ -21,7 +21,7 @@ logger = logging.getLogger("aizen")
 
 # Read version from installed package metadata (stays in sync with pyproject.toml).
 # Falls back to a hardcoded value only when running from source without installing.
-_FALLBACK_VERSION = "2.2.2"
+_FALLBACK_VERSION = "2.2.4"
 try:
     VERSION = _pkg_version("aizen-ai-cli")
 except PackageNotFoundError:
@@ -29,16 +29,16 @@ except PackageNotFoundError:
 CONFIG_PATH = os.path.expanduser("~/.aizen_config.json")
 SESSIONS_DIR = os.path.expanduser("~/.aizen_sessions")
 BACKUPS_DIR = os.path.expanduser("~/.aizen_backups")
-DEFAULT_MODEL = "nvidia/nemotron-3-super-120b-a12b:free"
-
-AIZEN_ASCII = r"""[bold magenta]
-    _    _
-   / \  (_)_______ _ __
-  / _ \ | |_  / _ \ '_ \
- / ___ \| |/ /  __/ | | |
-/_/   \_\_/___\___|_| |_|
-[/bold magenta]
-    [dim]by Irtaza Malik[/dim]
+DEFAULT_MODEL = "openrouter/free"
+
+AIZEN_ASCII = r"""[bold #ffabf3]
+ █████╗ ██╗███████╗███████╗███╗   ██╗
+██╔══██╗██║╚══███╔╝██╔════╝████╗  ██║
+███████║██║  ███╔╝ █████╗  ██╔██╗ ██║
+██╔══██║██║ ███╔╝  ██╔══╝  ██║╚██╗██║
+██║  ██║██║███████╗███████╗██║ ╚████║
+╚═╝  ╚═╝╚═╝╚══════╝╚══════╝╚═╝  ╚═══╝
+[/bold #ffabf3]
 """
 
 # Safe commands that auto-execute without confirmation
@@ -134,7 +134,7 @@ def build_system_prompt(config: dict | None = None) -> str:
                         f"The following rules are defined by the project maintainers "
                         f"(from {rules_file}):\n\n{project_rules}"
                     )
-                    console.print(f"  [dim]📋 Loaded project rules from {rules_file}[/dim]")
+                    console.print(f"[bold #ffabf3][SYSTEM][/bold #ffabf3] Project rules loaded from [#d3fbff]{rules_file}[/#d3fbff]")
                 break  # Only use the first rules file found
             except Exception as e:
                 logger.debug("Failed to load project rules from %s: %s", rules_file, e)
@@ -258,7 +258,7 @@ def _do_update_check(config: dict):
             ctx.load_verify_locations(cafile=certifi.where())
         except ImportError:
             ctx = ssl._create_unverified_context()
-            
+
         url = "https://pypi.org/pypi/aizen-ai-cli/json"
         req = urllib.request.Request(url, headers={"User-Agent": "aizen-ai-cli"})
         with urllib.request.urlopen(req, timeout=3, context=ctx) as response:
@@ -273,12 +273,20 @@ def _do_update_check(config: dict):
             except Exception as e:
                 logger.debug("Failed to save config after update check: %s", e)
 
-            if latest != VERSION:
+            try:
+                latest_parts = tuple(map(int, latest.split('.')))
+                current_parts = tuple(map(int, VERSION.split('.')))
+                is_newer = latest_parts > current_parts
+            except Exception:
+                is_newer = latest != VERSION
+                
+            if is_newer:
                 console.print(
                     f"\n[bold magenta]🔔 Update available:[/bold magenta] v{VERSION} → v{latest}"
                 )
                 console.print("[dim]Run: pip install -U aizen-ai-cli (or brew upgrade aizen)[/dim]")
-                console.print("[dim]Then restart Aizen to use the new version![/dim]\n")
+                console.print("[dim]Then restart Aizen to use the new version![/dim]\n"
+                )
     except Exception as e:
         logger.debug("Update check failed (network/parsing): %s", e)
 
@@ -295,11 +303,20 @@ def check_for_updates(config: dict | None = None):
         # Check if we have a cached latest version that's newer
         cached = config.get("_latest_version")
         if cached and cached != VERSION:
-            console.print(
-                f"\n[bold magenta]🔔 Update available:[/bold magenta] v{VERSION} → v{cached}"
-            )
-            console.print("[dim]Run: pip install -U aizen-ai-cli (or brew upgrade aizen)[/dim]")
-            console.print("[dim]Then restart Aizen to use the new version![/dim]\n")
+            try:
+                cached_parts = tuple(map(int, cached.split('.')))
+                current_parts = tuple(map(int, VERSION.split('.')))
+                is_newer = cached_parts > current_parts
+            except Exception:
+                is_newer = cached != VERSION
+                
+            if is_newer:
+                console.print(
+                    f"\n[bold magenta]🔔 Update available:[/bold magenta] v{VERSION} → v{cached}"
+                )
+                console.print("[dim]Run: pip install -U aizen-ai-cli (or brew upgrade aizen)[/dim]")
+                console.print("[dim]Then restart Aizen to use the new version![/dim]\n"
+                )
         return
 
     thread = threading.Thread(target=_do_update_check, args=(config,), daemon=True)
@@ -329,7 +346,7 @@ def _do_fetch_models():
             ctx.load_verify_locations(cafile=certifi.where())
         except ImportError:
             ctx = ssl._create_unverified_context()
-            
+
         req = urllib.request.Request("https://openrouter.ai/api/v1/models")
         with urllib.request.urlopen(req, timeout=5, context=ctx) as response:
             data = json.loads(response.read().decode())

{aizen_ai_cli-2.2.2 → aizen_ai_cli-2.2.4}/aizen/main.py

@@ -14,15 +14,15 @@ import sys
 from typing import Any
 
 from openai import APIConnectionError as OpenAIConnectionError
-from openai import APITimeoutError, AsyncOpenAI, AuthenticationError
+from openai import APITimeoutError, AsyncOpenAI, AuthenticationError, BadRequestError
 from openai import RateLimitError as OpenAIRateLimitError
 from prompt_toolkit import PromptSession
 from prompt_toolkit.filters import completion_is_selected, has_completions
-from prompt_toolkit.formatted_text import HTML
+from prompt_toolkit.formatted_text import FormattedText
 from prompt_toolkit.key_binding import KeyBindings
 from rich.live import Live
 from rich.markdown import Markdown
-from rich.panel import Panel
+from rich.spinner import Spinner
 from rich.text import Text
 
 from .commands import AizenCompleter, handle_slash_command
@@ -212,16 +212,12 @@ async def main_loop():
 
     # ── Header ──
     console.print(AIZEN_ASCII)
-    header = Text()
-    header.append(f"v{VERSION}", style="bold magenta")
-    header.append("  │  ", style="dim")
-    header.append(get_active_model(), style="cyan")
+    console.print(f"[bold #ffabf3][SYSTEM][/bold #ffabf3] Initializing Aizen AI v{VERSION}...")
+    console.print(f"[bold #ffabf3][SYSTEM][/bold #ffabf3] Model: {get_active_model()}")
     if auto_approve:
-        header.append("  │  ", style="dim")
-        header.append("YOLO MODE", style="bold red")
-    console.print(header)
+        console.print("[bold #ffabf3][SYSTEM][/bold #ffabf3] Mode: YOLO")
     console.print(
-        "[dim]Type /help for commands  •  @file to attach  •  exit to quit[/dim]\n"
+        "\n[dim]Type /help for commands  •  @file to attach  •  exit to quit[/dim]\n"
     )
 
     # ── Keybindings ──
@@ -239,10 +235,12 @@ async def main_loop():
         try:
             # ── Multi-line Input ──
             lines = []
-            prompt_html = HTML(
-                "<ansimagenta>╭─</ansimagenta> <ansimagenta><b>👤 You</b></ansimagenta>\n"
-                "<ansimagenta>╰─❯</ansimagenta> "
-            )
+            prompt_html = FormattedText([
+                ("fg:#ffabf3", "➜"),
+                ("", " "),
+                ("fg:#d3fbff", "~"),
+                ("", " ")
+            ])
             first_line = await session.prompt_async(prompt_html)
             lines.append(first_line)
 
@@ -250,7 +248,7 @@ async def main_loop():
             while lines[-1].rstrip().endswith("\\"):
                 lines[-1] = lines[-1].rstrip()[:-1]  # Remove trailing backslash
                 continuation = await session.prompt_async(
-                    HTML("<ansimagenta>  ⋮ </ansimagenta> ")
+                    FormattedText([("", "  ")])
                 )
                 lines.append(continuation)
 
@@ -346,9 +344,7 @@ async def main_loop():
                         "Exploring...",
                     ]
                 )
-                spinner_display = Text()
-                spinner_display.append("  ✦ ", style="bold magenta")
-                spinner_display.append(spinner_label, style="dim italic")
+                spinner_display = Spinner("dots", text=Text(spinner_label, style="#8e8e93 italic"), style="#ffabf3 bold")
 
                 try:
                     with Live(
@@ -378,23 +374,14 @@ async def main_loop():
                                 full_content += delta.content
                                 # Live-render Markdown in a panel
                                 try:
-                                    rendered = Panel(
-                                        Markdown(full_content),
-                                        title="[bold magenta]✦ Aizen[/bold magenta]",
-                                        border_style="magenta",
-                                        padding=(1, 2),
-                                    )
+                                    # Prepend AIZEN: styling before the markdown
+                                    display_content = f"**AIZEN:** {full_content}"
+                                    rendered = Markdown(display_content)
                                     live.update(rendered)
                                 except Exception:
                                     # Fallback for incomplete markdown
-                                    live.update(
-                                        Panel(
-                                            Text(full_content),
-                                            title="[bold magenta]✦ Aizen[/bold magenta]",
-                                            border_style="magenta",
-                                            padding=(1, 2),
-                                        )
-                                    )
+                                    display_text = Text.from_markup(f"[bold #ffabf3]AIZEN:[/bold #ffabf3] {full_content}")
+                                    live.update(display_text)
 
                             # ── Tool call tokens ──
                             if delta.tool_calls:
@@ -427,10 +414,9 @@ async def main_loop():
                                 ]
                                 if names and not full_content:
                                     tool_text = Text()
-                                    tool_text.append("  ⚙️  ", style="magenta")
+                                    tool_text.append("AIZEN: ", style="bold #ffabf3")
                                     tool_text.append(
-                                        f"Preparing: {', '.join(names)}",
-                                        style="dim italic",
+                                        f"Invoking [#d3fbff]{', '.join(names)}[/#d3fbff]...",
                                     )
                                     live.update(tool_text)
 
@@ -470,8 +456,15 @@ async def main_loop():
                         "[dim]Hint: Check your internet connection or API base URL.[/dim]"
                     )
                     break
+                except BadRequestError as e:
+                    logger.error("Bad request to API: %s", e)
+                    console.print(f"\n[bold red]Bad Request Error:[/bold red] {e}")
+                    console.print(
+                        "[dim]Hint: This usually means the model ID is invalid or the context length was exceeded.[/dim]"
+                    )
+                    break
                 except Exception as e:
-                    logger.exception("Unexpected API error: %s", e)
+                    logger.error("Unexpected API error: %s", e)
                     console.print(f"\n[bold red]API Error:[/bold red] {e}")
                     error_str = str(e).lower()
                     if "401" in error_str or "unauthorized" in error_str:

{aizen_ai_cli-2.2.2 → aizen_ai_cli-2.2.4}/aizen/tools.py

@@ -697,19 +697,28 @@ def run_command_impl(command: str, auto_approve: bool = False, timeout: int = 12
                     logger.warning("Command timed out after %ds: %s", timeout, command)
                     return f"Error: Command timed out after {timeout} seconds."
 
-                # Read available stdout non-blockingly
-                if proc.stdout:
-                    rlist, _, _ = select.select([proc.stdout], [], [], 0.1)
-                    if rlist:
-                        line = proc.stdout.readline()
-                        if line:
-                            stdout_lines.append(line)
-                            # Show live output tail (last 15 lines)
-                            tail = "".join(stdout_lines[-15:])
-                            display = Text()
-                            display.append(f"  ▶ Running ({elapsed:.0f}s)\n", style="dim italic")
-                            display.append(tail.rstrip(), style="dim")
-                            live.update(display)
+                # Read available stdout and stderr non-blockingly
+                reads = []
+                if proc.stdout: reads.append(proc.stdout)
+                if proc.stderr: reads.append(proc.stderr)
+                
+                if reads:
+                    rlist, _, _ = select.select(reads, [], [], 0.1)
+                    for fd in rlist:
+                        if fd == proc.stdout:
+                            line = proc.stdout.readline()
+                            if line:
+                                stdout_lines.append(line)
+                                # Show live output tail (last 15 lines)
+                                tail = "".join(stdout_lines[-15:])
+                                display = Text()
+                                display.append(f"  ▶ Running ({elapsed:.0f}s)\n", style="dim italic")
+                                display.append(tail.rstrip(), style="dim")
+                                live.update(display)
+                        elif fd == proc.stderr:
+                            line = proc.stderr.readline()
+                            if line:
+                                stderr_lines.append(line)
 
             # Read remaining output after process exits
             if proc.stdout:

{aizen_ai_cli-2.2.2 → aizen_ai_cli-2.2.4/aizen_ai_cli.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aizen-ai-cli
-Version: 2.2.2
+Version: 2.2.4
 Summary: Aizen AI Agent — A professional-grade AI coding assistant for your terminal.
 Author: Irtaza Malik
 License: MIT

{aizen_ai_cli-2.2.2 → aizen_ai_cli-2.2.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "aizen-ai-cli"
-version = "2.2.2"
+version = "2.2.4"
 description = "Aizen AI Agent — A professional-grade AI coding assistant for your terminal."
 readme = "README.md"
 license = {text = "MIT"}

{aizen_ai_cli-2.2.2 → aizen_ai_cli-2.2.4}/setup.py

@@ -8,7 +8,7 @@ def parse_requirements(filename):
 
 setup(
     name="aizen-ai-cli",
-    version="2.2.2",
+    version="2.2.4",
     description="Aizen AI Agent — A professional-grade AI coding assistant for your terminal.",
     packages=["aizen"],
     install_requires=parse_requirements("requirements.txt"),