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

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)
@@ -6,6 +6,7 @@ Authors:
 
 import inspect
 import os
+from typing import Any
 
 from bosa_core.telemetry import (
     FastAPIConfig,
@@ -20,7 +21,7 @@ from bosa_core.telemetry.opentelemetry.instrument.functions import (
 from dotenv import load_dotenv
 from fastapi import FastAPI
 
-from aip_agents.agent import BaseAgent, GoogleADKAgent, LangChainAgent, LangGraphAgent
+from aip_agents.agent import BaseAgent, LangChainAgent, LangGraphAgent
 from aip_agents.utils.logger import get_logger
 
 load_dotenv()
@@ -35,12 +36,32 @@ VERSION_NUMBER = os.getenv("VERSION_NUMBER", "0.0.0")
 BUILD_NUMBER = os.getenv("BUILD_NUMBER", "0")
 USE_OPENTELEMETRY = os.getenv("USE_OPENTELEMETRY", "true").lower() == "true"
 
-CLASSES_TO_INSTRUMENT = [
-    BaseAgent,
-    LangGraphAgent,
-    LangChainAgent,
-    GoogleADKAgent,
-]
+# Lazy import of GoogleADKAgent to avoid heavy dependencies when not needed.
+# This is initialized lazily by _get_classes_to_instrument() and can be
+# patched by tests for mocking purposes.
+CLASSES_TO_INSTRUMENT: list[type[Any]] | None = None
+
+
+def _get_classes_to_instrument() -> list[type[Any]]:
+    """Get the list of classes to instrument.
+
+    This lazily imports GoogleADKAgent only when telemetry is being set up,
+    avoiding the heavy Google ADK dependencies during module import.
+
+    Returns:
+        List of agent classes to instrument.
+    """
+    global CLASSES_TO_INSTRUMENT
+    if CLASSES_TO_INSTRUMENT is None:
+        from aip_agents.agent import GoogleADKAgent
+
+        CLASSES_TO_INSTRUMENT = [
+            BaseAgent,
+            LangGraphAgent,
+            LangChainAgent,
+            GoogleADKAgent,
+        ]
+    return CLASSES_TO_INSTRUMENT
 
 
 def get_all_methods(cls: type) -> list:
@@ -61,12 +82,12 @@ 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 = []
-    for cls in CLASSES_TO_INSTRUMENT:
+    for cls in _get_classes_to_instrument():
         agent_methods.extend(get_all_methods(cls))
     BOSAFunctionsInstrumentor().instrument(methods=agent_methods)
 
@@ -148,4 +169,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

@@ -1,7 +1,8 @@
 from _typeshed import Incomplete
-from aip_agents.agent import BaseAgent as BaseAgent, GoogleADKAgent as GoogleADKAgent, LangChainAgent as LangChainAgent, LangGraphAgent as LangGraphAgent
+from aip_agents.agent import BaseAgent as BaseAgent, LangChainAgent as LangChainAgent, LangGraphAgent as LangGraphAgent
 from aip_agents.utils.logger import get_logger as get_logger
 from fastapi import FastAPI
+from typing import Any
 
 logger: Incomplete
 SENTRY_DSN: Incomplete
@@ -10,7 +11,7 @@ SENTRY_PROJECT: Incomplete
 VERSION_NUMBER: Incomplete
 BUILD_NUMBER: Incomplete
 USE_OPENTELEMETRY: Incomplete
-CLASSES_TO_INSTRUMENT: Incomplete
+CLASSES_TO_INSTRUMENT: list[type[Any]] | None
 
 def get_all_methods(cls) -> list:
     """Get all methods from a class.
@@ -21,8 +22,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,26 @@
 from importlib import import_module
 from typing import TYPE_CHECKING
 
-from aip_agents.tools.bosa_tools import BOSA_AUTOMATED_TOOLS
+from aip_agents.tools.date_range_tool import DateRangeTool
 from aip_agents.tools.gl_connector import GLConnectorTool
+from aip_agents.tools.gl_connector_tools import (
+    BOSA_AUTOMATED_TOOLS,
+    GL_CONNECTORS_AUTOMATED_TOOLS,
+)
 from aip_agents.tools.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",
+    "DateRangeTool",
+]
 
 
 def _register_optional(module_path: str, export_name: str) -> None:
@@ -40,8 +51,14 @@ _register_optional("aip_agents.tools.code_sandbox", "E2BCodeSandboxTool")
 _register_optional("aip_agents.tools.document_loader", "DocxReaderTool")
 _register_optional("aip_agents.tools.document_loader", "ExcelReaderTool")
 _register_optional("aip_agents.tools.document_loader", "PDFReaderTool")
+_register_optional("aip_agents.tools.execute_ptc_code", "create_execute_ptc_code_tool")
 
 if TYPE_CHECKING:
     from aip_agents.tools.browser_use import BrowserUseTool
     from aip_agents.tools.code_sandbox import E2BCodeSandboxTool
-    from aip_agents.tools.document_loader import DocxReaderTool, ExcelReaderTool, PDFReaderTool
+    from aip_agents.tools.document_loader import (
+        DocxReaderTool,
+        ExcelReaderTool,
+        PDFReaderTool,
+    )
+    from aip_agents.tools.execute_ptc_code import create_execute_ptc_code_tool

aip_agents/tools/__init__.pyi

@@ -1,9 +1,11 @@
-from aip_agents.tools.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.date_range_tool import DateRangeTool as DateRangeTool
 from aip_agents.tools.document_loader import DocxReaderTool as DocxReaderTool, ExcelReaderTool as ExcelReaderTool, PDFReaderTool as PDFReaderTool
+from aip_agents.tools.execute_ptc_code import create_execute_ptc_code_tool as create_execute_ptc_code_tool
 from aip_agents.tools.gl_connector import GLConnectorTool as GLConnectorTool
+from aip_agents.tools.gl_connector_tools import BOSA_AUTOMATED_TOOLS as BOSA_AUTOMATED_TOOLS, GL_CONNECTORS_AUTOMATED_TOOLS as GL_CONNECTORS_AUTOMATED_TOOLS
 from aip_agents.tools.time_tool import TimeTool as TimeTool
 from aip_agents.tools.web_search import GoogleSerperTool as GoogleSerperTool
 
-__all__ = ['BOSA_AUTOMATED_TOOLS', 'GLConnectorTool', 'GoogleSerperTool', 'TimeTool', 'BrowserUseTool', 'E2BCodeSandboxTool', 'DocxReaderTool', 'ExcelReaderTool', 'PDFReaderTool']
+__all__ = ['BOSA_AUTOMATED_TOOLS', 'GL_CONNECTORS_AUTOMATED_TOOLS', 'GLConnectorTool', 'GoogleSerperTool', 'TimeTool', 'DateRangeTool', 'BrowserUseTool', 'E2BCodeSandboxTool', 'DocxReaderTool', 'ExcelReaderTool', 'PDFReaderTool', 'create_execute_ptc_code_tool']

aip_agents/tools/browser_use/browser_use_tool.py

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

aip_agents/tools/browser_use/streaming.py

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

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