arcade-x 0.1.5__py3-none-any.whl → 0.1.6__py3-none-any.whl

arcade_x/tools/tweets.py

@@ -1,4 +1,4 @@
-from typing import Annotated, Any
+from typing import Annotated, Any, Optional
 
 import httpx
 from arcade.sdk import ToolContext, tool
@@ -10,6 +10,7 @@ from arcade_x.tools.utils import (
     get_headers_with_token,
     get_tweet_url,
     parse_search_recent_tweets_response,
+    remove_none_values,
 )
 
 TWEETS_URL = "https://api.x.com/2/tweets"
@@ -63,8 +64,11 @@ async def search_recent_tweets_by_username(
     context: ToolContext,
     username: Annotated[str, "The username of the X (Twitter) user to look up"],
     max_results: Annotated[
-        int, "The maximum number of results to return. Cannot be less than 10"
+        int, "The maximum number of results to return. Must be in range [1, 100] inclusive"
     ] = 10,
+    next_token: Annotated[
+        Optional[str], "The pagination token starting from which to return results"
+    ] = None,
 ) -> Annotated[dict[str, Any], "Dictionary containing the search results"]:
     """Search for recent tweets (last 7 days) on X (Twitter) by username.
     Includes replies and reposts."""
@@ -72,8 +76,13 @@ async def search_recent_tweets_by_username(
     headers = get_headers_with_token(context)
     params: dict[str, int | str] = {
         "query": f"from:{username}",
-        "max_results": max(max_results, 10),  # X API does not allow 'max_results' less than 10
+        "max_results": min(
+            max(max_results, 10), 100
+        ),  # X API does not allow 'max_results' less than 10 or greater than 100
+        "next_token": next_token,
     }
+    params = remove_none_values(params)
+
     url = (
         "https://api.x.com/2/tweets/search/recent?"
         "expansions=author_id&user.fields=id,name,username,entities&tweet.fields=entities"
@@ -106,8 +115,11 @@ async def search_recent_tweets_by_keywords(
         list[str] | None, "List of phrases that must be present in the tweet"
     ] = None,
     max_results: Annotated[
-        int, "The maximum number of results to return. Cannot be less than 10"
+        int, "The maximum number of results to return. Must be in range [1, 100] inclusive"
     ] = 10,
+    next_token: Annotated[
+        Optional[str], "The pagination token starting from which to return results"
+    ] = None,
 ) -> Annotated[dict[str, Any], "Dictionary containing the search results"]:
     """
     Search for recent tweets (last 7 days) on X (Twitter) by required keywords and phrases.
@@ -131,8 +143,13 @@ async def search_recent_tweets_by_keywords(
 
     params: dict[str, int | str] = {
         "query": query.strip(),
-        "max_results": max(max_results, 10),  # X API does not allow 'max_results' less than 10
+        "max_results": min(
+            max(max_results, 10), 100
+        ),  # X API does not allow 'max_results' less than 10 or greater than 100
+        "next_token": next_token,
     }
+    params = remove_none_values(params)
+
     url = (
         "https://api.x.com/2/tweets/search/recent?"
         "expansions=author_id&user.fields=id,name,username,entities&tweet.fields=entities"

arcade_x/tools/utils.py

@@ -28,7 +28,7 @@ def parse_search_recent_tweets_response(response_data: dict[str, Any]) -> dict[s
     Returns the modified response data with added 'tweet_url', 'author_username', and 'author_name'.
     """
     if not sanity_check_tweets_data(response_data):
-        return {"data": []}
+        return {"data": [], "next_token": ""}
 
     # Add 'tweet_url' to each tweet
     for tweet in response_data["data"]:
@@ -110,3 +110,16 @@ def expand_urls_in_user_url(user_data: dict, delete_entities: bool = True) -> di
     if delete_entities:
         new_user_data.pop("entities", None)
     return new_user_data
+
+
+def remove_none_values(params: dict) -> dict:
+    """
+    Remove key/value pairs with None values from a dictionary.
+
+    Args:
+        params: The dictionary to clean
+
+    Returns:
+        A new dictionary with None values removed
+    """
+    return {k: v for k, v in params.items() if v is not None}

{arcade_x-0.1.5.dist-info → arcade_x-0.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: arcade_x
-Version: 0.1.5
+Version: 0.1.6
 Summary: LLM tools for interacting with X (Twitter)
 Author: Arcade AI
 Author-email: dev@arcade-ai.com
@@ -10,5 +10,5 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
-Requires-Dist: arcade-ai (==0.1.5)
+Requires-Dist: arcade-ai (==0.1.6)
 Requires-Dist: httpx (>=0.27.2,<0.28.0)

arcade_x-0.1.6.dist-info/RECORD

@@ -0,0 +1,9 @@
+arcade_x/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+arcade_x/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+arcade_x/tools/tweets.py,sha256=VchdglDVI15rsEyZuBLIj6qjGql0-UO8TdY6ylGl4yA,7158
+arcade_x/tools/users.py,sha256=zNACU9UhQJkGNnrei-E6GWvFnbDzg0YG9pkdplu8TzM,2148
+arcade_x/tools/utils.py,sha256=YKRGXcUPKcPnIvmiNF2-0jB4edxFmFZLJ90OQNR-DGk,4366
+arcade_x-0.1.6.dist-info/LICENSE,sha256=SphQPbiNmBD1J6yJ7oxG9bIZDbj8lNHKSv5Kl86zA40,1066
+arcade_x-0.1.6.dist-info/METADATA,sha256=5eq4z7KaY2GSDY8J-KLUDRdibTsCsHpt29AgW7qlSac,510
+arcade_x-0.1.6.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+arcade_x-0.1.6.dist-info/RECORD,,

arcade_x-0.1.5.dist-info/RECORD

@@ -1,9 +0,0 @@
-arcade_x/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-arcade_x/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-arcade_x/tools/tweets.py,sha256=IDAjx1j_XLoc7ILyuRKVyjTPwQeNtD47lnyKUMO-WpU,6594
-arcade_x/tools/users.py,sha256=zNACU9UhQJkGNnrei-E6GWvFnbDzg0YG9pkdplu8TzM,2148
-arcade_x/tools/utils.py,sha256=VEeUo715m6l6X9Syszy2E72rdvv4h_tc-P7wMaQEbGo,4044
-arcade_x-0.1.5.dist-info/LICENSE,sha256=SphQPbiNmBD1J6yJ7oxG9bIZDbj8lNHKSv5Kl86zA40,1066
-arcade_x-0.1.5.dist-info/METADATA,sha256=pa4GiYU352AShVYdd2rDjVa3AP5b1sFtJpi1gUihQmM,510
-arcade_x-0.1.5.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-arcade_x-0.1.5.dist-info/RECORD,,