Added several new actions to support various workflows

Author: cxbxmxcx

.gitignore

@@ -129,6 +129,8 @@ ENV/
 env.bak/
 venv.bak/
 
+assistant_outputs/
+
 logs.txt
 
 # Spyder project settings

btree.py

@@ -0,0 +1,91 @@
+import threading
+import time
+
+import py_trees
+
+from playground.assistants_api import api
+
+
+# Define the ActionWithThread class
+class ActionWrapper(py_trees.behaviour.Behaviour):
+    def __init__(self, name, action_function=None):
+        super(ActionWrapper, self).__init__(name=name)
+        self.thread = None
+        self.thread_running = False
+        self.thread_success = False
+        self.action_function = action_function
+
+    def setup(self):
+        # This is called once at the beginning to setup any necessary state or resources
+        print("%s.setup()" % self.name)
+        return py_trees.common.Status.SUCCESS
+
+    def initialise(self):
+        # This is called once each time the behavior is started
+        print("%s.initialise()" % self.name)
+        self.thread_running = True
+        self.thread_success = False
+        self.thread = threading.Thread(target=self.long_running_process)
+        self.thread.start()
+
+    def long_running_process(self):
+        # Simulate a long-running process
+        try:
+            print("%s: Thread started, running process..." % self.name)
+            result = self.action_function()
+            print(result)
+            self.thread_success = True
+            print("%s: Thread completed successfully." % self.name)
+        except Exception as e:
+            print("%s: Exception in thread: %s" % (self.name, str(e)))
+        finally:
+            self.thread_running = False
+
+    def update(self):
+        # This is called every tick to update the status of the behavior
+        print("%s.update()" % self.name)
+        if self.thread_running:
+            return py_trees.common.Status.RUNNING
+        else:
+            return (
+                py_trees.common.Status.SUCCESS
+                if self.thread_success
+                else py_trees.common.Status.FAILURE
+            )
+
+    def terminate(self, new_status):
+        # This is called once each time the behavior terminates
+        print("%s.terminate(%s)" % (self.name, new_status))
+        if self.thread is not None:
+            self.thread.join()
+        self.thread = None
+        self.thread_running = False
+        self.thread_success = False
+
+
+# Create the root node (sequence)
+root = py_trees.composites.Sequence("RootSequence", memory=True)
+
+thread = api.create_thread()
+assistant = api.get_assistant_by_name("manager")
+message = "check the status of the thread, if there are no recent user messages, do nothing and return the word 'success'"
+
+action_function = lambda: api.call_assistant_with_thread(thread, assistant.id, message)
+
+# Create an instance of ActionWithThread and add it to the root node
+long_running_action = ActionWithThread(
+    name="LongRunningAction", action_function=action_function
+)
+root.add_child(long_running_action)
+
+# Create the behavior tree
+tree = py_trees.trees.BehaviourTree(root)
+
+# Tick the tree to run it
+for i in range(100):
+    print(f"Tick {i + 1}")
+    tree.tick()
+    time.sleep(5)  # Simulate time between ticks
+
+# Add a blackboard watcher to visualize the tree
+# py_trees.display.render_dot_tree(tree.root)

logs/logs.txt

@@ -0,0 +1,30 @@
+Running on local URL:  http://127.0.0.1:7860
+
+To create a public link, set `share=True` in `launch()`.
+# >>> Code Interpreter
+import matplotlib.pyplot as plt
+import numpy as np
+
+# Define the linear function
+def linear_function(x):
+    return 3*x + 6
+
+# Generate x values
+x = np.linspace(-10, 10, 400)
+# Generate y values
+y = linear_function(x)
+
+# Create the plot
+plt.figure(figsize=(8, 6))
+plt.plot(x, y, label='y = 3x + 6', color='blue')
+plt.xlabel('x')
+plt.ylabel('y')
+plt.title('Plot of y = 3x + 6')
+plt.axhline(0, color='black',linewidth=0.5)
+plt.axvline(0, color='black',linewidth=0.5)
+plt.legend()
+plt.grid(True)
+plt.show()
+Output >
+assistant > File saved as assistant_outputs\file_20240526150025.png
+File saved as assistant_outputs\file_20240526150025.png

playground/assistant_actions/api_actions.py

@@ -17,6 +17,31 @@ def list_uploaded_files(purpose="assistants"):
     return json.dumps(files)
 
 
+@agent_action
+def list_all_working_files():
+    """Lists all the uploaded files for the given purpose."""
+    files = api.list_files(purpose=None).data
+    files = [
+        {"id": file.id, "filename": file.filename, "purpose": file.purpose}
+        for file in files
+    ]
+    return json.dumps(files)
+
+
+@agent_action
+def download_working_file_and_save(file_id, filename):
+    """Downloads the file with the given file_id and saves it with the given name."""
+    if file_id is None:
+        return "You must supply a file_id to download."
+
+    file = api.retrieve_file(file_id)
+    if not file:
+        return f"Unable to retrieve file with ID {file_id}."
+    with open(filename, "wb") as f:
+        f.write(file.content)
+    return f"File with ID {file_id} downloaded and saved as {filename}."
+
+
 @agent_action
 def delete_uploaded_file(file_id):
     """Deletes the uploaded file with the given file_id."""

playground/assistant_actions/date_actions.py

@@ -0,0 +1,9 @@
+from datetime import datetime
+
+from playground.actions_manager import agent_action
+
+
+@agent_action
+def get_current_date_and_time():
+    """Returns the current date and time."""
+    return datetime.now().strftime("%Y-%m-%d %H:%M:%S")

playground/assistant_actions/file_actions.py

@@ -0,0 +1,84 @@
+import os
+
+from playground.actions_manager import agent_action
+
+OUTPUT_FOLDER = "assistant_outputs"
+
+
+@agent_action
+def save_file(filename, content):
+    """
+    Save content to a file.
+
+    :param filename: The name of the file including extension.
+    :param content: The content to save in the file.
+    """
+    file_path = os.path.join(OUTPUT_FOLDER, filename)
+    with open(file_path, "w") as file:
+        file.write(content)
+    print(f"File '{filename}' saved successfully.")
+
+
+@agent_action
+def load_file(filename):
+    """
+    Load content from a file.
+
+    :param filename: The name of the file including extension.
+    :return: The content of the file.
+    """
+    file_path = os.path.join(OUTPUT_FOLDER, filename)
+    if not os.path.exists(file_path):
+        print(f"File '{filename}' does not exist.")
+        return None
+
+    with open(file_path, "r") as file:
+        content = file.read()
+    print(f"File '{filename}' loaded successfully.")
+    return content
+
+
+@agent_action
+def delete_file(filename):
+    """
+    Delete a file.
+
+    :param filename: The name of the file including extension.
+    """
+    file_path = os.path.join(OUTPUT_FOLDER, filename)
+    if os.path.exists(file_path):
+        os.remove(file_path)
+        print(f"File '{filename}' deleted successfully.")
+    else:
+        print(f"File '{filename}' does not exist.")
+
+
+@agent_action
+def create_folder(foldername):
+    """
+    Create a folder.
+
+    :param foldername: The name of the folder to create.
+    """
+    folder_path = os.path.join(OUTPUT_FOLDER, foldername)
+    if not os.path.exists(folder_path):
+        os.makedirs(folder_path)
+        print(f"Folder '{foldername}' created successfully.")
+    else:
+        print(f"Folder '{foldername}' already exists.")
+
+
+# # Example usage:
+# if __name__ == "__main__":
+#     # Save a file
+#     save_file('example.txt', 'Hello, world!')
+
+#     # Load a file
+#     content = load_file('example.txt')
+#     print(content)
+
+#     # Delete a file
+#     delete_file('example.txt')
+
+#     # Create a folder
+#     create_folder('example_folder')

playground/assistant_actions/trends_actions.py

@@ -0,0 +1,29 @@
+from pytrends.request import TrendReq
+
+from playground.actions_manager import agent_action
+
+
+@agent_action
+def get_google_trend_data(keywords):
+    """Returns the Google trends trend data for the given keywords.
+    keywords: list of keywords to search for.
+    """
+    # Connect to Google
+    pytrends = TrendReq(hl="en-US", tz=360)
+
+    if isinstance(keywords, str):
+        if "," in keywords:
+            keywords = keywords.split(",")
+        else:
+            keywords = [keywords]
+
+    # Define the keywords
+    kw_list = keywords
+
+    # Build the payload
+    pytrends.build_payload(kw_list, cat=0, timeframe="today 1-m", geo="", gprop="")
+
+    # Get interest over time
+    interest_over_time_df = pytrends.interest_over_time()
+
+    return interest_over_time_df.to_string()