mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-25 16:18:13 +00:00
Update tool usage logging to ensure tool arguments are consistently formatted as strings. Adjust agent test cases to reflect changes in maximum iterations and expected outputs, enhancing clarity in assertions. Update YAML cassettes to align with new response formats and improve overall consistency across tests.
This commit is contained in:
@@ -562,7 +562,7 @@ class ToolUsage:
|
|||||||
"run_attempts": self._run_attempts,
|
"run_attempts": self._run_attempts,
|
||||||
"delegations": self.task.delegations if self.task else 0,
|
"delegations": self.task.delegations if self.task else 0,
|
||||||
"tool_name": tool.name,
|
"tool_name": tool.name,
|
||||||
"tool_args": tool_calling.arguments,
|
"tool_args": str(tool_calling.arguments),
|
||||||
"tool_class": tool.__class__.__name__,
|
"tool_class": tool.__class__.__name__,
|
||||||
"agent_key": (
|
"agent_key": (
|
||||||
getattr(self.agent, "key", "unknown") if self.agent else "unknown"
|
getattr(self.agent, "key", "unknown") if self.agent else "unknown"
|
||||||
|
|||||||
@@ -530,7 +530,7 @@ def test_agent_moved_on_after_max_iterations():
|
|||||||
role="test role",
|
role="test role",
|
||||||
goal="test goal",
|
goal="test goal",
|
||||||
backstory="test backstory",
|
backstory="test backstory",
|
||||||
max_iter=3,
|
max_iter=5,
|
||||||
allow_delegation=False,
|
allow_delegation=False,
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -551,6 +551,7 @@ def test_agent_respect_the_max_rpm_set(capsys):
|
|||||||
def get_final_answer() -> float:
|
def get_final_answer() -> float:
|
||||||
"""Get the final answer but don't give it yet, just re-use this
|
"""Get the final answer but don't give it yet, just re-use this
|
||||||
tool non-stop."""
|
tool non-stop."""
|
||||||
|
return 42
|
||||||
|
|
||||||
agent = Agent(
|
agent = Agent(
|
||||||
role="test role",
|
role="test role",
|
||||||
@@ -572,7 +573,7 @@ def test_agent_respect_the_max_rpm_set(capsys):
|
|||||||
task=task,
|
task=task,
|
||||||
tools=[get_final_answer],
|
tools=[get_final_answer],
|
||||||
)
|
)
|
||||||
assert output == "The final answer is 42."
|
assert output == "42"
|
||||||
captured = capsys.readouterr()
|
captured = capsys.readouterr()
|
||||||
assert "Max RPM reached, waiting for next minute to start." in captured.out
|
assert "Max RPM reached, waiting for next minute to start." in captured.out
|
||||||
moveon.assert_called()
|
moveon.assert_called()
|
||||||
|
|||||||
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
@@ -355,7 +355,7 @@ def test_tools_emits_finished_events():
|
|||||||
assert received_events[0].agent_key == agent.key
|
assert received_events[0].agent_key == agent.key
|
||||||
assert received_events[0].agent_role == agent.role
|
assert received_events[0].agent_role == agent.role
|
||||||
assert received_events[0].tool_name == SayHiTool().name
|
assert received_events[0].tool_name == SayHiTool().name
|
||||||
assert received_events[0].tool_args == {}
|
assert received_events[0].tool_args == "{}" or received_events[0].tool_args == {}
|
||||||
assert received_events[0].type == "tool_usage_finished"
|
assert received_events[0].type == "tool_usage_finished"
|
||||||
assert isinstance(received_events[0].timestamp, datetime)
|
assert isinstance(received_events[0].timestamp, datetime)
|
||||||
|
|
||||||
@@ -401,7 +401,7 @@ def test_tools_emits_error_events():
|
|||||||
assert received_events[0].agent_key == agent.key
|
assert received_events[0].agent_key == agent.key
|
||||||
assert received_events[0].agent_role == agent.role
|
assert received_events[0].agent_role == agent.role
|
||||||
assert received_events[0].tool_name == "error_tool"
|
assert received_events[0].tool_name == "error_tool"
|
||||||
assert received_events[0].tool_args == {}
|
assert received_events[0].tool_args == "{}" or received_events[0].tool_args == {}
|
||||||
assert str(received_events[0].error) == "Simulated tool error"
|
assert str(received_events[0].error) == "Simulated tool error"
|
||||||
assert received_events[0].type == "tool_usage_error"
|
assert received_events[0].type == "tool_usage_error"
|
||||||
assert isinstance(received_events[0].timestamp, datetime)
|
assert isinstance(received_events[0].timestamp, datetime)
|
||||||
|
|||||||
Reference in New Issue
Block a user