@mastra/client-js 0.11.3-alpha.3 → 0.11.3-alpha.4
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/CHANGELOG.md +16 -0
- package/dist/client.d.ts +12 -2
- package/dist/client.d.ts.map +1 -1
- package/dist/index.cjs +398 -4
- package/dist/index.cjs.map +1 -1
- package/dist/index.js +398 -4
- package/dist/index.js.map +1 -1
- package/dist/resources/agent-builder.d.ts +148 -0
- package/dist/resources/agent-builder.d.ts.map +1 -0
- package/dist/resources/agent.d.ts +25 -0
- package/dist/resources/agent.d.ts.map +1 -1
- package/dist/resources/index.d.ts +1 -0
- package/dist/resources/index.d.ts.map +1 -1
- package/dist/types.d.ts +33 -0
- package/dist/types.d.ts.map +1 -1
- package/package.json +8 -4
- package/.turbo/turbo-build.log +0 -18
- package/eslint.config.js +0 -11
- package/integration-tests/agui-adapter.test.ts +0 -122
- package/integration-tests/package.json +0 -18
- package/integration-tests/src/mastra/index.ts +0 -35
- package/integration-tests/vitest.config.ts +0 -9
- package/src/adapters/agui.test.ts +0 -293
- package/src/adapters/agui.ts +0 -257
- package/src/client.ts +0 -644
- package/src/example.ts +0 -95
- package/src/index.test.ts +0 -1253
- package/src/index.ts +0 -3
- package/src/resources/a2a.ts +0 -98
- package/src/resources/agent.ts +0 -1460
- package/src/resources/base.ts +0 -77
- package/src/resources/index.ts +0 -11
- package/src/resources/legacy-workflow.ts +0 -242
- package/src/resources/mcp-tool.ts +0 -48
- package/src/resources/memory-thread.test.ts +0 -285
- package/src/resources/memory-thread.ts +0 -99
- package/src/resources/network-memory-thread.test.ts +0 -269
- package/src/resources/network-memory-thread.ts +0 -81
- package/src/resources/network.ts +0 -86
- package/src/resources/observability.ts +0 -53
- package/src/resources/tool.ts +0 -45
- package/src/resources/vNextNetwork.ts +0 -194
- package/src/resources/vector.ts +0 -83
- package/src/resources/workflow.ts +0 -410
- package/src/types.ts +0 -534
- package/src/utils/index.ts +0 -11
- package/src/utils/process-client-tools.ts +0 -32
- package/src/utils/process-mastra-stream.test.ts +0 -353
- package/src/utils/process-mastra-stream.ts +0 -49
- package/src/utils/zod-to-json-schema.ts +0 -30
- package/src/v2-messages.test.ts +0 -180
- package/tsconfig.build.json +0 -9
- package/tsconfig.json +0 -5
- package/tsup.config.ts +0 -17
- package/vitest.config.js +0 -8
package/src/resources/base.ts
DELETED
|
@@ -1,77 +0,0 @@
|
|
|
1
|
-
import type { RequestOptions, ClientOptions } from '../types';
|
|
2
|
-
|
|
3
|
-
export class BaseResource {
|
|
4
|
-
readonly options: ClientOptions;
|
|
5
|
-
|
|
6
|
-
constructor(options: ClientOptions) {
|
|
7
|
-
this.options = options;
|
|
8
|
-
}
|
|
9
|
-
|
|
10
|
-
/**
|
|
11
|
-
* Makes an HTTP request to the API with retries and exponential backoff
|
|
12
|
-
* @param path - The API endpoint path
|
|
13
|
-
* @param options - Optional request configuration
|
|
14
|
-
* @returns Promise containing the response data
|
|
15
|
-
*/
|
|
16
|
-
public async request<T>(path: string, options: RequestOptions = {}): Promise<T> {
|
|
17
|
-
let lastError: Error | null = null;
|
|
18
|
-
const { baseUrl, retries = 3, backoffMs = 100, maxBackoffMs = 1000, headers = {}, credentials } = this.options;
|
|
19
|
-
|
|
20
|
-
let delay = backoffMs;
|
|
21
|
-
|
|
22
|
-
for (let attempt = 0; attempt <= retries; attempt++) {
|
|
23
|
-
try {
|
|
24
|
-
const response = await fetch(`${baseUrl.replace(/\/$/, '')}${path}`, {
|
|
25
|
-
...options,
|
|
26
|
-
headers: {
|
|
27
|
-
...(options.body &&
|
|
28
|
-
!(options.body instanceof FormData) &&
|
|
29
|
-
(options.method === 'POST' || options.method === 'PUT')
|
|
30
|
-
? { 'content-type': 'application/json' }
|
|
31
|
-
: {}),
|
|
32
|
-
...headers,
|
|
33
|
-
...options.headers,
|
|
34
|
-
// TODO: Bring this back once we figure out what we/users need to do to make this work with cross-origin requests
|
|
35
|
-
// 'x-mastra-client-type': 'js',
|
|
36
|
-
},
|
|
37
|
-
signal: this.options.abortSignal,
|
|
38
|
-
credentials: options.credentials ?? credentials,
|
|
39
|
-
body:
|
|
40
|
-
options.body instanceof FormData ? options.body : options.body ? JSON.stringify(options.body) : undefined,
|
|
41
|
-
});
|
|
42
|
-
|
|
43
|
-
if (!response.ok) {
|
|
44
|
-
const errorBody = await response.text();
|
|
45
|
-
let errorMessage = `HTTP error! status: ${response.status}`;
|
|
46
|
-
try {
|
|
47
|
-
const errorJson = JSON.parse(errorBody);
|
|
48
|
-
errorMessage += ` - ${JSON.stringify(errorJson)}`;
|
|
49
|
-
} catch {
|
|
50
|
-
if (errorBody) {
|
|
51
|
-
errorMessage += ` - ${errorBody}`;
|
|
52
|
-
}
|
|
53
|
-
}
|
|
54
|
-
throw new Error(errorMessage);
|
|
55
|
-
}
|
|
56
|
-
|
|
57
|
-
if (options.stream) {
|
|
58
|
-
return response as unknown as T;
|
|
59
|
-
}
|
|
60
|
-
|
|
61
|
-
const data = await response.json();
|
|
62
|
-
return data as T;
|
|
63
|
-
} catch (error) {
|
|
64
|
-
lastError = error as Error;
|
|
65
|
-
|
|
66
|
-
if (attempt === retries) {
|
|
67
|
-
break;
|
|
68
|
-
}
|
|
69
|
-
|
|
70
|
-
await new Promise(resolve => setTimeout(resolve, delay));
|
|
71
|
-
delay = Math.min(delay * 2, maxBackoffMs);
|
|
72
|
-
}
|
|
73
|
-
}
|
|
74
|
-
|
|
75
|
-
throw lastError || new Error('Request failed');
|
|
76
|
-
}
|
|
77
|
-
}
|
package/src/resources/index.ts
DELETED
|
@@ -1,11 +0,0 @@
|
|
|
1
|
-
export * from './agent';
|
|
2
|
-
export * from './network';
|
|
3
|
-
export * from './memory-thread';
|
|
4
|
-
export * from './vector';
|
|
5
|
-
export * from './legacy-workflow';
|
|
6
|
-
export * from './tool';
|
|
7
|
-
export * from './base';
|
|
8
|
-
export * from './workflow';
|
|
9
|
-
export * from './a2a';
|
|
10
|
-
export * from './mcp-tool';
|
|
11
|
-
export * from './observability';
|
|
@@ -1,242 +0,0 @@
|
|
|
1
|
-
import type {
|
|
2
|
-
ClientOptions,
|
|
3
|
-
LegacyWorkflowRunResult,
|
|
4
|
-
GetLegacyWorkflowRunsResponse,
|
|
5
|
-
GetWorkflowRunsParams,
|
|
6
|
-
GetLegacyWorkflowResponse,
|
|
7
|
-
} from '../types';
|
|
8
|
-
|
|
9
|
-
import { BaseResource } from './base';
|
|
10
|
-
|
|
11
|
-
const RECORD_SEPARATOR = '\x1E';
|
|
12
|
-
|
|
13
|
-
export class LegacyWorkflow extends BaseResource {
|
|
14
|
-
constructor(
|
|
15
|
-
options: ClientOptions,
|
|
16
|
-
private workflowId: string,
|
|
17
|
-
) {
|
|
18
|
-
super(options);
|
|
19
|
-
}
|
|
20
|
-
|
|
21
|
-
/**
|
|
22
|
-
* Retrieves details about the legacy workflow
|
|
23
|
-
* @returns Promise containing legacy workflow details including steps and graphs
|
|
24
|
-
*/
|
|
25
|
-
details(): Promise<GetLegacyWorkflowResponse> {
|
|
26
|
-
return this.request(`/api/workflows/legacy/${this.workflowId}`);
|
|
27
|
-
}
|
|
28
|
-
|
|
29
|
-
/**
|
|
30
|
-
* Retrieves all runs for a legacy workflow
|
|
31
|
-
* @param params - Parameters for filtering runs
|
|
32
|
-
* @returns Promise containing legacy workflow runs array
|
|
33
|
-
*/
|
|
34
|
-
runs(params?: GetWorkflowRunsParams): Promise<GetLegacyWorkflowRunsResponse> {
|
|
35
|
-
const searchParams = new URLSearchParams();
|
|
36
|
-
if (params?.fromDate) {
|
|
37
|
-
searchParams.set('fromDate', params.fromDate.toISOString());
|
|
38
|
-
}
|
|
39
|
-
if (params?.toDate) {
|
|
40
|
-
searchParams.set('toDate', params.toDate.toISOString());
|
|
41
|
-
}
|
|
42
|
-
if (params?.limit) {
|
|
43
|
-
searchParams.set('limit', String(params.limit));
|
|
44
|
-
}
|
|
45
|
-
if (params?.offset) {
|
|
46
|
-
searchParams.set('offset', String(params.offset));
|
|
47
|
-
}
|
|
48
|
-
if (params?.resourceId) {
|
|
49
|
-
searchParams.set('resourceId', params.resourceId);
|
|
50
|
-
}
|
|
51
|
-
|
|
52
|
-
if (searchParams.size) {
|
|
53
|
-
return this.request(`/api/workflows/legacy/${this.workflowId}/runs?${searchParams}`);
|
|
54
|
-
} else {
|
|
55
|
-
return this.request(`/api/workflows/legacy/${this.workflowId}/runs`);
|
|
56
|
-
}
|
|
57
|
-
}
|
|
58
|
-
|
|
59
|
-
/**
|
|
60
|
-
* Creates a new legacy workflow run
|
|
61
|
-
* @returns Promise containing the generated run ID
|
|
62
|
-
*/
|
|
63
|
-
createRun(params?: { runId?: string }): Promise<{ runId: string }> {
|
|
64
|
-
const searchParams = new URLSearchParams();
|
|
65
|
-
|
|
66
|
-
if (!!params?.runId) {
|
|
67
|
-
searchParams.set('runId', params.runId);
|
|
68
|
-
}
|
|
69
|
-
|
|
70
|
-
return this.request(`/api/workflows/legacy/${this.workflowId}/create-run?${searchParams.toString()}`, {
|
|
71
|
-
method: 'POST',
|
|
72
|
-
});
|
|
73
|
-
}
|
|
74
|
-
|
|
75
|
-
/**
|
|
76
|
-
* Starts a legacy workflow run synchronously without waiting for the workflow to complete
|
|
77
|
-
* @param params - Object containing the runId and triggerData
|
|
78
|
-
* @returns Promise containing success message
|
|
79
|
-
*/
|
|
80
|
-
start(params: { runId: string; triggerData: Record<string, any> }): Promise<{ message: string }> {
|
|
81
|
-
return this.request(`/api/workflows/legacy/${this.workflowId}/start?runId=${params.runId}`, {
|
|
82
|
-
method: 'POST',
|
|
83
|
-
body: params?.triggerData,
|
|
84
|
-
});
|
|
85
|
-
}
|
|
86
|
-
|
|
87
|
-
/**
|
|
88
|
-
* Resumes a suspended legacy workflow step synchronously without waiting for the workflow to complete
|
|
89
|
-
* @param stepId - ID of the step to resume
|
|
90
|
-
* @param runId - ID of the legacy workflow run
|
|
91
|
-
* @param context - Context to resume the legacy workflow with
|
|
92
|
-
* @returns Promise containing the legacy workflow resume results
|
|
93
|
-
*/
|
|
94
|
-
resume({
|
|
95
|
-
stepId,
|
|
96
|
-
runId,
|
|
97
|
-
context,
|
|
98
|
-
}: {
|
|
99
|
-
stepId: string;
|
|
100
|
-
runId: string;
|
|
101
|
-
context: Record<string, any>;
|
|
102
|
-
}): Promise<{ message: string }> {
|
|
103
|
-
return this.request(`/api/workflows/legacy/${this.workflowId}/resume?runId=${runId}`, {
|
|
104
|
-
method: 'POST',
|
|
105
|
-
body: {
|
|
106
|
-
stepId,
|
|
107
|
-
context,
|
|
108
|
-
},
|
|
109
|
-
});
|
|
110
|
-
}
|
|
111
|
-
|
|
112
|
-
/**
|
|
113
|
-
* Starts a workflow run asynchronously and returns a promise that resolves when the workflow is complete
|
|
114
|
-
* @param params - Object containing the optional runId and triggerData
|
|
115
|
-
* @returns Promise containing the workflow execution results
|
|
116
|
-
*/
|
|
117
|
-
startAsync(params: { runId?: string; triggerData: Record<string, any> }): Promise<LegacyWorkflowRunResult> {
|
|
118
|
-
const searchParams = new URLSearchParams();
|
|
119
|
-
|
|
120
|
-
if (!!params?.runId) {
|
|
121
|
-
searchParams.set('runId', params.runId);
|
|
122
|
-
}
|
|
123
|
-
|
|
124
|
-
return this.request(`/api/workflows/legacy/${this.workflowId}/start-async?${searchParams.toString()}`, {
|
|
125
|
-
method: 'POST',
|
|
126
|
-
body: params?.triggerData,
|
|
127
|
-
});
|
|
128
|
-
}
|
|
129
|
-
|
|
130
|
-
/**
|
|
131
|
-
* Resumes a suspended legacy workflow step asynchronously and returns a promise that resolves when the workflow is complete
|
|
132
|
-
* @param params - Object containing the runId, stepId, and context
|
|
133
|
-
* @returns Promise containing the workflow resume results
|
|
134
|
-
*/
|
|
135
|
-
resumeAsync(params: {
|
|
136
|
-
runId: string;
|
|
137
|
-
stepId: string;
|
|
138
|
-
context: Record<string, any>;
|
|
139
|
-
}): Promise<LegacyWorkflowRunResult> {
|
|
140
|
-
return this.request(`/api/workflows/legacy/${this.workflowId}/resume-async?runId=${params.runId}`, {
|
|
141
|
-
method: 'POST',
|
|
142
|
-
body: {
|
|
143
|
-
stepId: params.stepId,
|
|
144
|
-
context: params.context,
|
|
145
|
-
},
|
|
146
|
-
});
|
|
147
|
-
}
|
|
148
|
-
|
|
149
|
-
/**
|
|
150
|
-
* Creates an async generator that processes a readable stream and yields records
|
|
151
|
-
* separated by the Record Separator character (\x1E)
|
|
152
|
-
*
|
|
153
|
-
* @param stream - The readable stream to process
|
|
154
|
-
* @returns An async generator that yields parsed records
|
|
155
|
-
*/
|
|
156
|
-
private async *streamProcessor(stream: ReadableStream): AsyncGenerator<LegacyWorkflowRunResult, void, unknown> {
|
|
157
|
-
const reader = stream.getReader();
|
|
158
|
-
|
|
159
|
-
// Track if we've finished reading from the stream
|
|
160
|
-
let doneReading = false;
|
|
161
|
-
// Buffer to accumulate partial chunks
|
|
162
|
-
let buffer = '';
|
|
163
|
-
|
|
164
|
-
try {
|
|
165
|
-
while (!doneReading) {
|
|
166
|
-
// Read the next chunk from the stream
|
|
167
|
-
const { done, value } = await reader.read();
|
|
168
|
-
doneReading = done;
|
|
169
|
-
|
|
170
|
-
// Skip processing if we're done and there's no value
|
|
171
|
-
if (done && !value) continue;
|
|
172
|
-
|
|
173
|
-
try {
|
|
174
|
-
// Decode binary data to text
|
|
175
|
-
const decoded = value ? new TextDecoder().decode(value) : '';
|
|
176
|
-
|
|
177
|
-
// Split the combined buffer and new data by record separator
|
|
178
|
-
const chunks = (buffer + decoded).split(RECORD_SEPARATOR);
|
|
179
|
-
|
|
180
|
-
// The last chunk might be incomplete, so save it for the next iteration
|
|
181
|
-
buffer = chunks.pop() || '';
|
|
182
|
-
|
|
183
|
-
// Process complete chunks
|
|
184
|
-
for (const chunk of chunks) {
|
|
185
|
-
if (chunk) {
|
|
186
|
-
// Only process non-empty chunks
|
|
187
|
-
if (typeof chunk === 'string') {
|
|
188
|
-
try {
|
|
189
|
-
const parsedChunk = JSON.parse(chunk);
|
|
190
|
-
yield parsedChunk;
|
|
191
|
-
} catch {
|
|
192
|
-
// Silently ignore parsing errors to maintain stream processing
|
|
193
|
-
// This allows the stream to continue even if one record is malformed
|
|
194
|
-
}
|
|
195
|
-
}
|
|
196
|
-
}
|
|
197
|
-
}
|
|
198
|
-
} catch {
|
|
199
|
-
// Silently ignore parsing errors to maintain stream processing
|
|
200
|
-
// This allows the stream to continue even if one record is malformed
|
|
201
|
-
}
|
|
202
|
-
}
|
|
203
|
-
|
|
204
|
-
// Process any remaining data in the buffer after stream is done
|
|
205
|
-
if (buffer) {
|
|
206
|
-
try {
|
|
207
|
-
yield JSON.parse(buffer);
|
|
208
|
-
} catch {
|
|
209
|
-
// Ignore parsing error for final chunk
|
|
210
|
-
}
|
|
211
|
-
}
|
|
212
|
-
} finally {
|
|
213
|
-
// Always ensure we clean up the reader
|
|
214
|
-
reader.cancel().catch(() => {
|
|
215
|
-
// Ignore cancel errors
|
|
216
|
-
});
|
|
217
|
-
}
|
|
218
|
-
}
|
|
219
|
-
|
|
220
|
-
/**
|
|
221
|
-
* Watches legacy workflow transitions in real-time
|
|
222
|
-
* @param runId - Optional run ID to filter the watch stream
|
|
223
|
-
* @returns AsyncGenerator that yields parsed records from the legacy workflow watch stream
|
|
224
|
-
*/
|
|
225
|
-
async watch({ runId }: { runId?: string }, onRecord: (record: LegacyWorkflowRunResult) => void) {
|
|
226
|
-
const response: Response = await this.request(`/api/workflows/legacy/${this.workflowId}/watch?runId=${runId}`, {
|
|
227
|
-
stream: true,
|
|
228
|
-
});
|
|
229
|
-
|
|
230
|
-
if (!response.ok) {
|
|
231
|
-
throw new Error(`Failed to watch legacy workflow: ${response.statusText}`);
|
|
232
|
-
}
|
|
233
|
-
|
|
234
|
-
if (!response.body) {
|
|
235
|
-
throw new Error('Response body is null');
|
|
236
|
-
}
|
|
237
|
-
|
|
238
|
-
for await (const record of this.streamProcessor(response.body)) {
|
|
239
|
-
onRecord(record);
|
|
240
|
-
}
|
|
241
|
-
}
|
|
242
|
-
}
|
|
@@ -1,48 +0,0 @@
|
|
|
1
|
-
import type { RuntimeContext } from '@mastra/core/runtime-context';
|
|
2
|
-
import type { ClientOptions, McpToolInfo } from '../types';
|
|
3
|
-
import { BaseResource } from './base';
|
|
4
|
-
|
|
5
|
-
/**
|
|
6
|
-
* Represents a specific tool available on a specific MCP server.
|
|
7
|
-
* Provides methods to get details and execute the tool.
|
|
8
|
-
*/
|
|
9
|
-
export class MCPTool extends BaseResource {
|
|
10
|
-
private serverId: string;
|
|
11
|
-
private toolId: string;
|
|
12
|
-
|
|
13
|
-
constructor(options: ClientOptions, serverId: string, toolId: string) {
|
|
14
|
-
super(options);
|
|
15
|
-
this.serverId = serverId;
|
|
16
|
-
this.toolId = toolId;
|
|
17
|
-
}
|
|
18
|
-
|
|
19
|
-
/**
|
|
20
|
-
* Retrieves details about this specific tool from the MCP server.
|
|
21
|
-
* @returns Promise containing the tool's information (name, description, schema).
|
|
22
|
-
*/
|
|
23
|
-
details(): Promise<McpToolInfo> {
|
|
24
|
-
return this.request(`/api/mcp/${this.serverId}/tools/${this.toolId}`);
|
|
25
|
-
}
|
|
26
|
-
|
|
27
|
-
/**
|
|
28
|
-
* Executes this specific tool on the MCP server.
|
|
29
|
-
* @param params - Parameters for tool execution, including data/args and optional runtimeContext.
|
|
30
|
-
* @returns Promise containing the result of the tool execution.
|
|
31
|
-
*/
|
|
32
|
-
execute(params: { data?: any; runtimeContext?: RuntimeContext }): Promise<any> {
|
|
33
|
-
const body: any = {};
|
|
34
|
-
if (params.data !== undefined) body.data = params.data;
|
|
35
|
-
// If none of data, args the body might be empty or just contain runtimeContext.
|
|
36
|
-
// The handler will look for these, so an empty args object might be appropriate if that's the intent.
|
|
37
|
-
// else body.data = {}; // Or let it be empty if no specific input fields are used
|
|
38
|
-
|
|
39
|
-
if (params.runtimeContext !== undefined) {
|
|
40
|
-
body.runtimeContext = params.runtimeContext;
|
|
41
|
-
}
|
|
42
|
-
|
|
43
|
-
return this.request(`/api/mcp/${this.serverId}/tools/${this.toolId}/execute`, {
|
|
44
|
-
method: 'POST',
|
|
45
|
-
body: Object.keys(body).length > 0 ? body : undefined,
|
|
46
|
-
});
|
|
47
|
-
}
|
|
48
|
-
}
|
|
@@ -1,285 +0,0 @@
|
|
|
1
|
-
import { describe, expect, beforeEach, it, vi } from 'vitest';
|
|
2
|
-
import type { ClientOptions } from '../types';
|
|
3
|
-
import { MemoryThread } from './memory-thread';
|
|
4
|
-
|
|
5
|
-
// Mock fetch globally
|
|
6
|
-
global.fetch = vi.fn();
|
|
7
|
-
|
|
8
|
-
describe('MemoryThread', () => {
|
|
9
|
-
let thread: MemoryThread;
|
|
10
|
-
const clientOptions: ClientOptions = {
|
|
11
|
-
baseUrl: 'http://localhost:4111',
|
|
12
|
-
headers: {
|
|
13
|
-
Authorization: 'Bearer test-key',
|
|
14
|
-
},
|
|
15
|
-
};
|
|
16
|
-
const threadId = 'test-thread-id';
|
|
17
|
-
const agentId = 'test-agent-id';
|
|
18
|
-
|
|
19
|
-
beforeEach(() => {
|
|
20
|
-
vi.clearAllMocks();
|
|
21
|
-
thread = new MemoryThread(clientOptions, threadId, agentId);
|
|
22
|
-
});
|
|
23
|
-
|
|
24
|
-
const mockFetchResponse = (data: any) => {
|
|
25
|
-
(global.fetch as any).mockResolvedValueOnce({
|
|
26
|
-
ok: true,
|
|
27
|
-
status: 200,
|
|
28
|
-
json: async () => data,
|
|
29
|
-
headers: new Headers({
|
|
30
|
-
'content-type': 'application/json',
|
|
31
|
-
}),
|
|
32
|
-
});
|
|
33
|
-
};
|
|
34
|
-
|
|
35
|
-
describe('get', () => {
|
|
36
|
-
it('should retrieve thread details', async () => {
|
|
37
|
-
const mockThread = {
|
|
38
|
-
id: threadId,
|
|
39
|
-
title: 'Test Thread',
|
|
40
|
-
metadata: { test: true },
|
|
41
|
-
createdAt: new Date().toISOString(),
|
|
42
|
-
updatedAt: new Date().toISOString(),
|
|
43
|
-
};
|
|
44
|
-
|
|
45
|
-
mockFetchResponse(mockThread);
|
|
46
|
-
|
|
47
|
-
const result = await thread.get();
|
|
48
|
-
|
|
49
|
-
expect(global.fetch).toHaveBeenCalledWith(
|
|
50
|
-
`http://localhost:4111/api/memory/threads/${threadId}?agentId=${agentId}`,
|
|
51
|
-
expect.objectContaining({
|
|
52
|
-
headers: expect.objectContaining({
|
|
53
|
-
Authorization: 'Bearer test-key',
|
|
54
|
-
}),
|
|
55
|
-
}),
|
|
56
|
-
);
|
|
57
|
-
expect(result).toEqual(mockThread);
|
|
58
|
-
});
|
|
59
|
-
});
|
|
60
|
-
|
|
61
|
-
describe('update', () => {
|
|
62
|
-
it('should update thread properties', async () => {
|
|
63
|
-
const updateParams = {
|
|
64
|
-
title: 'Updated Title',
|
|
65
|
-
metadata: { updated: true },
|
|
66
|
-
resourceid: 'resource-1',
|
|
67
|
-
};
|
|
68
|
-
|
|
69
|
-
const mockUpdatedThread = {
|
|
70
|
-
id: threadId,
|
|
71
|
-
...updateParams,
|
|
72
|
-
createdAt: new Date().toISOString(),
|
|
73
|
-
updatedAt: new Date().toISOString(),
|
|
74
|
-
};
|
|
75
|
-
|
|
76
|
-
mockFetchResponse(mockUpdatedThread);
|
|
77
|
-
|
|
78
|
-
const result = await thread.update(updateParams);
|
|
79
|
-
|
|
80
|
-
expect(global.fetch).toHaveBeenCalledWith(
|
|
81
|
-
`http://localhost:4111/api/memory/threads/${threadId}?agentId=${agentId}`,
|
|
82
|
-
expect.objectContaining({
|
|
83
|
-
method: 'PATCH',
|
|
84
|
-
headers: expect.objectContaining({
|
|
85
|
-
Authorization: 'Bearer test-key',
|
|
86
|
-
}),
|
|
87
|
-
body: JSON.stringify(updateParams),
|
|
88
|
-
}),
|
|
89
|
-
);
|
|
90
|
-
expect(result).toEqual(mockUpdatedThread);
|
|
91
|
-
});
|
|
92
|
-
});
|
|
93
|
-
|
|
94
|
-
describe('delete', () => {
|
|
95
|
-
it('should delete the thread', async () => {
|
|
96
|
-
const mockResponse = { result: 'Thread deleted' };
|
|
97
|
-
mockFetchResponse(mockResponse);
|
|
98
|
-
|
|
99
|
-
const result = await thread.delete();
|
|
100
|
-
|
|
101
|
-
expect(global.fetch).toHaveBeenCalledWith(
|
|
102
|
-
`http://localhost:4111/api/memory/threads/${threadId}?agentId=${agentId}`,
|
|
103
|
-
expect.objectContaining({
|
|
104
|
-
method: 'DELETE',
|
|
105
|
-
headers: expect.objectContaining({
|
|
106
|
-
Authorization: 'Bearer test-key',
|
|
107
|
-
}),
|
|
108
|
-
}),
|
|
109
|
-
);
|
|
110
|
-
expect(result).toEqual(mockResponse);
|
|
111
|
-
});
|
|
112
|
-
});
|
|
113
|
-
|
|
114
|
-
describe('getMessages', () => {
|
|
115
|
-
it('should retrieve thread messages', async () => {
|
|
116
|
-
const mockMessages = {
|
|
117
|
-
messages: [
|
|
118
|
-
{ id: 'msg-1', content: 'Hello', role: 'user' },
|
|
119
|
-
{ id: 'msg-2', content: 'Hi there', role: 'assistant' },
|
|
120
|
-
],
|
|
121
|
-
uiMessages: [
|
|
122
|
-
{ id: 'msg-1', content: 'Hello', role: 'user' },
|
|
123
|
-
{ id: 'msg-2', content: 'Hi there', role: 'assistant' },
|
|
124
|
-
],
|
|
125
|
-
};
|
|
126
|
-
|
|
127
|
-
mockFetchResponse(mockMessages);
|
|
128
|
-
|
|
129
|
-
const result = await thread.getMessages();
|
|
130
|
-
|
|
131
|
-
expect(global.fetch).toHaveBeenCalledWith(
|
|
132
|
-
`http://localhost:4111/api/memory/threads/${threadId}/messages?agentId=${agentId}`,
|
|
133
|
-
expect.objectContaining({
|
|
134
|
-
headers: expect.objectContaining({
|
|
135
|
-
Authorization: 'Bearer test-key',
|
|
136
|
-
}),
|
|
137
|
-
}),
|
|
138
|
-
);
|
|
139
|
-
expect(result).toEqual(mockMessages);
|
|
140
|
-
});
|
|
141
|
-
|
|
142
|
-
it('should retrieve thread messages with limit', async () => {
|
|
143
|
-
const mockMessages = {
|
|
144
|
-
messages: [{ id: 'msg-1', content: 'Hello', role: 'user' }],
|
|
145
|
-
uiMessages: [{ id: 'msg-1', content: 'Hello', role: 'user' }],
|
|
146
|
-
};
|
|
147
|
-
|
|
148
|
-
mockFetchResponse(mockMessages);
|
|
149
|
-
|
|
150
|
-
const result = await thread.getMessages({ limit: 5 });
|
|
151
|
-
|
|
152
|
-
expect(global.fetch).toHaveBeenCalledWith(
|
|
153
|
-
`http://localhost:4111/api/memory/threads/${threadId}/messages?agentId=${agentId}&limit=5`,
|
|
154
|
-
expect.objectContaining({
|
|
155
|
-
headers: expect.objectContaining({
|
|
156
|
-
Authorization: 'Bearer test-key',
|
|
157
|
-
}),
|
|
158
|
-
}),
|
|
159
|
-
);
|
|
160
|
-
expect(result).toEqual(mockMessages);
|
|
161
|
-
});
|
|
162
|
-
});
|
|
163
|
-
|
|
164
|
-
describe('deleteMessages', () => {
|
|
165
|
-
it('should delete a single message by string ID', async () => {
|
|
166
|
-
const messageId = 'test-message-id';
|
|
167
|
-
const mockResponse = { success: true, message: '1 message deleted successfully' };
|
|
168
|
-
|
|
169
|
-
mockFetchResponse(mockResponse);
|
|
170
|
-
|
|
171
|
-
const result = await thread.deleteMessages(messageId);
|
|
172
|
-
|
|
173
|
-
expect(global.fetch).toHaveBeenCalledWith(
|
|
174
|
-
`http://localhost:4111/api/memory/messages/delete?agentId=${agentId}`,
|
|
175
|
-
expect.objectContaining({
|
|
176
|
-
method: 'POST',
|
|
177
|
-
headers: expect.objectContaining({
|
|
178
|
-
'content-type': 'application/json',
|
|
179
|
-
Authorization: 'Bearer test-key',
|
|
180
|
-
}),
|
|
181
|
-
body: JSON.stringify({ messageIds: messageId }),
|
|
182
|
-
}),
|
|
183
|
-
);
|
|
184
|
-
expect(result).toEqual(mockResponse);
|
|
185
|
-
});
|
|
186
|
-
|
|
187
|
-
it('should delete multiple messages by array of string IDs', async () => {
|
|
188
|
-
const messageIds = ['msg-1', 'msg-2', 'msg-3'];
|
|
189
|
-
const mockResponse = { success: true, message: '3 messages deleted successfully' };
|
|
190
|
-
|
|
191
|
-
mockFetchResponse(mockResponse);
|
|
192
|
-
|
|
193
|
-
const result = await thread.deleteMessages(messageIds);
|
|
194
|
-
|
|
195
|
-
expect(global.fetch).toHaveBeenCalledWith(
|
|
196
|
-
`http://localhost:4111/api/memory/messages/delete?agentId=${agentId}`,
|
|
197
|
-
expect.objectContaining({
|
|
198
|
-
method: 'POST',
|
|
199
|
-
headers: expect.objectContaining({
|
|
200
|
-
'content-type': 'application/json',
|
|
201
|
-
Authorization: 'Bearer test-key',
|
|
202
|
-
}),
|
|
203
|
-
body: JSON.stringify({ messageIds }),
|
|
204
|
-
}),
|
|
205
|
-
);
|
|
206
|
-
expect(result).toEqual(mockResponse);
|
|
207
|
-
});
|
|
208
|
-
|
|
209
|
-
it('should delete a message by object with id property', async () => {
|
|
210
|
-
const messageObj = { id: 'test-message-id' };
|
|
211
|
-
const mockResponse = { success: true, message: '1 message deleted successfully' };
|
|
212
|
-
|
|
213
|
-
mockFetchResponse(mockResponse);
|
|
214
|
-
|
|
215
|
-
const result = await thread.deleteMessages(messageObj);
|
|
216
|
-
|
|
217
|
-
expect(global.fetch).toHaveBeenCalledWith(
|
|
218
|
-
`http://localhost:4111/api/memory/messages/delete?agentId=${agentId}`,
|
|
219
|
-
expect.objectContaining({
|
|
220
|
-
method: 'POST',
|
|
221
|
-
headers: expect.objectContaining({
|
|
222
|
-
'content-type': 'application/json',
|
|
223
|
-
Authorization: 'Bearer test-key',
|
|
224
|
-
}),
|
|
225
|
-
body: JSON.stringify({ messageIds: messageObj }),
|
|
226
|
-
}),
|
|
227
|
-
);
|
|
228
|
-
expect(result).toEqual(mockResponse);
|
|
229
|
-
});
|
|
230
|
-
|
|
231
|
-
it('should delete messages by array of objects', async () => {
|
|
232
|
-
const messageObjs = [{ id: 'msg-1' }, { id: 'msg-2' }];
|
|
233
|
-
const mockResponse = { success: true, message: '2 messages deleted successfully' };
|
|
234
|
-
|
|
235
|
-
mockFetchResponse(mockResponse);
|
|
236
|
-
|
|
237
|
-
const result = await thread.deleteMessages(messageObjs);
|
|
238
|
-
|
|
239
|
-
expect(global.fetch).toHaveBeenCalledWith(
|
|
240
|
-
`http://localhost:4111/api/memory/messages/delete?agentId=${agentId}`,
|
|
241
|
-
expect.objectContaining({
|
|
242
|
-
method: 'POST',
|
|
243
|
-
headers: expect.objectContaining({
|
|
244
|
-
'content-type': 'application/json',
|
|
245
|
-
Authorization: 'Bearer test-key',
|
|
246
|
-
}),
|
|
247
|
-
body: JSON.stringify({ messageIds: messageObjs }),
|
|
248
|
-
}),
|
|
249
|
-
);
|
|
250
|
-
expect(result).toEqual(mockResponse);
|
|
251
|
-
});
|
|
252
|
-
|
|
253
|
-
it('should handle empty array', async () => {
|
|
254
|
-
const messageIds: string[] = [];
|
|
255
|
-
|
|
256
|
-
(global.fetch as any).mockResolvedValueOnce({
|
|
257
|
-
ok: false,
|
|
258
|
-
status: 400,
|
|
259
|
-
statusText: 'Bad Request',
|
|
260
|
-
json: async () => ({ error: 'messageIds array cannot be empty' }),
|
|
261
|
-
headers: new Headers({
|
|
262
|
-
'content-type': 'application/json',
|
|
263
|
-
}),
|
|
264
|
-
});
|
|
265
|
-
|
|
266
|
-
await expect(thread.deleteMessages(messageIds)).rejects.toThrow();
|
|
267
|
-
});
|
|
268
|
-
|
|
269
|
-
it('should handle bulk delete errors', async () => {
|
|
270
|
-
const messageIds = ['msg-1', 'msg-2'];
|
|
271
|
-
|
|
272
|
-
(global.fetch as any).mockResolvedValueOnce({
|
|
273
|
-
ok: false,
|
|
274
|
-
status: 500,
|
|
275
|
-
statusText: 'Internal Server Error',
|
|
276
|
-
json: async () => ({ error: 'Database error' }),
|
|
277
|
-
headers: new Headers({
|
|
278
|
-
'content-type': 'application/json',
|
|
279
|
-
}),
|
|
280
|
-
});
|
|
281
|
-
|
|
282
|
-
await expect(thread.deleteMessages(messageIds)).rejects.toThrow();
|
|
283
|
-
});
|
|
284
|
-
});
|
|
285
|
-
});
|