padwan-cli 0.3.0__tar.gz → 0.4.0__tar.gz

{padwan_cli-0.3.0 → padwan_cli-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: padwan-cli
-Version: 0.3.0
+Version: 0.4.0
 Summary: CLI and TUI for padwan-llm
 Author-email: julien.brayere@polarsen.com
 Requires-Dist: padwan-llm>=0.6.0

{padwan_cli-0.3.0 → padwan_cli-0.4.0}/padwan_cli/chat.py

@@ -20,7 +20,13 @@ from padwan_llm import (
 from padwan_llm.gemini import GeminiClient
 from padwan_llm.gemini.models import ThinkingConfig
 from .utils import ALL_MODELS, console
-from .widgets import StreamingMessage, ThoughtMessage, ToolCallMessage, UserMessage
+from .widgets import (
+    ErrorMessage,
+    StreamingMessage,
+    ThoughtMessage,
+    ToolCallMessage,
+    UserMessage,
+)
 
 chat_group = CommandGroup(name="chat", help="Chat with an LLM")
 
@@ -95,6 +101,11 @@ async def chat_send_fn(
         "--extra-params",
         help="Extra JSON object merged into every request body (e.g. '{\"temperature\": 0}')",
     ),
+    mcp_urls: list[str] | None = Option(
+        None,
+        "--mcp",
+        help=f"Streamable-HTTP MCP server URL(s) to expose as tools (e.g. {DATAGOUV_MCP_URL})",
+    ),
     ctx: TuiContext = TuiOption(),
 ) -> None:
     """Start a conversation. Use Ctrl+C to exit."""
@@ -170,7 +181,7 @@ async def chat_send_fn(
         # the first turn under a new id lands in the store via `session.save()`.
         session = AgentSession.load(
             client=client,
-            mcp_tools=[McpStreamable(url=DATAGOUV_MCP_URL)],
+            mcp_tools=[McpStreamable(url=url) for url in mcp_urls or ()],
             on_tool=_on_tool,
             on_mcp_connect=_on_mcp_connect,
             store=_store,
@@ -193,23 +204,33 @@ async def chat_send_fn(
                         ctx.set_silent_queue(True)
                         needs_new_text_widget = True
                         widget: StreamingMessage | None = None
-                        async for chunk in session.stream(user_input):
-                            if needs_new_text_widget:
-                                widget = StreamingMessage()
-                                ctx.mount_widget(widget)
-                                needs_new_text_widget = False
-                                current_thought = None
-                            assert widget is not None
-                            widget.append(chunk)
-                            if (n := ctx.pending_count) > 0:
-                                ctx.set_hint(f"Responding... ({n} queued)")
+                        try:
+                            async for chunk in session.stream(user_input):
+                                if needs_new_text_widget or widget is None:
+                                    widget = StreamingMessage()
+                                    ctx.mount_widget(widget)
+                                    needs_new_text_widget = False
+                                    current_thought = None
+                                widget.append(chunk)
+                                if (n := ctx.pending_count) > 0:
+                                    ctx.set_hint(f"Responding... ({n} queued)")
+                        except Exception as e:
+                            # Surface the error inline: the outer handler is
+                            # only reached after session teardown, which can
+                            # hang on the MCP listener (see padwan-llm mcp.py).
+                            ctx.mount_widget(ErrorMessage(str(e)))
+                            needs_new_text_widget = True
                         ctx.set_silent_queue(False)
                         ctx.set_hint("Chat mode - press Ctrl+C to exit")
                         ctx.set_status_above(_format_tokens(session, ctx) or None)
                         user_input = await ctx.prompt()
                     else:
-                        async for chunk in session.stream(user_input):
-                            console.print(chunk, end="")
+                        try:
+                            async for chunk in session.stream(user_input):
+                                console.print(chunk, end="")
+                        except Exception as e:
+                            console.print(f"\n[red]✖ {e}[/red]")
+                            break
                         console.print()
                         if tokens := _format_tokens(session):
                             console.print(f"[dim]{tokens}[/dim]")

{padwan_cli-0.3.0 → padwan_cli-0.4.0}/padwan_cli/widgets.py

@@ -16,6 +16,7 @@ USER_COLOR = "#87ceeb"
 ASSISTANT_COLOR = "#98fb98"
 TOOL_CALL_COLOR = "#ffa500"
 THOUGHT_COLOR = "#9370db"
+ERROR_COLOR = "#ff6b6b"
 
 # Colors for batch job states
 STATE_COLORS = {
@@ -66,6 +67,14 @@ class ToolCallMessage(Static):
         self.update(text)
 
 
+class ErrorMessage(Static):
+    """A styled widget for displaying an error inline in the chat."""
+
+    def __init__(self, content: str, **kwargs) -> None:
+        text = Text(f"✖ {content}", style=ERROR_COLOR)
+        super().__init__(text, **kwargs)
+
+
 class ThoughtMessage(Static):
     """A styled widget for displaying a model reasoning/thought block.
 

{padwan_cli-0.3.0 → padwan_cli-0.4.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "padwan-cli"
-version = "0.3.0"
+version = "0.4.0"
 description = "CLI and TUI for padwan-llm"
 readme = "README.md"
 authors = [