vibecore 0.3.0b2__py3-none-any.whl → 0.3.1__py3-none-any.whl

vibecore/settings.py

@@ -155,6 +155,11 @@ class Settings(BaseSettings):
         description="List of MCP servers to connect to",
     )
 
+    rich_tool_names: list[str] = Field(
+        default_factory=list,
+        description="List of tools to render with RichToolMessage (temporary settings)",
+    )
+
     @property
     def model(self) -> str | Model:
         """Get the configured model.

vibecore/widgets/tool_message_factory.py

@@ -9,12 +9,14 @@ import contextlib
 import json
 from typing import Any
 
+from vibecore.settings import settings
 from vibecore.widgets.messages import MessageStatus
 from vibecore.widgets.tool_messages import (
     BaseToolMessage,
     MCPToolMessage,
     PythonToolMessage,
     ReadToolMessage,
+    RichToolMessage,
     TaskToolMessage,
     TodoWriteToolMessage,
     ToolMessage,
@@ -129,6 +131,12 @@ def create_tool_message(
         else:
             return WebFetchToolMessage(url=url, status=status)
 
+    elif tool_name in settings.rich_tool_names:
+        if output is not None:
+            return RichToolMessage(tool_name=tool_name, arguments=arguments, output=output, status=status)
+        else:
+            return RichToolMessage(tool_name=tool_name, arguments=arguments, status=status)
+
     # Default to generic ToolMessage for all other tools
     else:
         if output is not None:

vibecore/widgets/tool_messages.py

@@ -483,6 +483,118 @@ class MCPToolMessage(BaseToolMessage):
                         )
 
 
+class RichToolMessage(BaseToolMessage):
+    """A widget to display rich (json/markdown) custom tool execution messages."""
+
+    SQL_QUERY_KEY = "query"
+    tool_name: reactive[str] = reactive("")
+    arguments: reactive[str] = reactive("")
+
+    def __init__(
+        self,
+        tool_name: str,
+        arguments: str,
+        output: str = "",
+        status: MessageStatus = MessageStatus.EXECUTING,
+        **kwargs,
+    ) -> None:
+        """
+        Construct an RichToolMessage.
+
+        Args:
+            tool_name: The name of the tool being called.
+            arguments: JSON string of tool arguments.
+            output: The output from the tool (optional, can be set later).
+            status: The status of execution.
+            **kwargs: Additional keyword arguments for Widget.
+        """
+        super().__init__(status=status, **kwargs)
+        self.tool_name = tool_name
+        self.arguments = arguments
+        self.output = output
+
+    def _prettify_json_output(self, output: str) -> tuple[bool, str]:
+        """Try to prettify JSON output.
+
+        Args:
+            output: The raw output string.
+
+        Returns:
+            A tuple of (is_json, formatted_output).
+        """
+        if not output or not output.strip():
+            return False, output
+
+        try:
+            # Try to parse as JSON
+            json_obj = json.loads(output)
+            # Pretty print with 2-space indentation
+            formatted = json.dumps(json_obj, indent=2, ensure_ascii=False)
+            return True, formatted
+        except (json.JSONDecodeError, TypeError, ValueError):
+            # Not valid JSON, return as-is
+            return False, output
+
+    def on_button_pressed(self, event: Button.Pressed) -> None:
+        """Handle button press events."""
+        if event.button.has_class("copy-button"):
+            arg_dict = json.loads(self.arguments)
+            if self.SQL_QUERY_KEY in arg_dict:
+                # XXX(serialx): Special case for SQL queries
+                query = arg_dict[self.SQL_QUERY_KEY]
+                self.app.copy_to_clipboard(query)
+
+    def compose(self) -> ComposeResult:
+        """Create child widgets for the rich tool message."""
+        # Header line showing MCP server and tool
+        # Access the actual values, not the reactive descriptors
+        yield MessageHeader("⏺", self.tool_name, status=self.status)
+
+        # Arguments display (if any)
+        if self.arguments and self.arguments != "{}":
+            with Horizontal(classes="rich-arguments"):
+                yield Static("└─", classes="rich-arguments-prefix")
+                with Vertical(classes="rich-arguments-content"):
+                    # Truncate arguments if too long
+                    max_args_length = 100
+                    arg_dict = json.loads(self.arguments)
+                    if self.SQL_QUERY_KEY in arg_dict:
+                        # XXX(serialx): Special case for SQL queries
+                        query = arg_dict[self.SQL_QUERY_KEY]
+                        yield Button("Copy", classes="copy-button", variant="primary")
+                        yield ExpandableMarkdown(query, language="sql", truncated_lines=8, classes="rich-output-sql")
+                        del arg_dict[self.SQL_QUERY_KEY]
+
+                    display_args = arg_dict[:max_args_length] + "…" if len(arg_dict) > max_args_length else arg_dict
+                    yield Static(f"Args: {display_args}", classes="rich-arguments-text")
+
+        # Output - check if it's JSON and prettify if so
+        if self.output:
+            try:
+                json_output = json.loads(self.output)
+            except json.JSONDecodeError:
+                json_output = None
+            if json_output:
+                assert json_output.get("type") == "text", "Expected JSON output type to be 'text'"
+                is_json, processed_output = self._prettify_json_output(json_output.get("text", ""))
+            else:
+                # output should always be a JSON string, but if not, treat it as plain text
+                is_json, processed_output = False, self.output
+            with Horizontal(classes="tool-output"):
+                yield Static("└─", classes="tool-output-prefix")
+                with Vertical(classes="tool-output-content"):
+                    if is_json:
+                        # Use ExpandableMarkdown for JSON with syntax highlighting
+                        yield ExpandableMarkdown(
+                            processed_output, language="json", truncated_lines=8, classes="rich-output-json"
+                        )
+                    else:
+                        # Use ExpandableMarkdown for non-JSON content (renders as markdown without code block)
+                        yield ExpandableMarkdown(
+                            processed_output, language="", truncated_lines=5, classes="rich-output-markdown"
+                        )
+
+
 class WebSearchToolMessage(BaseToolMessage):
     """A widget to display web search results."""
 

vibecore/widgets/tool_messages.tcss

@@ -288,6 +288,75 @@ MCPToolMessage {
     }
 }
 
+RichToolMessage {
+    Horizontal.rich-arguments {
+        height: auto;
+        layers: main button;
+
+        .copy-button {
+            layer: button;
+            dock: right;
+            height: 1;
+            width: 8;
+            min-width: 8;
+            margin: 0;
+            padding: 0;
+            background: $secondary;
+            color: $text;
+            border: none;
+            
+            &:hover {
+                background: $secondary-lighten-1;
+            }
+            
+            &:focus {
+                background: $secondary-lighten-1;
+                text-style: bold;
+            }
+        }
+
+        &> .rich-arguments-prefix {
+            height: 1;
+            width: 5;
+            padding-left: 2;
+            padding-right: 1;
+            color: $text-muted;
+        }
+
+        &> Vertical.rich-arguments-content {
+            height: auto;
+            width: 1fr;
+
+            .rich-arguments-label {
+                color: $text-muted;
+                # margin-bottom: 1;
+            }
+
+            .rich-arguments-text {
+                color: $text-muted;
+                text-overflow: ellipsis;
+            }
+        }
+    }
+    
+    Horizontal.tool-output {
+        height: auto;
+
+        &> .tool-output-prefix {
+            height: 1;
+            width: 5;
+            padding-left: 2;
+            padding-right: 1;
+            color: $text-muted;
+        }
+
+        &> Vertical.tool-output-content {
+            height: auto;
+            width: 1fr;
+        }
+    }
+}
+
 WebSearchToolMessage {
     Horizontal.tool-output {
         height: auto;

{vibecore-0.3.0b2.dist-info → vibecore-0.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vibecore
-Version: 0.3.0b2
+Version: 0.3.1
 Summary: Build your own AI-powered automation tools in the terminal with this extensible agent framework
 Project-URL: Homepage, https://github.com/serialx/vibecore
 Project-URL: Repository, https://github.com/serialx/vibecore

{vibecore-0.3.0b2.dist-info → vibecore-0.3.1.dist-info}/RECORD

@@ -5,7 +5,7 @@ vibecore/flow.py,sha256=ZaKzMsz4YBvgelVzJOIHnTJzMWTmvkfvudwW_hllq6U,3384
 vibecore/main.py,sha256=MIn7Mpg_xO_20c6Mju8PgY-MmVCTER9cepHd5YFbtYs,20175
 vibecore/main.tcss,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vibecore/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vibecore/settings.py,sha256=OnEME61vi7qBiSbu7Q_7twAZa7l0sOXBIavxYNAVOWU,6897
+vibecore/settings.py,sha256=PwivzoJ-xyJu-Mmnv4E6G_E8up4GNvvz9_j4ttj-GUw,7063
 vibecore/agents/default.py,sha256=wxeP3Hsq9MVBChyMF_sNkVHCtFFXgy5ghDIxy9eH_fQ,2287
 vibecore/agents/prompts.py,sha256=0oO9QzcytIbzgZcKJQjWD9fSRNQqBqqK5ku0fcYZZrA,324
 vibecore/agents/task.py,sha256=cdhZDzKDA8eHHNYPhogFIKBms3oVAMvYeiBB2GqYNuE,1898
@@ -75,11 +75,11 @@ vibecore/widgets/info.py,sha256=hXtsRUOE13oHbIm9FNe1GCUX_FCht28pgT9SQWeJ69I,1567
 vibecore/widgets/info.tcss,sha256=v30IqNt1two-ezIcm18ZEInKRKcRkAW-h-UH2r8QzSo,201
 vibecore/widgets/messages.py,sha256=az4fJtdk3ItSoFZBG_adRDUHdTLttIV8F23E8LOb-mg,8156
 vibecore/widgets/messages.tcss,sha256=WtBbjf5LgFkUhzhVlxJB7NMbagWladJawDizvDm7hBE,1271
-vibecore/widgets/tool_message_factory.py,sha256=FMwavKMRNT8ik9RgcL37WuM19Ln-c5wuFmS0A2CkikM,5377
-vibecore/widgets/tool_messages.py,sha256=PKmPigeOlccowo0uLpgIPzsmmE-zkichzuXIS6hWsbQ,24501
-vibecore/widgets/tool_messages.tcss,sha256=mcFY58FE1AcfEvEiA_Yb7sMpIniTIC_IjDvv8M7vWOA,6924
-vibecore-0.3.0b2.dist-info/METADATA,sha256=7YQejvTynr_4g3MmrV0cRLHdH4P5OiqC403t8qYjVnM,18095
-vibecore-0.3.0b2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-vibecore-0.3.0b2.dist-info/entry_points.txt,sha256=i9mOKvpz07ciV_YYisxNCYZ53_Crjkn9mciiQ3aA6QM,51
-vibecore-0.3.0b2.dist-info/licenses/LICENSE,sha256=KXxxifvrcreHrZ4aOYgP-vA8DRHHueW389KKOeEbtjc,1069
-vibecore-0.3.0b2.dist-info/RECORD,,
+vibecore/widgets/tool_message_factory.py,sha256=yrZorT4HKo5b6rWUc0dgQle7q7cvLyq8JllE772RZS0,5730
+vibecore/widgets/tool_messages.py,sha256=hJOolN3iLTAjqfotfH1elXqsdDo1r_UHjsyRVH0GAeo,29415
+vibecore/widgets/tool_messages.tcss,sha256=gdChmHClURqn_sD9GkcOGQcQVYvUUl75mLUYp85sKz8,8442
+vibecore-0.3.1.dist-info/METADATA,sha256=wCQB7NKn35T0W3onSrpV1FNX13zHzO-VEn0WqU6ZIlw,18093
+vibecore-0.3.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+vibecore-0.3.1.dist-info/entry_points.txt,sha256=i9mOKvpz07ciV_YYisxNCYZ53_Crjkn9mciiQ3aA6QM,51
+vibecore-0.3.1.dist-info/licenses/LICENSE,sha256=KXxxifvrcreHrZ4aOYgP-vA8DRHHueW389KKOeEbtjc,1069
+vibecore-0.3.1.dist-info/RECORD,,