arcade-x 1.1.1__py3-none-any.whl → 1.3.0__py3-none-any.whl

arcade_x/tools/tweets.py

@@ -26,7 +26,24 @@ async def _post_tweet(context: ToolContext, payload: dict) -> str:
     async with httpx.AsyncClient() as client:
         response = await client.post(TWEETS_URL, headers=headers, json=payload, timeout=10)
         response.raise_for_status()
-    tweet_id = response.json()["data"]["id"]
+
+    response_json = response.json()
+
+    # Check if data exists in response
+    if "data" not in response_json or response_json.get("data") is None:
+        return (
+            "The post was successfully created, but the X API returned an unexpected response. "
+            f"The X API returned:\n {response_json}"
+        )
+
+    tweet_data = response_json["data"]
+    if "id" not in tweet_data:
+        return (
+            "The post was successfully created, but the X API returned an unexpected response. "
+            f"The X API returned:\n {response_json}"
+        )
+
+    tweet_id = tweet_data["id"]
     return f"Tweet with id {tweet_id} posted successfully. URL: {get_tweet_url(tweet_id)}"
 
 
@@ -40,8 +57,8 @@ async def post_tweet(
     tweet_text: Annotated[str, "The text content of the tweet you want to post"],
     quote_tweet_id: Annotated[
         str | None,
-        "The ID of the tweet you want to quote."
-        " It must be a valid integer as a string. Optional.",
+        "The ID of the tweet you want to quote. "
+        "It must be a valid integer as a string. Default is None.",
     ] = None,
 ) -> Annotated[str, "Success string and the URL of the tweet"]:
     """Post a tweet to X (Twitter).

arcade_x/tools/user_context.py

@@ -0,0 +1,76 @@
+"""User context tools for X (Twitter) toolkit."""
+
+from typing import Annotated
+
+import httpx
+from arcade_tdk import ToolContext, tool
+from arcade_tdk.auth import X
+from arcade_tdk.errors import ToolExecutionError
+
+from arcade_x.tools.utils import (
+    expand_urls_in_user_description,
+    expand_urls_in_user_url,
+    get_headers_with_token,
+)
+
+
+@tool(requires_auth=X(scopes=["users.read", "tweet.read"]))
+async def who_am_i(
+    context: ToolContext,
+) -> Annotated[dict, "Authenticated user's profile information"]:
+    """
+    Get information about the authenticated X (Twitter) user.
+
+    Returns the current user's profile including their username, name, description,
+    follower counts, and other account information.
+    """
+    headers = get_headers_with_token(context)
+
+    # User fields to retrieve
+    user_fields = ",".join([
+        "created_at",
+        "description",
+        "id",
+        "location",
+        "most_recent_tweet_id",
+        "name",
+        "pinned_tweet_id",
+        "profile_image_url",
+        "protected",
+        "public_metrics",
+        "url",
+        "username",
+        "verified",
+        "verified_type",
+        "withheld",
+        "entities",
+    ])
+
+    # Use the /2/users/me endpoint to get authenticated user's information
+    # API Reference: https://developer.x.com/en/docs/x-api/users/lookup/api-reference/get-users-me
+    # Returns user object with fields: id, name, username, and optional expansions
+    url = f"https://api.x.com/2/users/me?user.fields={user_fields}"
+
+    try:
+        async with httpx.AsyncClient() as client:
+            response = await client.get(url, headers=headers, timeout=10)
+            response.raise_for_status()
+    except httpx.HTTPStatusError as e:
+        raise ToolExecutionError(
+            f"X API returned an error: {e.response.status_code} {e.response.reason_phrase}"
+        ) from e
+    except httpx.RequestError as e:
+        raise ToolExecutionError(f"Failed to connect to X API: {e}") from e
+
+    # Parse the response JSON
+    response_data = response.json()
+    user_data = response_data.get("data")
+
+    if not user_data:
+        raise ToolExecutionError(f"X API response missing 'data' field. Response: {response_data}")
+
+    # Expand URLs in the user description and profile URL
+    user_data = expand_urls_in_user_description(user_data, delete_entities=False)
+    user_data = expand_urls_in_user_url(user_data, delete_entities=True)
+
+    return {"data": user_data}

arcade_x/tools/users.py

@@ -3,7 +3,6 @@ from typing import Annotated
 import httpx
 from arcade_tdk import ToolContext, tool
 from arcade_tdk.auth import X
-from arcade_tdk.errors import RetryableToolError
 
 from arcade_x.tools.utils import (
     expand_urls_in_user_description,
@@ -46,18 +45,22 @@ async def lookup_single_user_by_username(
 
     async with httpx.AsyncClient() as client:
         response = await client.get(url, headers=headers, timeout=10)
-        if response.status_code == 404:
-            # User not found
-            raise RetryableToolError(
-                "User not found",
-                developer_message=f"User with username '{username}' not found.",
-                additional_prompt_content="Please check the username and try again.",
-                retry_after_ms=500,  # Play nice with X API rate limits
-            )
         response.raise_for_status()
+
     # Parse the response JSON
-    user_data = response.json()["data"]
+    response_json = response.json()
+
+    # Check if data exists in response
+    if response_json.get("data") is None:
+        return {
+            "data": None,
+            "message": (
+                f"No user found with username '{username}'. "
+                "The account may not exist or may have been suspended."
+            ),
+        }
 
+    user_data = response_json["data"]
     user_data = expand_urls_in_user_description(user_data, delete_entities=False)
     user_data = expand_urls_in_user_url(user_data, delete_entities=True)
 

arcade_x/tools/utils.py

@@ -35,14 +35,20 @@ def parse_search_recent_tweets_response(response_data: dict[str, Any]) -> dict[s
 
     # Add 'tweet_url' to each tweet
     for tweet in response_data["data"]:
+        # Skip tweets without an id
+        if "id" not in tweet:
+            continue
         tweet["tweet_url"] = get_tweet_url(tweet["id"])
 
     # Add 'author_username' and 'author_name' to each tweet
     for tweet_data, user_data in zip(
         response_data["data"], response_data["includes"]["users"], strict=False
     ):
-        tweet_data["author_username"] = user_data["username"]
-        tweet_data["author_name"] = user_data["name"]
+        # Skip if user data is missing required fields
+        if "username" in user_data:
+            tweet_data["author_username"] = user_data["username"]
+        if "name" in user_data:
+            tweet_data["author_name"] = user_data["name"]
 
     return response_data
 
@@ -101,6 +107,9 @@ def expand_urls_in_user_description(user_data: dict, delete_entities: bool = Tru
     description_urls = new_user_data.get("entities", {}).get("description", {}).get("urls", [])
     description = new_user_data.get("description", "")
     for url_info in description_urls:
+        # Skip URL entities that don't have both url and expanded_url
+        if "url" not in url_info or "expanded_url" not in url_info:
+            continue
         t_co_link = url_info["url"]
         expanded_url = url_info["expanded_url"]
         description = description.replace(t_co_link, expanded_url)
@@ -119,6 +128,9 @@ def expand_urls_in_user_url(user_data: dict, delete_entities: bool = True) -> di
     url_urls = new_user_data.get("entities", {}).get("url", {}).get("urls", [])
     url = new_user_data.get("url", "")
     for url_info in url_urls:
+        # Skip URL entities that don't have both url and expanded_url
+        if "url" not in url_info or "expanded_url" not in url_info:
+            continue
         t_co_link = url_info["url"]
         expanded_url = url_info["expanded_url"]
         url = url.replace(t_co_link, expanded_url)

{arcade_x-1.1.1.dist-info → arcade_x-1.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arcade_x
-Version: 1.1.1
+Version: 1.3.0
 Summary: Arcade.dev LLM tools for X (Twitter)
 Author-email: Arcade <dev@arcade.dev>
 License: Proprietary - Arcade Software License Agreement v1.0

arcade_x-1.3.0.dist-info/RECORD

@@ -0,0 +1,11 @@
+arcade_x/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+arcade_x/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+arcade_x/tools/constants.py,sha256=d-OJK5Qx05JRUcpK5G1DcYB91wT37hFSzGmIYfwlEtA,42
+arcade_x/tools/tweets.py,sha256=KD93AY9EteE22txQGeVTK6N9_saJNZvBxJ8U1FOFLEQ,9986
+arcade_x/tools/user_context.py,sha256=nKlsM2LGdoIPl8FQ7FKqd6hkbPHVSJUiZdShZ7ZCsIk,2470
+arcade_x/tools/users.py,sha256=YObl3_3aBZpW-GqdSrR2Gmw3BHzdq075GWSxDd1Ery0,2033
+arcade_x/tools/utils.py,sha256=2s3hxpXplxIQs4rO1IJdO-1KbUj-Tm3f9P65n4drw1s,6259
+arcade_x-1.3.0.dist-info/METADATA,sha256=3dQr1nSmukSCVpxSj9xpqU_JeqxEKtuOqrsDI3Qi6l0,897
+arcade_x-1.3.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+arcade_x-1.3.0.dist-info/licenses/LICENSE,sha256=ixeE7aL9b2B-_ZYHTY1vQcJB4NufKeo-LWwKNObGDN0,1960
+arcade_x-1.3.0.dist-info/RECORD,,

{arcade_x-1.1.1.dist-info → arcade_x-1.3.0.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.27.0
+Generator: hatchling 1.28.0
 Root-Is-Purelib: true
 Tag: py3-none-any

arcade_x-1.1.1.dist-info/RECORD

@@ -1,10 +0,0 @@
-arcade_x/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-arcade_x/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-arcade_x/tools/constants.py,sha256=d-OJK5Qx05JRUcpK5G1DcYB91wT37hFSzGmIYfwlEtA,42
-arcade_x/tools/tweets.py,sha256=xfg31_Jh7sUuBfFjRIY_TwbUFCcYbm_xyUE_s35HB-8,9416
-arcade_x/tools/users.py,sha256=gopYKTBOwbZAeVdL_T3vwkepbpv1bdPbqcU_2oPC3SY,2132
-arcade_x/tools/utils.py,sha256=lR_shws08LWQD-4XU9r3xYmasZ2j2i-cUaCCDnuN8UE,5723
-arcade_x-1.1.1.dist-info/METADATA,sha256=Sy3KedtCp4On90A6gekRMTaOoeSp3G7wvLXxj9UlR4Y,897
-arcade_x-1.1.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-arcade_x-1.1.1.dist-info/licenses/LICENSE,sha256=ixeE7aL9b2B-_ZYHTY1vQcJB4NufKeo-LWwKNObGDN0,1960
-arcade_x-1.1.1.dist-info/RECORD,,