Added a list_datastore_category action

Author: frikky

shuffle-ai/1.1.0/api.yaml

@@ -37,7 +37,7 @@ actions:
     description: Input ANY kind of data in the format you want, and the format you want it in. Default is a business-y email. Uses ShuffleGPT, which is based on OpenAI and our own model.
     parameters:
       - name: apikey 
-        description: Your https://shuffler.io apikey
+        description: Your https://shuffler.io/settings apikey
         required: true
         multiline: false 
         example: ""

shuffle-tools/1.2.0/api.yaml

@@ -81,6 +81,32 @@ actions:
     returns:
       schema:
         type: string
+
+  - name: list_datastore_category
+    description: Lists data in the datastore category specified. 
+    parameters:
+      - name: category 
+        description: The category to list 
+        required: true
+        multiline: false
+        example: "shuffle-security_incidents"
+        schema:
+          type: string
+      - name: output_type 
+        description: The output type to get
+        required: false 
+        options:
+          - values 
+          - keys
+          - ndjson
+          - csv 
+          - meta 
+        example: "values"
+        schema:
+          type: string
+    returns:
+      schema:
+        type: string
   - name: set_datastore_value
     description: Set a value to be saved to your organization in Shuffle.
     parameters:

shuffle-tools/1.2.0/src/app.py

@@ -2222,6 +2222,26 @@ def change_cache_subkey(self, key, subkey, value, overwrite, category=""):
             self.logger.info("Value couldn't be parsed")
             return response.text
 
+    def list_datastore_category(self, category, output_type=""):
+        category = category.lower().replace(" ", "_")
+        if not output_type: 
+            output_type = "values"
+
+        org_id = self.full_execution["workflow"]["execution_org"]["id"]
+        url = "%s/api/v2/datastore/category/%s?top=10000&type=%s&execution_id=%s&authorization=%s&org_id=%s" % (self.url, category, output_type, self.current_execution_id, self.authorization, org_id)
+
+        parsed_headers = {}
+        data = requests.get(url, headers=parsed_headers, verify=False)
+        try:
+            allvalues = data.json()
+            return json.dumps(allvalues)
+        except Exception as e:
+            return {
+                "success": False,
+                "reason": f"Failed to parse datastore category list response: {e}",
+                "response": data.text,
+            }
+
     def delete_datastore_value(self, key, category=""):
         return self.delete_cache(key, category=category)