Add list stargazers tool (#130)
# PR Description As a celebration for arcade-ai becoming open sourced, this PR adds a tool to list the stargazers for a particular repository ### Example `arcade chat -h localhost` usage: 
This commit is contained in:
parent
d91d14aee4
commit
ddaeb4db53
7 changed files with 198 additions and 57 deletions
|
|
@ -1,55 +0,0 @@
|
||||||
from unittest.mock import AsyncMock, patch
|
|
||||||
|
|
||||||
import pytest
|
|
||||||
from arcade_github.tools.activity import set_starred
|
|
||||||
from httpx import Response
|
|
||||||
|
|
||||||
from arcade.sdk.errors import ToolExecutionError
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
|
||||||
def mock_context():
|
|
||||||
context = AsyncMock()
|
|
||||||
context.authorization.token = "mock_token" # noqa: S105
|
|
||||||
return context
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
|
||||||
def mock_client():
|
|
||||||
with patch("arcade_github.tools.activity.httpx.AsyncClient") as client:
|
|
||||||
yield client.return_value.__aenter__.return_value
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.asyncio
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
"starred,expected_message",
|
|
||||||
[
|
|
||||||
(True, "Successfully starred the repository owner/repo"),
|
|
||||||
(False, "Successfully unstarred the repository owner/repo"),
|
|
||||||
],
|
|
||||||
)
|
|
||||||
async def test_set_starred_success(mock_context, mock_client, starred, expected_message):
|
|
||||||
mock_client.put.return_value = mock_client.delete.return_value = Response(204)
|
|
||||||
|
|
||||||
result = await set_starred(mock_context, "owner", "repo", starred)
|
|
||||||
assert result == expected_message
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.asyncio
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
"status_code,error_message,expected_error",
|
|
||||||
[
|
|
||||||
(403, "Forbidden", "Error accessing.*: Forbidden"),
|
|
||||||
(404, "Not Found", "Error accessing.*: Resource not found"),
|
|
||||||
(500, "Internal Server Error", "Error accessing.*: Failed to process request"),
|
|
||||||
],
|
|
||||||
)
|
|
||||||
async def test_set_starred_errors(
|
|
||||||
mock_context, mock_client, status_code, error_message, expected_error
|
|
||||||
):
|
|
||||||
mock_client.put.return_value = mock_client.delete.return_value = Response(
|
|
||||||
status_code, json={"message": error_message}
|
|
||||||
)
|
|
||||||
|
|
||||||
with pytest.raises(ToolExecutionError, match=expected_error):
|
|
||||||
await set_starred(mock_context, "owner", "repo", True)
|
|
||||||
|
|
@ -1,4 +1,4 @@
|
||||||
from typing import Annotated
|
from typing import Annotated, Optional
|
||||||
|
|
||||||
import httpx
|
import httpx
|
||||||
|
|
||||||
|
|
@ -38,3 +38,56 @@ async def set_starred(
|
||||||
|
|
||||||
action = "starred" if starred else "unstarred"
|
action = "starred" if starred else "unstarred"
|
||||||
return f"Successfully {action} the repository {owner}/{name}"
|
return f"Successfully {action} the repository {owner}/{name}"
|
||||||
|
|
||||||
|
|
||||||
|
# Implements https://docs.github.com/en/rest/activity/starring?apiVersion=2022-11-28#list-stargazers
|
||||||
|
# Example `arcade chat` usage: "list the stargazers for the ArcadeAI/arcade-ai repo"
|
||||||
|
@tool(requires_auth=GitHub())
|
||||||
|
async def list_stargazers(
|
||||||
|
context: ToolContext,
|
||||||
|
owner: Annotated[str, "The owner of the repository"],
|
||||||
|
repo: Annotated[str, "The name of the repository"],
|
||||||
|
limit: Annotated[
|
||||||
|
Optional[int],
|
||||||
|
"The maximum number of stargazers to return. If not provided, all stargazers will be returned.",
|
||||||
|
] = None,
|
||||||
|
) -> Annotated[dict, "A dictionary containing the stargazers for the specified repository"]:
|
||||||
|
"""List the stargazers for a GitHub repository."""
|
||||||
|
url = get_url("repo_stargazers", owner=owner, repo=repo)
|
||||||
|
headers = get_github_json_headers(context.authorization.token)
|
||||||
|
|
||||||
|
per_page = min(limit, 100)
|
||||||
|
page = 1
|
||||||
|
stargazers = []
|
||||||
|
|
||||||
|
if limit is None:
|
||||||
|
limit = 2**64 - 1
|
||||||
|
|
||||||
|
async with httpx.AsyncClient() as client:
|
||||||
|
while len(stargazers) < limit:
|
||||||
|
response = await client.get(
|
||||||
|
url, headers=headers, params={"per_page": per_page, "page": page}
|
||||||
|
)
|
||||||
|
handle_github_response(response, url)
|
||||||
|
|
||||||
|
data = response.json()
|
||||||
|
if not data:
|
||||||
|
break
|
||||||
|
|
||||||
|
stargazers.extend([
|
||||||
|
{
|
||||||
|
"login": stargazer.get("login"),
|
||||||
|
"id": stargazer.get("id"),
|
||||||
|
"node_id": stargazer.get("node_id"),
|
||||||
|
"html_url": stargazer.get("html_url"),
|
||||||
|
}
|
||||||
|
for stargazer in data
|
||||||
|
])
|
||||||
|
|
||||||
|
if len(data) < per_page:
|
||||||
|
break
|
||||||
|
|
||||||
|
page += 1
|
||||||
|
|
||||||
|
stargazers = stargazers[:limit]
|
||||||
|
return {"number_of_stargazers": len(stargazers), "stargazers": stargazers}
|
||||||
|
|
|
||||||
|
|
@ -1,5 +1,5 @@
|
||||||
import arcade_github
|
import arcade_github
|
||||||
from arcade_github.tools.activity import set_starred
|
from arcade_github.tools.activity import list_stargazers, set_starred
|
||||||
|
|
||||||
from arcade.sdk import ToolCatalog
|
from arcade.sdk import ToolCatalog
|
||||||
from arcade.sdk.eval import (
|
from arcade.sdk.eval import (
|
||||||
|
|
@ -71,4 +71,44 @@ def github_activity_eval_suite() -> EvalSuite:
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
|
|
||||||
|
suite.add_case(
|
||||||
|
name="List stargazers for a repository",
|
||||||
|
user_message="List 42 stargazers for the ArcadeAI/arcade-ai repository.",
|
||||||
|
expected_tool_calls=[
|
||||||
|
(
|
||||||
|
list_stargazers,
|
||||||
|
{
|
||||||
|
"owner": "ArcadeAI",
|
||||||
|
"repo": "arcade-ai",
|
||||||
|
"limit": 42,
|
||||||
|
},
|
||||||
|
)
|
||||||
|
],
|
||||||
|
critics=[
|
||||||
|
BinaryCritic(critic_field="owner", weight=0.3),
|
||||||
|
BinaryCritic(critic_field="repo", weight=0.3),
|
||||||
|
BinaryCritic(critic_field="limit", weight=0.4),
|
||||||
|
],
|
||||||
|
)
|
||||||
|
|
||||||
|
suite.add_case(
|
||||||
|
name="List stargazers for a repository",
|
||||||
|
user_message="List all of the stargazers for the ArcadeAI/arcade-ai repo",
|
||||||
|
expected_tool_calls=[
|
||||||
|
(
|
||||||
|
list_stargazers,
|
||||||
|
{
|
||||||
|
"owner": "ArcadeAI",
|
||||||
|
"repo": "arcade-ai",
|
||||||
|
"limit": None,
|
||||||
|
},
|
||||||
|
)
|
||||||
|
],
|
||||||
|
critics=[
|
||||||
|
BinaryCritic(critic_field="owner", weight=0.3),
|
||||||
|
BinaryCritic(critic_field="repo", weight=0.3),
|
||||||
|
BinaryCritic(critic_field="limit", weight=0.4),
|
||||||
|
],
|
||||||
|
)
|
||||||
|
|
||||||
return suite
|
return suite
|
||||||
|
|
|
||||||
103
toolkits/github/tests/test_activity.py
Normal file
103
toolkits/github/tests/test_activity.py
Normal file
|
|
@ -0,0 +1,103 @@
|
||||||
|
from unittest.mock import AsyncMock, patch
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
from arcade_github.tools.activity import list_stargazers, set_starred
|
||||||
|
from httpx import Response
|
||||||
|
|
||||||
|
from arcade.sdk.errors import ToolExecutionError
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
def mock_context():
|
||||||
|
context = AsyncMock()
|
||||||
|
context.authorization.token = "mock_token" # noqa: S105
|
||||||
|
return context
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
def mock_client():
|
||||||
|
with patch("arcade_github.tools.activity.httpx.AsyncClient") as client:
|
||||||
|
yield client.return_value.__aenter__.return_value
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"starred,expected_message",
|
||||||
|
[
|
||||||
|
(True, "Successfully starred the repository owner/repo"),
|
||||||
|
(False, "Successfully unstarred the repository owner/repo"),
|
||||||
|
],
|
||||||
|
)
|
||||||
|
async def test_set_starred_success(mock_context, mock_client, starred, expected_message):
|
||||||
|
mock_client.put.return_value = mock_client.delete.return_value = Response(204)
|
||||||
|
|
||||||
|
result = await set_starred(mock_context, "owner", "repo", starred)
|
||||||
|
assert result == expected_message
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"status_code,error_message,expected_error",
|
||||||
|
[
|
||||||
|
(403, "Forbidden", "Error accessing.*: Forbidden"),
|
||||||
|
(404, "Not Found", "Error accessing.*: Resource not found"),
|
||||||
|
(500, "Internal Server Error", "Error accessing.*: Failed to process request"),
|
||||||
|
],
|
||||||
|
)
|
||||||
|
async def test_set_starred_errors(
|
||||||
|
mock_context, mock_client, status_code, error_message, expected_error
|
||||||
|
):
|
||||||
|
mock_client.put.return_value = mock_client.delete.return_value = Response(
|
||||||
|
status_code, json={"message": error_message}
|
||||||
|
)
|
||||||
|
|
||||||
|
with pytest.raises(ToolExecutionError, match=expected_error):
|
||||||
|
await set_starred(mock_context, "owner", "repo", True)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_list_stargazers_success(mock_context, mock_client):
|
||||||
|
mock_response_data = [
|
||||||
|
{
|
||||||
|
"login": "user1",
|
||||||
|
"id": 1,
|
||||||
|
"node_id": "MDQ6VXNlcjE=",
|
||||||
|
"html_url": "https://github.com/user1",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"login": "user2",
|
||||||
|
"id": 2,
|
||||||
|
"node_id": "MDQ6VXNlcjI=",
|
||||||
|
"html_url": "https://github.com/user2",
|
||||||
|
},
|
||||||
|
]
|
||||||
|
mock_client.get.return_value = Response(200, json=mock_response_data)
|
||||||
|
|
||||||
|
result = await list_stargazers(mock_context, "owner", "repo", limit=2)
|
||||||
|
assert result == {"number_of_stargazers": 2, "stargazers": mock_response_data}
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_list_stargazers_empty(mock_context, mock_client):
|
||||||
|
mock_client.get.return_value = Response(200, json=[])
|
||||||
|
|
||||||
|
result = await list_stargazers(mock_context, "owner", "repo")
|
||||||
|
assert result == {"number_of_stargazers": 0, "stargazers": []}
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"status_code,error_message,expected_error",
|
||||||
|
[
|
||||||
|
(403, "Forbidden", "Error accessing.*: Forbidden"),
|
||||||
|
(404, "Not Found", "Error accessing.*: Resource not found"),
|
||||||
|
(500, "Internal Server Error", "Error accessing.*: Failed to process request"),
|
||||||
|
],
|
||||||
|
)
|
||||||
|
async def test_list_stargazers_errors(
|
||||||
|
mock_context, mock_client, status_code, error_message, expected_error
|
||||||
|
):
|
||||||
|
mock_client.get.return_value = Response(status_code, json={"message": error_message})
|
||||||
|
|
||||||
|
with pytest.raises(ToolExecutionError, match=expected_error):
|
||||||
|
await list_stargazers(mock_context, "owner", "repo")
|
||||||
Loading…
Reference in a new issue