aip-agents-binary 0.6.3__py3-none-manylinux_2_31_x86_64.whl → 0.6.5__py3-none-manylinux_2_31_x86_64.whl

aip_agents/agent/__init__.py

@@ -6,13 +6,25 @@ Author:
     Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
 """
 
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Any
+
 from aip_agents.agent.base_agent import BaseAgent
+
+if TYPE_CHECKING:
+    from aip_agents.agent.google_adk_agent import GoogleADKAgent
+    from aip_agents.agent.langflow_agent import LangflowAgent
 from aip_agents.agent.base_langgraph_agent import BaseLangGraphAgent
-from aip_agents.agent.google_adk_agent import GoogleADKAgent
 from aip_agents.agent.interface import AgentInterface
-from aip_agents.agent.langflow_agent import LangflowAgent
-from aip_agents.agent.langgraph_memory_enhancer_agent import LangGraphMemoryEnhancerAgent
-from aip_agents.agent.langgraph_react_agent import LangChainAgent, LangGraphAgent, LangGraphReactAgent
+from aip_agents.agent.langgraph_memory_enhancer_agent import (
+    LangGraphMemoryEnhancerAgent,
+)
+from aip_agents.agent.langgraph_react_agent import (
+    LangChainAgent,
+    LangGraphAgent,
+    LangGraphReactAgent,
+)
 
 __all__ = [
     "AgentInterface",
@@ -25,3 +37,31 @@ __all__ = [
     "LangflowAgent",
     "LangGraphMemoryEnhancerAgent",
 ]
+
+
+def __getattr__(name: str) -> Any:
+    """Lazy import of heavy agent implementations.
+
+    This avoids importing heavy dependencies (Google ADK, etc.)
+    when they are not needed.
+
+    Args:
+        name: Attribute name to import.
+
+    Returns:
+        The requested class.
+
+    Raises:
+        AttributeError: If attribute is not found.
+    """
+    if name == "GoogleADKAgent":
+        from aip_agents.agent.google_adk_agent import (
+            GoogleADKAgent as _GoogleADKAgent,
+        )
+
+        return _GoogleADKAgent
+    elif name == "LangflowAgent":
+        from aip_agents.agent.langflow_agent import LangflowAgent as _LangflowAgent
+
+        return _LangflowAgent
+    raise AttributeError(f"module '{__name__}' has no attribute '{name}'")

aip_agents/agent/base_langgraph_agent.py

@@ -455,6 +455,14 @@ class BaseLangGraphAgent(BaseAgent):
 
         future.add_done_callback(_log_completion)
 
+    def _should_save_interaction(self, final_state: dict[str, Any] | None) -> bool:
+        """Return True when interaction should be saved to memory.
+
+        Subclasses can override this to skip persistence for specific response types.
+        """
+        del final_state
+        return True
+
     def _resolve_and_validate_tools(self) -> list[BaseTool]:
         """Resolve and validate regular tools for LangGraph usage.
 
@@ -898,7 +906,12 @@ class BaseLangGraphAgent(BaseAgent):
             formatted_output = self._format_graph_output(final_state_result)
 
             try:
-                self._memory_save_interaction(user_text=query, ai_text=formatted_output, memory_user_id=memory_user_id)
+                if self._should_save_interaction(final_state_result):
+                    self._memory_save_interaction(
+                        user_text=query,
+                        ai_text=formatted_output,
+                        memory_user_id=memory_user_id,
+                    )
             except Exception:
                 pass
 
@@ -2442,7 +2455,7 @@ class BaseLangGraphAgent(BaseAgent):
                     and isinstance(context.last_final_content, str)
                     and context.last_final_content
                 )
-                if should_save_early:
+                if should_save_early and self._should_save_interaction(context.final_state):
                     try:
                         logger.info(
                             "Agent '%s': A2A persisting memory early (len=%d) for user_id='%s'",
@@ -2461,6 +2474,8 @@ class BaseLangGraphAgent(BaseAgent):
                         context.saved_memory = True
                     except Exception:
                         pass
+                elif should_save_early:
+                    context.saved_memory = True
         except Exception:
             pass
 
@@ -2492,10 +2507,15 @@ class BaseLangGraphAgent(BaseAgent):
                     )
                 except Exception:
                     pass
-                self._memory_save_interaction(
-                    user_text=context.original_query, ai_text=final_text, memory_user_id=context.memory_user_id
-                )
-                context.saved_memory = True
+                if self._should_save_interaction(context.final_state):
+                    self._memory_save_interaction(
+                        user_text=context.original_query,
+                        ai_text=final_text,
+                        memory_user_id=context.memory_user_id,
+                    )
+                    context.saved_memory = True
+                else:
+                    context.saved_memory = True
         except Exception:
             pass