437 lines
14 KiB
Python
437 lines
14 KiB
Python
import json
|
|
|
|
from arcade.sdk import ToolCatalog
|
|
from arcade.sdk.eval import (
|
|
EvalRubric,
|
|
EvalSuite,
|
|
ExpectedToolCall,
|
|
tool_eval,
|
|
)
|
|
from arcade.sdk.eval.critic import BinaryCritic
|
|
|
|
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\n1. ENG-101: Implement the message queue\n2. 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
|