b6b4cd0a4c
### Overview Major restructuring from monolithic `arcade-ai` package to modular library architecture with standardized uv-based dependency management.  ### New Package Structure - **`arcade-tdk`** - Lightweight toolkit development kit (core decorators, auth) - **`arcade-core`** - Core execution engine and catalog functionality - **`arcade-serve`** - FastAPI/MCP server components - **`arcade-ai`** - Meta package that includes CLI functionality. Optionally include evals via the `evals` extra. Optionally include all packages via the `all` extra. ### Key Benefits - **Lighter Dependencies**: Toolkits now depend only on `arcade-tdk` (~2 deps) vs full `arcade-ai` (~30+ deps) - **Faster Builds**: uv provides 10-100x faster dependency resolution and installation - **Better Modularity**: Clear separation of concerns, consumers import only what they need - **Standard Tooling**: Eliminates custom poetry scripts, uses standard Python packaging ### Migration Impact - All 20 toolkits converted from poetry → uv with `arcade-tdk` dependencies plus `arcade-ai[evals]` and `arcade-serve` dev dependencies. When developing locally, devs should install toolkits via `make install-local`. - Modern Python 3.10+ type hints throughout - Standardized build system with hatchling backend - Enhanced Makefile with robust toolkit management commands - Removed `arcade dev` CLI command - Reduce the number of files created by `arcade new` and add an option to not generate a tests and evals folder. This foundation enables faster development cycles and cleaner dependency chains for the growing toolkit ecosystem. ### Todo After this PR is merged - [ ] Post-merge workflow(s) (release & publish containers, etc) - [ ] Release order plan. @EricGustin suggests releasing in the following order: 1. `arcade-core` version 0.1.0 2. `arcade-serve` version 0.1.0 and `arcade-tdk` version 0.1.0 3. `arcade-ai` version 2.0.0 4. Patch release for all toolkits (all changes in toolkits are internal refactors) - [ ] [Update docs](https://github.com/ArcadeAI/docs/pull/318) --------- Co-authored-by: Eric Gustin <eric@arcade.dev> Co-authored-by: Eric Gustin <34000337+EricGustin@users.noreply.github.com>
461 lines
15 KiB
Python
461 lines
15 KiB
Python
from arcade_evals import (
|
|
BinaryCritic,
|
|
EvalRubric,
|
|
EvalSuite,
|
|
ExpectedToolCall,
|
|
tool_eval,
|
|
)
|
|
from arcade_tdk import ToolCatalog
|
|
|
|
import arcade_reddit
|
|
from arcade_reddit.enums import RedditTimeFilter, SubredditListingType
|
|
from arcade_reddit.tools import (
|
|
get_content_of_post,
|
|
get_posts_in_subreddit,
|
|
get_top_level_comments,
|
|
)
|
|
from arcade_reddit.tools.read import (
|
|
check_subreddit_access,
|
|
get_content_of_multiple_posts,
|
|
get_my_posts,
|
|
get_subreddit_rules,
|
|
)
|
|
from evals.additional_messages import get_post_in_subreddit_messages
|
|
from evals.critics import AnyOfCritic, ListCritic
|
|
|
|
# Evaluation rubric
|
|
rubric = EvalRubric(
|
|
fail_threshold=0.85,
|
|
warn_threshold=0.95,
|
|
)
|
|
|
|
|
|
catalog = ToolCatalog()
|
|
catalog.add_module(arcade_reddit)
|
|
|
|
|
|
@tool_eval()
|
|
def reddit_get_posts_in_subreddit_eval_suite() -> EvalSuite:
|
|
suite = EvalSuite(
|
|
name="reddit_get_posts_in_subreddit_1",
|
|
system_message=(
|
|
"You are an AI assistant with access to reddit tools. "
|
|
"Use them to help the user with their tasks."
|
|
),
|
|
catalog=catalog,
|
|
rubric=rubric,
|
|
)
|
|
|
|
suite.add_case(
|
|
name="reddit_get_posts_in_subreddit_1",
|
|
user_message="Get 30 posts from AskReddit that are contentious at this moment.",
|
|
expected_tool_calls=[
|
|
ExpectedToolCall(
|
|
func=get_posts_in_subreddit,
|
|
args={
|
|
"subreddit": "AskReddit",
|
|
"listing": SubredditListingType.CONTROVERSIAL.value,
|
|
"limit": 30,
|
|
"cursor": None,
|
|
"time_range": RedditTimeFilter.NOW.value,
|
|
},
|
|
),
|
|
],
|
|
rubric=rubric,
|
|
critics=[
|
|
BinaryCritic(critic_field="subreddit", weight=0.3),
|
|
BinaryCritic(critic_field="listing", weight=0.2),
|
|
BinaryCritic(critic_field="limit", weight=0.2),
|
|
BinaryCritic(critic_field="cursor", weight=0.1),
|
|
BinaryCritic(critic_field="time_range", weight=0.2),
|
|
],
|
|
)
|
|
|
|
suite.add_case(
|
|
name="reddit_get_posts_in_subreddit_2",
|
|
user_message="Get the next 5 posts from AskReddit that are contentious at this moment.",
|
|
expected_tool_calls=[
|
|
ExpectedToolCall(
|
|
func=get_posts_in_subreddit,
|
|
args={
|
|
"subreddit": "AskReddit",
|
|
"listing": SubredditListingType.CONTROVERSIAL.value,
|
|
"limit": 5,
|
|
"cursor": "t3_1abcdef",
|
|
},
|
|
),
|
|
],
|
|
rubric=rubric,
|
|
critics=[
|
|
BinaryCritic(critic_field="subreddit", weight=0.2),
|
|
BinaryCritic(critic_field="listing", weight=0.2),
|
|
BinaryCritic(critic_field="limit", weight=0.1),
|
|
BinaryCritic(critic_field="cursor", weight=0.3),
|
|
BinaryCritic(critic_field="time_range", weight=0.2),
|
|
],
|
|
additional_messages=get_post_in_subreddit_messages,
|
|
)
|
|
|
|
suite.add_case( # time-based listing, but don't provide a specific time range
|
|
name="reddit_get_posts_in_subreddit_3",
|
|
user_message="Get 5 top posts from AskReddit",
|
|
expected_tool_calls=[
|
|
ExpectedToolCall(
|
|
func=get_posts_in_subreddit,
|
|
args={
|
|
"subreddit": "AskReddit",
|
|
"listing": SubredditListingType.TOP.value,
|
|
"limit": 5,
|
|
},
|
|
),
|
|
],
|
|
rubric=rubric,
|
|
critics=[
|
|
BinaryCritic(critic_field="subreddit", weight=0.3),
|
|
BinaryCritic(critic_field="listing", weight=0.3),
|
|
BinaryCritic(critic_field="limit", weight=0.3),
|
|
],
|
|
)
|
|
|
|
suite.add_case(
|
|
name="reddit_get_posts_in_subreddit_4",
|
|
user_message="Get posts from AskReddit that are gaining traction as we speak",
|
|
expected_tool_calls=[
|
|
ExpectedToolCall(
|
|
func=get_posts_in_subreddit,
|
|
args={
|
|
"subreddit": "AskReddit",
|
|
"listing": SubredditListingType.RISING.value,
|
|
},
|
|
),
|
|
],
|
|
rubric=rubric,
|
|
critics=[
|
|
BinaryCritic(critic_field="subreddit", weight=0.3),
|
|
BinaryCritic(critic_field="listing", weight=0.7),
|
|
],
|
|
)
|
|
|
|
return suite
|
|
|
|
|
|
@tool_eval()
|
|
def reddit_get_content_of_post_eval_suite() -> EvalSuite:
|
|
suite = EvalSuite(
|
|
name="reddit_get_content_of_post",
|
|
system_message=(
|
|
"You are an AI assistant with access to reddit tools. "
|
|
"Use them to help the user with their tasks."
|
|
),
|
|
catalog=catalog,
|
|
rubric=rubric,
|
|
)
|
|
|
|
suite.add_case(
|
|
name="reddit_get_content_of_post_1",
|
|
user_message="Get the content of the post with the id t3_1abcdef",
|
|
expected_tool_calls=[
|
|
ExpectedToolCall(
|
|
func=get_content_of_post,
|
|
args={
|
|
"post_identifier": [ # post_identifier can be any of the following
|
|
"1abcdef",
|
|
"t3_1abcdef",
|
|
"https://www.reddit.com/r/AskReddit/comments/1abcdef/why_does_my_dog_have_four_legs/",
|
|
"/r/AskReddit/comments/1abcdef/why_does_my_dog_have_four_legs/",
|
|
],
|
|
},
|
|
),
|
|
],
|
|
rubric=rubric,
|
|
critics=[
|
|
AnyOfCritic(
|
|
critic_field="post_identifier",
|
|
weight=1.0,
|
|
),
|
|
],
|
|
additional_messages=get_post_in_subreddit_messages,
|
|
)
|
|
|
|
return suite
|
|
|
|
|
|
@tool_eval()
|
|
def reddit_get_content_of_multiple_posts_eval_suite() -> EvalSuite:
|
|
suite = EvalSuite(
|
|
name="reddit_get_content_of_multiple_posts",
|
|
system_message=(
|
|
"You are an AI assistant with access to reddit tools. "
|
|
"Use them to help the user with their tasks."
|
|
),
|
|
catalog=catalog,
|
|
rubric=rubric,
|
|
)
|
|
|
|
suite.add_case(
|
|
name="reddit_get_content_of_multiple_posts_1",
|
|
user_message=(
|
|
"Get the content of the posts t3_1abcdef, "
|
|
"https://www.reddit.com/r/AskReddit/comments/1jdfgk1vn/why_is_water_wet/, "
|
|
"t3_3abcdef, t3_4abcdef, and t3_5abcdef, t3_6abcdef, and t3_7abcdef, 4jfnsklf, "
|
|
"/r/AskReddit/comments/2dsghr/, and /r/AskReddit/comments/1jdfg35dvn/"
|
|
),
|
|
expected_tool_calls=[
|
|
ExpectedToolCall(
|
|
func=get_content_of_multiple_posts,
|
|
args={
|
|
"post_identifiers": [
|
|
"t3_1abcdef",
|
|
"https://www.reddit.com/r/AskReddit/comments/1jdfgk1vn/why_is_water_wet/",
|
|
"t3_3abcdef",
|
|
"t3_4abcdef",
|
|
"t3_5abcdef",
|
|
"t3_6abcdef",
|
|
"t3_7abcdef",
|
|
"4jfnsklf",
|
|
"/r/AskReddit/comments/2dsghr/",
|
|
"/r/AskReddit/comments/1jdfg35dvn/",
|
|
],
|
|
},
|
|
),
|
|
],
|
|
rubric=rubric,
|
|
critics=[
|
|
ListCritic(
|
|
critic_field="post_identifiers",
|
|
weight=1.0,
|
|
order_matters=False,
|
|
duplicates_matter=True,
|
|
),
|
|
],
|
|
)
|
|
|
|
return suite
|
|
|
|
|
|
@tool_eval()
|
|
def reddit_get_top_level_comments_eval_suite() -> EvalSuite:
|
|
suite = EvalSuite(
|
|
name="reddit_get_top_level_comments",
|
|
system_message=(
|
|
"You are an AI assistant with access to reddit tools. "
|
|
"Use them to help the user with their tasks."
|
|
),
|
|
catalog=catalog,
|
|
rubric=rubric,
|
|
)
|
|
|
|
suite.add_case(
|
|
name="reddit_get_top_level_comments_1",
|
|
user_message="What are people saying in response?",
|
|
expected_tool_calls=[
|
|
ExpectedToolCall(
|
|
func=get_top_level_comments,
|
|
args={
|
|
"post_identifier": [ # post_identifier can be any of the following
|
|
"1abcdef",
|
|
"t3_1abcdef",
|
|
"https://www.reddit.com/r/AskReddit/comments/1abcdef/why_does_my_dog_have_four_legs/",
|
|
"/r/AskReddit/comments/1abcdef/why_does_my_dog_have_four_legs/",
|
|
],
|
|
},
|
|
),
|
|
],
|
|
rubric=rubric,
|
|
critics=[
|
|
AnyOfCritic(
|
|
critic_field="post_identifier",
|
|
weight=1.0,
|
|
),
|
|
],
|
|
additional_messages=get_post_in_subreddit_messages,
|
|
)
|
|
|
|
return suite
|
|
|
|
|
|
@tool_eval()
|
|
def reddit_check_subreddit_access_eval_suite() -> EvalSuite:
|
|
suite = EvalSuite(
|
|
name="reddit_check_subreddit_access",
|
|
system_message=(
|
|
"You are an AI assistant with access to reddit tools. "
|
|
"Use them to help the user with their tasks."
|
|
),
|
|
catalog=catalog,
|
|
rubric=rubric,
|
|
)
|
|
|
|
suite.add_case(
|
|
name="reddit_check_subreddit_access_1",
|
|
user_message="does r/WaterBottleCollecting exist?",
|
|
expected_tool_calls=[
|
|
ExpectedToolCall(
|
|
func=check_subreddit_access,
|
|
args={
|
|
"subreddit": "WaterBottleCollecting",
|
|
},
|
|
),
|
|
],
|
|
rubric=rubric,
|
|
critics=[
|
|
BinaryCritic(critic_field="subreddit", weight=1.0),
|
|
],
|
|
)
|
|
|
|
suite.add_case(
|
|
name="reddit_check_subreddit_access_2",
|
|
user_message=(
|
|
"so my friend is a part of the WaterBottleCollecting subreddit, "
|
|
"but i cant find it. Why?"
|
|
),
|
|
expected_tool_calls=[
|
|
ExpectedToolCall(
|
|
func=check_subreddit_access,
|
|
args={
|
|
"subreddit": "WaterBottleCollecting",
|
|
},
|
|
),
|
|
],
|
|
rubric=rubric,
|
|
critics=[
|
|
BinaryCritic(critic_field="subreddit", weight=1.0),
|
|
],
|
|
)
|
|
|
|
return suite
|
|
|
|
|
|
@tool_eval()
|
|
def reddit_get_subreddit_rules_eval_suite() -> EvalSuite:
|
|
suite = EvalSuite(
|
|
name="reddit_get_subreddit_rules",
|
|
system_message=(
|
|
"You are an AI assistant with access to reddit tools. "
|
|
"Use them to help the user with their tasks."
|
|
),
|
|
catalog=catalog,
|
|
rubric=rubric,
|
|
)
|
|
|
|
suite.add_case(
|
|
name="reddit_get_subreddit_rules_1",
|
|
user_message=(
|
|
"I'm going to be posting some stuff on WaterBottleCollecting, "
|
|
"but I'm scared that I might go against their terms & conditions "
|
|
"and get my post removed."
|
|
),
|
|
expected_tool_calls=[
|
|
ExpectedToolCall(
|
|
func=get_subreddit_rules,
|
|
args={"subreddit": "WaterBottleCollecting"},
|
|
),
|
|
],
|
|
rubric=rubric,
|
|
critics=[
|
|
BinaryCritic(critic_field="subreddit", weight=1.0),
|
|
],
|
|
)
|
|
|
|
suite.add_case(
|
|
name="reddit_get_subreddit_rules_2",
|
|
user_message=(
|
|
"What are WaterBottleCollecting's bannable offenses? I don't want to get banned!"
|
|
),
|
|
expected_tool_calls=[
|
|
ExpectedToolCall(
|
|
func=get_subreddit_rules,
|
|
args={"subreddit": "WaterBottleCollecting"},
|
|
),
|
|
],
|
|
rubric=rubric,
|
|
critics=[
|
|
BinaryCritic(critic_field="subreddit", weight=1.0),
|
|
],
|
|
)
|
|
|
|
return suite
|
|
|
|
|
|
@tool_eval()
|
|
def reddit_get_my_posts_eval_suite() -> EvalSuite:
|
|
get_my_posts_response = [
|
|
{"role": "user", "content": "get 1 of my posts"},
|
|
{
|
|
"role": "assistant",
|
|
"content": "",
|
|
"tool_calls": [
|
|
{
|
|
"id": "call_hPacHNSvuKKamoKKoPWBQosv",
|
|
"type": "function",
|
|
"function": {"name": "Reddit_GetMyPosts", "arguments": '{"limit":1}'},
|
|
}
|
|
],
|
|
},
|
|
{
|
|
"role": "tool",
|
|
"content": '{"cursor":"t3_1jt9jz4","posts":[{"author":"RedditUser123","body":"i just wanted to say that i love thisapp\\n\\n","created_utc":1743988489,"id":"1jt9jz4","is_video":false,"name":"t3_1jt9jz4","num_comments":1,"permalink":"/r/SparkingWater/comments/1jt9jz4/this_is_fun/","score":1,"subreddit":"SparklingWater","title":"this isfun","upvote_ratio":1,"upvotes":1,"url":"https://www.reddit.com/r/SparklingWater/comments/1jt9jz4/this_is_fun/"}]}', # noqa: E501
|
|
"tool_call_id": "call_hPacHNSvuKKamoKKoPWBQosv",
|
|
"name": "Reddit_GetMyPosts",
|
|
},
|
|
{
|
|
"role": "assistant",
|
|
"content": "Here is one of your posts on Reddit:\n\n**Title:** [this isfun](https://www.reddit.com/r/SparklingWater/comments/1jt9jz4/this_is_fun/)\n\n**Subreddit:**r/SparklingWater\n\n**Content:** \n```\ni just wanted to say that i love this app\n```\n\n**Upvotes:** 1 \n**Comments:** 1", # noqa: E501
|
|
},
|
|
]
|
|
|
|
suite = EvalSuite(
|
|
name="reddit_get_my_posts",
|
|
system_message=(
|
|
"You are an AI assistant with access to reddit tools. "
|
|
"Use them to help the user with their tasks."
|
|
),
|
|
catalog=catalog,
|
|
rubric=rubric,
|
|
)
|
|
|
|
suite.add_case(
|
|
name="reddit_get_my_posts_1",
|
|
user_message=(
|
|
"I want to train an AI on the voice I use for my reddit posts. "
|
|
"Help me out here & get my last 100"
|
|
),
|
|
expected_tool_calls=[
|
|
ExpectedToolCall(
|
|
func=get_my_posts,
|
|
args={
|
|
"limit": 100,
|
|
"include_body": True,
|
|
},
|
|
),
|
|
],
|
|
rubric=rubric,
|
|
critics=[
|
|
BinaryCritic(critic_field="limit", weight=0.5),
|
|
BinaryCritic(critic_field="include_body", weight=0.5),
|
|
],
|
|
)
|
|
|
|
suite.add_case(
|
|
name="reddit_get_my_posts_2",
|
|
user_message=("get 25 more but w/o their content"),
|
|
expected_tool_calls=[
|
|
ExpectedToolCall(
|
|
func=get_my_posts,
|
|
args={
|
|
"limit": 25,
|
|
"include_body": False,
|
|
"cursor": "t3_1jt9jz4",
|
|
},
|
|
),
|
|
],
|
|
rubric=rubric,
|
|
critics=[
|
|
BinaryCritic(critic_field="limit", weight=0.3),
|
|
BinaryCritic(critic_field="include_body", weight=0.3),
|
|
BinaryCritic(critic_field="cursor", weight=0.4),
|
|
],
|
|
additional_messages=get_my_posts_response,
|
|
)
|
|
|
|
return suite
|