zai-cli 0.1.9__tar.gz → 0.2.2__tar.gz

{zai_cli-0.1.9 → zai_cli-0.2.2}/CHANGELOG.md

@@ -5,6 +5,51 @@ All notable changes to this project are documented in this file.
 The format follows [Keep a Changelog](https://keepachangelog.com/en/1.1.0/),
 and the project uses [Semantic Versioning](https://semver.org/).
 
+## [0.2.2] - 2026-07-13
+
+### Added
+
+- Added structured live-provider validation results with the resolved model,
+  token usage, failure detail, and a recommended next action.
+
+### Changed
+
+- Authentication login validation now distinguishes missing or invalid
+  credentials, quota limits, model configuration errors, network failures, and
+  unexpected provider failures.
+- `zai model test` now reports actionable recovery commands and shows the token
+  cost of a successful validation request when providers return usage data.
+
+## [0.2.1] - 2026-06-30
+
+### Added
+
+- Added `@file` completion in normal interactive prompts.
+- Added safe `@file` mention expansion so prompts like `explain @app.py`
+  include bounded referenced file content.
+
+### Changed
+
+- Mention expansion skips missing, hidden, oversized, binary, and
+  sensitive-looking files without crashing the prompt.
+
+## [0.2.0] - 2026-06-30
+
+### Added
+
+- Added project rule loading from `ZAI.md`, `.zai/rules.md`, `AGENTS.md`,
+  `CLAUDE.md`, and `GEMINI.md`.
+- Added interactive `/rules` and top-level `zai rules` to show loaded project
+  instruction files.
+- Added project-rule visibility to `zai context` and interactive `/context`.
+
+### Changed
+
+- Interactive and one-shot chat now pin loaded project rules into model context
+  so the assistant can follow repo-specific instructions.
+- Project rule loading is bounded, skips unreadable/oversized files, and redacts
+  obvious secrets before rules are added to context.
+
 ## [0.1.9] - 2026-06-30
 
 ### Added

{zai_cli-0.1.9/zai_cli.egg-info → zai_cli-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: zai-cli
-Version: 0.1.9
+Version: 0.2.2
 Summary: Your personal AI CLI — free, fast, and smart
 License-Expression: MIT
 Project-URL: Homepage, https://github.com/HumaizaNaz/zai_cli
@@ -198,6 +198,10 @@ zai auth login openai
 zai auth login anthropic
 ```
 
+When prompted, authentication login can send one tiny live request to validate
+the key. The result identifies the resolved model and gives a recovery command
+for invalid credentials, quota limits, model errors, or network failures.
+
 Remove a saved API key from `zai`:
 
 ```bash
@@ -343,10 +347,49 @@ When `zai` starts in a folder that already has `.zai/project_memory.json`, it
 loads a short pinned summary so future answers can use that folder's context.
 Different folders keep different memories.
 
+## Project rules
+
+Add project instruction files when you want `zai` to consistently follow repo
+rules without copy-pasting them into every prompt.
+
+Supported files:
+
+```text
+ZAI.md
+.zai/rules.md
+AGENTS.md
+CLAUDE.md
+GEMINI.md
+```
+
+When `zai` starts, it loads bounded text from these files in the project root
+and current folder, redacts obvious secrets, and pins the rules into model
+context. Use these files for coding style, preferred test commands, folders to
+avoid, release steps, and project-specific constraints.
+
+Inspect loaded rules:
+
+```text
+zai rules
+/rules
+zai context
+/context
+```
+
+Project rules are different from project memory:
+
+- Project rules are explicit instructions written by the user/repo.
+- Project memory is a generated or remembered summary in
+  `.zai/project_memory.json`.
+- Repo maps are generated structure summaries used on demand.
+- Session history is the current or saved conversation.
+
 Memory types:
 
 - Session history: previous conversation messages saved globally under `~/.zai`.
 - Project memory: small per-folder facts in `.zai/project_memory.json`.
+- Project rules: explicit instruction files such as `ZAI.md` or
+  `.zai/rules.md`.
 - Repo map: generated structure summary used for repository questions; it is not
   long-term memory.
 
@@ -387,12 +430,13 @@ zai model add <name>        Add a custom model alias
 zai model remove <name>     Remove a custom model alias
 zai model configure <name>  Set timeout and retry policy
 zai model info <name>       Show effective model configuration
-zai model test <name>       Verify credentials and model availability
+zai model test <name>       Verify credentials/model and show recovery guidance
 zai auth status             Show API-key and external CLI auth status
 zai auth login <provider>   Connect a provider API key
 zai auth logout <provider>  Remove a provider API key from zai
 zai auth doctor             Show auth troubleshooting guidance
 zai context                 Show loaded project context status
+zai rules                   Show loaded project instruction files
 zai memory show             Show last task metadata
 zai memory projects         List remembered projects
 zai memory project          Show project-scoped memory
@@ -482,6 +526,7 @@ Inside `zai` interactive mode:
 /model <name>               Change model for this interactive session
 /providers                  Show provider connection status
 /context                    Show loaded project context status
+/rules                      Show loaded project instruction files
 /memory                     Show last task metadata
 /commands                   List built-in and plugin slash commands
 ```
@@ -497,6 +542,18 @@ Interactive input also provides:
 - `Ctrl+C` to cancel the current input or operation without ending the session;
 - a basic Rich prompt fallback in redirected or unsupported terminals.
 
+You can reference files in natural prompts with `@file`:
+
+```text
+explain @app.py
+fix @src/main.py
+compare @README.md and @CHANGELOG.md
+```
+
+`zai` loads bounded text from valid referenced files and appends it to the
+prompt context. Hidden files, sensitive-looking paths, binary files, and very
+large files are skipped with a note instead of crashing the session.
+
 ## How much of my project does zai see?
 
 `zai` starts from the folder where you run it. It does not load every file in a

{zai_cli-0.1.9 → zai_cli-0.2.2}/README.md

@@ -141,6 +141,10 @@ zai auth login openai
 zai auth login anthropic
 ```
 
+When prompted, authentication login can send one tiny live request to validate
+the key. The result identifies the resolved model and gives a recovery command
+for invalid credentials, quota limits, model errors, or network failures.
+
 Remove a saved API key from `zai`:
 
 ```bash
@@ -286,10 +290,49 @@ When `zai` starts in a folder that already has `.zai/project_memory.json`, it
 loads a short pinned summary so future answers can use that folder's context.
 Different folders keep different memories.
 
+## Project rules
+
+Add project instruction files when you want `zai` to consistently follow repo
+rules without copy-pasting them into every prompt.
+
+Supported files:
+
+```text
+ZAI.md
+.zai/rules.md
+AGENTS.md
+CLAUDE.md
+GEMINI.md
+```
+
+When `zai` starts, it loads bounded text from these files in the project root
+and current folder, redacts obvious secrets, and pins the rules into model
+context. Use these files for coding style, preferred test commands, folders to
+avoid, release steps, and project-specific constraints.
+
+Inspect loaded rules:
+
+```text
+zai rules
+/rules
+zai context
+/context
+```
+
+Project rules are different from project memory:
+
+- Project rules are explicit instructions written by the user/repo.
+- Project memory is a generated or remembered summary in
+  `.zai/project_memory.json`.
+- Repo maps are generated structure summaries used on demand.
+- Session history is the current or saved conversation.
+
 Memory types:
 
 - Session history: previous conversation messages saved globally under `~/.zai`.
 - Project memory: small per-folder facts in `.zai/project_memory.json`.
+- Project rules: explicit instruction files such as `ZAI.md` or
+  `.zai/rules.md`.
 - Repo map: generated structure summary used for repository questions; it is not
   long-term memory.
 
@@ -330,12 +373,13 @@ zai model add <name>        Add a custom model alias
 zai model remove <name>     Remove a custom model alias
 zai model configure <name>  Set timeout and retry policy
 zai model info <name>       Show effective model configuration
-zai model test <name>       Verify credentials and model availability
+zai model test <name>       Verify credentials/model and show recovery guidance
 zai auth status             Show API-key and external CLI auth status
 zai auth login <provider>   Connect a provider API key
 zai auth logout <provider>  Remove a provider API key from zai
 zai auth doctor             Show auth troubleshooting guidance
 zai context                 Show loaded project context status
+zai rules                   Show loaded project instruction files
 zai memory show             Show last task metadata
 zai memory projects         List remembered projects
 zai memory project          Show project-scoped memory
@@ -425,6 +469,7 @@ Inside `zai` interactive mode:
 /model <name>               Change model for this interactive session
 /providers                  Show provider connection status
 /context                    Show loaded project context status
+/rules                      Show loaded project instruction files
 /memory                     Show last task metadata
 /commands                   List built-in and plugin slash commands
 ```
@@ -440,6 +485,18 @@ Interactive input also provides:
 - `Ctrl+C` to cancel the current input or operation without ending the session;
 - a basic Rich prompt fallback in redirected or unsupported terminals.
 
+You can reference files in natural prompts with `@file`:
+
+```text
+explain @app.py
+fix @src/main.py
+compare @README.md and @CHANGELOG.md
+```
+
+`zai` loads bounded text from valid referenced files and appends it to the
+prompt context. Hidden files, sensitive-looking paths, binary files, and very
+large files are skipped with a note instead of crashing the session.
+
 ## How much of my project does zai see?
 
 `zai` starts from the folder where you run it. It does not load every file in a

{zai_cli-0.1.9 → zai_cli-0.2.2}/tests/test_auth.py

@@ -14,6 +14,7 @@ from zai.core.auth import (
     mask_secret,
     save_provider_key,
     validate_provider,
+    validate_provider_details,
 )
 
 
@@ -66,18 +67,58 @@ def test_external_login_command_for_codex_and_claude():
 def test_validate_provider_normalizes_results():
     provider = MagicMock()
     provider.is_available.return_value = True
-    provider.chat.return_value = MagicMock()
+    provider.chat.return_value = MagicMock(model="provider/model", tokens_used=7)
     with patch("zai.core.auth.get_provider", return_value=provider):
         assert validate_provider("groq") == "OK"
+        details = validate_provider_details("groq")
+        assert details.status == "OK"
+        assert details.model == "provider/model"
+        assert details.tokens_used == 7
+        assert "zai model set groq" in details.next_action
 
     provider.chat.side_effect = RuntimeError("unauthorized api key")
     with patch("zai.core.auth.get_provider", return_value=provider):
         assert validate_provider("groq") == "invalid"
+        details = validate_provider_details("groq")
+        assert details.status == "invalid"
+        assert "zai auth login groq" in details.next_action
 
     provider.chat.side_effect = RuntimeError("rate limit")
     with patch("zai.core.auth.get_provider", return_value=provider):
         assert validate_provider("groq") == "quota/rate limit"
 
+    provider.chat.side_effect = RuntimeError("unknown model does not exist")
+    with patch("zai.core.auth.get_provider", return_value=provider):
+        details = validate_provider_details("groq")
+        assert details.status == "model error"
+        assert "zai model info groq" in details.next_action
+
+    provider.chat.side_effect = RuntimeError("unexpected provider failure")
+    with patch("zai.core.auth.get_provider", return_value=provider):
+        details = validate_provider_details("groq")
+        assert details.status == "provider error"
+        assert "zai auth doctor" in details.next_action
+
+    provider.chat.side_effect = RuntimeError("api_key=super-secret-provider-token")
+    with patch("zai.core.auth.get_provider", return_value=provider):
+        details = validate_provider_details("groq")
+        assert "super-secret-provider-token" not in details.detail
+        assert "[REDACTED_SECRET]" in details.detail
+
+
+def test_validate_provider_reports_unknown_and_unavailable():
+    unknown = validate_provider_details("unknown")
+    assert unknown.status == "missing"
+    assert unknown.detail == "Unknown provider."
+
+    provider = MagicMock()
+    provider.is_available.return_value = False
+    with patch("zai.core.auth.get_provider", return_value=provider):
+        unavailable = validate_provider_details("groq")
+
+    assert unavailable.status == "missing"
+    assert "zai auth login groq" in unavailable.next_action
+
 
 def test_top_level_auth_status_does_not_print_secret(monkeypatch):
     monkeypatch.setenv("OPENAI_API_KEY", "sk-live-secret")

zai_cli-0.2.2/tests/test_file_mentions.py

@@ -0,0 +1,30 @@
+from zai.core.file_mentions import expand_file_mentions, extract_file_mentions
+
+
+def test_extract_file_mentions_keeps_order_and_uniqueness():
+    text = "explain @app.py then compare @README.md and @app.py."
+
+    assert extract_file_mentions(text) == ["app.py", "README.md"]
+
+
+def test_expand_file_mentions_loads_bounded_file_context(tmp_path):
+    (tmp_path / "app.py").write_text("print('hi')\n", encoding="utf-8")
+
+    expanded = expand_file_mentions("explain @app.py", tmp_path)
+
+    assert "explain @app.py" in expanded
+    assert "Referenced @file context" in expanded
+    assert "--- @app.py ---" in expanded
+    assert "print('hi')" in expanded
+
+
+def test_expand_file_mentions_skips_hidden_and_sensitive_files(tmp_path):
+    (tmp_path / ".env").write_text("API_KEY=secret\n", encoding="utf-8")
+    (tmp_path / "token.txt").write_text("secret\n", encoding="utf-8")
+
+    expanded = expand_file_mentions("read @.env and @token.txt", tmp_path)
+
+    assert "API_KEY=secret" not in expanded
+    assert "Mention notes" in expanded
+    assert "Skipped @.env" in expanded
+    assert "Skipped @token.txt" in expanded

{zai_cli-0.1.9 → zai_cli-0.2.2}/tests/test_input.py

@@ -97,6 +97,22 @@ def test_path_completion(tmp_path):
     assert not any(".secret" in candidate[0] for candidate in candidates)
 
 
+def test_file_mention_completion_in_natural_prompt(tmp_path):
+    (tmp_path / "src").mkdir()
+    (tmp_path / "sample.py").write_text("")
+    (tmp_path / ".secret").write_text("")
+
+    candidates = completion_candidates(
+        "explain @s",
+        str(tmp_path),
+        {"fix"},
+    )
+
+    assert ("@src/", "@src/", "file mention") in candidates
+    assert ("@sample.py", "@sample.py", "file mention") in candidates
+    assert not any(".secret" in candidate[0] for candidate in candidates)
+
+
 def test_prompt_falls_back_when_prompt_toolkit_missing(tmp_path):
     real_import = __import__
 

{zai_cli-0.1.9 → zai_cli-0.2.2}/tests/test_interactive.py

@@ -182,21 +182,22 @@ def test_interactive_project_memory_commands(tmp_path, monkeypatch, capsys):
     (tmp_path / "pyproject.toml").write_text("[project]\nname='demo'\n")
     with patch("zai.cli.interactive.has_available_provider", return_value=True):
         with patch("zai.cli.interactive.fire_hook", return_value=True):
-            with patch("zai.cli.interactive.plugin_loader.load_all", return_value={}):
-                with patch("zai.cli.interactive.plugin_loader.get_errors", return_value={}):
-                    with patch("zai.cli.interactive._connect_mcp_servers"):
-                        with patch(
-                            "zai.cli.interactive.InteractiveInput.prompt",
-                            side_effect=[
-                                "/memory project",
-                                "/memory refresh",
-                                "/remember uses pytest",
-                                "/forget 1",
-                                EOFError(),
-                            ],
-                        ):
-                            with patch("zai.cli.interactive.run_agent") as run_agent:
-                                run_interactive("groq")
+            with patch("zai.cli.interactive.add_project"):
+                with patch("zai.cli.interactive.plugin_loader.load_all", return_value={}):
+                    with patch("zai.cli.interactive.plugin_loader.get_errors", return_value={}):
+                        with patch("zai.cli.interactive._connect_mcp_servers"):
+                            with patch(
+                                "zai.cli.interactive.InteractiveInput.prompt",
+                                side_effect=[
+                                    "/memory project",
+                                    "/memory refresh",
+                                    "/remember uses pytest",
+                                    "/forget 1",
+                                    EOFError(),
+                                ],
+                            ):
+                                with patch("zai.cli.interactive.run_agent") as run_agent:
+                                    run_interactive("groq")
 
     output = capsys.readouterr().out
     assert "Project Memory" in output
@@ -210,26 +211,27 @@ def test_interactive_auth_commands_do_not_call_ai(tmp_path, monkeypatch, capsys)
     monkeypatch.chdir(tmp_path)
     with patch("zai.cli.interactive.has_available_provider", return_value=True):
         with patch("zai.cli.interactive.fire_hook", return_value=True):
-            with patch("zai.cli.interactive.plugin_loader.load_all", return_value={}):
-                with patch("zai.cli.interactive.plugin_loader.get_errors", return_value={}):
-                    with patch("zai.cli.interactive._connect_mcp_servers"):
-                        with patch(
-                            "zai.cli.interactive.InteractiveInput.prompt",
-                            side_effect=[
-                                "/auth status",
-                                "/auth login gemini",
-                                "/auth logout gemini",
-                                EOFError(),
-                            ],
-                        ):
-                            with (
-                                patch("zai.cli.interactive.Prompt.ask", return_value="gemini-key"),
-                                patch("zai.cli.interactive.Confirm.ask", return_value=False),
-                                patch("zai.cli.interactive.save_provider_key") as save,
-                                patch("zai.cli.interactive.delete_provider_key", return_value=True) as delete,
-                                patch("zai.cli.interactive.run_agent") as run_agent,
+            with patch("zai.cli.interactive.add_project"):
+                with patch("zai.cli.interactive.plugin_loader.load_all", return_value={}):
+                    with patch("zai.cli.interactive.plugin_loader.get_errors", return_value={}):
+                        with patch("zai.cli.interactive._connect_mcp_servers"):
+                            with patch(
+                                "zai.cli.interactive.InteractiveInput.prompt",
+                                side_effect=[
+                                    "/auth status",
+                                    "/auth login gemini",
+                                    "/auth logout gemini",
+                                    EOFError(),
+                                ],
                             ):
-                                run_interactive("groq")
+                                with (
+                                    patch("zai.cli.interactive.Prompt.ask", return_value="gemini-key"),
+                                    patch("zai.cli.interactive.Confirm.ask", return_value=False),
+                                    patch("zai.cli.interactive.save_provider_key") as save,
+                                    patch("zai.cli.interactive.delete_provider_key", return_value=True) as delete,
+                                    patch("zai.cli.interactive.run_agent") as run_agent,
+                                ):
+                                    run_interactive("groq")
 
     output = capsys.readouterr().out
     assert "Provider status" in output
@@ -242,6 +244,7 @@ def test_interactive_auth_commands_do_not_call_ai(tmp_path, monkeypatch, capsys)
 def test_interactive_context_and_providers_do_not_call_ai(tmp_path, monkeypatch, capsys):
     monkeypatch.chdir(tmp_path)
     (tmp_path / "app.py").write_text("print('hi')\n")
+    (tmp_path / "ZAI.md").write_text("Use pytest.\n")
     with patch("zai.cli.interactive.has_available_provider", return_value=True):
         with patch("zai.cli.interactive.fire_hook", return_value=True):
             with patch("zai.cli.interactive.plugin_loader.load_all", return_value={}):
@@ -252,6 +255,7 @@ def test_interactive_context_and_providers_do_not_call_ai(tmp_path, monkeypatch,
                             side_effect=[
                                 "/providers",
                                 "/context",
+                                "/rules",
                                 "/context tokens",
                                 "/context summary",
                                 EOFError(),
@@ -263,6 +267,9 @@ def test_interactive_context_and_providers_do_not_call_ai(tmp_path, monkeypatch,
     output = capsys.readouterr().out
     assert "Provider status" in output
     assert "Project context" in output
+    assert "Project rules loaded:" in output
+    assert "ZAI.md" in output
+    assert "Project Rules" in output
     assert "unread files are not loaded" in output
     assert "Estimated tokens used" in output
     assert "No compacted summary" in output
@@ -270,6 +277,60 @@ def test_interactive_context_and_providers_do_not_call_ai(tmp_path, monkeypatch,
     run_agent.assert_not_called()
 
 
+def test_interactive_pins_project_rules_before_agent_call(tmp_path, monkeypatch):
+    monkeypatch.chdir(tmp_path)
+    (tmp_path / "ZAI.md").write_text("Use the custom test command.\n")
+    with patch("zai.cli.interactive.has_available_provider", return_value=True):
+        with patch("zai.cli.interactive.fire_hook", return_value=True):
+            with patch("zai.cli.interactive.plugin_loader.load_all", return_value={}):
+                with patch("zai.cli.interactive.plugin_loader.get_errors", return_value={}):
+                    with patch("zai.cli.interactive._connect_mcp_servers"):
+                        with patch(
+                            "zai.cli.interactive.InteractiveInput.prompt",
+                            side_effect=["do work", EOFError()],
+                        ):
+                            with (
+                                patch(
+                                    "zai.cli.interactive.run_agent",
+                                    return_value="done",
+                                ) as run_agent,
+                                patch("zai.cli.interactive.Confirm.ask", return_value=False),
+                            ):
+                                run_interactive("groq")
+
+    history = run_agent.call_args.args[1]
+    assert any(
+        message.pinned and "Use the custom test command" in message.content
+        for message in history
+    )
+
+
+def test_interactive_expands_file_mentions_before_agent_call(tmp_path, monkeypatch):
+    monkeypatch.chdir(tmp_path)
+    (tmp_path / "app.py").write_text("print('hello')\n")
+    with patch("zai.cli.interactive.has_available_provider", return_value=True):
+        with patch("zai.cli.interactive.fire_hook", return_value=True):
+            with patch("zai.cli.interactive.plugin_loader.load_all", return_value={}):
+                with patch("zai.cli.interactive.plugin_loader.get_errors", return_value={}):
+                    with patch("zai.cli.interactive._connect_mcp_servers"):
+                        with patch(
+                            "zai.cli.interactive.InteractiveInput.prompt",
+                            side_effect=["explain @app.py", EOFError()],
+                        ):
+                            with (
+                                patch(
+                                    "zai.cli.interactive.run_agent",
+                                    return_value="done",
+                                ) as run_agent,
+                                patch("zai.cli.interactive.Confirm.ask", return_value=False),
+                            ):
+                                run_interactive("groq")
+
+    prompt = run_agent.call_args.args[0]
+    assert "Referenced @file context" in prompt
+    assert "print('hello')" in prompt
+
+
 def test_interactive_manual_compact_summarizes_history(tmp_path, monkeypatch, capsys):
     monkeypatch.chdir(tmp_path)
     prompts = [

zai_cli-0.2.2/tests/test_project_rules.py

@@ -0,0 +1,33 @@
+from zai.core.project_rules import (
+    format_project_rules,
+    load_project_rules,
+    project_rules_prompt,
+)
+
+
+def test_load_project_rules_from_root_and_current_folder(tmp_path):
+    (tmp_path / ".git").mkdir()
+    (tmp_path / "ZAI.md").write_text("Use pytest.\n", encoding="utf-8")
+    nested = tmp_path / "pkg"
+    nested.mkdir()
+    (nested / "AGENTS.md").write_text("Nested rule.\n", encoding="utf-8")
+
+    rules = load_project_rules(nested)
+
+    assert [rule.relative_path for rule in rules] == ["ZAI.md", "pkg/AGENTS.md"]
+    assert "Use pytest" in project_rules_prompt(rules)
+    assert "pkg/AGENTS.md" in format_project_rules(rules)
+
+
+def test_project_rules_redact_secrets_and_skip_large_files(tmp_path):
+    (tmp_path / "ZAI.md").write_text(
+        "api_key = sk-secret-value-1234567890\nFollow style.\n",
+        encoding="utf-8",
+    )
+    (tmp_path / "CLAUDE.md").write_text("x" * (70 * 1024), encoding="utf-8")
+
+    rules = load_project_rules(tmp_path)
+
+    assert len(rules) == 1
+    assert "sk-secret" not in rules[0].content
+    assert "[REDACTED_SECRET]" in rules[0].content

{zai_cli-0.1.9 → zai_cli-0.2.2}/tests/test_settings_cli.py

@@ -141,18 +141,35 @@ def test_context_command_is_honest_about_unloaded_files(tmp_path, monkeypatch):
     app = settings_app()
     monkeypatch.chdir(tmp_path)
     (tmp_path / "app.py").write_text("print('hi')\n")
+    (tmp_path / "ZAI.md").write_text("Use pytest.\n")
 
     result = CliRunner().invoke(app, ["context"])
 
     assert result.exit_code == 0
     assert f"Folder: {tmp_path}" in result.output
     assert "Project memory: missing" in result.output
+    assert "Project rules:" in result.output
+    assert "ZAI.md" in result.output
     assert "Repo map/index: not loaded yet" in result.output
     assert "unread files are not loaded" in result.output
     assert "Estimated tokens used" in result.output
     assert "app.py is loaded" not in result.output
 
 
+def test_rules_command_shows_loaded_project_rules(tmp_path, monkeypatch):
+    app = settings_app()
+    monkeypatch.chdir(tmp_path)
+    (tmp_path / ".git").mkdir()
+    (tmp_path / ".zai").mkdir()
+    (tmp_path / ".zai" / "rules.md").write_text("Never commit secrets.\n")
+
+    result = CliRunner().invoke(app, ["rules"])
+
+    assert result.exit_code == 0
+    assert "Project Rules" in result.output
+    assert ".zai/rules.md" in result.output
+
+
 def test_context_tokens_and_summary_commands(tmp_path, monkeypatch):
     app = settings_app()
     monkeypatch.chdir(tmp_path)
@@ -255,6 +272,25 @@ def test_model_test_passes_and_reports_unavailable():
     assert "Model unavailable" in unavailable.output
 
 
+def test_model_test_reports_actionable_auth_failure():
+    from zai.core.errors import AuthenticationError
+
+    app = settings_app()
+    runner = CliRunner()
+    provider = MagicMock()
+    provider.name = "groq"
+    provider.is_available.return_value = True
+    provider.chat.side_effect = AuthenticationError("groq", "bad key")
+
+    with patch("zai.core.fallback.get_provider", return_value=provider):
+        result = runner.invoke(app, ["model", "test", "groq"])
+
+    assert result.exit_code == 0
+    assert "invalid credentials" in result.output
+    assert "zai auth login groq" in result.output
+    assert "bad key" in result.output
+
+
 def test_model_configure_updates_builtin_runtime_settings():
     app = settings_app()
     config = {

zai_cli-0.2.2/zai/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.2.2"