aip-agents-binary 0.6.1__py3-none-macosx_13_0_arm64.whl → 0.6.3__py3-none-macosx_13_0_arm64.whl

aip_agents/examples/hello_world_ptc.py

@@ -7,8 +7,6 @@ Required environment variables:
 
 import asyncio
 
-from langchain_openai import ChatOpenAI
-
 from aip_agents.agent import LangGraphReactAgent
 from aip_agents.ptc import PromptConfig, PTCSandboxConfig
 
@@ -24,7 +22,7 @@ async def main() -> None:
     agent = LangGraphReactAgent(
         name="ptc_hello_world",
         instruction=instruction,
-        model=ChatOpenAI(model="gpt-5.2"),
+        model="openai/gpt-5.2",
         ptc_config=PTCSandboxConfig(enabled=True, sandbox_timeout=180.0, prompt=PromptConfig(mode="index")),
     )
     agent.add_mcp_server(

aip_agents/tools/__init__.py

@@ -3,8 +3,12 @@
 from importlib import import_module
 from typing import TYPE_CHECKING
 
+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.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
@@ -17,6 +21,7 @@ __all__ = [
     "GLConnectorTool",
     "GoogleSerperTool",
     "TimeTool",
+    "DateRangeTool",
 ]
 
 
@@ -51,5 +56,9 @@ _register_optional("aip_agents.tools.execute_ptc_code", "create_execute_ptc_code
 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,5 +1,6 @@
 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
@@ -7,4 +8,4 @@ from aip_agents.tools.gl_connector_tools import BOSA_AUTOMATED_TOOLS as BOSA_AUT
 from aip_agents.tools.time_tool import TimeTool as TimeTool
 from aip_agents.tools.web_search import GoogleSerperTool as GoogleSerperTool
 
-__all__ = ['BOSA_AUTOMATED_TOOLS', 'GL_CONNECTORS_AUTOMATED_TOOLS', 'GLConnectorTool', 'GoogleSerperTool', 'TimeTool', 'BrowserUseTool', 'E2BCodeSandboxTool', 'DocxReaderTool', 'ExcelReaderTool', 'PDFReaderTool', 'create_execute_ptc_code_tool']
+__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/date_range_tool.py

@@ -0,0 +1,554 @@
+"""Tool to get date ranges for common time periods.
+
+Authors:
+    Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
+    Fachriza Dian Adhiatma (fachriza.d.adhiatma@gdplabs.id)
+"""
+
+import json
+import re
+from collections.abc import Callable
+from datetime import datetime, timedelta
+
+from dateutil.relativedelta import relativedelta
+from langchain_core.tools import BaseTool
+from pydantic import BaseModel, Field
+
+FORMAT_STRING = "%m/%d/%y %H:%M:%S"
+
+MIN_DAYS_FOR_WEEK_SPLIT = 6  # Minimum days difference for week split (7 days inclusive = 6 days difference)
+MONTH_NAME_ABBR_THRESHOLD = 3  # len > 3 implies full month name, else try abbreviated form
+SUPPORTED_DATE_RANGES = [
+    "last_week",
+    "this_week",
+    "last_month",
+    "this_month",
+    "yesterday",
+    "today",
+    "last_7_days",
+    "last_30_days",
+    "this_quarter",
+    "last_quarter",
+    "this_year",
+    "last_year",
+    "[N]_days_ago",
+    "[N]_weeks_ago",
+    "[N]_months_ago",
+    "YYYY-MM-DD to YYYY-MM-DD",
+]
+
+
+class DateRangeToolInput(BaseModel):
+    """Input schema for the DateRangeTool."""
+
+    date_range: str = Field(
+        ...,
+        description="""
+        Date range identifier. Supported values:
+        - Relative periods: 'last_week', 'this_week', 'last_month', 'this_month'
+        - Relative days: 'yesterday', 'today', 'last_7_days', 'last_30_days'
+        - Custom range: 'N_days_ago', 'N_weeks_ago', 'N_months_ago' (replace N with number)
+        - Quarter: 'this_quarter', 'last_quarter'
+        - Year: 'this_year', 'last_year'
+        - Natural forms: 'January 2025', 'January-March 2025', 'Q1 2025'
+        - Explicit date range: 'YYYY-MM-DD to YYYY-MM-DD' (e.g., '2025-08-01 to 2025-08-07')
+
+        Note: For queries like 'first week of August 2025', convert to explicit format: '2025-08-01 to 2025-08-07'
+        """,
+    )
+    format: str = Field(
+        default=FORMAT_STRING,
+        description="Optional datetime format string. Default: '%m/%d/%y'",
+    )
+    split_weeks: bool = Field(
+        default=False,
+        description=(
+            "If True and the resulting period spans more than one week, include a 'weeks' array of Sunday–Saturday"
+            " splits that fit within the requested period (not recent fixed weeks)."
+        ),
+    )
+
+
+class DateRangeTool(BaseTool):
+    """Tool to get date ranges for common time periods."""
+
+    name: str = "date_range_tool"
+    description: str = """
+    Useful for getting date ranges for various time periods.
+    Supports relative dates, custom ranges, and standard periods.
+    Returns start and end dates for the specified period.
+    """
+    args_schema: type[BaseModel] = DateRangeToolInput
+
+    def _parse_custom_range(self, date_range: str) -> tuple[datetime, datetime]:
+        """Parse custom range patterns like '3_days_ago', '2_weeks_ago', etc.
+
+        Args:
+            date_range (str): A string representing the custom date range in the format '{number}_{unit}_ago',
+                              where {number} is an integer and {unit} is one of 'days', 'weeks', or 'months'.
+
+        Returns:
+            tuple[datetime, datetime]: A tuple containing the start date and end date corresponding to the custom range.
+        """
+        pattern = r"(\d+)_(days|weeks|months)_ago"
+        match = re.match(pattern, date_range)
+        start_date = None
+        end_date = None
+        if not match:
+            raise ValueError(f"Invalid custom range format: {date_range}")
+
+        number, unit = int(match.group(1)), match.group(2)
+        today = datetime.now()
+
+        if unit == "days":
+            start_date = today - timedelta(days=number)
+            end_date = today
+        elif unit == "weeks":
+            start_date = today - timedelta(weeks=number)
+            end_date = today
+        elif unit == "months":
+            start_date = today - relativedelta(months=number)
+            end_date = today
+
+        return start_date, end_date
+
+    def _parse_month(self, month_str: str) -> int:
+        """Parse month name (full or abbreviated) to month number (1-12).
+
+        Args:
+            month_str (int): The month name string to parse.
+
+        Returns:
+            int: The month number (1-12).
+        """
+        fmt = "%B" if len(month_str) > MONTH_NAME_ABBR_THRESHOLD else "%b"
+        try:
+            return datetime.strptime(month_str, fmt).month
+        except ValueError:
+            # Fallback to abbreviated format
+            return datetime.strptime(month_str, "%b").month
+
+    def _parse_explicit_range(self, dr: str) -> tuple[datetime, datetime] | None:
+        """Parse explicit date range: YYYY-MM-DD to YYYY-MM-DD.
+
+        Args:
+            dr (str): The date range string to parse.
+
+        Returns:
+            tuple[datetime, datetime] | None: A tuple of (start_date, end_date) if matched, otherwise None.
+        """
+        m_explicit = re.match(r"^(\d{4}-\d{2}-\d{2})\s+to\s+(\d{4}-\d{2}-\d{2})$", dr, flags=re.IGNORECASE)
+        if m_explicit:
+            try:
+                start = datetime.strptime(m_explicit.group(1), "%Y-%m-%d")
+                end = datetime.strptime(m_explicit.group(2), "%Y-%m-%d")
+                if start > end:
+                    raise ValueError("Start date must be before or equal to end date")
+                return start, end
+            except ValueError:
+                return None
+        return None
+
+    def _parse_quarter_with_year(self, dr: str) -> tuple[datetime, datetime] | None:
+        """Parse quarter with year: Q1 2025.
+
+        Args:
+            dr (str): The date range string to parse.
+
+        Returns:
+            tuple[datetime, datetime] | None: A tuple of (start_date, end_date) if matched, otherwise None.
+        """
+        m_q = re.match(r"^Q([1-4])\s+(\d{4})$", dr, flags=re.IGNORECASE)
+        if m_q:
+            q = int(m_q.group(1))
+            year = int(m_q.group(2))
+            start_month = (q - 1) * 3 + 1
+            start = datetime(year, start_month, 1)
+            end = start + relativedelta(months=3, days=-1)
+            return start, end
+        return None
+
+    def _parse_month_range_with_year(self, dr: str) -> tuple[datetime, datetime] | None:
+        """Parse month range within a year: January-March 2025.
+
+        Args:
+            dr (str): The date range string to parse.
+
+        Returns:
+            tuple[datetime, datetime] | None: A tuple of (start_date, end_date) if matched, otherwise None.
+        """
+        m_range = re.match(r"^([A-Za-z]{3,9})\s*-\s*([A-Za-z]{3,9})\s+(\d{4})$", dr)
+        if m_range:
+            m1, m2, year_s = m_range.groups()
+            year = int(year_s)
+            try:
+                start_month = self._parse_month(m1)
+                end_month = self._parse_month(m2)
+                start = datetime(year, start_month, 1)
+                end = datetime(year, end_month, 1) + relativedelta(months=1, days=-1)
+                return start, end
+            except ValueError:
+                return None
+        return None
+
+    def _parse_single_month_with_year(self, dr: str) -> tuple[datetime, datetime] | None:
+        """Parse single month with year: January 2025.
+
+        Args:
+            dr (str): The date range string to parse.
+
+        Returns:
+            tuple[datetime, datetime] | None: A tuple of (start_date, end_date) if matched, otherwise None.
+        """
+        m_single = re.match(r"^([A-Za-z]{3,9})\s+(\d{4})$", dr)
+        if m_single:
+            mname, year_s = m_single.groups()
+            year = int(year_s)
+            try:
+                month = self._parse_month(mname)
+                start = datetime(year, month, 1)
+                end = start + relativedelta(months=1, days=-1)
+                return start, end
+            except ValueError:
+                return None
+        return None
+
+    def _parse_month_or_range(self, date_range: str) -> tuple[datetime, datetime] | None:
+        """Parse inputs like 'January 2025', 'Jan 2025', 'January-March 2025', 'Jan-Mar 2025', 'Q1 2025'.
+
+        Also parses explicit date ranges like '2025-08-01 to 2025-08-07'.
+
+        Args:
+            date_range (str): The date range string to parse.
+                Can be natural month/quarter forms or explicit YYYY-MM-DD format.
+
+        Returns:
+            tuple[datetime, datetime] | None: A tuple of (start_date, end_date) if matched, otherwise None.
+
+        Raises:
+            ValueError: If start date is after end date (caught internally and returns None).
+
+        Examples:
+            >>> self._parse_month_or_range("January 2025")
+            (datetime(2025, 1, 1), datetime(2025, 1, 31))
+            >>> self._parse_month_or_range("2025-08-01 to 2025-08-07")
+            (datetime(2025, 8, 1), datetime(2025, 8, 7))
+        """
+        dr = date_range.strip()
+
+        # Try each parser in order
+        parsers = [
+            self._parse_explicit_range,
+            self._parse_quarter_with_year,
+            self._parse_month_range_with_year,
+            self._parse_single_month_with_year,
+        ]
+
+        for parser in parsers:
+            result = parser(dr)
+            if result:
+                return result
+
+        return None
+
+    def _get_quarter_dates(self, today: datetime, last_quarter: bool = False) -> tuple[datetime, datetime]:
+        """Calculate the start and end dates of the current or last quarter based on the given date.
+
+        Args:
+            today (datetime): The reference date to determine the quarter.
+            last_quarter (bool, optional): If True, calculate the dates for the last quarter.
+                                           If False, calculate the dates for the current quarter.
+                                           Defaults to False.
+
+        Returns:
+            tuple[datetime, datetime]: A tuple containing the start and end dates of the quarter.
+        """
+        current_quarter = (today.month - 1) // 3
+        if last_quarter:
+            if current_quarter == 0:
+                start_date = datetime(today.year - 1, 10, 1)
+                end_date = datetime(today.year - 1, 12, 31)
+            else:
+                start_date = datetime(today.year, 3 * (current_quarter - 1) + 1, 1)
+                end_date = datetime(today.year, 3 * current_quarter, 1) + relativedelta(months=1, days=-1)
+        else:
+            start_date = datetime(today.year, 3 * current_quarter + 1, 1)
+            end_date = datetime(today.year, 3 * (current_quarter + 1), 1) + relativedelta(months=1, days=-1)
+        return start_date, end_date
+
+    def _days_since_most_recent_sunday(self, date: datetime) -> int:
+        """Return days since most recent Sunday (0 if today is Sunday).
+
+        Args:
+            date (datetime): The reference date.
+
+        Returns:
+            int: Days since most recent Sunday (0-6).
+        """
+        # weekday(): Mon=0, Tue=1, ..., Sun=6
+        # We want Sun=0, Mon=1, ..., Sat=6
+        return (date.weekday() + 1) % 7
+
+    def _get_standard_period_dates(self, date_range: str, today: datetime) -> tuple[datetime, datetime]:
+        """Calculate the start and end dates for a given standard period relative to a specified date.
+
+        Args:
+            date_range (str): The standard period to calculate dates for.
+                              Supported values are "last_week", "this_week", "last_month", and "this_month".
+            today (datetime): The reference date to calculate the period from.
+
+        Returns:
+            tuple[datetime, datetime]: A tuple containing the start and end dates of the specified period.
+        """
+        if date_range == "last_week":
+            # Calculate the previous Saturday (end of last week)
+            days_since_sunday = self._days_since_most_recent_sunday(today)
+            last_saturday = today - timedelta(days=days_since_sunday + 1)
+
+            # Calculate the previous Sunday (start of last week)
+            last_sunday = last_saturday - timedelta(days=6)
+
+            return last_sunday, last_saturday
+
+        if date_range == "this_week":
+            # Calculate the most recent Sunday (start of this week)
+            days_since_sunday = self._days_since_most_recent_sunday(today)
+            sunday = today - timedelta(days=days_since_sunday)
+
+            # Calculate the upcoming Saturday (end of this week)
+            saturday = sunday + timedelta(days=6)
+
+            return sunday, saturday
+
+        if date_range == "last_month":
+            first_day = today.replace(day=1)
+            last_month = first_day - timedelta(days=1)
+            return last_month.replace(day=1), last_month
+
+        if date_range == "this_month":
+            start = today.replace(day=1)
+            return start, start + relativedelta(months=1, days=-1)
+
+        raise ValueError(f"Unknown standard period: {date_range}")
+
+    def _week_bounds_for(self, date: datetime) -> tuple[datetime, datetime]:
+        """Return the Sunday-to-Saturday week containing the given date.
+
+        - Sunday is the start of week.
+        - Saturday is the end of week.
+
+        Args:
+            date (datetime): The date to find the week bounds for.
+
+        Returns:
+            tuple[datetime, datetime]: A tuple containing the start and end dates of the week.
+        """
+        days_since_sunday = self._days_since_most_recent_sunday(date)
+        sunday = datetime(date.year, date.month, date.day) - timedelta(days=days_since_sunday)
+        saturday = sunday + timedelta(days=6)
+        return sunday, saturday
+
+    def _build_week_splits(self, start_date: datetime, end_date: datetime, fmt: str) -> list[dict[str, str]]:
+        """Build Sunday–Saturday week splits that intersect [start_date, end_date].
+
+        - Includes partial edge weeks if any day of the Sun–Sat week falls within the requested range.
+        - Week entries still use full Sunday–Saturday bounds for consistency.
+        - Labels weeks consecutively as week_1, week_2, ...
+
+        Args:
+            start_date (datetime): Start of requested period.
+            end_date (datetime): End of requested period.
+            fmt (str): Format string for dates.
+
+        Returns:
+            list[dict]: Weekly split entries with start_date, end_date, and human description.
+        """
+        weeks: list[dict] = []
+        # Anchor on the Sunday of the week that contains start_date
+        first_week_sunday, _ = self._week_bounds_for(start_date)
+        cursor = first_week_sunday
+        idx = 1
+        while cursor <= end_date:
+            full_week_start = cursor
+            full_week_end = cursor + timedelta(days=6)
+            # Include any week that intersects the requested range
+            if not (full_week_end < start_date or full_week_start > end_date):
+                desc = f"From {full_week_start.strftime('%B %d, %Y')} to {full_week_end.strftime('%B %d, %Y')}"
+                weeks.append(
+                    {
+                        "period": f"week_{idx}",
+                        "start_date": full_week_start.strftime(fmt),
+                        "end_date": full_week_end.strftime(fmt),
+                        "description": desc,
+                    }
+                )
+                idx += 1
+            cursor += timedelta(days=7)
+        return weeks
+
+    def _get_relative_day_dates(self, date_range: str, today: datetime) -> tuple[datetime, datetime]:
+        """Calculate the start and end dates for a given relative date range.
+
+        Args:
+            date_range (str): The relative date range. Supported values are "yesterday", "today",
+                              "last_7_days", and "last_30_days".
+            today (datetime): The reference date from which the relative date range is calculated.
+
+        Returns:
+            tuple[datetime, datetime]: A tuple containing the start and end dates for the specified
+                                       relative date range.
+        """
+        if date_range == "yesterday":
+            yesterday = today - timedelta(days=1)
+            return yesterday, yesterday
+
+        if date_range == "today":
+            return today, today
+
+        if date_range == "last_7_days":
+            return today - timedelta(days=7), today
+
+        if date_range == "last_30_days":
+            return today - timedelta(days=30), today
+
+        raise ValueError(f"Unknown relative day range: {date_range}")
+
+    def _get_year_dates(self, date_range: str, today: datetime) -> tuple[datetime, datetime]:
+        """Returns the start and end dates for the specified year range.
+
+        Args:
+            date_range (str): The year range to calculate dates for.
+                              Accepted values are "this_year" and "last_year".
+            today (datetime): The current date to base the year calculation on.
+
+        Returns:
+            Tuple[datetime, datetime]: A tuple containing the start and end dates for the specified year range.
+        """
+        if date_range == "this_year":
+            return datetime(today.year, 1, 1), datetime(today.year, 12, 31)
+
+        if date_range == "last_year":
+            return datetime(today.year - 1, 1, 1), datetime(today.year - 1, 12, 31)
+
+        raise ValueError(f"Unknown year range: {date_range}")
+
+    def _format_response(
+        self,
+        start_date: datetime,
+        end_date: datetime,
+        date_range: str,
+        format: str,
+        weeks: list[dict] | None = None,
+    ) -> str:
+        """Format the response as a JSON string with the given date range information.
+
+        Args:
+            start_date (datetime): The start date of the range.
+            end_date (datetime): The end date of the range.
+            date_range (str): A string representation of the date range.
+            format (str): The format string to use for formatting the dates.
+            weeks (list[dict] | None): Optional weekly splits (Sunday–Saturday) for the period. This
+                is included when `split_weeks` is True and the overall period spans more than one week.
+
+        Returns:
+            str: A JSON string containing the formatted start date, end date, period, and description.
+        """
+        if weeks:
+            payload: dict = {
+                "weeks": weeks,
+                "period": date_range,
+                "description": f"Split into {len(weeks)} weeks with Sunday-to-Saturday bounds",
+            }
+            return json.dumps(payload)
+        payload: dict = {
+            "start_date": start_date.strftime(format),
+            "end_date": end_date.strftime(format),
+            "period": date_range,
+            "description": f"From {start_date.strftime('%B %d, %Y')} to {end_date.strftime('%B %d, %Y')}",
+        }
+        return json.dumps(payload)
+
+    def _get_result_and_format(
+        self,
+        date_range: str,
+        format: str,
+        split_weeks: bool,
+        result: tuple[datetime, datetime],
+    ) -> str:
+        """Helper to format response with optional week splits.
+
+        Args:
+            date_range (str): The date range string to process.
+            format (str): The format string to use for the output.
+            split_weeks (bool): If True and the resulting period spans more than one week,
+                also include a 'weeks' array with Sunday-to-Saturday splits for the recent weeks.
+            result (tuple[datetime, datetime]): A tuple containing the start and end dates of the date range.
+
+        Returns:
+            str: A formatted date range string or an error message if the date range is unsupported or an error occurs.
+        """
+        start_date, end_date = result
+        weeks = None
+        # Only build week splits when requested and range spans more than one week
+        if split_weeks and (end_date - start_date).days >= MIN_DAYS_FOR_WEEK_SPLIT:
+            weeks = self._build_week_splits(start_date, end_date, format)
+        return self._format_response(start_date, end_date, date_range, format, weeks=weeks)
+
+    def _run(self, date_range: str, format: str = FORMAT_STRING, split_weeks: bool = False) -> str:
+        """Process a given date range string and return a formatted date range string.
+
+        Args:
+            date_range (str): The date range string to process. Supported formats include:
+                - "standard_periods"
+                - "relative_days"
+                - "quarters"
+                - "years"
+                - "custom" (e.g., "10_days_ago", "2_weeks_ago", "3_months_ago")
+            format (str, optional): The format string to use for the output. Defaults to FORMAT_STRING.
+            split_weeks (bool, optional): If True and the resulting period spans more than one week,
+                also include a 'weeks' array with Sunday-to-Saturday splits for the recent weeks.
+
+        Returns:
+            str: A formatted date range string or an error message if the date range is unsupported or an error occurs.
+        """
+        today = datetime.now()
+
+        try:
+            handlers: dict[str, Callable] = {
+                "standard_periods": lambda: self._get_standard_period_dates(date_range, today),
+                "relative_days": lambda: self._get_relative_day_dates(date_range, today),
+                # Only handle special tokens 'this_quarter' and 'last_quarter' here. Natural forms like 'Q1 2025'
+                # are parsed later by _parse_month_or_range(). For non-matching inputs, return None so we keep trying.
+                "quarters": lambda: (
+                    self._get_quarter_dates(today, last_quarter=(date_range == "last_quarter"))
+                    if date_range in {"this_quarter", "last_quarter"}
+                    else None
+                ),
+                "years": lambda: self._get_year_dates(date_range, today),
+                "custom": lambda: (
+                    self._parse_custom_range(date_range)
+                    if re.match(r"\d+_(days|weeks|months)_ago", date_range)
+                    else None
+                ),
+            }
+
+            for handler in handlers.values():
+                try:
+                    result = handler()
+                    if result:
+                        return self._get_result_and_format(date_range, format, split_weeks, result)
+                except ValueError:
+                    continue
+
+            # Try parsing natural month/quarter expressions
+            mr = self._parse_month_or_range(date_range)
+            if mr:
+                return self._get_result_and_format(date_range, format, split_weeks, mr)
+
+            return (
+                f"Unsupported date range: {date_range}, supported date ranges are: {SUPPORTED_DATE_RANGES} "
+                f"or natural month/quarter forms like 'January 2025', 'January-March 2025', 'Q1 2025'."
+            )
+
+        except Exception as e:
+            return f"Error processing date range: {str(e)}"

aip_agents/tools/date_range_tool.pyi

@@ -0,0 +1,21 @@
+from _typeshed import Incomplete
+from collections.abc import Callable as Callable
+from langchain_core.tools import BaseTool
+from pydantic import BaseModel
+
+FORMAT_STRING: str
+MIN_DAYS_FOR_WEEK_SPLIT: int
+MONTH_NAME_ABBR_THRESHOLD: int
+SUPPORTED_DATE_RANGES: Incomplete
+
+class DateRangeToolInput(BaseModel):
+    """Input schema for the DateRangeTool."""
+    date_range: str
+    format: str
+    split_weeks: bool
+
+class DateRangeTool(BaseTool):
+    """Tool to get date ranges for common time periods."""
+    name: str
+    description: str
+    args_schema: type[BaseModel]

aip_agents/tools/execute_ptc_code.py

@@ -18,6 +18,7 @@ from langchain_core.callbacks import (
     CallbackManagerForToolRun,
 )
 from langchain_core.tools import BaseTool
+from pydantic import BaseModel, Field
 
 from aip_agents.ptc.naming import sanitize_function_name
 from aip_agents.tools.tool_config_injector import TOOL_CONFIGS_KEY
@@ -31,6 +32,21 @@ if TYPE_CHECKING:
 logger = get_logger(__name__)
 
 
+class PTCCodeInput(BaseModel):
+    """Input schema for PTCCodeTool."""
+
+    code: str = Field(
+        ...,
+        description=(
+            "Python code to execute. Import MCP tools from the generated `tools` package, "
+            "for example: `from tools.yfinance import get_stock_history`. "
+            "The code runs in a sandboxed environment with access to all configured MCP tools. "
+            "Use print() to output results. The tool returns JSON with keys: "
+            "ok, stdout, stderr, exit_code."
+        ),
+    )
+
+
 def _merge_config_layer(
     merged: dict[str, dict[str, Any]],
     source: dict[str, Any],
@@ -123,6 +139,9 @@ class PTCCodeTool(BaseTool):
         "This tool is useful for chaining multiple MCP tool calls with local data processing."
     )
 
+    # Input schema for LangChain tool invocation
+    args_schema: type[BaseModel] = PTCCodeInput
+
     # Internal attributes (not exposed to LLM)
     _ptc_executor: "PTCSandboxExecutor" = None  # type: ignore[assignment]
     _ptc_runtime: "E2BSandboxRuntime" = None  # type: ignore[assignment]
@@ -149,22 +168,6 @@ class PTCCodeTool(BaseTool):
         object.__setattr__(self, "_ptc_runtime", runtime)
         object.__setattr__(self, "_agent_tool_configs", agent_tool_configs)
 
-    @property
-    def args(self) -> dict[str, Any]:
-        """Return the argument schema for the tool."""
-        return {
-            "code": {
-                "type": "string",
-                "description": (
-                    "Python code to execute. Import MCP tools from the generated `tools` package, "
-                    "for example: `from tools.yfinance import get_stock_history`. "
-                    "The code runs in a sandboxed environment with access to all configured MCP tools. "
-                    "Use print() to output results. The tool returns JSON with keys: "
-                    "ok, stdout, stderr, exit_code."
-                ),
-            }
-        }
-
     def _run(
         self,
         code: str,

aip_agents/tools/execute_ptc_code.pyi

@@ -6,10 +6,15 @@ from aip_agents.sandbox.e2b_runtime import E2BSandboxRuntime as E2BSandboxRuntim
 from aip_agents.tools.tool_config_injector import TOOL_CONFIGS_KEY as TOOL_CONFIGS_KEY
 from aip_agents.utils.logger import get_logger as get_logger
 from langchain_core.tools import BaseTool
+from pydantic import BaseModel
 from typing import Any
 
 logger: Incomplete
 
+class PTCCodeInput(BaseModel):
+    """Input schema for PTCCodeTool."""
+    code: str
+
 def merge_tool_configs(agent_configs: dict[str, Any] | None, runtime_configs: dict[str, Any] | None) -> dict[str, dict[str, Any]]:
     '''Merge agent-level and runtime tool configs with sanitized keys.
 
@@ -42,6 +47,7 @@ class PTCCodeTool(BaseTool):
     """
     name: str
     description: str
+    args_schema: type[BaseModel]
     def __init__(self, executor: PTCSandboxExecutor, runtime: E2BSandboxRuntime, agent_tool_configs: dict[str, Any] | None = None, **kwargs: Any) -> None:
         """Initialize the PTC code tool.
 
@@ -51,9 +57,6 @@ class PTCCodeTool(BaseTool):
             agent_tool_configs: Optional agent-level tool configs.
             **kwargs: Additional keyword arguments passed to BaseTool.
         """
-    @property
-    def args(self) -> dict[str, Any]:
-        """Return the argument schema for the tool."""
     async def cleanup(self) -> None:
         """Clean up the sandbox runtime."""
 

{aip_agents_binary-0.6.1.dist-info → aip_agents_binary-0.6.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aip-agents-binary
-Version: 0.6.1
+Version: 0.6.3
 Summary: A library for managing agents in Gen AI applications.
 Author-email: Raymond Christopher <raymond.christopher@gdplabs.id>
 Requires-Python: <3.13,>=3.11
@@ -24,6 +24,7 @@ Requires-Dist: langchain-experimental<0.4.0,>=0.3.4
 Requires-Dist: langgraph<0.7.0,>=0.6.0
 Requires-Dist: minio<8.0.0,>=7.2.20
 Requires-Dist: pydantic<3.0.0,>=2.11.7
+Requires-Dist: python-dateutil<3.0.0,>=2.9.0
 Requires-Dist: python-dotenv<2.0.0,>=1.1.0
 Requires-Dist: requests<3.0.0,>=2.32.4
 Requires-Dist: uvicorn<0.35.0,>=0.34.3

{aip_agents_binary-0.6.1.dist-info → aip_agents_binary-0.6.3.dist-info}/RECORD

@@ -209,7 +209,7 @@ aip_agents/examples/hello_world_multi_agent_langgraph_lm_invoker.py,sha256=sppLD
 aip_agents/examples/hello_world_multi_agent_langgraph_lm_invoker.pyi,sha256=zFNns9i-sPY8aBy4HaVS6tXDWTsU7f-ApLD8gDtb1uk,252
 aip_agents/examples/hello_world_pii_logger.py,sha256=pm9LK-doZwQL-VdJYFHziuvwH562c8wwAwmKZeAWQS0,584
 aip_agents/examples/hello_world_pii_logger.pyi,sha256=5gOVrvhQV2DxIwr7ocL1-oTUXsO8XSyl5WN70c7rl_w,134
-aip_agents/examples/hello_world_ptc.py,sha256=EkSHUfRiGWLhUBcwpjWTRzAILqYzEPpys0pXmndWjk0,1462
+aip_agents/examples/hello_world_ptc.py,sha256=A3aadYO3H41UgKZoHqrThXz2_hOFGX8-E4XwFnRi7P4,1410
 aip_agents/examples/hello_world_ptc.pyi,sha256=Dbo4myHbndrL6PyA_mgf3HTIYqLudTaF4h_nsdqOyKs,231
 aip_agents/examples/hello_world_sentry.py,sha256=Gccr6WCevSnMFZYT5M6uJ_22EvLmAMUE8Et_H510_kM,3976
 aip_agents/examples/hello_world_sentry.pyi,sha256=k0wFj46QqXEjBU0CEERmiEf2bovVFkupzuzQzeD6h00,671
@@ -434,12 +434,14 @@ aip_agents/storage/providers/memory.py,sha256=81E9yT_oRkH7tMVr5azZv1oXIaeTzTAtTu
 aip_agents/storage/providers/memory.pyi,sha256=r0meeCsjKyE9FHJqCysfQ5ZH_FWRRv2SWg-NOHCUD-g,2109
 aip_agents/storage/providers/object_storage.py,sha256=Q268Sn8Y09gS-ilP6fe4CR0YIOvdbunh3V6SmuQsAVs,6413
 aip_agents/storage/providers/object_storage.pyi,sha256=nBIKJjUAWkIKhYhBFmFcypjoOieIYOghy55f62x7AX8,2878
-aip_agents/tools/__init__.py,sha256=Kvd185p8zZ2xpNqimxURJUA6joPXP-3MIrr8K3Lf9Rw,2102
-aip_agents/tools/__init__.pyi,sha256=K2j6MjKo5aaMAbWkknLwyRbSu14LxJyRMoi88DbHROk,1027
+aip_agents/tools/__init__.py,sha256=ullhWvRVtKYAEO5XhXn1sc1dT1LYDViSc23uqywUwyg,2228
+aip_agents/tools/__init__.pyi,sha256=KmdVTX-WXRChSqCyGrAUoV8TwpUBXlo9ITgx4xpqYx8,1120
 aip_agents/tools/constants.py,sha256=AabnuPQG_mc2sVdr9jV23_6bFempAsxQv3kdCR_ztLA,5723
 aip_agents/tools/constants.pyi,sha256=kFY8dKgRqHKGvPqG3QUyuEc8C5yRaDj7DYQDH88K2T0,3552
-aip_agents/tools/execute_ptc_code.py,sha256=h1-bnW4u5HoDFZGQbROitREE4MGblvlwKqHXSNL3XUM,11408
-aip_agents/tools/execute_ptc_code.pyi,sha256=4wXrAPjUYoXWODxg90expdvIetTlULvFZwGvKbdbJrE,4046
+aip_agents/tools/date_range_tool.py,sha256=t0vQqb0du1Q2n4q8V_-TPmfjSuwr4MYRMi6OpXMvVrw,22807
+aip_agents/tools/date_range_tool.pyi,sha256=1Caen334PHliSJOmXpu94kIFP_PPOdOj3dhKqdfMe8o,570
+aip_agents/tools/execute_ptc_code.py,sha256=mN8-G50voxVqWH9r3uJGQOtgPtxsHaTOnJrJuojQLIM,11396
+aip_agents/tools/execute_ptc_code.pyi,sha256=nDlpxV-kcKuNmtghahjXAtjWvtNv6D38x8-VTNCKYjU,4089
 aip_agents/tools/gl_connector_tools.py,sha256=bxl_3VQYZDv3lFn6Y3kDVVRFwH4cntOLz3f74YzDcic,3936
 aip_agents/tools/gl_connector_tools.pyi,sha256=2ATn_MW_FRg5Uv7dLI_ToBOtlgTSfj0zgDQpN1N-cJs,1366
 aip_agents/tools/memory_search_tool.py,sha256=gqTTb_Fao_Hl-SavfaFsqPDhnFQUjzIQUkzizSD2L0A,653
@@ -606,7 +608,7 @@ aip_agents/utils/pii/pii_helper.py,sha256=g0yRzakfA2AA6vjUNLWHqFlcxyLql6MXQ90NN3
 aip_agents/utils/pii/pii_helper.pyi,sha256=dulZs150ikbAL3Bw2YLcz3_g4DsGmL3lciwf8mKxEjI,2939
 aip_agents/utils/pii/uuid_deanonymizer_mapping.py,sha256=Gks8l8t0cuS9pzoQnrpiK1CaLmWYksjOnTeiHh3_7EE,7348
 aip_agents/utils/pii/uuid_deanonymizer_mapping.pyi,sha256=gnWfD1rWZh_tloJjgKiZ6f6iNUuBaHpKqCSiP0d-9bs,3084
-aip_agents_binary-0.6.1.dist-info/METADATA,sha256=SHkg6ELXvsKcMmpLPVaJSxPYoSHbpMqUmhouUUFqAVg,22371
-aip_agents_binary-0.6.1.dist-info/WHEEL,sha256=KxCTaSkoYs_EnWvWxmau4HAvN-_rCRYV_bfRc_41A9k,106
-aip_agents_binary-0.6.1.dist-info/top_level.txt,sha256=PEz8vcwC1bH4UrkhF0LkIYCNfXGWZUHdSklbvkBe25E,11
-aip_agents_binary-0.6.1.dist-info/RECORD,,
+aip_agents_binary-0.6.3.dist-info/METADATA,sha256=7BAliBsUy5k6VWI4QIMhyqul_9tjgYDM4NtsxIr64ak,22416
+aip_agents_binary-0.6.3.dist-info/WHEEL,sha256=KxCTaSkoYs_EnWvWxmau4HAvN-_rCRYV_bfRc_41A9k,106
+aip_agents_binary-0.6.3.dist-info/top_level.txt,sha256=PEz8vcwC1bH4UrkhF0LkIYCNfXGWZUHdSklbvkBe25E,11
+aip_agents_binary-0.6.3.dist-info/RECORD,,