@mastra/client-js 0.0.0-fix-memory-xxhash-20250409202110 → 0.0.0-generate-message-id-20250512171942

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.
@@ -1,8 +1,8 @@
1
+ import { processDataStream } from '@ai-sdk/ui-utils';
1
2
  import type { GenerateReturn } from '@mastra/core';
2
3
  import type { JSONSchema7 } from 'json-schema';
3
4
  import { ZodSchema } from 'zod';
4
- import { zodToJsonSchema } from 'zod-to-json-schema';
5
- import { processDataStream } from '@ai-sdk/ui-utils';
5
+ import { zodToJsonSchema } from '../utils/zod-to-json-schema';
6
6
 
7
7
  import type {
8
8
  GenerateParams,
@@ -14,29 +14,7 @@ import type {
14
14
  } from '../types';
15
15
 
16
16
  import { BaseResource } from './base';
17
-
18
- export class AgentTool extends BaseResource {
19
- constructor(
20
- options: ClientOptions,
21
- private agentId: string,
22
- private toolId: string,
23
- ) {
24
- super(options);
25
- }
26
-
27
- /**
28
- * Executes a specific tool for an agent
29
- * @param params - Parameters required for tool execution
30
- * @returns Promise containing tool execution results
31
- */
32
- /** @deprecated use CreateRun/startRun */
33
- execute(params: { data: any }): Promise<any> {
34
- return this.request(`/api/agents/${this.agentId}/tools/${this.toolId}/execute`, {
35
- method: 'POST',
36
- body: params,
37
- });
38
- }
39
- }
17
+ import type { RuntimeContext } from '@mastra/core/di';
40
18
 
41
19
  export class AgentVoice extends BaseResource {
42
20
  constructor(
@@ -122,11 +100,9 @@ export class Agent extends BaseResource {
122
100
  ): Promise<GenerateReturn<T>> {
123
101
  const processedParams = {
124
102
  ...params,
125
- output: params.output instanceof ZodSchema ? zodToJsonSchema(params.output) : params.output,
126
- experimental_output:
127
- params.experimental_output instanceof ZodSchema
128
- ? zodToJsonSchema(params.experimental_output)
129
- : params.experimental_output,
103
+ output: zodToJsonSchema(params.output),
104
+ experimental_output: zodToJsonSchema(params.experimental_output),
105
+ runtimeContext: params.runtimeContext ? Object.fromEntries(params.runtimeContext.entries()) : undefined,
130
106
  };
131
107
 
132
108
  return this.request(`/api/agents/${this.agentId}/generate`, {
@@ -149,11 +125,9 @@ export class Agent extends BaseResource {
149
125
  > {
150
126
  const processedParams = {
151
127
  ...params,
152
- output: params.output instanceof ZodSchema ? zodToJsonSchema(params.output) : params.output,
153
- experimental_output:
154
- params.experimental_output instanceof ZodSchema
155
- ? zodToJsonSchema(params.experimental_output)
156
- : params.experimental_output,
128
+ output: zodToJsonSchema(params.output),
129
+ experimental_output: zodToJsonSchema(params.experimental_output),
130
+ runtimeContext: params.runtimeContext ? Object.fromEntries(params.runtimeContext.entries()) : undefined,
157
131
  };
158
132
 
159
133
  const response: Response & {
@@ -187,6 +161,23 @@ export class Agent extends BaseResource {
187
161
  return this.request(`/api/agents/${this.agentId}/tools/${toolId}`);
188
162
  }
189
163
 
164
+ /**
165
+ * Executes a tool for the agent
166
+ * @param toolId - ID of the tool to execute
167
+ * @param params - Parameters required for tool execution
168
+ * @returns Promise containing the tool execution results
169
+ */
170
+ executeTool(toolId: string, params: { data: any; runtimeContext?: RuntimeContext }): Promise<any> {
171
+ const body = {
172
+ data: params.data,
173
+ runtimeContext: params.runtimeContext ? Object.fromEntries(params.runtimeContext.entries()) : undefined,
174
+ };
175
+ return this.request(`/api/agents/${this.agentId}/tools/${toolId}/execute`, {
176
+ method: 'POST',
177
+ body,
178
+ });
179
+ }
180
+
190
181
  /**
191
182
  * Retrieves evaluation results for the agent
192
183
  * @returns Promise containing agent evaluations
@@ -1,4 +1,4 @@
1
- import type { RequestFunction, RequestOptions, ClientOptions } from '../types';
1
+ import type { RequestOptions, ClientOptions } from '../types';
2
2
 
3
3
  export class BaseResource {
4
4
  readonly options: ClientOptions;
@@ -5,3 +5,5 @@ export * from './vector';
5
5
  export * from './workflow';
6
6
  export * from './tool';
7
7
  export * from './base';
8
+ export * from './vnext-workflow';
9
+ export * from './a2a';
@@ -1,13 +1,6 @@
1
1
  import type { StorageThreadType } from '@mastra/core';
2
2
 
3
- import type {
4
- CreateMemoryThreadParams,
5
- GetMemoryThreadMessagesResponse,
6
- GetMemoryThreadResponse,
7
- ClientOptions,
8
- SaveMessageToMemoryParams,
9
- UpdateMemoryThreadParams,
10
- } from '../types';
3
+ import type { GetMemoryThreadMessagesResponse, ClientOptions, UpdateMemoryThreadParams } from '../types';
11
4
 
12
5
  import { BaseResource } from './base';
13
6
 
@@ -1,12 +1,12 @@
1
+ import { processDataStream } from '@ai-sdk/ui-utils';
1
2
  import type { GenerateReturn } from '@mastra/core';
2
3
  import type { JSONSchema7 } from 'json-schema';
3
4
  import { ZodSchema } from 'zod';
4
- import { zodToJsonSchema } from 'zod-to-json-schema';
5
+ import { zodToJsonSchema } from '../utils/zod-to-json-schema';
5
6
 
6
7
  import type { GenerateParams, ClientOptions, StreamParams, GetNetworkResponse } from '../types';
7
8
 
8
9
  import { BaseResource } from './base';
9
- import { processDataStream } from '@ai-sdk/ui-utils';
10
10
 
11
11
  export class Network extends BaseResource {
12
12
  constructor(
@@ -34,11 +34,8 @@ export class Network extends BaseResource {
34
34
  ): Promise<GenerateReturn<T>> {
35
35
  const processedParams = {
36
36
  ...params,
37
- output: params.output instanceof ZodSchema ? zodToJsonSchema(params.output) : params.output,
38
- experimental_output:
39
- params.experimental_output instanceof ZodSchema
40
- ? zodToJsonSchema(params.experimental_output)
41
- : params.experimental_output,
37
+ output: zodToJsonSchema(params.output),
38
+ experimental_output: zodToJsonSchema(params.experimental_output),
42
39
  };
43
40
 
44
41
  return this.request(`/api/networks/${this.networkId}/generate`, {
@@ -61,11 +58,8 @@ export class Network extends BaseResource {
61
58
  > {
62
59
  const processedParams = {
63
60
  ...params,
64
- output: params.output instanceof ZodSchema ? zodToJsonSchema(params.output) : params.output,
65
- experimental_output:
66
- params.experimental_output instanceof ZodSchema
67
- ? zodToJsonSchema(params.experimental_output)
68
- : params.experimental_output,
61
+ output: zodToJsonSchema(params.output),
62
+ experimental_output: zodToJsonSchema(params.experimental_output),
69
63
  };
70
64
 
71
65
  const response: Response & {
@@ -1,3 +1,4 @@
1
+ import type { RuntimeContext } from '@mastra/core/di';
1
2
  import type { GetToolResponse, ClientOptions } from '../types';
2
3
 
3
4
  import { BaseResource } from './base';
@@ -23,10 +24,21 @@ export class Tool extends BaseResource {
23
24
  * @param params - Parameters required for tool execution
24
25
  * @returns Promise containing the tool execution results
25
26
  */
26
- execute(params: { data: any }): Promise<any> {
27
- return this.request(`/api/tools/${this.toolId}/execute`, {
27
+ execute(params: { data: any; runId?: string; runtimeContext?: RuntimeContext }): Promise<any> {
28
+ const url = new URLSearchParams();
29
+
30
+ if (params.runId) {
31
+ url.set('runId', params.runId);
32
+ }
33
+
34
+ const body = {
35
+ data: params.data,
36
+ runtimeContext: params.runtimeContext ? Object.fromEntries(params.runtimeContext.entries()) : undefined,
37
+ };
38
+
39
+ return this.request(`/api/tools/${this.toolId}/execute?${url.toString()}`, {
28
40
  method: 'POST',
29
- body: params,
41
+ body,
30
42
  });
31
43
  }
32
44
  }
@@ -0,0 +1,261 @@
1
+ import { RuntimeContext } from '@mastra/core/runtime-context';
2
+ import type {
3
+ ClientOptions,
4
+ GetVNextWorkflowResponse,
5
+ GetWorkflowRunsParams,
6
+ GetWorkflowRunsResponse,
7
+ VNextWorkflowRunResult,
8
+ VNextWorkflowWatchResult,
9
+ } from '../types';
10
+
11
+ import { BaseResource } from './base';
12
+
13
+ const RECORD_SEPARATOR = '\x1E';
14
+
15
+ export class VNextWorkflow extends BaseResource {
16
+ constructor(
17
+ options: ClientOptions,
18
+ private workflowId: string,
19
+ ) {
20
+ super(options);
21
+ }
22
+
23
+ /**
24
+ * Creates an async generator that processes a readable stream and yields vNext workflow records
25
+ * separated by the Record Separator character (\x1E)
26
+ *
27
+ * @param stream - The readable stream to process
28
+ * @returns An async generator that yields parsed records
29
+ */
30
+ private async *streamProcessor(stream: ReadableStream): AsyncGenerator<VNextWorkflowWatchResult, void, unknown> {
31
+ const reader = stream.getReader();
32
+
33
+ // Track if we've finished reading from the stream
34
+ let doneReading = false;
35
+ // Buffer to accumulate partial chunks
36
+ let buffer = '';
37
+
38
+ try {
39
+ while (!doneReading) {
40
+ // Read the next chunk from the stream
41
+ const { done, value } = await reader.read();
42
+ doneReading = done;
43
+
44
+ // Skip processing if we're done and there's no value
45
+ if (done && !value) continue;
46
+
47
+ try {
48
+ // Decode binary data to text
49
+ const decoded = value ? new TextDecoder().decode(value) : '';
50
+
51
+ // Split the combined buffer and new data by record separator
52
+ const chunks = (buffer + decoded).split(RECORD_SEPARATOR);
53
+
54
+ // The last chunk might be incomplete, so save it for the next iteration
55
+ buffer = chunks.pop() || '';
56
+
57
+ // Process complete chunks
58
+ for (const chunk of chunks) {
59
+ if (chunk) {
60
+ // Only process non-empty chunks
61
+ if (typeof chunk === 'string') {
62
+ try {
63
+ const parsedChunk = JSON.parse(chunk);
64
+ yield parsedChunk;
65
+ } catch {
66
+ // Silently ignore parsing errors to maintain stream processing
67
+ // This allows the stream to continue even if one record is malformed
68
+ }
69
+ }
70
+ }
71
+ }
72
+ } catch {
73
+ // Silently ignore parsing errors to maintain stream processing
74
+ // This allows the stream to continue even if one record is malformed
75
+ }
76
+ }
77
+
78
+ // Process any remaining data in the buffer after stream is done
79
+ if (buffer) {
80
+ try {
81
+ yield JSON.parse(buffer);
82
+ } catch {
83
+ // Ignore parsing error for final chunk
84
+ }
85
+ }
86
+ } finally {
87
+ // Always ensure we clean up the reader
88
+ reader.cancel().catch(() => {
89
+ // Ignore cancel errors
90
+ });
91
+ }
92
+ }
93
+
94
+ /**
95
+ * Retrieves details about the vNext workflow
96
+ * @returns Promise containing vNext workflow details including steps and graphs
97
+ */
98
+ details(): Promise<GetVNextWorkflowResponse> {
99
+ return this.request(`/api/workflows/v-next/${this.workflowId}`);
100
+ }
101
+
102
+ /**
103
+ * Retrieves all runs for a vNext workflow
104
+ * @param params - Parameters for filtering runs
105
+ * @returns Promise containing vNext workflow runs array
106
+ */
107
+ runs(params?: GetWorkflowRunsParams): Promise<GetWorkflowRunsResponse> {
108
+ const searchParams = new URLSearchParams();
109
+ if (params?.fromDate) {
110
+ searchParams.set('fromDate', params.fromDate.toISOString());
111
+ }
112
+ if (params?.toDate) {
113
+ searchParams.set('toDate', params.toDate.toISOString());
114
+ }
115
+ if (params?.limit) {
116
+ searchParams.set('limit', String(params.limit));
117
+ }
118
+ if (params?.offset) {
119
+ searchParams.set('offset', String(params.offset));
120
+ }
121
+ if (params?.resourceId) {
122
+ searchParams.set('resourceId', params.resourceId);
123
+ }
124
+
125
+ if (searchParams.size) {
126
+ return this.request(`/api/workflows/v-next/${this.workflowId}/runs?${searchParams}`);
127
+ } else {
128
+ return this.request(`/api/workflows/v-next/${this.workflowId}/runs`);
129
+ }
130
+ }
131
+
132
+ /**
133
+ * Creates a new vNext workflow run
134
+ * @param params - Optional object containing the optional runId
135
+ * @returns Promise containing the runId of the created run
136
+ */
137
+ createRun(params?: { runId?: string }): Promise<{ runId: string }> {
138
+ const searchParams = new URLSearchParams();
139
+
140
+ if (!!params?.runId) {
141
+ searchParams.set('runId', params.runId);
142
+ }
143
+
144
+ return this.request(`/api/workflows/v-next/${this.workflowId}/create-run?${searchParams.toString()}`, {
145
+ method: 'POST',
146
+ });
147
+ }
148
+
149
+ /**
150
+ * Starts a vNext workflow run synchronously without waiting for the workflow to complete
151
+ * @param params - Object containing the runId, inputData and runtimeContext
152
+ * @returns Promise containing success message
153
+ */
154
+ start(params: {
155
+ runId: string;
156
+ inputData: Record<string, any>;
157
+ runtimeContext?: RuntimeContext;
158
+ }): Promise<{ message: string }> {
159
+ const runtimeContext = params.runtimeContext ? Object.fromEntries(params.runtimeContext.entries()) : undefined;
160
+ return this.request(`/api/workflows/v-next/${this.workflowId}/start?runId=${params.runId}`, {
161
+ method: 'POST',
162
+ body: { inputData: params?.inputData, runtimeContext },
163
+ });
164
+ }
165
+
166
+ /**
167
+ * Resumes a suspended vNext workflow step synchronously without waiting for the vNext workflow to complete
168
+ * @param params - Object containing the runId, step, resumeData and runtimeContext
169
+ * @returns Promise containing success message
170
+ */
171
+ resume({
172
+ step,
173
+ runId,
174
+ resumeData,
175
+ ...rest
176
+ }: {
177
+ step: string | string[];
178
+ runId: string;
179
+ resumeData?: Record<string, any>;
180
+ runtimeContext?: RuntimeContext;
181
+ }): Promise<{ message: string }> {
182
+ const runtimeContext = rest.runtimeContext ? Object.fromEntries(rest.runtimeContext.entries()) : undefined;
183
+ return this.request(`/api/workflows/v-next/${this.workflowId}/resume?runId=${runId}`, {
184
+ method: 'POST',
185
+ stream: true,
186
+ body: {
187
+ step,
188
+ resumeData,
189
+ runtimeContext,
190
+ },
191
+ });
192
+ }
193
+
194
+ /**
195
+ * Starts a vNext workflow run asynchronously and returns a promise that resolves when the vNext workflow is complete
196
+ * @param params - Object containing the optional runId, inputData and runtimeContext
197
+ * @returns Promise containing the vNext workflow execution results
198
+ */
199
+ startAsync(params: {
200
+ runId?: string;
201
+ inputData: Record<string, any>;
202
+ runtimeContext?: RuntimeContext;
203
+ }): Promise<VNextWorkflowRunResult> {
204
+ const searchParams = new URLSearchParams();
205
+
206
+ if (!!params?.runId) {
207
+ searchParams.set('runId', params.runId);
208
+ }
209
+
210
+ const runtimeContext = params.runtimeContext ? Object.fromEntries(params.runtimeContext.entries()) : undefined;
211
+ return this.request(`/api/workflows/v-next/${this.workflowId}/start-async?${searchParams.toString()}`, {
212
+ method: 'POST',
213
+ body: { inputData: params.inputData, runtimeContext },
214
+ });
215
+ }
216
+
217
+ /**
218
+ * Resumes a suspended vNext workflow step asynchronously and returns a promise that resolves when the vNext workflow is complete
219
+ * @param params - Object containing the runId, step, resumeData and runtimeContext
220
+ * @returns Promise containing the vNext workflow resume results
221
+ */
222
+ resumeAsync(params: {
223
+ runId: string;
224
+ step: string | string[];
225
+ resumeData?: Record<string, any>;
226
+ runtimeContext?: RuntimeContext;
227
+ }): Promise<VNextWorkflowRunResult> {
228
+ const runtimeContext = params.runtimeContext ? Object.fromEntries(params.runtimeContext.entries()) : undefined;
229
+ return this.request(`/api/workflows/v-next/${this.workflowId}/resume-async?runId=${params.runId}`, {
230
+ method: 'POST',
231
+ body: {
232
+ step: params.step,
233
+ resumeData: params.resumeData,
234
+ runtimeContext,
235
+ },
236
+ });
237
+ }
238
+
239
+ /**
240
+ * Watches vNext workflow transitions in real-time
241
+ * @param runId - Optional run ID to filter the watch stream
242
+ * @returns AsyncGenerator that yields parsed records from the vNext workflow watch stream
243
+ */
244
+ async watch({ runId }: { runId?: string }, onRecord: (record: VNextWorkflowWatchResult) => void) {
245
+ const response: Response = await this.request(`/api/workflows/v-next/${this.workflowId}/watch?runId=${runId}`, {
246
+ stream: true,
247
+ });
248
+
249
+ if (!response.ok) {
250
+ throw new Error(`Failed to watch vNext workflow: ${response.statusText}`);
251
+ }
252
+
253
+ if (!response.body) {
254
+ throw new Error('Response body is null');
255
+ }
256
+
257
+ for await (const record of this.streamProcessor(response.body)) {
258
+ onRecord(record);
259
+ }
260
+ }
261
+ }
@@ -1,4 +1,10 @@
1
- import type { GetWorkflowResponse, ClientOptions, WorkflowRunResult } from '../types';
1
+ import type {
2
+ GetWorkflowResponse,
3
+ ClientOptions,
4
+ WorkflowRunResult,
5
+ GetWorkflowRunsResponse,
6
+ GetWorkflowRunsParams,
7
+ } from '../types';
2
8
 
3
9
  import { BaseResource } from './base';
4
10
 
@@ -20,6 +26,36 @@ export class Workflow extends BaseResource {
20
26
  return this.request(`/api/workflows/${this.workflowId}`);
21
27
  }
22
28
 
29
+ /**
30
+ * Retrieves all runs for a workflow
31
+ * @param params - Parameters for filtering runs
32
+ * @returns Promise containing workflow runs array
33
+ */
34
+ runs(params?: GetWorkflowRunsParams): Promise<GetWorkflowRunsResponse> {
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/${this.workflowId}/runs?${searchParams}`);
54
+ } else {
55
+ return this.request(`/api/workflows/${this.workflowId}/runs`);
56
+ }
57
+ }
58
+
23
59
  /**
24
60
  * @deprecated Use `startAsync` instead
25
61
  * Executes the workflow with the provided parameters
@@ -168,7 +204,7 @@ export class Workflow extends BaseResource {
168
204
  }
169
205
  }
170
206
  }
171
- } catch (error) {
207
+ } catch {
172
208
  // Silently ignore parsing errors to maintain stream processing
173
209
  // This allows the stream to continue even if one record is malformed
174
210
  }
package/src/types.ts CHANGED
@@ -8,9 +8,11 @@ import type {
8
8
  StorageThreadType,
9
9
  BaseLogMessage,
10
10
  WorkflowRunResult as CoreWorkflowRunResult,
11
+ WorkflowRuns,
11
12
  } from '@mastra/core';
12
13
 
13
14
  import type { AgentGenerateOptions, AgentStreamOptions } from '@mastra/core/agent';
15
+ import type { NewWorkflow, WatchEvent, WorkflowResult as VNextWorkflowResult } from '@mastra/core/workflows/vNext';
14
16
  import type { JSONSchema7 } from 'json-schema';
15
17
  import type { ZodSchema } from 'zod';
16
18
 
@@ -46,14 +48,17 @@ export interface GetAgentResponse {
46
48
 
47
49
  export type GenerateParams<T extends JSONSchema7 | ZodSchema | undefined = undefined> = {
48
50
  messages: string | string[] | CoreMessage[] | AiMessageType[];
49
- } & Partial<AgentGenerateOptions<T>>;
51
+ } & Partial<Omit<AgentGenerateOptions<T>, 'experimental_generateMessageId'>>;
50
52
 
51
53
  export type StreamParams<T extends JSONSchema7 | ZodSchema | undefined = undefined> = {
52
54
  messages: string | string[] | CoreMessage[] | AiMessageType[];
53
- } & Omit<AgentStreamOptions<T>, 'onFinish' | 'onStepFinish' | 'telemetry'>;
55
+ } & Omit<AgentStreamOptions<T>, 'onFinish' | 'onStepFinish' | 'telemetry' | 'experimental_generateMessageId'>;
54
56
 
55
57
  export interface GetEvalsByAgentIdResponse extends GetAgentResponse {
56
58
  evals: any[];
59
+ instructions: string;
60
+ name: string;
61
+ id: string;
57
62
  }
58
63
 
59
64
  export interface GetToolResponse {
@@ -72,12 +77,43 @@ export interface GetWorkflowResponse {
72
77
  workflowId?: string;
73
78
  }
74
79
 
80
+ export interface GetWorkflowRunsParams {
81
+ fromDate?: Date;
82
+ toDate?: Date;
83
+ limit?: number;
84
+ offset?: number;
85
+ resourceId?: string;
86
+ }
87
+
88
+ export type GetWorkflowRunsResponse = WorkflowRuns;
89
+
75
90
  export type WorkflowRunResult = {
76
91
  activePaths: Record<string, { status: string; suspendPayload?: any; stepPath: string[] }>;
77
92
  results: CoreWorkflowRunResult<any, any, any>['results'];
78
93
  timestamp: number;
79
94
  runId: string;
80
95
  };
96
+
97
+ export interface GetVNextWorkflowResponse {
98
+ name: string;
99
+ steps: {
100
+ [key: string]: {
101
+ id: string;
102
+ description: string;
103
+ inputSchema: string;
104
+ outputSchema: string;
105
+ resumeSchema: string;
106
+ suspendSchema: string;
107
+ };
108
+ };
109
+ stepGraph: NewWorkflow['serializedStepGraph'];
110
+ inputSchema: string;
111
+ outputSchema: string;
112
+ }
113
+
114
+ export type VNextWorkflowWatchResult = WatchEvent & { runId: string };
115
+
116
+ export type VNextWorkflowRunResult = VNextWorkflowResult<any, any>;
81
117
  export interface UpsertVectorParams {
82
118
  indexName: string;
83
119
  vectors: number[][];
@@ -118,7 +154,7 @@ export type SaveMessageToMemoryResponse = MessageType[];
118
154
  export interface CreateMemoryThreadParams {
119
155
  title: string;
120
156
  metadata: Record<string, any>;
121
- resourceid: string;
157
+ resourceId: string;
122
158
  threadId: string;
123
159
  agentId: string;
124
160
  }
@@ -135,7 +171,7 @@ export type GetMemoryThreadResponse = StorageThreadType[];
135
171
  export interface UpdateMemoryThreadParams {
136
172
  title: string;
137
173
  metadata: Record<string, any>;
138
- resourceid: string;
174
+ resourceId: string;
139
175
  }
140
176
 
141
177
  export interface GetMemoryThreadMessagesResponse {
@@ -197,7 +233,7 @@ type Span = {
197
233
  };
198
234
 
199
235
  export interface GetTelemetryResponse {
200
- traces: { traces: Span[] };
236
+ traces: Span[];
201
237
  }
202
238
 
203
239
  export interface GetTelemetryParams {
@@ -206,6 +242,8 @@ export interface GetTelemetryParams {
206
242
  page?: number;
207
243
  perPage?: number;
208
244
  attribute?: Record<string, string>;
245
+ fromDate?: Date;
246
+ toDate?: Date;
209
247
  }
210
248
 
211
249
  export interface GetNetworkResponse {
@@ -0,0 +1,10 @@
1
+ import { ZodSchema } from 'zod';
2
+ import originalZodToJsonSchema from 'zod-to-json-schema';
3
+
4
+ export function zodToJsonSchema<T extends ZodSchema | any>(zodSchema: T) {
5
+ if (!(zodSchema instanceof ZodSchema)) {
6
+ return zodSchema;
7
+ }
8
+
9
+ return originalZodToJsonSchema(zodSchema, { $refStrategy: 'none' });
10
+ }