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

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/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",
 ]
 
 
@@ -46,8 +51,14 @@ _register_optional("aip_agents.tools.code_sandbox", "E2BCodeSandboxTool")
 _register_optional("aip_agents.tools.document_loader", "DocxReaderTool")
 _register_optional("aip_agents.tools.document_loader", "ExcelReaderTool")
 _register_optional("aip_agents.tools.document_loader", "PDFReaderTool")
+_register_optional("aip_agents.tools.execute_ptc_code", "create_execute_ptc_code_tool")
 
 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,9 +1,11 @@
 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
 from aip_agents.tools.gl_connector_tools import BOSA_AUTOMATED_TOOLS as BOSA_AUTOMATED_TOOLS, GL_CONNECTORS_AUTOMATED_TOOLS as GL_CONNECTORS_AUTOMATED_TOOLS
 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']
+__all__ = ['BOSA_AUTOMATED_TOOLS', 'GL_CONNECTORS_AUTOMATED_TOOLS', 'GLConnectorTool', 'GoogleSerperTool', 'TimeTool', 'DateRangeTool', 'BrowserUseTool', 'E2BCodeSandboxTool', 'DocxReaderTool', 'ExcelReaderTool', 'PDFReaderTool', 'create_execute_ptc_code_tool']

aip_agents/tools/browser_use/browser_use_tool.py

@@ -93,6 +93,7 @@ References:
 """
 
 import asyncio
+import copy
 import json
 from collections.abc import Callable
 from typing import Any, Literal
@@ -556,13 +557,20 @@ class BrowserUseTool(BaseTool):
             iframe_event = yield_iframe_activity(streaming_state.debug_url, "Receive streaming URL")
             self._log_stream_event("iframe_start", iframe_event)
             yield iframe_event
+            last_done_event: dict | None = None
             async for event in self._stream_agent_with_markers(agent, streaming_state, recorder):
                 self._log_stream_event("agent_event", event)
+                tool_info = event.get("tool_info") if isinstance(event, dict) else None
+                tool_calls = tool_info.get("tool_calls", []) if isinstance(tool_info, dict) else []
+                if any(call.get("name") == "done" for call in tool_calls if isinstance(call, dict)):
+                    last_done_event = event
                 yield event
             recording_event = self._recording_event(streaming_state)
             if recording_event:
                 self._log_stream_event("recording_event", recording_event)
                 yield recording_event
+                if last_done_event:
+                    yield copy.deepcopy(last_done_event)
         finally:
             await self._release_session(session, client)
 

aip_agents/tools/browser_use/streaming.py

@@ -223,6 +223,8 @@ def create_step_response(
     if is_done:
         final_tool = _get_done_tool_for_final_response(agent, tool_calls_dict)
         final_output = final_tool.get("output") or TASK_COMPLETED_MESSAGE
+        if not any(call.get("name") == "done" for call in tool_calls_dict):
+            tool_calls_dict.append(final_tool)
         tool_info = {
             "name": PRIMARY_TOOL_NAME,
             "args": final_tool.get("args", {}),