aip-agents-binary 0.5.10__py3-none-macosx_13_0_arm64.whl → 0.5.12__py3-none-macosx_13_0_arm64.whl

aip_agents/a2a/server/langgraph_executor.py

@@ -255,6 +255,7 @@ class LangGraphA2AExecutor(BaseA2AExecutor, ABC):
             return []
 
         normalized_files: list[str | dict[str, Any]] = []
+        invalid_entry_logged = False
         for entry in raw_files:
             if isinstance(entry, str) and entry:
                 normalized_files.append(entry)
@@ -262,6 +263,8 @@ class LangGraphA2AExecutor(BaseA2AExecutor, ABC):
             if isinstance(entry, dict):
                 normalized_files.append(entry)
                 continue
-            logger.warning("Invalid file metadata entry received; expected string or dict.")
+            if not invalid_entry_logged:
+                logger.warning("Invalid file metadata entry received; expected string or dict.")
+                invalid_entry_logged = True
 
         return normalized_files

aip_agents/agent/hitl/manager.py

@@ -15,6 +15,7 @@ from typing import Any
 
 from aip_agents.agent.hitl.config import ToolApprovalConfig
 from aip_agents.agent.hitl.prompt import BasePromptHandler, DeferredPromptHandler
+from aip_agents.agent.hitl.registry import hitl_registry
 from aip_agents.schema.hitl import ApprovalDecision, ApprovalDecisionType, ApprovalLogEntry, ApprovalRequest
 
 # Constants
@@ -197,6 +198,9 @@ class ApprovalManager:
 
         self._active_requests[request.request_id] = request
 
+        # Register ownership in global registry for hierarchical routing
+        hitl_registry.register(request.request_id, self)
+
     def get_pending_request(self, request_id: str) -> ApprovalRequest | None:
         """Get a pending approval request by ID.
 
@@ -312,6 +316,10 @@ class ApprovalManager:
             decision.latency_ms = int(latency.total_seconds() * 1000)
 
         self._active_requests.pop(request.request_id, None)
+
+        # Unregister from global registry when finalized
+        hitl_registry.unregister(request.request_id)
+
         return decision
 
     async def prompt_for_decision(
@@ -459,6 +467,8 @@ class ApprovalManager:
 
         for request_id in expired_ids:
             self._active_requests.pop(request_id, None)
+            # Unregister expired requests from global registry
+            hitl_registry.unregister(request_id)
 
         return len(expired_ids)
 

aip_agents/agent/hitl/manager.pyi

@@ -1,6 +1,7 @@
 from _typeshed import Incomplete
 from aip_agents.agent.hitl.config import ToolApprovalConfig as ToolApprovalConfig
 from aip_agents.agent.hitl.prompt import BasePromptHandler as BasePromptHandler, DeferredPromptHandler as DeferredPromptHandler
+from aip_agents.agent.hitl.registry import hitl_registry as hitl_registry
 from aip_agents.schema.hitl import ApprovalDecision as ApprovalDecision, ApprovalDecisionType as ApprovalDecisionType, ApprovalLogEntry as ApprovalLogEntry, ApprovalRequest as ApprovalRequest
 from collections.abc import Callable as Callable, Iterable
 from typing import Any

aip_agents/agent/hitl/registry.py

@@ -0,0 +1,149 @@
+"""Registry for tracking HITL approval manager ownership across agent hierarchies.
+
+This module provides a thread-safe singleton registry that maps request IDs to their
+owning ApprovalManager instances. This enables proper decision routing in hierarchical
+agent architectures where sub-agents have HITL enabled but parents do not.
+
+Authors:
+    Raymond Christopher (raymond.christopher@gdplabs.id)
+"""
+
+from __future__ import annotations
+
+import threading
+from typing import TYPE_CHECKING
+from weakref import WeakValueDictionary
+
+if TYPE_CHECKING:
+    from aip_agents.agent.hitl.manager import ApprovalManager
+
+
+class HITLManagerRegistry:
+    """Global registry mapping request_id → owning ApprovalManager.
+
+    Uses weak references to avoid preventing manager garbage collection.
+    Thread-safe for concurrent agent execution.
+
+    This singleton registry allows the HITL decision endpoint to route decisions
+    to the correct manager in hierarchical agent setups, where a sub-agent may
+    create a pending request but the parent agent receives the decision.
+
+    Example:
+        >>> from aip_agents.agent.hitl.registry import hitl_registry
+        >>> # Manager auto-registers when creating requests
+        >>> manager.create_approval_request(...)
+        >>> # Later, decision handler looks up the owning manager
+        >>> owning_manager = hitl_registry.get_manager(request_id)
+        >>> owning_manager.resolve_pending_request(request_id, "approved")
+    """
+
+    _instance: HITLManagerRegistry | None = None
+    _lock = threading.RLock()
+    _registry: WeakValueDictionary[str, ApprovalManager]
+    _registry_lock: threading.RLock
+
+    def __new__(cls) -> HITLManagerRegistry:
+        """Ensure only one instance exists (singleton pattern).
+
+        Returns:
+            The singleton HITLManagerRegistry instance.
+        """
+        if cls._instance is None:
+            with cls._lock:
+                if cls._instance is None:
+                    instance = super().__new__(cls)
+                    instance._registry = WeakValueDictionary()
+                    instance._registry_lock = threading.RLock()
+                    cls._instance = instance
+        return cls._instance
+
+    def register(self, request_id: str, manager: ApprovalManager) -> None:
+        """Register a pending request with its owning manager.
+
+        This method is typically called automatically by ApprovalManager when
+        creating a new pending request. It establishes the ownership mapping
+        needed for proper decision routing.
+
+        Args:
+            request_id: Unique identifier for the pending request
+            manager: ApprovalManager instance that owns this request
+
+        Example:
+            >>> hitl_registry.register("req_abc123", my_manager)
+        """
+        with self._registry_lock:
+            self._registry[request_id] = manager
+
+    def unregister(self, request_id: str) -> None:
+        """Remove a request from the registry.
+
+        This method is called when a request is resolved, times out, or expires.
+        It's important for cleanup to prevent the registry from growing unbounded.
+
+        Args:
+            request_id: Unique identifier to remove
+
+        Example:
+            >>> hitl_registry.unregister("req_abc123")
+        """
+        with self._registry_lock:
+            self._registry.pop(request_id, None)
+
+    def get_manager(self, request_id: str) -> ApprovalManager | None:
+        """Retrieve the manager owning a specific request.
+
+        This is the primary lookup method used by decision handlers to route
+        decisions to the correct manager in hierarchical agent setups.
+
+        Args:
+            request_id: Unique identifier to lookup
+
+        Returns:
+            ApprovalManager instance if found, None otherwise. None can indicate
+            the request was resolved, timed out, or the manager was garbage collected.
+
+        Example:
+            >>> manager = hitl_registry.get_manager("req_abc123")
+            >>> if manager:
+            ...     manager.resolve_pending_request("req_abc123", "approved")
+        """
+        with self._registry_lock:
+            return self._registry.get(request_id)
+
+    def clear(self) -> None:
+        """Clear all registrations.
+
+        This method is primarily intended for testing to ensure a clean state
+        between test cases. It removes all registered mappings.
+
+        Warning:
+            This should not be called in production code as it will prevent
+            pending requests from being resolved.
+
+        Example:
+            >>> hitl_registry.clear()  # For testing only
+        """
+        with self._registry_lock:
+            self._registry.clear()
+
+    def list_all(self) -> list[str]:
+        """List all currently registered request IDs.
+
+        Returns:
+            List of request IDs currently in the registry.
+
+        Note:
+            Due to weak references, managers may be garbage collected between
+            calling this method and accessing them, so the actual available
+            managers might be fewer than the returned list length.
+
+        Example:
+            >>> request_ids = hitl_registry.list_all()
+            >>> print(f"Pending requests: {request_ids}")
+        """
+        with self._registry_lock:
+            return list(self._registry.keys())
+
+
+# Global singleton instance
+hitl_registry = HITLManagerRegistry()

aip_agents/agent/hitl/registry.pyi

@@ -0,0 +1,101 @@
+from _typeshed import Incomplete
+from aip_agents.agent.hitl.manager import ApprovalManager as ApprovalManager
+
+class HITLManagerRegistry:
+    '''Global registry mapping request_id → owning ApprovalManager.
+
+    Uses weak references to avoid preventing manager garbage collection.
+    Thread-safe for concurrent agent execution.
+
+    This singleton registry allows the HITL decision endpoint to route decisions
+    to the correct manager in hierarchical agent setups, where a sub-agent may
+    create a pending request but the parent agent receives the decision.
+
+    Example:
+        >>> from aip_agents.agent.hitl.registry import hitl_registry
+        >>> # Manager auto-registers when creating requests
+        >>> manager.create_approval_request(...)
+        >>> # Later, decision handler looks up the owning manager
+        >>> owning_manager = hitl_registry.get_manager(request_id)
+        >>> owning_manager.resolve_pending_request(request_id, "approved")
+    '''
+    def __new__(cls) -> HITLManagerRegistry:
+        """Ensure only one instance exists (singleton pattern).
+
+        Returns:
+            The singleton HITLManagerRegistry instance.
+        """
+    def register(self, request_id: str, manager: ApprovalManager) -> None:
+        '''Register a pending request with its owning manager.
+
+        This method is typically called automatically by ApprovalManager when
+        creating a new pending request. It establishes the ownership mapping
+        needed for proper decision routing.
+
+        Args:
+            request_id: Unique identifier for the pending request
+            manager: ApprovalManager instance that owns this request
+
+        Example:
+            >>> hitl_registry.register("req_abc123", my_manager)
+        '''
+    def unregister(self, request_id: str) -> None:
+        '''Remove a request from the registry.
+
+        This method is called when a request is resolved, times out, or expires.
+        It\'s important for cleanup to prevent the registry from growing unbounded.
+
+        Args:
+            request_id: Unique identifier to remove
+
+        Example:
+            >>> hitl_registry.unregister("req_abc123")
+        '''
+    def get_manager(self, request_id: str) -> ApprovalManager | None:
+        '''Retrieve the manager owning a specific request.
+
+        This is the primary lookup method used by decision handlers to route
+        decisions to the correct manager in hierarchical agent setups.
+
+        Args:
+            request_id: Unique identifier to lookup
+
+        Returns:
+            ApprovalManager instance if found, None otherwise. None can indicate
+            the request was resolved, timed out, or the manager was garbage collected.
+
+        Example:
+            >>> manager = hitl_registry.get_manager("req_abc123")
+            >>> if manager:
+            ...     manager.resolve_pending_request("req_abc123", "approved")
+        '''
+    def clear(self) -> None:
+        """Clear all registrations.
+
+        This method is primarily intended for testing to ensure a clean state
+        between test cases. It removes all registered mappings.
+
+        Warning:
+            This should not be called in production code as it will prevent
+            pending requests from being resolved.
+
+        Example:
+            >>> hitl_registry.clear()  # For testing only
+        """
+    def list_all(self) -> list[str]:
+        '''List all currently registered request IDs.
+
+        Returns:
+            List of request IDs currently in the registry.
+
+        Note:
+            Due to weak references, managers may be garbage collected between
+            calling this method and accessing them, so the actual available
+            managers might be fewer than the returned list length.
+
+        Example:
+            >>> request_ids = hitl_registry.list_all()
+            >>> print(f"Pending requests: {request_ids}")
+        '''
+
+hitl_registry: Incomplete

{aip_agents_binary-0.5.10.dist-info → aip_agents_binary-0.5.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aip-agents-binary
-Version: 0.5.10
+Version: 0.5.12
 Summary: A library for managing agents in Gen AI applications.
 Author-email: Raymond Christopher <raymond.christopher@gdplabs.id>
 Requires-Python: <3.13,>=3.11

{aip_agents_binary-0.5.10.dist-info → aip_agents_binary-0.5.12.dist-info}/RECORD

@@ -14,7 +14,7 @@ aip_agents/a2a/server/google_adk_executor.py,sha256=s6mX7E1od8bDaVrIocEbJAbDnZvm
 aip_agents/a2a/server/google_adk_executor.pyi,sha256=bMU9GxPIA1xkA-sbowbEeDBS4eBgx-bpS-ZD_WX-1Jk,2600
 aip_agents/a2a/server/langflow_executor.py,sha256=6bVnwqRnqbBi949JHHWcIB-ZVQAPGJ84DhAMXnB19co,6646
 aip_agents/a2a/server/langflow_executor.pyi,sha256=nlWUIOEGFcp4qtyrmmxax0TfzwIwOGkvFK-YmG0cGMg,1837
-aip_agents/a2a/server/langgraph_executor.py,sha256=Rmt6u2jyAlJKPWut-W_xZDNevUyFTDwFiKWM1hOcOs8,10639
+aip_agents/a2a/server/langgraph_executor.py,sha256=45XRRwj7EQFBAkS69cZGtuL4ZtHZ0BDEyvDTp4Kw3yg,10765
 aip_agents/a2a/server/langgraph_executor.pyi,sha256=I6RX8_CjF0-gbJEYmNMO2P_i272k3R7X5iZNP5IfdTE,2287
 aip_agents/agent/__init__.py,sha256=KBT-e5nEBMVJypC8OFulmErUK63gmQZus0UcBu6EqBo,892
 aip_agents/agent/__init__.pyi,sha256=MxIAeAv1pPCtqfAa3lvmeCAN-IT5p3v77IeKhfKYvKo,855
@@ -44,10 +44,12 @@ aip_agents/agent/hitl/config.py,sha256=Stzga68KWD5sGQG0-Qn4zS1zSiwdQaelIaHcq9YEU
 aip_agents/agent/hitl/config.pyi,sha256=onrJHVuirFbL5kHgCmHLzrGsHm7pFaCtIsnT9N2vvX8,495
 aip_agents/agent/hitl/langgraph_hitl_mixin.py,sha256=wNgZTJhMoQm8yQiVus1muXmVFBfzCCh6gFO25hCzoz8,19054
 aip_agents/agent/hitl/langgraph_hitl_mixin.pyi,sha256=Hz_3AqvrEjX0iZC7Xia93L4O0UbknvLmUwYA8nLBZ8c,1978
-aip_agents/agent/hitl/manager.py,sha256=YNc-FpPhDUCueaIqMJ582P4gqOTaCjfVemv4jycLlq4,19969
-aip_agents/agent/hitl/manager.pyi,sha256=HlsB_xOoJGXiFNC7rl2RWYv0bOaCHIYxyeI8HbZcHRc,8616
+aip_agents/agent/hitl/manager.py,sha256=4dGu6QEHEzlVnF-bQdRScLdZ8WnGTJPZutz87pmzQtk,20381
+aip_agents/agent/hitl/manager.pyi,sha256=5wiU0MU4Ylf-wJ6Bn1wBDJeiVLFk7i3jD-boIMruf98,8690
 aip_agents/agent/hitl/models.py,sha256=Dwvc_TmkeWse6IAKE38k3IspLXnLFqjerUve4n82KQI,369
 aip_agents/agent/hitl/models.pyi,sha256=1iAKXmv17QDYNiVGbcAlr3fa1g_yvHTzkUbsxDKyy2k,287
+aip_agents/agent/hitl/registry.py,sha256=rDKw_5YbIic361RaxcwGRj9NP7y5KBkLjo2NkQDBK4Y,5390
+aip_agents/agent/hitl/registry.pyi,sha256=d0ylLEf-tPsA08_VfaHjHAhCzDLBMyevgu1u05Hb4qE,3913
 aip_agents/agent/hitl/prompt/__init__.py,sha256=HIF1MpWbtSwgSlqoVdDKEJTfhPq8YDXSIqGE1johshI,263
 aip_agents/agent/hitl/prompt/__init__.pyi,sha256=0ZULcLlR3WZPl7EivrYTVS4f0FaX2LIZeO8jaTX1yyk,240
 aip_agents/agent/hitl/prompt/base.py,sha256=NWMmLsFq_sasMv2WDlANUVWTMU0x4_D_mD2L6CDgn2w,1470
@@ -536,7 +538,7 @@ aip_agents/utils/pii/pii_helper.py,sha256=g0yRzakfA2AA6vjUNLWHqFlcxyLql6MXQ90NN3
 aip_agents/utils/pii/pii_helper.pyi,sha256=dulZs150ikbAL3Bw2YLcz3_g4DsGmL3lciwf8mKxEjI,2939
 aip_agents/utils/pii/uuid_deanonymizer_mapping.py,sha256=Gks8l8t0cuS9pzoQnrpiK1CaLmWYksjOnTeiHh3_7EE,7348
 aip_agents/utils/pii/uuid_deanonymizer_mapping.pyi,sha256=gnWfD1rWZh_tloJjgKiZ6f6iNUuBaHpKqCSiP0d-9bs,3084
-aip_agents_binary-0.5.10.dist-info/METADATA,sha256=EafPczm70GgmRRBtOT6HgTXt_jYdJ3FbiHalAK_Wj8Q,22617
-aip_agents_binary-0.5.10.dist-info/WHEEL,sha256=PaP4PvkDyiSc4C6Dhw6ccQmfxsWFrSv-lJQjBshu0hw,105
-aip_agents_binary-0.5.10.dist-info/top_level.txt,sha256=PEz8vcwC1bH4UrkhF0LkIYCNfXGWZUHdSklbvkBe25E,11
-aip_agents_binary-0.5.10.dist-info/RECORD,,
+aip_agents_binary-0.5.12.dist-info/METADATA,sha256=Q_EJoAONNrRwDF8M8KauU0z2JLMabnKX6dhiSoRD6rY,22617
+aip_agents_binary-0.5.12.dist-info/WHEEL,sha256=PaP4PvkDyiSc4C6Dhw6ccQmfxsWFrSv-lJQjBshu0hw,105
+aip_agents_binary-0.5.12.dist-info/top_level.txt,sha256=PEz8vcwC1bH4UrkhF0LkIYCNfXGWZUHdSklbvkBe25E,11
+aip_agents_binary-0.5.12.dist-info/RECORD,,