Rename text_download.py to push_notification.py

push_notification.py

@@ -0,0 +1,57 @@
+from exponent_server_sdk import (
+    DeviceNotRegisteredError,
+    PushClient,
+    PushMessage,
+    PushServerError,
+    PushTicketError,
+)
+import os
+
+from transformers.tools.base import Tool
+
+# Description for the tool
+PUSH_NOTIFICATION_DESCRIPTION = (
+    "This is a tool that sends a push notification using Expo. It takes two inputs: "
+    "`token`, which is the Expo push token of the recipient, and `message`, which is the "
+    "message to send. It does not return any output."
+)
+
+class PushNotificationTool(Tool):
+    description = PUSH_NOTIFICATION_DESCRIPTION
+    name = "push_notification"
+    inputs = ["token", "message"]
+    outputs = []
+
+    def __call__(self, token, message):
+        try:
+            # Replace 'your_expo_token' with your actual Expo push notification token
+            response = PushClient().publish(
+                PushMessage(to=token, body=message)
+            )
+        except PushServerError as exc:
+            # Encountered some likely formatting/validation error.
+            print(f"Error: {exc.errors}")
+            raise
+        except DeviceNotRegisteredError:
+            # Mark the push token as inactive
+            print("Device not registered.")
+            # Add logic to handle inactive push token in your application
+        except PushTicketError as exc:
+            # Encountered some other per-notification error.
+            print(f"Push ticket error: {exc.push_response._asdict()}")
+            raise
+
+        try:
+            # We got a response back, but we don't know whether it's an error yet.
+            # This call raises errors so we can handle them with normal exception flows.
+            response.validate_response()
+        except DeviceNotRegisteredError:
+            # Mark the push token as inactive
+            print("Device not registered.")
+            # Add logic to handle inactive push token in your application
+        except PushTicketError as exc:
+            # Encountered some other per-notification error.
+            print(f"Push ticket error: {exc.push_response._asdict()}")
+            raise
+
+        return None  # No output is expected

text_download.py

@@ -1,21 +0,0 @@
-import requests
-from bs4 import BeautifulSoup
-
-from transformers.tools.base import Tool
-
-
-TEXT_DOWNLOAD_DESCRIPTION = (
-    "This is a tool that downloads a file from a `url`. It takes the `url` as input, and returns the text"
-    " contained in the file."
-)
-
-
-class TextDownloadTool(Tool):
-
-    inputs = ['text']
-    outputs = ['text']
-    description = TEXT_DOWNLOAD_DESCRIPTION
-
-    def __call__(self, url):
-        return BeautifulSoup(requests.get(url).text, features="html.parser").get_text()
-