ai-agent-inspector 1.0.0__py3-none-any.whl → 1.1.1__py3-none-any.whl

agent_inspector/__init__.py

@@ -43,6 +43,11 @@ from .core.interfaces import Exporter, Sampler
 from .core.trace import (
     Trace,
     TraceContext,
+    agent_communication,
+    agent_handoff,
+    agent_join,
+    agent_leave,
+    agent_spawn,
     error,
     final,
     get_trace,
@@ -51,6 +56,8 @@ from .core.trace import (
     memory_write,
     run,
     set_trace,
+    task_assign,
+    task_complete,
     tool,
 )
 
@@ -71,6 +78,11 @@ except ImportError:
 
 # Event types
 from .core.events import (
+    AgentCommunicationEvent,
+    AgentHandoffEvent,
+    AgentJoinEvent,
+    AgentLeaveEvent,
+    AgentSpawnEvent,
     BaseEvent,
     ErrorEvent,
     EventStatus,
@@ -81,6 +93,8 @@ from .core.events import (
     MemoryWriteEvent,
     RunEndEvent,
     RunStartEvent,
+    TaskAssignmentEvent,
+    TaskCompletionEvent,
     ToolCallEvent,
 )
 
@@ -100,6 +114,14 @@ __all__ = [
     "final",
     "get_trace",
     "set_trace",
+    # Multi-agent tracing
+    "agent_spawn",
+    "agent_join",
+    "agent_leave",
+    "agent_communication",
+    "agent_handoff",
+    "task_assign",
+    "task_complete",
     # Configuration
     "TraceConfig",
     "Profile",
@@ -121,6 +143,14 @@ __all__ = [
     "MemoryWriteEvent",
     "ErrorEvent",
     "FinalAnswerEvent",
+    # Multi-agent event types
+    "AgentSpawnEvent",
+    "AgentJoinEvent",
+    "AgentLeaveEvent",
+    "AgentCommunicationEvent",
+    "AgentHandoffEvent",
+    "TaskAssignmentEvent",
+    "TaskCompletionEvent",
     # Adapters (optional)
     "enable_langchain",
     # API server (optional)

agent_inspector/cli.py

@@ -113,18 +113,30 @@ def cmd_prune(args):
     # Override with CLI arguments
     if args.retention_days is not None:
         config.retention_days = args.retention_days
+    if getattr(args, "retention_max_bytes", None) is not None:
+        config.retention_max_bytes = args.retention_max_bytes
 
     # Initialize database
     db = Database(config)
     db.initialize()
 
-    # Prune old runs
+    # Prune by age first
     deleted_count = db.prune_old_runs(retention_days=config.retention_days)
+    size_deleted = 0
+
+    # Then prune by size if configured
+    max_bytes = config.retention_max_bytes
+    if max_bytes is not None and max_bytes > 0:
+        size_deleted = db.prune_by_size(max_bytes)
+        deleted_count += size_deleted
+        if size_deleted > 0:
+            print(f"✅ Pruned {size_deleted} runs by size (max_bytes={max_bytes})")
 
     if deleted_count > 0:
-        print(f"✅ Pruned {deleted_count} old runs")
+        if size_deleted == 0:  # only age-based pruning had effect
+            print(f"✅ Pruned {deleted_count} old runs")
 
-        # Optionally vacuum to reclaim space
+        # Optionally vacuum to reclaim space (prune_by_size already vacuums)
         if args.vacuum:
             print("💾 Running VACUUM to reclaim disk space...")
             if db.vacuum():
@@ -408,6 +420,13 @@ Examples:
         type=int,
         help="Retention period in days (default: from config)",
     )
+    prune_parser.add_argument(
+        "--retention-max-bytes",
+        type=int,
+        default=None,
+        metavar="BYTES",
+        help="Prune oldest runs until DB size is at or below BYTES (optional; from config if set)",
+    )
     prune_parser.add_argument(
         "--vacuum",
         action="store_true",