agentic-blocks 0.1.19__py3-none-any.whl → 0.1.20__py3-none-any.whl

agentic_blocks/visualization/visualize.py

@@ -15,8 +15,6 @@ from typing import Any, Optional, Tuple, Union
 
 from pocketflow import Flow
 
-from async_flow import order_pipeline
-
 
 def build_mermaid(start):
     ids, visited, lines = {}, set(), ["graph LR"]
@@ -268,16 +266,16 @@ def create_d3_visualization(
             stroke-width: 1.5px;
         }
         .node-labels {
-            font-size: 12px;
+            font-size: 14px;
             pointer-events: none;
         }
         .link-labels {
-            font-size: 10px;
+            font-size: 12px;
             fill: #666;
             pointer-events: none;
         }
         .group-link-labels {
-            font-size: 11px;
+            font-size: 13px;
             font-weight: bold;
             fill: #333;
             pointer-events: none;
@@ -291,7 +289,7 @@ def create_d3_visualization(
             ry: 10;
         }
         .group-label {
-            font-size: 14px;
+            font-size: 16px;
             font-weight: bold;
             pointer-events: none;
         }
@@ -360,14 +358,14 @@ def create_d3_visualization(
             
             // Create a force simulation
             const simulation = d3.forceSimulation(data.nodes)
-                // Controls the distance between connected nodes
-                .force("link", d3.forceLink(data.links).id(d => d.id).distance(100))
-                // Controls how nodes repel each other - lower values bring nodes closer
-                .force("charge", d3.forceManyBody().strength(-30))
+                // Controls the distance between connected nodes - increased for larger visualization
+                .force("link", d3.forceLink(data.links).id(d => d.id).distance(200))
+                // Controls how nodes repel each other - increased repulsion for more spread
+                .force("charge", d3.forceManyBody().strength(-100))
                 // Centers the entire graph in the SVG
                 .force("center", d3.forceCenter(width / 2, height / 2))
-                // Prevents nodes from overlapping - acts like a minimum distance
-                .force("collide", d3.forceCollide().radius(50));
+                // Prevents nodes from overlapping - increased radius for larger spacing
+                .force("collide", d3.forceCollide().radius(80));
             
             // Group forces - create a force to keep nodes in the same group closer together
             // This creates the effect of nodes clustering within their group boxes
@@ -432,7 +430,7 @@ def create_d3_visualization(
                     
                     if (source && target) {
                         // Add a horizontal force to align groups
-                        const desiredDx = 300; // Desired horizontal distance between linked groups
+                        const desiredDx = 500; // Desired horizontal distance between linked groups - increased for larger layout
                         const dx = target.cx - source.cx;
                         const diff = desiredDx - Math.abs(dx);
                         
@@ -525,21 +523,21 @@ def create_d3_visualization(
                 .attr("font-size", "10px")
                 .attr("fill", "#666");
             
-            // Create nodes
+            // Create nodes - increased size for larger visualization
             const node = svg.append("g")
                 .attr("class", "nodes")
                 .selectAll("circle")
                 .data(data.nodes)
                 .enter()
                 .append("circle")
-                .attr("r", 15)
+                .attr("r", 25)
                 .attr("fill", d => color(d.group))
                 .call(d3.drag()
                     .on("start", dragstarted)
                     .on("drag", dragged)
                     .on("end", dragended));
             
-            // Create node labels
+            // Create node labels - increased offset for larger nodes
             const nodeLabel = svg.append("g")
                 .attr("class", "node-labels")
                 .selectAll("text")
@@ -548,7 +546,7 @@ def create_d3_visualization(
                 .append("text")
                 .text(d => d.name)
                 .attr("text-anchor", "middle")
-                .attr("dy", 25);
+                .attr("dy", 40);
             
             // Add tooltip on hover
             node.append("title")
@@ -659,16 +657,16 @@ def create_d3_visualization(
                     if (d.nodes.length > 0) {
                         let minX = Infinity, minY = Infinity, maxX = -Infinity, maxY = -Infinity;
                         
-                        // Find the bounding box for all nodes in the group
+                        // Find the bounding box for all nodes in the group - increased for larger nodes
                         d.nodes.forEach(n => {
-                            minX = Math.min(minX, n.x - 30);
-                            minY = Math.min(minY, n.y - 30);
-                            maxX = Math.max(maxX, n.x + 30);
-                            maxY = Math.max(maxY, n.y + 40); // Extra space for labels
+                            minX = Math.min(minX, n.x - 40);
+                            minY = Math.min(minY, n.y - 40);
+                            maxX = Math.max(maxX, n.x + 40);
+                            maxY = Math.max(maxY, n.y + 60); // Extra space for labels
                         });
                         
-                        // Add padding
-                        const padding = 20;
+                        // Add padding - increased for larger visualization
+                        const padding = 30;
                         minX -= padding;
                         minY -= padding;
                         maxX += padding;

{agentic_blocks-0.1.19.dist-info → agentic_blocks-0.1.20.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentic-blocks
-Version: 0.1.19
+Version: 0.1.20
 Summary: Simple building blocks for agentic AI systems with MCP client and conversation management
 Author-email: Magnus Bjelkenhed <bjelkenhed@gmail.com>
 License: MIT

{agentic_blocks-0.1.19.dist-info → agentic_blocks-0.1.20.dist-info}/RECORD

@@ -8,11 +8,9 @@ agentic_blocks/tracing/config.py,sha256=md2iBYzM_RarJotrBXXKk15Y14aKgIkOeG3b6Toa
 agentic_blocks/tracing/core.py,sha256=ZcW3vesqVqyL73Cd0YwflxqglUH9MK7u-xhEOSEIIJo,9119
 agentic_blocks/tracing/decorator.py,sha256=kE4_iNZfj4qNR9BSIf7EIpJToIwIPRaxU0K1-xcKCpw,9905
 agentic_blocks/utils/tools_utils.py,sha256=ZX6cqvtaV_evwMUnR_uukdvEtQah6xkDMBy7jwR7tJg,7145
-agentic_blocks/visualization/async_flow.py,sha256=lUIZUX6vU8eh0A8w1NNauQL5tKDoI0GSgzgtZwnFSiI,4596
-agentic_blocks/visualization/async_loop_flow.py,sha256=HnlDynb_j8fOern1jP1SX_ohclR0MLjnyUzkogJays0,2101
-agentic_blocks/visualization/visualize.py,sha256=YPNulA3Oy7Mkrneoq5w4Mk_boQBAesURPxFEd2wdALo,39463
-agentic_blocks-0.1.19.dist-info/licenses/LICENSE,sha256=r4IcBaAjTv3-yfjXgDPuRD953Qci0Y0nQn5JfHwLyBY,1073
-agentic_blocks-0.1.19.dist-info/METADATA,sha256=yb3CYZ4jY2YhT2sCYX36Zme7zHeh9YHEIKpXhLX3FOM,11994
-agentic_blocks-0.1.19.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-agentic_blocks-0.1.19.dist-info/top_level.txt,sha256=-1a4RAemqicXLU1rRzw4QHV3KlNeQDNxVs3m2gAT238,15
-agentic_blocks-0.1.19.dist-info/RECORD,,
+agentic_blocks/visualization/visualize.py,sha256=x59lFDKXqaFlereCEhN7GWDimiNRNcb9USeRPv8YY5I,39647
+agentic_blocks-0.1.20.dist-info/licenses/LICENSE,sha256=r4IcBaAjTv3-yfjXgDPuRD953Qci0Y0nQn5JfHwLyBY,1073
+agentic_blocks-0.1.20.dist-info/METADATA,sha256=lZmYZTTYfYyKFLgHfy3Mndyd1B4ob0FrYLUdbZavo88,11994
+agentic_blocks-0.1.20.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+agentic_blocks-0.1.20.dist-info/top_level.txt,sha256=-1a4RAemqicXLU1rRzw4QHV3KlNeQDNxVs3m2gAT238,15
+agentic_blocks-0.1.20.dist-info/RECORD,,

agentic_blocks/visualization/async_flow.py

@@ -1,165 +0,0 @@
-from pocketflow import AsyncNode, AsyncFlow
-import asyncio
-
-
-# Define Payment Nodes
-class ValidatePayment(AsyncNode):
-    async def exec_async(self, prep_res):
-        print("1.1.Validating payment...")
-        return "Payment validated successfully"
-
-    async def post_async(self, shared, prep_res, exec_res):
-        shared["payment_status"] = exec_res
-        return "default"
-
-
-class ProcessPayment(AsyncNode):
-    async def exec_async(self, prep_res):
-        print("1.2.Processing payment...")
-        return "Payment processed successfully"
-
-    async def post_async(self, shared, prep_res, exec_res):
-        shared["payment_result"] = exec_res
-        return "default"
-
-
-class PaymentConfirmation(AsyncNode):
-    async def exec_async(self, prep_res):
-        print("1.3.Confirming payment...")
-        return "Payment confirmed"
-
-    async def post_async(self, shared, prep_res, exec_res):
-        shared["payment_confirmation"] = exec_res
-        return "default"
-
-
-# Define Inventory Nodes
-class CheckStock(AsyncNode):
-    async def exec_async(self, prep_res):
-        print("2.1.Checking inventory stock...")
-        return "Stock available"
-
-    async def post_async(self, shared, prep_res, exec_res):
-        shared["stock_status"] = exec_res
-        return "default"
-
-
-class ReserveItems(AsyncNode):
-    async def exec_async(self, prep_res):
-        print("2.2.Reserving items...")
-        return "Items reserved"
-
-    async def post_async(self, shared, prep_res, exec_res):
-        shared["reservation_status"] = exec_res
-        return "default"
-
-
-class UpdateInventory(AsyncNode):
-    async def exec_async(self, prep_res):
-        print("2.3. Updating inventory...")
-        return "Inventory updated"
-
-    async def post_async(self, shared, prep_res, exec_res):
-        shared["inventory_update"] = exec_res
-        return "default"
-
-
-# Define Shipping Nodes
-class CreateLabel(AsyncNode):
-    async def exec_async(self, prep_res):
-        print("3.1 Creating shipping label...")
-        return "Shipping label created"
-
-    async def post_async(self, shared, prep_res, exec_res):
-        shared["shipping_label"] = exec_res
-        return "default"
-
-
-class AssignCarrier(AsyncNode):
-    async def exec_async(self, prep_res):
-        print("3.2 Assigning carrier...")
-        return "Carrier assigned"
-
-    async def post_async(self, shared, prep_res, exec_res):
-        shared["carrier"] = exec_res
-        return "default"
-
-
-class SchedulePickup(AsyncNode):
-    async def exec_async(self, prep_res):
-        print("3.3 Scheduling pickup...")
-        return "Pickup scheduled"
-
-    async def post_async(self, shared, prep_res, exec_res):
-        shared["pickup_status"] = exec_res
-        return "default"
-
-
-# Create node instances
-validate_payment = ValidatePayment()
-process_payment = ProcessPayment()
-payment_confirmation = PaymentConfirmation()
-
-check_stock = CheckStock()
-reserve_items = ReserveItems()
-update_inventory = UpdateInventory()
-
-create_label = CreateLabel()
-assign_carrier = AssignCarrier()
-schedule_pickup = SchedulePickup()
-
-# Payment processing sub-flow
-validate_payment >> process_payment >> payment_confirmation
-payment_flow = AsyncFlow(start=validate_payment)
-
-# Inventory sub-flow
-check_stock >> reserve_items >> update_inventory
-inventory_flow = AsyncFlow(start=check_stock)
-
-# Shipping sub-flow
-create_label >> assign_carrier >> schedule_pickup
-shipping_flow = AsyncFlow(start=create_label)
-
-# Connect the flows into a main order pipeline
-payment_flow >> inventory_flow >> shipping_flow
-# payment_flow >> inventory_flow >> create_label
-# payment_flow >> inventory_flow >> assign_carrier
-
-
-# Create the master flow
-class OrderFlow(AsyncFlow):
-    pass
-
-
-order_pipeline = OrderFlow(start=payment_flow)
-
-# Create shared data structure
-shared_data = {
-    "order_id": "ORD-12345",
-    "customer": "John Doe",
-    "items": [
-        {"id": "ITEM-001", "name": "Smartphone", "price": 999.99, "quantity": 1},
-        {"id": "ITEM-002", "name": "Phone case", "price": 29.99, "quantity": 1},
-    ],
-    "shipping_address": {
-        "street": "123 Main St",
-        "city": "Anytown",
-        "state": "CA",
-        "zip": "12345",
-    },
-}
-
-
-# Run the entire pipeline asynchronously
-async def main():
-    await order_pipeline.run_async(shared_data)
-
-    # Print final status
-    print("\nOrder processing completed!")
-    print(f"Payment: {shared_data.get('payment_confirmation')}")
-    print(f"Inventory: {shared_data.get('inventory_update')}")
-    print(f"Shipping: {shared_data.get('pickup_status')}")
-
-
-if __name__ == "__main__":
-    asyncio.run(main())

agentic_blocks/visualization/async_loop_flow.py

@@ -1,73 +0,0 @@
-from async_flow import *
-from pocketflow import Flow, AsyncParallelBatchNode, Node
-
-# Create node instances
-validate_payment = ValidatePayment()
-process_payment = ProcessPayment()
-payment_confirmation = PaymentConfirmation()
-
-check_stock = CheckStock()
-reserve_items = ReserveItems()
-update_inventory = UpdateInventory()
-
-create_label = CreateLabel()
-assign_carrier = AssignCarrier()
-schedule_pickup = SchedulePickup()
-
-# Payment processing sub-flow
-validate_payment >> process_payment
-validate_payment - "out_of_stock" >> validate_payment  # 循环重试
-process_payment - 'something fail' >> validate_payment
-process_payment - 'pass' >> payment_confirmation
-payment_flow = AsyncFlow(start=validate_payment)
-
-# Inventory sub-flow
-check_stock >> reserve_items >> update_inventory
-inventory_flow = AsyncFlow(start=check_stock)
-
-# Shipping sub-flow
-create_label >> assign_carrier >> schedule_pickup
-shipping_flow = AsyncFlow(start=create_label)
-
-# Connect the flows into a main order pipeline
-payment_flow >> inventory_flow >> shipping_flow
-# payment_flow >> inventory_flow >> create_label
-# payment_flow >> inventory_flow >> assign_carrier
-
-
-# Create the master flow
-class OrderFlow(AsyncFlow):
-    pass
-
-order_pipeline = OrderFlow(start=payment_flow)
-
-# Create shared data structure
-shared_data = {
-    "order_id": "ORD-12345",
-    "customer": "John Doe",
-    "items": [
-        {"id": "ITEM-001", "name": "Smartphone", "price": 999.99, "quantity": 1},
-        {"id": "ITEM-002", "name": "Phone case", "price": 29.99, "quantity": 1},
-    ],
-    "shipping_address": {
-        "street": "123 Main St",
-        "city": "Anytown",
-        "state": "CA",
-        "zip": "12345",
-    },
-}
-
-
-# Run the entire pipeline asynchronously
-async def main():
-    await order_pipeline.run_async(shared_data)
-
-    # Print final status
-    print("\nOrder processing completed!")
-    print(f"Payment: {shared_data.get('payment_confirmation')}")
-    print(f"Inventory: {shared_data.get('inventory_update')}")
-    print(f"Shipping: {shared_data.get('pickup_status')}")
-
-
-if __name__ == "__main__":
-    asyncio.run(main())