aury-agent 0.0.4__py3-none-any.whl

aury/agents/context_providers/skill.py

@@ -0,0 +1,50 @@
+"""SkillContextProvider - provides skill definitions."""
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from .base import BaseContextProvider, AgentContext
+
+if TYPE_CHECKING:
+    from ..core.context import InvocationContext
+    from ..skill import SkillLoader
+
+
+class SkillContextProvider(BaseContextProvider):
+    """Skill context provider.
+    
+    Provides skill definitions and formatted descriptions.
+    
+    Design: Provider provides skills list AND system_content.
+    Agent can use skills list for further processing.
+    """
+    
+    _name = "skills"
+    
+    def __init__(self, loader: "SkillLoader"):
+        """Initialize SkillContextProvider.
+        
+        Args:
+            loader: Skill loader for formatting skills
+        """
+        self.loader = loader
+    
+    async def fetch(self, ctx: "InvocationContext") -> AgentContext:
+        """Fetch skill context.
+        
+        Returns:
+            AgentContext with skills list and system_content
+        """
+        skills = self.loader.skills
+        skills_prompt = self.loader.format_for_prompt()
+        
+        if not skills:
+            return AgentContext.empty()
+        
+        return AgentContext(
+            skills=list(skills),
+            system_content=skills_prompt if skills_prompt else None,
+        )
+
+
+__all__ = ["SkillContextProvider"]

aury/agents/context_providers/subagent.py

@@ -0,0 +1,46 @@
+"""SubAgentContextProvider - provides sub-agent configurations."""
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from .base import BaseContextProvider, AgentContext
+
+if TYPE_CHECKING:
+    from ..core.context import InvocationContext
+    from ..backends.subagent import SubAgentBackend
+
+
+class SubAgentContextProvider(BaseContextProvider):
+    """Sub-agent context provider.
+    
+    Provides sub-agent configurations. Agent creates DelegateTool from this.
+    
+    Design: Provider only provides DATA (subagents list).
+    Agent decides how to use it (create DelegateTool).
+    """
+    
+    _name = "subagents"
+    
+    def __init__(self, backend: "SubAgentBackend"):
+        """Initialize SubAgentContextProvider.
+        
+        Args:
+            backend: Sub-agent backend for listing available agents
+        """
+        self.backend = backend
+    
+    async def fetch(self, ctx: "InvocationContext") -> AgentContext:
+        """Fetch sub-agent configurations.
+        
+        Returns:
+            AgentContext with subagents list (NOT DelegateTool)
+        """
+        agents = await self.backend.list()
+        if not agents:
+            return AgentContext.empty()
+        
+        # Return subagents list - Agent will create DelegateTool
+        return AgentContext(subagents=list(agents))
+
+
+__all__ = ["SubAgentContextProvider"]

aury/agents/context_providers/tool.py

@@ -0,0 +1,68 @@
+"""ToolContextProvider - example of dynamic tool discovery.
+
+NOTE: This provider is commented out as an EXAMPLE.
+In most cases, tools should be passed directly to ReactAgent.create(tools=[...]).
+
+This provider is useful for:
+- Dynamic tool discovery (MCP, database, etc.)
+- Permission-based tool filtering
+- Runtime tool injection
+
+Usage:
+    class MCPToolContextProvider(BaseContextProvider):
+        async def fetch(self, ctx):
+            # Discover tools from MCP server
+            tools = await self.mcp_client.list_tools()
+            return AgentContext(tools=tools)
+"""
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Any
+
+from .base import BaseContextProvider, AgentContext
+
+if TYPE_CHECKING:
+    from ..core.context import InvocationContext
+    from ..core.types.tool import BaseTool
+
+
+# =============================================================================
+# ToolContextProvider - Example (commented out)
+# =============================================================================
+# 
+# class ToolContextProvider(BaseContextProvider):
+#     """Tool context provider - for dynamic tool discovery.
+#     
+#     NOTE: Static tools should be passed directly to Agent.
+#     This provider is for DYNAMIC tool discovery only.
+#     
+#     Example use cases:
+#     1. MCP server - discover tools at runtime
+#     2. Permission-based - filter tools by user permissions
+#     3. Database - load tools from database
+#     
+#     Usage:
+#         class MCPToolContextProvider(ToolContextProvider):
+#             async def discover(self, ctx):
+#                 return await self.mcp_client.list_tools()
+#     """
+#     
+#     _name = "tools"
+#     
+#     async def fetch(self, ctx: "InvocationContext") -> AgentContext:
+#         """Fetch tools dynamically."""
+#         discovered = await self.discover(ctx)
+#         return AgentContext(tools=discovered)
+#     
+#     async def discover(self, ctx: "InvocationContext") -> list["BaseTool"]:
+#         """Override this for custom discovery logic.
+#         
+#         Examples:
+#         - Fetch from MCP server
+#         - Filter by user permissions
+#         - Load from database
+#         """
+#         return []
+
+
+__all__: list[str] = []  # No exports - this is an example file

aury/agents/core/__init__.py

@@ -0,0 +1,83 @@
+"""Core module - shared infrastructure for agents."""
+from .base import AgentConfig, BaseAgent, ToolInjectionMode
+from .context import InvocationContext, MaxDepthExceededError
+from .factory import AgentFactory
+from .isolator import StateIsolator, ChainMapIsolator
+from .logging import logger
+from .runner import Runner
+from .event_bus import EventBus, Events
+from .signals import SuspendSignal, HITLSuspend, PauseSuspend
+from .parallel import (
+    merge_agent_runs,
+    run_agents_parallel,
+    ParallelSubAgentContext,
+)
+from .types import (
+    Session,
+    Invocation,
+    InvocationState,
+    InvocationMode,
+    ControlFrame,
+    generate_id,
+    PromptInput,
+    Message,
+    MessageRole,
+    BlockEvent,
+    BlockKind,
+    BlockOp,
+    BlockHandle,
+    ToolContext,
+    ToolResult,
+    ToolInfo,
+    BaseTool,
+)
+
+__all__ = [
+    # Base
+    "AgentConfig",
+    "BaseAgent",
+    "ToolInjectionMode",
+    # Context
+    "InvocationContext",
+    "MaxDepthExceededError",
+    # Factory
+    "AgentFactory",
+    # Isolator
+    "StateIsolator",
+    "ChainMapIsolator",
+    # Logging
+    "logger",
+    # Runner
+    "Runner",
+    "EventBus",
+    "Events",
+    # Signals
+    "SuspendSignal",
+    "HITLSuspend",
+    "PauseSuspend",
+    # Parallel
+    "merge_agent_runs",
+    "run_agents_parallel",
+    "ParallelSubAgentContext",
+    # Types - Session
+    "Session",
+    "Invocation",
+    "InvocationState",
+    "InvocationMode",
+    "ControlFrame",
+    "generate_id",
+    # Types - Message
+    "PromptInput",
+    "Message",
+    "MessageRole",
+    # Types - Block
+    "BlockEvent",
+    "BlockKind",
+    "BlockOp",
+    "BlockHandle",
+    # Types - Tool
+    "ToolContext",
+    "ToolResult",
+    "ToolInfo",
+    "BaseTool",
+]