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

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.0b1.dist-info → vibecore-0.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vibecore
-Version: 0.3.0b1
+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
@@ -29,6 +29,7 @@ Classifier: Typing :: Typed
 Requires-Python: >=3.11
 Requires-Dist: ddgs>=9.5.4
 Requires-Dist: html2text>=2024.2.26
+Requires-Dist: httpx>=0.24.0
 Requires-Dist: litellm>=1.72.4
 Requires-Dist: openai-agents[litellm]>=0.2.2
 Requires-Dist: pydantic-settings>=2.10.1

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

@@ -1,21 +1,31 @@
 vibecore/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vibecore/cli.py,sha256=JspCzt8jlPur35yil7i6Whiw_u33DYeRldx4G-36ueo,4166
+vibecore/cli.py,sha256=HU2cjvEDzMWlauoArUItBlTpsVAdgVuKFMLshtaqvUE,7119
 vibecore/context.py,sha256=JUVkZpmKGUSlcchrHpxu-oSO8D21GDgHT1BXDzZDTeQ,844
 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=u6kWIMdd5IoLE_SJYqPChR1du8bJ3ZhuCHQQvsUy_cY,6041
+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
+vibecore/auth/__init__.py,sha256=lFxAcdjXpmCjU2tWBnej3fyUqJKSc41qHLF8929nMRM,451
+vibecore/auth/config.py,sha256=xEuChq61dygaDGhpOA9kK3VSximL42Usd-xBOY2XMpw,1650
+vibecore/auth/interceptor.py,sha256=KsMbnTi8DDVDnwiC1YT4APjcpvpIVQK3kMSS28wizNs,5127
+vibecore/auth/manager.py,sha256=xGhE8BUwmiAmT7zQWIqqYGD_3d1qdkBt9b-AslWIHJM,5980
+vibecore/auth/models.py,sha256=5dw_p2s2AMSCp_NqXxF_3gsb4dvauEMturK23t9tRA0,1083
+vibecore/auth/oauth_flow.py,sha256=LV8zQbNbkBRmMObsuvvc61PUQHypKyx4wByNd7PJzTc,4274
+vibecore/auth/pkce.py,sha256=YMQxddItPqUyy1pkKfYPNWIahWhIUVWL7vPFsQQvb7M,956
+vibecore/auth/storage.py,sha256=dWSDi4thRHHrSTJmAOInOdArTGQ_t0EdWHrk4Am8LWg,3495
+vibecore/auth/token_manager.py,sha256=Dwd0EkbBVeoIef7K6RW4QzaS7-6HAM-or0Ra4M1xddg,4611
 vibecore/handlers/__init__.py,sha256=pFogA2n3GeIi4lmlUEU5jFprNNOaA6AWRR8Wc9X-P4Y,148
-vibecore/handlers/stream_handler.py,sha256=N3fs7jO9QcjSClSyglrwJQ3ky08F70xS8j5m-aYNkyM,10336
+vibecore/handlers/stream_handler.py,sha256=e4VJOJcH84QmgKkGjOedwd85EZWmmvipmJpGuDBSPhA,11069
 vibecore/mcp/__init__.py,sha256=sl2_8pWjPx4TotO0ZojunVA6Jn6yOhbTQNbQG9-C-Jc,199
 vibecore/mcp/manager.py,sha256=RRpoFUiQjSg0-h9M7sF3odaSM7LURwrxVlaae5Yml3M,5644
 vibecore/mcp/server_wrapper.py,sha256=CkTyqZzmy7oeciSml0Q915orho_U2XcHoqhDVMOq7nY,3989
 vibecore/models/__init__.py,sha256=-vF0e_S5p7VdBM_TZuDI6oWFHcPecdX5KRjUzifHBow,116
 vibecore/models/anthropic.py,sha256=X48qj73cQ2jiChtoeMZYfLrkWTGABQZLEOPeU17hzuU,9750
+vibecore/models/anthropic_auth.py,sha256=_hRGbPHTs6IrkSc5mDI7mshdOZYCyqu82HX9WdQIb8Y,8856
 vibecore/prompts/common_system_prompt.txt,sha256=L-YlOfTJSQLtVg6d0r9lcD5FrgOLzoSWjGzVfQjcDBg,4916
 vibecore/session/__init__.py,sha256=FXbtw8DZVBw8e3qCA2cQharMzozblhwA0yU4U0JRSkM,121
 vibecore/session/file_lock.py,sha256=vCuDdfaOGDeVpTjJP6yBJx7onIT7JkkAeAuWAtuLJb8,3739
@@ -65,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.0b1.dist-info/METADATA,sha256=PYHen54ZzyRC9-8d6NA9GNxiV24aexWU8kPaQiiuJCk,18066
-vibecore-0.3.0b1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-vibecore-0.3.0b1.dist-info/entry_points.txt,sha256=YldTakc3dNelboaWXtzCcnM5MXvU2_6pVOjc2xPjDTY,47
-vibecore-0.3.0b1.dist-info/licenses/LICENSE,sha256=KXxxifvrcreHrZ4aOYgP-vA8DRHHueW389KKOeEbtjc,1069
-vibecore-0.3.0b1.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,,

vibecore-0.3.1.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+vibecore = vibecore.cli:cli_main

vibecore-0.3.0b1.dist-info/entry_points.txt

@@ -1,2 +0,0 @@
-[console_scripts]
-vibecore = vibecore.cli:main