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>
443 lines
14 KiB
Python
443 lines
14 KiB
Python
import json
|
|
|
|
from arcade_evals import (
|
|
EvalRubric,
|
|
EvalSuite,
|
|
ExpectedToolCall,
|
|
tool_eval,
|
|
)
|
|
from arcade_evals.critic import BinaryCritic
|
|
from arcade_tdk import ToolCatalog
|
|
|
|
import arcade_jira
|
|
from arcade_jira.critics import CaseInsensitiveBinaryCritic, HasSubstringCritic
|
|
from arcade_jira.tools.issues import (
|
|
get_issue_by_id,
|
|
get_issues_without_id,
|
|
list_issues,
|
|
search_issues_with_jql,
|
|
)
|
|
|
|
# Evaluation rubric
|
|
rubric = EvalRubric(
|
|
fail_threshold=0.85,
|
|
warn_threshold=0.95,
|
|
)
|
|
|
|
|
|
catalog = ToolCatalog()
|
|
catalog.add_module(arcade_jira)
|
|
|
|
|
|
@tool_eval()
|
|
def get_issue_by_id_eval_suite() -> EvalSuite:
|
|
suite = EvalSuite(
|
|
name="Get issue by ID eval suite",
|
|
system_message=(
|
|
"You are an AI assistant with access to Jira tools. "
|
|
"Use them to help the user with their tasks."
|
|
),
|
|
catalog=catalog,
|
|
rubric=rubric,
|
|
)
|
|
|
|
suite.add_case(
|
|
name="Get issue by ID",
|
|
user_message="Get the issue with ID '10000'.",
|
|
expected_tool_calls=[
|
|
ExpectedToolCall(
|
|
func=get_issue_by_id,
|
|
args={
|
|
"issue_id": "10000",
|
|
},
|
|
),
|
|
],
|
|
rubric=rubric,
|
|
critics=[
|
|
BinaryCritic(critic_field="issue_id", weight=1.0),
|
|
],
|
|
)
|
|
|
|
suite.add_case(
|
|
name="Get issue by Key",
|
|
user_message="Get the issue ENG-103.",
|
|
expected_tool_calls=[
|
|
ExpectedToolCall(
|
|
func=get_issue_by_id,
|
|
args={
|
|
"issue_id": "ENG-103",
|
|
},
|
|
),
|
|
],
|
|
rubric=rubric,
|
|
critics=[
|
|
BinaryCritic(critic_field="issue_id", weight=1.0),
|
|
],
|
|
)
|
|
|
|
return suite
|
|
|
|
|
|
@tool_eval()
|
|
def get_issues_without_id_eval_suite() -> EvalSuite:
|
|
suite = EvalSuite(
|
|
name="Get issues without an ID",
|
|
system_message=(
|
|
"You are an AI assistant with access to Jira tools. "
|
|
"Use them to help the user with their tasks. "
|
|
"Today is 2025-05-27 (Tuesday)."
|
|
),
|
|
catalog=catalog,
|
|
rubric=rubric,
|
|
)
|
|
|
|
suite.add_case(
|
|
name="Get issues by keywords",
|
|
user_message="Find the issue about implementing the message queue.",
|
|
expected_tool_calls=[
|
|
ExpectedToolCall(
|
|
func=get_issues_without_id,
|
|
args={
|
|
"keywords": "message queue",
|
|
},
|
|
),
|
|
],
|
|
rubric=rubric,
|
|
critics=[
|
|
HasSubstringCritic(critic_field="keywords", weight=1.0),
|
|
],
|
|
)
|
|
|
|
suite.add_case(
|
|
name="Get issues by due date",
|
|
user_message="Which issues are due this month?",
|
|
expected_tool_calls=[
|
|
ExpectedToolCall(
|
|
func=get_issues_without_id,
|
|
args={
|
|
"due_from": "2025-05-01",
|
|
"due_until": "2025-05-31",
|
|
},
|
|
),
|
|
],
|
|
rubric=rubric,
|
|
critics=[
|
|
BinaryCritic(critic_field="due_from", weight=0.5),
|
|
BinaryCritic(critic_field="due_until", weight=0.5),
|
|
],
|
|
)
|
|
|
|
suite.add_case(
|
|
name="Get issues by assignee, due date, status, priority and issue type",
|
|
user_message=(
|
|
"Find task issues assigned to John Doe that are in progress, "
|
|
"with high priority, and due until the end of this month"
|
|
),
|
|
expected_tool_calls=[
|
|
ExpectedToolCall(
|
|
func=get_issues_without_id,
|
|
args={
|
|
"assignee": "John Doe",
|
|
"due_from": None,
|
|
"due_until": "2025-05-31",
|
|
"status": "in progress",
|
|
"priority": "high",
|
|
"issue_type": "task",
|
|
},
|
|
),
|
|
],
|
|
rubric=rubric,
|
|
critics=[
|
|
BinaryCritic(critic_field="due_from", weight=0.1),
|
|
BinaryCritic(critic_field="due_until", weight=0.1),
|
|
CaseInsensitiveBinaryCritic(critic_field="assignee", weight=0.2),
|
|
CaseInsensitiveBinaryCritic(critic_field="status", weight=0.2),
|
|
CaseInsensitiveBinaryCritic(critic_field="priority", weight=0.2),
|
|
CaseInsensitiveBinaryCritic(critic_field="issue_type", weight=0.2),
|
|
],
|
|
)
|
|
|
|
suite.add_case(
|
|
name="Get issues by label and project name",
|
|
user_message="Find issues labeled with version 2 in the Engineering project",
|
|
expected_tool_calls=[
|
|
ExpectedToolCall(
|
|
func=get_issues_without_id,
|
|
args={
|
|
"project": "Engineering",
|
|
"labels": ["version 2"],
|
|
},
|
|
),
|
|
],
|
|
rubric=rubric,
|
|
critics=[
|
|
BinaryCritic(critic_field="project", weight=0.5),
|
|
BinaryCritic(critic_field="labels", weight=0.5),
|
|
],
|
|
)
|
|
|
|
suite.add_case(
|
|
name="Get issues by parent issue",
|
|
user_message="Get the children issues of ENG-123",
|
|
expected_tool_calls=[
|
|
ExpectedToolCall(
|
|
func=get_issues_without_id,
|
|
args={
|
|
"parent_issue": "ENG-123",
|
|
},
|
|
),
|
|
],
|
|
rubric=rubric,
|
|
critics=[
|
|
BinaryCritic(critic_field="parent_issue", weight=1.0),
|
|
],
|
|
)
|
|
|
|
suite.add_case(
|
|
name="Paginate issues in multiple chat turns",
|
|
user_message="Get the next page of issues",
|
|
expected_tool_calls=[
|
|
ExpectedToolCall(
|
|
func=get_issues_without_id,
|
|
args={
|
|
"assignee": "john doe",
|
|
"priority": "high",
|
|
"status": "in progress",
|
|
"issue_type": "task",
|
|
"limit": 2,
|
|
"offset": 4,
|
|
"next_page_token": "1234567890",
|
|
},
|
|
),
|
|
],
|
|
rubric=rubric,
|
|
critics=[
|
|
CaseInsensitiveBinaryCritic(critic_field="assignee", weight=1 / 7),
|
|
CaseInsensitiveBinaryCritic(critic_field="priority", weight=1 / 7),
|
|
CaseInsensitiveBinaryCritic(critic_field="status", weight=1 / 7),
|
|
CaseInsensitiveBinaryCritic(critic_field="issue_type", weight=1 / 7),
|
|
BinaryCritic(critic_field="limit", weight=1 / 7),
|
|
BinaryCritic(critic_field="offset", weight=1 / 7),
|
|
BinaryCritic(critic_field="next_page_token", weight=1 / 7),
|
|
],
|
|
additional_messages=[
|
|
{
|
|
"role": "user",
|
|
"content": (
|
|
"Find 2 tasks assigned to John Doe that are in progress, with high priority"
|
|
),
|
|
},
|
|
{
|
|
"role": "assistant",
|
|
"content": "",
|
|
"tool_calls": [
|
|
{
|
|
"id": "call_1",
|
|
"type": "function",
|
|
"function": {
|
|
"name": "Jira_GetIssuesWithoutId",
|
|
"arguments": json.dumps({
|
|
"assignee": "John Doe",
|
|
"priority": "high",
|
|
"status": "in progress",
|
|
"issue_type": "task",
|
|
"limit": 2,
|
|
"offset": 0,
|
|
}),
|
|
},
|
|
}
|
|
],
|
|
},
|
|
{
|
|
"role": "tool",
|
|
"content": json.dumps({
|
|
"issues": [
|
|
{
|
|
"id": "10001",
|
|
"key": "ENG-101",
|
|
"summary": "Implement the message queue",
|
|
"assignee": {
|
|
"id": "10010",
|
|
"name": "John Doe",
|
|
"email": "john.doe@example.com",
|
|
},
|
|
"status": {
|
|
"id": "10020",
|
|
"name": "In Progress",
|
|
},
|
|
"priority": {
|
|
"id": "10030",
|
|
"name": "High",
|
|
},
|
|
"issue_type": {
|
|
"id": "10040",
|
|
"name": "Task",
|
|
},
|
|
"project": {
|
|
"id": "10050",
|
|
"key": "ENG",
|
|
"name": "Engineering",
|
|
},
|
|
},
|
|
{
|
|
"id": "10002",
|
|
"key": "ENG-102",
|
|
"summary": "Deploy the message queue system",
|
|
"assignee": {
|
|
"id": "10010",
|
|
"name": "John Doe",
|
|
"email": "john.doe@example.com",
|
|
},
|
|
"status": {
|
|
"id": "10020",
|
|
"name": "In Progress",
|
|
},
|
|
"priority": {
|
|
"id": "10030",
|
|
"name": "High",
|
|
},
|
|
"issue_type": {
|
|
"id": "10040",
|
|
"name": "Task",
|
|
},
|
|
"project": {
|
|
"id": "10050",
|
|
"key": "ENG",
|
|
"name": "Engineering",
|
|
},
|
|
},
|
|
],
|
|
"pagination": {
|
|
"limit": 2,
|
|
"total_results": 2,
|
|
"next_page_token": "1234567890",
|
|
},
|
|
}),
|
|
"tool_call_id": "call_1",
|
|
"name": "Jira_GetIssuesWithoutId",
|
|
},
|
|
{
|
|
"role": "assistant",
|
|
"content": (
|
|
"Here are two issues:\n\n"
|
|
"1. ENG-101: Implement the message queue\n"
|
|
"2. ENG-102: Deploy the message queue system"
|
|
),
|
|
},
|
|
],
|
|
)
|
|
|
|
return suite
|
|
|
|
|
|
@tool_eval()
|
|
def search_issues_with_jql_eval_suite() -> EvalSuite:
|
|
suite = EvalSuite(
|
|
name="Search issues with JQL",
|
|
system_message=(
|
|
"You are an AI assistant with access to Jira tools. "
|
|
"Use them to help the user with their tasks. "
|
|
"Today is 2025-05-27 (Tuesday)."
|
|
),
|
|
catalog=catalog,
|
|
rubric=rubric,
|
|
)
|
|
|
|
jql_query_str = 'text ~ "message queue" AND dueDate <= 2025-05-31'
|
|
|
|
suite.add_case(
|
|
name="Search issues by keywords",
|
|
user_message=f"Search for up to 10 issues using the JQL query: {jql_query_str}",
|
|
expected_tool_calls=[
|
|
ExpectedToolCall(
|
|
func=search_issues_with_jql,
|
|
args={
|
|
"jql": jql_query_str,
|
|
"limit": 10,
|
|
"offset": 0,
|
|
},
|
|
),
|
|
],
|
|
rubric=rubric,
|
|
critics=[
|
|
HasSubstringCritic(critic_field="jql", weight=1 / 3),
|
|
BinaryCritic(critic_field="limit", weight=1 / 3),
|
|
BinaryCritic(critic_field="offset", weight=1 / 3),
|
|
],
|
|
)
|
|
|
|
return suite
|
|
|
|
|
|
@tool_eval()
|
|
def list_issues_eval_suite() -> EvalSuite:
|
|
suite = EvalSuite(
|
|
name="List issues eval suite",
|
|
system_message=(
|
|
"You are an AI assistant with access to Jira tools. "
|
|
"Use them to help the user with their tasks."
|
|
),
|
|
catalog=catalog,
|
|
rubric=rubric,
|
|
)
|
|
|
|
suite.add_case(
|
|
name="Get me any one issue in Jira",
|
|
user_message="Get me one issue in Jira.",
|
|
expected_tool_calls=[
|
|
ExpectedToolCall(
|
|
func=list_issues,
|
|
args={
|
|
"limit": 1,
|
|
"next_page_token": None,
|
|
},
|
|
),
|
|
],
|
|
rubric=rubric,
|
|
critics=[
|
|
BinaryCritic(critic_field="limit", weight=0.5),
|
|
BinaryCritic(critic_field="next_page_token", weight=0.5),
|
|
],
|
|
)
|
|
|
|
suite.add_case(
|
|
name="List 10 issues in Jira",
|
|
user_message="List 10 issues in Jira.",
|
|
expected_tool_calls=[
|
|
ExpectedToolCall(
|
|
func=list_issues,
|
|
args={
|
|
"limit": 10,
|
|
"next_page_token": None,
|
|
},
|
|
),
|
|
],
|
|
rubric=rubric,
|
|
critics=[
|
|
BinaryCritic(critic_field="limit", weight=0.5),
|
|
BinaryCritic(critic_field="next_page_token", weight=0.5),
|
|
],
|
|
)
|
|
|
|
suite.add_case(
|
|
name="List 10 issues in the Arcade project",
|
|
user_message="List 10 issues in the Arcade project.",
|
|
expected_tool_calls=[
|
|
ExpectedToolCall(
|
|
func=list_issues,
|
|
args={
|
|
"project": "Arcade",
|
|
"limit": 50,
|
|
"next_page_token": None,
|
|
},
|
|
),
|
|
],
|
|
rubric=rubric,
|
|
critics=[
|
|
CaseInsensitiveBinaryCritic(critic_field="project", weight=1 / 3),
|
|
BinaryCritic(critic_field="limit", weight=1 / 3),
|
|
BinaryCritic(critic_field="next_page_token", weight=1 / 3),
|
|
],
|
|
)
|
|
|
|
return suite
|