BREAKING_CHANGE(agentplatform): Migrate Memory Bank methods to client.memory_banks sub-module.

src/main/java/com/google/cloud/agentplatform/AgentEngines.java

@@ -55,15 +55,13 @@
 public final class AgentEngines {
   public final Sessions sessions;
   public final Sandboxes sandboxes;
-  public final Memories memories;
 
   final ApiClient apiClient;
 
   public AgentEngines(ApiClient apiClient) {
     this.apiClient = apiClient;
     this.sessions = new Sessions(apiClient);
     this.sandboxes = new Sandboxes(apiClient);
-    this.memories = new Memories(apiClient);
   }
 
   @ExcludeFromGeneratedCoverageReport

src/main/java/com/google/cloud/agentplatform/AsyncAgentEngines.java

@@ -39,7 +39,6 @@
 public final class AsyncAgentEngines {
   public final AsyncSessions sessions;
   public final AsyncSandboxes sandboxes;
-  public final AsyncMemories memories;
 
   AgentEngines agentEngines;
   ApiClient apiClient;
@@ -50,7 +49,6 @@ public AsyncAgentEngines(ApiClient apiClient) {
 
     this.sessions = new AsyncSessions(apiClient);
     this.sandboxes = new AsyncSandboxes(apiClient);
-    this.memories = new AsyncMemories(apiClient);
   }
 
   CompletableFuture<AgentEngineOperation> privateCreate(CreateAgentEngineConfig config) {

src/main/java/com/google/cloud/agentplatform/AsyncMemories.java

@@ -18,36 +18,32 @@
 
 package com.google.cloud.agentplatform;
 
-import com.google.cloud.agentplatform.types.AgentEngineGenerateMemoriesOperation;
-import com.google.cloud.agentplatform.types.AgentEngineMemoryConfig;
-import com.google.cloud.agentplatform.types.AgentEngineMemoryOperation;
-import com.google.cloud.agentplatform.types.AgentEnginePurgeMemoriesOperation;
-import com.google.cloud.agentplatform.types.AgentEngineRollbackMemoryOperation;
-import com.google.cloud.agentplatform.types.DeleteAgentEngineMemoryConfig;
-import com.google.cloud.agentplatform.types.DeleteAgentEngineMemoryOperation;
-import com.google.cloud.agentplatform.types.GenerateAgentEngineMemoriesConfig;
+import com.google.cloud.agentplatform.types.DeleteMemoryConfig;
+import com.google.cloud.agentplatform.types.DeleteMemoryOperation;
+import com.google.cloud.agentplatform.types.GenerateMemoriesConfig;
+import com.google.cloud.agentplatform.types.GenerateMemoriesOperation;
 import com.google.cloud.agentplatform.types.GenerateMemoriesRequestDirectContentsSource;
 import com.google.cloud.agentplatform.types.GenerateMemoriesRequestDirectMemoriesSource;
 import com.google.cloud.agentplatform.types.GenerateMemoriesRequestVertexSessionSource;
-import com.google.cloud.agentplatform.types.GetAgentEngineMemoryConfig;
-import com.google.cloud.agentplatform.types.GetAgentEngineOperationConfig;
-import com.google.cloud.agentplatform.types.IngestEventsConfig;
-import com.google.cloud.agentplatform.types.IngestionDirectContentsSource;
-import com.google.cloud.agentplatform.types.ListAgentEngineMemoryConfig;
-import com.google.cloud.agentplatform.types.ListReasoningEnginesMemoriesResponse;
+import com.google.cloud.agentplatform.types.GetMemoryBankOperationConfig;
+import com.google.cloud.agentplatform.types.GetMemoryConfig;
+import com.google.cloud.agentplatform.types.ListMemoriesConfig;
+import com.google.cloud.agentplatform.types.ListMemoriesResponse;
 import com.google.cloud.agentplatform.types.Memory;
-import com.google.cloud.agentplatform.types.MemoryBankIngestEventsOperation;
+import com.google.cloud.agentplatform.types.MemoryConfig;
 import com.google.cloud.agentplatform.types.MemoryConjunctionFilter;
-import com.google.cloud.agentplatform.types.MemoryGenerationTriggerConfig;
-import com.google.cloud.agentplatform.types.PurgeAgentEngineMemoriesConfig;
-import com.google.cloud.agentplatform.types.RetrieveAgentEngineMemoriesConfig;
+import com.google.cloud.agentplatform.types.MemoryOperation;
+import com.google.cloud.agentplatform.types.PurgeMemoriesConfig;
+import com.google.cloud.agentplatform.types.PurgeMemoriesOperation;
+import com.google.cloud.agentplatform.types.RetrieveMemoriesConfig;
 import com.google.cloud.agentplatform.types.RetrieveMemoriesRequestSimilaritySearchParams;
 import com.google.cloud.agentplatform.types.RetrieveMemoriesRequestSimpleRetrievalParams;
 import com.google.cloud.agentplatform.types.RetrieveMemoriesResponse;
 import com.google.cloud.agentplatform.types.RetrieveMemoryProfilesConfig;
 import com.google.cloud.agentplatform.types.RetrieveProfilesResponse;
-import com.google.cloud.agentplatform.types.RollbackAgentEngineMemoryConfig;
-import com.google.cloud.agentplatform.types.UpdateAgentEngineMemoryConfig;
+import com.google.cloud.agentplatform.types.RollbackMemoryConfig;
+import com.google.cloud.agentplatform.types.RollbackMemoryOperation;
+import com.google.cloud.agentplatform.types.UpdateMemoryConfig;
 import com.google.genai.ApiClient;
 import com.google.genai.ApiResponse;
 import com.google.genai.Common.BuiltRequest;
@@ -69,8 +65,8 @@ public AsyncMemories(ApiClient apiClient) {
     this.revisions = new AsyncMemoryRevisions(apiClient);
   }
 
-  CompletableFuture<AgentEngineMemoryOperation> privateCreate(
-      String name, String fact, Map<String, String> scope, AgentEngineMemoryConfig config) {
+  CompletableFuture<MemoryOperation> privateCreate(
+      String name, String fact, Map<String, String> scope, MemoryConfig config) {
 
     BuiltRequest builtRequest = memories.buildRequestForPrivateCreate(name, fact, scope, config);
     return this.apiClient
@@ -83,8 +79,7 @@ CompletableFuture<AgentEngineMemoryOperation> privateCreate(
             });
   }
 
-  public CompletableFuture<DeleteAgentEngineMemoryOperation> delete(
-      String name, DeleteAgentEngineMemoryConfig config) {
+  public CompletableFuture<DeleteMemoryOperation> delete(String name, DeleteMemoryConfig config) {
 
     BuiltRequest builtRequest = memories.buildRequestForDelete(name, config);
     return this.apiClient
@@ -98,13 +93,13 @@ public CompletableFuture<DeleteAgentEngineMemoryOperation> delete(
             });
   }
 
-  CompletableFuture<AgentEngineGenerateMemoriesOperation> privateGenerate(
+  CompletableFuture<GenerateMemoriesOperation> privateGenerate(
       String name,
       GenerateMemoriesRequestVertexSessionSource vertexSessionSource,
       GenerateMemoriesRequestDirectContentsSource directContentsSource,
       GenerateMemoriesRequestDirectMemoriesSource directMemoriesSource,
       Map<String, String> scope,
-      GenerateAgentEngineMemoriesConfig config) {
+      GenerateMemoriesConfig config) {
 
     BuiltRequest builtRequest =
         memories.buildRequestForPrivateGenerate(
@@ -119,7 +114,7 @@ CompletableFuture<AgentEngineGenerateMemoriesOperation> privateGenerate(
             });
   }
 
-  public CompletableFuture<Memory> get(String name, GetAgentEngineMemoryConfig config) {
+  public CompletableFuture<Memory> get(String name, GetMemoryConfig config) {
 
     BuiltRequest builtRequest = memories.buildRequestForGet(name, config);
     return this.apiClient
@@ -132,29 +127,7 @@ public CompletableFuture<Memory> get(String name, GetAgentEngineMemoryConfig con
             });
   }
 
-  CompletableFuture<MemoryBankIngestEventsOperation> privateIngestEvents(
-      String name,
-      String streamId,
-      IngestionDirectContentsSource directContentsSource,
-      Map<String, String> scope,
-      MemoryGenerationTriggerConfig generationTriggerConfig,
-      IngestEventsConfig config) {
-
-    BuiltRequest builtRequest =
-        memories.buildRequestForPrivateIngestEvents(
-            name, streamId, directContentsSource, scope, generationTriggerConfig, config);
-    return this.apiClient
-        .asyncRequest("post", builtRequest.path(), builtRequest.body(), builtRequest.httpOptions())
-        .thenApplyAsync(
-            response -> {
-              try (ApiResponse res = response) {
-                return memories.processResponseForPrivateIngestEvents(res, config);
-              }
-            });
-  }
-
-  CompletableFuture<ListReasoningEnginesMemoriesResponse> privateList(
-      String name, ListAgentEngineMemoryConfig config) {
+  CompletableFuture<ListMemoriesResponse> privateList(String name, ListMemoriesConfig config) {
 
     BuiltRequest builtRequest = memories.buildRequestForPrivateList(name, config);
     return this.apiClient
@@ -167,8 +140,8 @@ CompletableFuture<ListReasoningEnginesMemoriesResponse> privateList(
             });
   }
 
-  CompletableFuture<AgentEngineMemoryOperation> privateGetMemoryOperation(
-      String operationName, GetAgentEngineOperationConfig config) {
+  CompletableFuture<MemoryOperation> privateGetMemoryOperation(
+      String operationName, GetMemoryBankOperationConfig config) {
 
     BuiltRequest builtRequest =
         memories.buildRequestForPrivateGetMemoryOperation(operationName, config);
@@ -182,8 +155,8 @@ CompletableFuture<AgentEngineMemoryOperation> privateGetMemoryOperation(
             });
   }
 
-  CompletableFuture<AgentEngineGenerateMemoriesOperation> privateGetGenerateMemoriesOperation(
-      String operationName, GetAgentEngineOperationConfig config) {
+  CompletableFuture<GenerateMemoriesOperation> privateGetGenerateMemoriesOperation(
+      String operationName, GetMemoryBankOperationConfig config) {
 
     BuiltRequest builtRequest =
         memories.buildRequestForPrivateGetGenerateMemoriesOperation(operationName, config);
@@ -202,7 +175,7 @@ CompletableFuture<RetrieveMemoriesResponse> privateRetrieve(
       Map<String, String> scope,
       RetrieveMemoriesRequestSimilaritySearchParams similaritySearchParams,
       RetrieveMemoriesRequestSimpleRetrievalParams simpleRetrievalParams,
-      RetrieveAgentEngineMemoriesConfig config) {
+      RetrieveMemoriesConfig config) {
 
     BuiltRequest builtRequest =
         memories.buildRequestForPrivateRetrieve(
@@ -231,8 +204,8 @@ public CompletableFuture<RetrieveProfilesResponse> retrieveProfiles(
             });
   }
 
-  CompletableFuture<AgentEngineRollbackMemoryOperation> privateRollback(
-      String name, String targetRevisionId, RollbackAgentEngineMemoryConfig config) {
+  CompletableFuture<RollbackMemoryOperation> privateRollback(
+      String name, String targetRevisionId, RollbackMemoryConfig config) {
 
     BuiltRequest builtRequest =
         memories.buildRequestForPrivateRollback(name, targetRevisionId, config);
@@ -246,8 +219,8 @@ CompletableFuture<AgentEngineRollbackMemoryOperation> privateRollback(
             });
   }
 
-  CompletableFuture<AgentEngineMemoryOperation> privateUpdate(
-      String name, String fact, Map<String, String> scope, UpdateAgentEngineMemoryConfig config) {
+  CompletableFuture<MemoryOperation> privateUpdate(
+      String name, String fact, Map<String, String> scope, UpdateMemoryConfig config) {
 
     BuiltRequest builtRequest = memories.buildRequestForPrivateUpdate(name, fact, scope, config);
     return this.apiClient
@@ -260,12 +233,12 @@ CompletableFuture<AgentEngineMemoryOperation> privateUpdate(
             });
   }
 
-  CompletableFuture<AgentEnginePurgeMemoriesOperation> privatePurge(
+  CompletableFuture<PurgeMemoriesOperation> privatePurge(
       String name,
       String filter,
       List<MemoryConjunctionFilter> filterGroups,
       boolean force,
-      PurgeAgentEngineMemoriesConfig config) {
+      PurgeMemoriesConfig config) {
 
     BuiltRequest builtRequest =
         memories.buildRequestForPrivatePurge(name, filter, filterGroups, force, config);

src/main/java/com/google/cloud/agentplatform/AsyncMemoryBanks.java

@@ -0,0 +1,113 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+// Auto-generated code. Do not edit.
+
+package com.google.cloud.agentplatform;
+
+import com.google.cloud.agentplatform.types.CreateMemoryBankConfig;
+import com.google.cloud.agentplatform.types.DeleteMemoryBankConfig;
+import com.google.cloud.agentplatform.types.DeleteMemoryBankOperation;
+import com.google.cloud.agentplatform.types.GetMemoryBankOperationConfig;
+import com.google.cloud.agentplatform.types.IngestEventsConfig;
+import com.google.cloud.agentplatform.types.IngestionDirectContentsSource;
+import com.google.cloud.agentplatform.types.MemoryBankIngestEventsOperation;
+import com.google.cloud.agentplatform.types.MemoryBankOperation;
+import com.google.cloud.agentplatform.types.MemoryGenerationTriggerConfig;
+import com.google.genai.ApiClient;
+import com.google.genai.ApiResponse;
+import com.google.genai.Common.BuiltRequest;
+import java.util.Map;
+import java.util.concurrent.CompletableFuture;
+
+/** Async module of {@link MemoryBanks} */
+public final class AsyncMemoryBanks {
+  public final AsyncMemories memories;
+
+  MemoryBanks memoryBanks;
+  ApiClient apiClient;
+
+  public AsyncMemoryBanks(ApiClient apiClient) {
+    this.apiClient = apiClient;
+    this.memoryBanks = new MemoryBanks(apiClient);
+
+    this.memories = new AsyncMemories(apiClient);
+  }
+
+  CompletableFuture<MemoryBankOperation> privateCreate(CreateMemoryBankConfig config) {
+
+    BuiltRequest builtRequest = memoryBanks.buildRequestForPrivateCreate(config);
+    return this.apiClient
+        .asyncRequest("post", builtRequest.path(), builtRequest.body(), builtRequest.httpOptions())
+        .thenApplyAsync(
+            response -> {
+              try (ApiResponse res = response) {
+                return memoryBanks.processResponseForPrivateCreate(res, config);
+              }
+            });
+  }
+
+  CompletableFuture<DeleteMemoryBankOperation> privateDelete(
+      String name, boolean force, DeleteMemoryBankConfig config) {
+
+    BuiltRequest builtRequest = memoryBanks.buildRequestForPrivateDelete(name, force, config);
+    return this.apiClient
+        .asyncRequest(
+            "delete", builtRequest.path(), builtRequest.body(), builtRequest.httpOptions())
+        .thenApplyAsync(
+            response -> {
+              try (ApiResponse res = response) {
+                return memoryBanks.processResponseForPrivateDelete(res, config);
+              }
+            });
+  }
+
+  CompletableFuture<MemoryBankIngestEventsOperation> privateIngestEvents(
+      String name,
+      String streamId,
+      IngestionDirectContentsSource directContentsSource,
+      Map<String, String> scope,
+      MemoryGenerationTriggerConfig generationTriggerConfig,
+      IngestEventsConfig config) {
+
+    BuiltRequest builtRequest =
+        memoryBanks.buildRequestForPrivateIngestEvents(
+            name, streamId, directContentsSource, scope, generationTriggerConfig, config);
+    return this.apiClient
+        .asyncRequest("post", builtRequest.path(), builtRequest.body(), builtRequest.httpOptions())
+        .thenApplyAsync(
+            response -> {
+              try (ApiResponse res = response) {
+                return memoryBanks.processResponseForPrivateIngestEvents(res, config);
+              }
+            });
+  }
+
+  CompletableFuture<MemoryBankOperation> privateGetMemoryBankOperation(
+      String operationName, GetMemoryBankOperationConfig config) {
+
+    BuiltRequest builtRequest =
+        memoryBanks.buildRequestForPrivateGetMemoryBankOperation(operationName, config);
+    return this.apiClient
+        .asyncRequest("get", builtRequest.path(), builtRequest.body(), builtRequest.httpOptions())
+        .thenApplyAsync(
+            response -> {
+              try (ApiResponse res = response) {
+                return memoryBanks.processResponseForPrivateGetMemoryBankOperation(res, config);
+              }
+            });
+  }
+}

src/main/java/com/google/cloud/agentplatform/AsyncMemoryRevisions.java

@@ -18,9 +18,9 @@
 
 package com.google.cloud.agentplatform;
 
-import com.google.cloud.agentplatform.types.GetAgentEngineMemoryRevisionConfig;
-import com.google.cloud.agentplatform.types.ListAgentEngineMemoryRevisionsConfig;
-import com.google.cloud.agentplatform.types.ListAgentEngineMemoryRevisionsResponse;
+import com.google.cloud.agentplatform.types.GetMemoryRevisionConfig;
+import com.google.cloud.agentplatform.types.ListMemoryRevisionsConfig;
+import com.google.cloud.agentplatform.types.ListMemoryRevisionsResponse;
 import com.google.cloud.agentplatform.types.MemoryRevision;
 import com.google.genai.ApiClient;
 import com.google.genai.ApiResponse;
@@ -38,8 +38,7 @@ public AsyncMemoryRevisions(ApiClient apiClient) {
     this.memoryRevisions = new MemoryRevisions(apiClient);
   }
 
-  public CompletableFuture<MemoryRevision> get(
-      String name, GetAgentEngineMemoryRevisionConfig config) {
+  public CompletableFuture<MemoryRevision> get(String name, GetMemoryRevisionConfig config) {
 
     BuiltRequest builtRequest = memoryRevisions.buildRequestForGet(name, config);
     return this.apiClient
@@ -52,8 +51,8 @@ public CompletableFuture<MemoryRevision> get(
             });
   }
 
-  CompletableFuture<ListAgentEngineMemoryRevisionsResponse> privateList(
-      String name, ListAgentEngineMemoryRevisionsConfig config) {
+  CompletableFuture<ListMemoryRevisionsResponse> privateList(
+      String name, ListMemoryRevisionsConfig config) {
 
     BuiltRequest builtRequest = memoryRevisions.buildRequestForPrivateList(name, config);
     return this.apiClient