aip-agents-binary 0.5.22__py3-none-any.whl → 0.5.23__py3-none-any.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=pkchyUiTM2VlWUcW2TDZLzo
 aip_agents/examples/hello_world_langflow_agent.pyi,sha256=8cwZzGBqv_gaf-M7lGewOSL0XXxYjufgg2ZlXpKN2DQ,1286
 aip_agents/examples/hello_world_langgraph.py,sha256=uTsZ7X5up_BmQqKTKh0F1fsvkXpm9puOANXd33-pb9g,1287
 aip_agents/examples/hello_world_langgraph.pyi,sha256=nCbikXhA0CFeIgDFoD4UcwkAXvEUUgwZCUXwc8jLWno,256
-aip_agents/examples/hello_world_langgraph_gl_connector_twitter.py,sha256=FkNFkIG9GIj0rxqSGJ4maz-lyPHawqaKveB4vvhFF1M,1370
+aip_agents/examples/hello_world_langgraph_gl_connector_twitter.py,sha256=1ScB8V50LCPLadOFWDdDGAGMEc2U5oT4EZmkNfdHuzI,1372
 aip_agents/examples/hello_world_langgraph_gl_connector_twitter.pyi,sha256=Xku8mu9_2QbNpYEJH8imik4IDL6hQKYqpGtPs_AHTlU,269
 aip_agents/examples/hello_world_langgraph_mcp_http.py,sha256=8Vu3Q_RChDlBgPcbh9h7U4Cy7x4SakRGttOtQeHluyI,1039
 aip_agents/examples/hello_world_langgraph_mcp_http.pyi,sha256=M1LXhrQOUDquRcGOWOtu_tbnjcOU0qtmEvizDa1H4PA,269
@@ -209,7 +209,7 @@ aip_agents/examples/hello_world_multi_agent_langgraph_lm_invoker.py,sha256=B0I92
 aip_agents/examples/hello_world_multi_agent_langgraph_lm_invoker.pyi,sha256=OHkTLSqRAM7TOZEp6TH5dsPq9zbpi9UA_l0rDIrqFlM,257
 aip_agents/examples/hello_world_pii_logger.py,sha256=xH4PNGetuZv1TJj5Y1fqvm1Vto0pp1u5ju-ORccEFwo,605
 aip_agents/examples/hello_world_pii_logger.pyi,sha256=tZ0oOfd_gK97KHczzEh3PQfTt2eedh25IL7Q1PhVDWU,139
-aip_agents/examples/hello_world_sentry.py,sha256=tyvRNrHlAiaS0DB6Y_M8b_2zXQSxWD7tKFvHK_ydo6I,4139
+aip_agents/examples/hello_world_sentry.py,sha256=HRnIVTkaqcUGbVbVVCzB8HQv5nVqHbt_3aPiflrD31M,4109
 aip_agents/examples/hello_world_sentry.pyi,sha256=Z6t3ywrVlgBI9NFxbcbhRVAwt71Q3NsB_JvwZhcd0aQ,692
 aip_agents/examples/hello_world_step_limits.py,sha256=0f02AFrpfAertnjSHxQiU6uAyIWt0ovQ3aQ5QCJPSx4,10082
 aip_agents/examples/hello_world_step_limits.pyi,sha256=U9kx6rIgfZuNHxeAIiyW_Ae-KT65kNqCdCMQzKEKlBc,1077
@@ -292,7 +292,7 @@ aip_agents/mcp/client/persistent_session.py,sha256=x7YUX3EwIg4Ixs0D7Sb7-SyYxNYlj
 aip_agents/mcp/client/persistent_session.pyi,sha256=eMqoGE84JJtDDwiJJSNIsjwWrldB9luaVA7UeXlBGrc,4021
 aip_agents/mcp/client/session_pool.py,sha256=qYxtalGyT1Z4a5LIDe2X9QinduOz4f_U9yDzBH_qjKI,12968
 aip_agents/mcp/client/session_pool.pyi,sha256=dUXl8g4G4asZ1CplaeZaR24suyU4gAgKzvQNOlriQA0,3680
-aip_agents/mcp/client/transports.py,sha256=YF9ZwnVH1G4z6Wiv3IcFptq3D37bqlfjvF5s_rP0hnI,8387
+aip_agents/mcp/client/transports.py,sha256=VsxJmNDb86_kdEmuc9T0764nELboc4_VvCLQ8OUDUDk,8852
 aip_agents/mcp/client/transports.pyi,sha256=LCZfHvwHRwk4T3EyK2RerpEFyIfwfVFuon2igbEVH28,4699
 aip_agents/mcp/client/google_adk/__init__.py,sha256=mbVak_4MCHVNkgm6EQhkrR1HDAPo1Q7cP3nk55TPtu4,291
 aip_agents/mcp/client/google_adk/__init__.pyi,sha256=lG0PHv3Rcvl4BOeUkmJC5nx2HQJsmUMnYmC9iD99l0s,130
@@ -350,7 +350,7 @@ aip_agents/schema/storage.py,sha256=sH2VP0BE6-SNVb4YFs6QJdtKU_Cg9DQ2wmtMgnHpISA,
 aip_agents/schema/storage.pyi,sha256=pqfdgfT1kja5FM3aN_uqMpw5ewTpNUDr1Feg0K7Bfl0,578
 aip_agents/sentry/__init__.py,sha256=SluWkDegy1aOArrEos_-qINXlU7-qWuFLHK4VsQRFQs,272
 aip_agents/sentry/__init__.pyi,sha256=l3OSA6R-PI5NBsEcTd0QG9AX59uDzNrbasA6EJJMzTU,106
-aip_agents/sentry/sentry.py,sha256=vqI-5EXmOVygY8Q5I324NTqP77feudZr0AdQ1Lc5C-8,4753
+aip_agents/sentry/sentry.py,sha256=hIPaxrnG35Mn3DHZDluuAbMktGuy0N-ZRge3anOIbM0,4655
 aip_agents/sentry/sentry.pyi,sha256=kBHuQrfhqY9ifq3BbZdcjn0kedIWUiLtO78NX4-CdaE,1480
 aip_agents/storage/__init__.py,sha256=GIdxlSMf59hPPeurfnJpqh70JnF3LbEWH7tYFAO1tjI,1260
 aip_agents/storage/__init__.pyi,sha256=jVVcK1_V657EPjpZn5y_jcn-HbiiqMPG9oLIbeVaUlI,909
@@ -370,11 +370,11 @@ aip_agents/storage/providers/memory.py,sha256=Zd7QwYacEfaHLCpjvV3XaHSRulhfS464U0
 aip_agents/storage/providers/memory.pyi,sha256=NpsqKzboQGWtxtFV1ch_C3KT0KHOLyac6TtBgkLRgjI,2188
 aip_agents/storage/providers/object_storage.py,sha256=RlhXdaL1YTH9oqf7crvftsicHRxJlaLO3UgLPRCWlfg,6627
 aip_agents/storage/providers/object_storage.pyi,sha256=4pvDaNqhDleOm3ijMe_Yk3meBn2UCpLP9bKAH1l6C7w,2976
-aip_agents/tools/__init__.py,sha256=b36XoUy1CrJ1l2NxIOZSEBaOJAvx-a2BZs6nOJ9Q4_8,1988
+aip_agents/tools/__init__.py,sha256=-4NJEX914Zhvl7ivBy6ztWowtzpaR4r4ammXFydjWGw,1988
 aip_agents/tools/__init__.pyi,sha256=b4plRUwyRypopaZfKMFSj0kTGb0paYL1FPm0EsWRPVw,897
-aip_agents/tools/constants.py,sha256=JqML3QtFis8SxlpgL_t2AG5HN6xBKZuDaDAo_eHI3HU,5910
+aip_agents/tools/constants.py,sha256=xNf9_pv_Rf4TCODr7BYLO05Kw5cD55MWiYfDyLWPJ3Y,5900
 aip_agents/tools/constants.pyi,sha256=ftkzuxDMnVUqVeqm4W848olK-VSlRH4oSv1p66sZnHk,3690
-aip_agents/tools/gl_connector_tools.py,sha256=w8DIK09okXUHsxBc3UTsUE-F65i4AfabfF2dlYEA33Y,4093
+aip_agents/tools/gl_connector_tools.py,sha256=1VrXUKRiYO21v4JWOaoo1XNaIKSm_0KmXp3AMx_UL_8,4055
 aip_agents/tools/gl_connector_tools.pyi,sha256=wPK8hjFEZvQtRxBwX2MEebw6d4bXzn7nWxvcWPgYJ-c,1405
 aip_agents/tools/memory_search_tool.py,sha256=ClXW6BEjZNHb6NHe44KxbwZ6GSGo7v7AUzaV99LMouc,679
 aip_agents/tools/memory_search_tool.pyi,sha256=kjUaPWFrUIHrCTla_7ccnEWUNb44v0ZHLxyKGLPyeWg,456
@@ -412,9 +412,9 @@ aip_agents/tools/code_sandbox/__init__.py,sha256=jsBesvd07L8MGInVoBXkss_BxKK3XGK
 aip_agents/tools/code_sandbox/__init__.pyi,sha256=E-Ub6SWqRmx1bL8cQAbCBV4Ao8a8OjsC8skub2iZ9mw,137
 aip_agents/tools/code_sandbox/constant.py,sha256=pfDb03wmFcVh0uggWA_0bj6CWeUSHC7ZikVKokeFcRQ,692
 aip_agents/tools/code_sandbox/constant.pyi,sha256=seNiSpmQVwmC1mbYRloaIk511RuOPlYQv8GHCwBczDU,85
-aip_agents/tools/code_sandbox/e2b_cloud_sandbox_extended.py,sha256=mBX9Om3sQgXelgU2IeBSl1ZqIAkiR8e0UTw0JITvpLg,10078
-aip_agents/tools/code_sandbox/e2b_cloud_sandbox_extended.pyi,sha256=GlOGnJA_iHFpf5xrhzdbNaVkGHNB1xaCzN6eori9mxI,3640
-aip_agents/tools/code_sandbox/e2b_sandbox_tool.py,sha256=S5-THddk2cgEWISXcx8ETJ5Ad9LKfzvlDRn7QLounaY,17365
+aip_agents/tools/code_sandbox/e2b_cloud_sandbox_extended.py,sha256=vUNTXqpL1-7civtH5YqOnXKwb8_txHyVDpf1vns49oM,12319
+aip_agents/tools/code_sandbox/e2b_cloud_sandbox_extended.pyi,sha256=LxWVmVvBeoN1oWFYOsv8gVodtFxkIpVDq-fM_MthI4Y,4468
+aip_agents/tools/code_sandbox/e2b_sandbox_tool.py,sha256=l_bQC51UwIGYOIm4R6XNdLMjydrCvjUe31wBFzym05A,17348
 aip_agents/tools/code_sandbox/e2b_sandbox_tool.pyi,sha256=ANJMJJ4HIHtiWObJ_PVRjo_DyMwkjDiWEMY8mmYLfdo,1194
 aip_agents/tools/document_loader/__init__.py,sha256=D-NVmHr7hb21ctpFrpAoj5QaC0JBNYekNZ3zV6Zn4lE,1419
 aip_agents/tools/document_loader/__init__.pyi,sha256=T4JAnBCGWG1ZZz7-1fZwg210ltO_74dpHOhs9hgekqI,657
@@ -430,7 +430,7 @@ aip_agents/tools/document_loader/pdf_splitter.py,sha256=-phEFl9DBoG3epUmc9ajtZ57
 aip_agents/tools/document_loader/pdf_splitter.pyi,sha256=rUsKSMPevXY91GpYIKEBvcVzBjLIX4PKdBfL4ywD4QU,734
 aip_agents/tools/gl_connector/__init__.py,sha256=un2-A8iOnO3oKx6ZtUIMQTLgZLmtl8SQELMvrAXeTiE,142
 aip_agents/tools/gl_connector/__init__.pyi,sha256=cJXQnMCwZbn4Qs2umOwRsIK24kcg1zmTikO-LNHK_Q4,116
-aip_agents/tools/gl_connector/tool.py,sha256=Ur40UvXP50-K_L388b5yBIDJCDhQtTtTSRijQyRqm1s,13753
+aip_agents/tools/gl_connector/tool.py,sha256=D6CjeJiKvg0321Ok3Ylchl23-iEOVcUCOZQlxj5utxQ,13489
 aip_agents/tools/gl_connector/tool.pyi,sha256=gX30AG8P4YRpuFkt7xnj-l3O6Ri1SZENfnpJcZ2Vxl0,2844
 aip_agents/tools/memory_search/__init__.py,sha256=jonWk8gERcM-f0l_xBIzNyHqvKXSNWXBDUdtkCbvZ48,630
 aip_agents/tools/memory_search/__init__.pyi,sha256=9zElnDT6yiEQQvod0_YmIK6tL_ugHhBa6r21lRPp1Is,559
@@ -540,7 +540,7 @@ aip_agents/utils/pii/pii_helper.py,sha256=8QGVC9lb7dic_vSfLDUaDvqm45BUbYyPeQTVva
 aip_agents/utils/pii/pii_helper.pyi,sha256=wEgOasJxwKObtQ9Cb1UsiyIaqq2JUYmdRwOZR9PnIjA,3017
 aip_agents/utils/pii/uuid_deanonymizer_mapping.py,sha256=X9zeX1bhb3rlCc8P5QnbHCILx2AIhGmZwjsjh_2G4ZQ,7543
 aip_agents/utils/pii/uuid_deanonymizer_mapping.pyi,sha256=6H1xRV2Nr0LpP5K6fbz2uCobehTpM2626v8kiOd9W9Y,3157
-aip_agents_binary-0.5.22.dist-info/METADATA,sha256=tAd2H7_0vofhoz0wiwNGO-Ocla82oRKP9LacX7SyrPw,22895
-aip_agents_binary-0.5.22.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-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=_a-Fvb_Eb9t0DaR-rkSlMXX1WU0lO1kk9N9bNFUi-ic,22895
+aip_agents_binary-0.5.23.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aip_agents_binary-0.5.23.dist-info/top_level.txt,sha256=PEz8vcwC1bH4UrkhF0LkIYCNfXGWZUHdSklbvkBe25E,11
+aip_agents_binary-0.5.23.dist-info/RECORD,,