aip-agents-binary 0.6.4__py3-none-macosx_13_0_arm64.whl → 0.6.5__py3-none-macosx_13_0_arm64.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/sentry/sentry.pyi

@@ -1,7 +1,8 @@
 from _typeshed import Incomplete
-from aip_agents.agent import BaseAgent as BaseAgent, GoogleADKAgent as GoogleADKAgent, LangChainAgent as LangChainAgent, LangGraphAgent as LangGraphAgent
+from aip_agents.agent import BaseAgent as BaseAgent, LangChainAgent as LangChainAgent, LangGraphAgent as LangGraphAgent
 from aip_agents.utils.logger import get_logger as get_logger
 from fastapi import FastAPI
+from typing import Any
 
 logger: Incomplete
 SENTRY_DSN: Incomplete
@@ -10,7 +11,7 @@ SENTRY_PROJECT: Incomplete
 VERSION_NUMBER: Incomplete
 BUILD_NUMBER: Incomplete
 USE_OPENTELEMETRY: Incomplete
-CLASSES_TO_INSTRUMENT: Incomplete
+CLASSES_TO_INSTRUMENT: list[type[Any]] | None
 
 def get_all_methods(cls) -> list:
     """Get all methods from a class.

{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

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

@@ -16,7 +16,7 @@ aip_agents/a2a/server/langflow_executor.py,sha256=6bVnwqRnqbBi949JHHWcIB-ZVQAPGJ
 aip_agents/a2a/server/langflow_executor.pyi,sha256=nlWUIOEGFcp4qtyrmmxax0TfzwIwOGkvFK-YmG0cGMg,1837
 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__.py,sha256=v43Suwam9yhDIMhhr3bupMzOZqrzYjj_rm4Diw-wM30,1771
 aip_agents/agent/__init__.pyi,sha256=MxIAeAv1pPCtqfAa3lvmeCAN-IT5p3v77IeKhfKYvKo,855
 aip_agents/agent/base_agent.py,sha256=XH19lJZuumWBu2JMoretv6T4bSXcWMqK8jZ3mOWWzCk,39669
 aip_agents/agent/base_agent.pyi,sha256=xnWp05zTJrt1YfHmm9CsggBmrSsIY-SSy_G9EWGvEBQ,11118
@@ -302,7 +302,7 @@ aip_agents/guardrails/engines/phrase_matcher.py,sha256=4rilmNkZ3OCeUC6YhMVRT_UJe
 aip_agents/guardrails/engines/phrase_matcher.pyi,sha256=Gk978fctlBNoyoxtjIIQTwcoBC6pzb0DVP7LTHeKQfc,2060
 aip_agents/mcp/__init__.py,sha256=CCc2mHoabzEFzQyjEW4ULKglEGtR_BGgBZNix8QjWjU,60
 aip_agents/mcp/__init__.pyi,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-aip_agents/mcp/client/__init__.py,sha256=85jA6v-bjjm13kCqnu57__fIprCAk5fC628API9h6uc,493
+aip_agents/mcp/client/__init__.py,sha256=tA-LtCpvTL_OM0nzLLCm7uOxb46xMz4cuN_xBZojPmw,1430
 aip_agents/mcp/client/__init__.pyi,sha256=XgIkFbbmAh5cqQM4t1xT5y-Avu0vf4b-YCn6btXjIK8,339
 aip_agents/mcp/client/base_mcp_client.py,sha256=36bY_4BbzVBogDekwBOCL3A12MuXmKiNTpbxQjBgjlM,13973
 aip_agents/mcp/client/base_mcp_client.pyi,sha256=FuJBA4TLFPBJzuHqpxMGOlODgMpT7avOoAR34TsiQAk,5881
@@ -412,8 +412,8 @@ aip_agents/schema/storage.py,sha256=Tl0RFT32GWdsc5y8bTvpWMOU8hmQxXIlrMub3qdwx8Y,
 aip_agents/schema/storage.pyi,sha256=exaZAS49PYSuhYSPYukIRCRHIGRyCpBGucrdpnhV4zE,557
 aip_agents/sentry/__init__.py,sha256=l3nI-3pjBNNmcOZ7xPeIis1n_wkPpvnk2aHxQYWsflU,261
 aip_agents/sentry/__init__.pyi,sha256=OGxzQzEvbnqf02zXdBJUrHeKfjfBgvnOL7p-0gNvP8k,103
-aip_agents/sentry/sentry.py,sha256=zyhIZCnA2mcLuf5w-ezE_7ZWOjPdGt3A5g7AcJrPgxY,4504
-aip_agents/sentry/sentry.pyi,sha256=fTFb71FfeByBUIACjkp8qaHsWe6C52Y4tPTfizAPVv0,1432
+aip_agents/sentry/sentry.py,sha256=EBpHFzm_I--lZStmFuA2HQHXvJOEiYj2KG2t1LGn0Vc,5280
+aip_agents/sentry/sentry.pyi,sha256=ikoJuMEAvg_DSHf_IvBxNEXxhdagmsN1CtqvsMGjfg0,1433
 aip_agents/storage/__init__.py,sha256=cN4L2whui-DOm81tsYV88Wx5g5-cUjeLVyp55RtbJJU,1219
 aip_agents/storage/__init__.pyi,sha256=wn8VuS7x4WSzKuV01WrjPkZfnh3GylNum9FlPiaSdsA,901
 aip_agents/storage/base.py,sha256=dijzGJJRHPEneV-6QRC_xm0CzgYuIyhmnWloJC8oIx0,2413
@@ -606,7 +606,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.6.4.dist-info/METADATA,sha256=IEUX5CDrYnEC8vfhCQ4_UuRBm8OsYx-aY246fcjOpyE,22194
-aip_agents_binary-0.6.4.dist-info/WHEEL,sha256=KxCTaSkoYs_EnWvWxmau4HAvN-_rCRYV_bfRc_41A9k,106
-aip_agents_binary-0.6.4.dist-info/top_level.txt,sha256=PEz8vcwC1bH4UrkhF0LkIYCNfXGWZUHdSklbvkBe25E,11
-aip_agents_binary-0.6.4.dist-info/RECORD,,
+aip_agents_binary-0.6.5.dist-info/METADATA,sha256=oTbOgYNNZ2IWWrrf2uSGyocsC5qcFNh1WGX5pOpZDXA,22194
+aip_agents_binary-0.6.5.dist-info/WHEEL,sha256=KxCTaSkoYs_EnWvWxmau4HAvN-_rCRYV_bfRc_41A9k,106
+aip_agents_binary-0.6.5.dist-info/top_level.txt,sha256=PEz8vcwC1bH4UrkhF0LkIYCNfXGWZUHdSklbvkBe25E,11
+aip_agents_binary-0.6.5.dist-info/RECORD,,