arcade-slack 1.0.0__py3-none-any.whl → 1.1.0__py3-none-any.whl

arcade_slack/tools/system_context.py

@@ -0,0 +1,36 @@
+from typing import Annotated, Any
+
+from arcade_tdk import ToolContext, tool
+from arcade_tdk.auth import Slack
+from slack_sdk.web.async_client import AsyncWebClient
+
+from arcade_slack.who_am_i_util import build_who_am_i_response
+
+
+@tool(
+    requires_auth=Slack(
+        scopes=[
+            "users:read",
+            "users:read.email",
+        ]
+    )
+)
+async def who_am_i(
+    context: ToolContext,
+) -> Annotated[
+    dict[str, Any],
+    "Get comprehensive user profile and Slack information.",
+]:
+    """
+    Get comprehensive user profile and Slack information.
+
+    This tool provides detailed information about the authenticated user including
+    their name, email, profile picture, and other important profile details from
+    Slack services.
+    """
+
+    auth_token = context.get_auth_token_or_empty()
+    slack_client = AsyncWebClient(token=auth_token)
+    user_info = await build_who_am_i_response(context, slack_client)
+
+    return dict(user_info)

arcade_slack/who_am_i_util.py

@@ -0,0 +1,122 @@
+from typing import Any, TypedDict, cast
+
+from slack_sdk.web.async_client import AsyncWebClient
+
+
+class WhoAmIResponse(TypedDict, total=False):
+    user_id: str
+    username: str
+    display_name: str
+    real_name: str
+    email: str
+    profile_picture_url: str
+    title: str
+    phone: str
+    first_name: str
+    last_name: str
+    pronouns: str
+    status_text: str
+    status_emoji: str
+    slack_access: bool
+
+
+async def build_who_am_i_response(context: Any, slack_client: AsyncWebClient) -> WhoAmIResponse:
+    """Build complete who_am_i response from Slack APIs."""
+    auth_token = context.get_auth_token_or_empty()
+
+    auth_info = await _get_auth_info(slack_client)
+
+    user_profile = await _get_user_profile(auth_token, auth_info["user_id"])
+
+    user_info: dict[str, Any] = {}
+    user_info.update(_extract_auth_info(auth_info))
+    user_info.update(_extract_user_profile(user_profile))
+
+    return cast(WhoAmIResponse, user_info)
+
+
+async def _get_auth_info(slack_client: AsyncWebClient) -> dict[str, Any]:
+    """Get authentication information including user_id and team info."""
+    response = await slack_client.auth_test()
+    return cast(dict[str, Any], response.data)
+
+
+async def _get_user_profile(auth_token: str, user_id: str) -> dict[str, Any]:
+    """Get detailed user profile information."""
+    slack_client = AsyncWebClient(token=auth_token)
+    response = await slack_client.users_info(user=user_id)
+    response_dict = cast(dict[str, Any], response)
+    if not response_dict.get("ok"):
+        error = response_dict.get("error", "Unknown error")
+        raise RuntimeError(f"Failed to get user info: {error}")
+    return cast(dict[str, Any], response_dict["user"])
+
+
+def _extract_auth_info(auth_info: dict[str, Any]) -> dict[str, Any]:
+    """Extract authentication information."""
+    extracted = {}
+
+    if auth_info.get("user_id"):
+        extracted["user_id"] = auth_info["user_id"]
+    if auth_info.get("user"):
+        extracted["username"] = auth_info["user"]
+
+    extracted["slack_access"] = bool(auth_info.get("ok"))
+
+    return extracted
+
+
+def _extract_user_profile(user_profile: dict[str, Any]) -> dict[str, Any]:
+    """Extract user profile information."""
+    extracted = {}
+
+    if user_profile.get("real_name"):
+        extracted["real_name"] = user_profile["real_name"]
+    if user_profile.get("name"):
+        extracted["username"] = user_profile["name"]
+
+    profile = user_profile.get("profile", {})
+    extracted.update(_extract_profile_fields(profile))
+    extracted.update(_extract_profile_picture(profile))
+
+    return extracted
+
+
+def _extract_profile_fields(profile: dict[str, Any]) -> dict[str, Any]:
+    """Extract profile fields from user profile."""
+    extracted = {}
+
+    fields = [
+        ("display_name", "display_name"),
+        ("real_name", "real_name"),
+        ("email", "email"),
+        ("title", "title"),
+        ("phone", "phone"),
+        ("first_name", "first_name"),
+        ("last_name", "last_name"),
+        ("pronouns", "pronouns"),
+        ("status_text", "status_text"),
+        ("status_emoji", "status_emoji"),
+    ]
+
+    for profile_key, extracted_key in fields:
+        if profile.get(profile_key):
+            extracted[extracted_key] = profile[profile_key]
+
+    return extracted
+
+
+def _extract_profile_picture(profile: dict[str, Any]) -> dict[str, Any]:
+    """Extract profile picture URL from user profile."""
+    for size in [
+        "image_1024",
+        "image_512",
+        "image_192",
+        "image_72",
+        "image_48",
+        "image_32",
+        "image_24",
+    ]:
+        if profile.get(size):
+            return {"profile_picture_url": profile[size]}
+    return {}

{arcade_slack-1.0.0.dist-info → arcade_slack-1.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arcade_slack
-Version: 1.0.0
+Version: 1.1.0
 Summary: Arcade.dev LLM tools for Slack
 Author-email: Arcade <dev@arcade.dev>
 License: Proprietary - Arcade Software License Agreement v1.0

{arcade_slack-1.0.0.dist-info → arcade_slack-1.1.0.dist-info}/RECORD

@@ -8,10 +8,12 @@ arcade_slack/message_retrieval.py,sha256=XdzWLJdKtc5DL7H_qQBei6qJ-rz2uGs-Q0HfYqx
 arcade_slack/models.py,sha256=-zlkAxaDNEcIkQqePYTZXAqcRX4gO5jocJIixeWpK7A,12213
 arcade_slack/user_retrieval.py,sha256=cdfzBbm2fsVNCztt2XbIV2UKuCdHaotfFBdo10EpBcE,6774
 arcade_slack/utils.py,sha256=z6G7GCDXn_7_TshQRx-rnhgmQG8eCQSND5It9_uvhmE,21776
+arcade_slack/who_am_i_util.py,sha256=Nf6EdRiURva5jMhVx1lFOoKQ-2SuMwjXYbPCmtj6Zcw,3670
 arcade_slack/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 arcade_slack/tools/chat.py,sha256=KWDalEiOhay4Z71EB97dVllm3Uo7sXxjXFfmcpvuWYE,38125
+arcade_slack/tools/system_context.py,sha256=l2gZh_WhrIVL2LzttGAz5yiv0Cl6q_6AFyBBUBskz3E,961
 arcade_slack/tools/users.py,sha256=Kw3MWNOwuUY17X5HGf3JMAH_mfloLRdRzXSBC7Zo7ug,4233
-arcade_slack-1.0.0.dist-info/METADATA,sha256=zQpW_DaNQ_t9YEFDhXdt2C6WeG9GC686DbJEjua7meQ,1015
-arcade_slack-1.0.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-arcade_slack-1.0.0.dist-info/licenses/LICENSE,sha256=ixeE7aL9b2B-_ZYHTY1vQcJB4NufKeo-LWwKNObGDN0,1960
-arcade_slack-1.0.0.dist-info/RECORD,,
+arcade_slack-1.1.0.dist-info/METADATA,sha256=7xIdWupyKsoxoT8kgUBoJh-m72gSIa8NdH8IP8OTIhE,1015
+arcade_slack-1.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+arcade_slack-1.1.0.dist-info/licenses/LICENSE,sha256=ixeE7aL9b2B-_ZYHTY1vQcJB4NufKeo-LWwKNObGDN0,1960
+arcade_slack-1.1.0.dist-info/RECORD,,