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

aip_agents/tools/__init__.py

@@ -1,6 +1,47 @@
 """Initialize Tools Module for AIP Agents."""
 
+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.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"]
+
+
+def _register_optional(module_path: str, export_name: str) -> None:
+    """Try to import optional tool module and expose it in __all__.
+
+    Args:
+        module_path: The module path to import.
+        export_name: The name to export in __all__.
+    """
+    try:
+        module = import_module(module_path)
+        tool = getattr(module, export_name)
+        __all__.append(export_name)
+        globals()[export_name] = tool
+
+    except ImportError as e:
+        logger.debug(f"Module {module_path} not found: {e}")
+    except AttributeError as e:
+        logger.debug(f"Tool {export_name} not found in module {module_path}: {e}")
+    except Exception as e:
+        logger.debug(f"Unexpected error loading {export_name}: {e}")
+
+
+_register_optional("aip_agents.tools.browser_use", "BrowserUseTool")
+_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")
 
-__all__ = ["BOSA_AUTOMATED_TOOLS", "GLConnectorTool"]
+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

aip_agents/tools/__init__.pyi

@@ -1,4 +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.time_tool import TimeTool as TimeTool
+from aip_agents.tools.web_search import GoogleSerperTool as GoogleSerperTool
 
-__all__ = ['BOSA_AUTOMATED_TOOLS', 'GLConnectorTool']
+__all__ = ['BOSA_AUTOMATED_TOOLS', 'GLConnectorTool', 'GoogleSerperTool', 'TimeTool', 'BrowserUseTool', 'E2BCodeSandboxTool', 'DocxReaderTool', 'ExcelReaderTool', 'PDFReaderTool']

aip_agents/tools/time_tool.py

@@ -0,0 +1,117 @@
+"""Tool to get the current time.
+
+Authors:
+    Saul Sayers (saul.sayers@gdplabs.id)
+"""
+
+import re
+from datetime import datetime, timedelta, timezone, tzinfo
+
+from dateutil import tz
+from langchain_core.tools import BaseTool
+from pydantic import BaseModel, Field
+
+FORMAT_STRING = "%m/%d/%y %H:%M:%S"
+DEFAULT_TIMEZONE = "UTC+7"
+
+
+class TimeToolInput(BaseModel):
+    """Input schema for the TimeTool."""
+
+    datetime_format: str = Field(
+        default=FORMAT_STRING,
+        description="""
+        Optional datetime format string. Default: '%m/%d/%y %H:%M:%S'
+
+        Common format codes:
+        %Y: Year with century (2024)
+        %m: Month as number (01-12)
+        %d: Day of month (01-31)
+        %A: Full weekday name (Wednesday)
+        %a: Short weekday name (Wed)
+        %H: Hour (00-23)
+        %M: Minute (00-59)
+        %S: Second (00-59)
+        %Z: Timezone name
+        %j: Day of year (001-366)
+        %W: Week number (00-53, Monday first)
+        %U: Week number (00-53, Sunday first)
+        %c: Locale's date and time
+        %x: Locale's date
+        %X: Locale's time
+        """,
+    )
+    timezone: str | None = Field(
+        default=DEFAULT_TIMEZONE,
+        description="""
+        Optional timezone identifier. Supports IANA names (e.g., 'Asia/Jakarta')
+        or offsets (e.g., 'UTC+7', 'UTC-04:30'). Default: 'UTC+7' (Asia/Jakarta).
+        Highly recommended to be filled when getting the current time to ensure
+        the time is in the correct timezone.
+        """,
+    )
+
+
+def _parse_timezone(timezone_value: str | None) -> tzinfo | None:
+    """Return a tzinfo instance for a given timezone string.
+
+    Args:
+        timezone_value: Timezone string (e.g., "UTC+7", "UTC-5:30") or None.
+
+    Returns:
+        tzinfo instance or None if invalid/None input.
+    """
+    if timezone_value is None:
+        return None
+
+    tz_string = timezone_value.strip()
+    if not tz_string:
+        return None
+
+    offset_match = re.fullmatch(r"UTC([+-])(\d{1,2})(?::?(\d{2}))?$", tz_string.upper())
+    if offset_match:
+        sign, hours_str, minutes_str = offset_match.groups()
+        hours = int(hours_str)
+        minutes = int(minutes_str) if minutes_str else 0
+        MAX_HOURS_OFFSET = 12
+        MINUTES_OFFSETS = [0, 30, 45]
+        if hours > MAX_HOURS_OFFSET or minutes not in MINUTES_OFFSETS:
+            raise ValueError(f"Invalid timezone offset: {timezone_value}")
+        delta = timedelta(hours=hours, minutes=minutes)
+        if sign == "-":
+            delta = -delta
+        return timezone(delta)
+
+    tzinfo = tz.gettz(tz_string)
+    if tzinfo is not None:
+        return tzinfo
+
+    raise ValueError(f"Unknown timezone: {timezone_value}")
+
+
+class TimeTool(BaseTool):
+    """Tool to get the current time."""
+
+    name: str = "time_tool"
+    description: str = """
+    Useful for getting the current time in a specified format and timezone.
+    Default format: '%m/%d/%y %H:%M:%S' (e.g., '05/15/24 17:30:00')
+
+    Supports timezone specification to get the current time in any timezone.
+    It is highly recommended to specify a timezone to ensure accurate results.
+    """
+    args_schema: type[BaseModel] = TimeToolInput
+
+    def _run(self, datetime_format: str = FORMAT_STRING, timezone: str | None = DEFAULT_TIMEZONE) -> str:
+        """Get current time formatted according to the specified format and timezone.
+
+        Args:
+            datetime_format: Format string for datetime (default: ISO format).
+            timezone: Timezone string (default: UTC).
+
+        Returns:
+            Formatted datetime string.
+        """
+        tzinfo = _parse_timezone(timezone)
+        current_time = datetime.now(tz=tzinfo)
+        return current_time.strftime(datetime_format)

aip_agents/tools/time_tool.pyi

@@ -0,0 +1,16 @@
+from langchain_core.tools import BaseTool
+from pydantic import BaseModel
+
+FORMAT_STRING: str
+DEFAULT_TIMEZONE: str
+
+class TimeToolInput(BaseModel):
+    """Input schema for the TimeTool."""
+    datetime_format: str
+    timezone: str | None
+
+class TimeTool(BaseTool):
+    """Tool to get the current time."""
+    name: str
+    description: str
+    args_schema: type[BaseModel]

{aip_agents_binary-0.5.19.dist-info → aip_agents_binary-0.5.21.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aip-agents-binary
-Version: 0.5.19
+Version: 0.5.21
 Summary: A library for managing agents in Gen AI applications.
 Author-email: Raymond Christopher <raymond.christopher@gdplabs.id>
 Requires-Python: <3.13,>=3.11

{aip_agents_binary-0.5.19.dist-info → aip_agents_binary-0.5.21.dist-info}/RECORD

@@ -370,14 +370,16 @@ 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=1-kGfOGy-z1k_FHcCT9UOTDGGhM6ik2W72dkGYdcpPk,227
-aip_agents/tools/__init__.pyi,sha256=JvKGI44VA6i7hwKM9NuUqudYYvxX5u4n2D2U7w3RP5k,221
+aip_agents/tools/__init__.py,sha256=HoTjIH5o3xFwa5FYGugmwAhi7MXkBc0vf6IDG9qiOgU,1887
+aip_agents/tools/__init__.pyi,sha256=-yGiktAQqVJt-nspG79nQWEUCplAT4v-TadqQQb4rRc,792
 aip_agents/tools/bosa_tools.py,sha256=Zu5hhHx2LCBtvw9A30wYf1x5v4DQiB6MxePh12J77a4,3516
 aip_agents/tools/bosa_tools.pyi,sha256=ZidEMjX8TfdlSzLicwJQDbyoet9LeRqH8NSUSZ2tM7o,1165
 aip_agents/tools/constants.py,sha256=YZJFHKJvU9XkRKG_9hOpZ9ZreMoMoBXjF3SNsN51WhI,5439
 aip_agents/tools/constants.pyi,sha256=_M2Hm3OoCke51rxH4Pei5TPufD1stvpBqsQzmyFHNCk,3528
 aip_agents/tools/memory_search_tool.py,sha256=ClXW6BEjZNHb6NHe44KxbwZ6GSGo7v7AUzaV99LMouc,679
 aip_agents/tools/memory_search_tool.pyi,sha256=kjUaPWFrUIHrCTla_7ccnEWUNb44v0ZHLxyKGLPyeWg,456
+aip_agents/tools/time_tool.py,sha256=4m6PrLwLyyF-2UM7VqwhnyqPAWpbxub2SR1t6RIftkY,3849
+aip_agents/tools/time_tool.pyi,sha256=veJnkeyD_rco2Jah-m_LbtsOgziECPheCLFabKeuDH8,390
 aip_agents/tools/tool_config_injector.py,sha256=gEOzukaC3KzU00IVCKyhjJUKy6o59ple2PXiwiib5Rw,10859
 aip_agents/tools/tool_config_injector.pyi,sha256=9m6SDm3cq9fm3BvL2E1Q4_h_Vt9kY3VANB9edDnY8x4,1081
 aip_agents/tools/browser_use/__init__.py,sha256=ARD9Y7nfd3uEUUbNIly97-Bib5RKrniWOQA_9WptbUY,2200
@@ -538,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.19.dist-info/METADATA,sha256=evsObf3q80xbgQb8RkQNdYN3JWr6v5l9yGUbPU9molg,22888
-aip_agents_binary-0.5.19.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aip_agents_binary-0.5.19.dist-info/top_level.txt,sha256=PEz8vcwC1bH4UrkhF0LkIYCNfXGWZUHdSklbvkBe25E,11
-aip_agents_binary-0.5.19.dist-info/RECORD,,
+aip_agents_binary-0.5.21.dist-info/METADATA,sha256=L2JUdnr6SuTg8JVn4SIKQyl7e7hko0OPrrNcCe9DEQ4,22888
+aip_agents_binary-0.5.21.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aip_agents_binary-0.5.21.dist-info/top_level.txt,sha256=PEz8vcwC1bH4UrkhF0LkIYCNfXGWZUHdSklbvkBe25E,11
+aip_agents_binary-0.5.21.dist-info/RECORD,,