@mastra/client-js 0.11.3 → 0.12.0-alpha.0

package/dist/index.js

@@ -1,5 +1,3 @@
-import { AbstractAgent, EventType } from '@ag-ui/client';
-import { Observable } from 'rxjs';
 import { processDataStream, parsePartialJson } from '@ai-sdk/ui-utils';
 import { v4 } from '@lukeed/uuid';
 import { RuntimeContext } from '@mastra/core/runtime-context';
@@ -7,205 +5,7 @@ import { isVercelTool } from '@mastra/core/tools/is-vercel-tool';
 import { z } from 'zod';
 import originalZodToJsonSchema from 'zod-to-json-schema';
 
-// src/adapters/agui.ts
-var AGUIAdapter = class extends AbstractAgent {
-  agent;
-  resourceId;
-  constructor({ agent, agentId, resourceId, ...rest }) {
-    super({
-      agentId,
-      ...rest
-    });
-    this.agent = agent;
-    this.resourceId = resourceId;
-  }
-  run(input) {
-    return new Observable((subscriber) => {
-      const convertedMessages = convertMessagesToMastraMessages(input.messages);
-      subscriber.next({
-        type: EventType.RUN_STARTED,
-        threadId: input.threadId,
-        runId: input.runId
-      });
-      this.agent.stream({
-        threadId: input.threadId,
-        resourceId: this.resourceId ?? "",
-        runId: input.runId,
-        messages: convertedMessages,
-        clientTools: input.tools.reduce(
-          (acc, tool) => {
-            acc[tool.name] = {
-              id: tool.name,
-              description: tool.description,
-              inputSchema: tool.parameters
-            };
-            return acc;
-          },
-          {}
-        )
-      }).then((response) => {
-        let currentMessageId = void 0;
-        let isInTextMessage = false;
-        return response.processDataStream({
-          onTextPart: (text) => {
-            if (currentMessageId === void 0) {
-              currentMessageId = generateUUID();
-              const message2 = {
-                type: EventType.TEXT_MESSAGE_START,
-                messageId: currentMessageId,
-                role: "assistant"
-              };
-              subscriber.next(message2);
-              isInTextMessage = true;
-            }
-            const message = {
-              type: EventType.TEXT_MESSAGE_CONTENT,
-              messageId: currentMessageId,
-              delta: text
-            };
-            subscriber.next(message);
-          },
-          onFinishMessagePart: () => {
-            if (currentMessageId !== void 0) {
-              const message = {
-                type: EventType.TEXT_MESSAGE_END,
-                messageId: currentMessageId
-              };
-              subscriber.next(message);
-              isInTextMessage = false;
-            }
-            subscriber.next({
-              type: EventType.RUN_FINISHED,
-              threadId: input.threadId,
-              runId: input.runId
-            });
-            subscriber.complete();
-          },
-          onToolCallPart(streamPart) {
-            const parentMessageId = currentMessageId || generateUUID();
-            if (isInTextMessage) {
-              const message = {
-                type: EventType.TEXT_MESSAGE_END,
-                messageId: parentMessageId
-              };
-              subscriber.next(message);
-              isInTextMessage = false;
-            }
-            subscriber.next({
-              type: EventType.TOOL_CALL_START,
-              toolCallId: streamPart.toolCallId,
-              toolCallName: streamPart.toolName,
-              parentMessageId
-            });
-            subscriber.next({
-              type: EventType.TOOL_CALL_ARGS,
-              toolCallId: streamPart.toolCallId,
-              delta: JSON.stringify(streamPart.args),
-              parentMessageId
-            });
-            subscriber.next({
-              type: EventType.TOOL_CALL_END,
-              toolCallId: streamPart.toolCallId,
-              parentMessageId
-            });
-          }
-        });
-      }).catch((error) => {
-        console.error("error", error);
-        subscriber.error(error);
-      });
-      return () => {
-      };
-    });
-  }
-};
-function generateUUID() {
-  if (typeof crypto !== "undefined") {
-    if (typeof crypto.randomUUID === "function") {
-      return crypto.randomUUID();
-    }
-    if (typeof crypto.getRandomValues === "function") {
-      const buffer = new Uint8Array(16);
-      crypto.getRandomValues(buffer);
-      buffer[6] = buffer[6] & 15 | 64;
-      buffer[8] = buffer[8] & 63 | 128;
-      let hex = "";
-      for (let i = 0; i < 16; i++) {
-        hex += buffer[i].toString(16).padStart(2, "0");
-        if (i === 3 || i === 5 || i === 7 || i === 9) hex += "-";
-      }
-      return hex;
-    }
-  }
-  return "xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx".replace(/[xy]/g, (c) => {
-    const r = Math.random() * 16 | 0;
-    const v = c === "x" ? r : r & 3 | 8;
-    return v.toString(16);
-  });
-}
-function convertMessagesToMastraMessages(messages) {
-  const result = [];
-  const toolCallsWithResults = /* @__PURE__ */ new Set();
-  for (const message of messages) {
-    if (message.role === "tool" && message.toolCallId) {
-      toolCallsWithResults.add(message.toolCallId);
-    }
-  }
-  for (const message of messages) {
-    if (message.role === "assistant") {
-      const parts = message.content ? [{ type: "text", text: message.content }] : [];
-      for (const toolCall of message.toolCalls ?? []) {
-        parts.push({
-          type: "tool-call",
-          toolCallId: toolCall.id,
-          toolName: toolCall.function.name,
-          args: JSON.parse(toolCall.function.arguments)
-        });
-      }
-      result.push({
-        role: "assistant",
-        content: parts
-      });
-      if (message.toolCalls?.length) {
-        for (const toolCall of message.toolCalls) {
-          if (!toolCallsWithResults.has(toolCall.id)) {
-            result.push({
-              role: "tool",
-              content: [
-                {
-                  type: "tool-result",
-                  toolCallId: toolCall.id,
-                  toolName: toolCall.function.name,
-                  result: JSON.parse(toolCall.function.arguments)
-                  // This is still wrong but matches test expectations
-                }
-              ]
-            });
-          }
-        }
-      }
-    } else if (message.role === "user") {
-      result.push({
-        role: "user",
-        content: message.content || ""
-      });
-    } else if (message.role === "tool") {
-      result.push({
-        role: "tool",
-        content: [
-          {
-            type: "tool-result",
-            toolCallId: message.toolCallId || "unknown",
-            toolName: "unknown",
-            // toolName is not available in tool messages from CopilotKit
-            result: message.content
-          }
-        ]
-      });
-    }
-  }
-  return result;
-}
+// src/resources/agent.ts
 function parseClientRuntimeContext(runtimeContext) {
   if (runtimeContext) {
     if (runtimeContext instanceof RuntimeContext) {
@@ -2310,19 +2110,24 @@ var AgentBuilder = class extends BaseResource {
    * Creates a new agent builder action run and returns the runId.
    * This calls `/api/agent-builder/:actionId/create-run`.
    */
-  async createRun(params, runId) {
+  async createRun(params) {
     const searchParams = new URLSearchParams();
-    if (runId) {
-      searchParams.set("runId", runId);
+    if (!!params?.runId) {
+      searchParams.set("runId", params.runId);
     }
-    const runtimeContext = parseClientRuntimeContext(params.runtimeContext);
-    const { runtimeContext: _, ...actionParams } = params;
     const url = `/api/agent-builder/${this.actionId}/create-run${searchParams.toString() ? `?${searchParams.toString()}` : ""}`;
     return this.request(url, {
-      method: "POST",
-      body: { ...actionParams, runtimeContext }
+      method: "POST"
     });
   }
+  /**
+   * Creates a new workflow run (alias for createRun)
+   * @param params - Optional object containing the optional runId
+   * @returns Promise containing the runId of the created run
+   */
+  createRunAsync(params) {
+    return this.createRun(params);
+  }
   /**
    * Starts agent builder action asynchronously and waits for completion.
    * This calls `/api/agent-builder/:actionId/start-async`.
@@ -2536,8 +2341,8 @@ var AgentBuilder = class extends BaseResource {
    * and streams any remaining progress.
    * This calls `/api/agent-builder/:actionId/watch`.
    */
-  async watch({ runId }, onRecord) {
-    const url = `/api/agent-builder/${this.actionId}/watch?runId=${runId}`;
+  async watch({ runId, eventType }, onRecord) {
+    const url = `/api/agent-builder/${this.actionId}/watch?runId=${runId}${eventType ? `&eventType=${eventType}` : ""}`;
     const response = await this.request(url, {
       method: "GET",
       stream: true
@@ -2895,21 +2700,6 @@ var MastraClient = class extends BaseResource {
   getAgents() {
     return this.request("/api/agents");
   }
-  async getAGUI({ resourceId }) {
-    const agents = await this.getAgents();
-    return Object.entries(agents).reduce(
-      (acc, [agentId]) => {
-        const agent = this.getAgent(agentId);
-        acc[agentId] = new AGUIAdapter({
-          agentId,
-          agent,
-          resourceId
-        });
-        return acc;
-      },
-      {}
-    );
-  }
   /**
    * Gets an agent instance by ID
    * @param agentId - ID of the agent to retrieve