aip-agents-binary 0.6.2__py3-none-macosx_13_0_arm64.whl → 0.6.4__py3-none-macosx_13_0_arm64.whl

aip_agents/agent/langgraph_react_agent.py

@@ -54,7 +54,7 @@ from aip_agents.schema.a2a import A2AStreamEventType
 from aip_agents.schema.hitl import ApprovalDecision, HitlMetadata
 from aip_agents.schema.langgraph import ToolCallResult, ToolStorageParams
 from aip_agents.schema.step_limit import MaxStepsExceededError, StepLimitConfig
-from aip_agents.tools.memory_search_tool import MEMORY_SEARCH_TOOL_NAME
+from aip_agents.tools.memory_search_tool import MEMORY_DELETE_TOOL_NAME, MEMORY_SEARCH_TOOL_NAME
 from aip_agents.tools.tool_config_injector import TOOL_CONFIGS_KEY
 from aip_agents.utils import add_references_chunks
 from aip_agents.utils.langgraph import (
@@ -584,6 +584,29 @@ class LangGraphReactAgent(LangGraphHitLMixin, BaseLangGraphAgent):
 
         return memory_node
 
+    def _should_save_interaction(self, final_state: dict[str, Any] | None) -> bool:
+        """Return True when interaction should be saved to memory."""
+        if self._contains_memory_delete_action(final_state):
+            logger.info("Memory: Skipping save_interaction due to memory delete action in state.")
+            return False
+        return True
+
+    @staticmethod
+    def _contains_memory_delete_action(final_state: dict[str, Any] | None) -> bool:
+        """Return True when final state includes a delete memory action block."""
+        if not isinstance(final_state, dict):
+            return False
+        messages = final_state.get("messages")
+        if not isinstance(messages, list):
+            return False
+        for message in messages:
+            content = getattr(message, "content", None)
+            if not isinstance(content, str):
+                continue
+            if "<MEMORY_ACTION>" in content and "action=delete" in content:
+                return True
+        return False
+
     def _extract_user_query_from_messages(self, messages: list[Any]) -> str | None:
         """Get latest user query string from a list of messages.
 
@@ -2277,11 +2300,12 @@ class LangGraphReactAgent(LangGraphHitLMixin, BaseLangGraphAgent):
         """
         try:
             tool_cfgs = metadata.get(TOOL_CONFIGS_KEY, {})
-            per_tool_config = tool_cfgs.get(MEMORY_SEARCH_TOOL_NAME)
-            if not isinstance(per_tool_config, dict):
-                per_tool_config = {}
-            per_tool_config["user_id"] = memory_user_id
-            tool_cfgs[MEMORY_SEARCH_TOOL_NAME] = per_tool_config
+            for tool_name in (MEMORY_SEARCH_TOOL_NAME, MEMORY_DELETE_TOOL_NAME):
+                per_tool_config = tool_cfgs.get(tool_name)
+                if not isinstance(per_tool_config, dict):
+                    per_tool_config = {}
+                per_tool_config["user_id"] = memory_user_id
+                tool_cfgs[tool_name] = per_tool_config
             metadata[TOOL_CONFIGS_KEY] = tool_cfgs
         except Exception as e:
             # Non-fatal; metadata injection is best-effort

aip_agents/agent/langgraph_react_agent.pyi

@@ -11,7 +11,7 @@ from aip_agents.schema.a2a import A2AStreamEventType as A2AStreamEventType
 from aip_agents.schema.hitl import ApprovalDecision as ApprovalDecision, HitlMetadata as HitlMetadata
 from aip_agents.schema.langgraph import ToolCallResult as ToolCallResult, ToolStorageParams as ToolStorageParams
 from aip_agents.schema.step_limit import MaxStepsExceededError as MaxStepsExceededError, StepLimitConfig as StepLimitConfig
-from aip_agents.tools.memory_search_tool import MEMORY_SEARCH_TOOL_NAME as MEMORY_SEARCH_TOOL_NAME
+from aip_agents.tools.memory_search_tool import MEMORY_DELETE_TOOL_NAME as MEMORY_DELETE_TOOL_NAME, MEMORY_SEARCH_TOOL_NAME as MEMORY_SEARCH_TOOL_NAME
 from aip_agents.tools.tool_config_injector import TOOL_CONFIGS_KEY as TOOL_CONFIGS_KEY
 from aip_agents.utils import add_references_chunks as add_references_chunks
 from aip_agents.utils.langgraph import convert_langchain_messages_to_gllm_messages as convert_langchain_messages_to_gllm_messages, convert_lm_output_to_langchain_message as convert_lm_output_to_langchain_message

aip_agents/mcp/client/transports.py

@@ -185,7 +185,11 @@ class HTTPTransport(Transport):
         headers = _sanitize_headers(self.config)
         logger.debug(f"Attempting streamable HTTP connection to {url} with headers: {list(headers.keys())}")
         try:
-            http_client = httpx.AsyncClient(timeout=httpx.Timeout(timeout), headers=headers)
+            http_client = httpx.AsyncClient(
+                timeout=httpx.Timeout(timeout),
+                headers=headers,
+                follow_redirects=True,
+            )
             self._http_client = http_client
             self.ctx = streamable_http_client(url=url, http_client=http_client)
             read_stream, write_stream, _ = await self.ctx.__aenter__()

aip_agents/memory/adapters/base_adapter.py

@@ -256,6 +256,100 @@ class BaseMemoryAdapter(BaseMemory):
 
         return [self._chunk_to_hit(chunk) for chunk in chunks]
 
+    def delete_by_query(
+        self,
+        *,
+        query: str,
+        user_id: str,
+        metadata: dict[str, Any] | None = None,
+        threshold: float | None = None,
+        top_k: int | None = None,
+        categories: list[str] | None = None,
+    ) -> Any:
+        """Delete memories by semantic query.
+
+        Args:
+            query: Semantic query describing memories to delete.
+            user_id: User identifier for the deletion scope.
+            metadata: Optional metadata filters to constrain deletion.
+            threshold: Optional semantic threshold (if supported by backend).
+            top_k: Optional max number of memories to delete by query.
+            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_by_user_query,
+                    categories=categories,
+                    query=query,
+                    user_id=user_id or self.agent_id,
+                    agent_id=self.agent_id,
+                    scopes=DEFAULT_SCOPE,
+                    metadata=metadata,
+                    threshold=threshold,
+                    top_k=top_k,
+                )
+            )
+            duration = perf_counter() - start
+            logger.info(
+                "BaseMemoryAdapter: delete_by_query user_id='%s' query='%s' completed in %.2fs",
+                user_id,
+                query,
+                duration,
+            )
+            return result
+        except Exception as exc:  # noqa: BLE001
+            logger.debug("BaseMemoryAdapter.delete_by_query ignored error: %s", exc)
+            return None
+
+    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,32 @@ class BaseMemoryAdapter(BaseMemory):
         Returns:
             List of memory hits matching the criteria.
         """
+    def delete_by_query(self, *, query: str, user_id: str, metadata: dict[str, Any] | None = None, threshold: float | None = None, top_k: int | None = None, categories: list[str] | None = None) -> Any:
+        """Delete memories by semantic query.
+
+        Args:
+            query: Semantic query describing memories to delete.
+            user_id: User identifier for the deletion scope.
+            metadata: Optional metadata filters to constrain deletion.
+            threshold: Optional semantic threshold (if supported by backend).
+            top_k: Optional max number of memories to delete by query.
+            categories: Optional categories to filter by (best-effort).
+
+        Returns:
+            Backend-specific delete result or None on failure.
+        """
+    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/tools/__init__.py

@@ -3,8 +3,12 @@
 from importlib import import_module
 from typing import TYPE_CHECKING
 
+from aip_agents.tools.date_range_tool import DateRangeTool
 from aip_agents.tools.gl_connector import GLConnectorTool
-from aip_agents.tools.gl_connector_tools import BOSA_AUTOMATED_TOOLS, GL_CONNECTORS_AUTOMATED_TOOLS
+from aip_agents.tools.gl_connector_tools import (
+    BOSA_AUTOMATED_TOOLS,
+    GL_CONNECTORS_AUTOMATED_TOOLS,
+)
 from aip_agents.tools.time_tool import TimeTool
 from aip_agents.tools.web_search import GoogleSerperTool
 from aip_agents.utils.logger import get_logger
@@ -17,6 +21,7 @@ __all__ = [
     "GLConnectorTool",
     "GoogleSerperTool",
     "TimeTool",
+    "DateRangeTool",
 ]
 
 
@@ -51,5 +56,9 @@ _register_optional("aip_agents.tools.execute_ptc_code", "create_execute_ptc_code
 if TYPE_CHECKING:
     from aip_agents.tools.browser_use import BrowserUseTool
     from aip_agents.tools.code_sandbox import E2BCodeSandboxTool
-    from aip_agents.tools.document_loader import DocxReaderTool, ExcelReaderTool, PDFReaderTool
+    from aip_agents.tools.document_loader import (
+        DocxReaderTool,
+        ExcelReaderTool,
+        PDFReaderTool,
+    )
     from aip_agents.tools.execute_ptc_code import create_execute_ptc_code_tool

aip_agents/tools/__init__.pyi

@@ -1,5 +1,6 @@
 from aip_agents.tools.browser_use import BrowserUseTool as BrowserUseTool
 from aip_agents.tools.code_sandbox import E2BCodeSandboxTool as E2BCodeSandboxTool
+from aip_agents.tools.date_range_tool import DateRangeTool as DateRangeTool
 from aip_agents.tools.document_loader import DocxReaderTool as DocxReaderTool, ExcelReaderTool as ExcelReaderTool, PDFReaderTool as PDFReaderTool
 from aip_agents.tools.execute_ptc_code import create_execute_ptc_code_tool as create_execute_ptc_code_tool
 from aip_agents.tools.gl_connector import GLConnectorTool as GLConnectorTool
@@ -7,4 +8,4 @@ from aip_agents.tools.gl_connector_tools import BOSA_AUTOMATED_TOOLS as BOSA_AUT
 from aip_agents.tools.time_tool import TimeTool as TimeTool
 from aip_agents.tools.web_search import GoogleSerperTool as GoogleSerperTool
 
-__all__ = ['BOSA_AUTOMATED_TOOLS', 'GL_CONNECTORS_AUTOMATED_TOOLS', 'GLConnectorTool', 'GoogleSerperTool', 'TimeTool', 'BrowserUseTool', 'E2BCodeSandboxTool', 'DocxReaderTool', 'ExcelReaderTool', 'PDFReaderTool', 'create_execute_ptc_code_tool']
+__all__ = ['BOSA_AUTOMATED_TOOLS', 'GL_CONNECTORS_AUTOMATED_TOOLS', 'GLConnectorTool', 'GoogleSerperTool', 'TimeTool', 'DateRangeTool', 'BrowserUseTool', 'E2BCodeSandboxTool', 'DocxReaderTool', 'ExcelReaderTool', 'PDFReaderTool', 'create_execute_ptc_code_tool']