arcade-mcp/toolkits/slack/arcade_slack/tools/users.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

87 lines
2.9 KiB
Python

from typing import Annotated, Any, cast
from arcade_tdk import ToolContext, tool
from arcade_tdk.auth import Slack
from arcade_tdk.errors import RetryableToolError
from slack_sdk.errors import SlackApiError
from slack_sdk.web.async_client import AsyncWebClient
from arcade_slack.constants import MAX_PAGINATION_TIMEOUT_SECONDS
from arcade_slack.models import SlackPaginationNextCursor, SlackUser
from arcade_slack.utils import (
async_paginate,
extract_basic_user_info,
is_user_a_bot,
is_user_deleted,
)
@tool(
requires_auth=Slack(
scopes=["users:read", "users:read.email"],
)
)
async def get_user_info_by_id(
context: ToolContext,
user_id: Annotated[str, "The ID of the user to get"],
) -> Annotated[dict[str, Any], "The user's information"]:
"""Get the information of a user in Slack."""
token = (
context.authorization.token if context.authorization and context.authorization.token else ""
)
slackClient = AsyncWebClient(token=token)
try:
response = await slackClient.users_info(user=user_id)
except SlackApiError as e:
if e.response.get("error") == "user_not_found":
users = await list_users(context)
available_users = ", ".join(f"{user['id']} ({user['name']})" for user in users["users"])
raise RetryableToolError(
"User not found",
developer_message=f"User with ID '{user_id}' not found.",
additional_prompt_content=f"Available users: {available_users}",
retry_after_ms=500,
)
user_dict_raw: dict[str, Any] = response.get("user", {}) or {}
user_dict = cast(SlackUser, user_dict_raw)
user = SlackUser(**user_dict)
return dict(**extract_basic_user_info(user))
@tool(
requires_auth=Slack(
scopes=["users:read", "users:read.email"],
)
)
async def list_users(
context: ToolContext,
exclude_bots: Annotated[bool | None, "Whether to exclude bots from the results"] = True,
limit: Annotated[int | None, "The maximum number of users to return."] = None,
next_cursor: Annotated[str | None, "The next cursor token to use for pagination."] = None,
) -> Annotated[dict, "The users' info"]:
"""List all users in the authenticated user's Slack team."""
token = (
context.authorization.token if context.authorization and context.authorization.token else ""
)
slackClient = AsyncWebClient(token=token)
users, next_cursor = await async_paginate(
func=slackClient.users_list,
response_key="members",
limit=limit,
next_cursor=cast(SlackPaginationNextCursor, next_cursor),
max_pagination_timeout_seconds=MAX_PAGINATION_TIMEOUT_SECONDS,
)
users = [
extract_basic_user_info(user)
for user in users
if not is_user_deleted(user) and (not exclude_bots or not is_user_a_bot(user))
]
return {"users": users, "next_cursor": next_cursor}