aip-agents-binary 0.6.4__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/mcp/client/__init__.py

@@ -7,8 +7,44 @@ Authors:
     Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
 """
 
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Any
+
 from aip_agents.mcp.client.base_mcp_client import BaseMCPClient
-from aip_agents.mcp.client.google_adk.client import GoogleADKMCPClient
-from aip_agents.mcp.client.langchain.client import LangchainMCPClient
+
+if TYPE_CHECKING:
+    from aip_agents.mcp.client.google_adk.client import GoogleADKMCPClient
+    from aip_agents.mcp.client.langchain.client import LangchainMCPClient
 
 __all__ = ["GoogleADKMCPClient", "LangchainMCPClient", "BaseMCPClient"]
+
+
+def __getattr__(name: str) -> Any:
+    """Lazy import of MCP client implementations.
+
+    This avoids importing heavy dependencies (Google ADK, Vertex AI, 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 == "GoogleADKMCPClient":
+        from aip_agents.mcp.client.google_adk.client import (
+            GoogleADKMCPClient as _GoogleADKMCPClient,
+        )
+
+        return _GoogleADKMCPClient
+    elif name == "LangchainMCPClient":
+        from aip_agents.mcp.client.langchain.client import (
+            LangchainMCPClient as _LangchainMCPClient,
+        )
+
+        return _LangchainMCPClient
+    raise AttributeError(f"module '{__name__}' has no attribute '{name}'")

aip_agents/sentry/sentry.py

@@ -6,6 +6,7 @@ Authors:
 
 import inspect
 import os
+from typing import Any
 
 from bosa_core.telemetry import (
     FastAPIConfig,
@@ -20,7 +21,7 @@ from bosa_core.telemetry.opentelemetry.instrument.functions import (
 from dotenv import load_dotenv
 from fastapi import FastAPI
 
-from aip_agents.agent import BaseAgent, GoogleADKAgent, LangChainAgent, LangGraphAgent
+from aip_agents.agent import BaseAgent, LangChainAgent, LangGraphAgent
 from aip_agents.utils.logger import get_logger
 
 load_dotenv()
@@ -35,12 +36,32 @@ VERSION_NUMBER = os.getenv("VERSION_NUMBER", "0.0.0")
 BUILD_NUMBER = os.getenv("BUILD_NUMBER", "0")
 USE_OPENTELEMETRY = os.getenv("USE_OPENTELEMETRY", "true").lower() == "true"
 
-CLASSES_TO_INSTRUMENT = [
-    BaseAgent,
-    LangGraphAgent,
-    LangChainAgent,
-    GoogleADKAgent,
-]
+# Lazy import of GoogleADKAgent to avoid heavy dependencies when not needed.
+# This is initialized lazily by _get_classes_to_instrument() and can be
+# patched by tests for mocking purposes.
+CLASSES_TO_INSTRUMENT: list[type[Any]] | None = None
+
+
+def _get_classes_to_instrument() -> list[type[Any]]:
+    """Get the list of classes to instrument.
+
+    This lazily imports GoogleADKAgent only when telemetry is being set up,
+    avoiding the heavy Google ADK dependencies during module import.
+
+    Returns:
+        List of agent classes to instrument.
+    """
+    global CLASSES_TO_INSTRUMENT
+    if CLASSES_TO_INSTRUMENT is None:
+        from aip_agents.agent import GoogleADKAgent
+
+        CLASSES_TO_INSTRUMENT = [
+            BaseAgent,
+            LangGraphAgent,
+            LangChainAgent,
+            GoogleADKAgent,
+        ]
+    return CLASSES_TO_INSTRUMENT
 
 
 def get_all_methods(cls: type) -> list:
@@ -66,7 +87,7 @@ def instrument_gl_functions() -> None:
     if BOSAFunctionsInstrumentor is None:
         return
     agent_methods = []
-    for cls in CLASSES_TO_INSTRUMENT:
+    for cls in _get_classes_to_instrument():
         agent_methods.extend(get_all_methods(cls))
     BOSAFunctionsInstrumentor().instrument(methods=agent_methods)
 

{aip_agents_binary-0.6.4.dist-info → aip_agents_binary-0.6.5.dist-info}/METADATA

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