aip-agents-binary 0.0.0b2__py3-none-any.whl → 0.5.11__py3-none-any.whl

aip_agents/agent/google_adk_agent.pyi

@@ -1,18 +1,19 @@
 from _typeshed import Incomplete
-from a2a.types import AgentCard as AgentCard
+from a2a.types import AgentCard
 from aip_agents.agent.base_agent import BaseAgent as BaseAgent
-from aip_agents.mcp.client import GoogleADKMCPClient as GoogleADKMCPClient
+from aip_agents.agent.google_adk_constants import DEFAULT_AUTH_URL as DEFAULT_AUTH_URL
+from aip_agents.mcp.client.google_adk.client import GoogleADKMCPClient as GoogleADKMCPClient
 from aip_agents.utils.a2a_connector import A2AConnector as A2AConnector
-from aip_agents.utils.logger_manager import LoggerManager as LoggerManager
+from aip_agents.utils.logger import get_logger as get_logger
+from collections.abc import AsyncGenerator, AsyncIterator
 from google.adk.agents import LlmAgent
 from google.adk.tools.langchain_tool import LangchainTool
 from langchain.tools import BaseTool as LangchainBaseTool
 from pydantic import BaseModel
-from typing import Any, AsyncGenerator, AsyncIterator
+from typing import Any
 
 logger: Incomplete
 MODEL_TEMPERATURE: float
-DEFAULT_AUTH_URL: str
 
 class A2AToolInput(BaseModel):
     """Input for the A2ATool."""
@@ -26,14 +27,28 @@ class A2ATool(LangchainBaseTool):
     agent_card: AgentCard
 
 def create_a2a_tool(agent_card: AgentCard) -> LangchainTool:
-    """Create a LangChain tool from an A2A agent card."""
+    """Create a LangChain tool from an A2A agent card.
+
+    Args:
+        agent_card (AgentCard): The A2A agent card to create a tool for.
+
+    Returns:
+        LangchainTool: A LangChain tool that can communicate with the A2A agent.
+    """
 
 class GoogleADKAgent(BaseAgent):
-    """An agent that wraps a native Google ADK Agent.
+    """An agent that wraps a native Google ADK Agent with MCP support.
 
     This class implements the AgentInterface and uses Google's LlmAgent
     to handle the core conversation and tool execution logic via ADK's
-    async-first design.
+    async-first design. It includes persistent MCP session management for
+    stateful tool execution across multiple calls.
+
+    The agent supports:
+    - Native ADK tools (FunctionTool, LangchainTool)
+    - MCP tools via GoogleADKMCPClient with session persistence
+    - Sub-agent delegation using ADK's built-in multi-agent capabilities
+    - A2A communication through tool integration
     """
     adk_native_agent: LlmAgent
     model: Incomplete
@@ -43,7 +58,7 @@ class GoogleADKAgent(BaseAgent):
     session_service: Incomplete
     name: Incomplete
     def __init__(self, name: str, instruction: str, model: str, tools: list[Any] | None = None, description: str | None = None, max_iterations: int = 3, agents: list['GoogleADKAgent'] | None = None, **kwargs: Any) -> None:
-        '''Initializes the GoogleADKAgent.
+        '''Initializes the GoogleADKAgent with MCP support.
 
         Args:
             name: The name of this wrapper agent.
@@ -73,15 +88,20 @@ class GoogleADKAgent(BaseAgent):
                           or for other unhandled errors during synchronous execution.
         '''
     async def arun(self, query: str, **kwargs: Any) -> dict[str, Any]:
-        '''Asynchronously runs the agent with the given query and returns the response.
+        '''Asynchronously runs the agent with MCP tool support.
+
+        This method ensures MCP tools are properly initialized before execution
+        and provides persistent session management for stateful MCP tools.
 
         Args:
             query: The user\'s query to process.
             **kwargs: Additional keyword arguments. Supports "session_id", "user_id", "app_name".
 
         Returns:
-            A dictionary containing the output and other metadata.
+            A dictionary containing the output, tool_calls, and session_id.
         '''
+    async def cleanup(self) -> None:
+        """Clean up ADK and MCP resources."""
     async def arun_stream(self, query: str, **kwargs: Any) -> AsyncIterator[str]:
         '''Runs the agent with the given query and streams the response parts.
 
@@ -98,18 +118,11 @@ class GoogleADKAgent(BaseAgent):
         This method takes the agents from a2a_config.known_agents, creates A2AAgent
         instances for each one, and wraps them in LangChain tools.
 
-        Returns:
-            None: The tools are added to the existing tools list.
-        """
-    def add_mcp_server(self, mcp_config: dict[str, dict[str, Any]]) -> None:
-        """Adds a new MCP server configuration.
-
         Args:
-            mcp_config: Dictionary containing server name as key and its configuration as value.
+            agent_cards (list[AgentCard]): List of A2A agent cards to register as tools.
 
-        Raises:
-            ValueError: If mcp_config is empty or None, or if any server configuration is invalid.
-            KeyError: If any server name already exists in the configuration.
+        Returns:
+            None: The tools are added to the existing tools list.
         """
     async def arun_a2a_stream(self, query: str, configurable: dict[str, Any] | None = None, **kwargs: Any) -> AsyncGenerator[dict[str, Any], None]:
         '''Asynchronously streams the agent\'s response in a format compatible with A2A.

aip_agents/agent/google_adk_constants.py

@@ -0,0 +1,6 @@
+"""Shared constants for Google ADK agent integrations."""
+
+# Default authentication URL presented when the ADK agent requires auth.
+DEFAULT_AUTH_URL = "https://auth.example.com"
+
+__all__ = ["DEFAULT_AUTH_URL"]

aip_agents/agent/google_adk_constants.pyi

@@ -0,0 +1,3 @@
+__all__ = ['DEFAULT_AUTH_URL']
+
+DEFAULT_AUTH_URL: str

aip_agents/agent/hitl/__init__.py

@@ -0,0 +1,24 @@
+"""Human-in-the-Loop (HITL) approval system for LangGraph tools.
+
+This package provides components for intercepting tool calls and requiring
+human approval before execution. It includes configuration, models, approval
+management, and prompt handlers for deferred approval flows.
+
+Author:
+    Raymond Christopher (raymond.christopher@gdplabs.id)
+"""
+
+from aip_agents.agent.hitl.config import ToolApprovalConfig
+from aip_agents.agent.hitl.manager import ApprovalManager
+from aip_agents.agent.hitl.prompt import BasePromptHandler, DeferredPromptHandler
+from aip_agents.schema.hitl import ApprovalDecision, ApprovalLogEntry, ApprovalRequest
+
+__all__ = [
+    "ToolApprovalConfig",
+    "ApprovalManager",
+    "ApprovalDecision",
+    "ApprovalLogEntry",
+    "ApprovalRequest",
+    "BasePromptHandler",
+    "DeferredPromptHandler",
+]

aip_agents/agent/hitl/__init__.pyi

@@ -0,0 +1,6 @@
+from aip_agents.agent.hitl.config import ToolApprovalConfig as ToolApprovalConfig
+from aip_agents.agent.hitl.manager import ApprovalManager as ApprovalManager
+from aip_agents.agent.hitl.prompt import BasePromptHandler as BasePromptHandler, DeferredPromptHandler as DeferredPromptHandler
+from aip_agents.schema.hitl import ApprovalDecision as ApprovalDecision, ApprovalLogEntry as ApprovalLogEntry, ApprovalRequest as ApprovalRequest
+
+__all__ = ['ToolApprovalConfig', 'ApprovalManager', 'ApprovalDecision', 'ApprovalLogEntry', 'ApprovalRequest', 'BasePromptHandler', 'DeferredPromptHandler']

aip_agents/agent/hitl/config.py

@@ -0,0 +1,28 @@
+"""HITL configuration dataclasses.
+
+This module defines configuration structures for Human-in-the-Loop approval settings.
+
+Author:
+    Raymond Christopher (raymond.christopher@gdplabs.id)
+"""
+
+from dataclasses import dataclass
+
+
+@dataclass
+class ToolApprovalConfig:
+    """Configuration for HITL approval behavior attached to a tool.
+
+    Supplying this configuration marks the tool as requiring approval. Only the
+    timeout is configurable; timeouts always result in a safe skip.
+
+    Attributes:
+        timeout_seconds: Defaults to 300; must be >0.
+    """
+
+    timeout_seconds: int = 300
+
+    def __post_init__(self) -> None:
+        """Validate configuration after initialization."""
+        if self.timeout_seconds <= 0:
+            raise ValueError("timeout_seconds must be > 0")

aip_agents/agent/hitl/config.pyi

@@ -0,0 +1,15 @@
+from dataclasses import dataclass
+
+@dataclass
+class ToolApprovalConfig:
+    """Configuration for HITL approval behavior attached to a tool.
+
+    Supplying this configuration marks the tool as requiring approval. Only the
+    timeout is configurable; timeouts always result in a safe skip.
+
+    Attributes:
+        timeout_seconds: Defaults to 300; must be >0.
+    """
+    timeout_seconds: int = ...
+    def __post_init__(self) -> None:
+        """Validate configuration after initialization."""