@mastra/mssql 0.0.0-fix-message-list-merge-20250718043058 → 0.0.0-fix-tool-call-history-20250731222019

package/src/storage/index.ts

@@ -4,6 +4,7 @@ export type MastraMessageV2WithTypedContent = Omit<MastraMessageV2, 'content'> &
 import { ErrorCategory, ErrorDomain, MastraError } from '@mastra/core/error';
 import type { MetricResult } from '@mastra/core/eval';
 import type { MastraMessageV1, StorageThreadType } from '@mastra/core/memory';
+import type { ScoreRowData } from '@mastra/core/scores';
 import {
   MastraStorage,
   TABLE_MESSAGES,
@@ -23,6 +24,7 @@ import type {
   WorkflowRun,
   WorkflowRuns,
   PaginationArgs,
+  StoragePagination,
 } from '@mastra/core/storage';
 import { parseSqlIdentifier, parseFieldKey } from '@mastra/core/utils';
 import type { WorkflowRunState } from '@mastra/core/workflows';
@@ -129,10 +131,16 @@ export class MSSQLStore extends MastraStorage {
   public get supports(): {
     selectByIncludeResourceScope: boolean;
     resourceWorkingMemory: boolean;
+    hasColumn: boolean;
+    createTable: boolean;
+    deleteMessages: boolean;
   } {
     return {
       selectByIncludeResourceScope: true,
       resourceWorkingMemory: true,
+      hasColumn: true,
+      createTable: true,
+      deleteMessages: false,
     };
   }
 
@@ -2041,4 +2049,88 @@ export class MSSQLStore extends MastraStorage {
       throw mastraError;
     }
   }
+
+  async getScoreById({ id }: { id: string }): Promise<ScoreRowData | null> {
+    throw new MastraError({
+      id: 'STORAGE_MONGODB_STORE_GET_SCORE_BY_ID_FAILED',
+      domain: ErrorDomain.STORAGE,
+      category: ErrorCategory.THIRD_PARTY,
+      details: { id },
+      text: 'getScoreById is not implemented yet in MongoDBStore',
+    });
+  }
+
+  async saveScore(_score: Omit<ScoreRowData, 'id' | 'createdAt' | 'updatedAt'>): Promise<{ score: ScoreRowData }> {
+    throw new MastraError({
+      id: 'STORAGE_MONGODB_STORE_SAVE_SCORE_FAILED',
+      domain: ErrorDomain.STORAGE,
+      category: ErrorCategory.THIRD_PARTY,
+      details: {},
+      text: 'saveScore is not implemented yet in MongoDBStore',
+    });
+  }
+
+  async getScoresByScorerId({
+    scorerId,
+    pagination: _pagination,
+    entityId,
+    entityType,
+  }: {
+    scorerId: string;
+    pagination: StoragePagination;
+    entityId?: string;
+    entityType?: string;
+  }): Promise<{ pagination: PaginationInfo; scores: ScoreRowData[] }> {
+    throw new MastraError({
+      id: 'STORAGE_MONGODB_STORE_GET_SCORES_BY_SCORER_ID_FAILED',
+      domain: ErrorDomain.STORAGE,
+      category: ErrorCategory.THIRD_PARTY,
+      details: { scorerId, entityId: entityId || '', entityType: entityType || '' },
+      text: 'getScoresByScorerId is not implemented yet in MongoDBStore',
+    });
+  }
+
+  async getScoresByRunId({
+    runId,
+    pagination: _pagination,
+  }: {
+    runId: string;
+    pagination: StoragePagination;
+  }): Promise<{ pagination: PaginationInfo; scores: ScoreRowData[] }> {
+    throw new MastraError({
+      id: 'STORAGE_MONGODB_STORE_GET_SCORES_BY_RUN_ID_FAILED',
+      domain: ErrorDomain.STORAGE,
+      category: ErrorCategory.THIRD_PARTY,
+      details: { runId },
+      text: 'getScoresByRunId is not implemented yet in MongoDBStore',
+    });
+  }
+
+  async getScoresByEntityId({
+    entityId,
+    entityType,
+    pagination: _pagination,
+  }: {
+    pagination: StoragePagination;
+    entityId: string;
+    entityType: string;
+  }): Promise<{ pagination: PaginationInfo; scores: ScoreRowData[] }> {
+    throw new MastraError({
+      id: 'STORAGE_MONGODB_STORE_GET_SCORES_BY_ENTITY_ID_FAILED',
+      domain: ErrorDomain.STORAGE,
+      category: ErrorCategory.THIRD_PARTY,
+      details: { entityId, entityType },
+      text: 'getScoresByEntityId is not implemented yet in MongoDBStore',
+    });
+  }
+
+  async dropTable({ tableName }: { tableName: TABLE_NAMES }): Promise<void> {
+    throw new MastraError({
+      id: 'STORAGE_MONGODB_STORE_DROP_TABLE_FAILED',
+      domain: ErrorDomain.STORAGE,
+      category: ErrorCategory.THIRD_PARTY,
+      details: { tableName },
+      text: 'dropTable is not implemented yet in MongoDBStore',
+    });
+  }
 }

package/tsconfig.build.json

@@ -0,0 +1,9 @@
+{
+  "extends": ["./tsconfig.json", "../../tsconfig.build.json"],
+  "compilerOptions": {
+    "outDir": "./dist",
+    "rootDir": "./src"
+  },
+  "include": ["src/**/*"],
+  "exclude": ["node_modules", "**/*.test.ts", "src/**/*.mock.ts"]
+}

package/tsconfig.json

@@ -1,5 +1,5 @@
 {
   "extends": "../../tsconfig.node.json",
-  "include": ["src/**/*"],
+  "include": ["src/**/*", "tsup.config.ts"],
   "exclude": ["node_modules", "**/*.test.ts"]
 }

package/tsup.config.ts

@@ -0,0 +1,22 @@
+import { spawn } from 'child_process';
+import { promisify } from 'util';
+import { defineConfig } from 'tsup';
+
+const exec = promisify(spawn);
+
+export default defineConfig({
+  entry: ['src/index.ts'],
+  format: ['esm', 'cjs'],
+  clean: true,
+  dts: false,
+  splitting: true,
+  treeshake: {
+    preset: 'smallest',
+  },
+  sourcemap: true,
+  onSuccess: async () => {
+    await exec('pnpm', ['tsc', '-p', 'tsconfig.build.json'], {
+      stdio: 'inherit',
+    });
+  },
+});

package/dist/_tsup-dts-rollup.d.ts

@@ -1,213 +0,0 @@
-import type { EvalRow } from '@mastra/core/storage';
-import type { MastraMessageContentV2 } from '@mastra/core/agent';
-import type { MastraMessageV1 } from '@mastra/core/memory';
-import type { MastraMessageV2 } from '@mastra/core/agent';
-import { MastraStorage } from '@mastra/core/storage';
-import type { PaginationArgs } from '@mastra/core/storage';
-import type { PaginationInfo } from '@mastra/core/storage';
-import sql from 'mssql';
-import type { StorageColumn } from '@mastra/core/storage';
-import type { StorageGetMessagesArg } from '@mastra/core/storage';
-import type { StorageResourceType } from '@mastra/core/storage';
-import type { StorageThreadType } from '@mastra/core/memory';
-import type { TABLE_NAMES } from '@mastra/core/storage';
-import type { WorkflowRun } from '@mastra/core/storage';
-import type { WorkflowRuns } from '@mastra/core/storage';
-import type { WorkflowRunState } from '@mastra/core/workflows';
-
-declare type MastraMessageV2WithTypedContent = Omit<MastraMessageV2, 'content'> & {
-    content: MastraMessageContentV2;
-};
-export { MastraMessageV2WithTypedContent }
-export { MastraMessageV2WithTypedContent as MastraMessageV2WithTypedContent_alias_1 }
-
-declare type MSSQLConfig = MSSQLConfigType;
-export { MSSQLConfig }
-export { MSSQLConfig as MSSQLConfig_alias_1 }
-
-declare type MSSQLConfigType = {
-    schemaName?: string;
-} & ({
-    server: string;
-    port: number;
-    database: string;
-    user: string;
-    password: string;
-    options?: sql.IOptions;
-} | {
-    connectionString: string;
-});
-export { MSSQLConfigType }
-export { MSSQLConfigType as MSSQLConfigType_alias_1 }
-
-declare class MSSQLStore extends MastraStorage {
-    pool: sql.ConnectionPool;
-    private schema?;
-    private setupSchemaPromise;
-    private schemaSetupComplete;
-    private isConnected;
-    constructor(config: MSSQLConfigType);
-    init(): Promise<void>;
-    private _performInitializationAndStore;
-    get supports(): {
-        selectByIncludeResourceScope: boolean;
-        resourceWorkingMemory: boolean;
-    };
-    private getTableName;
-    private getSchemaName;
-    private transformEvalRow;
-    /** @deprecated use getEvals instead */
-    getEvalsByAgentName(agentName: string, type?: 'test' | 'live'): Promise<EvalRow[]>;
-    batchInsert({ tableName, records }: {
-        tableName: TABLE_NAMES;
-        records: Record<string, any>[];
-    }): Promise<void>;
-    /** @deprecated use getTracesPaginated instead*/
-    getTraces(args: {
-        name?: string;
-        scope?: string;
-        attributes?: Record<string, string>;
-        filters?: Record<string, any>;
-        page: number;
-        perPage?: number;
-        fromDate?: Date;
-        toDate?: Date;
-    }): Promise<any[]>;
-    getTracesPaginated(args: {
-        name?: string;
-        scope?: string;
-        attributes?: Record<string, string>;
-        filters?: Record<string, any>;
-    } & PaginationArgs): Promise<PaginationInfo & {
-        traces: any[];
-    }>;
-    private setupSchema;
-    protected getSqlType(type: StorageColumn['type'], isPrimaryKey?: boolean): string;
-    createTable({ tableName, schema, }: {
-        tableName: TABLE_NAMES;
-        schema: Record<string, StorageColumn>;
-    }): Promise<void>;
-    protected getDefaultValue(type: StorageColumn['type']): string;
-    alterTable({ tableName, schema, ifNotExists, }: {
-        tableName: TABLE_NAMES;
-        schema: Record<string, StorageColumn>;
-        ifNotExists: string[];
-    }): Promise<void>;
-    clearTable({ tableName }: {
-        tableName: TABLE_NAMES;
-    }): Promise<void>;
-    insert({ tableName, record }: {
-        tableName: TABLE_NAMES;
-        record: Record<string, any>;
-    }): Promise<void>;
-    load<R>({ tableName, keys }: {
-        tableName: TABLE_NAMES;
-        keys: Record<string, string>;
-    }): Promise<R | null>;
-    getThreadById({ threadId }: {
-        threadId: string;
-    }): Promise<StorageThreadType | null>;
-    getThreadsByResourceIdPaginated(args: {
-        resourceId: string;
-    } & PaginationArgs): Promise<PaginationInfo & {
-        threads: StorageThreadType[];
-    }>;
-    saveThread({ thread }: {
-        thread: StorageThreadType;
-    }): Promise<StorageThreadType>;
-    /**
-     * @deprecated use getThreadsByResourceIdPaginated instead
-     */
-    getThreadsByResourceId(args: {
-        resourceId: string;
-    }): Promise<StorageThreadType[]>;
-    /**
-     * Updates a thread's title and metadata, merging with existing metadata. Returns the updated thread.
-     */
-    updateThread({ id, title, metadata, }: {
-        id: string;
-        title: string;
-        metadata: Record<string, unknown>;
-    }): Promise<StorageThreadType>;
-    deleteThread({ threadId }: {
-        threadId: string;
-    }): Promise<void>;
-    private _getIncludedMessages;
-    /**
-     * @deprecated use getMessagesPaginated instead
-     */
-    getMessages(args: StorageGetMessagesArg & {
-        format?: 'v1';
-    }): Promise<MastraMessageV1[]>;
-    getMessages(args: StorageGetMessagesArg & {
-        format: 'v2';
-    }): Promise<MastraMessageV2[]>;
-    getMessagesPaginated(args: StorageGetMessagesArg & {
-        format?: 'v1' | 'v2';
-    }): Promise<PaginationInfo & {
-        messages: MastraMessageV1[] | MastraMessageV2[];
-    }>;
-    private _parseAndFormatMessages;
-    saveMessages(args: {
-        messages: MastraMessageV1[];
-        format?: undefined | 'v1';
-    }): Promise<MastraMessageV1[]>;
-    saveMessages(args: {
-        messages: MastraMessageV2[];
-        format: 'v2';
-    }): Promise<MastraMessageV2[]>;
-    persistWorkflowSnapshot({ workflowName, runId, snapshot, }: {
-        workflowName: string;
-        runId: string;
-        snapshot: WorkflowRunState;
-    }): Promise<void>;
-    loadWorkflowSnapshot({ workflowName, runId, }: {
-        workflowName: string;
-        runId: string;
-    }): Promise<WorkflowRunState | null>;
-    private hasColumn;
-    private parseWorkflowRun;
-    getWorkflowRuns({ workflowName, fromDate, toDate, limit, offset, resourceId, }?: {
-        workflowName?: string;
-        fromDate?: Date;
-        toDate?: Date;
-        limit?: number;
-        offset?: number;
-        resourceId?: string;
-    }): Promise<WorkflowRuns>;
-    getWorkflowRunById({ runId, workflowName, }: {
-        runId: string;
-        workflowName?: string;
-    }): Promise<WorkflowRun | null>;
-    updateMessages({ messages, }: {
-        messages: (Partial<Omit<MastraMessageV2, 'createdAt'>> & {
-            id: string;
-            content?: {
-                metadata?: MastraMessageContentV2['metadata'];
-                content?: MastraMessageContentV2['content'];
-            };
-        })[];
-    }): Promise<MastraMessageV2[]>;
-    close(): Promise<void>;
-    getEvals(options?: {
-        agentName?: string;
-        type?: 'test' | 'live';
-    } & PaginationArgs): Promise<PaginationInfo & {
-        evals: EvalRow[];
-    }>;
-    saveResource({ resource }: {
-        resource: StorageResourceType;
-    }): Promise<StorageResourceType>;
-    updateResource({ resourceId, workingMemory, metadata, }: {
-        resourceId: string;
-        workingMemory?: string;
-        metadata?: Record<string, unknown>;
-    }): Promise<StorageResourceType>;
-    getResourceById({ resourceId }: {
-        resourceId: string;
-    }): Promise<StorageResourceType | null>;
-}
-export { MSSQLStore }
-export { MSSQLStore as MSSQLStore_alias_1 }
-
-export { }

package/dist/index.d.cts

@@ -1,4 +0,0 @@
-export { MastraMessageV2WithTypedContent } from './_tsup-dts-rollup.cjs';
-export { MSSQLConfigType } from './_tsup-dts-rollup.cjs';
-export { MSSQLConfig } from './_tsup-dts-rollup.cjs';
-export { MSSQLStore } from './_tsup-dts-rollup.cjs';