349f2fd9fe
* feat: add claude support * feat: add script for end-to-end evaluation with logging and task distribution * feat&fix: add tool result handling and update model default in evaluation script * chore: remove run_test_env.py script * feat&fix: implement action parsing for tool calls and update default action space * fix: update text formatting in action parsing and replace logger import * feat&fix: implement action parsing for tool calls and add screen size handling * feat: add setup instructions for Anthropic API integration * feat: add notice about image size limitations for Anthropic API * Delete test_env/logger.py * Delete test_env/utils.py
34 lines
921 B
Python
34 lines
921 B
Python
"""Collection classes for managing multiple tools."""
|
|
|
|
from typing import Any
|
|
|
|
from anthropic.types.beta import BetaToolUnionParam
|
|
|
|
from .base import (
|
|
BaseAnthropicTool,
|
|
ToolError,
|
|
ToolFailure,
|
|
ToolResult,
|
|
)
|
|
|
|
|
|
class ToolCollection:
|
|
"""A collection of anthropic-defined tools."""
|
|
|
|
def __init__(self, *tools: BaseAnthropicTool):
|
|
self.tools = tools
|
|
self.tool_map = {tool.to_params()["name"]: tool for tool in tools}
|
|
|
|
def to_params(
|
|
self,
|
|
) -> list[BetaToolUnionParam]:
|
|
return [tool.to_params() for tool in self.tools]
|
|
|
|
async def run(self, *, name: str, tool_input: dict[str, Any]) -> ToolResult:
|
|
tool = self.tool_map.get(name)
|
|
if not tool:
|
|
return ToolFailure(error=f"Tool {name} is invalid")
|
|
try:
|
|
return await tool(**tool_input)
|
|
except ToolError as e:
|
|
return ToolFailure(error=e.message) |