connectonion 0.5.9__py3-none-any.whl → 0.6.0__py3-none-any.whl

connectonion/__init__.py

@@ -1,6 +1,6 @@
 """ConnectOnion - A simple agent framework with behavior tracking."""
 
-__version__ = "0.5.9"
+__version__ = "0.6.0"
 
 # Auto-load .env files for the entire framework
 from dotenv import load_dotenv
@@ -10,20 +10,8 @@ from pathlib import Path as _Path
 # NOT from the module's location (framework directory)
 load_dotenv(_Path.cwd() / ".env")
 
-from .agent import Agent
-from .tool_factory import create_tool_from_function
-from .llm import LLM
-from .logger import Logger
-from .llm_do import llm_do
-from .transcribe import transcribe
-from .prompts import load_system_prompt
-from .xray import xray
-from .decorators import replay, xray_replay
-from .useful_tools import send_email, get_emails, mark_read, mark_unread, Memory, Gmail, GoogleCalendar, Outlook, MicrosoftCalendar, WebFetch, Shell, DiffWriter, pick, yes_no, autocomplete, TodoList, SlashCommand
-from .auto_debug_exception import auto_debug_exception
-from .connect import connect, RemoteAgent
-from .host import host, create_app
-from .events import (
+from .core import Agent, LLM, create_tool_from_function
+from .core import (
     after_user_input,
     before_llm,
     after_llm,
@@ -32,8 +20,17 @@ from .events import (
     after_each_tool,
     after_tools,
     on_error,
-    on_complete
+    on_complete,
 )
+from .logger import Logger
+from .llm_do import llm_do
+from .transcribe import transcribe
+from .prompts import load_system_prompt
+from .debug import xray, auto_debug_exception, replay, xray_replay
+from .useful_tools import send_email, get_emails, mark_read, mark_unread, Memory, Gmail, GoogleCalendar, Outlook, MicrosoftCalendar, WebFetch, Shell, DiffWriter, pick, yes_no, autocomplete, TodoList, SlashCommand
+from .network import connect, RemoteAgent, host, create_app
+from .network import relay, announce
+from . import address
 
 __all__ = [
     "Agent",
@@ -68,6 +65,9 @@ __all__ = [
     "RemoteAgent",
     "host",
     "create_app",
+    "relay",
+    "announce",
+    "address",
     "after_user_input",
     "before_llm",
     "after_llm",

connectonion/cli/commands/copy_commands.py

@@ -39,6 +39,18 @@ PLUGINS = {
     "calendar_plugin": "calendar_plugin.py",
 }
 
+# Registry of copyable TUI components
+TUI = {
+    "chat": "chat.py",
+    "fuzzy": "fuzzy.py",
+    "divider": "divider.py",
+    "footer": "footer.py",
+    "status_bar": "status_bar.py",
+    "dropdown": "dropdown.py",
+    "pick": "pick.py",
+    "keys": "keys.py",
+}
+
 
 def handle_copy(
     names: List[str],
@@ -56,9 +68,11 @@ def handle_copy(
     # Get source directories using import system (works for installed packages)
     import connectonion.useful_tools as tools_module
     import connectonion.useful_plugins as plugins_module
+    import connectonion.tui as tui_module
 
     useful_tools_dir = Path(tools_module.__file__).parent
     useful_plugins_dir = Path(plugins_module.__file__).parent
+    tui_dir = Path(tui_module.__file__).parent
 
     current_dir = Path.cwd()
 
@@ -77,6 +91,12 @@ def handle_copy(
             dest_dir = Path(path) if path else current_dir / "plugins"
             copy_file(source, dest_dir, force)
 
+        # Check if it's a TUI component
+        elif name_lower in TUI:
+            source = tui_dir / TUI[name_lower]
+            dest_dir = Path(path) if path else current_dir / "tui"
+            copy_file(source, dest_dir, force)
+
         else:
             console.print(f"[red]Unknown: {name}[/red]")
             console.print("Use [cyan]co copy --list[/cyan] to see available items")
@@ -100,7 +120,7 @@ def copy_file(source: Path, dest_dir: Path, force: bool):
 
 
 def show_available_items():
-    """Display available tools and plugins."""
+    """Display available tools, plugins, and TUI components."""
     table = Table(title="Available Items to Copy")
     table.add_column("Name", style="cyan")
     table.add_column("Type", style="green")
@@ -112,5 +132,8 @@ def show_available_items():
     for name, file in sorted(PLUGINS.items()):
         table.add_row(name, "plugin", file)
 
+    for name, file in sorted(TUI.items()):
+        table.add_row(name, "tui", file)
+
     console.print(table)
     console.print("\n[dim]Usage: co copy <name> [--path ./custom/][/dim]")

connectonion/cli/commands/deploy_commands.py

@@ -217,4 +217,19 @@ def handle_deploy():
     # Always show URL if we have one
     if url:
         console.print(f"Agent URL: {url}")
+
+    # Always fetch and display container logs
+    if deployment_id:
+        logs_resp = requests.get(
+            f"{API_BASE}/api/v1/deploy/{deployment_id}/logs?tail=20",
+            headers={"Authorization": f"Bearer {api_key}"},
+            timeout=10,
+        )
+        if logs_resp.status_code == 200:
+            logs = logs_resp.json().get("logs", "")
+            if logs:
+                console.print()
+                console.print("[dim]Container logs:[/dim]")
+                console.print(f"[dim]{logs}[/dim]")
+
     console.print()

connectonion/cli/commands/project_cmd_lib.py

@@ -622,7 +622,7 @@ def generate_custom_template_with_name(description: str, api_key: str, model: st
     # Try to use AI to generate name and code
     if model or api_key:
         try:
-            from ...llm import create_llm
+            from ...core.llm import create_llm
 
             # Use the model specified or default to co/gemini-2.5-pro
             llm_model = model if model else "co/gemini-2.5-pro"

connectonion/core/__init__.py

@@ -0,0 +1,53 @@
+"""Core agent execution engine.
+
+This module contains the minimal set of components needed to run an agent:
+- Agent: Main orchestrator
+- LLM: Multi-provider LLM abstraction
+- Events: Event system for lifecycle hooks
+- Tools: Tool execution, factory, and registry
+- Usage: Token tracking and cost calculation
+"""
+
+from .agent import Agent
+from .llm import LLM, create_llm, TokenUsage
+from .events import (
+    EventHandler,
+    after_user_input,
+    before_llm,
+    after_llm,
+    before_each_tool,
+    before_tools,
+    after_each_tool,
+    after_tools,
+    on_error,
+    on_complete,
+)
+from .tool_factory import create_tool_from_function, extract_methods_from_instance, is_class_instance
+from .tool_registry import ToolRegistry
+from .tool_executor import execute_and_record_tools, execute_single_tool
+from .usage import TokenUsage, calculate_cost, get_context_limit
+
+__all__ = [
+    "Agent",
+    "LLM",
+    "create_llm",
+    "TokenUsage",
+    "EventHandler",
+    "after_user_input",
+    "before_llm",
+    "after_llm",
+    "before_each_tool",
+    "before_tools",
+    "after_each_tool",
+    "after_tools",
+    "on_error",
+    "on_complete",
+    "create_tool_from_function",
+    "extract_methods_from_instance",
+    "is_class_instance",
+    "ToolRegistry",
+    "execute_and_record_tools",
+    "execute_single_tool",
+    "calculate_cost",
+    "get_context_limit",
+]

connectonion/{agent.py → core/agent.py}

@@ -18,11 +18,11 @@ from .llm import LLM, create_llm, TokenUsage
 from .usage import get_context_limit
 from .tool_factory import create_tool_from_function, extract_methods_from_instance, is_class_instance
 from .tool_registry import ToolRegistry
-from .prompts import load_system_prompt
-from .decorators import (
+from ..prompts import load_system_prompt
+from ..debug.decorators import (
     _is_replay_enabled  # Only need this for replay check
 )
-from .logger import Logger
+from ..logger import Logger
 from .tool_executor import execute_and_record_tools, execute_single_tool
 from .events import EventHandler
 
@@ -444,7 +444,7 @@ class Agent:
             # Single prompt mode
             agent.auto_debug("Find information about Python")
         """
-        from .interactive_debugger import InteractiveDebugger
-        debugger = InteractiveDebugger(self)
+        from .debug import AutoDebugger
+        debugger = AutoDebugger(self)
         debugger.start_debug_session(prompt)
 

connectonion/{tool_executor.py → core/tool_executor.py}

@@ -13,7 +13,7 @@ import time
 import json
 from typing import List, Dict, Any, Optional, Callable
 
-from .xray import (
+from ..debug.xray import (
     inject_xray_context,
     clear_xray_context,
     is_xray_enabled
@@ -165,7 +165,8 @@ def execute_single_tool(
         agent.current_session['pending_tool'] = {
             'name': tool_name,
             'arguments': tool_args,
-            'id': tool_id
+            'id': tool_id,
+            'description': getattr(tool_func, 'description', '')
         }
 
         # Invoke before_each_tool events

connectonion/{tool_factory.py → core/tool_factory.py}

@@ -29,7 +29,9 @@ def create_tool_from_function(func: Callable) -> Callable:
     by inspecting its signature and docstring.
     """
     name = func.__name__
-    description = inspect.getdoc(func) or f"Execute the {name} tool."
+    raw_doc = inspect.getdoc(func)
+    # Extract summary only (first paragraph) - Args/Returns sections are not sent to LLM
+    description = raw_doc.split('\n\n')[0].strip() if raw_doc else f"Execute the {name} tool."
 
     # Build the parameters schema from the function signature
     sig = inspect.signature(func)

connectonion/debug/__init__.py

@@ -0,0 +1,51 @@
+"""Debug tools for agent development and troubleshooting.
+
+This module contains:
+- xray: Runtime context injection for tool inspection
+- decorators: replay, xray_replay for debugging
+- auto_debug: Interactive debugging with breakpoints (AutoDebugger, AutoDebugUI)
+- auto_debug_exception: Exception handling and debugging
+- runtime_inspector: AI-powered runtime state inspection for crash debugging
+- debug_explainer: AI-powered explanation of tool choices
+
+Note: Uses lazy imports to avoid circular dependency with agent.py
+"""
+
+# xray and decorators can be imported eagerly (no circular dependency)
+from .xray import xray
+from .decorators import replay, xray_replay
+
+__all__ = [
+    "xray",
+    "replay",
+    "xray_replay",
+    "AutoDebugger",
+    "AutoDebugUI",
+    "BreakpointContext",
+    "BreakpointAction",
+    "auto_debug_exception",
+    "create_debug_agent",
+    "RuntimeInspector",
+    "explain_tool_choice",
+    "RuntimeContext",
+]
+
+
+def __getattr__(name):
+    """Lazy import to avoid circular dependency with agent.py."""
+    if name == "AutoDebugger":
+        from .auto_debug import AutoDebugger
+        return AutoDebugger
+    elif name in ("AutoDebugUI", "BreakpointContext", "BreakpointAction"):
+        from .auto_debug_ui import AutoDebugUI, BreakpointContext, BreakpointAction
+        return {"AutoDebugUI": AutoDebugUI, "BreakpointContext": BreakpointContext, "BreakpointAction": BreakpointAction}[name]
+    elif name == "auto_debug_exception":
+        from .auto_debug_exception import auto_debug_exception
+        return auto_debug_exception
+    elif name in ("create_debug_agent", "RuntimeInspector"):
+        from .runtime_inspector import create_debug_agent, RuntimeInspector
+        return {"create_debug_agent": create_debug_agent, "RuntimeInspector": RuntimeInspector}[name]
+    elif name in ("explain_tool_choice", "RuntimeContext"):
+        from .debug_explainer import explain_tool_choice, RuntimeContext
+        return {"explain_tool_choice": explain_tool_choice, "RuntimeContext": RuntimeContext}[name]
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")

connectonion/{interactive_debugger.py → debug/auto_debug.py}

@@ -13,25 +13,25 @@ LLM-Note:
 """
 
 from typing import Any, Dict, Optional, List
-from .debugger_ui import DebuggerUI, BreakpointContext, BreakpointAction
+from .auto_debug_ui import AutoDebugUI, BreakpointContext, BreakpointAction
 
 
-class InteractiveDebugger:
+class AutoDebugger:
     """Orchestrates debugging sessions for AI agents.
 
     This class handles the debugging logic and intercepts tool execution,
     delegating all UI interactions to the DebuggerUI class.
     """
 
-    def __init__(self, agent: Any, ui: Optional[DebuggerUI] = None):
+    def __init__(self, agent: Any, ui: Optional[AutoDebugUI] = None):
         """Initialize debugger with an agent instance and optional UI.
 
         Args:
             agent: The Agent instance to debug
-            ui: Optional DebuggerUI instance (creates default if None)
+            ui: Optional AutoDebugUI instance (creates default if None)
         """
         self.agent = agent
-        self.ui = ui or DebuggerUI()
+        self.ui = ui or AutoDebugUI()
         self.original_execute_single_tool = None
 
     def start_debug_session(self, prompt: Optional[str] = None):
@@ -99,7 +99,7 @@ class InteractiveDebugger:
         - Pause execution and show UI if breakpoint conditions are met
         - Only affect this specific agent instance
         """
-        from . import tool_executor
+        from ..core import tool_executor
         from .xray import is_xray_enabled
 
         # Store original function for restoration later
@@ -141,7 +141,7 @@ class InteractiveDebugger:
         tool execution function.
         """
         if self.original_execute_single_tool:
-            from . import tool_executor
+            from ..core import tool_executor
             tool_executor.execute_single_tool = self.original_execute_single_tool
 
     def _show_breakpoint_ui_and_wait_for_continue(self, tool_name: str, tool_args: Dict, trace_entry: Dict):

connectonion/{auto_debug_exception.py → debug/auto_debug_exception.py}

@@ -57,7 +57,7 @@ def auto_debug_exception(model: str = "o4-mini"):
         original_hook(exc_type, exc_value, exception_traceback)
 
         # Then add our AI analysis
-        from .console import Console
+        from ..console import Console
         console = Console()
 
         # Find the most relevant frame (last user code, not library)
@@ -125,7 +125,7 @@ def auto_debug_exception(model: str = "o4-mini"):
 
         try:
             # Use debug agent with runtime inspection tools
-            from .debug_agent import create_debug_agent
+            from .runtime_inspector import create_debug_agent
 
             # Pass the actual frame and traceback for runtime inspection!
             agent = create_debug_agent(
@@ -176,6 +176,6 @@ You have LIVE ACCESS to the crashed program's state! Use your tools to investiga
     sys.excepthook = handle_exception
 
     # Simple confirmation
-    from .console import Console
+    from ..console import Console
     console = Console()
     console.print(f"[green]✅ Exception debugging enabled[/green] - AI will analyze uncaught exceptions with runtime inspection")

connectonion/{debugger_ui.py → debug/auto_debug_ui.py}

@@ -51,7 +51,7 @@ class BreakpointContext:
     tool_function: Optional[Any] = None  # The actual tool function for source inspection
 
 
-class DebuggerUI:
+class AutoDebugUI:
     """Handles all user interaction and display for the debugger."""
 
     def __init__(self):

connectonion/{debug_explainer → debug/debug_explainer}/explain_agent.py

@@ -31,7 +31,7 @@ def explain_tool_choice(
     Returns:
         Explanation string from the AI agent
     """
-    from ..agent import Agent
+    from ...core.agent import Agent
     import inspect
 
     # Get all the information we need

connectonion/{debug_explainer → debug/debug_explainer}/explain_context.py

@@ -12,7 +12,7 @@ LLM-Note:
 from typing import Dict, List, Any
 from pathlib import Path
 from pydantic import BaseModel
-from ..llm_do import llm_do
+from ...llm_do import llm_do
 
 
 class RootCauseAnalysis(BaseModel):

connectonion/{execution_analyzer → debug/execution_analyzer}/execution_analysis.py

@@ -16,7 +16,7 @@ Analyzes the entire execution trace and provides suggestions for improvement.
 from pathlib import Path
 from pydantic import BaseModel
 from typing import List
-from ..llm_do import llm_do
+from ...llm_do import llm_do
 
 
 class ExecutionAnalysis(BaseModel):

connectonion/debug/runtime_inspector/__init__.py

@@ -0,0 +1,13 @@
+"""Runtime inspector for AI-powered crash debugging.
+
+Provides RuntimeInspector class and factory function to create debug agents
+that can experiment, test, and validate fixes using actual crashed program data.
+"""
+
+from .agent import create_debug_agent
+from .runtime_inspector import RuntimeInspector
+
+__all__ = [
+    "create_debug_agent",
+    "RuntimeInspector"
+]

connectonion/{debug_agent → debug/runtime_inspector}/agent.py

@@ -10,7 +10,7 @@ LLM-Note:
 """
 
 from pathlib import Path
-from ..agent import Agent
+from ...core.agent import Agent
 from .runtime_inspector import RuntimeInspector
 
 

connectonion/{xray.py → debug/xray.py}

@@ -14,7 +14,7 @@ This module provides the @xray decorator and xray context for debugging AI agent
 See everything your agent is thinking during tool execution.
 
 Usage:
-    from connectonion.xray import xray
+    from connectonion import xray  # or: from connectonion.debug import xray
 
     @xray
     def my_tool(query: str):

connectonion/llm_do.py

@@ -222,7 +222,7 @@ from typing import Union, Type, Optional, TypeVar
 from pathlib import Path
 from pydantic import BaseModel
 from .prompts import load_system_prompt
-from .llm import create_llm
+from .core.llm import create_llm
 
 T = TypeVar('T', bound=BaseModel)
 

connectonion/network/__init__.py

@@ -0,0 +1,34 @@
+"""Network layer for hosting and connecting agents.
+
+This module contains:
+- host: Host an agent over HTTP/WebSocket
+- asgi: ASGI app implementation
+- relay: Agent relay server for P2P discovery
+- connect: Multi-agent networking (RemoteAgent)
+- announce: Service announcement protocol
+- trust: Trust verification system
+"""
+
+from .host import host, create_app, SessionStorage, Session
+from .connect import connect, RemoteAgent
+from .relay import connect as relay_connect, serve_loop
+from .announce import create_announce_message
+from .trust import create_trust_agent, get_default_trust_level, TRUST_LEVELS
+from . import relay, announce
+
+__all__ = [
+    "host",
+    "create_app",
+    "SessionStorage",
+    "Session",
+    "connect",
+    "RemoteAgent",
+    "relay_connect",
+    "serve_loop",
+    "create_announce_message",
+    "create_trust_agent",
+    "get_default_trust_level",
+    "TRUST_LEVELS",
+    "relay",
+    "announce",
+]

connectonion/{announce.py → network/announce.py}

@@ -72,7 +72,7 @@ def create_announce_message(
     message_bytes = message_json.encode('utf-8')
 
     # Sign with Ed25519
-    from . import address
+    from .. import address
     signature_bytes = address.sign(address_data, message_bytes)
 
     # Convert to hex string (NO 0x prefix - matches auth system convention)

connectonion/{connect.py → network/connect.py}

@@ -19,7 +19,7 @@ import time
 import uuid
 from typing import Any, Dict, List, Optional
 
-from . import address as addr
+from .. import address as addr
 
 
 class RemoteAgent: