@relevanceai/sdk 3.0.0-rc.1 → 3.0.0-rc.2

package/esm/agent.js

@@ -2,6 +2,7 @@ import { Client } from "./client.js";
 import { randomUUID } from "./utils.js";
 import { AgentStrategy, } from "./task/agent-strategy.js";
 import { resetBackoffDuration, Task } from "./task/task.js";
+import { TASK_PREFIX_DELIM } from "./const.js";
 /**
  * Converts an AgentTaskState to a simplified TaskStatus.
  *
@@ -87,7 +88,6 @@ function taskSortOptionsToParam(sort) {
     }
     throw new Error("invalid sort option");
 }
-const taskPrefixDelimiter = "_-_";
 export class Agent {
     static async get(id, client = Client.default()) {
         const config = await Agent.#fetchConfig(id, client);
@@ -187,7 +187,7 @@ export class Agent {
         let taskId;
         // embed keys require a task prefixing for new tasks
         if (!task && this.client.isEmbedKey()) {
-            taskId = [this.client.key.taskPrefix, await randomUUID()].join(taskPrefixDelimiter);
+            taskId = [this.client.key.taskPrefix, await randomUUID()].join(TASK_PREFIX_DELIM);
         }
         else if (task) {
             taskId = task.id;

package/esm/const.d.ts

@@ -0,0 +1 @@
+export declare const TASK_PREFIX_DELIM = "_-_";

package/esm/const.js

@@ -0,0 +1 @@
+export const TASK_PREFIX_DELIM = "_-_";

package/esm/message/agent.d.ts

@@ -1,9 +1,14 @@
-import { GenericMessage } from "./task.js";
+import type { Agent } from "../agent.js";
+import { GenericMessage, type TaskMessageData } from "./task.js";
 export interface AgentMessageContent {
     type: "agent-message";
     text: string;
+    thought_about_tool_calls?: boolean;
 }
 export declare class AgentMessage extends GenericMessage<AgentMessageContent> {
+    readonly agent?: Agent;
+    constructor(message: TaskMessageData<AgentMessageContent>, agent?: Agent);
     get text(): string;
     get agentId(): string;
+    isThought(): boolean;
 }

package/esm/message/agent.js

@@ -1,9 +1,17 @@
 import { GenericMessage } from "./task.js";
 export class AgentMessage extends GenericMessage {
+    agent;
+    constructor(message, agent) {
+        super(message);
+        this.agent = agent;
+    }
     get text() {
         return this.message.content.text;
     }
     get agentId() {
-        return "";
+        return this.agent?.id ?? "";
+    }
+    isThought() {
+        return this.message.content.thought_about_tool_calls ?? false;
     }
 }

package/esm/mod.d.ts

@@ -4,3 +4,4 @@ export { Key } from "./key.js";
 export { type Region, REGION_AU, REGION_EU, REGION_US } from "./region.js";
 export { Workforce } from "./workforce.js";
 export type { Task } from "./task/task.js";
+export type { AnyTaskMessage } from "./message/task.js";

package/esm/task/agent-strategy.js

@@ -50,7 +50,7 @@ export class AgentStrategy {
                 case "agent-error":
                     return new AgentErrorMessage(data);
                 case "agent-message":
-                    return new AgentMessage(data);
+                    return new AgentMessage(data, this.agent);
                 case "tool-run":
                     return new ToolMessage(data);
                 case "user-message":

package/esm/workforce.js

@@ -1,6 +1,8 @@
 import { Client } from "./client.js";
+import { TASK_PREFIX_DELIM } from "./const.js";
 import { resetBackoffDuration } from "./task/task.js";
 import { WorkforceStrategy } from "./task/workforce-strategy.js";
+import { randomUUID } from "./utils.js";
 export class Workforce {
     static async get(id, client = Client.default()) {
         const config = await client.fetch(`/workforce/items/${id}`);
@@ -28,11 +30,19 @@ export class Workforce {
         return await WorkforceStrategy.get(id, this, this.client);
     }
     async sendMessage(message, task) {
-        const { workforce_task_id: taskId } = await this.client.fetch("/workforce/trigger", {
+        let taskId;
+        // embed keys require a task prefixing for new tasks
+        if (!task && this.client.isEmbedKey()) {
+            taskId = [this.client.key.taskPrefix, await randomUUID()].join(TASK_PREFIX_DELIM);
+        }
+        else if (task) {
+            taskId = task.id;
+        }
+        const { workforce_task_id: taskIdResponse } = await this.client.fetch("/workforce/trigger", {
             method: "POST",
             body: JSON.stringify({
                 workforce_id: this.id,
-                workforce_task_id: task?.id,
+                workforce_task_id: taskId,
                 trigger: {
                     message: {
                         role: "user",
@@ -45,6 +55,6 @@ export class Workforce {
         if (task) {
             task[resetBackoffDuration]();
         }
-        return task ?? this.getTask(taskId);
+        return task ?? this.getTask(taskIdResponse);
     }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@relevanceai/sdk",
-  "version": "3.0.0-rc.1",
+  "version": "3.0.0-rc.2",
   "license": "MIT",
   "main": "./script/mod.js",
   "module": "./esm/mod.js",

package/script/agent.js

@@ -6,6 +6,7 @@ const client_js_1 = require("./client.js");
 const utils_js_1 = require("./utils.js");
 const agent_strategy_js_1 = require("./task/agent-strategy.js");
 const task_js_1 = require("./task/task.js");
+const const_js_1 = require("./const.js");
 /**
  * Converts an AgentTaskState to a simplified TaskStatus.
  *
@@ -91,7 +92,6 @@ function taskSortOptionsToParam(sort) {
     }
     throw new Error("invalid sort option");
 }
-const taskPrefixDelimiter = "_-_";
 class Agent {
     static async get(id, client = client_js_1.Client.default()) {
         const config = await Agent.#fetchConfig(id, client);
@@ -191,7 +191,7 @@ class Agent {
         let taskId;
         // embed keys require a task prefixing for new tasks
         if (!task && this.client.isEmbedKey()) {
-            taskId = [this.client.key.taskPrefix, await (0, utils_js_1.randomUUID)()].join(taskPrefixDelimiter);
+            taskId = [this.client.key.taskPrefix, await (0, utils_js_1.randomUUID)()].join(const_js_1.TASK_PREFIX_DELIM);
         }
         else if (task) {
             taskId = task.id;

package/script/const.d.ts

@@ -0,0 +1 @@
+export declare const TASK_PREFIX_DELIM = "_-_";

package/script/const.js

@@ -0,0 +1,4 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TASK_PREFIX_DELIM = void 0;
+exports.TASK_PREFIX_DELIM = "_-_";

package/script/message/agent.d.ts

@@ -1,9 +1,14 @@
-import { GenericMessage } from "./task.js";
+import type { Agent } from "../agent.js";
+import { GenericMessage, type TaskMessageData } from "./task.js";
 export interface AgentMessageContent {
     type: "agent-message";
     text: string;
+    thought_about_tool_calls?: boolean;
 }
 export declare class AgentMessage extends GenericMessage<AgentMessageContent> {
+    readonly agent?: Agent;
+    constructor(message: TaskMessageData<AgentMessageContent>, agent?: Agent);
     get text(): string;
     get agentId(): string;
+    isThought(): boolean;
 }

package/script/message/agent.js

@@ -3,11 +3,19 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.AgentMessage = void 0;
 const task_js_1 = require("./task.js");
 class AgentMessage extends task_js_1.GenericMessage {
+    agent;
+    constructor(message, agent) {
+        super(message);
+        this.agent = agent;
+    }
     get text() {
         return this.message.content.text;
     }
     get agentId() {
-        return "";
+        return this.agent?.id ?? "";
+    }
+    isThought() {
+        return this.message.content.thought_about_tool_calls ?? false;
     }
 }
 exports.AgentMessage = AgentMessage;

package/script/mod.d.ts

@@ -4,3 +4,4 @@ export { Key } from "./key.js";
 export { type Region, REGION_AU, REGION_EU, REGION_US } from "./region.js";
 export { Workforce } from "./workforce.js";
 export type { Task } from "./task/task.js";
+export type { AnyTaskMessage } from "./message/task.js";

package/script/task/agent-strategy.js

@@ -53,7 +53,7 @@ class AgentStrategy {
                 case "agent-error":
                     return new agent_error_js_1.AgentErrorMessage(data);
                 case "agent-message":
-                    return new agent_js_2.AgentMessage(data);
+                    return new agent_js_2.AgentMessage(data, this.agent);
                 case "tool-run":
                     return new tool_js_1.ToolMessage(data);
                 case "user-message":

package/script/workforce.js

@@ -2,8 +2,10 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.Workforce = void 0;
 const client_js_1 = require("./client.js");
+const const_js_1 = require("./const.js");
 const task_js_1 = require("./task/task.js");
 const workforce_strategy_js_1 = require("./task/workforce-strategy.js");
+const utils_js_1 = require("./utils.js");
 class Workforce {
     static async get(id, client = client_js_1.Client.default()) {
         const config = await client.fetch(`/workforce/items/${id}`);
@@ -31,11 +33,19 @@ class Workforce {
         return await workforce_strategy_js_1.WorkforceStrategy.get(id, this, this.client);
     }
     async sendMessage(message, task) {
-        const { workforce_task_id: taskId } = await this.client.fetch("/workforce/trigger", {
+        let taskId;
+        // embed keys require a task prefixing for new tasks
+        if (!task && this.client.isEmbedKey()) {
+            taskId = [this.client.key.taskPrefix, await (0, utils_js_1.randomUUID)()].join(const_js_1.TASK_PREFIX_DELIM);
+        }
+        else if (task) {
+            taskId = task.id;
+        }
+        const { workforce_task_id: taskIdResponse } = await this.client.fetch("/workforce/trigger", {
             method: "POST",
             body: JSON.stringify({
                 workforce_id: this.id,
-                workforce_task_id: task?.id,
+                workforce_task_id: taskId,
                 trigger: {
                     message: {
                         role: "user",
@@ -48,7 +58,7 @@ class Workforce {
         if (task) {
             task[task_js_1.resetBackoffDuration]();
         }
-        return task ?? this.getTask(taskId);
+        return task ?? this.getTask(taskIdResponse);
     }
 }
 exports.Workforce = Workforce;