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

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/client.d.ts

@@ -30,7 +30,7 @@ export declare class Client {
     get region(): Region;
     get project(): string;
     isEmbedKey(): boolean;
-    fetch<T>(input: "/agents/trigger" | `/agents/${string}/get` | `/agents/${string}/tasks/${string}/metadata` | `/agents/${string}/tasks/${string}/view` | "/agents/conversations/list" | "/services/get_temporary_file_upload_url" | `/workforce/items/${string}` | `/workforce/tasks/${string}/metadata` | `/workforce/items/${string}/tasks/${string}/messages` | "/workforce/trigger" | "/agents/list", init?: RequestInit): Promise<T>;
+    fetch<T>(input: "/agents/trigger" | `/agents/${string}/get` | `/agents/${string}/tasks/${string}/metadata` | `/agents/${string}/tasks/${string}/view` | "/agents/conversations/list" | "/services/get_temporary_file_upload_url" | `/workforce/items/${string}` | `/workforce/tasks/${string}/metadata` | `/workforce/items/${string}/tasks/${string}/messages` | "/workforce/trigger" | "/workforce/tasks/list" | "/agents/list", init?: RequestInit): Promise<T>;
     url(path: string): URL;
     uploadTempFile(file: File): Promise<Attachment>;
 }

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/key.d.ts

@@ -4,9 +4,13 @@ type CreateKeyOptions = {
     region: Region;
     project: string;
     agentId?: string;
+    workforceId?: string;
     taskPrefix?: string;
 };
-type GenerateEmbedKeyOptions = Omit<CreateKeyOptions, "key" | "taskPrefix">;
+type GenerateEmbedKeyOptions = {
+    region: Region;
+    project: string;
+};
 /**
  * Key is used to authenticate requests for the {@link Client}. A Key can be
  * either a _full_ key or an _embed_ key.
@@ -31,7 +35,13 @@ export declare class Key {
      *
      * @returns {Promise<Key>}
      */
-    static generateEmbedKey({ region, project, agentId, }: GenerateEmbedKeyOptions): Promise<Key>;
+    static generateEmbedKey(options: GenerateEmbedKeyOptions & ({
+        workforceId: string;
+    } | {
+        agentId: string;
+    })): Promise<Key>;
+    private static generateAgentEmbedKey;
+    private static generateWorkforceEmbedKey;
     /**
      * The region the key is scoped to.
      *
@@ -46,11 +56,18 @@ export declare class Key {
     readonly project: string;
     /**
      * The agent ID the embed key is scoped to. This is `undefined` for full
-     * keys.
+     * keys and workforce embed keys.
      *
      * @property {string | undefined} agentId
      */
     readonly agentId: string | undefined;
+    /**
+     * The workforce ID the embed key is scoped to. This is `undefined` for full
+     * keys
+     *
+     * @property {string | undefined} workforceId
+     */
+    readonly workforceId: string | undefined;
     /**
      * The task prefix used to namespace tasks created with the embed key. This
      * is `undefined` for full keys.
@@ -63,7 +80,7 @@ export declare class Key {
      *
      * @param {CreateKeyOptions} options
      */
-    constructor({ key, region, project, agentId, taskPrefix }: CreateKeyOptions);
+    constructor({ key, region, project, agentId, workforceId, taskPrefix }: CreateKeyOptions);
     /**
      * Returns whether the key is an embed key.
      *

package/esm/key.js

@@ -23,14 +23,23 @@ export class Key {
      *
      * @returns {Promise<Key>}
      */
-    static async generateEmbedKey({ region, project, agentId, }) {
+    static async generateEmbedKey(options) {
+        if ("workforceId" in options) {
+            return await Key.generateWorkforceEmbedKey(options);
+        }
+        if ("agentId" in options) {
+            return await Key.generateAgentEmbedKey(options);
+        }
+        throw new Error("invalid embed key options");
+    }
+    static async generateAgentEmbedKey({ region, project, agentId }) {
         const embedKeyURL = new URL(cleanPath("/agents/get_embed_key"), regionBaseURL(region));
         const response = await fetch(embedKeyURL, {
             method: "POST",
             body: JSON.stringify({ agent_id: agentId, project }),
         });
         if (!response.ok) {
-            throw new Error("failed to fetch embed key", { cause: response });
+            throw new Error("failed to fetch agent embed key", { cause: response });
         }
         const { embed_key: embedKey, conversation_prefix: taskPrefix } = await response.json();
         return new Key({
@@ -41,6 +50,26 @@ export class Key {
             taskPrefix,
         });
     }
+    static async generateWorkforceEmbedKey({ region, project, workforceId }) {
+        const embedKeyURL = new URL(cleanPath("/workforce/get_embed_key"), regionBaseURL(region));
+        const response = await fetch(embedKeyURL, {
+            method: "POST",
+            body: JSON.stringify({ workforce_id: workforceId, project }),
+        });
+        if (!response.ok) {
+            throw new Error("failed to fetch workforce embed key", {
+                cause: response,
+            });
+        }
+        const { embed_key: embedKey, conversation_prefix: taskPrefix } = await response.json();
+        return new Key({
+            key: embedKey,
+            region,
+            project,
+            workforceId,
+            taskPrefix,
+        });
+    }
     /**
      * The region the key is scoped to.
      *
@@ -62,11 +91,18 @@ export class Key {
     #key;
     /**
      * The agent ID the embed key is scoped to. This is `undefined` for full
-     * keys.
+     * keys and workforce embed keys.
      *
      * @property {string | undefined} agentId
      */
     agentId;
+    /**
+     * The workforce ID the embed key is scoped to. This is `undefined` for full
+     * keys
+     *
+     * @property {string | undefined} workforceId
+     */
+    workforceId;
     /**
      * The task prefix used to namespace tasks created with the embed key. This
      * is `undefined` for full keys.
@@ -79,12 +115,13 @@ export class Key {
      *
      * @param {CreateKeyOptions} options
      */
-    constructor({ key, region, project, agentId, taskPrefix }) {
+    constructor({ key, region, project, agentId, workforceId, taskPrefix }) {
         this.#key = key;
         this.region = region;
         this.project = project;
         this.agentId = agentId;
         this.taskPrefix = taskPrefix;
+        this.workforceId = workforceId;
     }
     /**
      * Returns whether the key is an embed key.
@@ -92,7 +129,8 @@ export class Key {
      * @returns {boolean}
      */
     isEmbed() {
-        return (this.agentId !== undefined && this.taskPrefix !== undefined);
+        return ((this.agentId !== undefined || this.workforceId !== undefined) &&
+            this.taskPrefix !== undefined);
     }
     /**
      * Returns the headers required for authenticating requests with this key.
@@ -114,8 +152,12 @@ export class Key {
             key: this.#key,
             region: this.region,
             project: this.project,
-            agentId: this.agentId,
-            taskPrefix: this.taskPrefix,
+            ...(this.agentId
+                ? { agentId: this.agentId, taskPrefix: this.taskPrefix }
+                : {}),
+            ...(this.workforceId
+                ? { workforceId: this.workforceId, taskPrefix: this.taskPrefix }
+                : {}),
         };
     }
 }

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/task/workforce-strategy.d.ts

@@ -2,16 +2,22 @@ import type { AnyTaskMessage } from "../message/task.js";
 import { Client } from "../client.js";
 import { Task, type TaskMetadata, type TaskStatus, type TaskStrategy } from "./task.js";
 import type { Workforce } from "../workforce.js";
+export type WorkforceTaskState = "running" | "completed" | "execution-limit-reached" | "pending-approval" | "errored-pending-approval" | "escalated";
+/**
+ * Converts a WorkforceTaskState to a simplified TaskStatus.
+ *
+ * @param {WorkforceTaskState} state The workforce task state to convert.
+ * @returns {TaskStatus} The simplified task status.
+ */
+export declare function stateToStatus(state: WorkforceTaskState): TaskStatus;
 export declare class WorkforceStrategy implements TaskStrategy<Workforce> {
     static get(id: string, workforce: Workforce, client?: Client): Promise<Task<Workforce>>;
-    static convertStatus(): TaskStatus;
+    static convertStatus(state: WorkforceTaskState): TaskStatus;
     private readonly id;
     private readonly workforce;
     private readonly client;
-    private constructor();
+    constructor(id: string, workforce: Workforce, client: Client);
     get subject(): Workforce;
-    getMessages({ after: _after }: {
-        after?: Date;
-    }): Promise<AnyTaskMessage[]>;
+    getMessages(): Promise<AnyTaskMessage[]>;
     getMetadata(): Promise<TaskMetadata>;
 }

package/esm/task/workforce-strategy.js

@@ -6,14 +6,37 @@ import { ToolMessage } from "../message/tool.js";
 import { UserMessage } from "../message/user.js";
 import { WorkforceAgentMessage, } from "../message/workforce-agent.js";
 import { WorkforceAgentHandoverMessage, } from "../message/workforce-agent-handover.js";
+/**
+ * Converts a WorkforceTaskState to a simplified TaskStatus.
+ *
+ * @param {WorkforceTaskState} state The workforce task state to convert.
+ * @returns {TaskStatus} The simplified task status.
+ */
+export function stateToStatus(state) {
+    switch (state) {
+        case "running":
+            return "running";
+        case "completed":
+            return "completed";
+        case "execution-limit-reached":
+            return "error";
+        case "pending-approval":
+        case "escalated":
+            return "action";
+        case "errored-pending-approval":
+            return "error";
+        default:
+            throw new Error(`unhandled workforce task state: ${state}`);
+    }
+}
 export class WorkforceStrategy {
     static async get(id, workforce, client = Client.default()) {
         const subject = new this(id, workforce, client);
         const metadata = await subject.getMetadata();
         return new Task(metadata, subject);
     }
-    static convertStatus() {
-        return "not-implemented";
+    static convertStatus(state) {
+        return stateToStatus(state);
     }
     id;
     workforce;
@@ -26,7 +49,7 @@ export class WorkforceStrategy {
     get subject() {
         return this.workforce;
     }
-    async getMessages({ after: _after }) {
+    async getMessages() {
         const { results } = await this.client.fetch(`/workforce/items/${this.workforce.id}/tasks/${this.id}/messages`, {
             method: "POST",
             body: JSON.stringify({
@@ -60,7 +83,7 @@ export class WorkforceStrategy {
             region: this.client.region,
             project: this.client.project,
             name: metadata.title,
-            status: WorkforceStrategy.convertStatus(), // @todo: not implemented
+            status: WorkforceStrategy.convertStatus(metadata.requested_state),
             createdAt: new Date(metadata.insert_date),
             updatedAt: new Date(metadata.update_date),
         };

package/esm/workforce.d.ts

@@ -1,12 +1,17 @@
 import { Client } from "./client.js";
 import type { Region } from "./region.js";
-import { type Task } from "./task/task.js";
+import { Task } from "./task/task.js";
 type WorkforceConfig = {
     id: string;
     workforce_metadata: {
         name: string;
     };
 };
+type GetTaskOptions = {
+    pageSize?: number;
+    cursor?: string;
+    search?: string;
+};
 export declare class Workforce {
     #private;
     static get(id: string, client?: Client): Promise<Workforce>;
@@ -17,6 +22,7 @@ export declare class Workforce {
     get project(): string;
     get name(): string;
     getTask(id: string): Promise<Task<Workforce>>;
+    getTasks({ pageSize, cursor, search, }?: GetTaskOptions): Promise<Task<Workforce>[]>;
     sendMessage(message: string, task?: Task<Workforce>): Promise<Task<Workforce>>;
 }
 export {};

package/esm/workforce.js

@@ -1,6 +1,8 @@
 import { Client } from "./client.js";
-import { resetBackoffDuration } from "./task/task.js";
-import { WorkforceStrategy } from "./task/workforce-strategy.js";
+import { TASK_PREFIX_DELIM } from "./const.js";
+import { resetBackoffDuration, Task } 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}`);
@@ -27,12 +29,45 @@ export class Workforce {
     async getTask(id) {
         return await WorkforceStrategy.get(id, this, this.client);
     }
+    async getTasks({ pageSize = 100, cursor, search, } = {}) {
+        const body = {
+            workforce_id: this.id,
+            page_size: pageSize,
+        };
+        if (search?.trim()) {
+            body.query = search.trim();
+        }
+        if (cursor?.trim()) {
+            body.cursor = cursor.trim();
+        }
+        const { results } = await this.client.fetch("/workforce/tasks/list", {
+            method: "POST",
+            body: JSON.stringify(body),
+        });
+        return results.map((t) => new Task({
+            id: t.workforce_task_id,
+            region: this.client.region,
+            project: this.client.project,
+            name: t.title,
+            status: WorkforceStrategy.convertStatus(t.state),
+            createdAt: new Date(t.insert_date),
+            updatedAt: new Date(t.update_date),
+        }, new WorkforceStrategy(t.workforce_task_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 +80,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.3",
   "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/client.d.ts

@@ -30,7 +30,7 @@ export declare class Client {
     get region(): Region;
     get project(): string;
     isEmbedKey(): boolean;
-    fetch<T>(input: "/agents/trigger" | `/agents/${string}/get` | `/agents/${string}/tasks/${string}/metadata` | `/agents/${string}/tasks/${string}/view` | "/agents/conversations/list" | "/services/get_temporary_file_upload_url" | `/workforce/items/${string}` | `/workforce/tasks/${string}/metadata` | `/workforce/items/${string}/tasks/${string}/messages` | "/workforce/trigger" | "/agents/list", init?: RequestInit): Promise<T>;
+    fetch<T>(input: "/agents/trigger" | `/agents/${string}/get` | `/agents/${string}/tasks/${string}/metadata` | `/agents/${string}/tasks/${string}/view` | "/agents/conversations/list" | "/services/get_temporary_file_upload_url" | `/workforce/items/${string}` | `/workforce/tasks/${string}/metadata` | `/workforce/items/${string}/tasks/${string}/messages` | "/workforce/trigger" | "/workforce/tasks/list" | "/agents/list", init?: RequestInit): Promise<T>;
     url(path: string): URL;
     uploadTempFile(file: File): Promise<Attachment>;
 }

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/key.d.ts

@@ -4,9 +4,13 @@ type CreateKeyOptions = {
     region: Region;
     project: string;
     agentId?: string;
+    workforceId?: string;
     taskPrefix?: string;
 };
-type GenerateEmbedKeyOptions = Omit<CreateKeyOptions, "key" | "taskPrefix">;
+type GenerateEmbedKeyOptions = {
+    region: Region;
+    project: string;
+};
 /**
  * Key is used to authenticate requests for the {@link Client}. A Key can be
  * either a _full_ key or an _embed_ key.
@@ -31,7 +35,13 @@ export declare class Key {
      *
      * @returns {Promise<Key>}
      */
-    static generateEmbedKey({ region, project, agentId, }: GenerateEmbedKeyOptions): Promise<Key>;
+    static generateEmbedKey(options: GenerateEmbedKeyOptions & ({
+        workforceId: string;
+    } | {
+        agentId: string;
+    })): Promise<Key>;
+    private static generateAgentEmbedKey;
+    private static generateWorkforceEmbedKey;
     /**
      * The region the key is scoped to.
      *
@@ -46,11 +56,18 @@ export declare class Key {
     readonly project: string;
     /**
      * The agent ID the embed key is scoped to. This is `undefined` for full
-     * keys.
+     * keys and workforce embed keys.
      *
      * @property {string | undefined} agentId
      */
     readonly agentId: string | undefined;
+    /**
+     * The workforce ID the embed key is scoped to. This is `undefined` for full
+     * keys
+     *
+     * @property {string | undefined} workforceId
+     */
+    readonly workforceId: string | undefined;
     /**
      * The task prefix used to namespace tasks created with the embed key. This
      * is `undefined` for full keys.
@@ -63,7 +80,7 @@ export declare class Key {
      *
      * @param {CreateKeyOptions} options
      */
-    constructor({ key, region, project, agentId, taskPrefix }: CreateKeyOptions);
+    constructor({ key, region, project, agentId, workforceId, taskPrefix }: CreateKeyOptions);
     /**
      * Returns whether the key is an embed key.
      *

package/script/key.js

@@ -26,14 +26,23 @@ class Key {
      *
      * @returns {Promise<Key>}
      */
-    static async generateEmbedKey({ region, project, agentId, }) {
+    static async generateEmbedKey(options) {
+        if ("workforceId" in options) {
+            return await Key.generateWorkforceEmbedKey(options);
+        }
+        if ("agentId" in options) {
+            return await Key.generateAgentEmbedKey(options);
+        }
+        throw new Error("invalid embed key options");
+    }
+    static async generateAgentEmbedKey({ region, project, agentId }) {
         const embedKeyURL = new URL((0, utils_js_1.cleanPath)("/agents/get_embed_key"), (0, region_js_1.regionBaseURL)(region));
         const response = await fetch(embedKeyURL, {
             method: "POST",
             body: JSON.stringify({ agent_id: agentId, project }),
         });
         if (!response.ok) {
-            throw new Error("failed to fetch embed key", { cause: response });
+            throw new Error("failed to fetch agent embed key", { cause: response });
         }
         const { embed_key: embedKey, conversation_prefix: taskPrefix } = await response.json();
         return new Key({
@@ -44,6 +53,26 @@ class Key {
             taskPrefix,
         });
     }
+    static async generateWorkforceEmbedKey({ region, project, workforceId }) {
+        const embedKeyURL = new URL((0, utils_js_1.cleanPath)("/workforce/get_embed_key"), (0, region_js_1.regionBaseURL)(region));
+        const response = await fetch(embedKeyURL, {
+            method: "POST",
+            body: JSON.stringify({ workforce_id: workforceId, project }),
+        });
+        if (!response.ok) {
+            throw new Error("failed to fetch workforce embed key", {
+                cause: response,
+            });
+        }
+        const { embed_key: embedKey, conversation_prefix: taskPrefix } = await response.json();
+        return new Key({
+            key: embedKey,
+            region,
+            project,
+            workforceId,
+            taskPrefix,
+        });
+    }
     /**
      * The region the key is scoped to.
      *
@@ -65,11 +94,18 @@ class Key {
     #key;
     /**
      * The agent ID the embed key is scoped to. This is `undefined` for full
-     * keys.
+     * keys and workforce embed keys.
      *
      * @property {string | undefined} agentId
      */
     agentId;
+    /**
+     * The workforce ID the embed key is scoped to. This is `undefined` for full
+     * keys
+     *
+     * @property {string | undefined} workforceId
+     */
+    workforceId;
     /**
      * The task prefix used to namespace tasks created with the embed key. This
      * is `undefined` for full keys.
@@ -82,12 +118,13 @@ class Key {
      *
      * @param {CreateKeyOptions} options
      */
-    constructor({ key, region, project, agentId, taskPrefix }) {
+    constructor({ key, region, project, agentId, workforceId, taskPrefix }) {
         this.#key = key;
         this.region = region;
         this.project = project;
         this.agentId = agentId;
         this.taskPrefix = taskPrefix;
+        this.workforceId = workforceId;
     }
     /**
      * Returns whether the key is an embed key.
@@ -95,7 +132,8 @@ class Key {
      * @returns {boolean}
      */
     isEmbed() {
-        return (this.agentId !== undefined && this.taskPrefix !== undefined);
+        return ((this.agentId !== undefined || this.workforceId !== undefined) &&
+            this.taskPrefix !== undefined);
     }
     /**
      * Returns the headers required for authenticating requests with this key.
@@ -117,8 +155,12 @@ class Key {
             key: this.#key,
             region: this.region,
             project: this.project,
-            agentId: this.agentId,
-            taskPrefix: this.taskPrefix,
+            ...(this.agentId
+                ? { agentId: this.agentId, taskPrefix: this.taskPrefix }
+                : {}),
+            ...(this.workforceId
+                ? { workforceId: this.workforceId, taskPrefix: this.taskPrefix }
+                : {}),
         };
     }
 }

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/task/workforce-strategy.d.ts

@@ -2,16 +2,22 @@ import type { AnyTaskMessage } from "../message/task.js";
 import { Client } from "../client.js";
 import { Task, type TaskMetadata, type TaskStatus, type TaskStrategy } from "./task.js";
 import type { Workforce } from "../workforce.js";
+export type WorkforceTaskState = "running" | "completed" | "execution-limit-reached" | "pending-approval" | "errored-pending-approval" | "escalated";
+/**
+ * Converts a WorkforceTaskState to a simplified TaskStatus.
+ *
+ * @param {WorkforceTaskState} state The workforce task state to convert.
+ * @returns {TaskStatus} The simplified task status.
+ */
+export declare function stateToStatus(state: WorkforceTaskState): TaskStatus;
 export declare class WorkforceStrategy implements TaskStrategy<Workforce> {
     static get(id: string, workforce: Workforce, client?: Client): Promise<Task<Workforce>>;
-    static convertStatus(): TaskStatus;
+    static convertStatus(state: WorkforceTaskState): TaskStatus;
     private readonly id;
     private readonly workforce;
     private readonly client;
-    private constructor();
+    constructor(id: string, workforce: Workforce, client: Client);
     get subject(): Workforce;
-    getMessages({ after: _after }: {
-        after?: Date;
-    }): Promise<AnyTaskMessage[]>;
+    getMessages(): Promise<AnyTaskMessage[]>;
     getMetadata(): Promise<TaskMetadata>;
 }

package/script/task/workforce-strategy.js

@@ -1,6 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.WorkforceStrategy = void 0;
+exports.stateToStatus = stateToStatus;
 const client_js_1 = require("../client.js");
 const task_js_1 = require("./task.js");
 const agent_error_js_1 = require("../message/agent-error.js");
@@ -9,14 +10,37 @@ const tool_js_1 = require("../message/tool.js");
 const user_js_1 = require("../message/user.js");
 const workforce_agent_js_1 = require("../message/workforce-agent.js");
 const workforce_agent_handover_js_1 = require("../message/workforce-agent-handover.js");
+/**
+ * Converts a WorkforceTaskState to a simplified TaskStatus.
+ *
+ * @param {WorkforceTaskState} state The workforce task state to convert.
+ * @returns {TaskStatus} The simplified task status.
+ */
+function stateToStatus(state) {
+    switch (state) {
+        case "running":
+            return "running";
+        case "completed":
+            return "completed";
+        case "execution-limit-reached":
+            return "error";
+        case "pending-approval":
+        case "escalated":
+            return "action";
+        case "errored-pending-approval":
+            return "error";
+        default:
+            throw new Error(`unhandled workforce task state: ${state}`);
+    }
+}
 class WorkforceStrategy {
     static async get(id, workforce, client = client_js_1.Client.default()) {
         const subject = new this(id, workforce, client);
         const metadata = await subject.getMetadata();
         return new task_js_1.Task(metadata, subject);
     }
-    static convertStatus() {
-        return "not-implemented";
+    static convertStatus(state) {
+        return stateToStatus(state);
     }
     id;
     workforce;
@@ -29,7 +53,7 @@ class WorkforceStrategy {
     get subject() {
         return this.workforce;
     }
-    async getMessages({ after: _after }) {
+    async getMessages() {
         const { results } = await this.client.fetch(`/workforce/items/${this.workforce.id}/tasks/${this.id}/messages`, {
             method: "POST",
             body: JSON.stringify({
@@ -63,7 +87,7 @@ class WorkforceStrategy {
             region: this.client.region,
             project: this.client.project,
             name: metadata.title,
-            status: WorkforceStrategy.convertStatus(), // @todo: not implemented
+            status: WorkforceStrategy.convertStatus(metadata.requested_state),
             createdAt: new Date(metadata.insert_date),
             updatedAt: new Date(metadata.update_date),
         };

package/script/workforce.d.ts

@@ -1,12 +1,17 @@
 import { Client } from "./client.js";
 import type { Region } from "./region.js";
-import { type Task } from "./task/task.js";
+import { Task } from "./task/task.js";
 type WorkforceConfig = {
     id: string;
     workforce_metadata: {
         name: string;
     };
 };
+type GetTaskOptions = {
+    pageSize?: number;
+    cursor?: string;
+    search?: string;
+};
 export declare class Workforce {
     #private;
     static get(id: string, client?: Client): Promise<Workforce>;
@@ -17,6 +22,7 @@ export declare class Workforce {
     get project(): string;
     get name(): string;
     getTask(id: string): Promise<Task<Workforce>>;
+    getTasks({ pageSize, cursor, search, }?: GetTaskOptions): Promise<Task<Workforce>[]>;
     sendMessage(message: string, task?: Task<Workforce>): Promise<Task<Workforce>>;
 }
 export {};

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}`);
@@ -30,12 +32,45 @@ class Workforce {
     async getTask(id) {
         return await workforce_strategy_js_1.WorkforceStrategy.get(id, this, this.client);
     }
+    async getTasks({ pageSize = 100, cursor, search, } = {}) {
+        const body = {
+            workforce_id: this.id,
+            page_size: pageSize,
+        };
+        if (search?.trim()) {
+            body.query = search.trim();
+        }
+        if (cursor?.trim()) {
+            body.cursor = cursor.trim();
+        }
+        const { results } = await this.client.fetch("/workforce/tasks/list", {
+            method: "POST",
+            body: JSON.stringify(body),
+        });
+        return results.map((t) => new task_js_1.Task({
+            id: t.workforce_task_id,
+            region: this.client.region,
+            project: this.client.project,
+            name: t.title,
+            status: workforce_strategy_js_1.WorkforceStrategy.convertStatus(t.state),
+            createdAt: new Date(t.insert_date),
+            updatedAt: new Date(t.update_date),
+        }, new workforce_strategy_js_1.WorkforceStrategy(t.workforce_task_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 +83,7 @@ class Workforce {
         if (task) {
             task[task_js_1.resetBackoffDuration]();
         }
-        return task ?? this.getTask(taskId);
+        return task ?? this.getTask(taskIdResponse);
     }
 }
 exports.Workforce = Workforce;