axe-cli 1.7.8__tar.gz → 1.7.9__tar.gz

{axe_cli-1.7.8 → axe_cli-1.7.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: axe-cli
-Version: 1.7.8
+Version: 1.7.9
 Summary: axe, yerrrr
 Requires-Dist: agent-client-protocol==0.7.0
 Requires-Dist: axe-dig

{axe_cli-1.7.8 → axe_cli-1.7.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "axe-cli"
-version = "1.7.8"
+version = "1.7.9"
 description = "axe, yerrrr"
 readme = "README.md"
 requires-python = ">=3.12"

{axe_cli-1.7.8 → axe_cli-1.7.9}/src/axe_cli/app.py

@@ -108,16 +108,19 @@ class AxeCLI:
 
         model: LLMModel | None = None
         provider: LLMProvider | None = None
+        model_config_name: str | None = None
 
         # try to use config file
         if not model_name and config.default_model:
             # no --model specified && default model is set in config
             model = config.models[config.default_model]
             provider = config.providers[model.provider]
+            model_config_name = config.default_model
         if model_name and model_name in config.models:
             # --model specified && model is set in config
             model = config.models[model_name]
             provider = config.providers[model.provider]
+            model_config_name = model_name
 
         if not model:
             model = LLMModel(provider="", model="", max_context_size=100_000)
@@ -142,7 +145,7 @@ class AxeCLI:
             logger.info("Using LLM model: {model}", model=model)
             logger.info("Thinking mode: {thinking}", thinking=thinking)
 
-        runtime = await Runtime.create(config, llm, session, yolo, skills_dir)
+        runtime = await Runtime.create(config, llm, model_config_name, session, yolo, skills_dir)
 
         if agent_file is None:
             agent_file = DEFAULT_AGENT_FILE
@@ -152,17 +155,19 @@ class AxeCLI:
         await context.restore()
 
         soul = AxeSoul(agent, context=context)
-        return AxeCLI(soul, runtime, env_overrides)
+        return AxeCLI(soul, runtime, env_overrides, model)
 
     def __init__(
         self,
         _soul: AxeSoul,
         _runtime: Runtime,
         _env_overrides: dict[str, str],
+        _model_config: LLMModel | None = None,
     ) -> None:
         self._soul = _soul
         self._runtime = _runtime
         self._env_overrides = _env_overrides
+        self._model_config = _model_config
 
     @property
     def soul(self) -> AxeSoul:
@@ -288,7 +293,7 @@ class AxeCLI:
             welcome_info.append(
                 WelcomeInfoItem(
                     name="Model",
-                    value=model_display_name(self._soul.model_name),
+                    value=model_display_name(self._model_config or self._soul.model_name),
                     level=WelcomeInfoItem.Level.INFO,
                 )
             )

{axe_cli-1.7.8 → axe_cli-1.7.9}/src/axe_cli/config.py

@@ -52,6 +52,8 @@ class LLMModel(BaseModel):
     """Provider name"""
     model: str
     """Model name"""
+    alias: str | None = None
+    """Model alias for nickname switch"""
     max_context_size: int
     """Maximum context size (unit: tokens)"""
     capabilities: set[ModelCapability] | None = None

{axe_cli-1.7.8 → axe_cli-1.7.9}/src/axe_cli/llm.py

@@ -45,10 +45,13 @@ class LLM:
         return self.chat_provider.model_name
 
 
-def model_display_name(model_name: str | None) -> str:
-    if not model_name:
+def model_display_name(model: LLMModel | str | None) -> str:
+    """Get display name for a model, using alias if available."""
+    if model is None:
         return ""
-    return model_name
+    if isinstance(model, str):
+        return model
+    return model.alias or model.model
 
 
 def augment_provider_with_env_vars(provider: LLMProvider, model: LLMModel) -> dict[str, str]:

{axe_cli-1.7.8 → axe_cli-1.7.9}/src/axe_cli/soul/agent.py

@@ -64,6 +64,7 @@ class Runtime:
 
     config: Config
     llm: LLM | None  # we do not freeze the `Runtime` dataclass because LLM can be changed
+    model_config_name: str | None  # name of the model config from config.toml
     session: Session
     builtin_args: BuiltinSystemPromptArgs
     denwa_renji: DenwaRenji
@@ -73,10 +74,10 @@ class Runtime:
     skills: dict[str, Skill]
     oauth: OAuthManager
 
-    @staticmethod
     async def create(
         config: Config,
         llm: LLM | None,
+        model_config_name: str | None,
         session: Session,
         yolo: bool,
         skills_dir: KaosPath | None = None,
@@ -106,6 +107,7 @@ class Runtime:
         return Runtime(
             config=config,
             llm=llm,
+            model_config_name=model_config_name,
             session=session,
             builtin_args=BuiltinSystemPromptArgs(
                 AXE_NOW=datetime.now().astimezone().isoformat(),
@@ -127,6 +129,7 @@ class Runtime:
         return Runtime(
             config=self.config,
             llm=self.llm,
+            model_config_name=self.model_config_name,
             session=self.session,
             builtin_args=self.builtin_args,
             denwa_renji=DenwaRenji(),  # subagent must have its own DenwaRenji
@@ -142,6 +145,7 @@ class Runtime:
         return Runtime(
             config=self.config,
             llm=self.llm,
+            model_config_name=self.model_config_name,
             session=self.session,
             builtin_args=self.builtin_args,
             denwa_renji=DenwaRenji(),  # subagent must have its own DenwaRenji

{axe_cli-1.7.8 → axe_cli-1.7.9}/src/axe_cli/soul/axesoul.py

@@ -133,6 +133,16 @@ class AxeSoul:
     def model_name(self) -> str:
         return self._runtime.llm.chat_provider.model_name if self._runtime.llm else ""
 
+    @property
+    def model_display_name(self) -> str:
+        """Get the display name for the model (uses alias if available)."""
+        if self._runtime.llm is None:
+            return ""
+        if self._runtime.model_config_name and self._runtime.model_config_name in self._runtime.config.models:
+            model_cfg = self._runtime.config.models[self._runtime.model_config_name]
+            return model_cfg.alias or model_cfg.model
+        return self._runtime.llm.chat_provider.model_name
+
     @property
     def model_capabilities(self) -> set[ModelCapability] | None:
         if self._runtime.llm is None:

{axe_cli-1.7.8 → axe_cli-1.7.9}/src/axe_cli/ui/shell/__init__.py

@@ -59,7 +59,7 @@ class Shell:
         else:
             self._start_background_task(self._auto_update())
 
-        _print_welcome_info(self.soul.name or "Axe Code CLI", self._welcome_info)
+        _print_welcome_info(self.soul.name or "axe", self._welcome_info)
 
         if isinstance(self.soul, AxeSoul):
             await replay_recent_history(
@@ -70,7 +70,7 @@ class Shell:
         with CustomPromptSession(
             status_provider=lambda: self.soul.status,
             model_capabilities=self.soul.model_capabilities or set(),
-            model_name=self.soul.model_name,
+            model_name=self.soul.model_display_name,
             thinking=self.soul.thinking or False,
             agent_mode_slash_commands=list(self._available_slash_commands.values()),
             shell_mode_slash_commands=shell_mode_registry.list_commands(),

{axe_cli-1.7.8 → axe_cli-1.7.9}/src/axe_cli/ui/shell/slash.py

@@ -173,7 +173,7 @@ async def model(app: Shell, args: str):
         model_cfg = config.models[name]
         provider_label = model_cfg.provider
         marker = " (current)" if name == curr_model_name else ""
-        label = f"{model_cfg.model} ({provider_label}){marker}"
+        label = f"{model_cfg.alias or model_cfg.model} ({provider_label}){marker}"
         model_choices.append((name, label))
 
     try: