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

arcade_slack/conversation_retrieval.py

@@ -1,8 +1,6 @@
 import json
-from typing import cast
 
-from arcade_tdk.errors import RetryableToolError, ToolExecutionError
-from slack_sdk.errors import SlackApiError
+from arcade_tdk.errors import RetryableToolError
 from slack_sdk.web.async_client import AsyncWebClient
 
 from arcade_slack.models import (
@@ -20,21 +18,13 @@ async def get_conversation_by_id(
     conversation_id: str,
 ) -> dict:
     """Get metadata of a conversation in Slack by the conversation_id."""
-    try:
-        slack_client = AsyncWebClient(token=auth_token)
-        response = await slack_client.conversations_info(
-            channel=conversation_id,
-            include_locale=True,
-            include_num_members=True,
-        )
-        return dict(**extract_conversation_metadata(response["channel"]))
-
-    except SlackApiError as e:
-        slack_error = cast(str, e.response.get("error", ""))
-        if "not_found" in slack_error.lower():
-            message = f"Conversation with ID '{conversation_id}' not found."
-            raise ToolExecutionError(message=message, developer_message=message)
-        raise
+    slack_client = AsyncWebClient(token=auth_token)
+    response = await slack_client.conversations_info(
+        channel=conversation_id,
+        include_locale=True,
+        include_num_members=True,
+    )
+    return dict(**extract_conversation_metadata(response["channel"]))
 
 
 async def get_channel_by_name(

arcade_slack/models.py

@@ -5,7 +5,6 @@ from contextlib import suppress
 from enum import Enum
 from typing import Any, Literal, TypedDict
 
-from arcade_tdk.errors import ToolExecutionError
 from slack_sdk.errors import SlackApiError
 
 from arcade_slack.custom_types import (
@@ -364,7 +363,4 @@ class GetUserByEmailCaller(AbstractConcurrencySafeCoroutineCaller):
                 if e.response.get("error") in ["user_not_found", "users_not_found"]:
                     return {"user": None, "email": self.email}
                 else:
-                    raise ToolExecutionError(
-                        message="Error getting user by email",
-                        developer_message=f"Error getting user by email: {e.response.get('error')}",
-                    )
+                    raise

arcade_slack/tools/chat.py

@@ -3,7 +3,6 @@ from typing import Annotated, cast
 from arcade_tdk import ToolContext, tool
 from arcade_tdk.auth import Slack
 from arcade_tdk.errors import ToolExecutionError
-from slack_sdk.errors import SlackApiError
 from slack_sdk.web.async_client import AsyncWebClient
 
 from arcade_slack.constants import MAX_PAGINATION_SIZE_LIMIT
@@ -349,12 +348,7 @@ async def get_conversation_metadata(
 
     slack_client = AsyncWebClient(token=auth_token)
 
-    try:
-        current_user = await slack_client.auth_test()
-    except SlackApiError as e:
-        message = "Failed to get currently authenticated user's info."
-        developer_message = f"{message} Slack error: '{e.response.get('error', 'unknown_error')}'"
-        raise ToolExecutionError(message, developer_message)
+    current_user = await slack_client.auth_test()
 
     if current_user["user_id"] not in user_ids_list:
         user_ids_list.append(current_user["user_id"])
@@ -367,16 +361,8 @@ async def get_conversation_metadata(
         )
         user_ids_list.extend([user["id"] for user in other_users])
 
-    try:
-        response = await slack_client.conversations_open(users=user_ids_list, return_im=True)
-        return dict(**extract_conversation_metadata(response["channel"]))
-    except SlackApiError as e:
-        message = "Failed to retrieve conversation metadata."
-        slack_error = e.response.get("error", "unknown_error")
-        raise ToolExecutionError(
-            message=message,
-            developer_message=f"{message} Slack error: '{slack_error}'",
-        )
+    response = await slack_client.conversations_open(users=user_ids_list, return_im=True)
+    return dict(**extract_conversation_metadata(response["channel"]))
 
 
 @tool(

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/user_retrieval.py

@@ -114,12 +114,7 @@ async def get_single_user_by_id(auth_token: str, user_id: str) -> dict[str, Any]
         if "not_found" in e.response.get("error", ""):
             return None
         else:
-            message = f"There was an error getting the user with ID {user_id}."
-            slack_error_message = e.response.get("error", "Unknown Slack API error")
-            raise ToolExecutionError(
-                message=message,
-                developer_message=f"{message}: {slack_error_message}",
-            ) from e
+            raise
 
 
 async def get_users_by_username(

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.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arcade_slack
-Version: 1.0.0
+Version: 1.2.0
 Summary: Arcade.dev LLM tools for Slack
 Author-email: Arcade <dev@arcade.dev>
 License: Proprietary - Arcade Software License Agreement v1.0
@@ -8,12 +8,12 @@ License-File: LICENSE
 Requires-Python: >=3.10
 Requires-Dist: aiodns<2.0.0,>=1.0
 Requires-Dist: aiohttp<4.0.0,>=3.7.3
-Requires-Dist: arcade-tdk<3.0.0,>=2.0.0
+Requires-Dist: arcade-tdk<3.0.0,>=2.5.0
 Requires-Dist: slack-sdk<4.0.0,>=3.31.0
 Requires-Dist: typing; python_version < '3.7'
 Provides-Extra: dev
-Requires-Dist: arcade-ai[evals]<3.0.0,>=2.0.0; extra == 'dev'
-Requires-Dist: arcade-serve<3.0.0,>=2.0.0; extra == 'dev'
+Requires-Dist: arcade-ai[evals]<3.0.0,>=2.2.1; extra == 'dev'
+Requires-Dist: arcade-serve<3.0.0,>=2.1.0; extra == 'dev'
 Requires-Dist: mypy<1.6.0,>=1.5.1; extra == 'dev'
 Requires-Dist: pre-commit<3.5.0,>=3.4.0; extra == 'dev'
 Requires-Dist: pytest-asyncio<0.25.0,>=0.24.0; extra == 'dev'

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

@@ -1,17 +1,19 @@
 arcade_slack/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 arcade_slack/constants.py,sha256=ANAZOmNSqrbZHDqEvQfau7_91if9EeKV-G5D0VBB9vo,524
-arcade_slack/conversation_retrieval.py,sha256=jsd64-IDpgTpuq-4Hbg1317VatxKnIy8edWhHB6bN7E,2338
+arcade_slack/conversation_retrieval.py,sha256=VKc0Cc0k4UZ0OON_1l-HxWRzk_nSCzgoP5lgGyUxLFs,1902
 arcade_slack/critics.py,sha256=Sx0nOmvym4LFNX1gZ34ndHkAKAB--X1zRPePcbiVzDo,1209
 arcade_slack/custom_types.py,sha256=-23JnfXjMQ-H8CyRL3CEv9opPMySox1KINH5rrPIOt0,928
 arcade_slack/exceptions.py,sha256=YQ4CTa1LbR-G7sjnQM8LB7LruxZDqPzvo-cptFYW7E8,385
 arcade_slack/message_retrieval.py,sha256=XdzWLJdKtc5DL7H_qQBei6qJ-rz2uGs-Q0HfYqx7EG4,2496
-arcade_slack/models.py,sha256=-zlkAxaDNEcIkQqePYTZXAqcRX4gO5jocJIixeWpK7A,12213
-arcade_slack/user_retrieval.py,sha256=cdfzBbm2fsVNCztt2XbIV2UKuCdHaotfFBdo10EpBcE,6774
+arcade_slack/models.py,sha256=R0dze0YMQW_v_GqarA2F3ZMR3ueN_ant9K855N9uqlE,11958
+arcade_slack/user_retrieval.py,sha256=F-ChLGHYFFyEnHVZnoSeGq6Gh-NZGda_pYn2Rc9oMv4,6464
 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/chat.py,sha256=-HvXuMTsW095DhtkhMINXmkdazbBdTZvUhaSx0BEIDw,37488
+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.2.0.dist-info/METADATA,sha256=RRPv1QPYgDO-Cn-b9udf55g-qtM3_3qOhP0kJY3Xssw,1015
+arcade_slack-1.2.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+arcade_slack-1.2.0.dist-info/licenses/LICENSE,sha256=ixeE7aL9b2B-_ZYHTY1vQcJB4NufKeo-LWwKNObGDN0,1960
+arcade_slack-1.2.0.dist-info/RECORD,,