aip-agents-binary 0.5.21__py3-none-macosx_13_0_arm64.whl → 0.6.8__py3-none-macosx_13_0_arm64.whl

aip_agents/memory/adapters/base_adapter.py

@@ -256,6 +256,104 @@ class BaseMemoryAdapter(BaseMemory):
 
         return [self._chunk_to_hit(chunk) for chunk in chunks]
 
+    def delete_by_query(
+        self,
+        *,
+        query: str,
+        user_id: str,
+        top_k: int | None = None,
+        threshold: float | None = 0.3,
+        filters: dict[str, Any] | None = None,
+    ) -> list[dict[str, Any]]:
+        """Delete memories matching a query.
+
+        Args:
+            query: The search query string used to find memories to delete.
+            user_id: User identifier for the deletion scope.
+            top_k: Maximum number of memories to delete.
+            threshold: Minimum similarity threshold for deletion.
+            filters: Optional filters to apply to the deletion scope.
+
+        Returns:
+            List of deleted memory hits.
+        """
+        options = self._build_retrieve_options(
+            user_id=user_id,
+            limit=top_k,
+            filters=filters,
+            page=None,
+        )
+        try:
+            start = perf_counter()
+            chunks = self._runner.run(
+                self._manager.delete_by_user_query(
+                    query=query,
+                    user_id=options.user_id,
+                    agent_id=self.agent_id,
+                    scopes=DEFAULT_SCOPE,
+                    metadata=options.metadata,
+                    threshold=threshold,
+                    top_k=options.top_k,
+                )
+            )
+            duration = perf_counter() - start
+            logger.info(
+                "BaseMemoryAdapter: delete_by_query user_id='%s' query='%s' deleted %d hits in %.2fs",
+                user_id,
+                query,
+                len(chunks),
+                duration,
+            )
+        except Exception as exc:  # noqa: BLE001
+            logger.debug("BaseMemoryAdapter.delete_by_query ignored error: %s", exc)
+            return []
+
+        return [self._chunk_to_hit(chunk) for chunk in chunks]
+
+    def delete(
+        self,
+        *,
+        memory_ids: list[str] | None,
+        user_id: str,
+        metadata: dict[str, Any] | None = None,
+        categories: list[str] | None = None,
+    ) -> Any:
+        """Delete memories by IDs or by user scope when IDs are None.
+
+        Args:
+            memory_ids: Optional list of memory IDs to delete.
+            user_id: User identifier for the deletion scope.
+            metadata: Optional metadata filters to constrain deletion.
+            categories: Optional categories to filter by (best-effort).
+
+        Returns:
+            Backend-specific delete result or None on failure.
+        """
+        try:
+            start = perf_counter()
+            result = self._runner.run(
+                self._call_manager_with_optional_categories(
+                    self._manager.delete,
+                    categories=categories,
+                    memory_ids=memory_ids,
+                    user_id=user_id or self.agent_id,
+                    agent_id=self.agent_id,
+                    scopes=DEFAULT_SCOPE,
+                    metadata=metadata,
+                )
+            )
+            duration = perf_counter() - start
+            logger.info(
+                "BaseMemoryAdapter: delete user_id='%s' memory_ids=%s completed in %.2fs",
+                user_id,
+                "None" if memory_ids is None else len(memory_ids),
+                duration,
+            )
+            return result
+        except Exception as exc:  # noqa: BLE001
+            logger.debug("BaseMemoryAdapter.delete ignored error: %s", exc)
+            return None
+
     def save_interaction(self, *, user_text: str, ai_text: str, user_id: str) -> None:
         """Save a user-AI interaction as memories.
 

aip_agents/memory/adapters/base_adapter.pyi

@@ -106,6 +106,31 @@ class BaseMemoryAdapter(BaseMemory):
         Returns:
             List of memory hits matching the criteria.
         """
+    def delete_by_query(self, *, query: str, user_id: str, top_k: int | None = None, threshold: float | None = 0.3, filters: dict[str, Any] | None = None) -> list[dict[str, Any]]:
+        """Delete memories matching a query.
+
+        Args:
+            query: The search query string used to find memories to delete.
+            user_id: User identifier for the deletion scope.
+            top_k: Maximum number of memories to delete.
+            threshold: Minimum similarity threshold for deletion.
+            filters: Optional filters to apply to the deletion scope.
+
+        Returns:
+            List of deleted memory hits.
+        """
+    def delete(self, *, memory_ids: list[str] | None, user_id: str, metadata: dict[str, Any] | None = None, categories: list[str] | None = None) -> Any:
+        """Delete memories by IDs or by user scope when IDs are None.
+
+        Args:
+            memory_ids: Optional list of memory IDs to delete.
+            user_id: User identifier for the deletion scope.
+            metadata: Optional metadata filters to constrain deletion.
+            categories: Optional categories to filter by (best-effort).
+
+        Returns:
+            Backend-specific delete result or None on failure.
+        """
     def save_interaction(self, *, user_text: str, ai_text: str, user_id: str) -> None:
         """Save a user-AI interaction as memories.
 

aip_agents/middleware/base.py

@@ -86,3 +86,11 @@ class AgentMiddleware(Protocol):
             if no updates are needed.
         """
         return {}  # pragma: no cover  # Protocol default - cannot be executed directly
+
+    async def abefore_model(self, state: dict[str, Any]) -> dict[str, Any]:
+        """Asynchronous version of before_model hook."""
+        return self.before_model(state)
+
+    async def aafter_model(self, state: dict[str, Any]) -> dict[str, Any]:
+        """Asynchronous version of after_model hook."""
+        return self.after_model(state)

aip_agents/middleware/base.pyi

@@ -69,3 +69,7 @@ class AgentMiddleware(Protocol):
             Dict of state updates to merge into the agent state. Return empty dict
             if no updates are needed.
         """
+    async def abefore_model(self, state: dict[str, Any]) -> dict[str, Any]:
+        """Asynchronous version of before_model hook."""
+    async def aafter_model(self, state: dict[str, Any]) -> dict[str, Any]:
+        """Asynchronous version of after_model hook."""

aip_agents/middleware/manager.py

@@ -126,3 +126,25 @@ class MiddlewareManager:
                 updates.update(mw_updates)
 
         return updates
+
+    async def abefore_model(self, state: dict[str, Any]) -> dict[str, Any]:
+        """Asynchronously execute before_model hooks for all middleware."""
+        state_updates: dict[str, Any] = {}
+
+        for mw in self.middleware:
+            mw_updates = await mw.abefore_model(state)
+            if mw_updates:
+                state_updates.update(mw_updates)
+
+        return state_updates
+
+    async def aafter_model(self, state: dict[str, Any]) -> dict[str, Any]:
+        """Asynchronously execute after_model hooks for all middleware in reverse order."""
+        updates: dict[str, Any] = {}
+
+        for mw in reversed(self.middleware):
+            mw_updates = await mw.aafter_model(state)
+            if mw_updates:
+                updates.update(mw_updates)
+
+        return updates

aip_agents/middleware/manager.pyi

@@ -78,3 +78,7 @@ class MiddlewareManager:
             Merged dictionary of all state updates from all middleware.
             Updates are accumulated in reverse order of execution.
         """
+    async def abefore_model(self, state: dict[str, Any]) -> dict[str, Any]:
+        """Asynchronously execute before_model hooks for all middleware."""
+    async def aafter_model(self, state: dict[str, Any]) -> dict[str, Any]:
+        """Asynchronously execute after_model hooks for all middleware in reverse order."""

aip_agents/ptc/__init__.py

@@ -0,0 +1,87 @@
+"""PTC (Programmatic Tool Calling) core module.
+
+This module provides core PTC functionality, including executor, prompt builder,
+sandbox bridge, and custom tool configuration validation.
+
+Authors:
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+"""
+
+from aip_agents.ptc.custom_tools import (
+    PTCCustomToolConfig,
+    PTCCustomToolValidationError,
+    PTCFileToolDef,
+    PTCPackageToolDef,
+    PTCToolDef,
+    enrich_tool_def_with_metadata,
+    extract_tool_metadata,
+    validate_custom_tool_config,
+)
+from aip_agents.ptc.custom_tools_payload import (
+    CustomToolPayloadResult,
+    build_custom_tools_payload,
+)
+from aip_agents.ptc.exceptions import PTCError, PTCToolError
+from aip_agents.ptc.prompt_builder import PromptConfig, build_ptc_prompt, compute_ptc_prompt_hash
+from aip_agents.ptc.tool_def_helpers import file_tool, package_tool
+from aip_agents.ptc.tool_enrichment import (
+    build_tool_lookup,
+    enrich_custom_tools_from_agent,
+    match_tool_by_name,
+)
+
+__all__ = [
+    # Exceptions
+    "PTCError",
+    "PTCToolError",
+    "PTCCustomToolValidationError",
+    # Executor
+    "PTCSandboxConfig",
+    "PTCSandboxExecutor",
+    # Custom tools
+    "PTCCustomToolConfig",
+    "PTCToolDef",
+    "PTCPackageToolDef",
+    "PTCFileToolDef",
+    "validate_custom_tool_config",
+    "extract_tool_metadata",
+    "enrich_tool_def_with_metadata",
+    # Tool enrichment
+    "build_tool_lookup",
+    "match_tool_by_name",
+    "enrich_custom_tools_from_agent",
+    # Tool definition helpers
+    "package_tool",
+    "file_tool",
+    # Custom tools payload
+    "CustomToolPayloadResult",
+    "build_custom_tools_payload",
+    # Prompt builder
+    "PromptConfig",
+    "build_ptc_prompt",
+    "compute_ptc_prompt_hash",
+    # Sandbox bridge
+    "build_sandbox_payload",
+    "wrap_ptc_code",
+]
+
+
+def __getattr__(name: str):
+    """Lazy import to avoid circular dependencies."""
+    if name == "PTCSandboxConfig":
+        from aip_agents.ptc.executor import PTCSandboxConfig
+
+        return PTCSandboxConfig
+    elif name == "PTCSandboxExecutor":
+        from aip_agents.ptc.executor import PTCSandboxExecutor
+
+        return PTCSandboxExecutor
+    elif name == "build_sandbox_payload":
+        from aip_agents.ptc.sandbox_bridge import build_sandbox_payload
+
+        return build_sandbox_payload
+    elif name == "wrap_ptc_code":
+        from aip_agents.ptc.sandbox_bridge import wrap_ptc_code
+
+        return wrap_ptc_code
+    raise AttributeError(f"module '{__name__}' has no attribute '{name}'")

aip_agents/ptc/__init__.pyi

@@ -0,0 +1,14 @@
+from aip_agents.ptc.custom_tools import PTCCustomToolConfig as PTCCustomToolConfig, PTCCustomToolValidationError as PTCCustomToolValidationError, PTCFileToolDef as PTCFileToolDef, PTCPackageToolDef as PTCPackageToolDef, PTCToolDef as PTCToolDef, enrich_tool_def_with_metadata as enrich_tool_def_with_metadata, extract_tool_metadata as extract_tool_metadata, validate_custom_tool_config as validate_custom_tool_config
+from aip_agents.ptc.custom_tools_payload import CustomToolPayloadResult as CustomToolPayloadResult, build_custom_tools_payload as build_custom_tools_payload
+from aip_agents.ptc.exceptions import PTCError as PTCError, PTCToolError as PTCToolError
+from aip_agents.ptc.prompt_builder import PromptConfig as PromptConfig, build_ptc_prompt as build_ptc_prompt, compute_ptc_prompt_hash as compute_ptc_prompt_hash
+from aip_agents.ptc.tool_def_helpers import file_tool as file_tool, package_tool as package_tool
+from aip_agents.ptc.tool_enrichment import build_tool_lookup as build_tool_lookup, enrich_custom_tools_from_agent as enrich_custom_tools_from_agent, match_tool_by_name as match_tool_by_name
+
+__all__ = ['PTCError', 'PTCToolError', 'PTCCustomToolValidationError', 'PTCSandboxConfig', 'PTCSandboxExecutor', 'PTCCustomToolConfig', 'PTCToolDef', 'PTCPackageToolDef', 'PTCFileToolDef', 'validate_custom_tool_config', 'extract_tool_metadata', 'enrich_tool_def_with_metadata', 'build_tool_lookup', 'match_tool_by_name', 'enrich_custom_tools_from_agent', 'package_tool', 'file_tool', 'CustomToolPayloadResult', 'build_custom_tools_payload', 'PromptConfig', 'build_ptc_prompt', 'compute_ptc_prompt_hash', 'build_sandbox_payload', 'wrap_ptc_code']
+
+# Names in __all__ with no definition:
+#   PTCSandboxConfig
+#   PTCSandboxExecutor
+#   build_sandbox_payload
+#   wrap_ptc_code