@mastra/elasticsearch 1.3.1 → 1.4.0

package/dist/index.d.cts

@@ -1,6 +1,11 @@
 import { Client } from "@elastic/elasticsearch";
 import { CreateIndexParams, DeleteIndexParams, DeleteVectorParams, DeleteVectorsParams, DescribeIndexParams, IndexStats, MastraVector, QueryResult, QueryVectorParams, UpdateVectorParams, UpsertVectorParams } from "@mastra/core/vector";
 import { BlacklistedRootOperators, LogicalOperatorValueMap, OperatorValueMap, VectorFilter } from "@mastra/core/vector/filter";
+import { MastraStorage, MemoryStorage, PaginationInfo, ScoreTenancyFilters, ScoresStorage, StorageCloneThreadInput, StorageCloneThreadOutput, StorageDomains, StorageListMessagesInput, StorageListMessagesOutput, StorageListThreadsInput, StorageListThreadsOutput, StorageListWorkflowRunsInput, StoragePagination, StorageResourceType, UpdateWorkflowStateOptions, WorkflowRun, WorkflowRuns, WorkflowsStorage } from "@mastra/core/storage";
+import { MastraMessageContentV2 } from "@mastra/core/agent";
+import { MastraDBMessage, StorageThreadType } from "@mastra/core/memory";
+import { StepResult, WorkflowRunState } from "@mastra/core/workflows";
+import { SaveScorePayload, ScoreRowData, ScoringSource } from "@mastra/core/evals";
 //#region src/vector/filter.d.ts
 type ElasticSearchOperatorValueMap = Omit<OperatorValueMap, '$options' | '$nor' | '$elemMatch'>;
 type ElasticSearchLogicalOperatorValueMap = Omit<LogicalOperatorValueMap, '$nor'>;
@@ -138,5 +143,272 @@ declare class ElasticSearchVector extends MastraVector<ElasticSearchVectorFilter
   deleteVectors({ indexName, filter, ids }: DeleteVectorsParams<ElasticSearchVectorFilter>): Promise<void>;
 }
 //#endregion
-export { type ElasticSearchAuth, ElasticSearchVector, type ElasticSearchVectorConfig, type ElasticSearchVectorFilter };
+//#region src/storage/types.d.ts
+/**
+ * ElasticSearch storage configuration type.
+ *
+ * Accepts either:
+ * - A pre-configured ElasticSearch client: `{ id, client }`
+ * - Connection parameters: `{ id, url, auth? }`
+ *
+ * This mirrors the config surface of `ElasticSearchVector` so both can share
+ * the same connection settings (or the same client instance).
+ */
+type ElasticSearchConfig = {
+  id: string;
+  /**
+   * When true, automatic initialization (index creation) is disabled.
+   * You must call `storage.init()` explicitly before use.
+   */
+  disableInit?: boolean;
+} & ({
+  /**
+   * Pre-configured ElasticSearch client (from `@elastic/elasticsearch`).
+   *
+   * @example
+   * ```typescript
+   * import { Client } from '@elastic/elasticsearch';
+   *
+   * const client = new Client({ node: 'http://localhost:9200' });
+   * const store = new ElasticSearchStore({ id: 'my-store', client });
+   * ```
+   */
+  client: Client;
+  url?: never;
+  auth?: never;
+} | {
+  /**
+   * ElasticSearch node URL.
+   *
+   * @example
+   * ```typescript
+   * const store = new ElasticSearchStore({
+   *   id: 'my-store',
+   *   url: 'http://localhost:9200',
+   *   auth: { apiKey: '...' },
+   * });
+   * ```
+   */
+  url: string;
+  auth?: ElasticSearchAuth;
+  client?: never;
+});
+//#endregion
+//#region src/storage/store.d.ts
+/**
+ * ElasticSearch storage adapter for Mastra.
+ *
+ * Implements the memory, workflows, and scores storage domains on top of
+ * ElasticSearch. Shares the same connection config surface as
+ * `ElasticSearchVector`, so both can reuse one client or connection config.
+ *
+ * @example
+ * ```typescript
+ * // Using connection parameters
+ * const storage = new ElasticSearchStore({
+ *   id: 'my-store',
+ *   url: 'http://localhost:9200',
+ *   auth: { apiKey: '...' },
+ * });
+ *
+ * // Access memory domain
+ * const memory = await storage.getStore('memory');
+ * await memory?.saveThread({ thread });
+ * ```
+ *
+ * @example
+ * ```typescript
+ * // Using a pre-configured client shared with ElasticSearchVector
+ * import { Client } from '@elastic/elasticsearch';
+ *
+ * const client = new Client({ node: 'http://localhost:9200' });
+ * const storage = new ElasticSearchStore({ id: 'my-store', client });
+ * const vector = new ElasticSearchVector({ id: 'my-vector', client });
+ * ```
+ */
+declare class ElasticSearchStore extends MastraStorage {
+  private client;
+  private shouldManageConnection;
+  stores: StorageDomains;
+  constructor(config: ElasticSearchConfig);
+  getClient(): Client;
+  close(): Promise<void>;
+}
+//#endregion
+//#region src/storage/db.d.ts
+interface ElasticSearchDomainConfig {
+  client: Client;
+}
+//#endregion
+//#region src/storage/domains/memory/index.d.ts
+declare class MemoryElasticSearch extends MemoryStorage {
+  readonly supportsPartialThreadUpdate = true;
+  private db;
+  constructor(config: ElasticSearchDomainConfig);
+  dangerouslyClearAll(): Promise<void>;
+  getThreadById({ threadId, resourceId }: {
+    threadId: string;
+    resourceId?: string;
+  }): Promise<StorageThreadType | null>;
+  listThreadsByResourceId(args: StorageListThreadsInput): Promise<StorageListThreadsOutput>;
+  listThreads(args: StorageListThreadsInput): Promise<StorageListThreadsOutput>;
+  saveThread({ thread }: {
+    thread: StorageThreadType;
+  }): Promise<StorageThreadType>;
+  updateThread({ id, title, metadata }: {
+    id: string;
+    title?: string;
+    metadata?: Record<string, unknown>;
+  }): Promise<StorageThreadType>;
+  deleteThread({ threadId }: {
+    threadId: string;
+  }): Promise<void>;
+  saveMessages(args: {
+    messages: MastraDBMessage[];
+  }): Promise<{
+    messages: MastraDBMessage[];
+  }>;
+  /**
+   * Returns all messages that belong to a thread, sorted in insertion order
+   * (createdAt with `_index` tiebreaker).
+   */
+  private listThreadMessages;
+  /** Returns all message documents across all threads (excludes index docs). */
+  private listAllMessages;
+  private getThreadIdForMessage;
+  /**
+   * Fetches the messages named by `include` together with their surrounding context.
+   *
+   * @param include - Message ids to pin, each with an optional before/after window.
+   * @param resourceId - When set, drops any pinned or context message owned by another
+   * resource so an id from another resource returns nothing.
+   */
+  private getIncludedMessages;
+  private parseStoredMessage;
+  listMessagesById({ messageIds }: {
+    messageIds: string[];
+  }): Promise<{
+    messages: MastraDBMessage[];
+  }>;
+  listMessages(args: StorageListMessagesInput): Promise<StorageListMessagesOutput>;
+  getResourceById({ resourceId }: {
+    resourceId: string;
+  }): Promise<StorageResourceType | null>;
+  saveResource({ resource }: {
+    resource: StorageResourceType;
+  }): Promise<StorageResourceType>;
+  updateResource({ resourceId, workingMemory, metadata }: {
+    resourceId: string;
+    workingMemory?: string;
+    metadata?: Record<string, unknown>;
+  }): Promise<StorageResourceType>;
+  updateMessages(args: {
+    messages: (Partial<Omit<MastraDBMessage, 'createdAt'>> & {
+      id: string;
+      content?: {
+        metadata?: MastraMessageContentV2['metadata'];
+        content?: MastraMessageContentV2['content'];
+      };
+    })[];
+  }): Promise<MastraDBMessage[]>;
+  deleteMessages(messageIds: string[]): Promise<void>;
+  private sortThreads;
+  cloneThread(args: StorageCloneThreadInput): Promise<StorageCloneThreadOutput>;
+}
+//#endregion
+//#region src/storage/domains/workflows/index.d.ts
+declare class WorkflowsElasticSearch extends WorkflowsStorage {
+  private db;
+  constructor(config: ElasticSearchDomainConfig);
+  supportsConcurrentUpdates(): boolean;
+  dangerouslyClearAll(): Promise<void>;
+  updateWorkflowResults({ workflowName, runId, stepId, result, requestContext }: {
+    workflowName: string;
+    runId: string;
+    stepId: string;
+    result: StepResult<unknown, unknown, unknown, unknown>;
+    requestContext: Record<string, unknown>;
+  }): Promise<Record<string, StepResult<unknown, unknown, unknown, unknown>>>;
+  updateWorkflowState({ workflowName, runId, opts }: {
+    workflowName: string;
+    runId: string;
+    opts: UpdateWorkflowStateOptions;
+  }): Promise<WorkflowRunState | undefined>;
+  persistWorkflowSnapshot(params: {
+    namespace?: string;
+    workflowName: string;
+    runId: string;
+    resourceId?: string;
+    snapshot: WorkflowRunState;
+    createdAt?: Date;
+    updatedAt?: Date;
+  }): Promise<void>;
+  loadWorkflowSnapshot(params: {
+    namespace: string;
+    workflowName: string;
+    runId: string;
+  }): Promise<WorkflowRunState | null>;
+  getWorkflowRunById({ runId, workflowName }: {
+    runId: string;
+    workflowName?: string;
+  }): Promise<WorkflowRun | null>;
+  deleteWorkflowRunById({ runId, workflowName }: {
+    runId: string;
+    workflowName: string;
+  }): Promise<void>;
+  listWorkflowRuns({ workflowName, fromDate, toDate, perPage, page, resourceId, status }?: StorageListWorkflowRunsInput): Promise<WorkflowRuns>;
+}
+//#endregion
+//#region src/storage/domains/scores/index.d.ts
+declare class ScoresElasticSearch extends ScoresStorage {
+  private db;
+  constructor(config: ElasticSearchDomainConfig);
+  dangerouslyClearAll(): Promise<void>;
+  getScoreById({ id }: {
+    id: string;
+  }): Promise<ScoreRowData | null>;
+  listScoresByScorerId({ scorerId, entityId, entityType, source, pagination, filters }: {
+    scorerId: string;
+    entityId?: string;
+    entityType?: string;
+    source?: ScoringSource;
+    pagination?: StoragePagination;
+    filters?: ScoreTenancyFilters;
+  }): Promise<{
+    scores: ScoreRowData[];
+    pagination: PaginationInfo;
+  }>;
+  saveScore(score: SaveScorePayload): Promise<{
+    score: ScoreRowData;
+  }>;
+  listScoresByRunId({ runId, pagination, filters }: {
+    runId: string;
+    pagination?: StoragePagination;
+    filters?: ScoreTenancyFilters;
+  }): Promise<{
+    scores: ScoreRowData[];
+    pagination: PaginationInfo;
+  }>;
+  listScoresByEntityId({ entityId, entityType, pagination, filters }: {
+    entityId: string;
+    entityType?: string;
+    pagination?: StoragePagination;
+    filters?: ScoreTenancyFilters;
+  }): Promise<{
+    scores: ScoreRowData[];
+    pagination: PaginationInfo;
+  }>;
+  listScoresBySpan({ traceId, spanId, pagination, filters }: {
+    traceId: string;
+    spanId: string;
+    pagination?: StoragePagination;
+    filters?: ScoreTenancyFilters;
+  }): Promise<{
+    scores: ScoreRowData[];
+    pagination: PaginationInfo;
+  }>;
+  private fetchAndFilterScores;
+}
+//#endregion
+export { type ElasticSearchAuth, type ElasticSearchConfig, ElasticSearchStore, ElasticSearchVector, type ElasticSearchVectorConfig, type ElasticSearchVectorFilter, MemoryElasticSearch, ScoresElasticSearch, WorkflowsElasticSearch };
 //# sourceMappingURL=index.d.cts.map

package/dist/index.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.cts","names":[],"sources":["../src/vector/filter.ts","../src/vector/index.ts"],"mappings":";;;;KAUK,gCAAgC,KAAK;KAErC,uCAAuC,KAAK;KAE5C,2BAA2B;KAEpB,4BAA4B,mBAChC,+BACN,+BACA,sCACA;;;KCaG,4BAA4B,kBAAkB;KAEvC;EAAsB;;EAAqB;EAAkB;;EAAuB;;KAEpF;EACN;EAAY,QAAQ;EAAqB;EAAa;;EACtD;EAAY;EAAa,OAAO;EAAmB;;cAE5C,4BAA4B,aAAa;UAC5C;;;;;;;;EASI,YAAA,QAAQ;;;;;;;;;EA6Bd,cAAc,WAAW,WAAW,UAAqB,oBAAoB;;;;;;EAkD7E,eAAe;;;;;YAwBL,sBAAsB,mBAAmB,mBAAmB,iBAAiB;;;;;;;EAoDvF,gBAAgB,aAAa,sBAAsB,QAAQ;;;;;;;EAqB3D,cAAc,aAAa,oBAAoB;;;;;;;;;;EA4B/C,SAAS,WAAW,SAAS,UAAe,OAAO,qBAAqB;;;;;;;;;;;EA+HxE,QACJ,WACA,aACA,QACA,MACA,iBACC,4BAA4B,QAAQ;;;;;;;;UA+D/B;;;;;;;UAYA;;;;;;;;;;;EAeF,aAAa,QAAQ,mBAAmB,6BAA6B;;;;UAwD7D;;;;UAuFA;;;;;;;;EAyDR,eAAe,WAAW,MAAM,qBAAqB;EA2BrD,gBAAgB,WAAW,QAAQ,OAAO,oBAAoB,6BAA6B"}
+{"version":3,"file":"index.d.cts","names":[],"sources":["../src/vector/filter.ts","../src/vector/index.ts","../src/storage/types.ts","../src/storage/store.ts","../src/storage/db.ts","../src/storage/domains/memory/index.ts","../src/storage/domains/workflows/index.ts","../src/storage/domains/scores/index.ts"],"mappings":";;;;;;;;;KAUK,gCAAgC,KAAK;KAErC,uCAAuC,KAAK;KAE5C,2BAA2B;KAEpB,4BAA4B,mBAChC,+BACN,+BACA,sCACA;;;KCaG,4BAA4B,kBAAkB;KAEvC;EAAsB;;EAAqB;EAAkB;;EAAuB;;KAEpF;EACN;EAAY,QAAQ;EAAqB;EAAa;;EACtD;EAAY;EAAa,OAAO;EAAmB;;cAE5C,4BAA4B,aAAa;UAC5C;;;;;;;;EASI,YAAA,QAAQ;;;;;;;;;EA6Bd,cAAc,WAAW,WAAW,UAAqB,oBAAoB;;;;;;EAkD7E,eAAe;;;;;YAwBL,sBAAsB,mBAAmB,mBAAmB,iBAAiB;;;;;;;EAoDvF,gBAAgB,aAAa,sBAAsB,QAAQ;;;;;;;EAqB3D,cAAc,aAAa,oBAAoB;;;;;;;;;;EA4B/C,SAAS,WAAW,SAAS,UAAe,OAAO,qBAAqB;;;;;;;;;;;EA+HxE,QACJ,WACA,aACA,QACA,MACA,iBACC,4BAA4B,QAAQ;;;;;;;;UA+D/B;;;;;;;UAYA;;;;;;;;;;;EAeF,aAAa,QAAQ,mBAAmB,6BAA6B;;;;UAwD7D;;;;UAuFA;;;;;;;;EAyDR,eAAe,WAAW,MAAM,qBAAqB;EA2BrD,gBAAgB,WAAW,QAAQ,OAAO,oBAAoB,6BAA6B;;;;;;;;;;;;;;KCnrBvF;EACV;;;;;EAKA;;;;;;;;;;;;;EAcI,QAAQ;EACR;EACA;;;;;;;;;;;;;;EAeA;EACA,OAAO;EACP;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cCXO,2BAA2B;UAC9B;UACA;EACD,QAAQ;EAEH,YAAA,QAAQ;EA8Bb,aAAa;EAIP,SAAS;;;;UCiNP;EACf,QAAQ;;;;cC9PG,4BAA4B;WACrB;UACV;EAEI,YAAA,QAAQ;EAKP,uBAAuB;EAMvB,gBACX,UACA;IAEA;IACA;MACE,QAAQ;EAgCC,wBAAwB,MAAM,0BAA0B,QAAQ;EAIhE,YAAY,MAAM,0BAA0B,QAAQ;EAqGpD,aAAa;IAAY,QAAQ;MAAsB,QAAQ;EAyB/D,eACX,IACA,OACA;IAEA;IACA;IACA,WAAW;MACT,QAAQ;EA0CC,eAAe;IAAc;MAAqB;EA4BlD,aAAa;IAAQ,UAAU;MAAsB;IAAU,UAAU;;;;;;UA4FxE;;UASA;UAOA;;;;;;;;UAiCA;UAqCN;EAUK,mBAAmB;IAAgB;MAAyB;IAAU,UAAU;;EA8DhF,aAAa,MAAM,2BAA2B,QAAQ;EAkMtD,kBAAkB;IAAgB;MAAuB,QAAQ;EAwBjE,eAAe;IAAc,UAAU;MAAwB,QAAQ;EAsBvE,iBACX,YACA,eACA;IAEA;IACA;IACA,WAAW;MACT,QAAQ;EAiCC,eAAe;IAC1B,WAAW,QAAQ,KAAK;MACtB;MACA;QAAY,WAAW;QAAoC,UAAU;;;MAErE,QAAQ;EA4IC,eAAe,uBAAuB;UAoD3C;EAYK,YAAY,MAAM,0BAA0B,QAAQ;;;;cC39BtD,+BAA+B;UAClC;EAEI,YAAA,QAAQ;EAKb;EAIM,uBAAuB;EAIvB,wBACX,cACA,OACA,QACA,QACA;IAEA;IACA;IACA;IACA,QAAQ;IACR,gBAAgB;MACd,QAAQ,eAAe;EA2Dd,sBACX,cACA,OACA;IAEA;IACA;IACA,MAAM;MACJ,QAAQ;EAmDC,wBAAwB;IACnC;IACA;IACA;IACA;IACA,UAAU;IACV,YAAY;IACZ,YAAY;MACV;EA6CS,qBAAqB;IAChC;IACA;IACA;MACE,QAAQ;EA6BC,qBACX,OACA;IAEA;IACA;MACE,QAAQ;EA2CC,wBAAwB,OAAO;IAAkB;IAAe;MAAyB;EAqBzF,mBACX,cACA,UACA,QACA,SACA,MACA,YACA,WACC,+BAAoC,QAAQ;;;;cChVpC,4BAA4B;UAC/B;EAEI,YAAA,QAAQ;EAKP,uBAAuB;EAIvB,eAAe;IAAQ;MAAe,QAAQ;EA2B9C,uBACX,UACA,UACA,YACA,QACA,YACA;IAEA;IACA;IACA;IACA,SAAS;IACT,aAAa;IACb,UAAU;MACR;IACF,QAAQ;IACR,YAAY;;EAsBD,UAAU,OAAO,mBAAmB;IAAU,OAAO;;EAiDrD,oBACX,OACA,YACA;IAEA;IACA,aAAa;IACb,UAAU;MACR;IACF,QAAQ;IACR,YAAY;;EAKD,uBACX,UACA,YACA,YACA;IAEA;IACA;IACA,aAAa;IACb,UAAU;MACR;IACF,QAAQ;IACR,YAAY;;EAgBD,mBACX,SACA,QACA,YACA;IAEA;IACA;IACA,aAAa;IACb,UAAU;MACR;IACF,QAAQ;IACR,YAAY;;UAQA"}

package/dist/index.d.ts

@@ -1,6 +1,11 @@
 import { Client } from "@elastic/elasticsearch";
+import { MastraStorage, MemoryStorage, PaginationInfo, ScoreTenancyFilters, ScoresStorage, StorageCloneThreadInput, StorageCloneThreadOutput, StorageDomains, StorageListMessagesInput, StorageListMessagesOutput, StorageListThreadsInput, StorageListThreadsOutput, StorageListWorkflowRunsInput, StoragePagination, StorageResourceType, UpdateWorkflowStateOptions, WorkflowRun, WorkflowRuns, WorkflowsStorage } from "@mastra/core/storage";
 import { CreateIndexParams, DeleteIndexParams, DeleteVectorParams, DeleteVectorsParams, DescribeIndexParams, IndexStats, MastraVector, QueryResult, QueryVectorParams, UpdateVectorParams, UpsertVectorParams } from "@mastra/core/vector";
 import { BaseFilterTranslator, BlacklistedRootOperators, LogicalOperatorValueMap, OperatorValueMap, VectorFilter } from "@mastra/core/vector/filter";
+import { MastraMessageContentV2 } from "@mastra/core/agent";
+import { SaveScorePayload, ScoreRowData, ScoringSource } from "@mastra/core/evals";
+import { MastraDBMessage, StorageThreadType } from "@mastra/core/memory";
+import { StepResult, WorkflowRunState } from "@mastra/core/workflows";
 //#region src/vector/filter.d.ts
 type ElasticSearchOperatorValueMap = Omit<OperatorValueMap, '$options' | '$nor' | '$elemMatch'>;
 type ElasticSearchLogicalOperatorValueMap = Omit<LogicalOperatorValueMap, '$nor'>;
@@ -138,5 +143,272 @@ declare class ElasticSearchVector extends MastraVector<ElasticSearchVectorFilter
   deleteVectors({ indexName, filter, ids }: DeleteVectorsParams<ElasticSearchVectorFilter>): Promise<void>;
 }
 //#endregion
-export { type ElasticSearchAuth, ElasticSearchVector, type ElasticSearchVectorConfig, type ElasticSearchVectorFilter };
+//#region src/storage/types.d.ts
+/**
+ * ElasticSearch storage configuration type.
+ *
+ * Accepts either:
+ * - A pre-configured ElasticSearch client: `{ id, client }`
+ * - Connection parameters: `{ id, url, auth? }`
+ *
+ * This mirrors the config surface of `ElasticSearchVector` so both can share
+ * the same connection settings (or the same client instance).
+ */
+type ElasticSearchConfig = {
+  id: string;
+  /**
+   * When true, automatic initialization (index creation) is disabled.
+   * You must call `storage.init()` explicitly before use.
+   */
+  disableInit?: boolean;
+} & ({
+  /**
+   * Pre-configured ElasticSearch client (from `@elastic/elasticsearch`).
+   *
+   * @example
+   * ```typescript
+   * import { Client } from '@elastic/elasticsearch';
+   *
+   * const client = new Client({ node: 'http://localhost:9200' });
+   * const store = new ElasticSearchStore({ id: 'my-store', client });
+   * ```
+   */
+  client: Client;
+  url?: never;
+  auth?: never;
+} | {
+  /**
+   * ElasticSearch node URL.
+   *
+   * @example
+   * ```typescript
+   * const store = new ElasticSearchStore({
+   *   id: 'my-store',
+   *   url: 'http://localhost:9200',
+   *   auth: { apiKey: '...' },
+   * });
+   * ```
+   */
+  url: string;
+  auth?: ElasticSearchAuth;
+  client?: never;
+});
+//#endregion
+//#region src/storage/store.d.ts
+/**
+ * ElasticSearch storage adapter for Mastra.
+ *
+ * Implements the memory, workflows, and scores storage domains on top of
+ * ElasticSearch. Shares the same connection config surface as
+ * `ElasticSearchVector`, so both can reuse one client or connection config.
+ *
+ * @example
+ * ```typescript
+ * // Using connection parameters
+ * const storage = new ElasticSearchStore({
+ *   id: 'my-store',
+ *   url: 'http://localhost:9200',
+ *   auth: { apiKey: '...' },
+ * });
+ *
+ * // Access memory domain
+ * const memory = await storage.getStore('memory');
+ * await memory?.saveThread({ thread });
+ * ```
+ *
+ * @example
+ * ```typescript
+ * // Using a pre-configured client shared with ElasticSearchVector
+ * import { Client } from '@elastic/elasticsearch';
+ *
+ * const client = new Client({ node: 'http://localhost:9200' });
+ * const storage = new ElasticSearchStore({ id: 'my-store', client });
+ * const vector = new ElasticSearchVector({ id: 'my-vector', client });
+ * ```
+ */
+declare class ElasticSearchStore extends MastraStorage {
+  private client;
+  private shouldManageConnection;
+  stores: StorageDomains;
+  constructor(config: ElasticSearchConfig);
+  getClient(): Client;
+  close(): Promise<void>;
+}
+//#endregion
+//#region src/storage/db.d.ts
+interface ElasticSearchDomainConfig {
+  client: Client;
+}
+//#endregion
+//#region src/storage/domains/memory/index.d.ts
+declare class MemoryElasticSearch extends MemoryStorage {
+  readonly supportsPartialThreadUpdate = true;
+  private db;
+  constructor(config: ElasticSearchDomainConfig);
+  dangerouslyClearAll(): Promise<void>;
+  getThreadById({ threadId, resourceId }: {
+    threadId: string;
+    resourceId?: string;
+  }): Promise<StorageThreadType | null>;
+  listThreadsByResourceId(args: StorageListThreadsInput): Promise<StorageListThreadsOutput>;
+  listThreads(args: StorageListThreadsInput): Promise<StorageListThreadsOutput>;
+  saveThread({ thread }: {
+    thread: StorageThreadType;
+  }): Promise<StorageThreadType>;
+  updateThread({ id, title, metadata }: {
+    id: string;
+    title?: string;
+    metadata?: Record<string, unknown>;
+  }): Promise<StorageThreadType>;
+  deleteThread({ threadId }: {
+    threadId: string;
+  }): Promise<void>;
+  saveMessages(args: {
+    messages: MastraDBMessage[];
+  }): Promise<{
+    messages: MastraDBMessage[];
+  }>;
+  /**
+   * Returns all messages that belong to a thread, sorted in insertion order
+   * (createdAt with `_index` tiebreaker).
+   */
+  private listThreadMessages;
+  /** Returns all message documents across all threads (excludes index docs). */
+  private listAllMessages;
+  private getThreadIdForMessage;
+  /**
+   * Fetches the messages named by `include` together with their surrounding context.
+   *
+   * @param include - Message ids to pin, each with an optional before/after window.
+   * @param resourceId - When set, drops any pinned or context message owned by another
+   * resource so an id from another resource returns nothing.
+   */
+  private getIncludedMessages;
+  private parseStoredMessage;
+  listMessagesById({ messageIds }: {
+    messageIds: string[];
+  }): Promise<{
+    messages: MastraDBMessage[];
+  }>;
+  listMessages(args: StorageListMessagesInput): Promise<StorageListMessagesOutput>;
+  getResourceById({ resourceId }: {
+    resourceId: string;
+  }): Promise<StorageResourceType | null>;
+  saveResource({ resource }: {
+    resource: StorageResourceType;
+  }): Promise<StorageResourceType>;
+  updateResource({ resourceId, workingMemory, metadata }: {
+    resourceId: string;
+    workingMemory?: string;
+    metadata?: Record<string, unknown>;
+  }): Promise<StorageResourceType>;
+  updateMessages(args: {
+    messages: (Partial<Omit<MastraDBMessage, 'createdAt'>> & {
+      id: string;
+      content?: {
+        metadata?: MastraMessageContentV2['metadata'];
+        content?: MastraMessageContentV2['content'];
+      };
+    })[];
+  }): Promise<MastraDBMessage[]>;
+  deleteMessages(messageIds: string[]): Promise<void>;
+  private sortThreads;
+  cloneThread(args: StorageCloneThreadInput): Promise<StorageCloneThreadOutput>;
+}
+//#endregion
+//#region src/storage/domains/workflows/index.d.ts
+declare class WorkflowsElasticSearch extends WorkflowsStorage {
+  private db;
+  constructor(config: ElasticSearchDomainConfig);
+  supportsConcurrentUpdates(): boolean;
+  dangerouslyClearAll(): Promise<void>;
+  updateWorkflowResults({ workflowName, runId, stepId, result, requestContext }: {
+    workflowName: string;
+    runId: string;
+    stepId: string;
+    result: StepResult<unknown, unknown, unknown, unknown>;
+    requestContext: Record<string, unknown>;
+  }): Promise<Record<string, StepResult<unknown, unknown, unknown, unknown>>>;
+  updateWorkflowState({ workflowName, runId, opts }: {
+    workflowName: string;
+    runId: string;
+    opts: UpdateWorkflowStateOptions;
+  }): Promise<WorkflowRunState | undefined>;
+  persistWorkflowSnapshot(params: {
+    namespace?: string;
+    workflowName: string;
+    runId: string;
+    resourceId?: string;
+    snapshot: WorkflowRunState;
+    createdAt?: Date;
+    updatedAt?: Date;
+  }): Promise<void>;
+  loadWorkflowSnapshot(params: {
+    namespace: string;
+    workflowName: string;
+    runId: string;
+  }): Promise<WorkflowRunState | null>;
+  getWorkflowRunById({ runId, workflowName }: {
+    runId: string;
+    workflowName?: string;
+  }): Promise<WorkflowRun | null>;
+  deleteWorkflowRunById({ runId, workflowName }: {
+    runId: string;
+    workflowName: string;
+  }): Promise<void>;
+  listWorkflowRuns({ workflowName, fromDate, toDate, perPage, page, resourceId, status }?: StorageListWorkflowRunsInput): Promise<WorkflowRuns>;
+}
+//#endregion
+//#region src/storage/domains/scores/index.d.ts
+declare class ScoresElasticSearch extends ScoresStorage {
+  private db;
+  constructor(config: ElasticSearchDomainConfig);
+  dangerouslyClearAll(): Promise<void>;
+  getScoreById({ id }: {
+    id: string;
+  }): Promise<ScoreRowData | null>;
+  listScoresByScorerId({ scorerId, entityId, entityType, source, pagination, filters }: {
+    scorerId: string;
+    entityId?: string;
+    entityType?: string;
+    source?: ScoringSource;
+    pagination?: StoragePagination;
+    filters?: ScoreTenancyFilters;
+  }): Promise<{
+    scores: ScoreRowData[];
+    pagination: PaginationInfo;
+  }>;
+  saveScore(score: SaveScorePayload): Promise<{
+    score: ScoreRowData;
+  }>;
+  listScoresByRunId({ runId, pagination, filters }: {
+    runId: string;
+    pagination?: StoragePagination;
+    filters?: ScoreTenancyFilters;
+  }): Promise<{
+    scores: ScoreRowData[];
+    pagination: PaginationInfo;
+  }>;
+  listScoresByEntityId({ entityId, entityType, pagination, filters }: {
+    entityId: string;
+    entityType?: string;
+    pagination?: StoragePagination;
+    filters?: ScoreTenancyFilters;
+  }): Promise<{
+    scores: ScoreRowData[];
+    pagination: PaginationInfo;
+  }>;
+  listScoresBySpan({ traceId, spanId, pagination, filters }: {
+    traceId: string;
+    spanId: string;
+    pagination?: StoragePagination;
+    filters?: ScoreTenancyFilters;
+  }): Promise<{
+    scores: ScoreRowData[];
+    pagination: PaginationInfo;
+  }>;
+  private fetchAndFilterScores;
+}
+//#endregion
+export { type ElasticSearchAuth, type ElasticSearchConfig, ElasticSearchStore, ElasticSearchVector, type ElasticSearchVectorConfig, type ElasticSearchVectorFilter, MemoryElasticSearch, ScoresElasticSearch, WorkflowsElasticSearch };
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","names":[],"sources":["../src/vector/filter.ts","../src/vector/index.ts"],"mappings":";;;;KAUK,gCAAgC,KAAK;KAErC,uCAAuC,KAAK;KAE5C,2BAA2B;KAEpB,4BAA4B,mBAChC,+BACN,+BACA,sCACA;;;KCaG,4BAA4B,kBAAkB;KAEvC;EAAsB;;EAAqB;EAAkB;;EAAuB;;KAEpF;EACN;EAAY,QAAQ;EAAqB;EAAa;;EACtD;EAAY;EAAa,OAAO;EAAmB;;cAE5C,4BAA4B,aAAa;UAC5C;;;;;;;;EASI,YAAA,QAAQ;;;;;;;;;EA6Bd,cAAc,WAAW,WAAW,UAAqB,oBAAoB;;;;;;EAkD7E,eAAe;;;;;YAwBL,sBAAsB,mBAAmB,mBAAmB,iBAAiB;;;;;;;EAoDvF,gBAAgB,aAAa,sBAAsB,QAAQ;;;;;;;EAqB3D,cAAc,aAAa,oBAAoB;;;;;;;;;;EA4B/C,SAAS,WAAW,SAAS,UAAe,OAAO,qBAAqB;;;;;;;;;;;EA+HxE,QACJ,WACA,aACA,QACA,MACA,iBACC,4BAA4B,QAAQ;;;;;;;;UA+D/B;;;;;;;UAYA;;;;;;;;;;;EAeF,aAAa,QAAQ,mBAAmB,6BAA6B;;;;UAwD7D;;;;UAuFA;;;;;;;;EAyDR,eAAe,WAAW,MAAM,qBAAqB;EA2BrD,gBAAgB,WAAW,QAAQ,OAAO,oBAAoB,6BAA6B"}
+{"version":3,"file":"index.d.ts","names":[],"sources":["../src/vector/filter.ts","../src/vector/index.ts","../src/storage/types.ts","../src/storage/store.ts","../src/storage/db.ts","../src/storage/domains/memory/index.ts","../src/storage/domains/workflows/index.ts","../src/storage/domains/scores/index.ts"],"mappings":";;;;;;;;;KAUK,gCAAgC,KAAK;KAErC,uCAAuC,KAAK;KAE5C,2BAA2B;KAEpB,4BAA4B,mBAChC,+BACN,+BACA,sCACA;;;KCaG,4BAA4B,kBAAkB;KAEvC;EAAsB;;EAAqB;EAAkB;;EAAuB;;KAEpF;EACN;EAAY,QAAQ;EAAqB;EAAa;;EACtD;EAAY;EAAa,OAAO;EAAmB;;cAE5C,4BAA4B,aAAa;UAC5C;;;;;;;;EASI,YAAA,QAAQ;;;;;;;;;EA6Bd,cAAc,WAAW,WAAW,UAAqB,oBAAoB;;;;;;EAkD7E,eAAe;;;;;YAwBL,sBAAsB,mBAAmB,mBAAmB,iBAAiB;;;;;;;EAoDvF,gBAAgB,aAAa,sBAAsB,QAAQ;;;;;;;EAqB3D,cAAc,aAAa,oBAAoB;;;;;;;;;;EA4B/C,SAAS,WAAW,SAAS,UAAe,OAAO,qBAAqB;;;;;;;;;;;EA+HxE,QACJ,WACA,aACA,QACA,MACA,iBACC,4BAA4B,QAAQ;;;;;;;;UA+D/B;;;;;;;UAYA;;;;;;;;;;;EAeF,aAAa,QAAQ,mBAAmB,6BAA6B;;;;UAwD7D;;;;UAuFA;;;;;;;;EAyDR,eAAe,WAAW,MAAM,qBAAqB;EA2BrD,gBAAgB,WAAW,QAAQ,OAAO,oBAAoB,6BAA6B;;;;;;;;;;;;;;KCnrBvF;EACV;;;;;EAKA;;;;;;;;;;;;;EAcI,QAAQ;EACR;EACA;;;;;;;;;;;;;;EAeA;EACA,OAAO;EACP;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cCXO,2BAA2B;UAC9B;UACA;EACD,QAAQ;EAEH,YAAA,QAAQ;EA8Bb,aAAa;EAIP,SAAS;;;;UCiNP;EACf,QAAQ;;;;cC9PG,4BAA4B;WACrB;UACV;EAEI,YAAA,QAAQ;EAKP,uBAAuB;EAMvB,gBACX,UACA;IAEA;IACA;MACE,QAAQ;EAgCC,wBAAwB,MAAM,0BAA0B,QAAQ;EAIhE,YAAY,MAAM,0BAA0B,QAAQ;EAqGpD,aAAa;IAAY,QAAQ;MAAsB,QAAQ;EAyB/D,eACX,IACA,OACA;IAEA;IACA;IACA,WAAW;MACT,QAAQ;EA0CC,eAAe;IAAc;MAAqB;EA4BlD,aAAa;IAAQ,UAAU;MAAsB;IAAU,UAAU;;;;;;UA4FxE;;UASA;UAOA;;;;;;;;UAiCA;UAqCN;EAUK,mBAAmB;IAAgB;MAAyB;IAAU,UAAU;;EA8DhF,aAAa,MAAM,2BAA2B,QAAQ;EAkMtD,kBAAkB;IAAgB;MAAuB,QAAQ;EAwBjE,eAAe;IAAc,UAAU;MAAwB,QAAQ;EAsBvE,iBACX,YACA,eACA;IAEA;IACA;IACA,WAAW;MACT,QAAQ;EAiCC,eAAe;IAC1B,WAAW,QAAQ,KAAK;MACtB;MACA;QAAY,WAAW;QAAoC,UAAU;;;MAErE,QAAQ;EA4IC,eAAe,uBAAuB;UAoD3C;EAYK,YAAY,MAAM,0BAA0B,QAAQ;;;;cC39BtD,+BAA+B;UAClC;EAEI,YAAA,QAAQ;EAKb;EAIM,uBAAuB;EAIvB,wBACX,cACA,OACA,QACA,QACA;IAEA;IACA;IACA;IACA,QAAQ;IACR,gBAAgB;MACd,QAAQ,eAAe;EA2Dd,sBACX,cACA,OACA;IAEA;IACA;IACA,MAAM;MACJ,QAAQ;EAmDC,wBAAwB;IACnC;IACA;IACA;IACA;IACA,UAAU;IACV,YAAY;IACZ,YAAY;MACV;EA6CS,qBAAqB;IAChC;IACA;IACA;MACE,QAAQ;EA6BC,qBACX,OACA;IAEA;IACA;MACE,QAAQ;EA2CC,wBAAwB,OAAO;IAAkB;IAAe;MAAyB;EAqBzF,mBACX,cACA,UACA,QACA,SACA,MACA,YACA,WACC,+BAAoC,QAAQ;;;;cChVpC,4BAA4B;UAC/B;EAEI,YAAA,QAAQ;EAKP,uBAAuB;EAIvB,eAAe;IAAQ;MAAe,QAAQ;EA2B9C,uBACX,UACA,UACA,YACA,QACA,YACA;IAEA;IACA;IACA;IACA,SAAS;IACT,aAAa;IACb,UAAU;MACR;IACF,QAAQ;IACR,YAAY;;EAsBD,UAAU,OAAO,mBAAmB;IAAU,OAAO;;EAiDrD,oBACX,OACA,YACA;IAEA;IACA,aAAa;IACb,UAAU;MACR;IACF,QAAQ;IACR,YAAY;;EAKD,uBACX,UACA,YACA,YACA;IAEA;IACA;IACA,aAAa;IACb,UAAU;MACR;IACF,QAAQ;IACR,YAAY;;EAgBD,mBACX,SACA,QACA,YACA;IAEA;IACA;IACA,aAAa;IACb,UAAU;MACR;IACF,QAAQ;IACR,YAAY;;UAQA"}