aip-agents-binary 0.5.22__py3-none-macosx_13_0_arm64.whl → 0.5.23__py3-none-macosx_13_0_arm64.whl

aip_agents/examples/hello_world_langgraph_gl_connector_twitter.py

@@ -6,11 +6,12 @@ Authors:
 
 import asyncio
 
-from aip_agents.agent import LangGraphAgent
-from aip_agents.tools import GL_CONNECTORS_AUTOMATED_TOOLS
 from dotenv import load_dotenv
 from langchain_openai import ChatOpenAI
 
+from aip_agents.agent import LangGraphAgent
+from aip_agents.tools import GL_CONNECTORS_AUTOMATED_TOOLS
+
 load_dotenv(override=True)
 
 

aip_agents/examples/hello_world_sentry.py

@@ -20,13 +20,14 @@ import threading
 import time
 
 import uvicorn
-from aip_agents.examples.hello_world_langgraph import langgraph_example
-from aip_agents.sentry import setup_telemetry
-from aip_agents.utils.logger import get_logger
 from fastapi import FastAPI
 from fastapi.responses import JSONResponse
 from fastapi.staticfiles import StaticFiles
 
+from aip_agents.examples.hello_world_langgraph import langgraph_example
+from aip_agents.sentry import setup_telemetry
+from aip_agents.utils.logger import get_logger
+
 logger = get_logger(__name__)
 
 BASE_URL = "http://localhost:8585"
@@ -49,9 +50,7 @@ def fetch_endpoints():
             # Because using request.get() will trigger sentry, but the ep name is not saved.
             # Instead of GET /langgraph-hello-world, sentry will save it as GET
             url = f"http://127.0.0.1:8585{ep}"
-            result = subprocess.run(
-                ["curl", "-i", url], capture_output=True, text=True, check=False
-            )
+            result = subprocess.run(["curl", "-i", url], capture_output=True, text=True, check=False)
             logger.info(result.stdout)
         except Exception as e:
             logger.error(f"Error fetching {ep}: {e}")

aip_agents/mcp/client/transports.py

@@ -45,6 +45,19 @@ DEFAULT_TIMEOUT: float = 30.0
 """Default connection timeout in seconds."""
 
 
+def _sanitize_headers(config: MCPConfiguration) -> dict[str, str]:
+    """Remove headers with None values to avoid invalid HTTP headers.
+
+    Args:
+        config (MCPConfiguration): Transport configuration containing optional headers.
+
+    Returns:
+        dict[str, str]: Filtered headers with None values removed.
+    """
+    headers = config.get("headers", {}) or {}
+    return {key: value for key, value in headers.items() if value is not None}
+
+
 class TransportType(StrEnum):
     """Enum for supported MCP transport types."""
 
@@ -115,7 +128,7 @@ class SSETransport(Transport):
 
         url = f"{base_url}/sse" if not base_url.endswith("/sse") else base_url
         timeout = self.config.get("timeout", DEFAULT_TIMEOUT)
-        headers = self.config.get("headers", {})
+        headers = _sanitize_headers(self.config)
         logger.debug(f"Attempting SSE connection to {url} with headers: {list(headers.keys())}")
         try:
             self.ctx = sse_client(url=url, timeout=timeout, sse_read_timeout=300.0, headers=headers)
@@ -147,7 +160,7 @@ class HTTPTransport(Transport):
 
         url = f"{base_url}/mcp" if not base_url.endswith("/mcp") else base_url
         timeout = self.config.get("timeout", DEFAULT_TIMEOUT)
-        headers = self.config.get("headers", {})
+        headers = _sanitize_headers(self.config)
         logger.debug(f"Attempting streamable HTTP connection to {url} with headers: {list(headers.keys())}")
         try:
             self.ctx = streamablehttp_client(url=url, timeout=timeout, headers=headers)

aip_agents/sentry/sentry.py

@@ -7,8 +7,6 @@ Authors:
 import inspect
 import os
 
-from aip_agents.agent import BaseAgent, GoogleADKAgent, LangChainAgent, LangGraphAgent
-from aip_agents.utils.logger import get_logger
 from bosa_core.telemetry import (
     FastAPIConfig,
     OpenTelemetryConfig,
@@ -22,6 +20,9 @@ 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.utils.logger import get_logger
+
 load_dotenv()
 
 
@@ -55,11 +56,7 @@ def get_all_methods(cls: type) -> list:
     for name, member in inspect.getmembers(cls):
         if name.startswith("_"):
             continue  # skip dunder and private
-        if (
-            inspect.isfunction(member)
-            or inspect.ismethod(member)
-            or inspect.iscoroutinefunction(member)
-        ):
+        if inspect.isfunction(member) or inspect.ismethod(member) or inspect.iscoroutinefunction(member):
             methods.append(member)
     return methods
 
@@ -114,9 +111,7 @@ def setup_sentry_with_open_telemetry(app: FastAPI) -> None:
 
         telemetry_config = TelemetryConfig(sentry_config=sentry_config)
         init_telemetry(telemetry_config)
-        logger.info(
-            f"Telemetry initialized with OpenTelemetry for environment: {SENTRY_ENVIRONMENT}"
-        )
+        logger.info(f"Telemetry initialized with OpenTelemetry for environment: {SENTRY_ENVIRONMENT}")
     except Exception as e:
         logger.error(f"Failed to initialize telemetry with OpenTelemetry: {e}")
 
@@ -135,9 +130,7 @@ def setup_sentry_only() -> None:
 
         telemetry_config = TelemetryConfig(sentry_config=sentry_config)
         init_telemetry(telemetry_config)
-        logger.info(
-            f"Telemetry initialized with Sentry only for environment: {SENTRY_ENVIRONMENT}"
-        )
+        logger.info(f"Telemetry initialized with Sentry only for environment: {SENTRY_ENVIRONMENT}")
     except Exception as e:
         logger.error(f"Failed to initialize telemetry with Sentry only: {e}")
 

aip_agents/tools/__init__.py

@@ -3,8 +3,8 @@
 from importlib import import_module
 from typing import TYPE_CHECKING
 
-from aip_agents.tools.gl_connector_tools import GL_CONNECTORS_AUTOMATED_TOOLS, BOSA_AUTOMATED_TOOLS
 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.time_tool import TimeTool
 from aip_agents.tools.web_search import GoogleSerperTool
 from aip_agents.utils.logger import get_logger

aip_agents/tools/code_sandbox/e2b_cloud_sandbox_extended.py

@@ -11,12 +11,10 @@ from http import HTTPStatus
 from typing import Any
 
 import requests
+from e2b_code_interpreter import Sandbox
 from gllm_inference.schema import Attachment
-from gllm_tools.code_interpreter.code_sandbox.e2b_cloud_sandbox import E2BCloudSandbox
-from gllm_tools.code_interpreter.code_sandbox.models import (
-    ExecutionResult,
-    ExecutionStatus,
-)
+from gllm_tools.code_interpreter.code_sandbox.e2b_sandbox import E2BSandbox
+from gllm_tools.code_interpreter.code_sandbox.models import ExecutionResult, ExecutionStatus
 from gllm_tools.code_interpreter.code_sandbox.utils import calculate_duration_ms
 
 from aip_agents.tools.code_sandbox.constant import DATA_FILE_PATH
@@ -123,19 +121,57 @@ class SandboxFileWatcher:
         return self._created_files.copy()
 
 
-class MyE2BCloudSandbox(E2BCloudSandbox):
-    """Extended E2B Cloud Sandbox with filesystem monitoring capabilities."""
+class MyE2BCloudSandbox(E2BSandbox):
+    """Extended E2B sandbox with filesystem monitoring capabilities.
 
-    def __init__(self, *args, **kwargs):
-        """Initialize the sandbox with monitoring capabilities.
+    Use `create()` in production to build a fully initialized sandbox wrapper.
+    Direct construction is intentionally blocked to prevent partially initialized
+    instances that lack the underlying E2B sandbox clients.
+    """
+
+    def __init__(self, language: str = "python", *, _unsafe_allow_init: bool = False) -> None:
+        """Initialize the sandbox wrapper.
 
         Args:
-            *args: Positional arguments forwarded to ``E2BCloudSandbox``.
-            **kwargs: Keyword arguments forwarded to ``E2BCloudSandbox``.
+            language (str): Language to execute inside the sandbox.
+            _unsafe_allow_init (bool): Escape hatch for tests/mocks only.
+
+        Raises:
+            RuntimeError: When instantiated directly without `create()`.
         """
-        super().__init__(*args, **kwargs)
+        if not _unsafe_allow_init:
+            raise RuntimeError("Use MyE2BCloudSandbox.create(...) to initialize a sandbox instance.")
+        super().__init__(language=language)
         self.file_watcher: SandboxFileWatcher | None = None
 
+    @classmethod
+    async def create(
+        cls,
+        api_key: str,
+        domain: str | None = None,
+        template: str | None = None,
+        language: str = "python",
+        additional_packages: list[str] | None = None,
+        **kwargs: Any,
+    ) -> "MyE2BCloudSandbox":
+        """Create a fully initialized sandbox wrapper.
+
+        This is the supported construction path for production usage. It wires
+        the E2B sandbox instance and its filesystem/command clients, then
+        installs language dependencies.
+        """
+        sandbox = Sandbox.create(api_key=api_key, domain=domain, template=template, **kwargs)
+
+        instance = cls(language=language, _unsafe_allow_init=True)
+        instance.sandbox = sandbox
+        instance.files = sandbox.files
+        instance.commands = sandbox.commands
+        instance.additional_packages = additional_packages or []
+
+        instance._install_language_dependencies()
+
+        return instance
+
     async def execute_code(
         self,
         code: str,
@@ -161,11 +197,10 @@ class MyE2BCloudSandbox(E2BCloudSandbox):
         Raises:
             RuntimeError: If sandbox is not initialized.
         """
-        if not self.sandbox:
+        if not self.sandbox or not self.files or not self.commands:
             raise RuntimeError("Sandbox is not initialized")
 
         start_time = time.time()
-
         try:
             # Initialize filesystem monitoring
             self.file_watcher = SandboxFileWatcher(self.sandbox)
@@ -176,8 +211,12 @@ class MyE2BCloudSandbox(E2BCloudSandbox):
             # Pre-populate the variable `df` for direct use in the code
             if files:
                 logger.info("Pre-populating the variable `df` with the data from the file.")
-                self.sandbox.run_code(f"import pandas as pd; df = pd.read_csv('{DATA_FILE_PATH}')", timeout=timeout)
-            execution = self.sandbox.run_code(code, timeout=timeout)
+                self.sandbox.run_code(
+                    f"import pandas as pd; df = pd.read_csv('{DATA_FILE_PATH}')",
+                    language=self.language,
+                    timeout=timeout,
+                )
+            execution = self.sandbox.run_code(code, language=self.language, timeout=timeout)
             duration_ms = calculate_duration_ms(start_time)
             status = ExecutionStatus.ERROR if execution.error else ExecutionStatus.SUCCESS
 
@@ -191,8 +230,8 @@ class MyE2BCloudSandbox(E2BCloudSandbox):
             return ExecutionResult.create(
                 status=status,
                 code=code,
-                stdout=(execution.logs.stdout[0] if execution.logs and execution.logs.stdout else ""),
-                stderr=(execution.logs.stderr[0] if execution.logs and execution.logs.stderr else ""),
+                stdout=("\n".join(execution.logs.stdout) if execution.logs and execution.logs.stdout else ""),
+                stderr=("\n".join(execution.logs.stderr) if execution.logs and execution.logs.stderr else ""),
                 error=(str(execution.error) if execution.error else ""),  # Convert to string here
                 duration_ms=duration_ms,
             )
@@ -218,8 +257,8 @@ class MyE2BCloudSandbox(E2BCloudSandbox):
     def download_file(self, file_path: str) -> bytes | None:
         """Download file content from the sandbox.
 
-        Uses download_url method to get a direct URL and downloads via HTTP,
-        which avoids the binary corruption issue with files.read().
+        Uses download_url when available to avoid binary corruption issues.
+        Falls back to the filesystem API when download_url fails or is unavailable.
 
         Args:
             file_path (str): Path to the file in the sandbox.
@@ -237,18 +276,28 @@ class MyE2BCloudSandbox(E2BCloudSandbox):
             if hasattr(self.sandbox, "download_url"):
                 logger.info(f"Downloading {file_path} via download_url method")
 
-                # Get the download URL
-                url = self.sandbox.download_url(file_path)
-                logger.debug(f"Got download URL: {url}")
-
-                response = requests.get(url, timeout=30)
-
-                if response.status_code == HTTPStatus.OK:
-                    content = response.content
-                    logger.info(f"Successfully downloaded {len(content)} bytes via URL")
-                    return content
+                try:
+                    url = self.sandbox.download_url(file_path)
+                except Exception as e:
+                    logger.warning(f"Failed to get download URL: {str(e)}")
                 else:
-                    logger.warning(f"URL download failed with status {response.status_code}")
+                    logger.debug(f"Got download URL: {url}")
+
+                    try:
+                        response = requests.get(url, timeout=30)
+                    except Exception as e:
+                        logger.warning(f"URL download failed with error: {str(e)}")
+                    else:
+                        if response.status_code == HTTPStatus.OK:
+                            content = response.content
+                            logger.info(f"Successfully downloaded {len(content)} bytes via URL")
+                            return content
+                        logger.warning(f"URL download failed with status {response.status_code}")
+
+            if self.files:
+                logger.info(f"Downloading {file_path} via filesystem API")
+                content = self.files.read(file_path, format="bytes")
+                return bytes(content)
 
             return None
 

aip_agents/tools/code_sandbox/e2b_cloud_sandbox_extended.pyi

@@ -2,7 +2,7 @@ from _typeshed import Incomplete
 from aip_agents.tools.code_sandbox.constant import DATA_FILE_PATH as DATA_FILE_PATH
 from aip_agents.utils.logger import get_logger as get_logger
 from gllm_inference.schema import Attachment as Attachment
-from gllm_tools.code_interpreter.code_sandbox.e2b_cloud_sandbox import E2BCloudSandbox
+from gllm_tools.code_interpreter.code_sandbox.e2b_sandbox import E2BSandbox
 from gllm_tools.code_interpreter.code_sandbox.models import ExecutionResult
 from typing import Any
 
@@ -34,15 +34,31 @@ class SandboxFileWatcher:
             list[str]: List of file paths that were created.
         """
 
-class MyE2BCloudSandbox(E2BCloudSandbox):
-    """Extended E2B Cloud Sandbox with filesystem monitoring capabilities."""
+class MyE2BCloudSandbox(E2BSandbox):
+    """Extended E2B sandbox with filesystem monitoring capabilities.
+
+    Use `create()` in production to build a fully initialized sandbox wrapper.
+    Direct construction is intentionally blocked to prevent partially initialized
+    instances that lack the underlying E2B sandbox clients.
+    """
     file_watcher: SandboxFileWatcher | None
-    def __init__(self, *args, **kwargs) -> None:
-        """Initialize the sandbox with monitoring capabilities.
+    def __init__(self, language: str = 'python', *, _unsafe_allow_init: bool = False) -> None:
+        """Initialize the sandbox wrapper.
 
         Args:
-            *args: Positional arguments forwarded to ``E2BCloudSandbox``.
-            **kwargs: Keyword arguments forwarded to ``E2BCloudSandbox``.
+            language (str): Language to execute inside the sandbox.
+            _unsafe_allow_init (bool): Escape hatch for tests/mocks only.
+
+        Raises:
+            RuntimeError: When instantiated directly without `create()`.
+        """
+    @classmethod
+    async def create(cls, api_key: str, domain: str | None = None, template: str | None = None, language: str = 'python', additional_packages: list[str] | None = None, **kwargs: Any) -> MyE2BCloudSandbox:
+        """Create a fully initialized sandbox wrapper.
+
+        This is the supported construction path for production usage. It wires
+        the E2B sandbox instance and its filesystem/command clients, then
+        installs language dependencies.
         """
     async def execute_code(self, code: str, timeout: int = 30, files: list[Attachment] | None = None, **kwargs: Any) -> ExecutionResult:
         """Execute code in the E2B Cloud sandbox with filesystem monitoring.
@@ -72,8 +88,8 @@ class MyE2BCloudSandbox(E2BCloudSandbox):
     def download_file(self, file_path: str) -> bytes | None:
         """Download file content from the sandbox.
 
-        Uses download_url method to get a direct URL and downloads via HTTP,
-        which avoids the binary corruption issue with files.read().
+        Uses download_url when available to avoid binary corruption issues.
+        Falls back to the filesystem API when download_url fails or is unavailable.
 
         Args:
             file_path (str): Path to the file in the sandbox.

aip_agents/tools/code_sandbox/e2b_sandbox_tool.py

@@ -12,7 +12,7 @@ from typing import Any
 
 import pandas as pd
 from gllm_inference.schema import Attachment
-from gllm_tools.code_interpreter.code_sandbox.e2b_cloud_sandbox import E2BCloudSandbox
+from gllm_tools.code_interpreter.code_sandbox.sandbox import BaseSandbox
 from langchain_core.tools import BaseTool
 from langgraph.types import Command
 from pydantic import BaseModel, Field
@@ -260,7 +260,7 @@ class E2BCodeSandboxTool(BaseTool):
         """
         unique_packages = self._prepare_packages(additional_packages)
 
-        return MyE2BCloudSandbox(
+        return await MyE2BCloudSandbox.create(
             api_key=self.api_key,
             language=language,
             additional_packages=unique_packages,
@@ -304,11 +304,11 @@ class E2BCodeSandboxTool(BaseTool):
             except Exception as e:
                 logger.warning(f"Error terminating sandbox: {str(e)}")
 
-    def _create_artifacts_from_files(self, sandbox: E2BCloudSandbox, file_paths: list[str]) -> list[dict[str, Any]]:
+    def _create_artifacts_from_files(self, sandbox: BaseSandbox, file_paths: list[str]) -> list[dict[str, Any]]:
         """Create artifacts from a list of file paths using ArtifactHandler.
 
         Args:
-            sandbox (E2BCloudSandbox): The active sandbox instance.
+            sandbox (BaseSandbox): The active sandbox instance.
             file_paths (list[str]): List of file paths to download.
 
         Returns:
@@ -358,11 +358,11 @@ class E2BCodeSandboxTool(BaseTool):
 
         return execution_summary
 
-    def _download_and_create_artifact(self, sandbox: E2BCloudSandbox, file_path: str) -> dict[str, Any] | None:
+    def _download_and_create_artifact(self, sandbox: BaseSandbox, file_path: str) -> dict[str, Any] | None:
         """Download a single file and create an artifact using ArtifactHandler.
 
         Args:
-            sandbox (E2BCloudSandbox): The active sandbox instance.
+            sandbox (BaseSandbox): The active sandbox instance.
             file_path (str): Path to the file in the sandbox.
 
         Returns:

aip_agents/tools/constants.py

@@ -15,9 +15,7 @@ GL_CONNECTORS_BASE_URL = (
 )
 GL_CONNECTORS_API_KEY = os.getenv("GL_CONNECTORS_API_KEY") or os.getenv("BOSA_API_KEY")
 GL_CONNECTORS_FETCH_MAX_RETRIES = int(
-    os.getenv("GL_CONNECTORS_FETCH_MAX_RETRIES")
-    or os.getenv("BOSA_FETCH_MAX_RETRIES")
-    or 3
+    os.getenv("GL_CONNECTORS_FETCH_MAX_RETRIES") or os.getenv("BOSA_FETCH_MAX_RETRIES") or 3
 )
 
 # For backward compatibility

aip_agents/tools/gl_connector/tool.py

@@ -14,12 +14,13 @@ import os
 from collections.abc import Iterable
 from typing import Any
 
-from aip_agents.tools.constants import ToolType
 from bosa_connectors import BosaConnector, BOSAConnectorToolGenerator
 from langchain_core.runnables import RunnableConfig
 from langchain_core.tools import BaseTool
 from pydantic import ConfigDict, PrivateAttr
 
+from aip_agents.tools.constants import ToolType
+
 _REQUIRED_ENV_VARS: tuple[str, ...] = (
     "GL_CONNECTORS_BASE_URL",
     "GL_CONNECTORS_API_KEY",
@@ -66,9 +67,7 @@ class _InjectedTool(BaseTool):
         Returns:
             None
         """
-        base_fields = {
-            field: getattr(base_tool, field) for field in BaseTool.model_fields
-        }
+        base_fields = {field: getattr(base_tool, field) for field in BaseTool.model_fields}
         super().__init__(**base_fields)
         self._base_tool = base_tool
         self._token = token
@@ -98,9 +97,7 @@ class _InjectedTool(BaseTool):
         """
         return await self._base_tool._arun(*args, **kwargs)
 
-    def invoke(
-        self, input: Any, config: RunnableConfig | None = None, **kwargs: Any
-    ) -> Any:
+    def invoke(self, input: Any, config: RunnableConfig | None = None, **kwargs: Any) -> Any:
         """Invoke the tool with token and optional identifier injected.
 
         Args:
@@ -114,9 +111,7 @@ class _InjectedTool(BaseTool):
         injected = _inject_params(input, self._token, self._identifier, self._base_tool)
         return super().invoke(injected, config=config, **kwargs)
 
-    async def ainvoke(
-        self, input: Any, config: RunnableConfig | None = None, **kwargs: Any
-    ) -> Any:
+    async def ainvoke(self, input: Any, config: RunnableConfig | None = None, **kwargs: Any) -> Any:
         """Invoke the tool asynchronously with token and optional identifier injected.
 
         Args:
@@ -140,9 +135,7 @@ class _InjectedTool(BaseTool):
         Returns:
             The result of running the tool with injected parameters.
         """
-        injected = _inject_params(
-            tool_input, self._token, self._identifier, self._base_tool
-        )
+        injected = _inject_params(tool_input, self._token, self._identifier, self._base_tool)
         return super().run(injected, **kwargs)
 
     async def arun(self, tool_input: Any, **kwargs: Any) -> Any:
@@ -155,9 +148,7 @@ class _InjectedTool(BaseTool):
         Returns:
             The result of running the tool with injected parameters.
         """
-        injected = _inject_params(
-            tool_input, self._token, self._identifier, self._base_tool
-        )
+        injected = _inject_params(tool_input, self._token, self._identifier, self._base_tool)
         return await super().arun(injected, **kwargs)
 
 
@@ -200,9 +191,7 @@ def GLConnectorTool(
     if not matching:
         raise ValueError(f"Tool '{tool_name}' not found in module '{module_name}'")
     if len(matching) > 1:
-        raise ValueError(
-            f"Multiple tools named '{tool_name}' found in module '{module_name}'"
-        )
+        raise ValueError(f"Multiple tools named '{tool_name}' found in module '{module_name}'")
 
     token = _create_token(
         connector,
@@ -249,9 +238,7 @@ def _load_env(*, api_key: str | None, identifier: str | None) -> dict[str, str]:
         for m in missing:
             preferred = _ENV_VAR_MAPPING[m][0]
             friendly_missing.append(preferred)
-        raise ValueError(
-            f"Missing required environment variables: {', '.join(friendly_missing)}"
-        )
+        raise ValueError(f"Missing required environment variables: {', '.join(friendly_missing)}")
 
     return {k: v for k, v in env.items() if v is not None}
 
@@ -291,21 +278,13 @@ def _resolve_module(tool_name: str, modules: Iterable[str]) -> str:
     Raises:
         ValueError: If no matching module is found or multiple ambiguous matches exist.
     """
-    candidates = [
-        module
-        for module in modules
-        if tool_name == module or tool_name.startswith(f"{module}_")
-    ]
+    candidates = [module for module in modules if tool_name == module or tool_name.startswith(f"{module}_")]
     if not candidates:
-        raise ValueError(
-            f"Unable to resolve module for tool '{tool_name}'. Available modules: {', '.join(modules)}"
-        )
+        raise ValueError(f"Unable to resolve module for tool '{tool_name}'. Available modules: {', '.join(modules)}")
 
     candidates.sort(key=len, reverse=True)
     if len(candidates) > 1 and len(candidates[0]) == len(candidates[1]):
-        raise ValueError(
-            f"Ambiguous module match for tool '{tool_name}'. Matches: {', '.join(candidates)}"
-        )
+        raise ValueError(f"Ambiguous module match for tool '{tool_name}'. Matches: {', '.join(candidates)}")
     return candidates[0]
 
 
@@ -334,9 +313,7 @@ def _create_token(connector: BosaConnector, username: str, password: str) -> str
     return token
 
 
-def _inject_params(
-    tool_input: Any, token: str, identifier: str | None, base_tool: BaseTool
-) -> dict[str, Any]:
+def _inject_params(tool_input: Any, token: str, identifier: str | None, base_tool: BaseTool) -> dict[str, Any]:
     """Inject token and optional identifier into tool input.
 
     Args:
@@ -374,9 +351,7 @@ def _inject_params(
     return injected
 
 
-def _wrap_request_if_needed(
-    tool_input: dict[str, Any], base_tool: BaseTool
-) -> dict[str, Any]:
+def _wrap_request_if_needed(tool_input: dict[str, Any], base_tool: BaseTool) -> dict[str, Any]:
     """Wrap flat inputs into a 'request' payload when required by schema.
 
     Args:
@@ -387,9 +362,7 @@ def _wrap_request_if_needed(
         Dictionary with inputs wrapped in 'request' key if needed, otherwise unchanged.
     """
     args_schema = getattr(base_tool, "args_schema", None)
-    if not (
-        isinstance(args_schema, dict) and "request" in args_schema.get("properties", {})
-    ):
+    if not (isinstance(args_schema, dict) and "request" in args_schema.get("properties", {})):
         return tool_input
 
     request_payload = {}

aip_agents/tools/gl_connector_tools.py

@@ -4,6 +4,9 @@ Authors:
     Saul Sayers (saul.sayers@gdplabs.id)
 """
 
+from bosa_connectors import BosaConnector, BOSAConnectorToolGenerator
+from langchain_core.tools import BaseTool
+
 from aip_agents.tools.constants import (
     GL_CONNECTORS_API_KEY,
     GL_CONNECTORS_BASE_URL,
@@ -11,8 +14,6 @@ from aip_agents.tools.constants import (
     ToolType,
 )
 from aip_agents.utils.logger import get_logger
-from bosa_connectors import BosaConnector, BOSAConnectorToolGenerator
-from langchain_core.tools import BaseTool
 
 logger = get_logger(__name__)
 
@@ -24,14 +25,10 @@ def get_gl_connector_modules_with_retry() -> list[str]:
         List of available modules.
     """
     if not GL_CONNECTORS_BASE_URL or not GL_CONNECTORS_API_KEY:
-        logger.warning(
-            "GL Connectors credentials missing (base_url or api_key); returning empty modules list"
-        )
+        logger.warning("GL Connectors credentials missing (base_url or api_key); returning empty modules list")
         return []
 
-    connector = BosaConnector(
-        api_base_url=GL_CONNECTORS_BASE_URL, api_key=GL_CONNECTORS_API_KEY
-    )
+    connector = BosaConnector(api_base_url=GL_CONNECTORS_BASE_URL, api_key=GL_CONNECTORS_API_KEY)
     modules = []
     for attempt in range(GL_CONNECTORS_FETCH_MAX_RETRIES):
         try:

{aip_agents_binary-0.5.22.dist-info → aip_agents_binary-0.5.23.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aip-agents-binary
-Version: 0.5.22
+Version: 0.5.23
 Summary: A library for managing agents in Gen AI applications.
 Author-email: Raymond Christopher <raymond.christopher@gdplabs.id>
 Requires-Python: <3.13,>=3.11
@@ -15,7 +15,7 @@ Requires-Dist: deprecated<2.0.0,>=1.2.18
 Requires-Dist: fastapi<0.121.0,>=0.120.0
 Requires-Dist: gllm-core-binary<0.4.0,>=0.3.18
 Requires-Dist: gllm-inference-binary[anthropic,bedrock,google-genai,google-vertexai,openai]<0.6.0,>=0.5.90
-Requires-Dist: gllm-tools-binary<0.2.0,>=0.1.3
+Requires-Dist: gllm-tools-binary<0.2.0,>=0.1.5
 Requires-Dist: google-adk<0.6.0,>=0.5.0
 Requires-Dist: langchain<0.4.0,>=0.3.0
 Requires-Dist: langchain-openai<0.4.0,>=0.3.17
@@ -34,7 +34,7 @@ Requires-Dist: gllm-privacy-binary<0.5.0,>=0.4.12; extra == "privacy"
 Provides-Extra: gl-connector
 Requires-Dist: bosa-connectors-binary<0.4.0,>=0.3.1; extra == "gl-connector"
 Provides-Extra: local
-Requires-Dist: e2b<2.0.0,>=1.5.4; extra == "local"
+Requires-Dist: e2b<3.0.0,>=2.3.0; extra == "local"
 Requires-Dist: browser-use==0.5.9; extra == "local"
 Requires-Dist: steel-sdk>=0.7.0; extra == "local"
 Requires-Dist: json-repair>=0.52.3; extra == "local"

{aip_agents_binary-0.5.22.dist-info → aip_agents_binary-0.5.23.dist-info}/RECORD

@@ -179,7 +179,7 @@ aip_agents/examples/hello_world_langflow_agent.py,sha256=NFgI_jJ0mDpeRpVhw5TESRS
 aip_agents/examples/hello_world_langflow_agent.pyi,sha256=sl3sF36CcY_s_zQ61gvneRyTiHmPV8p9xzPL2IeB1zo,1251
 aip_agents/examples/hello_world_langgraph.py,sha256=CSlZzVOtwbnCxw3uci_qGFzO7eYZe0uzs-4kmGNih7A,1248
 aip_agents/examples/hello_world_langgraph.pyi,sha256=ix3j1wqj79wkMMuOjEQUhZA4Isr5JSFEJw49WJ2NmM4,251
-aip_agents/examples/hello_world_langgraph_gl_connector_twitter.py,sha256=rZ517PtYjE9azJjtSOXkZgyc_g7pHejVS7Ij4gz-BqE,1327
+aip_agents/examples/hello_world_langgraph_gl_connector_twitter.py,sha256=cOvEKKr1cvyl1R3CX5c6JFUpwnOWYmAWtbJuqvwUsWA,1328
 aip_agents/examples/hello_world_langgraph_gl_connector_twitter.pyi,sha256=zvvkXFJw8NV4W7B8XB4b4F3s-HKixsasSMfJAuteqoQ,264
 aip_agents/examples/hello_world_langgraph_mcp_http.py,sha256=BmfwomTLNbGH3jpMMxGmSV_waYuvfbtdYdeKqVVzEPw,1008
 aip_agents/examples/hello_world_langgraph_mcp_http.pyi,sha256=8Ye3T6ip-_qykUEsYz5oPrJ99Td89OvFYwTdiP6-hAk,264
@@ -209,7 +209,7 @@ aip_agents/examples/hello_world_multi_agent_langgraph_lm_invoker.py,sha256=sppLD
 aip_agents/examples/hello_world_multi_agent_langgraph_lm_invoker.pyi,sha256=zFNns9i-sPY8aBy4HaVS6tXDWTsU7f-ApLD8gDtb1uk,252
 aip_agents/examples/hello_world_pii_logger.py,sha256=pm9LK-doZwQL-VdJYFHziuvwH562c8wwAwmKZeAWQS0,584
 aip_agents/examples/hello_world_pii_logger.pyi,sha256=5gOVrvhQV2DxIwr7ocL1-oTUXsO8XSyl5WN70c7rl_w,134
-aip_agents/examples/hello_world_sentry.py,sha256=UI9bkfoui4SjhHWSllJGUdaup3kYzGtoomZkY1zhcXg,4005
+aip_agents/examples/hello_world_sentry.py,sha256=Gccr6WCevSnMFZYT5M6uJ_22EvLmAMUE8Et_H510_kM,3976
 aip_agents/examples/hello_world_sentry.pyi,sha256=k0wFj46QqXEjBU0CEERmiEf2bovVFkupzuzQzeD6h00,671
 aip_agents/examples/hello_world_step_limits.py,sha256=hRV9XERlq2qW_yZ4kHAe1hTgukrSZjThIrsY9NuTdJg,9809
 aip_agents/examples/hello_world_step_limits.pyi,sha256=V9KP6r5OEVq2lNUfnqMeCGH1nsNewDsRL1Uo9up8xis,1060
@@ -292,7 +292,7 @@ aip_agents/mcp/client/persistent_session.py,sha256=I8JjHe_ZqU1PPB2OF6hUX4kcrraVW
 aip_agents/mcp/client/persistent_session.pyi,sha256=in3TgtC-eHD6j2payC9_TryuOLOmqoeDKfl7UaZ_kBA,3908
 aip_agents/mcp/client/session_pool.py,sha256=_J8WduSo3HAfhE5n4u67IQQ71m_L2aPUY1NOgX5e7yA,12617
 aip_agents/mcp/client/session_pool.pyi,sha256=RtzN-5QDLS5MFAlnR5TiarY1xW4xo780n7lQL0sQbRU,3579
-aip_agents/mcp/client/transports.py,sha256=1HqmFqpmktFhR-vKU85xuvWcwogkIjQIh9se0qo4LEM,8172
+aip_agents/mcp/client/transports.py,sha256=i7cJ1_vUUrE4yVTq-nFf0AnfKv0j9k1d8EcVZBykTUI,8624
 aip_agents/mcp/client/transports.pyi,sha256=_V6ZaQyKtTMhHzNQH943asy0O6z2bHTOX8hc-lZlV2s,4576
 aip_agents/mcp/client/google_adk/__init__.py,sha256=ZJZE7IusoWFzEwaWsHOk-8VMdR-LUmWacepw_HHuOlI,280
 aip_agents/mcp/client/google_adk/__init__.pyi,sha256=TAbiDbysxbCtHQSASGdko6JIaZEnPbCmUqt4Jf966cs,127
@@ -350,7 +350,7 @@ 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=DgYaIph-VQp0b36sUFvBqxe9lmbBZv1i1edg0Tx5nbE,4595
+aip_agents/sentry/sentry.py,sha256=zyhIZCnA2mcLuf5w-ezE_7ZWOjPdGt3A5g7AcJrPgxY,4504
 aip_agents/sentry/sentry.pyi,sha256=fTFb71FfeByBUIACjkp8qaHsWe6C52Y4tPTfizAPVv0,1432
 aip_agents/storage/__init__.py,sha256=cN4L2whui-DOm81tsYV88Wx5g5-cUjeLVyp55RtbJJU,1219
 aip_agents/storage/__init__.pyi,sha256=wn8VuS7x4WSzKuV01WrjPkZfnh3GylNum9FlPiaSdsA,901
@@ -370,11 +370,11 @@ aip_agents/storage/providers/memory.py,sha256=81E9yT_oRkH7tMVr5azZv1oXIaeTzTAtTu
 aip_agents/storage/providers/memory.pyi,sha256=r0meeCsjKyE9FHJqCysfQ5ZH_FWRRv2SWg-NOHCUD-g,2109
 aip_agents/storage/providers/object_storage.py,sha256=Q268Sn8Y09gS-ilP6fe4CR0YIOvdbunh3V6SmuQsAVs,6413
 aip_agents/storage/providers/object_storage.pyi,sha256=nBIKJjUAWkIKhYhBFmFcypjoOieIYOghy55f62x7AX8,2878
-aip_agents/tools/__init__.py,sha256=HWj2nsjMBadA6aJdPlBQBSBxbvDAYmMKe3PJeNyDRbA,1935
+aip_agents/tools/__init__.py,sha256=J0wDdcKre3Qc_z1hhzKfPiaa1m2OPOXyHjl6uwOkApU,1935
 aip_agents/tools/__init__.pyi,sha256=w1gRWL5iRdLtuBNXBhmD87uzfvIRrXwq3uZ1meku8js,888
-aip_agents/tools/constants.py,sha256=S8-ctDo4FXp5Si8dFCJ2FORPPS1SKD2OQT5II3j1yak,5731
+aip_agents/tools/constants.py,sha256=AabnuPQG_mc2sVdr9jV23_6bFempAsxQv3kdCR_ztLA,5723
 aip_agents/tools/constants.pyi,sha256=kFY8dKgRqHKGvPqG3QUyuEc8C5yRaDj7DYQDH88K2T0,3552
-aip_agents/tools/gl_connector_tools.py,sha256=yXLuGTTAUxCYpOG808pKcHEZIntNI_x-JUq3OqzxexI,3971
+aip_agents/tools/gl_connector_tools.py,sha256=bxl_3VQYZDv3lFn6Y3kDVVRFwH4cntOLz3f74YzDcic,3936
 aip_agents/tools/gl_connector_tools.pyi,sha256=2ATn_MW_FRg5Uv7dLI_ToBOtlgTSfj0zgDQpN1N-cJs,1366
 aip_agents/tools/memory_search_tool.py,sha256=gqTTb_Fao_Hl-SavfaFsqPDhnFQUjzIQUkzizSD2L0A,653
 aip_agents/tools/memory_search_tool.pyi,sha256=BCVFEEgH3pETnI9b6vRlMm2_PnnIeBe_vuMlL9y3LpU,453
@@ -412,9 +412,9 @@ aip_agents/tools/code_sandbox/__init__.py,sha256=QPUBMzmSDGjvjnLRRJRHK7shP16LH6K
 aip_agents/tools/code_sandbox/__init__.pyi,sha256=hBgsW_ZdnGonBx7PtCwz3ohs8RuZcDpQMU6iK_VZIYg,134
 aip_agents/tools/code_sandbox/constant.py,sha256=55uM9sHOISdT40O3pVdLfL88YU-1Npy7CXkYxlLFzug,679
 aip_agents/tools/code_sandbox/constant.pyi,sha256=3bwRpYOZZNVYjBBvPXnTR_cq6dH-pALsy_5yCbxdPQU,81
-aip_agents/tools/code_sandbox/e2b_cloud_sandbox_extended.py,sha256=H91-1a3keS6Si5hLzQ7Z1tHM7ZWEv6p2fJQ6HSqttw0,9821
-aip_agents/tools/code_sandbox/e2b_cloud_sandbox_extended.pyi,sha256=ZnW7S-Wla5J-ZD8P92-sXuzTGZ2-z5zyv4aH97FlshI,3554
-aip_agents/tools/code_sandbox/e2b_sandbox_tool.py,sha256=sTFTQd9Li9lB2iXgsARfpJEfFz9O2_sdFYFh8gFvC6A,16954
+aip_agents/tools/code_sandbox/e2b_cloud_sandbox_extended.py,sha256=78mrQ4Tfv0HQst92NugC2_ogw7LU0ubx8XKFAbW53vY,12013
+aip_agents/tools/code_sandbox/e2b_cloud_sandbox_extended.pyi,sha256=W2NIk0b9dK-HTResYt1ppQdAcBLp2963D94XHX9hq6E,4366
+aip_agents/tools/code_sandbox/e2b_sandbox_tool.py,sha256=TChVXscOR1WXmpjelwcZ_Ws4pYtuYhLALbvPXYz6iSU,16937
 aip_agents/tools/code_sandbox/e2b_sandbox_tool.pyi,sha256=ylrBQaqNBa1Jppld-mHjaskSa9SOjsQD8TdcK2bnl4s,1165
 aip_agents/tools/document_loader/__init__.py,sha256=rnQFLYJqvit5eegGIWGdjOUoEJiyOh3nW0mLSRd7xfE,1375
 aip_agents/tools/document_loader/__init__.pyi,sha256=RYZb-EdfR1btPxFBUwfmrOWs51aVgwJuw9epguxnVgQ,650
@@ -430,7 +430,7 @@ aip_agents/tools/document_loader/pdf_splitter.py,sha256=-QyrlnN4AXDqY_dxeUzxcgCV
 aip_agents/tools/document_loader/pdf_splitter.pyi,sha256=3IiRDQWDz8QR5LH_sni9O-N2qJFheFjKOQMAhiWxB7E,716
 aip_agents/tools/gl_connector/__init__.py,sha256=f8F4mdBFj0ulxewCQwG5qN2SDzlgI2jA0Kfj31A5iD0,137
 aip_agents/tools/gl_connector/__init__.pyi,sha256=96wtNkB3VUSI66aRlYxVrzMiPtqOYviRMKviRgX3_fc,113
-aip_agents/tools/gl_connector/tool.py,sha256=EmfAYU5ZEtG93osWPWd4ElfXIhBe7mqz0yIH6Ntc9sE,13343
+aip_agents/tools/gl_connector/tool.py,sha256=jzT8XmTfFQC9ZcQplVcRs2VmCtKewH9FzT7wSFtUJac,13106
 aip_agents/tools/gl_connector/tool.pyi,sha256=a5l0MHSOe_iWDvjMRzYtcbMdX0bFlK1m7Hl52HQ__iQ,2770
 aip_agents/tools/memory_search/__init__.py,sha256=YSsObYlHjdZEbJj4MVYy3Ht8JPlo42YhjnkI-yFNWV0,608
 aip_agents/tools/memory_search/__init__.pyi,sha256=NG0g94OoC_xw66yIqiPViqTnpj01QdnRsVBGzkSxJFI,554
@@ -540,7 +540,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.5.22.dist-info/METADATA,sha256=CocpyU4hmWP-sT9ti3uGtzFDcULe2Vr2sJVH-Ae2ftM,22214
-aip_agents_binary-0.5.22.dist-info/WHEEL,sha256=PaP4PvkDyiSc4C6Dhw6ccQmfxsWFrSv-lJQjBshu0hw,105
-aip_agents_binary-0.5.22.dist-info/top_level.txt,sha256=PEz8vcwC1bH4UrkhF0LkIYCNfXGWZUHdSklbvkBe25E,11
-aip_agents_binary-0.5.22.dist-info/RECORD,,
+aip_agents_binary-0.5.23.dist-info/METADATA,sha256=VlQHX0DiRpcp68ytt7zTHreClijm1LVv-IKEoz23044,22214
+aip_agents_binary-0.5.23.dist-info/WHEEL,sha256=PaP4PvkDyiSc4C6Dhw6ccQmfxsWFrSv-lJQjBshu0hw,105
+aip_agents_binary-0.5.23.dist-info/top_level.txt,sha256=PEz8vcwC1bH4UrkhF0LkIYCNfXGWZUHdSklbvkBe25E,11
+aip_agents_binary-0.5.23.dist-info/RECORD,,