aip-agents-binary 0.5.20__py3-none-manylinux_2_31_x86_64.whl

aip_agents/tools/bosa_tools.py

@@ -0,0 +1,105 @@
+"""Auto-generated tools from BOSA connector.
+
+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 BOSA_API_BASE_URL, BOSA_API_KEY, BOSA_FETCH_MAX_RETRIES, ToolType
+from aip_agents.utils.logger import get_logger
+
+logger = get_logger(__name__)
+
+
+def get_bosa_modules_with_retry() -> list[str]:
+    """Try to get available modules with retries.
+
+    Returns:
+        List of available modules.
+    """
+    if not BOSA_API_BASE_URL or not BOSA_API_KEY:
+        logger.warning("BOSA credentials missing (base_url or api_key); returning empty modules list")
+        return []
+
+    connector = BosaConnector(api_base_url=BOSA_API_BASE_URL, api_key=BOSA_API_KEY)
+    modules = []
+    for attempt in range(BOSA_FETCH_MAX_RETRIES):
+        try:
+            modules = list(connector.get_available_modules())
+            if modules:
+                return modules
+            logger.warning(
+                f"Failed to get BOSA available modules, retrying... (attempt {attempt + 1} / {BOSA_FETCH_MAX_RETRIES})"
+            )
+        except Exception as e:
+            logger.exception(
+                f"Exception when getting BOSA available modules (attempt {attempt + 1} / {BOSA_FETCH_MAX_RETRIES}): {e}"
+            )
+    logger.error("Failed to get BOSA available modules after maximum retries")
+    return modules
+
+
+class LazyBosaToolsDict(dict):
+    """Lazy dictionary for BOSA tools."""
+
+    def __missing__(self, app):
+        """When a key is missing, create the tools and store them in the dictionary.
+
+        Args:
+            app: Name of the BOSA connector.
+
+        Returns:
+            List of tools generated by BOSAConnectorToolGenerator.
+        """
+        if app not in get_bosa_modules():
+            return []
+        tools = []
+        for attempt in range(BOSA_FETCH_MAX_RETRIES):
+            try:
+                tools = BOSAConnectorToolGenerator(
+                    api_base_url=BOSA_API_BASE_URL,
+                    api_key=BOSA_API_KEY,
+                    app_name=app,
+                ).generate_tools(tool_type=ToolType.LANGCHAIN)
+                if tools:
+                    self[app] = tools
+                    return tools
+                logger.warning(
+                    f"Failed to create BOSA tools, retrying... (attempt {attempt + 1} / {BOSA_FETCH_MAX_RETRIES})"
+                )
+            except Exception as e:
+                logger.exception(
+                    f"Exception when creating BOSA tools for app '{app}' "
+                    f"(attempt {attempt + 1} / {BOSA_FETCH_MAX_RETRIES}): {e}"
+                )
+        logger.error("Failed to create BOSA tools after maximum retries")
+        return tools
+
+
+# Supported modules (dynamic)
+def get_bosa_modules() -> list[str]:
+    """Lazily fetch and cache BOSA modules.
+
+    This is for backwards compatibility with the old BOSA modules.
+
+    Returns:
+        List of BOSA modules.
+    """
+    if not hasattr(get_bosa_modules, "_cache"):
+        get_bosa_modules._cache = get_bosa_modules_with_retry()
+    return get_bosa_modules._cache
+
+
+# FOR BACKWARDS COMPATIBILITY
+BOSA_MODULES = [
+    "github",
+    "twitter",
+    "google",
+    "google_drive",
+    "google_mail",
+    "google_docs",
+]
+
+BOSA_AUTOMATED_TOOLS: dict[str, list[BaseTool]] = LazyBosaToolsDict()

aip_agents/tools/bosa_tools.pyi

@@ -0,0 +1,37 @@
+from _typeshed import Incomplete
+from aip_agents.tools.constants import BOSA_API_BASE_URL as BOSA_API_BASE_URL, BOSA_API_KEY as BOSA_API_KEY, BOSA_FETCH_MAX_RETRIES as BOSA_FETCH_MAX_RETRIES, ToolType as ToolType
+from aip_agents.utils.logger import get_logger as get_logger
+from langchain_core.tools import BaseTool as BaseTool
+
+logger: Incomplete
+
+def get_bosa_modules_with_retry() -> list[str]:
+    """Try to get available modules with retries.
+
+    Returns:
+        List of available modules.
+    """
+
+class LazyBosaToolsDict(dict):
+    """Lazy dictionary for BOSA tools."""
+    def __missing__(self, app):
+        """When a key is missing, create the tools and store them in the dictionary.
+
+        Args:
+            app: Name of the BOSA connector.
+
+        Returns:
+            List of tools generated by BOSAConnectorToolGenerator.
+        """
+
+def get_bosa_modules() -> list[str]:
+    """Lazily fetch and cache BOSA modules.
+
+    This is for backwards compatibility with the old BOSA modules.
+
+    Returns:
+        List of BOSA modules.
+    """
+
+BOSA_MODULES: Incomplete
+BOSA_AUTOMATED_TOOLS: dict[str, list[BaseTool]]

aip_agents/tools/browser_use/__init__.py

@@ -0,0 +1,82 @@
+# flake8: noqa: F401
+"""Browser use tools package for AI Agent Platform.
+
+Authors:
+    Reinhart Linanda (reinhart.linanda@gdplabs.id)
+"""
+
+import warnings
+from enum import StrEnum
+
+try:
+    import json_repair
+    import minio
+    from browser_use import Agent
+    from steel import Steel
+
+    _TOOL_AVAILABLE = True
+
+except ImportError:
+    _TOOL_AVAILABLE = False
+    warnings.warn(
+        "Browser use tools not available. Install with: pip install aip-agents[local]",
+        ImportWarning,
+        stacklevel=2,
+    )
+
+
+class ImportableName(StrEnum):
+    """Names of the importable attributes."""
+
+    BROWSER_USE_TOOL = "BrowserUseTool"
+    BROWSER_USE_TOOL_INPUT = "BrowserUseToolInput"
+    BROWSER_USE_TOOL_CONFIG = "BrowserUseToolConfig"
+
+
+if _TOOL_AVAILABLE:
+    __all__ = [
+        ImportableName.BROWSER_USE_TOOL,
+        ImportableName.BROWSER_USE_TOOL_INPUT,
+        ImportableName.BROWSER_USE_TOOL_CONFIG,
+    ]
+
+    _LAZY_IMPORTS = {}
+else:
+    # No tools available
+    __all__ = []
+
+
+def __getattr__(name: str):
+    """Lazy import to avoid circular dependencies and import errors in tests.
+
+    Args:
+        name (str): The name of the attribute to get.
+
+    Returns:
+        The attribute value.
+    """
+    if not _TOOL_AVAILABLE:
+        raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+
+    if name in _LAZY_IMPORTS:
+        return _LAZY_IMPORTS[name]
+
+    if name == ImportableName.BROWSER_USE_TOOL:
+        from aip_agents.tools.browser_use.browser_use_tool import BrowserUseTool
+
+        _LAZY_IMPORTS[name] = BrowserUseTool
+        return BrowserUseTool
+
+    if name == ImportableName.BROWSER_USE_TOOL_INPUT:
+        from aip_agents.tools.browser_use.schemas import BrowserUseToolInput
+
+        _LAZY_IMPORTS[name] = BrowserUseToolInput
+        return BrowserUseToolInput
+
+    if name == ImportableName.BROWSER_USE_TOOL_CONFIG:
+        from aip_agents.tools.browser_use.schemas import BrowserUseToolConfig
+
+        _LAZY_IMPORTS[name] = BrowserUseToolConfig
+        return BrowserUseToolConfig
+
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")

aip_agents/tools/browser_use/__init__.pyi

@@ -0,0 +1,14 @@
+from enum import StrEnum
+
+__all__ = [<ImportableName.BROWSER_USE_TOOL: 'BrowserUseTool'>, <ImportableName.BROWSER_USE_TOOL_INPUT: 'BrowserUseToolInput'>, <ImportableName.BROWSER_USE_TOOL_CONFIG: 'BrowserUseToolConfig'>]
+
+class ImportableName(StrEnum):
+    """Names of the importable attributes."""
+    BROWSER_USE_TOOL: str
+    BROWSER_USE_TOOL_INPUT: str
+    BROWSER_USE_TOOL_CONFIG: str
+
+# Names in __all__ with no definition:
+#   BrowserUseTool
+#   BrowserUseToolConfig
+#   BrowserUseToolInput

aip_agents/tools/browser_use/action_parser.py

@@ -0,0 +1,103 @@
+"""Helper utilities for parsing browser-use agent actions.
+
+Authors:
+    Raymond Christopher (raymond.christopher@gdplabs.id)
+"""
+
+from __future__ import annotations
+
+from typing import Any
+
+from browser_use.agent.views import ActionResult, AgentOutput
+from browser_use.controller.registry.views import ActionModel
+
+from aip_agents.tools.browser_use.types import ToolCallInfo
+from aip_agents.utils.logger import get_logger
+
+logger = get_logger(__name__)
+
+
+class ActionParser:
+    """Dedicated class for parsing agent actions with improved error handling."""
+
+    @staticmethod
+    def extract_actions(
+        model_output: AgentOutput | None, last_result: list[ActionResult] | None = None
+    ) -> list[ToolCallInfo]:
+        """Extract action information from model output.
+
+        Args:
+            model_output: The model output containing action information.
+            last_result: The last result from the agent state for output extraction.
+
+        Returns:
+            list[ToolCallInfo]: Structured action information.
+        """
+        if model_output is None or getattr(model_output, "action", None) is None:
+            return []
+
+        actions = model_output.action
+        outputs = ActionParser._extract_outputs_from_result(last_result)
+
+        return [ActionParser._parse_single_action(action_model, outputs, i) for i, action_model in enumerate(actions)]
+
+    @staticmethod
+    def _extract_outputs_from_result(last_result: list[ActionResult] | None) -> list[str]:
+        """Extract output strings from agent result.
+
+        Args:
+            last_result: The last result from the agent state for output extraction.
+
+        Returns:
+            list[str]: List of output strings.
+        """
+        if not last_result:
+            return []
+
+        outputs: list[str] = []
+        for result in last_result:
+            if result.extracted_content:
+                outputs.append(result.extracted_content)
+            elif result.error:
+                outputs.append(f"Error: {result.error}")
+            else:
+                outputs.append("")
+
+        return outputs
+
+    @staticmethod
+    def _parse_single_action(action_model: ActionModel, outputs: list[str], index: int) -> ToolCallInfo:
+        """Parse a single action model into ToolCallInfo.
+
+        Args:
+            action_model: The action model to parse.
+            outputs: List of output strings.
+            index: Index of the action in the outputs list.
+
+        Returns:
+            ToolCallInfo: Parsed action information.
+        """
+        try:
+            specific_action = action_model.root
+            action_data: dict[str, Any] = specific_action.model_dump(exclude_unset=True)
+
+            tool_name = list(action_data.keys())[0]
+            tool_args = action_data[tool_name]
+
+            if not isinstance(tool_args, dict):
+                tool_args = {"value": tool_args}
+
+            output = outputs[index] if index < len(outputs) else ""
+
+            return ToolCallInfo(name=tool_name, args=tool_args, output=output)
+
+        except Exception as exc:
+            logger.warning("Error parsing action structure: %s", exc)
+            return ToolCallInfo(
+                name="parsing_error",
+                args={"error": str(exc)},
+                output=outputs[index] if index < len(outputs) else "",
+            )
+
+
+__all__ = ["ActionParser"]

aip_agents/tools/browser_use/action_parser.pyi

@@ -0,0 +1,18 @@
+from aip_agents.tools.browser_use.types import ToolCallInfo
+from browser_use.agent.views import ActionResult, AgentOutput
+
+__all__ = ['ActionParser']
+
+class ActionParser:
+    """Dedicated class for parsing agent actions with improved error handling."""
+    @staticmethod
+    def extract_actions(model_output: AgentOutput | None, last_result: list[ActionResult] | None = None) -> list[ToolCallInfo]:
+        """Extract action information from model output.
+
+        Args:
+            model_output: The model output containing action information.
+            last_result: The last result from the agent state for output extraction.
+
+        Returns:
+            list[ToolCallInfo]: Structured action information.
+        """