arcade-x 1.2.0__py3-none-any.whl → 1.3.1__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).
@@ -134,7 +151,7 @@ async def search_recent_tweets_by_username(
         "next_token": next_token,
         "expansions": "author_id",
         "user.fields": "id,name,username,entities",
-        "tweet.fields": "entities,note_tweet",
+        "tweet.fields": "entities,note_tweet,public_metrics",
     }
     params = expand_attached_media(remove_none_values(params))
 
@@ -204,7 +221,7 @@ async def search_recent_tweets_by_keywords(
         "next_token": next_token,
         "expansions": "author_id",
         "user.fields": "id,name,username,entities",
-        "tweet.fields": "entities,note_tweet",
+        "tweet.fields": "entities,note_tweet,public_metrics",
     }
     params = expand_attached_media(remove_none_values(params))
 
@@ -242,7 +259,7 @@ async def lookup_tweet_by_id(
     params = {
         "expansions": "author_id",
         "user.fields": "id,name,username,entities",
-        "tweet.fields": "entities,note_tweet",
+        "tweet.fields": "entities,note_tweet,public_metrics",
     }
     params = expand_attached_media(params)
 

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arcade_x
-Version: 1.2.0
+Version: 1.3.1
 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.1.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=pN3V_B3P6genFI1Wh6FXfGw1ggp3GOLk4W2pgZaAtZ0,10031
+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.1.dist-info/METADATA,sha256=39055VcWrVEvQ9w0-5d4o8XgYdJBJhd_Xox4jMELGg0,897
+arcade_x-1.3.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+arcade_x-1.3.1.dist-info/licenses/LICENSE,sha256=ixeE7aL9b2B-_ZYHTY1vQcJB4NufKeo-LWwKNObGDN0,1960
+arcade_x-1.3.1.dist-info/RECORD,,

arcade_x-1.2.0.dist-info/RECORD

@@ -1,11 +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/user_context.py,sha256=nKlsM2LGdoIPl8FQ7FKqd6hkbPHVSJUiZdShZ7ZCsIk,2470
-arcade_x/tools/users.py,sha256=gopYKTBOwbZAeVdL_T3vwkepbpv1bdPbqcU_2oPC3SY,2132
-arcade_x/tools/utils.py,sha256=lR_shws08LWQD-4XU9r3xYmasZ2j2i-cUaCCDnuN8UE,5723
-arcade_x-1.2.0.dist-info/METADATA,sha256=JJjv675epNksYQsVM50LV1jT6RuO3PwdafIcs2hpuOM,897
-arcade_x-1.2.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-arcade_x-1.2.0.dist-info/licenses/LICENSE,sha256=ixeE7aL9b2B-_ZYHTY1vQcJB4NufKeo-LWwKNObGDN0,1960
-arcade_x-1.2.0.dist-info/RECORD,,