aip-agents-binary 0.5.21__py3-none-any.whl → 0.5.23__py3-none-any.whl

aip_agents/examples/{hello_world_langgraph_bosa_twitter.py → hello_world_langgraph_gl_connector_twitter.py}

@@ -1,4 +1,4 @@
-"""Minimal LangGraph agent with bosa support example demonstrating asynchronous run.
+"""Minimal LangGraph agent with GL Connectors support example demonstrating asynchronous run.
 
 Authors:
     Saul Sayers (saul.sayers@gdplabs.id)
@@ -6,22 +6,25 @@ Authors:
 
 import asyncio
 
+from dotenv import load_dotenv
 from langchain_openai import ChatOpenAI
 
 from aip_agents.agent import LangGraphAgent
-from aip_agents.tools import BOSA_AUTOMATED_TOOLS
+from aip_agents.tools import GL_CONNECTORS_AUTOMATED_TOOLS
 
+load_dotenv(override=True)
 
-async def langgraph_bosa_example():
+
+async def langgraph_gl_connector_example():
     """Demonstrates the LangGraphAgent's arun method."""
     model = ChatOpenAI(model="gpt-4.1", temperature=0)
-    agent_name = "BOSAConnectorTwitterAgent"
+    agent_name = "GLConnectorTwitterAgent"
 
     langgraph_agent = LangGraphAgent(
         name=agent_name,
-        instruction="You are a helpful assistant that use BOSA connector to connect with Twitter API.",
+        instruction="You are a helpful assistant that uses GL Connectors to connect with the Twitter API.",
         model=model,
-        tools=BOSA_AUTOMATED_TOOLS["twitter"],
+        tools=GL_CONNECTORS_AUTOMATED_TOOLS["twitter"],
     )
 
     query = "Get 3 tweets about the latest Air India Incident"
@@ -38,4 +41,4 @@ async def langgraph_bosa_example():
 
 
 if __name__ == "__main__":
-    asyncio.run(langgraph_bosa_example())
+    asyncio.run(langgraph_gl_connector_example())

aip_agents/examples/hello_world_langgraph_gl_connector_twitter.pyi

@@ -0,0 +1,5 @@
+from aip_agents.agent import LangGraphAgent as LangGraphAgent
+from aip_agents.tools import GL_CONNECTORS_AUTOMATED_TOOLS as GL_CONNECTORS_AUTOMATED_TOOLS
+
+async def langgraph_gl_connector_example() -> None:
+    """Demonstrates the LangGraphAgent's arun method."""

aip_agents/examples/hello_world_sentry.py

@@ -1,4 +1,4 @@
-"""This is a simple FastAPI app to test Sentry with Open Telemetry from BOSA SDK.
+"""This is a simple FastAPI app to test Sentry with Open Telemetry from GL Connectors SDK.
 
 To run this example, make sure to set the following environment variables:
     SENTRY_DSN
@@ -63,7 +63,7 @@ def create_app() -> FastAPI:
         FastAPI: The configured FastAPI application.
     """
     app = FastAPI(
-        title="Bosa SDK - Sentry with Open Telemetry",
+        title="GL Connectors SDK - Sentry with Open Telemetry",
         description="This is a simple FastAPI app to test Sentry with Open Telemetry.",
         version="0.0.1",
         docs_url=None,

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/__init__.py

@@ -1,4 +1,4 @@
-"""This module provides a centralized way to handle Sentry and OpenTelemetry configuration for BOSA SDK.
+"""This module provides a centralized way to handle Sentry and OpenTelemetry configuration for GL Connectors SDK.
 
 Authors:
     Saul Sayers (saul.sayers@gdplabs.id)

aip_agents/sentry/sentry.py

@@ -1,4 +1,4 @@
-"""This file contains the Sentry and OpenTelemetry configuration for BOSA SDK.
+"""This file contains the Sentry and OpenTelemetry configuration for GL Connectors SDK.
 
 Authors:
     Saul Sayers (saul.sayers@gdplabs.id)
@@ -61,8 +61,8 @@ def get_all_methods(cls: type) -> list:
     return methods
 
 
-def instrument_bosa_functions() -> None:
-    """Instrument BOSA functions."""
+def instrument_gl_functions() -> None:
+    """Instrument GL functions."""
     if BOSAFunctionsInstrumentor is None:
         return
     agent_methods = []
@@ -148,4 +148,4 @@ def setup_telemetry(app: FastAPI) -> None:
         setup_sentry_with_open_telemetry(app)
     else:
         setup_sentry_only()
-    instrument_bosa_functions()
+    instrument_gl_functions()

aip_agents/sentry/sentry.pyi

@@ -21,8 +21,8 @@ def get_all_methods(cls) -> list:
     Returns:
         list: A list of methods.
     """
-def instrument_bosa_functions() -> None:
-    """Instrument BOSA functions."""
+def instrument_gl_functions() -> None:
+    """Instrument GL functions."""
 def traces_sampler(*args) -> float:
     """Determine appropriate sampling rate for Sentry transactions.
 

aip_agents/tools/__init__.py

@@ -3,15 +3,21 @@
 from importlib import import_module
 from typing import TYPE_CHECKING
 
-from aip_agents.tools.bosa_tools import 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
 
 logger = get_logger(__name__)
 
-__all__ = ["BOSA_AUTOMATED_TOOLS", "GLConnectorTool", "GoogleSerperTool", "TimeTool"]
+__all__ = [
+    "BOSA_AUTOMATED_TOOLS",
+    "GL_CONNECTORS_AUTOMATED_TOOLS",
+    "GLConnectorTool",
+    "GoogleSerperTool",
+    "TimeTool",
+]
 
 
 def _register_optional(module_path: str, export_name: str) -> None:

aip_agents/tools/__init__.pyi

@@ -1,9 +1,9 @@
-from aip_agents.tools.bosa_tools import BOSA_AUTOMATED_TOOLS as BOSA_AUTOMATED_TOOLS
 from aip_agents.tools.browser_use import BrowserUseTool as BrowserUseTool
 from aip_agents.tools.code_sandbox import E2BCodeSandboxTool as E2BCodeSandboxTool
 from aip_agents.tools.document_loader import DocxReaderTool as DocxReaderTool, ExcelReaderTool as ExcelReaderTool, PDFReaderTool as PDFReaderTool
 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', 'GLConnectorTool', 'GoogleSerperTool', 'TimeTool', 'BrowserUseTool', 'E2BCodeSandboxTool', 'DocxReaderTool', 'ExcelReaderTool', 'PDFReaderTool']
+__all__ = ['BOSA_AUTOMATED_TOOLS', 'GL_CONNECTORS_AUTOMATED_TOOLS', 'GLConnectorTool', 'GoogleSerperTool', 'TimeTool', 'BrowserUseTool', 'E2BCodeSandboxTool', 'DocxReaderTool', 'ExcelReaderTool', 'PDFReaderTool']

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

@@ -1,4 +1,4 @@
-"""Constants for tools using BOSA Connector.
+"""Constants for tools using GL Connectors.
 
 Authors:
     Saul Sayers (saul.sayers@gdplabs.id)
@@ -7,20 +7,32 @@ Authors:
 import os
 from enum import Enum, StrEnum
 
-BOSA_API_BASE_URL = os.getenv("BOSA_API_BASE_URL")
-BOSA_API_KEY = os.getenv("BOSA_API_KEY")
-BOSA_FETCH_MAX_RETRIES = 3
+GL_CONNECTORS_BASE_URL = (
+    os.getenv("GL_CONNECTORS_BASE_URL")
+    or os.getenv("GL_CONNECTORS_API_BASE_URL")
+    or os.getenv("BOSA_API_BASE_URL")
+    or os.getenv("BOSA_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
+)
+
+# For backward compatibility
+BOSA_API_BASE_URL = GL_CONNECTORS_BASE_URL
+BOSA_API_KEY = GL_CONNECTORS_API_KEY
+BOSA_FETCH_MAX_RETRIES = GL_CONNECTORS_FETCH_MAX_RETRIES
 
 
 class ToolType(StrEnum):
-    """Tool types for BOSA Connector."""
+    """Tool types for GL Connectors."""
 
     GLLM = "gllm"
     LANGCHAIN = "langchain"
 
 
 class Action(Enum):
-    """Actions for BOSA Connector."""
+    """Actions for GL Connectors."""
 
     GITHUB = "github"
     GOOGLE = "google"
@@ -30,7 +42,7 @@ class Action(Enum):
 
 
 class GitHubEndpoint(Enum):
-    """GitHub endpoints for BOSA Connector."""
+    """GitHub endpoints for GL Connectors."""
 
     INTEGRATIONS = "integrations"
     USER_HAS_INTEGRATION = "integration-exists"
@@ -59,7 +71,7 @@ class GitHubEndpoint(Enum):
 
 
 class GoogleDriveEndpoint(Enum):
-    """Google Drive endpoints for BOSA Connector."""
+    """Google Drive endpoints for GL Connectors."""
 
     INTEGRATIONS = "integrations"
     USER_HAS_INTEGRATION = "integration-exists"
@@ -84,7 +96,7 @@ class GoogleDriveEndpoint(Enum):
 
 
 class GoogleDocsEndpoint(Enum):
-    """Google Docs endpoints for BOSA Connector."""
+    """Google Docs endpoints for GL Connectors."""
 
     INTEGRATIONS = "integrations"
     USER_HAS_INTEGRATION = "integration-exists"
@@ -100,7 +112,7 @@ class GoogleDocsEndpoint(Enum):
 
 
 class GoogleEndpoint(Enum):
-    """Google endpoints for BOSA Connector."""
+    """Google endpoints for GL Connectors."""
 
     INTEGRATIONS = "integrations"
     USER_HAS_INTEGRATION = "integration-exists"
@@ -109,7 +121,7 @@ class GoogleEndpoint(Enum):
 
 
 class TwitterEndpoint(Enum):
-    """Twitter endpoints for BOSA Connector."""
+    """Twitter endpoints for GL Connectors."""
 
     INTEGRATIONS = "integrations"
     USER_HAS_INTEGRATION = "integration-exists"
@@ -121,7 +133,7 @@ class TwitterEndpoint(Enum):
 
 
 class GoogleMailEndpoint(Enum):
-    """Google Mail endpoints for BOSA Connector."""
+    """Google Mail endpoints for GL Connectors."""
 
     INTEGRATIONS = "integrations"
     USER_HAS_INTEGRATION = "integration-exists"

aip_agents/tools/constants.pyi

@@ -1,17 +1,20 @@
 from _typeshed import Incomplete
 from enum import Enum, StrEnum
 
-BOSA_API_BASE_URL: Incomplete
-BOSA_API_KEY: Incomplete
-BOSA_FETCH_MAX_RETRIES: int
+GL_CONNECTORS_BASE_URL: Incomplete
+GL_CONNECTORS_API_KEY: Incomplete
+GL_CONNECTORS_FETCH_MAX_RETRIES: Incomplete
+BOSA_API_BASE_URL = GL_CONNECTORS_BASE_URL
+BOSA_API_KEY = GL_CONNECTORS_API_KEY
+BOSA_FETCH_MAX_RETRIES = GL_CONNECTORS_FETCH_MAX_RETRIES
 
 class ToolType(StrEnum):
-    """Tool types for BOSA Connector."""
+    """Tool types for GL Connectors."""
     GLLM: str
     LANGCHAIN: str
 
 class Action(Enum):
-    """Actions for BOSA Connector."""
+    """Actions for GL Connectors."""
     GITHUB: str
     GOOGLE: str
     GOOGLE_DRIVE: str
@@ -19,7 +22,7 @@ class Action(Enum):
     TWITTER: str
 
 class GitHubEndpoint(Enum):
-    """GitHub endpoints for BOSA Connector."""
+    """GitHub endpoints for GL Connectors."""
     INTEGRATIONS: str
     USER_HAS_INTEGRATION: str
     SUCCESS_AUTHORIZE_CALLBACK: str
@@ -46,7 +49,7 @@ class GitHubEndpoint(Enum):
     LIST_PROJECTS: str
 
 class GoogleDriveEndpoint(Enum):
-    """Google Drive endpoints for BOSA Connector."""
+    """Google Drive endpoints for GL Connectors."""
     INTEGRATIONS: str
     USER_HAS_INTEGRATION: str
     SUCCESS_AUTHORIZE_CALLBACK: str
@@ -69,7 +72,7 @@ class GoogleDriveEndpoint(Enum):
     DOWNLOAD_FILE: str
 
 class GoogleDocsEndpoint(Enum):
-    """Google Docs endpoints for BOSA Connector."""
+    """Google Docs endpoints for GL Connectors."""
     INTEGRATIONS: str
     USER_HAS_INTEGRATION: str
     SUCCESS_AUTHORIZE_CALLBACK: str
@@ -83,14 +86,14 @@ class GoogleDocsEndpoint(Enum):
     SUMMARIZE_COMMENTS: str
 
 class GoogleEndpoint(Enum):
-    """Google endpoints for BOSA Connector."""
+    """Google endpoints for GL Connectors."""
     INTEGRATIONS: str
     USER_HAS_INTEGRATION: str
     SUCCESS_AUTHORIZE_CALLBACK: str
     USERINFO: str
 
 class TwitterEndpoint(Enum):
-    """Twitter endpoints for BOSA Connector."""
+    """Twitter endpoints for GL Connectors."""
     INTEGRATIONS: str
     USER_HAS_INTEGRATION: str
     SUCCESS_AUTHORIZE_CALLBACK: str
@@ -100,7 +103,7 @@ class TwitterEndpoint(Enum):
     GET_USERS: str
 
 class GoogleMailEndpoint(Enum):
-    """Google Mail endpoints for BOSA Connector."""
+    """Google Mail endpoints for GL Connectors."""
     INTEGRATIONS: str
     USER_HAS_INTEGRATION: str
     SUCCESS_AUTHORIZE_CALLBACK: str

aip_agents/tools/gl_connector/__init__.py

@@ -1,4 +1,4 @@
-"""GL Connector tool wrapper exports."""
+"""GL Connectors tool wrapper exports."""
 
 from aip_agents.tools.gl_connector.tool import GLConnectorTool