@mastra/memory 1.17.2 → 1.17.3-alpha.0

package/dist/index.js

@@ -6,8 +6,7 @@ import * as z4 from 'zod/v4';
 import { z as z$1 } from 'zod/v4';
 import { MessageList } from '@mastra/core/agent';
 import { coreFeatures } from '@mastra/core/features';
-import { MastraMemory, filterSystemReminderMessages, removeWorkingMemoryTags, extractWorkingMemoryContent } from '@mastra/core/memory';
-export { extractWorkingMemoryContent, extractWorkingMemoryTags, removeWorkingMemoryTags } from '@mastra/core/memory';
+import { MastraMemory } from '@mastra/core/memory';
 import { EntityType, SpanType } from '@mastra/core/observability';
 import { generateEmptyFromSchema } from '@mastra/core/utils';
 import { isStandardSchemaWithJSON, toStandardSchema, standardSchemaToJSONSchema } from '@mastra/schema-compat/schema';
@@ -4562,7 +4561,7 @@ function withUserAgentSuffix(headers, ...userAgentSuffixParts) {
   );
   return Object.fromEntries(normalizedHeaders.entries());
 }
-var VERSION2 = "3.0.22";
+var VERSION2 = "3.0.23";
 var getOriginalFetch = () => globalThis.fetch;
 var getFromApi = async ({
   url,
@@ -6609,6 +6608,7 @@ async function parseAuthMethod(headers) {
 var gatewayAuthMethodSchema = lazyValidator(
   () => zodSchema2(z$1.union([z$1.literal("api-key"), z$1.literal("oidc")]))
 );
+var KNOWN_MODEL_TYPES = ["embedding", "image", "language"];
 var GatewayFetchMetadata = class {
   constructor(config) {
     this.config = config;
@@ -6679,8 +6679,12 @@ var gatewayAvailableModelsResponseSchema = lazyValidator(
             provider: z$1.string(),
             modelId: z$1.string()
           }),
-          modelType: z$1.enum(["language", "embedding", "image"]).nullish()
+          modelType: z$1.string().nullish()
         })
+      ).transform(
+        (models) => models.filter(
+          (m) => m.modelType == null || KNOWN_MODEL_TYPES.includes(m.modelType)
+        )
       )
     })
   )
@@ -6696,6 +6700,187 @@ var gatewayCreditsResponseSchema = lazyValidator(
     }))
   )
 );
+var GatewaySpendReport = class {
+  constructor(config) {
+    this.config = config;
+  }
+  async getSpendReport(params) {
+    try {
+      const baseUrl = new URL(this.config.baseURL);
+      const searchParams = new URLSearchParams();
+      searchParams.set("start_date", params.startDate);
+      searchParams.set("end_date", params.endDate);
+      if (params.groupBy) {
+        searchParams.set("group_by", params.groupBy);
+      }
+      if (params.datePart) {
+        searchParams.set("date_part", params.datePart);
+      }
+      if (params.userId) {
+        searchParams.set("user_id", params.userId);
+      }
+      if (params.model) {
+        searchParams.set("model", params.model);
+      }
+      if (params.provider) {
+        searchParams.set("provider", params.provider);
+      }
+      if (params.credentialType) {
+        searchParams.set("credential_type", params.credentialType);
+      }
+      if (params.tags && params.tags.length > 0) {
+        searchParams.set("tags", params.tags.join(","));
+      }
+      const { value } = await getFromApi({
+        url: `${baseUrl.origin}/v1/report?${searchParams.toString()}`,
+        headers: await resolve(this.config.headers()),
+        successfulResponseHandler: createJsonResponseHandler(
+          gatewaySpendReportResponseSchema
+        ),
+        failedResponseHandler: createJsonErrorResponseHandler({
+          errorSchema: z$1.any(),
+          errorToMessage: (data) => data
+        }),
+        fetch: this.config.fetch
+      });
+      return value;
+    } catch (error) {
+      throw await asGatewayError(error);
+    }
+  }
+};
+var gatewaySpendReportResponseSchema = lazySchema(
+  () => zodSchema2(
+    z$1.object({
+      results: z$1.array(
+        z$1.object({
+          day: z$1.string().optional(),
+          hour: z$1.string().optional(),
+          user: z$1.string().optional(),
+          model: z$1.string().optional(),
+          tag: z$1.string().optional(),
+          provider: z$1.string().optional(),
+          credential_type: z$1.enum(["byok", "system"]).optional(),
+          total_cost: z$1.number(),
+          market_cost: z$1.number().optional(),
+          input_tokens: z$1.number().optional(),
+          output_tokens: z$1.number().optional(),
+          cached_input_tokens: z$1.number().optional(),
+          cache_creation_input_tokens: z$1.number().optional(),
+          reasoning_tokens: z$1.number().optional(),
+          request_count: z$1.number().optional()
+        }).transform(
+          ({
+            credential_type,
+            total_cost,
+            market_cost,
+            input_tokens,
+            output_tokens,
+            cached_input_tokens,
+            cache_creation_input_tokens,
+            reasoning_tokens,
+            request_count,
+            ...rest
+          }) => ({
+            ...rest,
+            ...credential_type !== void 0 ? { credentialType: credential_type } : {},
+            totalCost: total_cost,
+            ...market_cost !== void 0 ? { marketCost: market_cost } : {},
+            ...input_tokens !== void 0 ? { inputTokens: input_tokens } : {},
+            ...output_tokens !== void 0 ? { outputTokens: output_tokens } : {},
+            ...cached_input_tokens !== void 0 ? { cachedInputTokens: cached_input_tokens } : {},
+            ...cache_creation_input_tokens !== void 0 ? { cacheCreationInputTokens: cache_creation_input_tokens } : {},
+            ...reasoning_tokens !== void 0 ? { reasoningTokens: reasoning_tokens } : {},
+            ...request_count !== void 0 ? { requestCount: request_count } : {}
+          })
+        )
+      )
+    })
+  )
+);
+var GatewayGenerationInfoFetcher = class {
+  constructor(config) {
+    this.config = config;
+  }
+  async getGenerationInfo(params) {
+    try {
+      const baseUrl = new URL(this.config.baseURL);
+      const { value } = await getFromApi({
+        url: `${baseUrl.origin}/v1/generation?id=${encodeURIComponent(params.id)}`,
+        headers: await resolve(this.config.headers()),
+        successfulResponseHandler: createJsonResponseHandler(
+          gatewayGenerationInfoResponseSchema
+        ),
+        failedResponseHandler: createJsonErrorResponseHandler({
+          errorSchema: z$1.any(),
+          errorToMessage: (data) => data
+        }),
+        fetch: this.config.fetch
+      });
+      return value;
+    } catch (error) {
+      throw await asGatewayError(error);
+    }
+  }
+};
+var gatewayGenerationInfoResponseSchema = lazySchema(
+  () => zodSchema2(
+    z$1.object({
+      data: z$1.object({
+        id: z$1.string(),
+        total_cost: z$1.number(),
+        upstream_inference_cost: z$1.number(),
+        usage: z$1.number(),
+        created_at: z$1.string(),
+        model: z$1.string(),
+        is_byok: z$1.boolean(),
+        provider_name: z$1.string(),
+        streamed: z$1.boolean(),
+        finish_reason: z$1.string(),
+        latency: z$1.number(),
+        generation_time: z$1.number(),
+        native_tokens_prompt: z$1.number(),
+        native_tokens_completion: z$1.number(),
+        native_tokens_reasoning: z$1.number(),
+        native_tokens_cached: z$1.number(),
+        native_tokens_cache_creation: z$1.number(),
+        billable_web_search_calls: z$1.number()
+      }).transform(
+        ({
+          total_cost,
+          upstream_inference_cost,
+          created_at,
+          is_byok,
+          provider_name,
+          finish_reason,
+          generation_time,
+          native_tokens_prompt,
+          native_tokens_completion,
+          native_tokens_reasoning,
+          native_tokens_cached,
+          native_tokens_cache_creation,
+          billable_web_search_calls,
+          ...rest
+        }) => ({
+          ...rest,
+          totalCost: total_cost,
+          upstreamInferenceCost: upstream_inference_cost,
+          createdAt: created_at,
+          isByok: is_byok,
+          providerName: provider_name,
+          finishReason: finish_reason,
+          generationTime: generation_time,
+          promptTokens: native_tokens_prompt,
+          completionTokens: native_tokens_completion,
+          reasoningTokens: native_tokens_reasoning,
+          cachedTokens: native_tokens_cached,
+          cacheCreationTokens: native_tokens_cache_creation,
+          billableWebSearchCalls: billable_web_search_calls
+        })
+      )
+    }).transform(({ data }) => data)
+  )
+);
 var GatewayLanguageModel = class {
   constructor(modelId, config) {
     this.modelId = modelId;
@@ -7198,7 +7383,7 @@ async function getVercelRequestId() {
   var _a932;
   return (_a932 = getContext().headers) == null ? void 0 : _a932["x-vercel-id"];
 }
-var VERSION3 = "2.0.59";
+var VERSION3 = "2.0.82";
 var AI_GATEWAY_PROTOCOL_VERSION = "0.0.1";
 function createGatewayProvider(options = {}) {
   var _a932, _b93;
@@ -7296,6 +7481,30 @@ function createGatewayProvider(options = {}) {
       );
     });
   };
+  const getSpendReport = async (params) => {
+    return new GatewaySpendReport({
+      baseURL,
+      headers: getHeaders,
+      fetch: options.fetch
+    }).getSpendReport(params).catch(async (error) => {
+      throw await asGatewayError(
+        error,
+        await parseAuthMethod(await getHeaders())
+      );
+    });
+  };
+  const getGenerationInfo = async (params) => {
+    return new GatewayGenerationInfoFetcher({
+      baseURL,
+      headers: getHeaders,
+      fetch: options.fetch
+    }).getGenerationInfo(params).catch(async (error) => {
+      throw await asGatewayError(
+        error,
+        await parseAuthMethod(await getHeaders())
+      );
+    });
+  };
   const provider = function(modelId) {
     if (new.target) {
       throw new Error(
@@ -7306,6 +7515,8 @@ function createGatewayProvider(options = {}) {
   };
   provider.getAvailableModels = getAvailableModels;
   provider.getCredits = getCredits;
+  provider.getSpendReport = getSpendReport;
+  provider.getGenerationInfo = getGenerationInfo;
   provider.imageModel = (modelId) => {
     return new GatewayImageModel(modelId, {
       provider: "gateway",
@@ -8159,7 +8370,7 @@ function getGlobalProvider() {
   var _a163;
   return (_a163 = globalThis.AI_SDK_DEFAULT_PROVIDER) != null ? _a163 : gateway;
 }
-var VERSION32 = "5.0.155";
+var VERSION32 = "5.0.179";
 var dataContentSchema2 = z$1.union([
   z$1.string(),
   z$1.instanceof(Uint8Array),
@@ -9770,7 +9981,7 @@ function withUserAgentSuffix2(headers, ...userAgentSuffixParts) {
   );
   return Object.fromEntries(normalizedHeaders.entries());
 }
-var VERSION4 = "4.0.19";
+var VERSION4 = "4.0.23";
 var getOriginalFetch3 = () => globalThis.fetch;
 var getFromApi2 = async ({
   url,
@@ -11855,6 +12066,13 @@ async function parseAuthMethod2(headers) {
 var gatewayAuthMethodSchema2 = lazySchema2(
   () => zodSchema3(z$1.union([z$1.literal("api-key"), z$1.literal("oidc")]))
 );
+var KNOWN_MODEL_TYPES2 = [
+  "embedding",
+  "image",
+  "language",
+  "reranking",
+  "video"
+];
 var GatewayFetchMetadata2 = class {
   constructor(config) {
     this.config = config;
@@ -11925,8 +12143,12 @@ var gatewayAvailableModelsResponseSchema2 = lazySchema2(
             provider: z$1.string(),
             modelId: z$1.string()
           }),
-          modelType: z$1.enum(["embedding", "image", "language", "video"]).nullish()
+          modelType: z$1.string().nullish()
         })
+      ).transform(
+        (models) => models.filter(
+          (m) => m.modelType == null || KNOWN_MODEL_TYPES2.includes(m.modelType)
+        )
       )
     })
   )
@@ -11942,6 +12164,187 @@ var gatewayCreditsResponseSchema2 = lazySchema2(
     }))
   )
 );
+var GatewaySpendReport2 = class {
+  constructor(config) {
+    this.config = config;
+  }
+  async getSpendReport(params) {
+    try {
+      const baseUrl = new URL(this.config.baseURL);
+      const searchParams = new URLSearchParams();
+      searchParams.set("start_date", params.startDate);
+      searchParams.set("end_date", params.endDate);
+      if (params.groupBy) {
+        searchParams.set("group_by", params.groupBy);
+      }
+      if (params.datePart) {
+        searchParams.set("date_part", params.datePart);
+      }
+      if (params.userId) {
+        searchParams.set("user_id", params.userId);
+      }
+      if (params.model) {
+        searchParams.set("model", params.model);
+      }
+      if (params.provider) {
+        searchParams.set("provider", params.provider);
+      }
+      if (params.credentialType) {
+        searchParams.set("credential_type", params.credentialType);
+      }
+      if (params.tags && params.tags.length > 0) {
+        searchParams.set("tags", params.tags.join(","));
+      }
+      const { value } = await getFromApi2({
+        url: `${baseUrl.origin}/v1/report?${searchParams.toString()}`,
+        headers: await resolve2(this.config.headers()),
+        successfulResponseHandler: createJsonResponseHandler2(
+          gatewaySpendReportResponseSchema2
+        ),
+        failedResponseHandler: createJsonErrorResponseHandler2({
+          errorSchema: z$1.any(),
+          errorToMessage: (data) => data
+        }),
+        fetch: this.config.fetch
+      });
+      return value;
+    } catch (error) {
+      throw await asGatewayError2(error);
+    }
+  }
+};
+var gatewaySpendReportResponseSchema2 = lazySchema2(
+  () => zodSchema3(
+    z$1.object({
+      results: z$1.array(
+        z$1.object({
+          day: z$1.string().optional(),
+          hour: z$1.string().optional(),
+          user: z$1.string().optional(),
+          model: z$1.string().optional(),
+          tag: z$1.string().optional(),
+          provider: z$1.string().optional(),
+          credential_type: z$1.enum(["byok", "system"]).optional(),
+          total_cost: z$1.number(),
+          market_cost: z$1.number().optional(),
+          input_tokens: z$1.number().optional(),
+          output_tokens: z$1.number().optional(),
+          cached_input_tokens: z$1.number().optional(),
+          cache_creation_input_tokens: z$1.number().optional(),
+          reasoning_tokens: z$1.number().optional(),
+          request_count: z$1.number().optional()
+        }).transform(
+          ({
+            credential_type,
+            total_cost,
+            market_cost,
+            input_tokens,
+            output_tokens,
+            cached_input_tokens,
+            cache_creation_input_tokens,
+            reasoning_tokens,
+            request_count,
+            ...rest
+          }) => ({
+            ...rest,
+            ...credential_type !== void 0 ? { credentialType: credential_type } : {},
+            totalCost: total_cost,
+            ...market_cost !== void 0 ? { marketCost: market_cost } : {},
+            ...input_tokens !== void 0 ? { inputTokens: input_tokens } : {},
+            ...output_tokens !== void 0 ? { outputTokens: output_tokens } : {},
+            ...cached_input_tokens !== void 0 ? { cachedInputTokens: cached_input_tokens } : {},
+            ...cache_creation_input_tokens !== void 0 ? { cacheCreationInputTokens: cache_creation_input_tokens } : {},
+            ...reasoning_tokens !== void 0 ? { reasoningTokens: reasoning_tokens } : {},
+            ...request_count !== void 0 ? { requestCount: request_count } : {}
+          })
+        )
+      )
+    })
+  )
+);
+var GatewayGenerationInfoFetcher2 = class {
+  constructor(config) {
+    this.config = config;
+  }
+  async getGenerationInfo(params) {
+    try {
+      const baseUrl = new URL(this.config.baseURL);
+      const { value } = await getFromApi2({
+        url: `${baseUrl.origin}/v1/generation?id=${encodeURIComponent(params.id)}`,
+        headers: await resolve2(this.config.headers()),
+        successfulResponseHandler: createJsonResponseHandler2(
+          gatewayGenerationInfoResponseSchema2
+        ),
+        failedResponseHandler: createJsonErrorResponseHandler2({
+          errorSchema: z$1.any(),
+          errorToMessage: (data) => data
+        }),
+        fetch: this.config.fetch
+      });
+      return value;
+    } catch (error) {
+      throw await asGatewayError2(error);
+    }
+  }
+};
+var gatewayGenerationInfoResponseSchema2 = lazySchema2(
+  () => zodSchema3(
+    z$1.object({
+      data: z$1.object({
+        id: z$1.string(),
+        total_cost: z$1.number(),
+        upstream_inference_cost: z$1.number(),
+        usage: z$1.number(),
+        created_at: z$1.string(),
+        model: z$1.string(),
+        is_byok: z$1.boolean(),
+        provider_name: z$1.string(),
+        streamed: z$1.boolean(),
+        finish_reason: z$1.string(),
+        latency: z$1.number(),
+        generation_time: z$1.number(),
+        native_tokens_prompt: z$1.number(),
+        native_tokens_completion: z$1.number(),
+        native_tokens_reasoning: z$1.number(),
+        native_tokens_cached: z$1.number(),
+        native_tokens_cache_creation: z$1.number(),
+        billable_web_search_calls: z$1.number()
+      }).transform(
+        ({
+          total_cost,
+          upstream_inference_cost,
+          created_at,
+          is_byok,
+          provider_name,
+          finish_reason,
+          generation_time,
+          native_tokens_prompt,
+          native_tokens_completion,
+          native_tokens_reasoning,
+          native_tokens_cached,
+          native_tokens_cache_creation,
+          billable_web_search_calls,
+          ...rest
+        }) => ({
+          ...rest,
+          totalCost: total_cost,
+          upstreamInferenceCost: upstream_inference_cost,
+          createdAt: created_at,
+          isByok: is_byok,
+          providerName: provider_name,
+          finishReason: finish_reason,
+          generationTime: generation_time,
+          promptTokens: native_tokens_prompt,
+          completionTokens: native_tokens_completion,
+          reasoningTokens: native_tokens_reasoning,
+          cachedTokens: native_tokens_cached,
+          cacheCreationTokens: native_tokens_cache_creation,
+          billableWebSearchCalls: billable_web_search_calls
+        })
+      )
+    }).transform(({ data }) => data)
+  )
+);
 var GatewayLanguageModel2 = class {
   constructor(modelId, config) {
     this.modelId = modelId;
@@ -12489,6 +12892,86 @@ var gatewayVideoEventSchema = z$1.discriminatedUnion("type", [
     param: z$1.unknown().nullable()
   })
 ]);
+var GatewayRerankingModel = class {
+  constructor(modelId, config) {
+    this.modelId = modelId;
+    this.config = config;
+    this.specificationVersion = "v3";
+  }
+  get provider() {
+    return this.config.provider;
+  }
+  async doRerank({
+    documents,
+    query,
+    topN,
+    headers,
+    abortSignal,
+    providerOptions
+  }) {
+    const resolvedHeaders = await resolve2(this.config.headers());
+    try {
+      const {
+        responseHeaders,
+        value: responseBody,
+        rawValue
+      } = await postJsonToApi2({
+        url: this.getUrl(),
+        headers: combineHeaders2(
+          resolvedHeaders,
+          headers != null ? headers : {},
+          this.getModelConfigHeaders(),
+          await resolve2(this.config.o11yHeaders)
+        ),
+        body: {
+          documents,
+          query,
+          ...topN != null ? { topN } : {},
+          ...providerOptions ? { providerOptions } : {}
+        },
+        successfulResponseHandler: createJsonResponseHandler2(
+          gatewayRerankingResponseSchema
+        ),
+        failedResponseHandler: createJsonErrorResponseHandler2({
+          errorSchema: z$1.any(),
+          errorToMessage: (data) => data
+        }),
+        ...abortSignal && { abortSignal },
+        fetch: this.config.fetch
+      });
+      return {
+        ranking: responseBody.ranking,
+        providerMetadata: responseBody.providerMetadata,
+        response: { headers: responseHeaders, body: rawValue },
+        warnings: []
+      };
+    } catch (error) {
+      throw await asGatewayError2(error, await parseAuthMethod2(resolvedHeaders));
+    }
+  }
+  getUrl() {
+    return `${this.config.baseURL}/reranking-model`;
+  }
+  getModelConfigHeaders() {
+    return {
+      "ai-reranking-model-specification-version": "3",
+      "ai-model-id": this.modelId
+    };
+  }
+};
+var gatewayRerankingResponseSchema = lazySchema2(
+  () => zodSchema3(
+    z$1.object({
+      ranking: z$1.array(
+        z$1.object({
+          index: z$1.number(),
+          relevanceScore: z$1.number()
+        })
+      ),
+      providerMetadata: z$1.record(z$1.string(), z$1.record(z$1.string(), z$1.unknown())).optional()
+    })
+  )
+);
 var parallelSearchInputSchema2 = lazySchema2(
   () => zodSchema3(
     z.object({
@@ -12665,7 +13148,7 @@ async function getVercelRequestId2() {
   var _a932;
   return (_a932 = getContext2().headers) == null ? void 0 : _a932["x-vercel-id"];
 }
-var VERSION5 = "3.0.66";
+var VERSION5 = "3.0.104";
 var AI_GATEWAY_PROTOCOL_VERSION2 = "0.0.1";
 function createGatewayProvider2(options = {}) {
   var _a932, _b93;
@@ -12765,6 +13248,30 @@ function createGatewayProvider2(options = {}) {
       );
     });
   };
+  const getSpendReport = async (params) => {
+    return new GatewaySpendReport2({
+      baseURL,
+      headers: getHeaders,
+      fetch: options.fetch
+    }).getSpendReport(params).catch(async (error) => {
+      throw await asGatewayError2(
+        error,
+        await parseAuthMethod2(await getHeaders())
+      );
+    });
+  };
+  const getGenerationInfo = async (params) => {
+    return new GatewayGenerationInfoFetcher2({
+      baseURL,
+      headers: getHeaders,
+      fetch: options.fetch
+    }).getGenerationInfo(params).catch(async (error) => {
+      throw await asGatewayError2(
+        error,
+        await parseAuthMethod2(await getHeaders())
+      );
+    });
+  };
   const provider = function(modelId) {
     if (new.target) {
       throw new Error(
@@ -12776,6 +13283,8 @@ function createGatewayProvider2(options = {}) {
   provider.specificationVersion = "v3";
   provider.getAvailableModels = getAvailableModels;
   provider.getCredits = getCredits;
+  provider.getSpendReport = getSpendReport;
+  provider.getGenerationInfo = getGenerationInfo;
   provider.imageModel = (modelId) => {
     return new GatewayImageModel2(modelId, {
       provider: "gateway",
@@ -12806,6 +13315,17 @@ function createGatewayProvider2(options = {}) {
       o11yHeaders: createO11yHeaders()
     });
   };
+  const createRerankingModel = (modelId) => {
+    return new GatewayRerankingModel(modelId, {
+      provider: "gateway",
+      baseURL,
+      headers: getHeaders,
+      fetch: options.fetch,
+      o11yHeaders: createO11yHeaders()
+    });
+  };
+  provider.rerankingModel = createRerankingModel;
+  provider.reranking = createRerankingModel;
   provider.chat = provider.languageModel;
   provider.embedding = provider.embeddingModel;
   provider.image = provider.imageModel;
@@ -13738,7 +14258,7 @@ function getTotalTimeoutMs(timeout) {
   }
   return timeout.totalMs;
 }
-var VERSION33 = "6.0.116";
+var VERSION33 = "6.0.168";
 var dataContentSchema3 = z$1.union([
   z$1.string(),
   z$1.instanceof(Uint8Array),
@@ -16440,6 +16960,69 @@ var __experimental_updateWorkingMemoryToolVNext = (config) => {
     }
   });
 };
+var WORKING_MEMORY_START_TAG = "<working_memory>";
+var WORKING_MEMORY_END_TAG = "</working_memory>";
+var LEGACY_SYSTEM_REMINDER_METADATA_KEY = "dynamicAgentsMdReminder";
+function isRecord(value) {
+  return typeof value === "object" && value !== null;
+}
+function extractWorkingMemoryTags(text4) {
+  const results = [];
+  let pos = 0;
+  while (pos < text4.length) {
+    const start = text4.indexOf(WORKING_MEMORY_START_TAG, pos);
+    if (start === -1) break;
+    const end = text4.indexOf(WORKING_MEMORY_END_TAG, start + WORKING_MEMORY_START_TAG.length);
+    if (end === -1) break;
+    results.push(text4.substring(start, end + WORKING_MEMORY_END_TAG.length));
+    pos = end + WORKING_MEMORY_END_TAG.length;
+  }
+  return results.length > 0 ? results : null;
+}
+function removeWorkingMemoryTags(text4) {
+  let result = "";
+  let pos = 0;
+  while (pos < text4.length) {
+    const start = text4.indexOf(WORKING_MEMORY_START_TAG, pos);
+    if (start === -1) {
+      result += text4.substring(pos);
+      break;
+    }
+    result += text4.substring(pos, start);
+    const end = text4.indexOf(WORKING_MEMORY_END_TAG, start + WORKING_MEMORY_START_TAG.length);
+    if (end === -1) {
+      result += text4.substring(start);
+      break;
+    }
+    pos = end + WORKING_MEMORY_END_TAG.length;
+  }
+  return result;
+}
+function extractWorkingMemoryContent(text4) {
+  const start = text4.indexOf(WORKING_MEMORY_START_TAG);
+  if (start === -1) return null;
+  const contentStart = start + WORKING_MEMORY_START_TAG.length;
+  const end = text4.indexOf(WORKING_MEMORY_END_TAG, contentStart);
+  if (end === -1) return null;
+  return text4.substring(contentStart, end);
+}
+function isSystemReminderMessage(message) {
+  if (message.role !== "user" || !isRecord(message.content)) {
+    return false;
+  }
+  const metadata = message.content.metadata;
+  if (isRecord(metadata) && (isRecord(metadata.systemReminder) || LEGACY_SYSTEM_REMINDER_METADATA_KEY in metadata)) {
+    return true;
+  }
+  const firstTextPart = message.content.parts.find((part) => part.type === "text");
+  return typeof firstTextPart?.text === "string" && firstTextPart.text.startsWith("<system-reminder");
+}
+function filterSystemReminderMessages(messages, includeSystemReminders) {
+  if (includeSystemReminders) {
+    return messages;
+  }
+  return messages.filter((message) => !isSystemReminderMessage(message));
+}
 function normalizeObservationalMemoryConfig(config) {
   if (config === true) return { model: "google/gemini-2.5-flash" };
   if (config === false || config === void 0) return void 0;
@@ -18277,6 +18860,6 @@ Notes:
   }
 };
 
-export { Memory, deepMergeWorkingMemory };
+export { Memory, deepMergeWorkingMemory, extractWorkingMemoryContent, extractWorkingMemoryTags, removeWorkingMemoryTags };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map