@@ -5,15 +5,15 @@ import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio";
55import { StreamableHTTPClientTransport } from "@modelcontextprotocol/sdk/client/streamableHttp" ;
66import { Transport } from "@modelcontextprotocol/sdk/shared/transport" ;
77
8- import { McpTestPromptGetResponse } from "./prompts/McpTestPromptGetResponse " ;
9- import { McpTestPromptsListResponse } from "./prompts/McpTestPromptsListResponse " ;
10- import { McpTestResourceTemplatesListResponse } from "./resource-templates/McpTestResourceTemplatesListResponse " ;
11- import { McpTestResourcesListResponse } from "./resources/McpTestResourcesListResponse " ;
12- import { McpTestResourcesReadResponse } from "./resources/McpTestResourcesReadResponse " ;
13- import { McpTestToolCallResponse } from "./tools/McpTestToolCallResponse " ;
14- import { McpTestToolsListResponse } from "./tools/McpTestToolsListResponse " ;
15-
16- export class McpTestClient {
8+ import { McpPromptGetResponse } from "./prompts/McpPromptGetResponse " ;
9+ import { McpPromptsListResponse } from "./prompts/McpPromptsListResponse " ;
10+ import { McpResourceTemplatesListResponse } from "./resource-templates/McpResourceTemplatesListResponse " ;
11+ import { McpResourcesListResponse } from "./resources/McpResourcesListResponse " ;
12+ import { McpResourcesReadResponse } from "./resources/McpResourcesReadResponse " ;
13+ import { McpToolCallResponse } from "./tools/McpToolCallResponse " ;
14+ import { McpToolsListResponse } from "./tools/McpToolsListResponse " ;
15+
16+ export class McpClient {
1717 private readonly client : Client ;
1818 private readonly transport : Transport ;
1919
@@ -23,7 +23,7 @@ export class McpTestClient {
2323 ) {
2424 this . client = new Client (
2525 {
26- name : "mcp-test- client" ,
26+ name : "mcp-client" ,
2727 version : "1.0.0" ,
2828 } ,
2929 {
@@ -53,27 +53,25 @@ export class McpTestClient {
5353 await this . client . close ( ) ;
5454 }
5555
56- async listTools ( ) : Promise < McpTestToolsListResponse > {
56+ async listTools ( ) : Promise < McpToolsListResponse > {
5757 const response = await this . client . listTools ( ) ;
5858
59- return McpTestToolsListResponse . fromPrimitives (
60- response as Primitives < McpTestToolsListResponse > ,
61- ) ;
59+ return McpToolsListResponse . fromPrimitives ( response as Primitives < McpToolsListResponse > ) ;
6260 }
6361
64- async listResources ( ) : Promise < McpTestResourcesListResponse > {
62+ async listResources ( ) : Promise < McpResourcesListResponse > {
6563 const response = await this . client . listResources ( ) ;
6664
67- return McpTestResourcesListResponse . fromPrimitives (
68- response as Primitives < McpTestResourcesListResponse > ,
65+ return McpResourcesListResponse . fromPrimitives (
66+ response as Primitives < McpResourcesListResponse > ,
6967 ) ;
7068 }
7169
72- async listResourceTemplates ( ) : Promise < McpTestResourceTemplatesListResponse > {
70+ async listResourceTemplates ( ) : Promise < McpResourceTemplatesListResponse > {
7371 const response = await this . client . listResourceTemplates ( ) ;
7472
75- return McpTestResourceTemplatesListResponse . fromPrimitives (
76- response as Primitives < McpTestResourceTemplatesListResponse > ,
73+ return McpResourceTemplatesListResponse . fromPrimitives (
74+ response as Primitives < McpResourceTemplatesListResponse > ,
7775 ) ;
7876 }
7977
@@ -96,30 +94,27 @@ export class McpTestClient {
9694 return response . completion . values ;
9795 }
9896
99- async readResource ( uri : string ) : Promise < McpTestResourcesReadResponse > {
97+ async readResource ( uri : string ) : Promise < McpResourcesReadResponse > {
10098 const response = await this . client . readResource ( { uri } ) ;
10199
102- return McpTestResourcesReadResponse . fromPrimitives (
103- response as Primitives < McpTestResourcesReadResponse > ,
100+ return McpResourcesReadResponse . fromPrimitives (
101+ response as Primitives < McpResourcesReadResponse > ,
104102 ) ;
105103 }
106104
107- async callTool (
108- name : string ,
109- args : Record < string , unknown > = { } ,
110- ) : Promise < McpTestToolCallResponse > {
105+ async callTool ( name : string , args : Record < string , unknown > = { } ) : Promise < McpToolCallResponse > {
111106 const response = await this . client . callTool ( {
112107 name,
113108 arguments : args ,
114109 } ) ;
115110
116- return McpTestToolCallResponse . fromPrimitives ( response as Primitives < McpTestToolCallResponse > ) ;
111+ return McpToolCallResponse . fromPrimitives ( response as Primitives < McpToolCallResponse > ) ;
117112 }
118113
119- async listPrompts ( ) : Promise < McpTestPromptsListResponse > {
114+ async listPrompts ( ) : Promise < McpPromptsListResponse > {
120115 const response = await this . client . listPrompts ( ) ;
121116
122- return McpTestPromptsListResponse . fromPrimitives ( {
117+ return McpPromptsListResponse . fromPrimitives ( {
123118 prompts : response . prompts . map ( ( prompt : any ) => ( {
124119 name : prompt . name ,
125120 title : prompt . title ?? "" ,
@@ -129,18 +124,15 @@ export class McpTestClient {
129124 } ) ;
130125 }
131126
132- async getPrompt (
133- name : string ,
134- args : Record < string , unknown > = { } ,
135- ) : Promise < McpTestPromptGetResponse > {
127+ async getPrompt ( name : string , args : Record < string , unknown > = { } ) : Promise < McpPromptGetResponse > {
136128 const response = await this . client . getPrompt ( {
137129 name,
138130 arguments : Object . fromEntries (
139131 Object . entries ( args ) . map ( ( [ key , value ] ) => [ key , String ( value ) ] ) ,
140132 ) ,
141133 } ) ;
142134
143- return McpTestPromptGetResponse . fromPrimitives ( {
135+ return McpPromptGetResponse . fromPrimitives ( {
144136 messages : response . messages . map ( ( message : any ) => ( {
145137 role : message . role ,
146138 content :
0 commit comments