Get Run Status action

Author: GTFalcao

components/utopian_labs/actions/get-run-status/get-run-status.mjs

@@ -0,0 +1,22 @@
+import utopianLabs from "../../utopian_labs.app.mjs";
+
+export default {
+  key: "utopian_labs-get-run-status",
+  name: "Get Run Status",
+  description: "Retrieve the status of an initiated run. [See the documentation](https://docs.utopianlabs.ai/research#retrieve-research-run-status)",
+  version: "0.0.1",
+  type: "action",
+  props: {
+    utopianLabs,
+    runId: {
+      type: "string",
+      label: "Run ID",
+      description: "The ID of the run you want to retrieve the status for.",
+    },
+  },
+  async run({ $ }) {
+    const response = await this.utopianLabs.getRunStatus(this.runId);
+    $.export("$summary", `Successfully retrieved status for run ${this.runId}`);
+    return response;
+  },
+};

components/utopian_labs/utopian_labs.app.mjs

@@ -5,10 +5,18 @@ export default {
   app: "utopian_labs",
   propDefinitions: {},
   methods: {
-    _getClient() {
+    _client() {
       return new UtopianLabs({
         apiKey: this.$auth.api_key,
       });
     },
+    initiateRun(args) {
+      return this._client().agents.runs.create(args);
+    },
+    getRunStatus(run) {
+      return this._client().agents.runs.get({
+        run,
+      });
+    },
   },
 };