ai 5.0.0-canary.0 → 5.0.0-canary.10

package/mcp-stdio/mcp-stdio-transport.test.ts

@@ -1,262 +0,0 @@
-import type { ChildProcess } from 'node:child_process';
-import { EventEmitter } from 'node:events';
-import { beforeEach, describe, expect, it, vi } from 'vitest';
-import { JSONRPCMessage } from '../core/tool/mcp/json-rpc-message';
-import { MCPClientError } from '../errors';
-import { createChildProcess } from './create-child-process';
-import { StdioMCPTransport } from './mcp-stdio-transport';
-
-vi.mock('./create-child-process', { spy: true });
-
-interface MockChildProcess {
-  stdin: EventEmitter & { write?: ReturnType<typeof vi.fn> };
-  stdout: EventEmitter;
-  stderr: EventEmitter;
-  on: ReturnType<typeof vi.fn>;
-  removeAllListeners: ReturnType<typeof vi.fn>;
-}
-
-describe('StdioMCPTransport', () => {
-  let transport: StdioMCPTransport;
-  let mockChildProcess: MockChildProcess;
-  let mockStdin: EventEmitter & { write?: ReturnType<typeof vi.fn> };
-  let mockStdout: EventEmitter;
-
-  beforeEach(() => {
-    vi.clearAllMocks();
-
-    mockStdin = new EventEmitter();
-    mockStdout = new EventEmitter();
-    mockChildProcess = {
-      stdin: mockStdin,
-      stdout: mockStdout,
-      stderr: new EventEmitter(),
-      on: vi.fn(),
-      removeAllListeners: vi.fn(),
-    };
-
-    vi.mocked(createChildProcess).mockResolvedValue(
-      mockChildProcess as unknown as ChildProcess,
-    );
-
-    transport = new StdioMCPTransport({
-      command: 'test-command',
-      args: ['--test'],
-    });
-  });
-
-  afterEach(() => {
-    transport.close();
-  });
-
-  describe('start', () => {
-    it('should successfully start the transport', async () => {
-      const stdinOnSpy = vi.spyOn(mockStdin, 'on');
-      const stdoutOnSpy = vi.spyOn(mockStdout, 'on');
-
-      mockChildProcess.on.mockImplementation(
-        (event: string, callback: () => void) => {
-          if (event === 'spawn') {
-            callback();
-          }
-        },
-      );
-
-      const startPromise = transport.start();
-      await expect(startPromise).resolves.toBeUndefined();
-
-      expect(mockChildProcess.on).toHaveBeenCalledWith(
-        'error',
-        expect.any(Function),
-      );
-      expect(mockChildProcess.on).toHaveBeenCalledWith(
-        'spawn',
-        expect.any(Function),
-      );
-      expect(mockChildProcess.on).toHaveBeenCalledWith(
-        'close',
-        expect.any(Function),
-      );
-
-      expect(stdinOnSpy).toHaveBeenCalledWith('error', expect.any(Function));
-      expect(stdoutOnSpy).toHaveBeenCalledWith('error', expect.any(Function));
-      expect(stdoutOnSpy).toHaveBeenCalledWith('data', expect.any(Function));
-    });
-
-    it('should throw error if already started', async () => {
-      mockChildProcess.on.mockImplementation(
-        (event: string, callback: () => void) => {
-          if (event === 'spawn') {
-            callback();
-          }
-        },
-      );
-      const firstStart = transport.start();
-      await expect(firstStart).resolves.toBeUndefined();
-      const secondStart = transport.start();
-      await expect(secondStart).rejects.toThrow(MCPClientError);
-    });
-
-    it('should handle spawn errors', async () => {
-      const error = new Error('Spawn failed');
-      const onErrorSpy = vi.fn();
-      transport.onerror = onErrorSpy;
-
-      // simulate `spawn` failure by emitting error event after returning child process
-      mockChildProcess.on.mockImplementation(
-        (event: string, callback: (err: Error) => void) => {
-          if (event === 'error') {
-            callback(error);
-          }
-        },
-      );
-
-      const startPromise = transport.start();
-      await expect(startPromise).rejects.toThrow('Spawn failed');
-      expect(onErrorSpy).toHaveBeenCalledWith(error);
-    });
-
-    it('should handle child_process import errors', async () => {
-      vi.mocked(createChildProcess).mockRejectedValue(
-        new MCPClientError({
-          message: 'Failed to load child_process module dynamically',
-        }),
-      );
-
-      const startPromise = transport.start();
-      await expect(startPromise).rejects.toThrow(
-        'Failed to load child_process module dynamically',
-      );
-    });
-  });
-
-  describe('send', () => {
-    beforeEach(async () => {
-      mockChildProcess.on.mockImplementation(
-        (event: string, callback: () => void) => {
-          if (event === 'spawn') {
-            callback();
-          }
-        },
-      );
-      await transport.start();
-    });
-
-    it('should successfully send a message', async () => {
-      const message: JSONRPCMessage = {
-        jsonrpc: '2.0',
-        id: '1',
-        method: 'test',
-        params: {},
-      };
-
-      mockStdin.write = vi.fn().mockReturnValue(true);
-
-      await transport.send(message);
-
-      expect(mockStdin.write).toHaveBeenCalledWith(
-        JSON.stringify(message) + '\n',
-      );
-    });
-
-    it('should handle write backpressure', async () => {
-      const message: JSONRPCMessage = {
-        jsonrpc: '2.0',
-        id: '1',
-        method: 'test',
-        params: {},
-      };
-
-      mockStdin.write = vi.fn().mockReturnValue(false);
-
-      const sendPromise = transport.send(message);
-
-      mockStdin.emit('drain');
-
-      await expect(sendPromise).resolves.toBeUndefined();
-    });
-
-    it('should throw error if transport is not connected', async () => {
-      await transport.close();
-
-      const message: JSONRPCMessage = {
-        jsonrpc: '2.0',
-        id: '1',
-        method: 'test',
-        params: {},
-      };
-
-      await expect(transport.send(message)).rejects.toThrow(MCPClientError);
-    });
-  });
-
-  describe('message handling', () => {
-    const onMessageSpy = vi.fn();
-
-    beforeEach(async () => {
-      mockChildProcess.on.mockImplementation(
-        (event: string, callback: () => void) => {
-          if (event === 'spawn') {
-            callback();
-          }
-        },
-      );
-      transport.onmessage = onMessageSpy;
-      await transport.start();
-    });
-
-    it('should handle incoming messages correctly', async () => {
-      const message: JSONRPCMessage = {
-        jsonrpc: '2.0',
-        id: '1',
-        method: 'test',
-        params: {},
-      };
-
-      mockStdout.emit('data', Buffer.from(JSON.stringify(message) + '\n'));
-      expect(onMessageSpy).toHaveBeenCalledWith(message);
-    });
-
-    it('should handle partial messages correctly', async () => {
-      const message = {
-        jsonrpc: '2.0',
-        id: '1',
-        method: 'test',
-        params: {},
-      };
-
-      const messageStr = JSON.stringify(message);
-      mockStdout.emit('data', Buffer.from(messageStr.slice(0, 10)));
-      mockStdout.emit('data', Buffer.from(messageStr.slice(10) + '\n'));
-      expect(onMessageSpy).toHaveBeenCalledWith(message);
-    });
-  });
-
-  describe('close', () => {
-    const onCloseSpy = vi.fn();
-
-    beforeEach(async () => {
-      mockChildProcess.on.mockImplementation(
-        (event: string, callback: (code?: number) => void) => {
-          if (event === 'spawn') {
-            callback();
-          } else if (event === 'close') {
-            callback(0);
-          }
-        },
-      );
-      transport.onclose = onCloseSpy;
-      await transport.start();
-    });
-
-    it('should close the transport successfully', async () => {
-      await transport.close();
-
-      expect(mockChildProcess.on).toHaveBeenCalledWith(
-        'close',
-        expect.any(Function),
-      );
-      expect(onCloseSpy).toHaveBeenCalled();
-    });
-  });
-});

package/mcp-stdio/mcp-stdio-transport.ts

@@ -1,157 +0,0 @@
-import type { ChildProcess, IOType } from 'node:child_process';
-import { Stream } from 'node:stream';
-import {
-  JSONRPCMessage,
-  JSONRPCMessageSchema,
-} from '../core/tool/mcp/json-rpc-message';
-import { MCPTransport } from '../core/tool/mcp/mcp-transport';
-import { MCPClientError } from '../errors';
-import { createChildProcess } from './create-child-process';
-
-export interface StdioConfig {
-  command: string;
-  args?: string[];
-  env?: Record<string, string>;
-  stderr?: IOType | Stream | number;
-  cwd?: string;
-}
-
-export class StdioMCPTransport implements MCPTransport {
-  private process?: ChildProcess;
-  private abortController: AbortController = new AbortController();
-  private readBuffer: ReadBuffer = new ReadBuffer();
-  private serverParams: StdioConfig;
-
-  onclose?: () => void;
-  onerror?: (error: unknown) => void;
-  onmessage?: (message: JSONRPCMessage) => void;
-
-  constructor(server: StdioConfig) {
-    this.serverParams = server;
-  }
-
-  async start(): Promise<void> {
-    if (this.process) {
-      throw new MCPClientError({
-        message: 'StdioMCPTransport already started.',
-      });
-    }
-
-    return new Promise(async (resolve, reject) => {
-      try {
-        const process = await createChildProcess(
-          this.serverParams,
-          this.abortController.signal,
-        );
-
-        this.process = process;
-
-        this.process.on('error', error => {
-          if (error.name === 'AbortError') {
-            this.onclose?.();
-            return;
-          }
-
-          reject(error);
-          this.onerror?.(error);
-        });
-
-        this.process.on('spawn', () => {
-          resolve();
-        });
-
-        this.process.on('close', _code => {
-          this.process = undefined;
-          this.onclose?.();
-        });
-
-        this.process.stdin?.on('error', error => {
-          this.onerror?.(error);
-        });
-
-        this.process.stdout?.on('data', chunk => {
-          this.readBuffer.append(chunk);
-          this.processReadBuffer();
-        });
-
-        this.process.stdout?.on('error', error => {
-          this.onerror?.(error);
-        });
-      } catch (error) {
-        reject(error);
-        this.onerror?.(error);
-      }
-    });
-  }
-
-  private processReadBuffer() {
-    while (true) {
-      try {
-        const message = this.readBuffer.readMessage();
-        if (message === null) {
-          break;
-        }
-
-        this.onmessage?.(message);
-      } catch (error) {
-        this.onerror?.(error as Error);
-      }
-    }
-  }
-
-  async close(): Promise<void> {
-    this.abortController.abort();
-    this.process = undefined;
-    this.readBuffer.clear();
-  }
-
-  send(message: JSONRPCMessage): Promise<void> {
-    return new Promise(resolve => {
-      if (!this.process?.stdin) {
-        throw new MCPClientError({
-          message: 'StdioClientTransport not connected',
-        });
-      }
-
-      const json = serializeMessage(message);
-      if (this.process.stdin.write(json)) {
-        resolve();
-      } else {
-        this.process.stdin.once('drain', resolve);
-      }
-    });
-  }
-}
-
-class ReadBuffer {
-  private buffer?: Buffer;
-
-  append(chunk: Buffer): void {
-    this.buffer = this.buffer ? Buffer.concat([this.buffer, chunk]) : chunk;
-  }
-
-  readMessage(): JSONRPCMessage | null {
-    if (!this.buffer) return null;
-
-    const index = this.buffer.indexOf('\n');
-    if (index === -1) {
-      return null;
-    }
-
-    const line = this.buffer.toString('utf8', 0, index);
-    this.buffer = this.buffer.subarray(index + 1);
-    return deserializeMessage(line);
-  }
-
-  clear(): void {
-    this.buffer = undefined;
-  }
-}
-
-function serializeMessage(message: JSONRPCMessage): string {
-  return JSON.stringify(message) + '\n';
-}
-
-export function deserializeMessage(line: string): JSONRPCMessage {
-  return JSONRPCMessageSchema.parse(JSON.parse(line));
-}

package/react/dist/index.d.mts

@@ -1,18 +0,0 @@
-import * as _ai_sdk_react from '@ai-sdk/react';
-import { useChat as useChat$1, useCompletion as useCompletion$1 } from '@ai-sdk/react';
-export { CreateMessage, Message, UseChatHelpers, UseChatOptions } from '@ai-sdk/react';
-
-/**
- * @deprecated Use `@ai-sdk/react` instead.
- */
-declare const useChat: typeof useChat$1;
-/**
- * @deprecated Use `@ai-sdk/react` instead.
- */
-declare const useCompletion: typeof useCompletion$1;
-/**
- * @deprecated Use `@ai-sdk/react` instead.
- */
-declare const experimental_useObject: <RESULT, INPUT = any>({ api, id, schema, initialValue, fetch, onError, onFinish, headers, credentials, }: _ai_sdk_react.Experimental_UseObjectOptions<RESULT>) => _ai_sdk_react.Experimental_UseObjectHelpers<RESULT, INPUT>;
-
-export { experimental_useObject, useChat, useCompletion };

package/react/dist/index.d.ts

@@ -1,18 +0,0 @@
-import * as _ai_sdk_react from '@ai-sdk/react';
-import { useChat as useChat$1, useCompletion as useCompletion$1 } from '@ai-sdk/react';
-export { CreateMessage, Message, UseChatHelpers, UseChatOptions } from '@ai-sdk/react';
-
-/**
- * @deprecated Use `@ai-sdk/react` instead.
- */
-declare const useChat: typeof useChat$1;
-/**
- * @deprecated Use `@ai-sdk/react` instead.
- */
-declare const useCompletion: typeof useCompletion$1;
-/**
- * @deprecated Use `@ai-sdk/react` instead.
- */
-declare const experimental_useObject: <RESULT, INPUT = any>({ api, id, schema, initialValue, fetch, onError, onFinish, headers, credentials, }: _ai_sdk_react.Experimental_UseObjectOptions<RESULT>) => _ai_sdk_react.Experimental_UseObjectHelpers<RESULT, INPUT>;
-
-export { experimental_useObject, useChat, useCompletion };

package/react/dist/index.js

@@ -1,39 +0,0 @@
-'use client'
-"use strict";
-var __defProp = Object.defineProperty;
-var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
-var __getOwnPropNames = Object.getOwnPropertyNames;
-var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __export = (target, all) => {
-  for (var name in all)
-    __defProp(target, name, { get: all[name], enumerable: true });
-};
-var __copyProps = (to, from, except, desc) => {
-  if (from && typeof from === "object" || typeof from === "function") {
-    for (let key of __getOwnPropNames(from))
-      if (!__hasOwnProp.call(to, key) && key !== except)
-        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
-  }
-  return to;
-};
-var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
-
-// react/index.ts
-var react_exports = {};
-__export(react_exports, {
-  experimental_useObject: () => experimental_useObject,
-  useChat: () => useChat,
-  useCompletion: () => useCompletion
-});
-module.exports = __toCommonJS(react_exports);
-var import_react = require("@ai-sdk/react");
-var useChat = import_react.useChat;
-var useCompletion = import_react.useCompletion;
-var experimental_useObject = import_react.experimental_useObject;
-// Annotate the CommonJS export names for ESM import in node:
-0 && (module.exports = {
-  experimental_useObject,
-  useChat,
-  useCompletion
-});
-//# sourceMappingURL=index.js.map

package/react/dist/index.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../index.ts"],"sourcesContent":["import {\n  useChat as useChatReact,\n  useCompletion as useCompletionReact,\n  experimental_useObject as experimental_useObjectReact,\n} from '@ai-sdk/react';\n\n/**\n * @deprecated Use `@ai-sdk/react` instead.\n */\nexport const useChat = useChatReact;\n\n/**\n * @deprecated Use `@ai-sdk/react` instead.\n */\nexport const useCompletion = useCompletionReact;\n\n/**\n * @deprecated Use `@ai-sdk/react` instead.\n */\nexport const experimental_useObject = experimental_useObjectReact;\n\nexport type {\n  /**\n   * @deprecated Use `@ai-sdk/react` instead.\n   */\n  CreateMessage,\n\n  /**\n   * @deprecated Use `@ai-sdk/react` instead.\n   */\n  Message,\n\n  /**\n   * @deprecated Use `@ai-sdk/react` instead.\n   */\n  UseChatOptions,\n\n  /**\n   * @deprecated Use `@ai-sdk/react` instead.\n   */\n  UseChatHelpers,\n} from '@ai-sdk/react';\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,mBAIO;AAKA,IAAM,UAAU,aAAAA;AAKhB,IAAM,gBAAgB,aAAAC;AAKtB,IAAM,yBAAyB,aAAAC;","names":["useChatReact","useCompletionReact","experimental_useObjectReact"]}

package/react/dist/index.mjs

@@ -1,17 +0,0 @@
-'use client'
-
-// react/index.ts
-import {
-  useChat as useChatReact,
-  useCompletion as useCompletionReact,
-  experimental_useObject as experimental_useObjectReact
-} from "@ai-sdk/react";
-var useChat = useChatReact;
-var useCompletion = useCompletionReact;
-var experimental_useObject = experimental_useObjectReact;
-export {
-  experimental_useObject,
-  useChat,
-  useCompletion
-};
-//# sourceMappingURL=index.mjs.map

package/react/dist/index.mjs.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../index.ts"],"sourcesContent":["import {\n  useChat as useChatReact,\n  useCompletion as useCompletionReact,\n  experimental_useObject as experimental_useObjectReact,\n} from '@ai-sdk/react';\n\n/**\n * @deprecated Use `@ai-sdk/react` instead.\n */\nexport const useChat = useChatReact;\n\n/**\n * @deprecated Use `@ai-sdk/react` instead.\n */\nexport const useCompletion = useCompletionReact;\n\n/**\n * @deprecated Use `@ai-sdk/react` instead.\n */\nexport const experimental_useObject = experimental_useObjectReact;\n\nexport type {\n  /**\n   * @deprecated Use `@ai-sdk/react` instead.\n   */\n  CreateMessage,\n\n  /**\n   * @deprecated Use `@ai-sdk/react` instead.\n   */\n  Message,\n\n  /**\n   * @deprecated Use `@ai-sdk/react` instead.\n   */\n  UseChatOptions,\n\n  /**\n   * @deprecated Use `@ai-sdk/react` instead.\n   */\n  UseChatHelpers,\n} from '@ai-sdk/react';\n"],"mappings":";;;AAAA;AAAA,EACE,WAAW;AAAA,EACX,iBAAiB;AAAA,EACjB,0BAA0B;AAAA,OACrB;AAKA,IAAM,UAAU;AAKhB,IAAM,gBAAgB;AAKtB,IAAM,yBAAyB;","names":[]}