7c228a59d5
# PR Description
This PR renames `ExpectedToolCall` to `NamedExpectedToolCall` and then
creates a new dataclass called `ExpectedToolCall`. `ExpectedToolCall`
can be passed to the `EvalSuite.add_case` and `EvalSuite.extend_case`
methods.
1. Enhance `EvalSuite.add_case` and `EvalSuite.extend_case` by accepting
a list of `ExpectedToolCall` as their `expected_tool_calls` input
parameter. This helps create a scaffolding for developers. Previously,
the expected type was `list[tuple[Callable, dict[str, Any]]]`, which is
still valid for backward compatibility.
```python
# Before (still valid for backward compatibility)
expected_tool_calls=[
(
adjust_playback_position,
{
"absolute_position_ms": 10000,
},
)
]
# After
expected_tool_calls=[
ExpectedToolCall(
func=adjust_playback_position,
args={"absolute_position_ms": 10000},
)
]
```
2. Removed any references to arcade.core in toolkits directory.
3. Some linting for import organization.
119 lines
2.8 KiB
Python
119 lines
2.8 KiB
Python
import arcade_code_sandbox
|
|
from arcade_code_sandbox.tools.e2b import create_static_matplotlib_chart, run_code
|
|
from arcade_code_sandbox.tools.models import E2BSupportedLanguage
|
|
|
|
from arcade.sdk import ToolCatalog
|
|
from arcade.sdk.eval import (
|
|
BinaryCritic,
|
|
EvalRubric,
|
|
EvalSuite,
|
|
ExpectedToolCall,
|
|
SimilarityCritic,
|
|
tool_eval,
|
|
)
|
|
|
|
merge_sort_code = """
|
|
def merge_sort(arr):
|
|
if len(arr) <= 1:
|
|
return arr
|
|
|
|
mid = len(arr) // 2
|
|
left = merge_sort(arr[:mid])
|
|
right = merge_sort(arr[mid:])
|
|
|
|
return merge(left, right)
|
|
|
|
def merge(left, right):
|
|
result = []
|
|
i, j = 0, 0
|
|
|
|
while i < len(left) and j < len(right):
|
|
if left[i] < right[j]:
|
|
result.append(left[i])
|
|
i += 1
|
|
else:
|
|
result.append(right[j])
|
|
j += 1
|
|
|
|
result.extend(left[i:])
|
|
result.extend(right[j:])
|
|
|
|
return result
|
|
|
|
sample_list = ["banana", "apple", "cherry", "date", "elderberry"]
|
|
|
|
sorted_list = merge_sort(sample_list)
|
|
print("Sorted list:", sorted_list)
|
|
"""
|
|
|
|
matplotlib_chart_code = """
|
|
import matplotlib.pyplot as plt
|
|
|
|
labels = ['Apples', 'Bananas', 'Cherries', 'Dates']
|
|
sizes = [30, 25, 20, 25]
|
|
colors = ['red', 'yellow', 'purple', 'brown']
|
|
|
|
plt.pie(sizes, labels=labels, colors=colors, autopct='%1.1f%%', startangle=90)
|
|
|
|
plt.axis('equal')
|
|
|
|
plt.title('Fruit Distribution')
|
|
|
|
plt.savefig('fruit_pie_chart.png')
|
|
"""
|
|
|
|
# Evaluation rubric
|
|
rubric = EvalRubric(
|
|
fail_threshold=0.85,
|
|
warn_threshold=0.95,
|
|
)
|
|
|
|
|
|
catalog = ToolCatalog()
|
|
catalog.add_module(arcade_code_sandbox)
|
|
|
|
|
|
@tool_eval()
|
|
def code_sandbox_eval_suite():
|
|
suite = EvalSuite(
|
|
name="code_sandbox Tools Evaluation",
|
|
system_message="You are an AI assistant with access to code_sandbox tools. Use them to help the user with their tasks.",
|
|
catalog=catalog,
|
|
rubric=rubric,
|
|
)
|
|
|
|
suite.add_case(
|
|
name="Run code",
|
|
user_message=f"Can you please run my merge sort algo?\n\n{merge_sort_code}",
|
|
expected_tool_calls=[
|
|
ExpectedToolCall(
|
|
func=run_code,
|
|
args={
|
|
"code": merge_sort_code,
|
|
"language": E2BSupportedLanguage.PYTHON,
|
|
},
|
|
)
|
|
],
|
|
critics=[
|
|
SimilarityCritic(critic_field="code", weight=0.8),
|
|
BinaryCritic(critic_field="language", weight=0.2),
|
|
],
|
|
)
|
|
|
|
suite.add_case(
|
|
name="Create static matplotlib chart",
|
|
user_message=f"Run this code:\n\n{matplotlib_chart_code}",
|
|
expected_tool_calls=[
|
|
ExpectedToolCall(
|
|
func=create_static_matplotlib_chart,
|
|
args={
|
|
"code": matplotlib_chart_code,
|
|
},
|
|
)
|
|
],
|
|
critics=[
|
|
SimilarityCritic(critic_field="code", weight=1.0),
|
|
],
|
|
)
|
|
|
|
return suite
|