@opentiny/next-sdk 0.1.4 → 0.1.6

package/WebMcpClient.ts

@@ -1,6 +1,7 @@
 import { Client } from '@modelcontextprotocol/sdk/client/index.js'
 import { SSEClientTransport } from '@modelcontextprotocol/sdk/client/sse.js'
 import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/streamableHttp.js'
+import { WebSocketClientTransport } from '@modelcontextprotocol/sdk/client/websocket.js'
 import { z, ZodObject, ZodLiteral, ZodType } from 'zod'
 import {
   ElicitRequestSchema,
@@ -18,8 +19,9 @@ import {
   sseOptions,
   streamOptions,
   attemptConnection,
-  createStreamProxy,
   createSseProxy,
+  createStreamProxy,
+  createSocketProxy,
   AuthClientProvider
 } from '@opentiny/next'
 import type {
@@ -60,7 +62,7 @@ export interface ClientConnectOptions {
   token?: string
   sessionId?: string
   authProvider?: AuthClientProvider
-  type?: 'channel' | 'sse'
+  type?: 'channel' | 'sse' | 'stream' | 'socket'
   agent?: boolean
   onError?: (error: Error) => void
   onUnauthorized?: (connect: () => Promise<void>) => Promise<void>
@@ -126,7 +128,11 @@ export class WebMcpClient {
 
       const connectProxy = async () => {
         const { transport, sessionId } =
-          type === 'sse' ? await createSseProxy(proxyOptions) : await createStreamProxy(proxyOptions)
+          type === 'sse'
+            ? await createSseProxy(proxyOptions)
+            : type === 'socket'
+              ? await createSocketProxy(proxyOptions)
+              : await createStreamProxy(proxyOptions)
 
         transport.onerror = async (error: Error) => {
           onError?.(error)
@@ -169,6 +175,12 @@ export class WebMcpClient {
       }
     }
 
+    if (type === 'socket') {
+      transport = new WebSocketClientTransport(new URL(`${url}?sessionId=${sessionId}&token=${token}`))
+      transport.sessionId = sessionId
+      await this.client.connect(transport)
+    }
+
     if (typeof transport === 'undefined') {
       if (authProvider) {
         const createTransport = () => new StreamableHTTPClientTransport(endpoint, { authProvider })

package/agent/AgentModelProvider.ts

@@ -36,6 +36,8 @@ export class AgentModelProvider {
 
   /** chat 时，自动更新 所有的tools 后的事件 */
   onUpdatedTools: (() => void) | undefined
+  /** 内部报错时，抛出错误事件 */
+  onError: ((msg: string, err?: any) => void) | undefined
 
   constructor({ llmConfig, mcpServers, llm }: IAgentModelProviderOption) {
     // 1、保存 mcpServer
@@ -72,8 +74,11 @@ export class AgentModelProvider {
         transport = serverConfig as MCPClientConfig['transport']
       }
 
-      return createMCPClient({ transport: transport as MCPClientConfig['transport'] })
+      return await createMCPClient({ transport: transport as MCPClientConfig['transport'] })
     } catch (error) {
+      if (this.onError) {
+        this.onError(`Failed to create MCP client`, error)
+      }
       console.error(`Failed to create MCP client`, serverConfig, error)
       return null
     }
@@ -94,6 +99,10 @@ export class AgentModelProvider {
         try {
           return client ? await client?.tools?.() : null
         } catch (error) {
+          if (this.onError) {
+            this.onError(`Failed to query tools`, error)
+          }
+          console.error(`Failed to query tools`, error)
           return null
         }
       })
@@ -122,7 +131,7 @@ export class AgentModelProvider {
   }
 
   async insertMcpServer(mcpServer: McpServerConfig) {
-    const find = this.mcpServers.find((item: any) => item.url === mcpServer.url)
+    const find = this.mcpServers.find((item: any) => 'url' in item && 'url' in mcpServer && item.url === mcpServer.url)
 
     if (!find) {
       this.mcpServers = [...this.mcpServers, mcpServer]
@@ -167,7 +176,6 @@ export class AgentModelProvider {
     this.ignoreToolnames.forEach((name) => {
       delete toolsResult[name]
     })
-
     return toolsResult
   }
 
@@ -187,9 +195,9 @@ export class AgentModelProvider {
     return chatMethod({
       // @ts-ignore  ProviderV2 是所有llm的父类， 在每一个具体的llm 类都有一个选择model的函数用法
       model: this.llm(model),
-      tools: this.tempMergeTools(options.tools) as ToolSet,
       stopWhen: stepCountIs(maxSteps),
-      ...options
+      ...options,
+      tools: this.tempMergeTools(options.tools) as ToolSet
     })
   }
 

package/agent/type.ts

@@ -14,7 +14,7 @@ export interface IAgentModelProviderLlmConfig {
 }
 
 /** Mcp Server的配置对象 */
-export type McpServerConfig = { type: 'streamableHttp' | 'sse'; url: string } & { transport: MCPTransport }
+export type McpServerConfig = { type: 'streamableHttp' | 'sse'; url: string } | { transport: MCPTransport }
 
 /** */
 export interface IAgentModelProviderOption {

package/dist/WebMcpClient.d.ts

@@ -18,7 +18,7 @@ export interface ClientConnectOptions {
     token?: string;
     sessionId?: string;
     authProvider?: AuthClientProvider;
-    type?: 'channel' | 'sse';
+    type?: 'channel' | 'sse' | 'stream' | 'socket';
     agent?: boolean;
     onError?: (error: Error) => void;
     onUnauthorized?: (connect: () => Promise<void>) => Promise<void>;

package/dist/WebMcpClient.js

@@ -10,8 +10,9 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
 import { Client } from '@modelcontextprotocol/sdk/client/index.js';
 import { SSEClientTransport } from '@modelcontextprotocol/sdk/client/sse.js';
 import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/streamableHttp.js';
+import { WebSocketClientTransport } from '@modelcontextprotocol/sdk/client/websocket.js';
 import { ElicitRequestSchema, CallToolResultSchema, ListRootsRequestSchema, CreateMessageRequestSchema, LoggingMessageNotificationSchema, ToolListChangedNotificationSchema, ResourceUpdatedNotificationSchema, PromptListChangedNotificationSchema, ResourceListChangedNotificationSchema } from '@modelcontextprotocol/sdk/types.js';
-import { MessageChannelClientTransport, sseOptions, streamOptions, attemptConnection, createStreamProxy, createSseProxy } from '@opentiny/next';
+import { MessageChannelClientTransport, sseOptions, streamOptions, attemptConnection, createSseProxy, createStreamProxy, createSocketProxy } from '@opentiny/next';
 /**
  * An MCP client on top of a pluggable transport.
  * The client will automatically begin the initialization flow with the server when connect() is called.
@@ -56,7 +57,11 @@ export class WebMcpClient {
                 let reconnect = false;
                 let response;
                 const connectProxy = () => __awaiter(this, void 0, void 0, function* () {
-                    const { transport, sessionId } = type === 'sse' ? yield createSseProxy(proxyOptions) : yield createStreamProxy(proxyOptions);
+                    const { transport, sessionId } = type === 'sse'
+                        ? yield createSseProxy(proxyOptions)
+                        : type === 'socket'
+                            ? yield createSocketProxy(proxyOptions)
+                            : yield createStreamProxy(proxyOptions);
                     transport.onerror = (error) => __awaiter(this, void 0, void 0, function* () {
                         onError === null || onError === void 0 ? void 0 : onError(error);
                         if (error.message === 'Unauthorized' && !reconnect) {
@@ -93,6 +98,11 @@ export class WebMcpClient {
                     yield this.client.connect(transport);
                 }
             }
+            if (type === 'socket') {
+                transport = new WebSocketClientTransport(new URL(`${url}?sessionId=${sessionId}&token=${token}`));
+                transport.sessionId = sessionId;
+                yield this.client.connect(transport);
+            }
             if (typeof transport === 'undefined') {
                 if (authProvider) {
                     const createTransport = () => new StreamableHTTPClientTransport(endpoint, { authProvider });

package/dist/agent/AgentModelProvider.d.ts

@@ -28,6 +28,8 @@ export declare class AgentModelProvider {
     autoUpdateTools: boolean;
     /** chat 时，自动更新 所有的tools 后的事件 */
     onUpdatedTools: (() => void) | undefined;
+    /** 内部报错时，抛出错误事件 */
+    onError: ((msg: string, err?: any) => void) | undefined;
     constructor({ llmConfig, mcpServers, llm }: IAgentModelProviderOption);
     /** 创建一个 ai-sdk的 mcpClient, 创建失败则返回 Null */
     private _createOneClient;

package/dist/agent/AgentModelProvider.js

@@ -79,9 +79,12 @@ export class AgentModelProvider {
                 else {
                     transport = serverConfig;
                 }
-                return createMCPClient({ transport: transport });
+                return yield createMCPClient({ transport: transport });
             }
             catch (error) {
+                if (this.onError) {
+                    this.onError(`Failed to create MCP client`, error);
+                }
                 console.error(`Failed to create MCP client`, serverConfig, error);
                 return null;
             }
@@ -105,6 +108,10 @@ export class AgentModelProvider {
                     return client ? yield ((_a = client === null || client === void 0 ? void 0 : client.tools) === null || _a === void 0 ? void 0 : _a.call(client)) : null;
                 }
                 catch (error) {
+                    if (this.onError) {
+                        this.onError(`Failed to query tools`, error);
+                    }
+                    console.error(`Failed to query tools`, error);
                     return null;
                 }
             })));
@@ -137,7 +144,7 @@ export class AgentModelProvider {
     insertMcpServer(mcpServer) {
         return __awaiter(this, void 0, void 0, function* () {
             var _a;
-            const find = this.mcpServers.find((item) => item.url === mcpServer.url);
+            const find = this.mcpServers.find((item) => 'url' in item && 'url' in mcpServer && item.url === mcpServer.url);
             if (!find) {
                 this.mcpServers = [...this.mcpServers, mcpServer];
                 const client = yield this._createOneClient(mcpServer);
@@ -190,9 +197,9 @@ export class AgentModelProvider {
                 yield this._createMpcTools();
                 (_b = this.onUpdatedTools) === null || _b === void 0 ? void 0 : _b.call(this);
             }
-            return chatMethod(Object.assign({ 
+            return chatMethod(Object.assign(Object.assign({ 
                 // @ts-ignore  ProviderV2 是所有llm的父类， 在每一个具体的llm 类都有一个选择model的函数用法
-                model: this.llm(model), tools: this.tempMergeTools(options.tools), stopWhen: stepCountIs(maxSteps) }, options));
+                model: this.llm(model), stopWhen: stepCountIs(maxSteps) }, options), { tools: this.tempMergeTools(options.tools) }));
         });
     }
     chat(options) {

package/dist/agent/type.d.ts

@@ -15,7 +15,7 @@ export interface IAgentModelProviderLlmConfig {
 export type McpServerConfig = {
     type: 'streamableHttp' | 'sse';
     url: string;
-} & {
+} | {
     transport: MCPTransport;
 };
 /** */