agentic-blocks 0.1.23__tar.gz → 0.1.24__tar.gz

{agentic_blocks-0.1.23/src/agentic_blocks.egg-info → agentic_blocks-0.1.24}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentic-blocks
-Version: 0.1.23
+Version: 0.1.24
 Summary: Simple building blocks for agentic AI systems with MCP client and conversation management
 Author-email: Magnus Bjelkenhed <bjelkenhed@gmail.com>
 License: MIT
@@ -27,6 +27,10 @@ Requires-Dist: langchain-core
 Requires-Dist: langfuse
 Requires-Dist: pocketflow
 Requires-Dist: pocketflow-tracing
+Requires-Dist: agno
+Requires-Dist: ipywidgets
+Requires-Dist: rich[jupyter]
+Requires-Dist: fastapi
 Provides-Extra: test
 Requires-Dist: pytest; extra == "test"
 Provides-Extra: dev

{agentic_blocks-0.1.23 → agentic_blocks-0.1.24}/pyproject.toml

@@ -14,7 +14,7 @@ agentic_blocks = []
 
 [project]
 name = "agentic-blocks"
-version = "0.1.23"
+version = "0.1.24"
 description = "Simple building blocks for agentic AI systems with MCP client and conversation management"
 readme = "README.md"
 requires-python = ">=3.11"
@@ -43,6 +43,10 @@ dependencies = [
     "langfuse",
     "pocketflow",
     "pocketflow-tracing",
+    "agno",
+    "ipywidgets", 
+    "rich[jupyter]",
+    "fastapi",
 ]
 
 [project.urls]

{agentic_blocks-0.1.23 → agentic_blocks-0.1.24}/src/agentic_blocks/agent.py

@@ -4,6 +4,18 @@ from agentic_blocks.utils.tools_utils import (
     execute_pending_tool_calls,
 )
 from agentic_blocks import call_llm, Messages
+from rich.panel import Panel
+from rich.box import HEAVY
+from rich.console import Console
+from rich.console import Group
+
+console = Console(
+    style="black on bright_white",
+    force_terminal=True,
+    width=None,
+    legacy_windows=False,
+    color_system="truecolor",
+)
 
 
 class Agent:
@@ -11,6 +23,7 @@ class Agent:
         self.system_prompt = system_prompt
         self.tools = tools
         self.tool_registry = create_tool_registry(tools)
+        self.panels = []
 
         # Create nodes
         self.llm_node = self._create_llm_node()
@@ -32,7 +45,8 @@ class Agent:
                 self.tools = tools
 
             def prep(self, shared):
-                return shared["messages"]
+                messages = shared["messages"]
+                return messages
 
             def exec(self, messages) -> Messages:
                 response = call_llm(messages=messages, tools=self.tools)
@@ -49,14 +63,31 @@ class Agent:
 
     def _create_tool_node(self):
         class ToolNode(Node):
-            def __init__(self, tool_registry):
+            def __init__(self, tool_registry, agent):
                 super().__init__()
                 self.tool_registry = tool_registry
+                self.agent = agent
 
             def prep(self, shared):
                 return shared["messages"]
 
             def exec(self, messages) -> Messages:
+                tool_calls = messages.get_pending_tool_calls()[0]
+                tool_name = tool_calls["tool_name"]
+                tool_arguments = tool_calls["arguments"]
+
+                # Format arguments nicely
+                if isinstance(tool_arguments, dict):
+                    args_str = ", ".join(
+                        [f"{k}={v}" for k, v in tool_arguments.items()]
+                    )
+                    formatted_call = f"{tool_name}({args_str})"
+                else:
+                    formatted_call = f"{tool_name}({tool_arguments})"
+
+                tool_panel = self.agent.create_panel(formatted_call, "Tool Calls")
+                self.agent.panels.append(tool_panel)
+
                 tool_responses = execute_pending_tool_calls(
                     messages, self.tool_registry
                 )
@@ -66,7 +97,7 @@ class Agent:
             def post(self, shared, prep_res, messages):
                 return "llm_node"
 
-        return ToolNode(self.tool_registry)
+        return ToolNode(self.tool_registry, self)
 
     def _create_answer_node(self):
         class AnswerNode(Node):
@@ -86,3 +117,29 @@ class Agent:
         self.flow.run(shared)
 
         return shared["answer"]
+
+    def print_response(self, user_prompt: str, stream: bool = False):
+        # Reset panels and start with message
+        self.panels = []
+        message_panel = self.create_panel(user_prompt, "Message")
+        self.panels.append(message_panel)
+
+        # Always collect all panels first
+        response = self.invoke(user_prompt)
+        response_panel = self.create_panel(response, "Response")
+        self.panels.append(response_panel)
+
+        # Print all panels as a group (no gaps)
+        panel_group = Group(*self.panel)
+        console.print(panel_group)
+
+    def create_panel(self, content, title, border_style="blue"):
+        return Panel(
+            content,
+            title=title,
+            title_align="left",
+            border_style=border_style,
+            box=HEAVY,
+            expand=True,  # Full terminal width
+            padding=(1, 1),  # Internal padding
+        )

{agentic_blocks-0.1.23 → agentic_blocks-0.1.24}/src/agentic_blocks/messages.py

@@ -175,6 +175,13 @@ class Messages:
         """Get the current messages list."""
         return self.messages
 
+    def get_user_message(self) -> str:
+        """Get the user message."""
+        for message in reversed(self.messages):
+            if message.get("role") == "user":
+                return message.get("content") or ""
+        return ""
+
     def has_pending_tool_calls(self) -> bool:
         """
         Check if the last message has tool calls that need execution.
@@ -213,7 +220,7 @@ class Messages:
             List of dictionaries with 'tool_name', 'arguments', and 'tool_call_id' keys
         """
         pending_calls = []
-        
+
         if not self.messages:
             return pending_calls
 
@@ -234,19 +241,22 @@ class Messages:
                     function_info = tool_call.get("function", {})
                     tool_name = function_info.get("name")
                     arguments_str = function_info.get("arguments", "{}")
-                    
+
                     # Parse arguments JSON string to dict
                     import json
+
                     try:
                         arguments = json.loads(arguments_str)
                     except json.JSONDecodeError:
                         arguments = {}
-                    
-                    pending_calls.append({
-                        "tool_name": tool_name,
-                        "arguments": arguments,
-                        "tool_call_id": tool_call_id
-                    })
+
+                    pending_calls.append(
+                        {
+                            "tool_name": tool_name,
+                            "arguments": arguments,
+                            "tool_call_id": tool_call_id,
+                        }
+                    )
 
         return pending_calls
 

{agentic_blocks-0.1.23 → agentic_blocks-0.1.24/src/agentic_blocks.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentic-blocks
-Version: 0.1.23
+Version: 0.1.24
 Summary: Simple building blocks for agentic AI systems with MCP client and conversation management
 Author-email: Magnus Bjelkenhed <bjelkenhed@gmail.com>
 License: MIT
@@ -27,6 +27,10 @@ Requires-Dist: langchain-core
 Requires-Dist: langfuse
 Requires-Dist: pocketflow
 Requires-Dist: pocketflow-tracing
+Requires-Dist: agno
+Requires-Dist: ipywidgets
+Requires-Dist: rich[jupyter]
+Requires-Dist: fastapi
 Provides-Extra: test
 Requires-Dist: pytest; extra == "test"
 Provides-Extra: dev

{agentic_blocks-0.1.23 → agentic_blocks-0.1.24}/src/agentic_blocks.egg-info/requires.txt

@@ -6,6 +6,10 @@ langchain-core
 langfuse
 pocketflow
 pocketflow-tracing
+agno
+ipywidgets
+rich[jupyter]
+fastapi
 
 [dev]
 pytest