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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aizen-ai-cli
-Version: 2.2.2
+Version: 2.2.3
 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.3}/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.3"
 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:
@@ -329,7 +329,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.3}/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.3/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.3
 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.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "aizen-ai-cli"
-version = "2.2.2"
+version = "2.2.3"
 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.3}/setup.py

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