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>
73 lines
2.6 KiB
Python
73 lines
2.6 KiB
Python
from unittest.mock import AsyncMock, patch
|
|
|
|
import pytest
|
|
from arcade_tdk.errors import ToolExecutionError
|
|
from httpx import Response
|
|
|
|
from arcade_github.tools.models import RepoType
|
|
from arcade_github.tools.repositories import (
|
|
count_stargazers,
|
|
get_repository,
|
|
list_org_repositories,
|
|
list_repository_activities,
|
|
list_review_comments_in_a_repository,
|
|
)
|
|
|
|
|
|
@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.repositories.httpx.AsyncClient") as client:
|
|
yield client.return_value.__aenter__.return_value
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize(
|
|
"status_code,error_message,expected_error",
|
|
[
|
|
(422, "Validation Failed", "Error accessing.*: Validation failed"),
|
|
(301, "Moved Permanently", "Error accessing.*: Moved permanently"),
|
|
(404, "Not Found", "Error accessing.*: Resource not found"),
|
|
(503, "Service Unavailable", "Error accessing.*: Service unavailable"),
|
|
(410, "Gone", "Error accessing.*: Gone"),
|
|
],
|
|
)
|
|
async def test_error_responses(
|
|
mock_context, mock_client, status_code, error_message, expected_error
|
|
):
|
|
mock_client.get.return_value = Response(status_code, json={"message": error_message})
|
|
mock_client.post.return_value = Response(status_code, json={"message": error_message})
|
|
|
|
with pytest.raises(ToolExecutionError, match=expected_error):
|
|
if status_code == 422:
|
|
await list_org_repositories(mock_context, "org", repo_type=RepoType.ALL)
|
|
elif status_code == 301:
|
|
await count_stargazers(mock_context, "owner", "repo")
|
|
elif status_code == 404:
|
|
await list_org_repositories(mock_context, "non_existent_org")
|
|
elif status_code == 503:
|
|
await get_repository(mock_context, "owner", "repo")
|
|
elif status_code == 410:
|
|
await list_review_comments_in_a_repository(mock_context, "owner", "repo")
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_list_repository_activities_invalid_cursor(mock_context, mock_client):
|
|
mock_client.get.return_value = Response(422, json={"message": "Validation Failed"})
|
|
|
|
with pytest.raises(ToolExecutionError, match="Error accessing.*: Validation failed"):
|
|
await list_repository_activities(mock_context, "owner", "repo", before="invalid_cursor")
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_count_stargazers_success(mock_context, mock_client):
|
|
mock_client.get.return_value = Response(200, json={"stargazers_count": 42})
|
|
|
|
result = await count_stargazers(mock_context, "owner", "repo")
|
|
assert result == 42
|