arcade-mcp/toolkits/github/arcade_github/tools/activity.py
Sam Partee b6b4cd0a4c
🏗️ Restructure: Multi-Package Architecture + uv Migration (#412)
### Overview
Major restructuring from monolithic `arcade-ai` package to modular
library architecture with standardized uv-based dependency management.

![arcade-ai Monorepo
(2)](https://github.com/user-attachments/assets/25f102b0-bb87-4a04-9701-d227d05664b1)

### 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>
2025-06-11 16:48:17 -07:00

98 lines
3.6 KiB
Python

from typing import Annotated
import httpx
from arcade_tdk import ToolContext, tool
from arcade_tdk.auth import GitHub
from arcade_github.tools.utils import get_github_json_headers, get_url, handle_github_response
# Implements https://docs.github.com/en/rest/activity/starring?apiVersion=2022-11-28#star-a-repository-for-the-authenticated-user and https://docs.github.com/en/rest/activity/starring?apiVersion=2022-11-28#unstar-a-repository-for-the-authenticated-user # noqa: E501
# Example `arcade chat` usage: "star the vscode repo owned by microsoft"
@tool(requires_auth=GitHub())
async def set_starred(
context: ToolContext,
owner: Annotated[str, "The owner of the repository"],
name: Annotated[str, "The name of the repository"],
starred: Annotated[bool, "Whether to star the repository or not"] = True,
) -> Annotated[
str, "A message indicating whether the repository was successfully starred or unstarred"
]:
"""
Star or un-star a GitHub repository.
For example, to star microsoft/vscode, you would use:
```
set_starred(owner="microsoft", name="vscode", starred=True)
```
"""
url = get_url("user_starred", owner=owner, repo=name)
headers = get_github_json_headers(
context.authorization.token if context.authorization and context.authorization.token else ""
)
async with httpx.AsyncClient() as client:
if starred:
response = await client.put(url, headers=headers)
else:
response = await client.delete(url, headers=headers)
handle_github_response(response, url)
action = "starred" if starred else "unstarred"
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[
int | None,
"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 if context.authorization and context.authorization.token else ""
)
if limit is None:
limit = 2**64 - 1
per_page = min(limit, 100)
page = 1
stargazers: list[dict] = []
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}