@multi-agent-protocol/sdk 0.1.0 → 0.1.2

package/dist/testing.cjs

@@ -151,7 +151,12 @@ var EVENT_TYPES = {
   SCOPE_MEMBER_LEFT: "scope_member_left",
   // Permission events
   PERMISSIONS_CLIENT_UPDATED: "permissions_client_updated",
-  PERMISSIONS_AGENT_UPDATED: "permissions_agent_updated"};
+  PERMISSIONS_AGENT_UPDATED: "permissions_agent_updated",
+  // Task events
+  TASK_CREATED: "task.created",
+  TASK_ASSIGNED: "task.assigned",
+  TASK_STATUS: "task.status",
+  TASK_COMPLETED: "task.completed"};
 var CORE_METHODS = {
   CONNECT: "map/connect",
   DISCONNECT: "map/disconnect",
@@ -217,6 +222,12 @@ var MAIL_METHODS = {
   MAIL_SUMMARY: "mail/summary",
   MAIL_REPLAY: "mail/replay"
 };
+var TASK_METHODS = {
+  TASKS_CREATE: "map/tasks/create",
+  TASKS_ASSIGN: "map/tasks/assign",
+  TASKS_UPDATE: "map/tasks/update",
+  TASKS_LIST: "map/tasks/list"
+};
 var NOTIFICATION_METHODS = {
   EVENT: "map/event",
   MESSAGE: "map/message",
@@ -1165,6 +1176,7 @@ var TestServer = class {
   #agents = /* @__PURE__ */ new Map();
   #scopes = /* @__PURE__ */ new Map();
   #subscriptions = /* @__PURE__ */ new Map();
+  #tasks = /* @__PURE__ */ new Map();
   #messages = [];
   #eventHistory = [];
   #maxEventHistory;
@@ -1179,6 +1191,7 @@ var TestServer = class {
   #nextScopeId = 1;
   #nextSubscriptionId = 1;
   #nextMessageId = 1;
+  #nextTaskId = 1;
   constructor(options = {}) {
     this.#options = options;
     this.#sessionId = `session-${Date.now()}`;
@@ -1210,6 +1223,12 @@ var TestServer = class {
   get scopes() {
     return this.#scopes;
   }
+  /**
+   * Get all tasks
+   */
+  get tasks() {
+    return this.#tasks;
+  }
   /**
    * Get all messages sent through the server
    */
@@ -1367,6 +1386,17 @@ var TestServer = class {
           connection,
           params
         );
+      // =======================================================================
+      // Task Methods
+      // =======================================================================
+      case TASK_METHODS.TASKS_CREATE:
+        return this.#handleTasksCreate(params);
+      case TASK_METHODS.TASKS_ASSIGN:
+        return this.#handleTasksAssign(params);
+      case TASK_METHODS.TASKS_UPDATE:
+        return this.#handleTasksUpdate(params);
+      case TASK_METHODS.TASKS_LIST:
+        return this.#handleTasksList(params);
       default:
         throw MAPRequestError.methodNotFound(method);
     }
@@ -2185,6 +2215,94 @@ var TestServer = class {
       effectivePermissions
     };
   }
+  // ===========================================================================
+  // Task Handlers
+  // ===========================================================================
+  #handleTasksCreate(params) {
+    const { task: taskInput } = params;
+    const id = taskInput.id ?? `task-${this.#nextTaskId++}`;
+    const task = {
+      id,
+      ...taskInput.assignee !== void 0 && { assignee: taskInput.assignee },
+      ...taskInput.title !== void 0 && { title: taskInput.title },
+      ...taskInput.status !== void 0 && { status: taskInput.status },
+      ...taskInput.description !== void 0 && { description: taskInput.description },
+      ...taskInput.meta !== void 0 && { meta: taskInput.meta }
+    };
+    this.#tasks.set(id, task);
+    this.emitEvent({
+      type: EVENT_TYPES.TASK_CREATED,
+      data: { task }
+    });
+    return { task };
+  }
+  #handleTasksAssign(params) {
+    const { taskId, agentId } = params;
+    const task = this.#tasks.get(taskId);
+    if (!task) {
+      throw MAPRequestError.invalidParams(`Task ${taskId} not found`);
+    }
+    task.assignee = agentId;
+    this.#tasks.set(taskId, task);
+    this.emitEvent({
+      type: EVENT_TYPES.TASK_ASSIGNED,
+      data: { taskId, agentId }
+    });
+    return { task };
+  }
+  #handleTasksUpdate(params) {
+    const { taskId, ...updates } = params;
+    const task = this.#tasks.get(taskId);
+    if (!task) {
+      throw MAPRequestError.invalidParams(`Task ${taskId} not found`);
+    }
+    const previousStatus = task.status;
+    if (updates.status !== void 0) task.status = updates.status;
+    if (updates.title !== void 0) task.title = updates.title;
+    if (updates.description !== void 0) task.description = updates.description;
+    if (updates.assignee !== void 0) task.assignee = updates.assignee;
+    if (updates.meta !== void 0) task.meta = { ...task.meta, ...updates.meta };
+    this.#tasks.set(taskId, task);
+    if (updates.status !== void 0 && updates.status !== previousStatus) {
+      this.emitEvent({
+        type: EVENT_TYPES.TASK_STATUS,
+        data: {
+          taskId,
+          previous: previousStatus ?? "open",
+          current: updates.status
+        }
+      });
+    }
+    if (updates.status === "completed") {
+      this.emitEvent({
+        type: EVENT_TYPES.TASK_COMPLETED,
+        data: { taskId }
+      });
+    }
+    return { task };
+  }
+  #handleTasksList(params) {
+    let tasks = Array.from(this.#tasks.values());
+    if (params?.filter) {
+      const { assignee, status } = params.filter;
+      if (assignee) {
+        tasks = tasks.filter((t) => t.assignee === assignee);
+      }
+      if (status) {
+        const statuses = Array.isArray(status) ? status : [status];
+        tasks = tasks.filter((t) => t.status !== void 0 && statuses.includes(t.status));
+      }
+    }
+    const limit = params?.limit ?? 100;
+    const startIndex = params?.cursor ? parseInt(params.cursor, 10) : 0;
+    const slice = tasks.slice(startIndex, startIndex + limit);
+    const hasMore = startIndex + limit < tasks.length;
+    return {
+      tasks: slice,
+      hasMore,
+      nextCursor: hasMore ? String(startIndex + limit) : void 0
+    };
+  }
 };
 
 // src/stream/index.ts
@@ -4240,6 +4358,58 @@ var ClientConnection = class _ClientConnection {
     });
   }
   // ===========================================================================
+  // Tasks
+  // ===========================================================================
+  /**
+   * Create a new task.
+   *
+   * @param params - Task creation parameters
+   * @returns The created task
+   */
+  async createTask(params) {
+    return this.#connection.sendRequest(
+      TASK_METHODS.TASKS_CREATE,
+      params
+    );
+  }
+  /**
+   * Assign a task to an agent.
+   *
+   * @param taskId - ID of the task to assign
+   * @param agentId - ID of the agent to assign to
+   * @returns The updated task
+   */
+  async assignTask(taskId, agentId) {
+    return this.#connection.sendRequest(
+      TASK_METHODS.TASKS_ASSIGN,
+      { taskId, agentId }
+    );
+  }
+  /**
+   * Update a task's status or fields.
+   *
+   * @param params - Update parameters including taskId and fields to change
+   * @returns The updated task
+   */
+  async updateTask(params) {
+    return this.#connection.sendRequest(
+      TASK_METHODS.TASKS_UPDATE,
+      params
+    );
+  }
+  /**
+   * List tasks with optional filters.
+   *
+   * @param params - Optional filter, limit, and cursor parameters
+   * @returns Paginated list of tasks
+   */
+  async listTasks(params) {
+    return this.#connection.sendRequest(
+      TASK_METHODS.TASKS_LIST,
+      params ?? {}
+    );
+  }
+  // ===========================================================================
   // Reconnection
   // ===========================================================================
   /**
@@ -5382,6 +5552,22 @@ var AgentConnection = class _AgentConnection {
     });
   }
   // ===========================================================================
+  // Extensions
+  // ===========================================================================
+  /**
+   * Call an extension method on the MAP server.
+   *
+   * Sends a raw JSON-RPC request with the given method and params.
+   * Use this for protocol extensions like trajectory, macro-agent, etc.
+   *
+   * @param method - The extension method name (e.g., 'trajectory/checkpoint')
+   * @param params - Optional parameters for the method
+   * @returns The result from the server
+   */
+  async callExtension(method, params) {
+    return await this.#connection.sendRequest(method, params);
+  }
+  // ===========================================================================
   // Internal
   // ===========================================================================
   /**