jpmschweitzerandClaude
78066fab1b
Mechanical only, and separated from the judgment calls that follow so the reviewable changes are not buried in a 98-file whitespace diff. 227 automatic fixes: 60 blank lines carrying whitespace, 60 unsorted import blocks, 34 Optional[X] to X | None, 28 unused imports, 16 deprecated typing imports, 12 datetime.timezone.utc to datetime.UTC, and assorted smaller modernisations. Then `ruff format` over src and tests: 98 files reformatted, 35 already conforming. No file among the unused-import findings defines __all__ or is an __init__.py, so nothing here removes a re-export. `make test`: 658 passed, unchanged from HEAD. Two things observed while verifying, neither addressed here: `pytest tests/` cannot collect — tests/e2e/test_orchestration_e2e.py uses an `e2e` marker that is not registered, and the config is strict about markers. This fails identically at HEAD, so it predates this change; `make test` passes because it ignores tests/e2e, tests/integration and tests/contracts. test_tatlock_tool_call_logging_calculator is flaky. It failed once in a full run with these changes and passed on the next, passes in isolation with them, and fails in isolation at HEAD. It is order- or timing-dependent, not a regression from this commit — established by running the full suite both ways rather than by reasoning about which change could have caused it. Co-Authored-By: Claude <noreply@anthropic.com>
79 lines
2.1 KiB
Python
79 lines
2.1 KiB
Python
"""
|
|
PydanticAI toolset for Tatlock's core tools.
|
|
|
|
Converts the core tool functions into PydanticAI tool definitions
|
|
that can be registered with agents and the household registry.
|
|
"""
|
|
|
|
from pydantic_ai.tools import Tool
|
|
|
|
from . import tools
|
|
|
|
# Create tool definitions for PydanticAI
|
|
calculator_tool = Tool(
|
|
function=tools.calculate,
|
|
name="calculate",
|
|
description=(
|
|
"Safely evaluate mathematical expressions. "
|
|
"Supports basic arithmetic (+, -, *, /, %, **), "
|
|
"functions (sqrt, sin, cos, log, exp, etc.), "
|
|
"and constants (pi, e). "
|
|
"Use this for ALL mathematical calculations."
|
|
),
|
|
)
|
|
|
|
current_datetime_tool = Tool(
|
|
function=tools.get_current_datetime,
|
|
name="get_current_datetime",
|
|
description=(
|
|
"Get the current date and time. "
|
|
"Supports various formats: 'full' (datetime), 'date' (YYYY-MM-DD), "
|
|
"'time' (HH:MM:SS), 'iso' (ISO 8601), or custom strftime format. "
|
|
"Use this instead of guessing the current date/time."
|
|
),
|
|
)
|
|
|
|
time_offset_tool = Tool(
|
|
function=tools.calculate_time_offset,
|
|
name="calculate_time_offset",
|
|
description=(
|
|
"Calculate a date/time relative to now. "
|
|
"Accepts natural language like '1 week ago', '2 days from now', "
|
|
"'3 months ago', etc. "
|
|
"Use this for calculating past or future dates."
|
|
),
|
|
)
|
|
|
|
time_difference_tool = Tool(
|
|
function=tools.time_difference,
|
|
name="time_difference",
|
|
description=(
|
|
"Calculate the difference between two dates. "
|
|
"Accepts dates in YYYY-MM-DD or YYYY-MM-DD HH:MM:SS format. "
|
|
"Second date can be 'now'. "
|
|
"Returns human-readable difference (e.g., '5 days, 3 hours')."
|
|
),
|
|
)
|
|
|
|
# NOTE: Web search has been moved to The Librarian agent.
|
|
# Use delegate_to_librarian(task="search web for ...") for web search.
|
|
|
|
|
|
# Combined toolset of all core tools
|
|
tatlock_core_tools = [
|
|
calculator_tool,
|
|
current_datetime_tool,
|
|
time_offset_tool,
|
|
time_difference_tool,
|
|
]
|
|
|
|
|
|
def get_core_tools():
|
|
"""
|
|
Get list of Tatlock's core tool definitions.
|
|
|
|
Returns:
|
|
List of PydanticAI Tool objects
|
|
"""
|
|
return tatlock_core_tools
|