feat(tools): implement X Twitter tools

langgenius · Dec 23, 2024 · f66d90a · f66d90a
1 parent 6d1cb4a
commit f66d90a
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 9 deletions.
diff --git a/api/core/tools/provider/builtin/x/tools/get_user_timeline.py b/api/core/tools/provider/builtin/x/tools/get_user_timeline.py
@@ -31,7 +31,7 @@ def _convert_tweet_to_dict(self, tweet: tweepy.Tweet) -> dict[str, Any]:
         """
         tweet_dict = {}
         # Convert Tweet object to dictionary
-        for field in tweet.data.keys():
+        for field in tweet.data:
             value = tweet.data[field]
             # Skip None values
             if value is None:
@@ -166,7 +166,7 @@ def _invoke(
 
                 # Convert user data
                 user_dict = {}
-                for field in user_data.keys():
+                for field in user_data:
                     value = user_data[field]
                     if value is None:
                         continue

diff --git a/api/core/tools/provider/builtin/x/tools/like_tweet.py b/api/core/tools/provider/builtin/x/tools/like_tweet.py
@@ -2,7 +2,7 @@
 Like Tweet Tool for liking/unliking tweets
 """
 
-from typing import Any, Dict, List, Union
+from typing import Any, Union
 
 import tweepy
 
@@ -12,8 +12,8 @@
 
 class LikeTweetTool(BuiltinTool):
     def _invoke(
-        self, user_id: str, tool_parameters: Dict[str, Any]
-    ) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
+        self, user_id: str, tool_parameters: dict[str, Any]
+    ) -> Union[ToolInvokeMessage, list[ToolInvokeMessage]]:
         """
         Like or unlike a tweet
 

diff --git a/api/core/tools/provider/builtin/x/tools/media_upload.py b/api/core/tools/provider/builtin/x/tools/media_upload.py
@@ -3,7 +3,7 @@
 """
 
 import io
-from typing import Any, Dict, List, Union
+from typing import Any, Union
 
 import tweepy
 
@@ -14,7 +14,7 @@
 
 
 class MediaUploadTool(BuiltinTool):
-    def _get_twitter_api_v1(self, credentials: Dict[str, str]) -> tweepy.API:
+    def _get_twitter_api_v1(self, credentials: dict[str, str]) -> tweepy.API:
         """
         Get Twitter API v1.1 connection for media upload
         """
@@ -33,8 +33,8 @@ def _validate_file_type(self, file_type: FileType) -> bool:
         return file_type in [FileType.IMAGE, FileType.VIDEO]
 
     def _invoke(
-        self, user_id: str, tool_parameters: Dict[str, Any]
-    ) -> Union[ToolInvokeMessage, List[ToolInvokeMessage]]:
+        self, user_id: str, tool_parameters: dict[str, Any]
+    ) -> Union[ToolInvokeMessage, list[ToolInvokeMessage]]:
         """
         Upload media to Twitter
         """