2dc70a7814
| Name | Description | |----------------------------------|-------------------------------------------------------------------------| | Reddit.CheckSubredditAccess | Checks whether the specified subreddit exists and also if it is accessible | | Reddit.GetSubredditRules | Gets the rules of the specified subreddit | | Reddit.GetMyUsername | Get the Reddit username of the authenticated user | | Reddit.GetMyPosts | Get posts that were created by the authenticated user sorted by newest | Four new tools that I had to create for my demo app.
461 lines
15 KiB
Python
461 lines
15 KiB
Python
from arcade.sdk import ToolCatalog
|
|
from arcade.sdk.eval import (
|
|
EvalRubric,
|
|
EvalSuite,
|
|
ExpectedToolCall,
|
|
tool_eval,
|
|
)
|
|
from arcade.sdk.eval.critic import BinaryCritic
|
|
|
|
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
|