@mastra/client-js 0.11.3-alpha.2 → 0.11.3-alpha.4

package/src/adapters/agui.test.ts

@@ -1,293 +0,0 @@
-import type { Message, BaseEvent } from '@ag-ui/client';
-import { describe, it, expect, vi } from 'vitest';
-import { generateUUID, convertMessagesToMastraMessages, AGUIAdapter } from './agui';
-
-describe('generateUUID', () => {
-  it('should generate a valid UUID v4 string', () => {
-    const uuid = generateUUID();
-    // Check UUID format (8-4-4-4-12 hex digits)
-    expect(uuid).toMatch(/^[0-9a-f]{8}-[0-9a-f]{4}-4[0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$/i);
-  });
-
-  it('should generate unique UUIDs', () => {
-    const uuids = new Set();
-    for (let i = 0; i < 100; i++) {
-      uuids.add(generateUUID());
-    }
-    // All UUIDs should be unique
-    expect(uuids.size).toBe(100);
-  });
-});
-
-describe('convertMessagesToMastraMessages', () => {
-  it('should convert user messages correctly', () => {
-    const messages: Message[] = [
-      {
-        id: '1',
-        role: 'user',
-        content: 'Hello, world!',
-      },
-    ];
-
-    const result = convertMessagesToMastraMessages(messages);
-
-    expect(result).toEqual([
-      {
-        role: 'user',
-        content: 'Hello, world!',
-      },
-    ]);
-  });
-
-  it('should convert assistant messages correctly', () => {
-    const messages: Message[] = [
-      {
-        id: '1',
-        role: 'assistant',
-        content: 'Hello, I am an assistant',
-      },
-    ];
-
-    const result = convertMessagesToMastraMessages(messages);
-
-    expect(result).toEqual([
-      {
-        role: 'assistant',
-        content: [{ type: 'text', text: 'Hello, I am an assistant' }],
-      },
-    ]);
-  });
-
-  it('should convert assistant messages with tool calls correctly', () => {
-    const messages: Message[] = [
-      {
-        id: '1',
-        role: 'assistant',
-        content: undefined,
-        toolCalls: [
-          {
-            id: 'tool-call-1',
-            type: 'function',
-            function: {
-              name: 'getWeather',
-              arguments: '{"location":"San Francisco"}',
-            },
-          },
-        ],
-      },
-    ];
-
-    const result = convertMessagesToMastraMessages(messages);
-
-    expect(result).toEqual([
-      {
-        role: 'assistant',
-        content: [
-          {
-            type: 'tool-call',
-            toolCallId: 'tool-call-1',
-            toolName: 'getWeather',
-            args: { location: 'San Francisco' },
-          },
-        ],
-      },
-      {
-        role: 'tool',
-        content: [
-          {
-            type: 'tool-result',
-            toolCallId: 'tool-call-1',
-            toolName: 'getWeather',
-            result: { location: 'San Francisco' },
-          },
-        ],
-      },
-    ]);
-  });
-
-  it('should convert tool messages correctly', () => {
-    const messages: Message[] = [
-      {
-        id: '1',
-        role: 'tool',
-        toolCallId: 'tool-call-1',
-        content: '{"temperature":72,"unit":"F"}',
-      },
-    ];
-
-    const result = convertMessagesToMastraMessages(messages);
-
-    expect(result).toEqual([
-      {
-        role: 'tool',
-        content: [
-          {
-            type: 'tool-result',
-            toolCallId: 'tool-call-1',
-            toolName: 'unknown',
-            result: '{"temperature":72,"unit":"F"}',
-          },
-        ],
-      },
-    ]);
-  });
-
-  it('should convert a complex conversation correctly', () => {
-    const messages: Message[] = [
-      {
-        id: '1',
-        role: 'user',
-        content: "What's the weather in San Francisco?",
-      },
-      {
-        id: '2',
-        role: 'assistant',
-        content: undefined,
-        toolCalls: [
-          {
-            id: 'tool-call-1',
-            type: 'function',
-            function: {
-              name: 'getWeather',
-              arguments: '{"location":"San Francisco"}',
-            },
-          },
-        ],
-      },
-      {
-        id: '4',
-        role: 'assistant',
-        content: 'The weather in San Francisco is 72°F.',
-      },
-    ];
-
-    const result = convertMessagesToMastraMessages(messages);
-
-    expect(result).toHaveLength(4);
-    expect(result[0].role).toBe('user');
-    expect(result[1].role).toBe('assistant');
-    expect(result[2].role).toBe('tool');
-    expect(result[2].content).toEqual([
-      {
-        type: 'tool-result',
-        toolCallId: 'tool-call-1',
-        toolName: 'getWeather',
-        result: { location: 'San Francisco' },
-      },
-    ]);
-    expect(result[3].role).toBe('assistant');
-  });
-});
-
-describe('AGUIAdapter', () => {
-  it('should correctly pass parameters to agent stream method', async () => {
-    // Create a mock client agent that simulates the expected behavior
-    const clientAgent = {
-      stream: vi.fn().mockImplementation(async (params: any) => {
-        // Verify the parameters are passed correctly
-        expect(params).toHaveProperty('messages');
-        expect(params).toHaveProperty('threadId');
-        expect(params).toHaveProperty('resourceId');
-        expect(params).toHaveProperty('runId');
-        expect(params).toHaveProperty('clientTools');
-
-        // Verify that messages array is passed, not the entire request object
-        expect(Array.isArray(params.messages)).toBe(true);
-        expect(params.messages[0]).toHaveProperty('role');
-        expect(params.messages[0]).toHaveProperty('content');
-
-        // Return a mock processDataStream that mimics the expected behavior
-        return {
-          processDataStream: vi.fn().mockImplementation(async ({ onTextPart, onFinishMessagePart }: any) => {
-            // Simulate streaming text
-            if (onTextPart) {
-              onTextPart('Hello from agent');
-            }
-            if (onFinishMessagePart) {
-              onFinishMessagePart();
-            }
-            return Promise.resolve();
-          }),
-        };
-      }),
-    };
-
-    const adapter = new AGUIAdapter({
-      agent: clientAgent as any,
-      agentId: 'test',
-      resourceId: 'testAgent',
-    });
-
-    const input = {
-      threadId: 'test-thread-id',
-      runId: 'test-run-id',
-      messages: [
-        {
-          id: '1',
-          role: 'user' as const,
-          content: 'Hello',
-        },
-      ],
-      tools: [],
-      context: [],
-    };
-
-    const observable = adapter['run'](input);
-    const events: BaseEvent[] = [];
-
-    await new Promise<void>((resolve, reject) => {
-      observable.subscribe({
-        next: (event: BaseEvent) => events.push(event),
-        complete: () => resolve(),
-        error: (error: any) => reject(error),
-      });
-    });
-
-    // Verify we received the expected events
-    expect(events).toHaveLength(5); // RUN_STARTED, TEXT_MESSAGE_START, TEXT_MESSAGE_CONTENT, TEXT_MESSAGE_END, RUN_FINISHED
-    expect(events[0].type).toBe('RUN_STARTED');
-    expect(events[1].type).toBe('TEXT_MESSAGE_START');
-    expect(events[2].type).toBe('TEXT_MESSAGE_CONTENT');
-    expect(events[3].type).toBe('TEXT_MESSAGE_END');
-    expect(events[4].type).toBe('RUN_FINISHED');
-
-    // Verify the stream method was called with the correct parameters
-    expect(clientAgent.stream).toHaveBeenCalledWith({
-      threadId: 'test-thread-id',
-      resourceId: 'testAgent',
-      runId: 'test-run-id',
-      messages: [{ role: 'user', content: 'Hello' }],
-      clientTools: {},
-    });
-  });
-
-  it('should handle messages without role property in request objects', async () => {
-    // This test demonstrates that request objects without role property
-    // would cause validation errors if passed directly to MessageList
-    const requestObject = {
-      threadId: 'test-thread-id',
-      resourceId: 'testAgent',
-      runId: 'test-run-id',
-      messages: [
-        {
-          role: 'user',
-          content: 'Hello',
-        },
-      ],
-      clientTools: {},
-    };
-
-    // Request objects don't have role property
-    expect('role' in requestObject).toBe(false);
-    expect('messages' in requestObject).toBe(true);
-    expect('content' in requestObject).toBe(false);
-    expect('parts' in requestObject).toBe(false);
-
-    // This structure would cause validation errors if treated as a message
-    // because it lacks required message properties (role, content/parts)
-    const hasValidMessageStructure =
-      'role' in requestObject && ('content' in requestObject || 'parts' in requestObject);
-
-    expect(hasValidMessageStructure).toBe(false);
-  });
-});

package/src/adapters/agui.ts

@@ -1,257 +0,0 @@
-// Cross-platform UUID generation function
-import type {
-  AgentConfig,
-  BaseEvent,
-  Message,
-  RunAgentInput,
-  RunFinishedEvent,
-  RunStartedEvent,
-  TextMessageContentEvent,
-  TextMessageEndEvent,
-  TextMessageStartEvent,
-  ToolCallArgsEvent,
-  ToolCallEndEvent,
-  ToolCallStartEvent,
-} from '@ag-ui/client';
-import { AbstractAgent, EventType } from '@ag-ui/client';
-import type { CoreMessage } from '@mastra/core/llm';
-import { Observable } from 'rxjs';
-import type { Agent } from '../resources/agent';
-
-interface MastraAgentConfig extends AgentConfig {
-  agent: Agent;
-  agentId: string;
-  resourceId?: string;
-}
-
-export class AGUIAdapter extends AbstractAgent {
-  agent: Agent;
-  resourceId?: string;
-  constructor({ agent, agentId, resourceId, ...rest }: MastraAgentConfig) {
-    super({
-      agentId,
-      ...rest,
-    });
-    this.agent = agent;
-    this.resourceId = resourceId;
-  }
-
-  protected run(input: RunAgentInput): Observable<BaseEvent> {
-    return new Observable<BaseEvent>(subscriber => {
-      const convertedMessages = convertMessagesToMastraMessages(input.messages);
-      subscriber.next({
-        type: EventType.RUN_STARTED,
-        threadId: input.threadId,
-        runId: input.runId,
-      } as RunStartedEvent);
-
-      this.agent
-        .stream({
-          threadId: input.threadId,
-          resourceId: this.resourceId ?? '',
-          runId: input.runId,
-          messages: convertedMessages,
-          clientTools: input.tools.reduce(
-            (acc, tool) => {
-              acc[tool.name as string] = {
-                id: tool.name,
-                description: tool.description,
-                inputSchema: tool.parameters,
-              };
-              return acc;
-            },
-            {} as Record<string, any>,
-          ),
-        })
-        .then(response => {
-          let currentMessageId: string | undefined = undefined;
-          let isInTextMessage = false;
-          return response.processDataStream({
-            onTextPart: text => {
-              if (currentMessageId === undefined) {
-                currentMessageId = generateUUID();
-                const message: TextMessageStartEvent = {
-                  type: EventType.TEXT_MESSAGE_START,
-                  messageId: currentMessageId,
-                  role: 'assistant',
-                };
-                subscriber.next(message);
-                isInTextMessage = true;
-              }
-
-              const message: TextMessageContentEvent = {
-                type: EventType.TEXT_MESSAGE_CONTENT,
-                messageId: currentMessageId,
-                delta: text,
-              };
-              subscriber.next(message);
-            },
-            onFinishMessagePart: () => {
-              if (currentMessageId !== undefined) {
-                const message: TextMessageEndEvent = {
-                  type: EventType.TEXT_MESSAGE_END,
-                  messageId: currentMessageId,
-                };
-                subscriber.next(message);
-                isInTextMessage = false;
-              }
-              // Emit run finished event
-              subscriber.next({
-                type: EventType.RUN_FINISHED,
-                threadId: input.threadId,
-                runId: input.runId,
-              } as RunFinishedEvent);
-
-              // Complete the observable
-              subscriber.complete();
-            },
-            onToolCallPart(streamPart) {
-              const parentMessageId = currentMessageId || generateUUID();
-              if (isInTextMessage) {
-                const message: TextMessageEndEvent = {
-                  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,
-              } as ToolCallStartEvent);
-
-              subscriber.next({
-                type: EventType.TOOL_CALL_ARGS,
-                toolCallId: streamPart.toolCallId,
-                delta: JSON.stringify(streamPart.args),
-                parentMessageId,
-              } as ToolCallArgsEvent);
-
-              subscriber.next({
-                type: EventType.TOOL_CALL_END,
-                toolCallId: streamPart.toolCallId,
-                parentMessageId,
-              } as ToolCallEndEvent);
-            },
-          });
-        })
-        .catch(error => {
-          console.error('error', error);
-          // Handle error
-          subscriber.error(error);
-        });
-
-      return () => {};
-    });
-  }
-}
-
-/**
- * Generates a UUID v4 that works in both browser and Node.js environments
- */
-export function generateUUID(): string {
-  // Use crypto.randomUUID() if available (Node.js environment or modern browsers)
-  if (typeof crypto !== 'undefined') {
-    // Browser crypto API or Node.js crypto global
-    if (typeof crypto.randomUUID === 'function') {
-      return crypto.randomUUID();
-    }
-    // Fallback for older browsers
-    if (typeof crypto.getRandomValues === 'function') {
-      const buffer = new Uint8Array(16);
-      crypto.getRandomValues(buffer);
-      // Set version (4) and variant (8, 9, A, or B)
-      buffer[6] = (buffer[6]! & 0x0f) | 0x40; // version 4
-      buffer[8] = (buffer[8]! & 0x3f) | 0x80; // variant
-
-      // Convert to hex string in UUID format
-      let hex = '';
-      for (let i = 0; i < 16; i++) {
-        hex += buffer[i]!.toString(16).padStart(2, '0');
-        // Add hyphens at standard positions
-        if (i === 3 || i === 5 || i === 7 || i === 9) hex += '-';
-      }
-      return hex;
-    }
-  }
-
-  // Last resort fallback (less secure but works everywhere)
-  return 'xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx'.replace(/[xy]/g, c => {
-    const r = (Math.random() * 16) | 0;
-    const v = c === 'x' ? r : (r & 0x3) | 0x8;
-    return v.toString(16);
-  });
-}
-
-export function convertMessagesToMastraMessages(messages: Message[]): CoreMessage[] {
-  const result: CoreMessage[] = [];
-
-  // First pass: identify which tool calls already have corresponding tool messages
-  const toolCallsWithResults = new Set<string>();
-  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: any[] = 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,
-      });
-
-      // Only create automatic tool results if there are no corresponding tool messages
-      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') {
-      // For tool messages from CopilotKit, we need to handle them properly
-      // CopilotKit sends tool messages as responses to tool calls
-      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;
-}