🛂 feat: Payload limits and Validation for User-created Memories (#8974)

packages/api/src/index.ts

@@ -15,6 +15,8 @@ export * from './crypto';
 export * from './flow/manager';
 /* Middleware */
 export * from './middleware';
+/* Memory */
+export * from './memory';
 /* Agents */
 export * from './agents';
 /* Endpoints */

packages/api/src/memory/config.ts

@@ -0,0 +1,28 @@
+import { memorySchema } from 'librechat-data-provider';
+import type { TCustomConfig, TMemoryConfig } from 'librechat-data-provider';
+
+const hasValidAgent = (agent: TMemoryConfig['agent']) =>
+  !!agent &&
+  (('id' in agent && !!agent.id) ||
+    ('provider' in agent && 'model' in agent && !!agent.provider && !!agent.model));
+
+const isDisabled = (config?: TMemoryConfig | TCustomConfig['memory']) =>
+  !config || config.disabled === true;
+
+export function loadMemoryConfig(config: TCustomConfig['memory']): TMemoryConfig | undefined {
+  if (!config) return undefined;
+  if (isDisabled(config)) return config as TMemoryConfig;
+
+  if (!hasValidAgent(config.agent)) {
+    return { ...config, disabled: true } as TMemoryConfig;
+  }
+
+  const charLimit = memorySchema.shape.charLimit.safeParse(config.charLimit).data ?? 10000;
+
+  return { ...config, charLimit };
+}
+
+export function isMemoryEnabled(config: TMemoryConfig | undefined): boolean {
+  if (isDisabled(config)) return false;
+  return hasValidAgent(config!.agent);
+}

packages/api/src/memory/index.ts

@@ -0,0 +1 @@
+export * from './config';

packages/data-provider/src/config.ts

@@ -727,6 +727,7 @@ export const memorySchema = z.object({
   disabled: z.boolean().optional(),
   validKeys: z.array(z.string()).optional(),
   tokenLimit: z.number().optional(),
+  charLimit: z.number().optional().default(10000),
   personalize: z.boolean().default(true),
   messageWindowSize: z.number().optional().default(5),
   agent: z

packages/data-provider/src/index.ts

@@ -13,8 +13,6 @@ export * from './generate';
 export * from './models';
 /* mcp */
 export * from './mcp';
-/* memory */
-export * from './memory';
 /* RBAC */
 export * from './permissions';
 export * from './roles';

packages/data-provider/src/memory.ts

@@ -1,62 +0,0 @@
-import type { TCustomConfig, TMemoryConfig } from './config';
-
-/**
- * Loads the memory configuration and validates it
- * @param config - The memory configuration from librechat.yaml
- * @returns The validated memory configuration
- */
-export function loadMemoryConfig(config: TCustomConfig['memory']): TMemoryConfig | undefined {
-  if (!config) {
-    return undefined;
-  }
-
-  // If disabled is explicitly true, return the config as-is
-  if (config.disabled === true) {
-    return config;
-  }
-
-  // Check if the agent configuration is valid
-  const hasValidAgent =
-    config.agent &&
-    (('id' in config.agent && !!config.agent.id) ||
-      ('provider' in config.agent &&
-        'model' in config.agent &&
-        !!config.agent.provider &&
-        !!config.agent.model));
-
-  // If agent config is invalid, treat as disabled
-  if (!hasValidAgent) {
-    return {
-      ...config,
-      disabled: true,
-    };
-  }
-
-  return config;
-}
-
-/**
- * Checks if memory feature is enabled based on the configuration
- * @param config - The memory configuration
- * @returns True if memory is enabled, false otherwise
- */
-export function isMemoryEnabled(config: TMemoryConfig | undefined): boolean {
-  if (!config) {
-    return false;
-  }
-
-  if (config.disabled === true) {
-    return false;
-  }
-
-  // Check if agent configuration is valid
-  const hasValidAgent =
-    config.agent &&
-    (('id' in config.agent && !!config.agent.id) ||
-      ('provider' in config.agent &&
-        'model' in config.agent &&
-        !!config.agent.provider &&
-        !!config.agent.model));
-
-  return !!hasValidAgent;
-}