Skip to content

Tool Call Error

Source code in OpAgentsOlympus/practice/tool_call_error.py
OpAgentsOlympus/practice/tool_call_error.py
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
from agents import RunContextWrapper, Agent, Runner, function_tool
from typing import Any
from config import config


def custom_error_handler(ctx: RunContextWrapper[Any], error: Exception) -> str:
    print("Error occurred")
    return f"Error: Tool failed with {error.__class__.__name__}: {str(error)}"


@function_tool(failure_error_function=custom_error_handler)
def multiply_by_2(x: int) -> int:
    raise ValueError()
    print(f"Multiplying {x} by 2")
    return x * 2


assistant = Agent(
    name="assistant",
    instructions="You are a friendly assistant.",
    tools=[multiply_by_2],
)

result = Runner.run_sync(assistant, "multiply 4 with 2", run_config=config)
print(result.final_output)