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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langstage-cli
-Version: 0.6.44
+Version: 0.6.45
 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.45}/langstage_cli/cli.py

@@ -1140,20 +1140,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 +1506,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:

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langstage-cli
-Version: 0.6.44
+Version: 0.6.45
 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.45}/langstage_cli.egg-info/SOURCES.txt

@@ -59,6 +59,7 @@ 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

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

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "langstage-cli"
-version = "0.6.44"
+version = "0.6.45"
 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.45/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