@mastra/core 0.10.6-alpha.0 → 0.10.6-alpha.2
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/dist/agent/index.cjs +2 -2
- package/dist/agent/index.d.cts +1 -1
- package/dist/agent/index.d.ts +1 -1
- package/dist/agent/index.js +1 -1
- package/dist/{base-DF29Fhcx.d.ts → base-BGLE5ewf.d.ts} +37 -34
- package/dist/{base-vDxe4rcD.d.cts → base-BP-zf-hZ.d.cts} +37 -34
- package/dist/{chunk-BMED2N3R.js → chunk-2XKJ7TUK.js} +1 -1
- package/dist/{chunk-U3F3AQJP.js → chunk-6ACH5N7Q.js} +28 -1
- package/dist/{chunk-VRDDTY4O.cjs → chunk-6U377XOJ.cjs} +2 -2
- package/dist/{chunk-QQT4XTOG.cjs → chunk-AQRPDE2V.cjs} +24 -11
- package/dist/{chunk-6PQRSQZF.js → chunk-BE2B3X5T.js} +5 -5
- package/dist/{chunk-HIQ3FPBN.js → chunk-CD2WGTW5.js} +1 -1
- package/dist/{chunk-F5XU7FFR.cjs → chunk-CEM46IBM.cjs} +4 -4
- package/dist/{chunk-QQ3KHYKE.cjs → chunk-JDOKE3OQ.cjs} +5 -5
- package/dist/{chunk-DNK4SXPA.js → chunk-K5SVOEUZ.js} +1 -1
- package/dist/{chunk-VZDPK54I.js → chunk-ODKQT6IG.js} +21 -8
- package/dist/{chunk-YLCFUSJ4.js → chunk-PYJKBUWB.js} +1 -1
- package/dist/{chunk-EFJIVMMM.cjs → chunk-UCKXM3N7.cjs} +21 -21
- package/dist/{chunk-EXOTT52R.cjs → chunk-WC7XENI5.cjs} +29 -2
- package/dist/{chunk-F2DBWJK3.js → chunk-WQYP3JIQ.js} +1 -1
- package/dist/{chunk-D2ULEJL3.cjs → chunk-WZRMRYK5.cjs} +2 -2
- package/dist/{chunk-LRMKKZ6M.cjs → chunk-X2JO3IZM.cjs} +2 -2
- package/dist/eval/index.d.cts +1 -1
- package/dist/eval/index.d.ts +1 -1
- package/dist/index.cjs +28 -28
- package/dist/index.d.cts +2 -2
- package/dist/index.d.ts +2 -2
- package/dist/index.js +8 -8
- package/dist/integration/index.d.cts +1 -1
- package/dist/integration/index.d.ts +1 -1
- package/dist/llm/index.d.cts +1 -1
- package/dist/llm/index.d.ts +1 -1
- package/dist/mastra/index.cjs +2 -2
- package/dist/mastra/index.d.cts +1 -1
- package/dist/mastra/index.d.ts +1 -1
- package/dist/mastra/index.js +1 -1
- package/dist/mcp/index.d.cts +1 -1
- package/dist/mcp/index.d.ts +1 -1
- package/dist/memory/index.d.cts +1 -1
- package/dist/memory/index.d.ts +1 -1
- package/dist/network/index.cjs +2 -2
- package/dist/network/index.d.cts +1 -1
- package/dist/network/index.d.ts +1 -1
- package/dist/network/index.js +1 -1
- package/dist/relevance/index.cjs +4 -4
- package/dist/relevance/index.d.cts +1 -1
- package/dist/relevance/index.d.ts +1 -1
- package/dist/relevance/index.js +1 -1
- package/dist/server/index.cjs +2 -2
- package/dist/server/index.d.cts +2 -2
- package/dist/server/index.d.ts +2 -2
- package/dist/server/index.js +1 -1
- package/dist/storage/index.cjs +10 -10
- package/dist/storage/index.d.cts +2 -2
- package/dist/storage/index.d.ts +2 -2
- package/dist/storage/index.js +3 -3
- package/dist/telemetry/index.cjs +7 -7
- package/dist/telemetry/index.d.cts +1 -1
- package/dist/telemetry/index.d.ts +1 -1
- package/dist/telemetry/index.js +1 -1
- package/dist/tools/index.d.cts +2 -2
- package/dist/tools/index.d.ts +2 -2
- package/dist/tts/index.cjs +2 -2
- package/dist/tts/index.js +1 -1
- package/dist/utils.d.cts +1 -1
- package/dist/utils.d.ts +1 -1
- package/dist/voice/index.cjs +4 -4
- package/dist/voice/index.d.cts +1 -1
- package/dist/voice/index.d.ts +1 -1
- package/dist/voice/index.js +1 -1
- package/dist/workflows/index.cjs +9 -9
- package/dist/workflows/index.d.cts +2 -2
- package/dist/workflows/index.d.ts +2 -2
- package/dist/workflows/index.js +1 -1
- package/dist/workflows/legacy/index.cjs +22 -22
- package/dist/workflows/legacy/index.d.cts +2 -2
- package/dist/workflows/legacy/index.d.ts +2 -2
- package/dist/workflows/legacy/index.js +1 -1
- package/package.json +1 -1
package/dist/agent/index.cjs
CHANGED
|
@@ -1,13 +1,13 @@
|
|
|
1
1
|
'use strict';
|
|
2
2
|
|
|
3
|
-
var
|
|
3
|
+
var chunkAQRPDE2V_cjs = require('../chunk-AQRPDE2V.cjs');
|
|
4
4
|
var chunkTGK6JO2D_cjs = require('../chunk-TGK6JO2D.cjs');
|
|
5
5
|
|
|
6
6
|
|
|
7
7
|
|
|
8
8
|
Object.defineProperty(exports, "Agent", {
|
|
9
9
|
enumerable: true,
|
|
10
|
-
get: function () { return
|
|
10
|
+
get: function () { return chunkAQRPDE2V_cjs.Agent; }
|
|
11
11
|
});
|
|
12
12
|
Object.defineProperty(exports, "MessageList", {
|
|
13
13
|
enumerable: true,
|
package/dist/agent/index.d.cts
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
export { Message as AiMessageType } from 'ai';
|
|
2
2
|
import 'json-schema';
|
|
3
3
|
import 'zod';
|
|
4
|
-
export { A as Agent, as as AgentConfig, bV as AgentGenerateOptions, bW as AgentStreamOptions, bU as DynamicArgument, bs as MastraLanguageModel, bX as MastraMessageContentV2, s as MastraMessageV2, bS as MessageList, T as ToolsInput, bT as ToolsetsInput } from '../base-
|
|
4
|
+
export { A as Agent, as as AgentConfig, bV as AgentGenerateOptions, bW as AgentStreamOptions, bU as DynamicArgument, bs as MastraLanguageModel, bX as MastraMessageContentV2, s as MastraMessageV2, bS as MessageList, T as ToolsInput, bT as ToolsetsInput } from '../base-BP-zf-hZ.cjs';
|
|
5
5
|
import '../base-DVVTzIQy.cjs';
|
|
6
6
|
import '../types-Bo1uigWx.cjs';
|
|
7
7
|
import '../runtime-context/index.cjs';
|
package/dist/agent/index.d.ts
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
export { Message as AiMessageType } from 'ai';
|
|
2
2
|
import 'json-schema';
|
|
3
3
|
import 'zod';
|
|
4
|
-
export { A as Agent, as as AgentConfig, bV as AgentGenerateOptions, bW as AgentStreamOptions, bU as DynamicArgument, bs as MastraLanguageModel, bX as MastraMessageContentV2, s as MastraMessageV2, bS as MessageList, T as ToolsInput, bT as ToolsetsInput } from '../base-
|
|
4
|
+
export { A as Agent, as as AgentConfig, bV as AgentGenerateOptions, bW as AgentStreamOptions, bU as DynamicArgument, bs as MastraLanguageModel, bX as MastraMessageContentV2, s as MastraMessageV2, bS as MessageList, T as ToolsInput, bT as ToolsetsInput } from '../base-BGLE5ewf.js';
|
|
5
5
|
import '../base-BbdvunNJ.js';
|
|
6
6
|
import '../types-Bo1uigWx.js';
|
|
7
7
|
import '../runtime-context/index.js';
|
package/dist/agent/index.js
CHANGED
|
@@ -1,2 +1,2 @@
|
|
|
1
|
-
export { Agent } from '../chunk-
|
|
1
|
+
export { Agent } from '../chunk-ODKQT6IG.js';
|
|
2
2
|
export { MessageList } from '../chunk-DPKEVI6Q.js';
|
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import * as ai from 'ai';
|
|
2
|
-
import { Tool as Tool$1, Schema, ToolExecutionOptions, TextStreamPart, UserContent, AssistantContent, ToolContent, EmbeddingModel, CoreMessage as CoreMessage$1, UIMessage, IDGenerator as IDGenerator$1,
|
|
2
|
+
import { Tool as Tool$1, Schema, ToolExecutionOptions, TextStreamPart, UserContent, AssistantContent, ToolContent, EmbeddingModel, CoreMessage as CoreMessage$1, UIMessage, Message, IDGenerator as IDGenerator$1, CoreSystemMessage as CoreSystemMessage$1, LanguageModel as LanguageModel$1, GenerateTextResult, GenerateObjectResult, StreamTextResult, StreamObjectResult, DeepPartial, generateText, generateObject, streamText, streamObject, TelemetrySettings, CoreAssistantMessage as CoreAssistantMessage$1, CoreUserMessage as CoreUserMessage$1, CoreToolMessage as CoreToolMessage$1, EmbedResult as EmbedResult$1, EmbedManyResult as EmbedManyResult$1, LanguageModelV1, GenerateTextOnStepFinishCallback, StreamTextOnFinishCallback, StreamObjectOnFinishCallback, StreamTextOnStepFinishCallback } from 'ai';
|
|
3
3
|
import { M as MastraBase, a as Telemetry, O as OtelConfig, T as Trace } from './base-BbdvunNJ.js';
|
|
4
4
|
import { z, ZodSchema } from 'zod';
|
|
5
5
|
import { B as BaseLogMessage, R as RegisteredLogger, I as IMastraLogger, a as LogLevel } from './logger-DtVDdb81.js';
|
|
@@ -1487,6 +1487,7 @@ declare class Workflow<TEngineType = any, TSteps extends Step<string, any, any,
|
|
|
1487
1487
|
resourceId?: string;
|
|
1488
1488
|
}): Promise<WorkflowRuns>;
|
|
1489
1489
|
getWorkflowRunById(runId: string): Promise<WorkflowRun | null>;
|
|
1490
|
+
getWorkflowRunExecutionResult(runId: string): Promise<WatchEvent['payload']['workflowState'] | null>;
|
|
1490
1491
|
}
|
|
1491
1492
|
/**
|
|
1492
1493
|
* Represents a workflow run that can be executed
|
|
@@ -2350,9 +2351,10 @@ interface AgentConfig<TAgentId extends string = string, TTools extends ToolsInpu
|
|
|
2350
2351
|
}
|
|
2351
2352
|
/**
|
|
2352
2353
|
* Options for generating responses with an agent
|
|
2353
|
-
* @template
|
|
2354
|
+
* @template OUTPUT - The schema type for structured output (Zod schema or JSON schema)
|
|
2355
|
+
* @template EXPERIMENTAL_OUTPUT - The schema type for structured output generation alongside tool calls (Zod schema or JSON schema)
|
|
2354
2356
|
*/
|
|
2355
|
-
type AgentGenerateOptions<
|
|
2357
|
+
type AgentGenerateOptions<OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined, EXPERIMENTAL_OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined> = {
|
|
2356
2358
|
/** Optional instructions to override the agent's default instructions */
|
|
2357
2359
|
instructions?: string;
|
|
2358
2360
|
/** Additional tool sets that can be used for this generation */
|
|
@@ -2365,13 +2367,13 @@ type AgentGenerateOptions<Z extends ZodSchema | JSONSchema7 | undefined = undefi
|
|
|
2365
2367
|
/** Unique ID for this generation run */
|
|
2366
2368
|
runId?: string;
|
|
2367
2369
|
/** Callback fired after each generation step completes */
|
|
2368
|
-
onStepFinish?:
|
|
2370
|
+
onStepFinish?: OUTPUT extends undefined ? EXPERIMENTAL_OUTPUT extends undefined ? GenerateTextOnStepFinishCallback<any> : GenerateTextOnStepFinishCallback<any> : never;
|
|
2369
2371
|
/** Maximum number of steps allowed for generation */
|
|
2370
2372
|
maxSteps?: number;
|
|
2371
2373
|
/** Schema for structured output, does not work with tools, use experimental_output instead */
|
|
2372
|
-
output?: OutputType |
|
|
2374
|
+
output?: OutputType | OUTPUT;
|
|
2373
2375
|
/** Schema for structured output generation alongside tool calls. */
|
|
2374
|
-
experimental_output?:
|
|
2376
|
+
experimental_output?: EXPERIMENTAL_OUTPUT;
|
|
2375
2377
|
/** Controls how tools are selected during generation */
|
|
2376
2378
|
toolChoice?: 'auto' | 'none' | 'required' | {
|
|
2377
2379
|
type: 'tool';
|
|
@@ -2387,12 +2389,13 @@ type AgentGenerateOptions<Z extends ZodSchema | JSONSchema7 | undefined = undefi
|
|
|
2387
2389
|
} | {
|
|
2388
2390
|
resourceId: string;
|
|
2389
2391
|
threadId: string;
|
|
2390
|
-
}) & (
|
|
2392
|
+
}) & (OUTPUT extends undefined ? DefaultLLMTextOptions : DefaultLLMTextObjectOptions);
|
|
2391
2393
|
/**
|
|
2392
2394
|
* Options for streaming responses with an agent
|
|
2393
|
-
* @template
|
|
2395
|
+
* @template OUTPUT - The schema type for structured output (Zod schema or JSON schema)
|
|
2396
|
+
* @template EXPERIMENTAL_OUTPUT - The schema type for structured output generation alongside tool calls (Zod schema or JSON schema)
|
|
2394
2397
|
*/
|
|
2395
|
-
type AgentStreamOptions<
|
|
2398
|
+
type AgentStreamOptions<OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined, EXPERIMENTAL_OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined> = {
|
|
2396
2399
|
/** Optional instructions to override the agent's default instructions */
|
|
2397
2400
|
instructions?: string;
|
|
2398
2401
|
/** Additional tool sets that can be used for this generation */
|
|
@@ -2405,13 +2408,13 @@ type AgentStreamOptions<Z extends ZodSchema | JSONSchema7 | undefined = undefine
|
|
|
2405
2408
|
/** Unique ID for this generation run */
|
|
2406
2409
|
runId?: string;
|
|
2407
2410
|
/** Callback fired when streaming completes */
|
|
2408
|
-
onFinish?:
|
|
2411
|
+
onFinish?: OUTPUT extends undefined ? StreamTextOnFinishCallback<any> : OUTPUT extends ZodSchema ? StreamObjectOnFinishCallback<z.infer<OUTPUT>> : StreamObjectOnFinishCallback<any>;
|
|
2409
2412
|
/** Callback fired after each generation step completes */
|
|
2410
|
-
onStepFinish?:
|
|
2413
|
+
onStepFinish?: OUTPUT extends undefined ? EXPERIMENTAL_OUTPUT extends undefined ? StreamTextOnStepFinishCallback<any> : StreamTextOnStepFinishCallback<any> : never;
|
|
2411
2414
|
/** Maximum number of steps allowed for generation */
|
|
2412
2415
|
maxSteps?: number;
|
|
2413
2416
|
/** Schema for structured output */
|
|
2414
|
-
output?: OutputType |
|
|
2417
|
+
output?: OutputType | OUTPUT;
|
|
2415
2418
|
/** Temperature parameter for controlling randomness */
|
|
2416
2419
|
temperature?: number;
|
|
2417
2420
|
/** Controls how tools are selected during generation */
|
|
@@ -2420,7 +2423,7 @@ type AgentStreamOptions<Z extends ZodSchema | JSONSchema7 | undefined = undefine
|
|
|
2420
2423
|
toolName: string;
|
|
2421
2424
|
};
|
|
2422
2425
|
/** Experimental schema for structured output */
|
|
2423
|
-
experimental_output?:
|
|
2426
|
+
experimental_output?: EXPERIMENTAL_OUTPUT;
|
|
2424
2427
|
/** Telemetry settings */
|
|
2425
2428
|
telemetry?: TelemetrySettings;
|
|
2426
2429
|
/** RuntimeContext for dependency injection */
|
|
@@ -2431,7 +2434,7 @@ type AgentStreamOptions<Z extends ZodSchema | JSONSchema7 | undefined = undefine
|
|
|
2431
2434
|
} | {
|
|
2432
2435
|
resourceId: string;
|
|
2433
2436
|
threadId: string;
|
|
2434
|
-
}) & (
|
|
2437
|
+
}) & (OUTPUT extends undefined ? DefaultLLMStreamOptions : DefaultLLMStreamObjectOptions);
|
|
2435
2438
|
|
|
2436
2439
|
type AgentNetworkConfig = {
|
|
2437
2440
|
name: string;
|
|
@@ -3301,11 +3304,11 @@ declare class Agent<TAgentId extends string = string, TTools extends ToolsInput
|
|
|
3301
3304
|
*/
|
|
3302
3305
|
__setTools(tools: TTools): void;
|
|
3303
3306
|
generateTitleFromUserMessage({ message, runtimeContext, }: {
|
|
3304
|
-
message:
|
|
3307
|
+
message: string | MessageInput;
|
|
3305
3308
|
runtimeContext?: RuntimeContext;
|
|
3306
3309
|
}): Promise<string>;
|
|
3307
3310
|
getMostRecentUserMessage(messages: Array<UIMessage>): UIMessage | undefined;
|
|
3308
|
-
genTitle(userMessage:
|
|
3311
|
+
genTitle(userMessage: string | MessageInput | undefined, runtimeContext: RuntimeContext): Promise<string>;
|
|
3309
3312
|
fetchMemory({ threadId, thread: passedThread, memoryConfig, resourceId, runId, messageList, }: {
|
|
3310
3313
|
resourceId: string;
|
|
3311
3314
|
threadId: string;
|
|
@@ -3358,33 +3361,33 @@ declare class Agent<TAgentId extends string = string, TTools extends ToolsInput
|
|
|
3358
3361
|
messageList: MessageList;
|
|
3359
3362
|
}) => Promise<void>;
|
|
3360
3363
|
};
|
|
3361
|
-
generate<
|
|
3364
|
+
generate<OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined, EXPERIMENTAL_OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined>(messages: string | string[] | CoreMessage$1[] | Message[], args?: AgentGenerateOptions<OUTPUT, EXPERIMENTAL_OUTPUT> & {
|
|
3362
3365
|
output?: never;
|
|
3363
3366
|
experimental_output?: never;
|
|
3364
|
-
}): Promise<GenerateTextResult<any,
|
|
3365
|
-
generate<
|
|
3366
|
-
output?:
|
|
3367
|
+
}): Promise<GenerateTextResult<any, OUTPUT extends ZodSchema ? z.infer<OUTPUT> : unknown>>;
|
|
3368
|
+
generate<OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined, EXPERIMENTAL_OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined>(messages: string | string[] | CoreMessage$1[] | Message[], args?: AgentGenerateOptions<OUTPUT, EXPERIMENTAL_OUTPUT> & {
|
|
3369
|
+
output?: OUTPUT;
|
|
3367
3370
|
experimental_output?: never;
|
|
3368
|
-
}): Promise<GenerateObjectResult<
|
|
3369
|
-
generate<
|
|
3371
|
+
}): Promise<GenerateObjectResult<OUTPUT extends ZodSchema ? z.infer<OUTPUT> : unknown>>;
|
|
3372
|
+
generate<OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined, EXPERIMENTAL_OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined>(messages: string | string[] | CoreMessage$1[] | Message[], args?: AgentGenerateOptions<OUTPUT, EXPERIMENTAL_OUTPUT> & {
|
|
3370
3373
|
output?: never;
|
|
3371
|
-
experimental_output?:
|
|
3372
|
-
}): Promise<GenerateTextResult<any,
|
|
3373
|
-
object:
|
|
3374
|
+
experimental_output?: EXPERIMENTAL_OUTPUT;
|
|
3375
|
+
}): Promise<GenerateTextResult<any, OUTPUT extends ZodSchema ? z.infer<OUTPUT> : unknown> & {
|
|
3376
|
+
object: OUTPUT extends ZodSchema ? z.infer<OUTPUT> : EXPERIMENTAL_OUTPUT extends ZodSchema ? z.infer<EXPERIMENTAL_OUTPUT> : unknown;
|
|
3374
3377
|
}>;
|
|
3375
|
-
stream<
|
|
3378
|
+
stream<OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined, EXPERIMENTAL_OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined>(messages: string | string[] | CoreMessage$1[] | Message[], args?: AgentStreamOptions<OUTPUT, EXPERIMENTAL_OUTPUT> & {
|
|
3376
3379
|
output?: never;
|
|
3377
3380
|
experimental_output?: never;
|
|
3378
|
-
}): Promise<StreamTextResult<any,
|
|
3379
|
-
stream<
|
|
3380
|
-
output?:
|
|
3381
|
+
}): Promise<StreamTextResult<any, OUTPUT extends ZodSchema ? z.infer<OUTPUT> : unknown>>;
|
|
3382
|
+
stream<OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined, EXPERIMENTAL_OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined>(messages: string | string[] | CoreMessage$1[] | Message[], args?: AgentStreamOptions<OUTPUT, EXPERIMENTAL_OUTPUT> & {
|
|
3383
|
+
output?: OUTPUT;
|
|
3381
3384
|
experimental_output?: never;
|
|
3382
|
-
}): Promise<StreamObjectResult<any,
|
|
3383
|
-
stream<
|
|
3385
|
+
}): Promise<StreamObjectResult<any, OUTPUT extends ZodSchema ? z.infer<OUTPUT> : unknown, any>>;
|
|
3386
|
+
stream<OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined, EXPERIMENTAL_OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined>(messages: string | string[] | CoreMessage$1[] | Message[], args?: AgentStreamOptions<OUTPUT, EXPERIMENTAL_OUTPUT> & {
|
|
3384
3387
|
output?: never;
|
|
3385
|
-
experimental_output?:
|
|
3386
|
-
}): Promise<StreamTextResult<any,
|
|
3387
|
-
partialObjectStream: StreamTextResult<any,
|
|
3388
|
+
experimental_output?: EXPERIMENTAL_OUTPUT;
|
|
3389
|
+
}): Promise<StreamTextResult<any, OUTPUT extends ZodSchema ? z.infer<OUTPUT> : unknown> & {
|
|
3390
|
+
partialObjectStream: StreamTextResult<any, OUTPUT extends ZodSchema ? z.infer<OUTPUT> : EXPERIMENTAL_OUTPUT extends ZodSchema ? z.infer<EXPERIMENTAL_OUTPUT> : unknown>['experimental_partialOutputStream'];
|
|
3388
3391
|
}>;
|
|
3389
3392
|
/**
|
|
3390
3393
|
* Convert text to speech using the configured voice provider
|
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import * as ai from 'ai';
|
|
2
|
-
import { Tool as Tool$1, Schema, ToolExecutionOptions, TextStreamPart, UserContent, AssistantContent, ToolContent, EmbeddingModel, CoreMessage as CoreMessage$1, UIMessage, IDGenerator as IDGenerator$1,
|
|
2
|
+
import { Tool as Tool$1, Schema, ToolExecutionOptions, TextStreamPart, UserContent, AssistantContent, ToolContent, EmbeddingModel, CoreMessage as CoreMessage$1, UIMessage, Message, IDGenerator as IDGenerator$1, CoreSystemMessage as CoreSystemMessage$1, LanguageModel as LanguageModel$1, GenerateTextResult, GenerateObjectResult, StreamTextResult, StreamObjectResult, DeepPartial, generateText, generateObject, streamText, streamObject, TelemetrySettings, CoreAssistantMessage as CoreAssistantMessage$1, CoreUserMessage as CoreUserMessage$1, CoreToolMessage as CoreToolMessage$1, EmbedResult as EmbedResult$1, EmbedManyResult as EmbedManyResult$1, LanguageModelV1, GenerateTextOnStepFinishCallback, StreamTextOnFinishCallback, StreamObjectOnFinishCallback, StreamTextOnStepFinishCallback } from 'ai';
|
|
3
3
|
import { M as MastraBase, a as Telemetry, O as OtelConfig, T as Trace } from './base-DVVTzIQy.cjs';
|
|
4
4
|
import { z, ZodSchema } from 'zod';
|
|
5
5
|
import { B as BaseLogMessage, R as RegisteredLogger, I as IMastraLogger, a as LogLevel } from './logger-CpL0z5v_.cjs';
|
|
@@ -1487,6 +1487,7 @@ declare class Workflow<TEngineType = any, TSteps extends Step<string, any, any,
|
|
|
1487
1487
|
resourceId?: string;
|
|
1488
1488
|
}): Promise<WorkflowRuns>;
|
|
1489
1489
|
getWorkflowRunById(runId: string): Promise<WorkflowRun | null>;
|
|
1490
|
+
getWorkflowRunExecutionResult(runId: string): Promise<WatchEvent['payload']['workflowState'] | null>;
|
|
1490
1491
|
}
|
|
1491
1492
|
/**
|
|
1492
1493
|
* Represents a workflow run that can be executed
|
|
@@ -2350,9 +2351,10 @@ interface AgentConfig<TAgentId extends string = string, TTools extends ToolsInpu
|
|
|
2350
2351
|
}
|
|
2351
2352
|
/**
|
|
2352
2353
|
* Options for generating responses with an agent
|
|
2353
|
-
* @template
|
|
2354
|
+
* @template OUTPUT - The schema type for structured output (Zod schema or JSON schema)
|
|
2355
|
+
* @template EXPERIMENTAL_OUTPUT - The schema type for structured output generation alongside tool calls (Zod schema or JSON schema)
|
|
2354
2356
|
*/
|
|
2355
|
-
type AgentGenerateOptions<
|
|
2357
|
+
type AgentGenerateOptions<OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined, EXPERIMENTAL_OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined> = {
|
|
2356
2358
|
/** Optional instructions to override the agent's default instructions */
|
|
2357
2359
|
instructions?: string;
|
|
2358
2360
|
/** Additional tool sets that can be used for this generation */
|
|
@@ -2365,13 +2367,13 @@ type AgentGenerateOptions<Z extends ZodSchema | JSONSchema7 | undefined = undefi
|
|
|
2365
2367
|
/** Unique ID for this generation run */
|
|
2366
2368
|
runId?: string;
|
|
2367
2369
|
/** Callback fired after each generation step completes */
|
|
2368
|
-
onStepFinish?:
|
|
2370
|
+
onStepFinish?: OUTPUT extends undefined ? EXPERIMENTAL_OUTPUT extends undefined ? GenerateTextOnStepFinishCallback<any> : GenerateTextOnStepFinishCallback<any> : never;
|
|
2369
2371
|
/** Maximum number of steps allowed for generation */
|
|
2370
2372
|
maxSteps?: number;
|
|
2371
2373
|
/** Schema for structured output, does not work with tools, use experimental_output instead */
|
|
2372
|
-
output?: OutputType |
|
|
2374
|
+
output?: OutputType | OUTPUT;
|
|
2373
2375
|
/** Schema for structured output generation alongside tool calls. */
|
|
2374
|
-
experimental_output?:
|
|
2376
|
+
experimental_output?: EXPERIMENTAL_OUTPUT;
|
|
2375
2377
|
/** Controls how tools are selected during generation */
|
|
2376
2378
|
toolChoice?: 'auto' | 'none' | 'required' | {
|
|
2377
2379
|
type: 'tool';
|
|
@@ -2387,12 +2389,13 @@ type AgentGenerateOptions<Z extends ZodSchema | JSONSchema7 | undefined = undefi
|
|
|
2387
2389
|
} | {
|
|
2388
2390
|
resourceId: string;
|
|
2389
2391
|
threadId: string;
|
|
2390
|
-
}) & (
|
|
2392
|
+
}) & (OUTPUT extends undefined ? DefaultLLMTextOptions : DefaultLLMTextObjectOptions);
|
|
2391
2393
|
/**
|
|
2392
2394
|
* Options for streaming responses with an agent
|
|
2393
|
-
* @template
|
|
2395
|
+
* @template OUTPUT - The schema type for structured output (Zod schema or JSON schema)
|
|
2396
|
+
* @template EXPERIMENTAL_OUTPUT - The schema type for structured output generation alongside tool calls (Zod schema or JSON schema)
|
|
2394
2397
|
*/
|
|
2395
|
-
type AgentStreamOptions<
|
|
2398
|
+
type AgentStreamOptions<OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined, EXPERIMENTAL_OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined> = {
|
|
2396
2399
|
/** Optional instructions to override the agent's default instructions */
|
|
2397
2400
|
instructions?: string;
|
|
2398
2401
|
/** Additional tool sets that can be used for this generation */
|
|
@@ -2405,13 +2408,13 @@ type AgentStreamOptions<Z extends ZodSchema | JSONSchema7 | undefined = undefine
|
|
|
2405
2408
|
/** Unique ID for this generation run */
|
|
2406
2409
|
runId?: string;
|
|
2407
2410
|
/** Callback fired when streaming completes */
|
|
2408
|
-
onFinish?:
|
|
2411
|
+
onFinish?: OUTPUT extends undefined ? StreamTextOnFinishCallback<any> : OUTPUT extends ZodSchema ? StreamObjectOnFinishCallback<z.infer<OUTPUT>> : StreamObjectOnFinishCallback<any>;
|
|
2409
2412
|
/** Callback fired after each generation step completes */
|
|
2410
|
-
onStepFinish?:
|
|
2413
|
+
onStepFinish?: OUTPUT extends undefined ? EXPERIMENTAL_OUTPUT extends undefined ? StreamTextOnStepFinishCallback<any> : StreamTextOnStepFinishCallback<any> : never;
|
|
2411
2414
|
/** Maximum number of steps allowed for generation */
|
|
2412
2415
|
maxSteps?: number;
|
|
2413
2416
|
/** Schema for structured output */
|
|
2414
|
-
output?: OutputType |
|
|
2417
|
+
output?: OutputType | OUTPUT;
|
|
2415
2418
|
/** Temperature parameter for controlling randomness */
|
|
2416
2419
|
temperature?: number;
|
|
2417
2420
|
/** Controls how tools are selected during generation */
|
|
@@ -2420,7 +2423,7 @@ type AgentStreamOptions<Z extends ZodSchema | JSONSchema7 | undefined = undefine
|
|
|
2420
2423
|
toolName: string;
|
|
2421
2424
|
};
|
|
2422
2425
|
/** Experimental schema for structured output */
|
|
2423
|
-
experimental_output?:
|
|
2426
|
+
experimental_output?: EXPERIMENTAL_OUTPUT;
|
|
2424
2427
|
/** Telemetry settings */
|
|
2425
2428
|
telemetry?: TelemetrySettings;
|
|
2426
2429
|
/** RuntimeContext for dependency injection */
|
|
@@ -2431,7 +2434,7 @@ type AgentStreamOptions<Z extends ZodSchema | JSONSchema7 | undefined = undefine
|
|
|
2431
2434
|
} | {
|
|
2432
2435
|
resourceId: string;
|
|
2433
2436
|
threadId: string;
|
|
2434
|
-
}) & (
|
|
2437
|
+
}) & (OUTPUT extends undefined ? DefaultLLMStreamOptions : DefaultLLMStreamObjectOptions);
|
|
2435
2438
|
|
|
2436
2439
|
type AgentNetworkConfig = {
|
|
2437
2440
|
name: string;
|
|
@@ -3301,11 +3304,11 @@ declare class Agent<TAgentId extends string = string, TTools extends ToolsInput
|
|
|
3301
3304
|
*/
|
|
3302
3305
|
__setTools(tools: TTools): void;
|
|
3303
3306
|
generateTitleFromUserMessage({ message, runtimeContext, }: {
|
|
3304
|
-
message:
|
|
3307
|
+
message: string | MessageInput;
|
|
3305
3308
|
runtimeContext?: RuntimeContext;
|
|
3306
3309
|
}): Promise<string>;
|
|
3307
3310
|
getMostRecentUserMessage(messages: Array<UIMessage>): UIMessage | undefined;
|
|
3308
|
-
genTitle(userMessage:
|
|
3311
|
+
genTitle(userMessage: string | MessageInput | undefined, runtimeContext: RuntimeContext): Promise<string>;
|
|
3309
3312
|
fetchMemory({ threadId, thread: passedThread, memoryConfig, resourceId, runId, messageList, }: {
|
|
3310
3313
|
resourceId: string;
|
|
3311
3314
|
threadId: string;
|
|
@@ -3358,33 +3361,33 @@ declare class Agent<TAgentId extends string = string, TTools extends ToolsInput
|
|
|
3358
3361
|
messageList: MessageList;
|
|
3359
3362
|
}) => Promise<void>;
|
|
3360
3363
|
};
|
|
3361
|
-
generate<
|
|
3364
|
+
generate<OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined, EXPERIMENTAL_OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined>(messages: string | string[] | CoreMessage$1[] | Message[], args?: AgentGenerateOptions<OUTPUT, EXPERIMENTAL_OUTPUT> & {
|
|
3362
3365
|
output?: never;
|
|
3363
3366
|
experimental_output?: never;
|
|
3364
|
-
}): Promise<GenerateTextResult<any,
|
|
3365
|
-
generate<
|
|
3366
|
-
output?:
|
|
3367
|
+
}): Promise<GenerateTextResult<any, OUTPUT extends ZodSchema ? z.infer<OUTPUT> : unknown>>;
|
|
3368
|
+
generate<OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined, EXPERIMENTAL_OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined>(messages: string | string[] | CoreMessage$1[] | Message[], args?: AgentGenerateOptions<OUTPUT, EXPERIMENTAL_OUTPUT> & {
|
|
3369
|
+
output?: OUTPUT;
|
|
3367
3370
|
experimental_output?: never;
|
|
3368
|
-
}): Promise<GenerateObjectResult<
|
|
3369
|
-
generate<
|
|
3371
|
+
}): Promise<GenerateObjectResult<OUTPUT extends ZodSchema ? z.infer<OUTPUT> : unknown>>;
|
|
3372
|
+
generate<OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined, EXPERIMENTAL_OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined>(messages: string | string[] | CoreMessage$1[] | Message[], args?: AgentGenerateOptions<OUTPUT, EXPERIMENTAL_OUTPUT> & {
|
|
3370
3373
|
output?: never;
|
|
3371
|
-
experimental_output?:
|
|
3372
|
-
}): Promise<GenerateTextResult<any,
|
|
3373
|
-
object:
|
|
3374
|
+
experimental_output?: EXPERIMENTAL_OUTPUT;
|
|
3375
|
+
}): Promise<GenerateTextResult<any, OUTPUT extends ZodSchema ? z.infer<OUTPUT> : unknown> & {
|
|
3376
|
+
object: OUTPUT extends ZodSchema ? z.infer<OUTPUT> : EXPERIMENTAL_OUTPUT extends ZodSchema ? z.infer<EXPERIMENTAL_OUTPUT> : unknown;
|
|
3374
3377
|
}>;
|
|
3375
|
-
stream<
|
|
3378
|
+
stream<OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined, EXPERIMENTAL_OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined>(messages: string | string[] | CoreMessage$1[] | Message[], args?: AgentStreamOptions<OUTPUT, EXPERIMENTAL_OUTPUT> & {
|
|
3376
3379
|
output?: never;
|
|
3377
3380
|
experimental_output?: never;
|
|
3378
|
-
}): Promise<StreamTextResult<any,
|
|
3379
|
-
stream<
|
|
3380
|
-
output?:
|
|
3381
|
+
}): Promise<StreamTextResult<any, OUTPUT extends ZodSchema ? z.infer<OUTPUT> : unknown>>;
|
|
3382
|
+
stream<OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined, EXPERIMENTAL_OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined>(messages: string | string[] | CoreMessage$1[] | Message[], args?: AgentStreamOptions<OUTPUT, EXPERIMENTAL_OUTPUT> & {
|
|
3383
|
+
output?: OUTPUT;
|
|
3381
3384
|
experimental_output?: never;
|
|
3382
|
-
}): Promise<StreamObjectResult<any,
|
|
3383
|
-
stream<
|
|
3385
|
+
}): Promise<StreamObjectResult<any, OUTPUT extends ZodSchema ? z.infer<OUTPUT> : unknown, any>>;
|
|
3386
|
+
stream<OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined, EXPERIMENTAL_OUTPUT extends ZodSchema | JSONSchema7 | undefined = undefined>(messages: string | string[] | CoreMessage$1[] | Message[], args?: AgentStreamOptions<OUTPUT, EXPERIMENTAL_OUTPUT> & {
|
|
3384
3387
|
output?: never;
|
|
3385
|
-
experimental_output?:
|
|
3386
|
-
}): Promise<StreamTextResult<any,
|
|
3387
|
-
partialObjectStream: StreamTextResult<any,
|
|
3388
|
+
experimental_output?: EXPERIMENTAL_OUTPUT;
|
|
3389
|
+
}): Promise<StreamTextResult<any, OUTPUT extends ZodSchema ? z.infer<OUTPUT> : unknown> & {
|
|
3390
|
+
partialObjectStream: StreamTextResult<any, OUTPUT extends ZodSchema ? z.infer<OUTPUT> : EXPERIMENTAL_OUTPUT extends ZodSchema ? z.infer<EXPERIMENTAL_OUTPUT> : unknown>['experimental_partialOutputStream'];
|
|
3388
3391
|
}>;
|
|
3389
3392
|
/**
|
|
3390
3393
|
* Convert text to speech using the configured voice provider
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import { TABLE_WORKFLOW_SNAPSHOT, TABLE_EVALS, TABLE_MESSAGES, TABLE_THREADS, TABLE_TRACES, TABLE_SCHEMAS } from './chunk-
|
|
1
|
+
import { TABLE_WORKFLOW_SNAPSHOT, TABLE_EVALS, TABLE_MESSAGES, TABLE_THREADS, TABLE_TRACES, TABLE_SCHEMAS } from './chunk-BE2B3X5T.js';
|
|
2
2
|
import { MastraBase } from './chunk-5IEKR756.js';
|
|
3
3
|
|
|
4
4
|
// src/storage/base.ts
|
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import { EMITTER_SYMBOL } from './chunk-Y7D2JLKS.js';
|
|
2
|
-
import { Agent } from './chunk-
|
|
2
|
+
import { Agent } from './chunk-ODKQT6IG.js';
|
|
3
3
|
import { Tool } from './chunk-C4LMN2IR.js';
|
|
4
4
|
import { MastraError } from './chunk-LDNCERLU.js';
|
|
5
5
|
import { MastraBase } from './chunk-5IEKR756.js';
|
|
@@ -1382,6 +1382,33 @@ var Workflow = class extends MastraBase {
|
|
|
1382
1382
|
const run = await storage.getWorkflowRunById({ runId, workflowName: this.id });
|
|
1383
1383
|
return run ?? (this.#runs.get(runId) ? { ...this.#runs.get(runId), workflowName: this.id } : null);
|
|
1384
1384
|
}
|
|
1385
|
+
async getWorkflowRunExecutionResult(runId) {
|
|
1386
|
+
const storage = this.#mastra?.getStorage();
|
|
1387
|
+
if (!storage) {
|
|
1388
|
+
this.logger.debug("Cannot get workflow run execution result. Mastra storage is not initialized");
|
|
1389
|
+
return null;
|
|
1390
|
+
}
|
|
1391
|
+
const run = await storage.getWorkflowRunById({ runId, workflowName: this.id });
|
|
1392
|
+
let snapshot = run?.snapshot;
|
|
1393
|
+
if (!snapshot) {
|
|
1394
|
+
return null;
|
|
1395
|
+
}
|
|
1396
|
+
if (typeof snapshot === "string") {
|
|
1397
|
+
try {
|
|
1398
|
+
snapshot = JSON.parse(snapshot);
|
|
1399
|
+
} catch (e) {
|
|
1400
|
+
this.logger.debug("Cannot get workflow run execution result. Snapshot is not a valid JSON string", e);
|
|
1401
|
+
return null;
|
|
1402
|
+
}
|
|
1403
|
+
}
|
|
1404
|
+
return {
|
|
1405
|
+
status: snapshot.status,
|
|
1406
|
+
result: snapshot.result,
|
|
1407
|
+
error: snapshot.error,
|
|
1408
|
+
payload: snapshot.context?.input,
|
|
1409
|
+
steps: snapshot.context
|
|
1410
|
+
};
|
|
1411
|
+
}
|
|
1385
1412
|
};
|
|
1386
1413
|
var Run = class {
|
|
1387
1414
|
emitter;
|
|
@@ -1,13 +1,13 @@
|
|
|
1
1
|
'use strict';
|
|
2
2
|
|
|
3
|
-
var
|
|
3
|
+
var chunkJDOKE3OQ_cjs = require('./chunk-JDOKE3OQ.cjs');
|
|
4
4
|
var chunkOLSQRA2V_cjs = require('./chunk-OLSQRA2V.cjs');
|
|
5
5
|
var chunkP3Q73CAW_cjs = require('./chunk-P3Q73CAW.cjs');
|
|
6
6
|
var chunkRWTSGWWL_cjs = require('./chunk-RWTSGWWL.cjs');
|
|
7
7
|
|
|
8
8
|
// src/voice/voice.ts
|
|
9
9
|
var _MastraVoice_decorators, _init, _a;
|
|
10
|
-
_MastraVoice_decorators = [
|
|
10
|
+
_MastraVoice_decorators = [chunkJDOKE3OQ_cjs.InstrumentClass({
|
|
11
11
|
prefix: "voice",
|
|
12
12
|
excludeMethods: ["__setTools", "__setLogger", "__setTelemetry", "#log"]
|
|
13
13
|
})];
|
|
@@ -1,10 +1,10 @@
|
|
|
1
1
|
'use strict';
|
|
2
2
|
|
|
3
|
-
var
|
|
3
|
+
var chunk6U377XOJ_cjs = require('./chunk-6U377XOJ.cjs');
|
|
4
4
|
var chunkTGK6JO2D_cjs = require('./chunk-TGK6JO2D.cjs');
|
|
5
5
|
var chunkST5RMVLG_cjs = require('./chunk-ST5RMVLG.cjs');
|
|
6
6
|
var chunkHVI3SKIH_cjs = require('./chunk-HVI3SKIH.cjs');
|
|
7
|
-
var
|
|
7
|
+
var chunkJDOKE3OQ_cjs = require('./chunk-JDOKE3OQ.cjs');
|
|
8
8
|
var chunkJ7MLMW6O_cjs = require('./chunk-J7MLMW6O.cjs');
|
|
9
9
|
var chunkOLSQRA2V_cjs = require('./chunk-OLSQRA2V.cjs');
|
|
10
10
|
var chunkP3Q73CAW_cjs = require('./chunk-P3Q73CAW.cjs');
|
|
@@ -68,7 +68,7 @@ function resolveMaybePromise(value, cb) {
|
|
|
68
68
|
return cb(value);
|
|
69
69
|
}
|
|
70
70
|
var _Agent_decorators, _init, _a;
|
|
71
|
-
_Agent_decorators = [
|
|
71
|
+
_Agent_decorators = [chunkJDOKE3OQ_cjs.InstrumentClass({
|
|
72
72
|
prefix: "agent",
|
|
73
73
|
excludeMethods: ["hasOwnMemory", "getMemory", "__primitive", "__registerMastra", "__registerPrimitives", "__setTools", "__setLogger", "__setTelemetry", "log", "getModel", "getInstructions", "getTools", "getLLM", "getWorkflows", "getDefaultGenerateOptions", "getDefaultStreamOptions", "getDescription"]
|
|
74
74
|
})];
|
|
@@ -146,7 +146,7 @@ exports.Agent = class Agent extends (_a = chunkP3Q73CAW_cjs.MastraBase) {
|
|
|
146
146
|
this.#voice?.addInstructions(config.instructions);
|
|
147
147
|
}
|
|
148
148
|
} else {
|
|
149
|
-
this.#voice = new
|
|
149
|
+
this.#voice = new chunk6U377XOJ_cjs.DefaultVoice();
|
|
150
150
|
}
|
|
151
151
|
}
|
|
152
152
|
hasOwnMemory() {
|
|
@@ -210,7 +210,7 @@ exports.Agent = class Agent extends (_a = chunkP3Q73CAW_cjs.MastraBase) {
|
|
|
210
210
|
}));
|
|
211
211
|
return voice;
|
|
212
212
|
} else {
|
|
213
|
-
return new
|
|
213
|
+
return new chunk6U377XOJ_cjs.DefaultVoice();
|
|
214
214
|
}
|
|
215
215
|
}
|
|
216
216
|
get instructions() {
|
|
@@ -443,8 +443,12 @@ exports.Agent = class Agent extends (_a = chunkP3Q73CAW_cjs.MastraBase) {
|
|
|
443
443
|
const llm = await this.getLLM({
|
|
444
444
|
runtimeContext
|
|
445
445
|
});
|
|
446
|
+
const normMessage = new chunkTGK6JO2D_cjs.MessageList().add(message, "user").get.all.ui().at(-1);
|
|
447
|
+
if (!normMessage) {
|
|
448
|
+
throw new Error(`Could not generate title from input ${JSON.stringify(message)}`);
|
|
449
|
+
}
|
|
446
450
|
const partsToGen = [];
|
|
447
|
-
for (const part of
|
|
451
|
+
for (const part of normMessage.parts) {
|
|
448
452
|
if (part.type === `text`) {
|
|
449
453
|
partsToGen.push(part);
|
|
450
454
|
} else if (part.type === `source`) {
|
|
@@ -488,13 +492,16 @@ exports.Agent = class Agent extends (_a = chunkP3Q73CAW_cjs.MastraBase) {
|
|
|
488
492
|
let title = `New Thread ${(/* @__PURE__ */new Date()).toISOString()}`;
|
|
489
493
|
try {
|
|
490
494
|
if (userMessage) {
|
|
491
|
-
|
|
492
|
-
|
|
493
|
-
|
|
494
|
-
|
|
495
|
+
const normMessage = new chunkTGK6JO2D_cjs.MessageList().add(userMessage, "user").get.all.ui().at(-1);
|
|
496
|
+
if (normMessage) {
|
|
497
|
+
title = await this.generateTitleFromUserMessage({
|
|
498
|
+
message: normMessage,
|
|
499
|
+
runtimeContext
|
|
500
|
+
});
|
|
501
|
+
}
|
|
495
502
|
}
|
|
496
503
|
} catch (e) {
|
|
497
|
-
|
|
504
|
+
this.logger.error("Error generating title:", e);
|
|
498
505
|
}
|
|
499
506
|
return title;
|
|
500
507
|
}
|
|
@@ -1328,6 +1335,8 @@ ${JSON.stringify(
|
|
|
1328
1335
|
telemetry,
|
|
1329
1336
|
memory: this.getMemory(),
|
|
1330
1337
|
runtimeContext,
|
|
1338
|
+
threadId,
|
|
1339
|
+
resourceId,
|
|
1331
1340
|
...args
|
|
1332
1341
|
});
|
|
1333
1342
|
const newStreamResult = streamResult;
|
|
@@ -1370,6 +1379,8 @@ ${JSON.stringify(
|
|
|
1370
1379
|
telemetry,
|
|
1371
1380
|
memory: this.getMemory(),
|
|
1372
1381
|
runtimeContext,
|
|
1382
|
+
threadId,
|
|
1383
|
+
resourceId,
|
|
1373
1384
|
...args
|
|
1374
1385
|
});
|
|
1375
1386
|
}
|
|
@@ -1409,6 +1420,8 @@ ${JSON.stringify(
|
|
|
1409
1420
|
telemetry,
|
|
1410
1421
|
memory: this.getMemory(),
|
|
1411
1422
|
runtimeContext,
|
|
1423
|
+
threadId,
|
|
1424
|
+
resourceId,
|
|
1412
1425
|
...args
|
|
1413
1426
|
});
|
|
1414
1427
|
}
|
|
@@ -373,10 +373,10 @@ var Telemetry = class _Telemetry {
|
|
|
373
373
|
*/
|
|
374
374
|
static init(config = {}) {
|
|
375
375
|
try {
|
|
376
|
-
if (!
|
|
377
|
-
|
|
376
|
+
if (!globalThis.__TELEMETRY__) {
|
|
377
|
+
globalThis.__TELEMETRY__ = new _Telemetry(config);
|
|
378
378
|
}
|
|
379
|
-
return
|
|
379
|
+
return globalThis.__TELEMETRY__;
|
|
380
380
|
} catch (error) {
|
|
381
381
|
const wrappedError = new MastraError(
|
|
382
382
|
{
|
|
@@ -400,7 +400,7 @@ var Telemetry = class _Telemetry {
|
|
|
400
400
|
* @returns {Telemetry} The global telemetry instance
|
|
401
401
|
*/
|
|
402
402
|
static get() {
|
|
403
|
-
if (!
|
|
403
|
+
if (!globalThis.__TELEMETRY__) {
|
|
404
404
|
throw new MastraError({
|
|
405
405
|
id: "TELEMETRY_GETTER_FAILED_GLOBAL_TELEMETRY_NOT_INITIALIZED",
|
|
406
406
|
text: "Telemetry not initialized",
|
|
@@ -408,7 +408,7 @@ var Telemetry = class _Telemetry {
|
|
|
408
408
|
category: "USER" /* USER */
|
|
409
409
|
});
|
|
410
410
|
}
|
|
411
|
-
return
|
|
411
|
+
return globalThis.__TELEMETRY__;
|
|
412
412
|
}
|
|
413
413
|
/**
|
|
414
414
|
* Wraps a class instance with telemetry tracing
|
|
@@ -1,14 +1,14 @@
|
|
|
1
1
|
'use strict';
|
|
2
2
|
|
|
3
3
|
var chunkLXFZUKP3_cjs = require('./chunk-LXFZUKP3.cjs');
|
|
4
|
-
var
|
|
4
|
+
var chunkJDOKE3OQ_cjs = require('./chunk-JDOKE3OQ.cjs');
|
|
5
5
|
var chunk2D7NOO7S_cjs = require('./chunk-2D7NOO7S.cjs');
|
|
6
6
|
var chunkZIZ3CVHN_cjs = require('./chunk-ZIZ3CVHN.cjs');
|
|
7
7
|
var chunkRWTSGWWL_cjs = require('./chunk-RWTSGWWL.cjs');
|
|
8
8
|
|
|
9
9
|
// src/mastra/index.ts
|
|
10
10
|
var _Mastra_decorators, _init;
|
|
11
|
-
_Mastra_decorators = [
|
|
11
|
+
_Mastra_decorators = [chunkJDOKE3OQ_cjs.InstrumentClass({
|
|
12
12
|
prefix: "mastra",
|
|
13
13
|
excludeMethods: ["getLogger", "getTelemetry"]
|
|
14
14
|
})];
|
|
@@ -72,7 +72,7 @@ exports.Mastra = class Mastra {
|
|
|
72
72
|
if (storage) {
|
|
73
73
|
storage = chunkLXFZUKP3_cjs.augmentWithInit(storage);
|
|
74
74
|
}
|
|
75
|
-
this.#telemetry =
|
|
75
|
+
this.#telemetry = chunkJDOKE3OQ_cjs.Telemetry.init(config?.telemetry);
|
|
76
76
|
if (this.#telemetry && storage) {
|
|
77
77
|
this.#storage = this.#telemetry.traceClass(storage, {
|
|
78
78
|
excludeMethods: ["__setTelemetry", "__getTelemetry", "batchTraceInsert", "getTraces", "getEvalsByAgentName"]
|
|
@@ -323,7 +323,7 @@ do:
|
|
|
323
323
|
}
|
|
324
324
|
}
|
|
325
325
|
setTelemetry(telemetry) {
|
|
326
|
-
this.#telemetry =
|
|
326
|
+
this.#telemetry = chunkJDOKE3OQ_cjs.Telemetry.init(telemetry);
|
|
327
327
|
if (this.#agents) {
|
|
328
328
|
Object.keys(this.#agents).forEach(key => {
|
|
329
329
|
if (this.#telemetry) {
|