langstage-cli 0.6.44__tar.gz → 0.6.46__tar.gz

{langstage_cli-0.6.44/langstage_cli.egg-info → langstage_cli-0.6.46}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langstage-cli
-Version: 0.6.44
+Version: 0.6.46
 Summary: The terminal stage for your LangGraph agent — Claude Code-style CLI for any CompiledGraph
 Author-email: Kedar Dabhadkar <kdabhadk@gmail.com>
 License-Expression: MIT

{langstage_cli-0.6.44 → langstage_cli-0.6.46}/langstage_cli/cli.py

@@ -1001,6 +1001,15 @@ def _default_agent_path() -> Optional[Path]:
     return path if path.exists() else None
 
 
+# A MESSAGE that is really an agent spec: ``my_agent.py:graph`` or ``pkg.module:agent``.
+_SPEC_LIKE = re.compile(r"^(?:(?:[A-Za-z]:)?[^\s:]+\.py|[A-Za-z_][\w.]*):[A-Za-z_]\w*$")
+
+
+def _looks_like_agent_spec(message: Optional[str]) -> bool:
+    """Whether a positional MESSAGE looks like an agent spec meant for ``-a`` (gh #186)."""
+    return bool(message) and bool(_SPEC_LIKE.match(message.strip()))
+
+
 def _no_agent_toml_hint(cfg: Any, toml_sources: list) -> List[str]:
     """Lines explaining why a ``langstage.toml`` that was read supplied no spec (gh #148).
 
@@ -1140,20 +1149,31 @@ def _absolutize_file_spec(spec: str) -> str:
     return f"{Path(path_part).expanduser().resolve()}{suffix}"
 
 
-def get_tool_arg_preview(args: Dict[str, Any]) -> str:
-    """Get a preview of the first argument value (nanocode style).
+# Per-value cap on the tool-call card (default output; ``-v`` shows values in full).
+_TOOL_ARG_VALUE_MAX = 50
+
 
-    The value is escaped before it is truncated, so a tool arg can't move the cursor
-    or retitle the terminal, and a cut can't leave half an escape sequence live (gh #180).
+def get_tool_arg_preview(args: Dict[str, Any], verbose: bool = False) -> str:
+    """The arguments a tool was called with, for the tool-call card.
+
+    A lone argument shows as its bare value (``read_file └─ docs/a.md``). Several show
+    as ``name=value`` pairs, every one of them: showing only the first misstated the
+    call, ``add(a=17, b=25)`` read as "add 17" (gh #184). Each value is capped at 50
+    characters unless ``verbose``. Values are escaped before they are truncated, so a
+    tool arg can't move the cursor or retitle the terminal, and a cut can't leave half
+    an escape sequence live (gh #180).
     """
     if not args:
         return ""
-    # Get first value
-    first_val = _escape_controls(str(list(args.values())[0]))
-    # Truncate if needed
-    if len(first_val) > 50:
-        return first_val[:50] + "..."
-    return first_val
+    limit = None if verbose else _TOOL_ARG_VALUE_MAX
+    if not isinstance(args, dict):
+        return _approval_text(args, limit)
+    if len(args) == 1:
+        (value,) = args.values()
+        return _approval_text(str(value), limit)
+    return ", ".join(
+        f"{_approval_text(str(k), None)}={_approval_text(v, limit)}" for k, v in args.items()
+    )
 
 
 # Per-value cap for the HITL approval preview. Values are truncated, never dropped: the
@@ -1495,7 +1515,7 @@ def print_chunk(chunk: Dict[str, Any], verbose: bool = False):
             for tool_call in chunk["tool_calls"]:
                 tool_name = _escape_controls(str(tool_call["name"]))  # gh #180
                 args = tool_call.get("args", {})
-                arg_preview = get_tool_arg_preview(args)
+                arg_preview = get_tool_arg_preview(args, verbose)
 
                 safe_print(f"\n{GREEN}● {tool_name}{RESET}")
                 if arg_preview:
@@ -3457,10 +3477,18 @@ def main(
                 # of only pointing at the env var (gh #148).
                 for line in _no_agent_toml_hint(cfg, toml_sources):
                     _status(f"{YELLOW}{line}{RESET}")
+                # The spec went in as the MESSAGE: name the command that works (gh #186).
+                if _looks_like_agent_spec(message):
+                    _status(
+                        f"{YELLOW}The agent spec goes after -a: langstage-cli -a {message.strip()}{RESET}"
+                    )
                 _status(f"\n{DIM}Usage:{RESET}")
-                _status("  langstage-cli path/to/agent.py:graph")
-                _status("  langstage-cli mypackage.module:agent")
-                _status(f"\n{DIM}Or set the LANGSTAGE_AGENT_SPEC environment variable{RESET}")
+                _status("  langstage-cli -a path/to/agent.py:graph")
+                _status("  langstage-cli -a mypackage.module:agent")
+                _status(
+                    f"\n{DIM}Or set LANGSTAGE_AGENT_SPEC, put the spec under [agent] spec in "
+                    f"langstage.toml, or run `langstage-cli init` for a starter agent{RESET}"
+                )
                 sys.exit(1)
 
         # The BASE directory a relative spec resolves against depends on WHERE it came

{langstage_cli-0.6.44 → langstage_cli-0.6.46/langstage_cli.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langstage-cli
-Version: 0.6.44
+Version: 0.6.46
 Summary: The terminal stage for your LangGraph agent — Claude Code-style CLI for any CompiledGraph
 Author-email: Kedar Dabhadkar <kdabhadk@gmail.com>
 License-Expression: MIT

{langstage_cli-0.6.44 → langstage_cli-0.6.46}/langstage_cli.egg-info/SOURCES.txt

@@ -59,10 +59,12 @@ tests/test_stream_mode.py
 tests/test_streaming_marker.py
 tests/test_terminal_width.py
 tests/test_toml_relative_paths.py
+tests/test_tool_card_args.py
 tests/test_tool_output_escape.py
 tests/test_turn_error_verbose.py
 tests/test_turn_exit_and_render.py
 tests/test_unicode_console.py
+tests/test_usage_hint_186.py
 tests/test_verbose_command.py
 tests/test_verify_flag.py
 tests/test_workspace.py

{langstage_cli-0.6.44 → langstage_cli-0.6.46}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "langstage-cli"
-version = "0.6.44"
+version = "0.6.46"
 description = "The terminal stage for your LangGraph agent — Claude Code-style CLI for any CompiledGraph"
 readme = "README.md"
 requires-python = ">=3.11"

langstage_cli-0.6.46/tests/test_tool_card_args.py

@@ -0,0 +1,100 @@
+"""The tool-call card shows every argument, by name (gh #184).
+
+It used to show only the first argument's value, so ``add(a=17, b=25)`` rendered as
+``add └─ 17`` and ``transfer_funds(to, amount, currency)`` as ``└─ alice``, in default
+and ``-v`` output alike. Under ``--no-interactive`` the card is the only record of
+what ran.
+"""
+
+import io
+import re
+from contextlib import redirect_stdout
+
+import pytest
+
+pytest.importorskip("ag_ui_langgraph")
+
+from langchain_core.messages import AIMessage, ToolMessage  # noqa: E402
+from langgraph.graph import END, START, StateGraph  # noqa: E402
+from langgraph.graph.message import MessagesState  # noqa: E402
+
+from langstage_cli import cli  # noqa: E402
+from langstage_cli.agui_stream import build_session_agent  # noqa: E402
+from langstage_cli.cli import get_tool_arg_preview, run_single_turn_agui  # noqa: E402
+
+_SGR = re.compile(r"\x1b\[[0-9;]*m")
+TRANSFER = {"to": "alice", "amount": 1000000, "currency": "USD"}
+
+
+def test_every_argument_is_named():
+    assert get_tool_arg_preview({"a": 17, "b": 25}) == "a=17, b=25"
+    assert get_tool_arg_preview(TRANSFER) == "to=alice, amount=1000000, currency=USD"
+
+
+def test_a_single_argument_stays_a_bare_value():
+    assert get_tool_arg_preview({"path": "docs/a.md"}) == "docs/a.md"
+    assert get_tool_arg_preview({}) == ""
+
+
+def test_values_are_capped_but_never_dropped():
+    long = "x" * 200
+    preview = get_tool_arg_preview({"path": "a.md", "content": long, "mode": "w"})
+    assert preview == f"path=a.md, content={'x' * 50}..., mode=w"
+    # Structured values read as JSON, not Python reprs.
+    assert (
+        get_tool_arg_preview({"ids": [1, 2], "opts": {"k": "v"}}) == 'ids=[1, 2], opts={"k": "v"}'
+    )
+
+
+def test_verbose_shows_values_in_full():
+    long = "y" * 200
+    assert (
+        get_tool_arg_preview({"path": "a.md", "content": long}, verbose=True)
+        == f"path=a.md, content={long}"
+    )
+    assert get_tool_arg_preview({"q": long}, verbose=True) == long
+
+
+def test_names_and_values_are_escaped():
+    preview = get_tool_arg_preview({"p\x1b[2K": "a\x1b]0;t\x07", "b": "c\nd"})
+    assert "\x1b" not in preview and "\x07" not in preview and "\n" not in preview
+    assert preview == r"p\x1b[2K=a\x1b]0;t\x07, b=c\nd"
+
+
+def _transfer_graph():
+    def call(state):
+        tc = {"id": "t1", "name": "transfer_funds", "args": TRANSFER}
+        return {"messages": [AIMessage(content="", tool_calls=[tc])]}
+
+    def tool(state):
+        return {"messages": [ToolMessage(content="ok", tool_call_id="t1")]}
+
+    def reply(state):
+        return {"messages": [AIMessage(content="Transferred.")]}
+
+    g = StateGraph(MessagesState)
+    for n, f in [("call", call), ("tool", tool), ("reply", reply)]:
+        g.add_node(n, f)
+    g.add_edge(START, "call")
+    g.add_edge("call", "tool")
+    g.add_edge("tool", "reply")
+    g.add_edge("reply", END)
+    return g.compile()
+
+
+@pytest.mark.parametrize("verbose", [False, True])
+async def test_card_in_a_real_turn(monkeypatch, verbose):
+    """The issue's keyless repro, through the AG-UI stream the CLI renders."""
+    monkeypatch.setattr(cli, "Spinner", lambda *a, **k: None)
+    buf = io.StringIO()
+    with redirect_stdout(buf):
+        await run_single_turn_agui(
+            build_session_agent(_transfer_graph()),
+            "pay alice",
+            "t-184",
+            interactive=False,
+            verbose=verbose,
+        )
+    plain = _SGR.sub("", buf.getvalue())
+    assert "● transfer_funds" in plain, plain
+    assert "└─ to=alice, amount=1000000, currency=USD" in plain, plain

langstage_cli-0.6.46/tests/test_usage_hint_186.py

@@ -0,0 +1,52 @@
+"""gh #186: the "No agent specified" hint shows commands that work (with -a)."""
+
+import pytest
+from click.testing import CliRunner
+
+from langstage_cli import cli
+from langstage_cli.cli import _looks_like_agent_spec, main
+
+
+@pytest.fixture
+def no_agent(tmp_path, monkeypatch):
+    monkeypatch.chdir(tmp_path)
+    for var in ("LANGSTAGE_AGENT_SPEC", "DEEPAGENT_AGENT_SPEC", "DEEPAGENT_SPEC"):
+        monkeypatch.delenv(var, raising=False)
+    monkeypatch.setattr(cli, "_default_agent_path", lambda: None)
+
+
+def test_usage_hint_uses_dash_a(no_agent):
+    r = CliRunner().invoke(main, ["hi"])
+    assert r.exit_code == 1, r.output
+    assert "No agent specified" in r.output
+    assert "langstage-cli -a path/to/agent.py:graph" in r.output
+    assert "langstage-cli -a mypackage.module:agent" in r.output
+    assert "[agent] spec" in r.output and "langstage-cli init" in r.output
+    # Not a spec-looking message: no "goes after -a" line.
+    assert "goes after -a" not in r.output
+
+
+@pytest.mark.parametrize("spec", ["my_agent.py:graph", "mypackage.module:agent"])
+def test_spec_passed_as_message_names_the_working_command(no_agent, spec):
+    r = CliRunner().invoke(main, [spec])
+    assert r.exit_code == 1, r.output
+    assert f"The agent spec goes after -a: langstage-cli -a {spec}" in r.output
+
+
+@pytest.mark.parametrize(
+    "message,expected",
+    [
+        ("my_agent.py:graph", True),
+        ("./agents/a.py:graph", True),
+        (r"C:\work\agent.py:graph", True),
+        ("pkg.module:agent", True),
+        ("hello", False),
+        ("what is 2:3", False),
+        ("ratio 3:1 please", False),
+        ("http://example.org:80", False),
+        ("", False),
+        (None, False),
+    ],
+)
+def test_looks_like_agent_spec(message, expected):
+    assert _looks_like_agent_spec(message) is expected