@mastra/mssql 0.0.0-share-agent-metadata-with-cloud-20250718123411 → 0.0.0-span-scorring-test-20251124132129

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.
@@ -0,0 +1,69 @@
1
+ import type { MastraMessageContentV2 } from '@mastra/core/agent';
2
+ import type { MastraDBMessage, StorageThreadType } from '@mastra/core/memory';
3
+ import { MemoryStorage } from '@mastra/core/storage';
4
+ import type { StorageResourceType, StorageListMessagesInput, StorageListMessagesOutput, StorageListThreadsByResourceIdInput, StorageListThreadsByResourceIdOutput } from '@mastra/core/storage';
5
+ import sql from 'mssql';
6
+ import type { StoreOperationsMSSQL } from '../operations/index.js';
7
+ export declare class MemoryMSSQL extends MemoryStorage {
8
+ private pool;
9
+ private schema;
10
+ private operations;
11
+ private _parseAndFormatMessages;
12
+ constructor({ pool, schema, operations, }: {
13
+ pool: sql.ConnectionPool;
14
+ schema: string;
15
+ operations: StoreOperationsMSSQL;
16
+ });
17
+ getThreadById({ threadId }: {
18
+ threadId: string;
19
+ }): Promise<StorageThreadType | null>;
20
+ listThreadsByResourceId(args: StorageListThreadsByResourceIdInput): Promise<StorageListThreadsByResourceIdOutput>;
21
+ saveThread({ thread }: {
22
+ thread: StorageThreadType;
23
+ }): Promise<StorageThreadType>;
24
+ /**
25
+ * Updates a thread's title and metadata, merging with existing metadata. Returns the updated thread.
26
+ */
27
+ updateThread({ id, title, metadata, }: {
28
+ id: string;
29
+ title: string;
30
+ metadata: Record<string, unknown>;
31
+ }): Promise<StorageThreadType>;
32
+ deleteThread({ threadId }: {
33
+ threadId: string;
34
+ }): Promise<void>;
35
+ private _getIncludedMessages;
36
+ listMessagesById({ messageIds }: {
37
+ messageIds: string[];
38
+ }): Promise<{
39
+ messages: MastraDBMessage[];
40
+ }>;
41
+ listMessages(args: StorageListMessagesInput): Promise<StorageListMessagesOutput>;
42
+ saveMessages({ messages }: {
43
+ messages: MastraDBMessage[];
44
+ }): Promise<{
45
+ messages: MastraDBMessage[];
46
+ }>;
47
+ updateMessages({ messages, }: {
48
+ messages: (Partial<Omit<MastraDBMessage, 'createdAt'>> & {
49
+ id: string;
50
+ content?: {
51
+ metadata?: MastraMessageContentV2['metadata'];
52
+ content?: MastraMessageContentV2['content'];
53
+ };
54
+ })[];
55
+ }): Promise<MastraDBMessage[]>;
56
+ deleteMessages(messageIds: string[]): Promise<void>;
57
+ getResourceById({ resourceId }: {
58
+ resourceId: string;
59
+ }): Promise<StorageResourceType | null>;
60
+ saveResource({ resource }: {
61
+ resource: StorageResourceType;
62
+ }): Promise<StorageResourceType>;
63
+ updateResource({ resourceId, workingMemory, metadata, }: {
64
+ resourceId: string;
65
+ workingMemory?: string;
66
+ metadata?: Record<string, unknown>;
67
+ }): Promise<StorageResourceType>;
68
+ }
69
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/memory/index.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,oBAAoB,CAAC;AAEjE,OAAO,KAAK,EAAmB,eAAe,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAC/F,OAAO,EACL,aAAa,EAOd,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EACV,mBAAmB,EACnB,wBAAwB,EACxB,yBAAyB,EACzB,mCAAmC,EACnC,oCAAoC,EACrC,MAAM,sBAAsB,CAAC;AAC9B,OAAO,GAAG,MAAM,OAAO,CAAC;AACxB,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC;AAG1D,qBAAa,WAAY,SAAQ,aAAa;IAC5C,OAAO,CAAC,IAAI,CAAqB;IACjC,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,UAAU,CAAuB;IAEzC,OAAO,CAAC,uBAAuB;gBAsBnB,EACV,IAAI,EACJ,MAAM,EACN,UAAU,GACX,EAAE;QACD,IAAI,EAAE,GAAG,CAAC,cAAc,CAAC;QACzB,MAAM,EAAE,MAAM,CAAC;QACf,UAAU,EAAE,oBAAoB,CAAC;KAClC;IAOK,aAAa,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,iBAAiB,GAAG,IAAI,CAAC;IAuC7E,uBAAuB,CAClC,IAAI,EAAE,mCAAmC,GACxC,OAAO,CAAC,oCAAoC,CAAC;IA4FnC,UAAU,CAAC,EAAE,MAAM,EAAE,EAAE;QAAE,MAAM,EAAE,iBAAiB,CAAA;KAAE,GAAG,OAAO,CAAC,iBAAiB,CAAC;IA6C9F;;OAEG;IACG,YAAY,CAAC,EACjB,EAAE,EACF,KAAK,EACL,QAAQ,GACT,EAAE;QACD,EAAE,EAAE,MAAM,CAAC;QACX,KAAK,EAAE,MAAM,CAAC;QACd,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACnC,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAyExB,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;YA6BvD,oBAAoB;IA2FrB,gBAAgB,CAAC,EAAE,UAAU,EAAE,EAAE;QAAE,UAAU,EAAE,MAAM,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,CAAC;IAkDpG,YAAY,CAAC,IAAI,EAAE,wBAAwB,GAAG,OAAO,CAAC,yBAAyB,CAAC;IAyKvF,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,CAAC;IAiGrG,cAAc,CAAC,EACnB,QAAQ,GACT,EAAE;QACD,QAAQ,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,eAAe,EAAE,WAAW,CAAC,CAAC,GAAG;YACvD,EAAE,EAAE,MAAM,CAAC;YACX,OAAO,CAAC,EAAE;gBACR,QAAQ,CAAC,EAAE,sBAAsB,CAAC,UAAU,CAAC,CAAC;gBAC9C,OAAO,CAAC,EAAE,sBAAsB,CAAC,SAAS,CAAC,CAAC;aAC7C,CAAC;SACH,CAAC,EAAE,CAAC;KACN,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IA4GxB,cAAc,CAAC,UAAU,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAsEnD,eAAe,CAAC,EAAE,UAAU,EAAE,EAAE;QAAE,UAAU,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,mBAAmB,GAAG,IAAI,CAAC;IAkC5F,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,mBAAmB,CAAA;KAAE,GAAG,OAAO,CAAC,mBAAmB,CAAC;IAY3F,cAAc,CAAC,EACnB,UAAU,EACV,aAAa,EACb,QAAQ,GACT,EAAE;QACD,UAAU,EAAE,MAAM,CAAC;QACnB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACpC,GAAG,OAAO,CAAC,mBAAmB,CAAC;CA8DjC"}
@@ -0,0 +1,44 @@
1
+ import type { TracingStorageStrategy } from '@mastra/core/observability';
2
+ import { ObservabilityStorage } from '@mastra/core/storage';
3
+ import type { SpanRecord, TraceRecord, TracesPaginatedArg, CreateSpanRecord, PaginationInfo, UpdateSpanRecord } from '@mastra/core/storage';
4
+ import type { ConnectionPool } from 'mssql';
5
+ import type { StoreOperationsMSSQL } from '../operations/index.js';
6
+ export declare class ObservabilityMSSQL extends ObservabilityStorage {
7
+ pool: ConnectionPool;
8
+ private operations;
9
+ private schema?;
10
+ constructor({ pool, operations, schema, }: {
11
+ pool: ConnectionPool;
12
+ operations: StoreOperationsMSSQL;
13
+ schema?: string;
14
+ });
15
+ get tracingStrategy(): {
16
+ preferred: TracingStorageStrategy;
17
+ supported: TracingStorageStrategy[];
18
+ };
19
+ createSpan(span: CreateSpanRecord): Promise<void>;
20
+ getTrace(traceId: string): Promise<TraceRecord | null>;
21
+ updateSpan({ spanId, traceId, updates, }: {
22
+ spanId: string;
23
+ traceId: string;
24
+ updates: Partial<UpdateSpanRecord>;
25
+ }): Promise<void>;
26
+ getTracesPaginated({ filters, pagination, }: TracesPaginatedArg): Promise<{
27
+ pagination: PaginationInfo;
28
+ spans: SpanRecord[];
29
+ }>;
30
+ batchCreateSpans(args: {
31
+ records: CreateSpanRecord[];
32
+ }): Promise<void>;
33
+ batchUpdateSpans(args: {
34
+ records: {
35
+ traceId: string;
36
+ spanId: string;
37
+ updates: Partial<UpdateSpanRecord>;
38
+ }[];
39
+ }): Promise<void>;
40
+ batchDeleteTraces(args: {
41
+ traceIds: string[];
42
+ }): Promise<void>;
43
+ }
44
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/observability/index.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,4BAA4B,CAAC;AACzE,OAAO,EAAe,oBAAoB,EAAe,MAAM,sBAAsB,CAAC;AACtF,OAAO,KAAK,EACV,UAAU,EACV,WAAW,EACX,kBAAkB,EAClB,gBAAgB,EAChB,cAAc,EACd,gBAAgB,EACjB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,OAAO,CAAC;AAC5C,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC;AAG1D,qBAAa,kBAAmB,SAAQ,oBAAoB;IACnD,IAAI,EAAE,cAAc,CAAC;IAC5B,OAAO,CAAC,UAAU,CAAuB;IACzC,OAAO,CAAC,MAAM,CAAC,CAAS;gBAEZ,EACV,IAAI,EACJ,UAAU,EACV,MAAM,GACP,EAAE;QACD,IAAI,EAAE,cAAc,CAAC;QACrB,UAAU,EAAE,oBAAoB,CAAC;QACjC,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB;IAOD,IAAW,eAAe,IAAI;QAC5B,SAAS,EAAE,sBAAsB,CAAC;QAClC,SAAS,EAAE,sBAAsB,EAAE,CAAC;KACrC,CAKA;IAEK,UAAU,CAAC,IAAI,EAAE,gBAAgB,GAAG,OAAO,CAAC,IAAI,CAAC;IA+BjD,QAAQ,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC;IAgDtD,UAAU,CAAC,EACf,MAAM,EACN,OAAO,EACP,OAAO,GACR,EAAE;QACD,MAAM,EAAE,MAAM,CAAC;QACf,OAAO,EAAE,MAAM,CAAC;QAChB,OAAO,EAAE,OAAO,CAAC,gBAAgB,CAAC,CAAC;KACpC,GAAG,OAAO,CAAC,IAAI,CAAC;IAgCX,kBAAkB,CAAC,EACvB,OAAO,EACP,UAAU,GACX,EAAE,kBAAkB,GAAG,OAAO,CAAC;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,KAAK,EAAE,UAAU,EAAE,CAAA;KAAE,CAAC;IAoH9E,gBAAgB,CAAC,IAAI,EAAE;QAAE,OAAO,EAAE,gBAAgB,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IA6BtE,gBAAgB,CAAC,IAAI,EAAE;QAC3B,OAAO,EAAE;YACP,OAAO,EAAE,MAAM,CAAC;YAChB,MAAM,EAAE,MAAM,CAAC;YACf,OAAO,EAAE,OAAO,CAAC,gBAAgB,CAAC,CAAC;SACpC,EAAE,CAAC;KACL,GAAG,OAAO,CAAC,IAAI,CAAC;IAwCX,iBAAiB,CAAC,IAAI,EAAE;QAAE,QAAQ,EAAE,MAAM,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;CA0BrE"}
@@ -0,0 +1,114 @@
1
+ import { StoreOperations } from '@mastra/core/storage';
2
+ import type { StorageColumn, TABLE_NAMES, CreateIndexOptions, IndexInfo, StorageIndexStats } from '@mastra/core/storage';
3
+ import sql from 'mssql';
4
+ export type { CreateIndexOptions, IndexInfo, StorageIndexStats };
5
+ export declare class StoreOperationsMSSQL extends StoreOperations {
6
+ pool: sql.ConnectionPool;
7
+ schemaName?: string;
8
+ private setupSchemaPromise;
9
+ private schemaSetupComplete;
10
+ protected getSqlType(type: StorageColumn['type'], isPrimaryKey?: boolean, useLargeStorage?: boolean): string;
11
+ constructor({ pool, schemaName }: {
12
+ pool: sql.ConnectionPool;
13
+ schemaName?: string;
14
+ });
15
+ hasColumn(table: string, column: string): Promise<boolean>;
16
+ private setupSchema;
17
+ insert({ tableName, record, transaction, }: {
18
+ tableName: TABLE_NAMES;
19
+ record: Record<string, any>;
20
+ transaction?: sql.Transaction;
21
+ }): Promise<void>;
22
+ clearTable({ tableName }: {
23
+ tableName: TABLE_NAMES;
24
+ }): Promise<void>;
25
+ protected getDefaultValue(type: StorageColumn['type']): string;
26
+ createTable({ tableName, schema, }: {
27
+ tableName: TABLE_NAMES;
28
+ schema: Record<string, StorageColumn>;
29
+ }): Promise<void>;
30
+ /**
31
+ * Alters table schema to add columns if they don't exist
32
+ * @param tableName Name of the table
33
+ * @param schema Schema of the table
34
+ * @param ifNotExists Array of column names to add if they don't exist
35
+ */
36
+ alterTable({ tableName, schema, ifNotExists, }: {
37
+ tableName: TABLE_NAMES;
38
+ schema: Record<string, StorageColumn>;
39
+ ifNotExists: string[];
40
+ }): Promise<void>;
41
+ load<R>({ tableName, keys }: {
42
+ tableName: TABLE_NAMES;
43
+ keys: Record<string, any>;
44
+ }): Promise<R | null>;
45
+ batchInsert({ tableName, records }: {
46
+ tableName: TABLE_NAMES;
47
+ records: Record<string, any>[];
48
+ }): Promise<void>;
49
+ dropTable({ tableName }: {
50
+ tableName: TABLE_NAMES;
51
+ }): Promise<void>;
52
+ /**
53
+ * Prepares a value for database operations, handling Date objects and JSON serialization
54
+ */
55
+ private prepareValue;
56
+ /**
57
+ * Maps TABLE_SCHEMAS types to mssql param types (used when value is null)
58
+ */
59
+ private getMssqlType;
60
+ /**
61
+ * Update a single record in the database
62
+ */
63
+ update({ tableName, keys, data, transaction, }: {
64
+ tableName: TABLE_NAMES;
65
+ keys: Record<string, any>;
66
+ data: Record<string, any>;
67
+ transaction?: sql.Transaction;
68
+ }): Promise<void>;
69
+ /**
70
+ * Update multiple records in a single batch transaction
71
+ */
72
+ batchUpdate({ tableName, updates, }: {
73
+ tableName: TABLE_NAMES;
74
+ updates: Array<{
75
+ keys: Record<string, any>;
76
+ data: Record<string, any>;
77
+ }>;
78
+ }): Promise<void>;
79
+ /**
80
+ * Delete multiple records by keys
81
+ */
82
+ batchDelete({ tableName, keys }: {
83
+ tableName: TABLE_NAMES;
84
+ keys: Record<string, any>[];
85
+ }): Promise<void>;
86
+ /**
87
+ * Create a new index on a table
88
+ */
89
+ createIndex(options: CreateIndexOptions): Promise<void>;
90
+ /**
91
+ * Drop an existing index
92
+ */
93
+ dropIndex(indexName: string): Promise<void>;
94
+ /**
95
+ * List indexes for a specific table or all tables
96
+ */
97
+ listIndexes(tableName?: string): Promise<IndexInfo[]>;
98
+ /**
99
+ * Get detailed statistics for a specific index
100
+ */
101
+ describeIndex(indexName: string): Promise<StorageIndexStats>;
102
+ /**
103
+ * Returns definitions for automatic performance indexes
104
+ * IMPORTANT: Uses seq_id DESC instead of createdAt DESC for MSSQL due to millisecond accuracy limitations
105
+ * NOTE: Using NVARCHAR(400) for text columns (800 bytes) leaves room for composite indexes
106
+ */
107
+ protected getAutomaticIndexDefinitions(): CreateIndexOptions[];
108
+ /**
109
+ * Creates automatic indexes for optimal query performance
110
+ * Uses getAutomaticIndexDefinitions() to determine which indexes to create
111
+ */
112
+ createAutomaticIndexes(): Promise<void>;
113
+ }
114
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/operations/index.ts"],"names":[],"mappings":"AACA,OAAO,EACL,eAAe,EAQhB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EACV,aAAa,EACb,WAAW,EACX,kBAAkB,EAClB,SAAS,EACT,iBAAiB,EAClB,MAAM,sBAAsB,CAAC;AAE9B,OAAO,GAAG,MAAM,OAAO,CAAC;AAIxB,YAAY,EAAE,kBAAkB,EAAE,SAAS,EAAE,iBAAiB,EAAE,CAAC;AAEjE,qBAAa,oBAAqB,SAAQ,eAAe;IAChD,IAAI,EAAE,GAAG,CAAC,cAAc,CAAC;IACzB,UAAU,CAAC,EAAE,MAAM,CAAC;IAC3B,OAAO,CAAC,kBAAkB,CAA8B;IACxD,OAAO,CAAC,mBAAmB,CAAkC;IAE7D,SAAS,CAAC,UAAU,CAAC,IAAI,EAAE,aAAa,CAAC,MAAM,CAAC,EAAE,YAAY,UAAQ,EAAE,eAAe,UAAQ,GAAG,MAAM;gBAmC5F,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE;QAAE,IAAI,EAAE,GAAG,CAAC,cAAc,CAAC;QAAC,UAAU,CAAC,EAAE,MAAM,CAAA;KAAE;IAM7E,SAAS,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;YAalD,WAAW;IA2CnB,MAAM,CAAC,EACX,SAAS,EACT,MAAM,EACN,WAAW,GACZ,EAAE;QACD,SAAS,EAAE,WAAW,CAAC;QACvB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAC5B,WAAW,CAAC,EAAE,GAAG,CAAC,WAAW,CAAC;KAC/B,GAAG,OAAO,CAAC,IAAI,CAAC;IAqCX,UAAU,CAAC,EAAE,SAAS,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IA6B1E,SAAS,CAAC,eAAe,CAAC,IAAI,EAAE,aAAa,CAAC,MAAM,CAAC,GAAG,MAAM;IAaxD,WAAW,CAAC,EAChB,SAAS,EACT,MAAM,GACP,EAAE;QACD,SAAS,EAAE,WAAW,CAAC;QACvB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;KACvC,GAAG,OAAO,CAAC,IAAI,CAAC;IAoGjB;;;;;OAKG;IACG,UAAU,CAAC,EACf,SAAS,EACT,MAAM,EACN,WAAW,GACZ,EAAE;QACD,SAAS,EAAE,WAAW,CAAC;QACvB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;QACtC,WAAW,EAAE,MAAM,EAAE,CAAC;KACvB,GAAG,OAAO,CAAC,IAAI,CAAC;IAqDX,IAAI,CAAC,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAA;KAAE,GAAG,OAAO,CAAC,CAAC,GAAG,IAAI,CAAC;IA0CtG,WAAW,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAC;QAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAyB9G,SAAS,CAAC,EAAE,SAAS,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAmBzE;;OAEG;IACH,OAAO,CAAC,YAAY;IA4CpB;;OAEG;IACH,OAAO,CAAC,YAAY;IAwBpB;;OAEG;IACG,MAAM,CAAC,EACX,SAAS,EACT,IAAI,EACJ,IAAI,EACJ,WAAW,GACZ,EAAE;QACD,SAAS,EAAE,WAAW,CAAC;QACvB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAC1B,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAC1B,WAAW,CAAC,EAAE,GAAG,CAAC,WAAW,CAAC;KAC/B,GAAG,OAAO,CAAC,IAAI,CAAC;IA0EjB;;OAEG;IACG,WAAW,CAAC,EAChB,SAAS,EACT,OAAO,GACR,EAAE;QACD,SAAS,EAAE,WAAW,CAAC;QACvB,OAAO,EAAE,KAAK,CAAC;YACb,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;YAC1B,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;SAC3B,CAAC,CAAC;KACJ,GAAG,OAAO,CAAC,IAAI,CAAC;IA2BjB;;OAEG;IACG,WAAW,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAqD9G;;OAEG;IACG,WAAW,CAAC,OAAO,EAAE,kBAAkB,GAAG,OAAO,CAAC,IAAI,CAAC;IAqE7D;;OAEG;IACG,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IA2DjD;;OAEG;IACG,WAAW,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IA2F3D;;OAEG;IACG,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAgFlE;;;;OAIG;IACH,SAAS,CAAC,4BAA4B,IAAI,kBAAkB,EAAE;IAiD9D;;;OAGG;IACG,sBAAsB,IAAI,OAAO,CAAC,IAAI,CAAC;CAuB9C"}
@@ -0,0 +1,55 @@
1
+ import type { ScoreRowData, ScoringSource } from '@mastra/core/evals';
2
+ import type { PaginationInfo, StoragePagination } from '@mastra/core/storage';
3
+ import { ScoresStorage } from '@mastra/core/storage';
4
+ import type { ConnectionPool } from 'mssql';
5
+ import type { StoreOperationsMSSQL } from '../operations/index.js';
6
+ export declare class ScoresMSSQL extends ScoresStorage {
7
+ pool: ConnectionPool;
8
+ private operations;
9
+ private schema?;
10
+ constructor({ pool, operations, schema, }: {
11
+ pool: ConnectionPool;
12
+ operations: StoreOperationsMSSQL;
13
+ schema?: string;
14
+ });
15
+ getScoreById({ id }: {
16
+ id: string;
17
+ }): Promise<ScoreRowData | null>;
18
+ saveScore(score: Omit<ScoreRowData, 'id' | 'createdAt' | 'updatedAt'>): Promise<{
19
+ score: ScoreRowData;
20
+ }>;
21
+ listScoresByScorerId({ scorerId, pagination, entityId, entityType, source, }: {
22
+ scorerId: string;
23
+ pagination: StoragePagination;
24
+ entityId?: string;
25
+ entityType?: string;
26
+ source?: ScoringSource;
27
+ }): Promise<{
28
+ pagination: PaginationInfo;
29
+ scores: ScoreRowData[];
30
+ }>;
31
+ listScoresByRunId({ runId, pagination, }: {
32
+ runId: string;
33
+ pagination: StoragePagination;
34
+ }): Promise<{
35
+ pagination: PaginationInfo;
36
+ scores: ScoreRowData[];
37
+ }>;
38
+ listScoresByEntityId({ entityId, entityType, pagination, }: {
39
+ pagination: StoragePagination;
40
+ entityId: string;
41
+ entityType: string;
42
+ }): Promise<{
43
+ pagination: PaginationInfo;
44
+ scores: ScoreRowData[];
45
+ }>;
46
+ listScoresBySpan({ traceId, spanId, pagination, }: {
47
+ traceId: string;
48
+ spanId: string;
49
+ pagination: StoragePagination;
50
+ }): Promise<{
51
+ pagination: PaginationInfo;
52
+ scores: ScoreRowData[];
53
+ }>;
54
+ }
55
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/scores/index.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,YAAY,EAAE,aAAa,EAA6B,MAAM,oBAAoB,CAAC;AAEjG,OAAO,KAAK,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAC9E,OAAO,EACL,aAAa,EAKd,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,OAAO,CAAC;AAC5C,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC;AAoB1D,qBAAa,WAAY,SAAQ,aAAa;IACrC,IAAI,EAAE,cAAc,CAAC;IAC5B,OAAO,CAAC,UAAU,CAAuB;IACzC,OAAO,CAAC,MAAM,CAAC,CAAS;gBAEZ,EACV,IAAI,EACJ,UAAU,EACV,MAAM,GACP,EAAE;QACD,IAAI,EAAE,cAAc,CAAC;QACrB,UAAU,EAAE,oBAAoB,CAAC;QACjC,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB;IAOK,YAAY,CAAC,EAAE,EAAE,EAAE,EAAE;QAAE,EAAE,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;IA0BlE,SAAS,CAAC,KAAK,EAAE,IAAI,CAAC,YAAY,EAAE,IAAI,GAAG,WAAW,GAAG,WAAW,CAAC,GAAG,OAAO,CAAC;QAAE,KAAK,EAAE,YAAY,CAAA;KAAE,CAAC;IAiExG,oBAAoB,CAAC,EACzB,QAAQ,EACR,UAAU,EACV,QAAQ,EACR,UAAU,EACV,MAAM,GACP,EAAE;QACD,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,iBAAiB,CAAC;QAC9B,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,MAAM,CAAC,EAAE,aAAa,CAAC;KACxB,GAAG,OAAO,CAAC;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,MAAM,EAAE,YAAY,EAAE,CAAA;KAAE,CAAC;IAwF7D,iBAAiB,CAAC,EACtB,KAAK,EACL,UAAU,GACX,EAAE;QACD,KAAK,EAAE,MAAM,CAAC;QACd,UAAU,EAAE,iBAAiB,CAAC;KAC/B,GAAG,OAAO,CAAC;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,MAAM,EAAE,YAAY,EAAE,CAAA;KAAE,CAAC;IA4D7D,oBAAoB,CAAC,EACzB,QAAQ,EACR,UAAU,EACV,UAAU,GACX,EAAE;QACD,UAAU,EAAE,iBAAiB,CAAC;QAC9B,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,MAAM,CAAC;KACpB,GAAG,OAAO,CAAC;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,MAAM,EAAE,YAAY,EAAE,CAAA;KAAE,CAAC;IA6D7D,gBAAgB,CAAC,EACrB,OAAO,EACP,MAAM,EACN,UAAU,GACX,EAAE;QACD,OAAO,EAAE,MAAM,CAAC;QAChB,MAAM,EAAE,MAAM,CAAC;QACf,UAAU,EAAE,iBAAiB,CAAC;KAC/B,GAAG,OAAO,CAAC;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,MAAM,EAAE,YAAY,EAAE,CAAA;KAAE,CAAC;CA6DpE"}
@@ -0,0 +1,25 @@
1
+ import type { PaginationArgs, StorageColumn, TABLE_NAMES } from '@mastra/core/storage';
2
+ export declare function getSchemaName(schema?: string): string | undefined;
3
+ export declare function getTableName({ indexName, schemaName }: {
4
+ indexName: string;
5
+ schemaName?: string;
6
+ }): string;
7
+ /**
8
+ * Build date range filter for queries
9
+ */
10
+ export declare function buildDateRangeFilter(dateRange: PaginationArgs['dateRange'], fieldName: string): Record<string, any>;
11
+ /**
12
+ * Prepare WHERE clause for MSSQL queries with @param substitution
13
+ */
14
+ export declare function prepareWhereClause(filters: Record<string, any>, _schema?: Record<string, StorageColumn>): {
15
+ sql: string;
16
+ params: Record<string, any>;
17
+ };
18
+ /**
19
+ * Transform SQL row to record format, handling JSON columns
20
+ */
21
+ export declare function transformFromSqlRow<T>({ tableName, sqlRow, }: {
22
+ tableName: TABLE_NAMES;
23
+ sqlRow: Record<string, any>;
24
+ }): T;
25
+ //# sourceMappingURL=utils.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/storage/domains/utils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAIvF,wBAAgB,aAAa,CAAC,MAAM,CAAC,EAAE,MAAM,sBAE5C;AAED,wBAAgB,YAAY,CAAC,EAAE,SAAS,EAAE,UAAU,EAAE,EAAE;IAAE,SAAS,EAAE,MAAM,CAAC;IAAC,UAAU,CAAC,EAAE,MAAM,CAAA;CAAE,UAKjG;AAED;;GAEG;AACH,wBAAgB,oBAAoB,CAAC,SAAS,EAAE,cAAc,CAAC,WAAW,CAAC,EAAE,SAAS,EAAE,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CASnH;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAChC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAC5B,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,aAAa,CAAC,GACtC;IAAE,GAAG,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAA;CAAE,CA+B9C;AAED;;GAEG;AACH,wBAAgB,mBAAmB,CAAC,CAAC,EAAE,EACrC,SAAS,EACT,MAAM,GACP,EAAE;IACD,SAAS,EAAE,WAAW,CAAC;IACvB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAC7B,GAAG,CAAC,CA8BJ"}
@@ -0,0 +1,50 @@
1
+ import { WorkflowsStorage } from '@mastra/core/storage';
2
+ import type { StorageListWorkflowRunsInput, WorkflowRun, WorkflowRuns } from '@mastra/core/storage';
3
+ import type { StepResult, WorkflowRunState } from '@mastra/core/workflows';
4
+ import sql from 'mssql';
5
+ import type { StoreOperationsMSSQL } from '../operations/index.js';
6
+ export declare class WorkflowsMSSQL extends WorkflowsStorage {
7
+ pool: sql.ConnectionPool;
8
+ private operations;
9
+ private schema;
10
+ constructor({ pool, operations, schema, }: {
11
+ pool: sql.ConnectionPool;
12
+ operations: StoreOperationsMSSQL;
13
+ schema: string;
14
+ });
15
+ private parseWorkflowRun;
16
+ updateWorkflowResults({ workflowName, runId, stepId, result, requestContext, }: {
17
+ workflowName: string;
18
+ runId: string;
19
+ stepId: string;
20
+ result: StepResult<any, any, any, any>;
21
+ requestContext: Record<string, any>;
22
+ }): Promise<Record<string, StepResult<any, any, any, any>>>;
23
+ updateWorkflowState({ workflowName, runId, opts, }: {
24
+ workflowName: string;
25
+ runId: string;
26
+ opts: {
27
+ status: string;
28
+ result?: StepResult<any, any, any, any>;
29
+ error?: string;
30
+ suspendedPaths?: Record<string, number[]>;
31
+ waitingPaths?: Record<string, number[]>;
32
+ };
33
+ }): Promise<WorkflowRunState | undefined>;
34
+ persistWorkflowSnapshot({ workflowName, runId, resourceId, snapshot, }: {
35
+ workflowName: string;
36
+ runId: string;
37
+ resourceId?: string;
38
+ snapshot: WorkflowRunState;
39
+ }): Promise<void>;
40
+ loadWorkflowSnapshot({ workflowName, runId, }: {
41
+ workflowName: string;
42
+ runId: string;
43
+ }): Promise<WorkflowRunState | null>;
44
+ getWorkflowRunById({ runId, workflowName, }: {
45
+ runId: string;
46
+ workflowName?: string;
47
+ }): Promise<WorkflowRun | null>;
48
+ listWorkflowRuns({ workflowName, fromDate, toDate, page, perPage, resourceId, status, }?: StorageListWorkflowRunsInput): Promise<WorkflowRuns>;
49
+ }
50
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/workflows/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,gBAAgB,EAA6C,MAAM,sBAAsB,CAAC;AACnG,OAAO,KAAK,EAAE,4BAA4B,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AACpG,OAAO,KAAK,EAAE,UAAU,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAC3E,OAAO,GAAG,MAAM,OAAO,CAAC;AACxB,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC;AAG1D,qBAAa,cAAe,SAAQ,gBAAgB;IAC3C,IAAI,EAAE,GAAG,CAAC,cAAc,CAAC;IAChC,OAAO,CAAC,UAAU,CAAuB;IACzC,OAAO,CAAC,MAAM,CAAS;gBAEX,EACV,IAAI,EACJ,UAAU,EACV,MAAM,GACP,EAAE;QACD,IAAI,EAAE,GAAG,CAAC,cAAc,CAAC;QACzB,UAAU,EAAE,oBAAoB,CAAC;QACjC,MAAM,EAAE,MAAM,CAAC;KAChB;IAOD,OAAO,CAAC,gBAAgB;IAmBlB,qBAAqB,CAAC,EAC1B,YAAY,EACZ,KAAK,EACL,MAAM,EACN,MAAM,EACN,cAAc,GACf,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,EAAE,MAAM,CAAC;QACf,MAAM,EAAE,UAAU,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QACvC,cAAc,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KACrC,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;IAoFrD,mBAAmB,CAAC,EACxB,YAAY,EACZ,KAAK,EACL,IAAI,GACL,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;QACd,IAAI,EAAE;YACJ,MAAM,EAAE,MAAM,CAAC;YACf,MAAM,CAAC,EAAE,UAAU,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YACxC,KAAK,CAAC,EAAE,MAAM,CAAC;YACf,cAAc,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;YAC1C,YAAY,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;SACzC,CAAC;KACH,GAAG,OAAO,CAAC,gBAAgB,GAAG,SAAS,CAAC;IA8EnC,uBAAuB,CAAC,EAC5B,YAAY,EACZ,KAAK,EACL,UAAU,EACV,QAAQ,GACT,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;QACd,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,QAAQ,EAAE,gBAAgB,CAAC;KAC5B,GAAG,OAAO,CAAC,IAAI,CAAC;IAqCX,oBAAoB,CAAC,EACzB,YAAY,EACZ,KAAK,GACN,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;KACf,GAAG,OAAO,CAAC,gBAAgB,GAAG,IAAI,CAAC;IA6B9B,kBAAkB,CAAC,EACvB,KAAK,EACL,YAAY,GACb,EAAE;QACD,KAAK,EAAE,MAAM,CAAC;QACd,YAAY,CAAC,EAAE,MAAM,CAAC;KACvB,GAAG,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC;IA2CzB,gBAAgB,CAAC,EACrB,YAAY,EACZ,QAAQ,EACR,MAAM,EACN,IAAI,EACJ,OAAO,EACP,UAAU,EACV,MAAM,GACP,GAAE,4BAAiC,GAAG,OAAO,CAAC,YAAY,CAAC;CA+E7D"}
@@ -0,0 +1,233 @@
1
+ import type { MastraMessageContentV2, MastraDBMessage } from '@mastra/core/agent';
2
+ import type { ScoreRowData, ScoringSource } from '@mastra/core/evals';
3
+ import type { StorageThreadType } from '@mastra/core/memory';
4
+ import { MastraStorage } from '@mastra/core/storage';
5
+ export type MastraDBMessageWithTypedContent = Omit<MastraDBMessage, 'content'> & {
6
+ content: MastraMessageContentV2;
7
+ };
8
+ import type { PaginationInfo, StorageColumn, StorageResourceType, TABLE_NAMES, WorkflowRun, WorkflowRuns, StoragePagination, StorageDomains, SpanRecord, TraceRecord, TracesPaginatedArg, UpdateSpanRecord, CreateIndexOptions, IndexInfo, StorageIndexStats, StorageListWorkflowRunsInput } from '@mastra/core/storage';
9
+ import type { StepResult, WorkflowRunState } from '@mastra/core/workflows';
10
+ import sql from 'mssql';
11
+ export type MSSQLConfigType = {
12
+ id: string;
13
+ schemaName?: string;
14
+ } & ({
15
+ server: string;
16
+ port: number;
17
+ database: string;
18
+ user: string;
19
+ password: string;
20
+ options?: sql.IOptions;
21
+ } | {
22
+ connectionString: string;
23
+ });
24
+ export type MSSQLConfig = MSSQLConfigType;
25
+ export declare class MSSQLStore extends MastraStorage {
26
+ pool: sql.ConnectionPool;
27
+ private schema?;
28
+ private isConnected;
29
+ stores: StorageDomains;
30
+ constructor(config: MSSQLConfigType);
31
+ init(): Promise<void>;
32
+ private _performInitializationAndStore;
33
+ get supports(): {
34
+ selectByIncludeResourceScope: boolean;
35
+ resourceWorkingMemory: boolean;
36
+ hasColumn: boolean;
37
+ createTable: boolean;
38
+ deleteMessages: boolean;
39
+ listScoresBySpan: boolean;
40
+ observabilityInstance: boolean;
41
+ indexManagement: boolean;
42
+ };
43
+ createTable({ tableName, schema, }: {
44
+ tableName: TABLE_NAMES;
45
+ schema: Record<string, StorageColumn>;
46
+ }): Promise<void>;
47
+ alterTable({ tableName, schema, ifNotExists, }: {
48
+ tableName: TABLE_NAMES;
49
+ schema: Record<string, StorageColumn>;
50
+ ifNotExists: string[];
51
+ }): Promise<void>;
52
+ clearTable({ tableName }: {
53
+ tableName: TABLE_NAMES;
54
+ }): Promise<void>;
55
+ dropTable({ tableName }: {
56
+ tableName: TABLE_NAMES;
57
+ }): Promise<void>;
58
+ insert({ tableName, record }: {
59
+ tableName: TABLE_NAMES;
60
+ record: Record<string, any>;
61
+ }): Promise<void>;
62
+ batchInsert({ tableName, records }: {
63
+ tableName: TABLE_NAMES;
64
+ records: Record<string, any>[];
65
+ }): Promise<void>;
66
+ load<R>({ tableName, keys }: {
67
+ tableName: TABLE_NAMES;
68
+ keys: Record<string, string>;
69
+ }): Promise<R | null>;
70
+ /**
71
+ * Memory
72
+ */
73
+ getThreadById({ threadId }: {
74
+ threadId: string;
75
+ }): Promise<StorageThreadType | null>;
76
+ saveThread({ thread }: {
77
+ thread: StorageThreadType;
78
+ }): Promise<StorageThreadType>;
79
+ updateThread({ id, title, metadata, }: {
80
+ id: string;
81
+ title: string;
82
+ metadata: Record<string, unknown>;
83
+ }): Promise<StorageThreadType>;
84
+ deleteThread({ threadId }: {
85
+ threadId: string;
86
+ }): Promise<void>;
87
+ listMessagesById({ messageIds }: {
88
+ messageIds: string[];
89
+ }): Promise<{
90
+ messages: MastraDBMessage[];
91
+ }>;
92
+ saveMessages(args: {
93
+ messages: MastraDBMessage[];
94
+ }): Promise<{
95
+ messages: MastraDBMessage[];
96
+ }>;
97
+ updateMessages({ messages, }: {
98
+ messages: (Partial<Omit<MastraDBMessage, 'createdAt'>> & {
99
+ id: string;
100
+ content?: {
101
+ metadata?: MastraMessageContentV2['metadata'];
102
+ content?: MastraMessageContentV2['content'];
103
+ };
104
+ })[];
105
+ }): Promise<MastraDBMessage[]>;
106
+ deleteMessages(messageIds: string[]): Promise<void>;
107
+ getResourceById({ resourceId }: {
108
+ resourceId: string;
109
+ }): Promise<StorageResourceType | null>;
110
+ saveResource({ resource }: {
111
+ resource: StorageResourceType;
112
+ }): Promise<StorageResourceType>;
113
+ updateResource({ resourceId, workingMemory, metadata, }: {
114
+ resourceId: string;
115
+ workingMemory?: string;
116
+ metadata?: Record<string, unknown>;
117
+ }): Promise<StorageResourceType>;
118
+ /**
119
+ * Workflows
120
+ */
121
+ updateWorkflowResults({ workflowName, runId, stepId, result, requestContext, }: {
122
+ workflowName: string;
123
+ runId: string;
124
+ stepId: string;
125
+ result: StepResult<any, any, any, any>;
126
+ requestContext: Record<string, any>;
127
+ }): Promise<Record<string, StepResult<any, any, any, any>>>;
128
+ updateWorkflowState({ workflowName, runId, opts, }: {
129
+ workflowName: string;
130
+ runId: string;
131
+ opts: {
132
+ status: string;
133
+ result?: StepResult<any, any, any, any>;
134
+ error?: string;
135
+ suspendedPaths?: Record<string, number[]>;
136
+ waitingPaths?: Record<string, number[]>;
137
+ };
138
+ }): Promise<WorkflowRunState | undefined>;
139
+ persistWorkflowSnapshot({ workflowName, runId, resourceId, snapshot, }: {
140
+ workflowName: string;
141
+ runId: string;
142
+ resourceId?: string;
143
+ snapshot: WorkflowRunState;
144
+ }): Promise<void>;
145
+ loadWorkflowSnapshot({ workflowName, runId, }: {
146
+ workflowName: string;
147
+ runId: string;
148
+ }): Promise<WorkflowRunState | null>;
149
+ listWorkflowRuns(args?: StorageListWorkflowRunsInput): Promise<WorkflowRuns>;
150
+ getWorkflowRunById({ runId, workflowName, }: {
151
+ runId: string;
152
+ workflowName?: string;
153
+ }): Promise<WorkflowRun | null>;
154
+ close(): Promise<void>;
155
+ /**
156
+ * Index Management
157
+ */
158
+ createIndex(options: CreateIndexOptions): Promise<void>;
159
+ listIndexes(tableName?: string): Promise<IndexInfo[]>;
160
+ describeIndex(indexName: string): Promise<StorageIndexStats>;
161
+ dropIndex(indexName: string): Promise<void>;
162
+ /**
163
+ * Tracing / Observability
164
+ */
165
+ private getObservabilityStore;
166
+ createSpan(span: SpanRecord): Promise<void>;
167
+ updateSpan({ spanId, traceId, updates, }: {
168
+ spanId: string;
169
+ traceId: string;
170
+ updates: Partial<UpdateSpanRecord>;
171
+ }): Promise<void>;
172
+ getTrace(traceId: string): Promise<TraceRecord | null>;
173
+ getTracesPaginated(args: TracesPaginatedArg): Promise<{
174
+ pagination: PaginationInfo;
175
+ spans: SpanRecord[];
176
+ }>;
177
+ batchCreateSpans(args: {
178
+ records: SpanRecord[];
179
+ }): Promise<void>;
180
+ batchUpdateSpans(args: {
181
+ records: {
182
+ traceId: string;
183
+ spanId: string;
184
+ updates: Partial<UpdateSpanRecord>;
185
+ }[];
186
+ }): Promise<void>;
187
+ batchDeleteTraces(args: {
188
+ traceIds: string[];
189
+ }): Promise<void>;
190
+ /**
191
+ * Scorers
192
+ */
193
+ getScoreById({ id: _id }: {
194
+ id: string;
195
+ }): Promise<ScoreRowData | null>;
196
+ listScoresByScorerId({ scorerId: _scorerId, pagination: _pagination, entityId: _entityId, entityType: _entityType, source: _source, }: {
197
+ scorerId: string;
198
+ pagination: StoragePagination;
199
+ entityId?: string;
200
+ entityType?: string;
201
+ source?: ScoringSource;
202
+ }): Promise<{
203
+ pagination: PaginationInfo;
204
+ scores: ScoreRowData[];
205
+ }>;
206
+ saveScore(_score: ScoreRowData): Promise<{
207
+ score: ScoreRowData;
208
+ }>;
209
+ listScoresByRunId({ runId: _runId, pagination: _pagination, }: {
210
+ runId: string;
211
+ pagination: StoragePagination;
212
+ }): Promise<{
213
+ pagination: PaginationInfo;
214
+ scores: ScoreRowData[];
215
+ }>;
216
+ listScoresByEntityId({ entityId: _entityId, entityType: _entityType, pagination: _pagination, }: {
217
+ pagination: StoragePagination;
218
+ entityId: string;
219
+ entityType: string;
220
+ }): Promise<{
221
+ pagination: PaginationInfo;
222
+ scores: ScoreRowData[];
223
+ }>;
224
+ listScoresBySpan({ traceId, spanId, pagination: _pagination, }: {
225
+ traceId: string;
226
+ spanId: string;
227
+ pagination: StoragePagination;
228
+ }): Promise<{
229
+ pagination: PaginationInfo;
230
+ scores: ScoreRowData[];
231
+ }>;
232
+ }
233
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/storage/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,sBAAsB,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAElF,OAAO,KAAK,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACtE,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAC7D,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAErD,MAAM,MAAM,+BAA+B,GAAG,IAAI,CAAC,eAAe,EAAE,SAAS,CAAC,GAAG;IAAE,OAAO,EAAE,sBAAsB,CAAA;CAAE,CAAC;AACrH,OAAO,KAAK,EACV,cAAc,EACd,aAAa,EACb,mBAAmB,EACnB,WAAW,EACX,WAAW,EACX,YAAY,EACZ,iBAAiB,EACjB,cAAc,EACd,UAAU,EACV,WAAW,EACX,kBAAkB,EAClB,gBAAgB,EAChB,kBAAkB,EAClB,SAAS,EACT,iBAAiB,EACjB,4BAA4B,EAC7B,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAAE,UAAU,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAC3E,OAAO,GAAG,MAAM,OAAO,CAAC;AAOxB,MAAM,MAAM,eAAe,GAAG;IAC5B,EAAE,EAAE,MAAM,CAAC;IACX,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB,GAAG,CACA;IACE,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,GAAG,CAAC,QAAQ,CAAC;CACxB,GACD;IACE,gBAAgB,EAAE,MAAM,CAAC;CAC1B,CACJ,CAAC;AAEF,MAAM,MAAM,WAAW,GAAG,eAAe,CAAC;AAE1C,qBAAa,UAAW,SAAQ,aAAa;IACpC,IAAI,EAAE,GAAG,CAAC,cAAc,CAAC;IAChC,OAAO,CAAC,MAAM,CAAC,CAAS;IACxB,OAAO,CAAC,WAAW,CAAiC;IACpD,MAAM,EAAE,cAAc,CAAC;gBAEX,MAAM,EAAE,eAAe;IA6D7B,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;YA8Bb,8BAA8B;IAS5C,IAAW,QAAQ,IAAI;QACrB,4BAA4B,EAAE,OAAO,CAAC;QACtC,qBAAqB,EAAE,OAAO,CAAC;QAC/B,SAAS,EAAE,OAAO,CAAC;QACnB,WAAW,EAAE,OAAO,CAAC;QACrB,cAAc,EAAE,OAAO,CAAC;QACxB,gBAAgB,EAAE,OAAO,CAAC;QAC1B,qBAAqB,EAAE,OAAO,CAAC;QAC/B,eAAe,EAAE,OAAO,CAAC;KAC1B,CAWA;IAEK,WAAW,CAAC,EAChB,SAAS,EACT,MAAM,GACP,EAAE;QACD,SAAS,EAAE,WAAW,CAAC;QACvB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;KACvC,GAAG,OAAO,CAAC,IAAI,CAAC;IAIX,UAAU,CAAC,EACf,SAAS,EACT,MAAM,EACN,WAAW,GACZ,EAAE;QACD,SAAS,EAAE,WAAW,CAAC;QACvB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;QACtC,WAAW,EAAE,MAAM,EAAE,CAAC;KACvB,GAAG,OAAO,CAAC,IAAI,CAAC;IAIX,UAAU,CAAC,EAAE,SAAS,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAIpE,SAAS,CAAC,EAAE,SAAS,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAInE,MAAM,CAAC,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAIrG,WAAW,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAC;QAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAI9G,IAAI,CAAC,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;KAAE,GAAG,OAAO,CAAC,CAAC,GAAG,IAAI,CAAC;IAI/G;;OAEG;IAEG,aAAa,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,iBAAiB,GAAG,IAAI,CAAC;IAIpF,UAAU,CAAC,EAAE,MAAM,EAAE,EAAE;QAAE,MAAM,EAAE,iBAAiB,CAAA;KAAE,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAIjF,YAAY,CAAC,EACjB,EAAE,EACF,KAAK,EACL,QAAQ,GACT,EAAE;QACD,EAAE,EAAE,MAAM,CAAC;QACX,KAAK,EAAE,MAAM,CAAC;QACd,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACnC,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAIxB,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAI/D,gBAAgB,CAAC,EAAE,UAAU,EAAE,EAAE;QAAE,UAAU,EAAE,MAAM,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,CAAC;IAIpG,YAAY,CAAC,IAAI,EAAE;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,CAAC;IAI7F,cAAc,CAAC,EACnB,QAAQ,GACT,EAAE;QACD,QAAQ,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,eAAe,EAAE,WAAW,CAAC,CAAC,GAAG;YACvD,EAAE,EAAE,MAAM,CAAC;YACX,OAAO,CAAC,EAAE;gBACR,QAAQ,CAAC,EAAE,sBAAsB,CAAC,UAAU,CAAC,CAAC;gBAC9C,OAAO,CAAC,EAAE,sBAAsB,CAAC,SAAS,CAAC,CAAC;aAC7C,CAAC;SACH,CAAC,EAAE,CAAC;KACN,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IAIxB,cAAc,CAAC,UAAU,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAInD,eAAe,CAAC,EAAE,UAAU,EAAE,EAAE;QAAE,UAAU,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,mBAAmB,GAAG,IAAI,CAAC;IAI5F,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,mBAAmB,CAAA;KAAE,GAAG,OAAO,CAAC,mBAAmB,CAAC;IAI3F,cAAc,CAAC,EACnB,UAAU,EACV,aAAa,EACb,QAAQ,GACT,EAAE;QACD,UAAU,EAAE,MAAM,CAAC;QACnB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACpC,GAAG,OAAO,CAAC,mBAAmB,CAAC;IAIhC;;OAEG;IACG,qBAAqB,CAAC,EAC1B,YAAY,EACZ,KAAK,EACL,MAAM,EACN,MAAM,EACN,cAAc,GACf,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,EAAE,MAAM,CAAC;QACf,MAAM,EAAE,UAAU,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QACvC,cAAc,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KACrC,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;IAIrD,mBAAmB,CAAC,EACxB,YAAY,EACZ,KAAK,EACL,IAAI,GACL,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;QACd,IAAI,EAAE;YACJ,MAAM,EAAE,MAAM,CAAC;YACf,MAAM,CAAC,EAAE,UAAU,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YACxC,KAAK,CAAC,EAAE,MAAM,CAAC;YACf,cAAc,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;YAC1C,YAAY,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;SACzC,CAAC;KACH,GAAG,OAAO,CAAC,gBAAgB,GAAG,SAAS,CAAC;IAInC,uBAAuB,CAAC,EAC5B,YAAY,EACZ,KAAK,EACL,UAAU,EACV,QAAQ,GACT,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;QACd,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,QAAQ,EAAE,gBAAgB,CAAC;KAC5B,GAAG,OAAO,CAAC,IAAI,CAAC;IAIX,oBAAoB,CAAC,EACzB,YAAY,EACZ,KAAK,GACN,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;KACf,GAAG,OAAO,CAAC,gBAAgB,GAAG,IAAI,CAAC;IAI9B,gBAAgB,CAAC,IAAI,GAAE,4BAAiC,GAAG,OAAO,CAAC,YAAY,CAAC;IAIhF,kBAAkB,CAAC,EACvB,KAAK,EACL,YAAY,GACb,EAAE;QACD,KAAK,EAAE,MAAM,CAAC;QACd,YAAY,CAAC,EAAE,MAAM,CAAC;KACvB,GAAG,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC;IAIzB,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAI5B;;OAEG;IACG,WAAW,CAAC,OAAO,EAAE,kBAAkB,GAAG,OAAO,CAAC,IAAI,CAAC;IAIvD,WAAW,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IAIrD,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAI5D,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIjD;;OAEG;IACH,OAAO,CAAC,qBAAqB;IAYvB,UAAU,CAAC,IAAI,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;IAI3C,UAAU,CAAC,EACf,MAAM,EACN,OAAO,EACP,OAAO,GACR,EAAE;QACD,MAAM,EAAE,MAAM,CAAC;QACf,OAAO,EAAE,MAAM,CAAC;QAChB,OAAO,EAAE,OAAO,CAAC,gBAAgB,CAAC,CAAC;KACpC,GAAG,OAAO,CAAC,IAAI,CAAC;IAIX,QAAQ,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC;IAItD,kBAAkB,CAAC,IAAI,EAAE,kBAAkB,GAAG,OAAO,CAAC;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,KAAK,EAAE,UAAU,EAAE,CAAA;KAAE,CAAC;IAI1G,gBAAgB,CAAC,IAAI,EAAE;QAAE,OAAO,EAAE,UAAU,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAIhE,gBAAgB,CAAC,IAAI,EAAE;QAC3B,OAAO,EAAE;YACP,OAAO,EAAE,MAAM,CAAC;YAChB,MAAM,EAAE,MAAM,CAAC;YACf,OAAO,EAAE,OAAO,CAAC,gBAAgB,CAAC,CAAC;SACpC,EAAE,CAAC;KACL,GAAG,OAAO,CAAC,IAAI,CAAC;IAIX,iBAAiB,CAAC,IAAI,EAAE;QAAE,QAAQ,EAAE,MAAM,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAIpE;;OAEG;IACG,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE;QAAE,EAAE,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;IAIvE,oBAAoB,CAAC,EACzB,QAAQ,EAAE,SAAS,EACnB,UAAU,EAAE,WAAW,EACvB,QAAQ,EAAE,SAAS,EACnB,UAAU,EAAE,WAAW,EACvB,MAAM,EAAE,OAAO,GAChB,EAAE;QACD,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,iBAAiB,CAAC;QAC9B,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,MAAM,CAAC,EAAE,aAAa,CAAC;KACxB,GAAG,OAAO,CAAC;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,MAAM,EAAE,YAAY,EAAE,CAAA;KAAE,CAAC;IAU7D,SAAS,CAAC,MAAM,EAAE,YAAY,GAAG,OAAO,CAAC;QAAE,KAAK,EAAE,YAAY,CAAA;KAAE,CAAC;IAIjE,iBAAiB,CAAC,EACtB,KAAK,EAAE,MAAM,EACb,UAAU,EAAE,WAAW,GACxB,EAAE;QACD,KAAK,EAAE,MAAM,CAAC;QACd,UAAU,EAAE,iBAAiB,CAAC;KAC/B,GAAG,OAAO,CAAC;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,MAAM,EAAE,YAAY,EAAE,CAAA;KAAE,CAAC;IAI7D,oBAAoB,CAAC,EACzB,QAAQ,EAAE,SAAS,EACnB,UAAU,EAAE,WAAW,EACvB,UAAU,EAAE,WAAW,GACxB,EAAE;QACD,UAAU,EAAE,iBAAiB,CAAC;QAC9B,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,MAAM,CAAC;KACpB,GAAG,OAAO,CAAC;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,MAAM,EAAE,YAAY,EAAE,CAAA;KAAE,CAAC;IAQ7D,gBAAgB,CAAC,EACrB,OAAO,EACP,MAAM,EACN,UAAU,EAAE,WAAW,GACxB,EAAE;QACD,OAAO,EAAE,MAAM,CAAC;QAChB,MAAM,EAAE,MAAM,CAAC;QACf,UAAU,EAAE,iBAAiB,CAAC;KAC/B,GAAG,OAAO,CAAC;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,MAAM,EAAE,YAAY,EAAE,CAAA;KAAE,CAAC;CAGpE"}