diff --git a/openmetadata-spec/src/main/resources/json/schema/configuration/aiPlatformConfiguration.json b/openmetadata-spec/src/main/resources/json/schema/configuration/aiPlatformConfiguration.json
index 61335f37062a..0f9b95df4fa6 100644
--- a/openmetadata-spec/src/main/resources/json/schema/configuration/aiPlatformConfiguration.json
+++ b/openmetadata-spec/src/main/resources/json/schema/configuration/aiPlatformConfiguration.json
@@ -28,6 +28,26 @@
         }
       },
       "required": ["port"]
+    },
+    "contextMemoryConfiguration": {
+      "description": "Tuning for the Context Memory T0 (user preferences) block shipped to the AI Platform via the gRPC user_memory_context field.",
+      "javaType": "org.openmetadata.schema.configuration.ContextMemoryConfiguration",
+      "type": "object",
+      "properties": {
+        "tokenBudget": {
+          "description": "Maximum number of tokens worth of T0 user-preference memories to include. Memories are ranked by freshness + usage and filled in order until the budget or item cap is reached.",
+          "type": "integer",
+          "default": 500,
+          "minimum": 0
+        },
+        "maxItems": {
+          "description": "Maximum number of T0 user-preference memories to include regardless of token budget.",
+          "type": "integer",
+          "default": 5,
+          "minimum": 0
+        }
+      },
+      "additionalProperties": false
     }
   },
   "properties": {
@@ -59,6 +79,10 @@
     "grpc": {
       "description": "gRPC configuration for the AI Platform server",
       "$ref": "#/definitions/grpcConfiguration"
+    },
+    "contextMemory": {
+      "description": "Context Memory tuning for the T0 user-preference block assembled by Collate and shipped to the AI Platform.",
+      "$ref": "#/definitions/contextMemoryConfiguration"
     }
   },
   "required": ["host", "port", "grpc"],
diff --git a/openmetadata-ui/src/main/resources/ui/src/generated/configuration/aiPlatformConfiguration.ts b/openmetadata-ui/src/main/resources/ui/src/generated/configuration/aiPlatformConfiguration.ts
index 4fd103fdf835..2f6a178db988 100644
--- a/openmetadata-ui/src/main/resources/ui/src/generated/configuration/aiPlatformConfiguration.ts
+++ b/openmetadata-ui/src/main/resources/ui/src/generated/configuration/aiPlatformConfiguration.ts
@@ -14,6 +14,11 @@
  * This schema defines the JWT Configuration.
  */
 export interface AIPlatformConfiguration {
+    /**
+     * Context Memory tuning for the T0 user-preference block assembled by Collate and shipped
+     * to the AI Platform.
+     */
+    contextMemory?: ContextMemoryConfiguration;
     /**
      * Indicates whether the AI Platform is enabled
      */
@@ -44,6 +49,25 @@ export interface AIPlatformConfiguration {
     trustedCertsPath?: string;
 }
 
+/**
+ * Context Memory tuning for the T0 user-preference block assembled by Collate and shipped
+ * to the AI Platform.
+ *
+ * Tuning for the Context Memory T0 (user preferences) block shipped to the AI Platform via
+ * the gRPC user_memory_context field.
+ */
+export interface ContextMemoryConfiguration {
+    /**
+     * Maximum number of T0 user-preference memories to include regardless of token budget.
+     */
+    maxItems?: number;
+    /**
+     * Maximum number of tokens worth of T0 user-preference memories to include. Memories are
+     * ranked by freshness + usage and filled in order until the budget or item cap is reached.
+     */
+    tokenBudget?: number;
+}
+
 /**
  * gRPC configuration for the AI Platform server
  */