mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-10 00:28:31 +00:00
feat: remove test from tool execution erro:
This commit is contained in:
@@ -395,51 +395,3 @@ def test_deepseek_r1_with_open_router():
|
|||||||
result = llm.call("What is the capital of France?")
|
result = llm.call("What is the capital of France?")
|
||||||
assert isinstance(result, str)
|
assert isinstance(result, str)
|
||||||
assert "Paris" in result
|
assert "Paris" in result
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.vcr(filter_headers=["authorization"])
|
|
||||||
def test_tool_execution_error_event():
|
|
||||||
llm = LLM(model="gpt-4o-mini")
|
|
||||||
|
|
||||||
def failing_tool(param: str) -> str:
|
|
||||||
"""This tool always fails."""
|
|
||||||
raise Exception("Tool execution failed!")
|
|
||||||
|
|
||||||
tool_schema = {
|
|
||||||
"type": "function",
|
|
||||||
"function": {
|
|
||||||
"name": "failing_tool",
|
|
||||||
"description": "This tool always fails.",
|
|
||||||
"parameters": {
|
|
||||||
"type": "object",
|
|
||||||
"properties": {
|
|
||||||
"param": {"type": "string", "description": "A test parameter"}
|
|
||||||
},
|
|
||||||
"required": ["param"],
|
|
||||||
},
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
received_events = []
|
|
||||||
|
|
||||||
@crewai_event_bus.on(ToolExecutionErrorEvent)
|
|
||||||
def event_handler(source, event):
|
|
||||||
received_events.append(event)
|
|
||||||
|
|
||||||
available_functions = {"failing_tool": failing_tool}
|
|
||||||
|
|
||||||
messages = [{"role": "user", "content": "Use the failing tool"}]
|
|
||||||
|
|
||||||
llm.call(
|
|
||||||
messages,
|
|
||||||
tools=[tool_schema],
|
|
||||||
available_functions=available_functions,
|
|
||||||
)
|
|
||||||
|
|
||||||
assert len(received_events) == 1
|
|
||||||
event = received_events[0]
|
|
||||||
assert isinstance(event, ToolExecutionErrorEvent)
|
|
||||||
assert event.tool_name == "failing_tool"
|
|
||||||
assert event.tool_args == {"param": "test"}
|
|
||||||
assert event.tool_class == failing_tool
|
|
||||||
assert "Tool execution failed!" in event.error
|
|
||||||
|
|||||||
Reference in New Issue
Block a user