diff --git a/src/main/java/com/google/cloud/agentplatform/AgentEngines.java b/src/main/java/com/google/cloud/agentplatform/AgentEngines.java index bd1a171..43ce3e4 100644 --- a/src/main/java/com/google/cloud/agentplatform/AgentEngines.java +++ b/src/main/java/com/google/cloud/agentplatform/AgentEngines.java @@ -55,7 +55,6 @@ public final class AgentEngines { public final Sessions sessions; public final Sandboxes sandboxes; - public final Memories memories; final ApiClient apiClient; @@ -63,7 +62,6 @@ public AgentEngines(ApiClient apiClient) { this.apiClient = apiClient; this.sessions = new Sessions(apiClient); this.sandboxes = new Sandboxes(apiClient); - this.memories = new Memories(apiClient); } @ExcludeFromGeneratedCoverageReport diff --git a/src/main/java/com/google/cloud/agentplatform/AsyncAgentEngines.java b/src/main/java/com/google/cloud/agentplatform/AsyncAgentEngines.java index 7444dae..950ecd0 100644 --- a/src/main/java/com/google/cloud/agentplatform/AsyncAgentEngines.java +++ b/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 privateCreate(CreateAgentEngineConfig config) { diff --git a/src/main/java/com/google/cloud/agentplatform/AsyncMemories.java b/src/main/java/com/google/cloud/agentplatform/AsyncMemories.java index 8e1f3a7..4239d3e 100644 --- a/src/main/java/com/google/cloud/agentplatform/AsyncMemories.java +++ b/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 privateCreate( - String name, String fact, Map scope, AgentEngineMemoryConfig config) { + CompletableFuture privateCreate( + String name, String fact, Map scope, MemoryConfig config) { BuiltRequest builtRequest = memories.buildRequestForPrivateCreate(name, fact, scope, config); return this.apiClient @@ -83,8 +79,7 @@ CompletableFuture privateCreate( }); } - public CompletableFuture delete( - String name, DeleteAgentEngineMemoryConfig config) { + public CompletableFuture delete(String name, DeleteMemoryConfig config) { BuiltRequest builtRequest = memories.buildRequestForDelete(name, config); return this.apiClient @@ -98,13 +93,13 @@ public CompletableFuture delete( }); } - CompletableFuture privateGenerate( + CompletableFuture privateGenerate( String name, GenerateMemoriesRequestVertexSessionSource vertexSessionSource, GenerateMemoriesRequestDirectContentsSource directContentsSource, GenerateMemoriesRequestDirectMemoriesSource directMemoriesSource, Map scope, - GenerateAgentEngineMemoriesConfig config) { + GenerateMemoriesConfig config) { BuiltRequest builtRequest = memories.buildRequestForPrivateGenerate( @@ -119,7 +114,7 @@ CompletableFuture privateGenerate( }); } - public CompletableFuture get(String name, GetAgentEngineMemoryConfig config) { + public CompletableFuture get(String name, GetMemoryConfig config) { BuiltRequest builtRequest = memories.buildRequestForGet(name, config); return this.apiClient @@ -132,29 +127,7 @@ public CompletableFuture get(String name, GetAgentEngineMemoryConfig con }); } - CompletableFuture privateIngestEvents( - String name, - String streamId, - IngestionDirectContentsSource directContentsSource, - Map 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 privateList( - String name, ListAgentEngineMemoryConfig config) { + CompletableFuture privateList(String name, ListMemoriesConfig config) { BuiltRequest builtRequest = memories.buildRequestForPrivateList(name, config); return this.apiClient @@ -167,8 +140,8 @@ CompletableFuture privateList( }); } - CompletableFuture privateGetMemoryOperation( - String operationName, GetAgentEngineOperationConfig config) { + CompletableFuture privateGetMemoryOperation( + String operationName, GetMemoryBankOperationConfig config) { BuiltRequest builtRequest = memories.buildRequestForPrivateGetMemoryOperation(operationName, config); @@ -182,8 +155,8 @@ CompletableFuture privateGetMemoryOperation( }); } - CompletableFuture privateGetGenerateMemoriesOperation( - String operationName, GetAgentEngineOperationConfig config) { + CompletableFuture privateGetGenerateMemoriesOperation( + String operationName, GetMemoryBankOperationConfig config) { BuiltRequest builtRequest = memories.buildRequestForPrivateGetGenerateMemoriesOperation(operationName, config); @@ -202,7 +175,7 @@ CompletableFuture privateRetrieve( Map scope, RetrieveMemoriesRequestSimilaritySearchParams similaritySearchParams, RetrieveMemoriesRequestSimpleRetrievalParams simpleRetrievalParams, - RetrieveAgentEngineMemoriesConfig config) { + RetrieveMemoriesConfig config) { BuiltRequest builtRequest = memories.buildRequestForPrivateRetrieve( @@ -231,8 +204,8 @@ public CompletableFuture retrieveProfiles( }); } - CompletableFuture privateRollback( - String name, String targetRevisionId, RollbackAgentEngineMemoryConfig config) { + CompletableFuture privateRollback( + String name, String targetRevisionId, RollbackMemoryConfig config) { BuiltRequest builtRequest = memories.buildRequestForPrivateRollback(name, targetRevisionId, config); @@ -246,8 +219,8 @@ CompletableFuture privateRollback( }); } - CompletableFuture privateUpdate( - String name, String fact, Map scope, UpdateAgentEngineMemoryConfig config) { + CompletableFuture privateUpdate( + String name, String fact, Map scope, UpdateMemoryConfig config) { BuiltRequest builtRequest = memories.buildRequestForPrivateUpdate(name, fact, scope, config); return this.apiClient @@ -260,12 +233,12 @@ CompletableFuture privateUpdate( }); } - CompletableFuture privatePurge( + CompletableFuture privatePurge( String name, String filter, List filterGroups, boolean force, - PurgeAgentEngineMemoriesConfig config) { + PurgeMemoriesConfig config) { BuiltRequest builtRequest = memories.buildRequestForPrivatePurge(name, filter, filterGroups, force, config); diff --git a/src/main/java/com/google/cloud/agentplatform/AsyncMemoryBanks.java b/src/main/java/com/google/cloud/agentplatform/AsyncMemoryBanks.java new file mode 100644 index 0000000..6b44900 --- /dev/null +++ b/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 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 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 privateIngestEvents( + String name, + String streamId, + IngestionDirectContentsSource directContentsSource, + Map 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 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); + } + }); + } +} diff --git a/src/main/java/com/google/cloud/agentplatform/AsyncMemoryRevisions.java b/src/main/java/com/google/cloud/agentplatform/AsyncMemoryRevisions.java index ddf4216..d068aa5 100644 --- a/src/main/java/com/google/cloud/agentplatform/AsyncMemoryRevisions.java +++ b/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 get( - String name, GetAgentEngineMemoryRevisionConfig config) { + public CompletableFuture get(String name, GetMemoryRevisionConfig config) { BuiltRequest builtRequest = memoryRevisions.buildRequestForGet(name, config); return this.apiClient @@ -52,8 +51,8 @@ public CompletableFuture get( }); } - CompletableFuture privateList( - String name, ListAgentEngineMemoryRevisionsConfig config) { + CompletableFuture privateList( + String name, ListMemoryRevisionsConfig config) { BuiltRequest builtRequest = memoryRevisions.buildRequestForPrivateList(name, config); return this.apiClient diff --git a/src/main/java/com/google/cloud/agentplatform/Memories.java b/src/main/java/com/google/cloud/agentplatform/Memories.java index 49b0add..58680d8 100644 --- a/src/main/java/com/google/cloud/agentplatform/Memories.java +++ b/src/main/java/com/google/cloud/agentplatform/Memories.java @@ -20,49 +20,44 @@ import com.fasterxml.jackson.databind.JsonNode; import com.fasterxml.jackson.databind.node.ObjectNode; -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.CreateAgentEngineMemoryRequestParameters; -import com.google.cloud.agentplatform.types.DeleteAgentEngineMemoryConfig; -import com.google.cloud.agentplatform.types.DeleteAgentEngineMemoryOperation; -import com.google.cloud.agentplatform.types.DeleteAgentEngineMemoryRequestParameters; -import com.google.cloud.agentplatform.types.GenerateAgentEngineMemoriesConfig; -import com.google.cloud.agentplatform.types.GenerateAgentEngineMemoriesRequestParameters; +import com.google.cloud.agentplatform.types.CreateMemoryRequestParameters; +import com.google.cloud.agentplatform.types.DeleteMemoryConfig; +import com.google.cloud.agentplatform.types.DeleteMemoryOperation; +import com.google.cloud.agentplatform.types.DeleteMemoryRequestParameters; +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.GenerateMemoriesRequestParameters; import com.google.cloud.agentplatform.types.GenerateMemoriesRequestVertexSessionSource; -import com.google.cloud.agentplatform.types.GetAgentEngineGenerateMemoriesOperationParameters; -import com.google.cloud.agentplatform.types.GetAgentEngineMemoryConfig; -import com.google.cloud.agentplatform.types.GetAgentEngineMemoryOperationParameters; -import com.google.cloud.agentplatform.types.GetAgentEngineMemoryRequestParameters; -import com.google.cloud.agentplatform.types.GetAgentEngineOperationConfig; -import com.google.cloud.agentplatform.types.IngestEventsConfig; -import com.google.cloud.agentplatform.types.IngestEventsRequestParameters; -import com.google.cloud.agentplatform.types.IngestionDirectContentsSource; -import com.google.cloud.agentplatform.types.ListAgentEngineMemoryConfig; -import com.google.cloud.agentplatform.types.ListAgentEngineMemoryRequestParameters; -import com.google.cloud.agentplatform.types.ListReasoningEnginesMemoriesResponse; +import com.google.cloud.agentplatform.types.GetGenerateMemoriesOperationParameters; +import com.google.cloud.agentplatform.types.GetMemoryBankOperationConfig; +import com.google.cloud.agentplatform.types.GetMemoryConfig; +import com.google.cloud.agentplatform.types.GetMemoryOperationParameters; +import com.google.cloud.agentplatform.types.GetMemoryRequestParameters; +import com.google.cloud.agentplatform.types.ListMemoriesConfig; +import com.google.cloud.agentplatform.types.ListMemoriesRequestParameters; +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.PurgeAgentEngineMemoriesRequestParameters; -import com.google.cloud.agentplatform.types.RetrieveAgentEngineMemoriesConfig; -import com.google.cloud.agentplatform.types.RetrieveAgentEngineMemoriesRequestParameters; +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.PurgeMemoriesRequestParameters; +import com.google.cloud.agentplatform.types.RetrieveMemoriesConfig; +import com.google.cloud.agentplatform.types.RetrieveMemoriesRequestParameters; 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.RetrieveMemoryProfilesRequestParameters; import com.google.cloud.agentplatform.types.RetrieveProfilesResponse; -import com.google.cloud.agentplatform.types.RollbackAgentEngineMemoryConfig; -import com.google.cloud.agentplatform.types.RollbackAgentEngineMemoryRequestParameters; -import com.google.cloud.agentplatform.types.UpdateAgentEngineMemoryConfig; -import com.google.cloud.agentplatform.types.UpdateAgentEngineMemoryRequestParameters; +import com.google.cloud.agentplatform.types.RollbackMemoryConfig; +import com.google.cloud.agentplatform.types.RollbackMemoryOperation; +import com.google.cloud.agentplatform.types.RollbackMemoryRequestParameters; +import com.google.cloud.agentplatform.types.UpdateMemoryConfig; +import com.google.cloud.agentplatform.types.UpdateMemoryRequestParameters; import com.google.genai.ApiClient; import com.google.genai.ApiResponse; import com.google.genai.Common; @@ -87,85 +82,7 @@ public Memories(ApiClient apiClient) { } @ExcludeFromGeneratedCoverageReport - ObjectNode agentEngineMemoryConfigToVertex(JsonNode fromObject, ObjectNode parentObject) { - ObjectNode toObject = JsonSerializable.objectMapper().createObjectNode(); - - if (Common.getValueByPath(fromObject, new String[] {"displayName"}) != null) { - Common.setValueByPath( - parentObject, - new String[] {"displayName"}, - Common.getValueByPath(fromObject, new String[] {"displayName"})); - } - - if (Common.getValueByPath(fromObject, new String[] {"description"}) != null) { - Common.setValueByPath( - parentObject, - new String[] {"description"}, - Common.getValueByPath(fromObject, new String[] {"description"})); - } - - if (Common.getValueByPath(fromObject, new String[] {"ttl"}) != null) { - Common.setValueByPath( - parentObject, - new String[] {"ttl"}, - Common.getValueByPath(fromObject, new String[] {"ttl"})); - } - - if (Common.getValueByPath(fromObject, new String[] {"expireTime"}) != null) { - Common.setValueByPath( - parentObject, - new String[] {"expireTime"}, - Common.getValueByPath(fromObject, new String[] {"expireTime"})); - } - - if (Common.getValueByPath(fromObject, new String[] {"revisionExpireTime"}) != null) { - Common.setValueByPath( - parentObject, - new String[] {"revisionExpireTime"}, - Common.getValueByPath(fromObject, new String[] {"revisionExpireTime"})); - } - - if (Common.getValueByPath(fromObject, new String[] {"revisionTtl"}) != null) { - Common.setValueByPath( - parentObject, - new String[] {"revisionTtl"}, - Common.getValueByPath(fromObject, new String[] {"revisionTtl"})); - } - - if (Common.getValueByPath(fromObject, new String[] {"disableMemoryRevisions"}) != null) { - Common.setValueByPath( - parentObject, - new String[] {"disableMemoryRevisions"}, - Common.getValueByPath(fromObject, new String[] {"disableMemoryRevisions"})); - } - - if (Common.getValueByPath(fromObject, new String[] {"topics"}) != null) { - Common.setValueByPath( - parentObject, - new String[] {"topics"}, - Common.getValueByPath(fromObject, new String[] {"topics"})); - } - - if (Common.getValueByPath(fromObject, new String[] {"metadata"}) != null) { - Common.setValueByPath( - parentObject, - new String[] {"metadata"}, - Common.getValueByPath(fromObject, new String[] {"metadata"})); - } - - if (Common.getValueByPath(fromObject, new String[] {"memoryId"}) != null) { - Common.setValueByPath( - parentObject, - new String[] {"_query", "memoryId"}, - Common.getValueByPath(fromObject, new String[] {"memoryId"})); - } - - return toObject; - } - - @ExcludeFromGeneratedCoverageReport - ObjectNode createAgentEngineMemoryRequestParametersToVertex( - JsonNode fromObject, ObjectNode parentObject) { + ObjectNode createMemoryRequestParametersToVertex(JsonNode fromObject, ObjectNode parentObject) { ObjectNode toObject = JsonSerializable.objectMapper().createObjectNode(); if (Common.getValueByPath(fromObject, new String[] {"name"}) != null) { Common.setValueByPath( @@ -190,7 +107,7 @@ ObjectNode createAgentEngineMemoryRequestParametersToVertex( if (Common.getValueByPath(fromObject, new String[] {"config"}) != null) { JsonNode unused = - agentEngineMemoryConfigToVertex( + memoryConfigToVertex( JsonSerializable.toJsonNode( Common.getValueByPath(fromObject, new String[] {"config"})), toObject); @@ -200,8 +117,7 @@ ObjectNode createAgentEngineMemoryRequestParametersToVertex( } @ExcludeFromGeneratedCoverageReport - ObjectNode deleteAgentEngineMemoryRequestParametersToVertex( - JsonNode fromObject, ObjectNode parentObject) { + ObjectNode deleteMemoryRequestParametersToVertex(JsonNode fromObject, ObjectNode parentObject) { ObjectNode toObject = JsonSerializable.objectMapper().createObjectNode(); if (Common.getValueByPath(fromObject, new String[] {"name"}) != null) { Common.setValueByPath( @@ -214,8 +130,7 @@ ObjectNode deleteAgentEngineMemoryRequestParametersToVertex( } @ExcludeFromGeneratedCoverageReport - ObjectNode generateAgentEngineMemoriesConfigToVertex( - JsonNode fromObject, ObjectNode parentObject) { + ObjectNode generateMemoriesConfigToVertex(JsonNode fromObject, ObjectNode parentObject) { ObjectNode toObject = JsonSerializable.objectMapper().createObjectNode(); if (Common.getValueByPath(fromObject, new String[] {"disableConsolidation"}) != null) { @@ -278,7 +193,7 @@ ObjectNode generateAgentEngineMemoriesConfigToVertex( } @ExcludeFromGeneratedCoverageReport - ObjectNode generateAgentEngineMemoriesRequestParametersToVertex( + ObjectNode generateMemoriesRequestParametersToVertex( JsonNode fromObject, ObjectNode parentObject) { ObjectNode toObject = JsonSerializable.objectMapper().createObjectNode(); if (Common.getValueByPath(fromObject, new String[] {"name"}) != null) { @@ -318,7 +233,7 @@ ObjectNode generateAgentEngineMemoriesRequestParametersToVertex( if (Common.getValueByPath(fromObject, new String[] {"config"}) != null) { JsonNode unused = - generateAgentEngineMemoriesConfigToVertex( + generateMemoriesConfigToVertex( JsonSerializable.toJsonNode( Common.getValueByPath(fromObject, new String[] {"config"})), toObject); @@ -328,7 +243,7 @@ ObjectNode generateAgentEngineMemoriesRequestParametersToVertex( } @ExcludeFromGeneratedCoverageReport - ObjectNode getAgentEngineGenerateMemoriesOperationParametersToVertex( + ObjectNode getGenerateMemoriesOperationParametersToVertex( JsonNode fromObject, ObjectNode parentObject) { ObjectNode toObject = JsonSerializable.objectMapper().createObjectNode(); if (Common.getValueByPath(fromObject, new String[] {"operationName"}) != null) { @@ -342,8 +257,7 @@ ObjectNode getAgentEngineGenerateMemoriesOperationParametersToVertex( } @ExcludeFromGeneratedCoverageReport - ObjectNode getAgentEngineMemoryOperationParametersToVertex( - JsonNode fromObject, ObjectNode parentObject) { + ObjectNode getMemoryOperationParametersToVertex(JsonNode fromObject, ObjectNode parentObject) { ObjectNode toObject = JsonSerializable.objectMapper().createObjectNode(); if (Common.getValueByPath(fromObject, new String[] {"operationName"}) != null) { Common.setValueByPath( @@ -356,8 +270,7 @@ ObjectNode getAgentEngineMemoryOperationParametersToVertex( } @ExcludeFromGeneratedCoverageReport - ObjectNode getAgentEngineMemoryRequestParametersToVertex( - JsonNode fromObject, ObjectNode parentObject) { + ObjectNode getMemoryRequestParametersToVertex(JsonNode fromObject, ObjectNode parentObject) { ObjectNode toObject = JsonSerializable.objectMapper().createObjectNode(); if (Common.getValueByPath(fromObject, new String[] {"name"}) != null) { Common.setValueByPath( @@ -370,63 +283,42 @@ ObjectNode getAgentEngineMemoryRequestParametersToVertex( } @ExcludeFromGeneratedCoverageReport - ObjectNode ingestEventsConfigToVertex(JsonNode fromObject, ObjectNode parentObject) { + ObjectNode listMemoriesConfigToVertex(JsonNode fromObject, ObjectNode parentObject) { ObjectNode toObject = JsonSerializable.objectMapper().createObjectNode(); - if (Common.getValueByPath(fromObject, new String[] {"forceFlush"}) != null) { - Common.setValueByPath( - parentObject, - new String[] {"forceFlush"}, - Common.getValueByPath(fromObject, new String[] {"forceFlush"})); - } - - if (Common.getValueByPath(fromObject, new String[] {"revisionLabels"}) != null) { - Common.setValueByPath( - parentObject, - new String[] {"revisionLabels"}, - Common.getValueByPath(fromObject, new String[] {"revisionLabels"})); - } - - if (Common.getValueByPath(fromObject, new String[] {"revisionExpireTime"}) != null) { - Common.setValueByPath( - parentObject, - new String[] {"revisionExpireTime"}, - Common.getValueByPath(fromObject, new String[] {"revisionExpireTime"})); - } - - if (Common.getValueByPath(fromObject, new String[] {"revisionTtl"}) != null) { + if (Common.getValueByPath(fromObject, new String[] {"pageSize"}) != null) { Common.setValueByPath( parentObject, - new String[] {"revisionTtl"}, - Common.getValueByPath(fromObject, new String[] {"revisionTtl"})); + new String[] {"_query", "pageSize"}, + Common.getValueByPath(fromObject, new String[] {"pageSize"})); } - if (Common.getValueByPath(fromObject, new String[] {"disableMemoryRevisions"}) != null) { + if (Common.getValueByPath(fromObject, new String[] {"pageToken"}) != null) { Common.setValueByPath( parentObject, - new String[] {"disableMemoryRevisions"}, - Common.getValueByPath(fromObject, new String[] {"disableMemoryRevisions"})); + new String[] {"_query", "pageToken"}, + Common.getValueByPath(fromObject, new String[] {"pageToken"})); } - if (Common.getValueByPath(fromObject, new String[] {"metadata"}) != null) { + if (Common.getValueByPath(fromObject, new String[] {"filter"}) != null) { Common.setValueByPath( parentObject, - new String[] {"metadata"}, - Common.getValueByPath(fromObject, new String[] {"metadata"})); + new String[] {"_query", "filter"}, + Common.getValueByPath(fromObject, new String[] {"filter"})); } - if (Common.getValueByPath(fromObject, new String[] {"metadataMergeStrategy"}) != null) { + if (Common.getValueByPath(fromObject, new String[] {"orderBy"}) != null) { Common.setValueByPath( parentObject, - new String[] {"metadataMergeStrategy"}, - Common.getValueByPath(fromObject, new String[] {"metadataMergeStrategy"})); + new String[] {"_query", "orderBy"}, + Common.getValueByPath(fromObject, new String[] {"orderBy"})); } return toObject; } @ExcludeFromGeneratedCoverageReport - ObjectNode ingestEventsRequestParametersToVertex(JsonNode fromObject, ObjectNode parentObject) { + ObjectNode listMemoriesRequestParametersToVertex(JsonNode fromObject, ObjectNode parentObject) { ObjectNode toObject = JsonSerializable.objectMapper().createObjectNode(); if (Common.getValueByPath(fromObject, new String[] {"name"}) != null) { Common.setValueByPath( @@ -435,37 +327,9 @@ ObjectNode ingestEventsRequestParametersToVertex(JsonNode fromObject, ObjectNode Common.getValueByPath(fromObject, new String[] {"name"})); } - if (Common.getValueByPath(fromObject, new String[] {"streamId"}) != null) { - Common.setValueByPath( - toObject, - new String[] {"streamId"}, - Common.getValueByPath(fromObject, new String[] {"streamId"})); - } - - if (Common.getValueByPath(fromObject, new String[] {"directContentsSource"}) != null) { - Common.setValueByPath( - toObject, - new String[] {"directContentsSource"}, - Common.getValueByPath(fromObject, new String[] {"directContentsSource"})); - } - - if (Common.getValueByPath(fromObject, new String[] {"scope"}) != null) { - Common.setValueByPath( - toObject, - new String[] {"scope"}, - Common.getValueByPath(fromObject, new String[] {"scope"})); - } - - if (Common.getValueByPath(fromObject, new String[] {"generationTriggerConfig"}) != null) { - Common.setValueByPath( - toObject, - new String[] {"generationTriggerConfig"}, - Common.getValueByPath(fromObject, new String[] {"generationTriggerConfig"})); - } - if (Common.getValueByPath(fromObject, new String[] {"config"}) != null) { JsonNode unused = - ingestEventsConfigToVertex( + listMemoriesConfigToVertex( JsonSerializable.toJsonNode( Common.getValueByPath(fromObject, new String[] {"config"})), toObject); @@ -475,65 +339,84 @@ ObjectNode ingestEventsRequestParametersToVertex(JsonNode fromObject, ObjectNode } @ExcludeFromGeneratedCoverageReport - ObjectNode listAgentEngineMemoryConfigToVertex(JsonNode fromObject, ObjectNode parentObject) { + ObjectNode memoryConfigToVertex(JsonNode fromObject, ObjectNode parentObject) { ObjectNode toObject = JsonSerializable.objectMapper().createObjectNode(); - if (Common.getValueByPath(fromObject, new String[] {"pageSize"}) != null) { + if (Common.getValueByPath(fromObject, new String[] {"displayName"}) != null) { Common.setValueByPath( parentObject, - new String[] {"_query", "pageSize"}, - Common.getValueByPath(fromObject, new String[] {"pageSize"})); + new String[] {"displayName"}, + Common.getValueByPath(fromObject, new String[] {"displayName"})); } - if (Common.getValueByPath(fromObject, new String[] {"pageToken"}) != null) { + if (Common.getValueByPath(fromObject, new String[] {"description"}) != null) { Common.setValueByPath( parentObject, - new String[] {"_query", "pageToken"}, - Common.getValueByPath(fromObject, new String[] {"pageToken"})); + new String[] {"description"}, + Common.getValueByPath(fromObject, new String[] {"description"})); } - if (Common.getValueByPath(fromObject, new String[] {"filter"}) != null) { + if (Common.getValueByPath(fromObject, new String[] {"ttl"}) != null) { Common.setValueByPath( parentObject, - new String[] {"_query", "filter"}, - Common.getValueByPath(fromObject, new String[] {"filter"})); + new String[] {"ttl"}, + Common.getValueByPath(fromObject, new String[] {"ttl"})); } - if (Common.getValueByPath(fromObject, new String[] {"orderBy"}) != null) { + if (Common.getValueByPath(fromObject, new String[] {"expireTime"}) != null) { Common.setValueByPath( parentObject, - new String[] {"_query", "orderBy"}, - Common.getValueByPath(fromObject, new String[] {"orderBy"})); + new String[] {"expireTime"}, + Common.getValueByPath(fromObject, new String[] {"expireTime"})); } - return toObject; - } + if (Common.getValueByPath(fromObject, new String[] {"revisionExpireTime"}) != null) { + Common.setValueByPath( + parentObject, + new String[] {"revisionExpireTime"}, + Common.getValueByPath(fromObject, new String[] {"revisionExpireTime"})); + } - @ExcludeFromGeneratedCoverageReport - ObjectNode listAgentEngineMemoryRequestParametersToVertex( - JsonNode fromObject, ObjectNode parentObject) { - ObjectNode toObject = JsonSerializable.objectMapper().createObjectNode(); - if (Common.getValueByPath(fromObject, new String[] {"name"}) != null) { + if (Common.getValueByPath(fromObject, new String[] {"revisionTtl"}) != null) { Common.setValueByPath( - toObject, - new String[] {"_url", "name"}, - Common.getValueByPath(fromObject, new String[] {"name"})); + parentObject, + new String[] {"revisionTtl"}, + Common.getValueByPath(fromObject, new String[] {"revisionTtl"})); } - if (Common.getValueByPath(fromObject, new String[] {"config"}) != null) { - JsonNode unused = - listAgentEngineMemoryConfigToVertex( - JsonSerializable.toJsonNode( - Common.getValueByPath(fromObject, new String[] {"config"})), - toObject); + if (Common.getValueByPath(fromObject, new String[] {"disableMemoryRevisions"}) != null) { + Common.setValueByPath( + parentObject, + new String[] {"disableMemoryRevisions"}, + Common.getValueByPath(fromObject, new String[] {"disableMemoryRevisions"})); + } + + if (Common.getValueByPath(fromObject, new String[] {"topics"}) != null) { + Common.setValueByPath( + parentObject, + new String[] {"topics"}, + Common.getValueByPath(fromObject, new String[] {"topics"})); + } + + if (Common.getValueByPath(fromObject, new String[] {"metadata"}) != null) { + Common.setValueByPath( + parentObject, + new String[] {"metadata"}, + Common.getValueByPath(fromObject, new String[] {"metadata"})); + } + + if (Common.getValueByPath(fromObject, new String[] {"memoryId"}) != null) { + Common.setValueByPath( + parentObject, + new String[] {"_query", "memoryId"}, + Common.getValueByPath(fromObject, new String[] {"memoryId"})); } return toObject; } @ExcludeFromGeneratedCoverageReport - ObjectNode purgeAgentEngineMemoriesRequestParametersToVertex( - JsonNode fromObject, ObjectNode parentObject) { + ObjectNode purgeMemoriesRequestParametersToVertex(JsonNode fromObject, ObjectNode parentObject) { ObjectNode toObject = JsonSerializable.objectMapper().createObjectNode(); if (Common.getValueByPath(fromObject, new String[] {"name"}) != null) { Common.setValueByPath( @@ -567,8 +450,7 @@ ObjectNode purgeAgentEngineMemoriesRequestParametersToVertex( } @ExcludeFromGeneratedCoverageReport - ObjectNode retrieveAgentEngineMemoriesConfigToVertex( - JsonNode fromObject, ObjectNode parentObject) { + ObjectNode retrieveMemoriesConfigToVertex(JsonNode fromObject, ObjectNode parentObject) { ObjectNode toObject = JsonSerializable.objectMapper().createObjectNode(); if (Common.getValueByPath(fromObject, new String[] {"filter"}) != null) { @@ -596,7 +478,7 @@ ObjectNode retrieveAgentEngineMemoriesConfigToVertex( } @ExcludeFromGeneratedCoverageReport - ObjectNode retrieveAgentEngineMemoriesRequestParametersToVertex( + ObjectNode retrieveMemoriesRequestParametersToVertex( JsonNode fromObject, ObjectNode parentObject) { ObjectNode toObject = JsonSerializable.objectMapper().createObjectNode(); if (Common.getValueByPath(fromObject, new String[] {"name"}) != null) { @@ -629,7 +511,7 @@ ObjectNode retrieveAgentEngineMemoriesRequestParametersToVertex( if (Common.getValueByPath(fromObject, new String[] {"config"}) != null) { JsonNode unused = - retrieveAgentEngineMemoriesConfigToVertex( + retrieveMemoriesConfigToVertex( JsonSerializable.toJsonNode( Common.getValueByPath(fromObject, new String[] {"config"})), toObject); @@ -660,8 +542,7 @@ ObjectNode retrieveMemoryProfilesRequestParametersToVertex( } @ExcludeFromGeneratedCoverageReport - ObjectNode rollbackAgentEngineMemoryRequestParametersToVertex( - JsonNode fromObject, ObjectNode parentObject) { + ObjectNode rollbackMemoryRequestParametersToVertex(JsonNode fromObject, ObjectNode parentObject) { ObjectNode toObject = JsonSerializable.objectMapper().createObjectNode(); if (Common.getValueByPath(fromObject, new String[] {"name"}) != null) { Common.setValueByPath( @@ -681,7 +562,7 @@ ObjectNode rollbackAgentEngineMemoryRequestParametersToVertex( } @ExcludeFromGeneratedCoverageReport - ObjectNode updateAgentEngineMemoryConfigToVertex(JsonNode fromObject, ObjectNode parentObject) { + ObjectNode updateMemoryConfigToVertex(JsonNode fromObject, ObjectNode parentObject) { ObjectNode toObject = JsonSerializable.objectMapper().createObjectNode(); if (Common.getValueByPath(fromObject, new String[] {"displayName"}) != null) { @@ -765,8 +646,7 @@ ObjectNode updateAgentEngineMemoryConfigToVertex(JsonNode fromObject, ObjectNode } @ExcludeFromGeneratedCoverageReport - ObjectNode updateAgentEngineMemoryRequestParametersToVertex( - JsonNode fromObject, ObjectNode parentObject) { + ObjectNode updateMemoryRequestParametersToVertex(JsonNode fromObject, ObjectNode parentObject) { ObjectNode toObject = JsonSerializable.objectMapper().createObjectNode(); if (Common.getValueByPath(fromObject, new String[] {"name"}) != null) { Common.setValueByPath( @@ -791,7 +671,7 @@ ObjectNode updateAgentEngineMemoryRequestParametersToVertex( if (Common.getValueByPath(fromObject, new String[] {"config"}) != null) { JsonNode unused = - updateAgentEngineMemoryConfigToVertex( + updateMemoryConfigToVertex( JsonSerializable.toJsonNode( Common.getValueByPath(fromObject, new String[] {"config"})), toObject); @@ -802,10 +682,10 @@ ObjectNode updateAgentEngineMemoryRequestParametersToVertex( /** A shared buildRequest method for both sync and async methods. */ BuiltRequest buildRequestForPrivateCreate( - String name, String fact, Map scope, AgentEngineMemoryConfig config) { + String name, String fact, Map scope, MemoryConfig config) { - CreateAgentEngineMemoryRequestParameters.Builder parameterBuilder = - CreateAgentEngineMemoryRequestParameters.builder(); + CreateMemoryRequestParameters.Builder parameterBuilder = + CreateMemoryRequestParameters.builder(); if (!Common.isZero(name)) { parameterBuilder.name(name); @@ -824,7 +704,7 @@ BuiltRequest buildRequestForPrivateCreate( ObjectNode body; String path; if (this.apiClient.vertexAI()) { - body = createAgentEngineMemoryRequestParametersToVertex(parameterNode, null); + body = createMemoryRequestParametersToVertex(parameterNode, null); path = Common.formatMap("{name}/memories", body.get("_url")); } else { throw new UnsupportedOperationException( @@ -849,8 +729,7 @@ BuiltRequest buildRequestForPrivateCreate( } /** A shared processResponse function for both sync and async methods. */ - AgentEngineMemoryOperation processResponseForPrivateCreate( - ApiResponse response, AgentEngineMemoryConfig config) { + MemoryOperation processResponseForPrivateCreate(ApiResponse response, MemoryConfig config) { ResponseBody responseBody = response.getBody(); String responseString; try { @@ -867,11 +746,11 @@ AgentEngineMemoryOperation processResponseForPrivateCreate( + " Developer API mode."); } - return JsonSerializable.fromJsonNode(responseNode, AgentEngineMemoryOperation.class); + return JsonSerializable.fromJsonNode(responseNode, MemoryOperation.class); } - public AgentEngineMemoryOperation privateCreate( - String name, String fact, Map scope, AgentEngineMemoryConfig config) { + public MemoryOperation privateCreate( + String name, String fact, Map scope, MemoryConfig config) { BuiltRequest builtRequest = buildRequestForPrivateCreate(name, fact, scope, config); try (ApiResponse response = @@ -882,10 +761,10 @@ public AgentEngineMemoryOperation privateCreate( } /** A shared buildRequest method for both sync and async methods. */ - BuiltRequest buildRequestForDelete(String name, DeleteAgentEngineMemoryConfig config) { + BuiltRequest buildRequestForDelete(String name, DeleteMemoryConfig config) { - DeleteAgentEngineMemoryRequestParameters.Builder parameterBuilder = - DeleteAgentEngineMemoryRequestParameters.builder(); + DeleteMemoryRequestParameters.Builder parameterBuilder = + DeleteMemoryRequestParameters.builder(); if (!Common.isZero(name)) { parameterBuilder.name(name); @@ -898,7 +777,7 @@ BuiltRequest buildRequestForDelete(String name, DeleteAgentEngineMemoryConfig co ObjectNode body; String path; if (this.apiClient.vertexAI()) { - body = deleteAgentEngineMemoryRequestParametersToVertex(parameterNode, null); + body = deleteMemoryRequestParametersToVertex(parameterNode, null); path = Common.formatMap("{name}", body.get("_url")); } else { throw new UnsupportedOperationException( @@ -923,8 +802,7 @@ BuiltRequest buildRequestForDelete(String name, DeleteAgentEngineMemoryConfig co } /** A shared processResponse function for both sync and async methods. */ - DeleteAgentEngineMemoryOperation processResponseForDelete( - ApiResponse response, DeleteAgentEngineMemoryConfig config) { + DeleteMemoryOperation processResponseForDelete(ApiResponse response, DeleteMemoryConfig config) { ResponseBody responseBody = response.getBody(); String responseString; try { @@ -941,11 +819,10 @@ DeleteAgentEngineMemoryOperation processResponseForDelete( + " Developer API mode."); } - return JsonSerializable.fromJsonNode(responseNode, DeleteAgentEngineMemoryOperation.class); + return JsonSerializable.fromJsonNode(responseNode, DeleteMemoryOperation.class); } - public DeleteAgentEngineMemoryOperation delete( - String name, DeleteAgentEngineMemoryConfig config) { + public DeleteMemoryOperation delete(String name, DeleteMemoryConfig config) { BuiltRequest builtRequest = buildRequestForDelete(name, config); try (ApiResponse response = @@ -962,10 +839,10 @@ BuiltRequest buildRequestForPrivateGenerate( GenerateMemoriesRequestDirectContentsSource directContentsSource, GenerateMemoriesRequestDirectMemoriesSource directMemoriesSource, Map scope, - GenerateAgentEngineMemoriesConfig config) { + GenerateMemoriesConfig config) { - GenerateAgentEngineMemoriesRequestParameters.Builder parameterBuilder = - GenerateAgentEngineMemoriesRequestParameters.builder(); + GenerateMemoriesRequestParameters.Builder parameterBuilder = + GenerateMemoriesRequestParameters.builder(); if (!Common.isZero(name)) { parameterBuilder.name(name); @@ -990,7 +867,7 @@ BuiltRequest buildRequestForPrivateGenerate( ObjectNode body; String path; if (this.apiClient.vertexAI()) { - body = generateAgentEngineMemoriesRequestParametersToVertex(parameterNode, null); + body = generateMemoriesRequestParametersToVertex(parameterNode, null); path = Common.formatMap("{name}/memories:generate", body.get("_url")); } else { throw new UnsupportedOperationException( @@ -1015,8 +892,8 @@ BuiltRequest buildRequestForPrivateGenerate( } /** A shared processResponse function for both sync and async methods. */ - AgentEngineGenerateMemoriesOperation processResponseForPrivateGenerate( - ApiResponse response, GenerateAgentEngineMemoriesConfig config) { + GenerateMemoriesOperation processResponseForPrivateGenerate( + ApiResponse response, GenerateMemoriesConfig config) { ResponseBody responseBody = response.getBody(); String responseString; try { @@ -1033,16 +910,16 @@ AgentEngineGenerateMemoriesOperation processResponseForPrivateGenerate( + " Developer API mode."); } - return JsonSerializable.fromJsonNode(responseNode, AgentEngineGenerateMemoriesOperation.class); + return JsonSerializable.fromJsonNode(responseNode, GenerateMemoriesOperation.class); } - public AgentEngineGenerateMemoriesOperation privateGenerate( + public GenerateMemoriesOperation privateGenerate( String name, GenerateMemoriesRequestVertexSessionSource vertexSessionSource, GenerateMemoriesRequestDirectContentsSource directContentsSource, GenerateMemoriesRequestDirectMemoriesSource directMemoriesSource, Map scope, - GenerateAgentEngineMemoriesConfig config) { + GenerateMemoriesConfig config) { BuiltRequest builtRequest = buildRequestForPrivateGenerate( name, vertexSessionSource, directContentsSource, directMemoriesSource, scope, config); @@ -1055,10 +932,9 @@ public AgentEngineGenerateMemoriesOperation privateGenerate( } /** A shared buildRequest method for both sync and async methods. */ - BuiltRequest buildRequestForGet(String name, GetAgentEngineMemoryConfig config) { + BuiltRequest buildRequestForGet(String name, GetMemoryConfig config) { - GetAgentEngineMemoryRequestParameters.Builder parameterBuilder = - GetAgentEngineMemoryRequestParameters.builder(); + GetMemoryRequestParameters.Builder parameterBuilder = GetMemoryRequestParameters.builder(); if (!Common.isZero(name)) { parameterBuilder.name(name); @@ -1071,7 +947,7 @@ BuiltRequest buildRequestForGet(String name, GetAgentEngineMemoryConfig config) ObjectNode body; String path; if (this.apiClient.vertexAI()) { - body = getAgentEngineMemoryRequestParametersToVertex(parameterNode, null); + body = getMemoryRequestParametersToVertex(parameterNode, null); path = Common.formatMap("{name}", body.get("_url")); } else { throw new UnsupportedOperationException( @@ -1096,7 +972,7 @@ BuiltRequest buildRequestForGet(String name, GetAgentEngineMemoryConfig config) } /** A shared processResponse function for both sync and async methods. */ - Memory processResponseForGet(ApiResponse response, GetAgentEngineMemoryConfig config) { + Memory processResponseForGet(ApiResponse response, GetMemoryConfig config) { ResponseBody responseBody = response.getBody(); String responseString; try { @@ -1116,7 +992,7 @@ Memory processResponseForGet(ApiResponse response, GetAgentEngineMemoryConfig co return JsonSerializable.fromJsonNode(responseNode, Memory.class); } - public Memory get(String name, GetAgentEngineMemoryConfig config) { + public Memory get(String name, GetMemoryConfig config) { BuiltRequest builtRequest = buildRequestForGet(name, config); try (ApiResponse response = @@ -1127,109 +1003,10 @@ public Memory get(String name, GetAgentEngineMemoryConfig config) { } /** A shared buildRequest method for both sync and async methods. */ - BuiltRequest buildRequestForPrivateIngestEvents( - String name, - String streamId, - IngestionDirectContentsSource directContentsSource, - Map scope, - MemoryGenerationTriggerConfig generationTriggerConfig, - IngestEventsConfig config) { - - IngestEventsRequestParameters.Builder parameterBuilder = - IngestEventsRequestParameters.builder(); - - if (!Common.isZero(name)) { - parameterBuilder.name(name); - } - if (!Common.isZero(streamId)) { - parameterBuilder.streamId(streamId); - } - if (!Common.isZero(directContentsSource)) { - parameterBuilder.directContentsSource(directContentsSource); - } - if (!Common.isZero(scope)) { - parameterBuilder.scope(scope); - } - if (!Common.isZero(generationTriggerConfig)) { - parameterBuilder.generationTriggerConfig(generationTriggerConfig); - } - if (!Common.isZero(config)) { - parameterBuilder.config(config); - } - JsonNode parameterNode = JsonSerializable.toJsonNode(parameterBuilder.build()); - - ObjectNode body; - String path; - if (this.apiClient.vertexAI()) { - body = ingestEventsRequestParametersToVertex(parameterNode, null); - path = Common.formatMap("{name}/memories:ingestEvents", body.get("_url")); - } else { - throw new UnsupportedOperationException( - "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini" - + " Developer API mode."); - } - body.remove("_url"); - - JsonNode queryParams = body.get("_query"); - if (queryParams != null) { - body.remove("_query"); - path = String.format("%s?%s", path, Common.urlEncode((ObjectNode) queryParams)); - } - - // TODO: Remove the hack that removes config. - Optional requestHttpOptions = Optional.empty(); - if (config != null) { - requestHttpOptions = config.httpOptions(); - } - - return new BuiltRequest(path, JsonSerializable.toJsonString(body), requestHttpOptions); - } - - /** A shared processResponse function for both sync and async methods. */ - MemoryBankIngestEventsOperation processResponseForPrivateIngestEvents( - ApiResponse response, IngestEventsConfig config) { - ResponseBody responseBody = response.getBody(); - String responseString; - try { - responseString = responseBody.string(); - } catch (IOException e) { - throw new GenAiIOException("Failed to read HTTP response.", e); - } - - JsonNode responseNode = JsonSerializable.stringToJsonNode(responseString); - - if (!this.apiClient.vertexAI()) { - throw new UnsupportedOperationException( - "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini" - + " Developer API mode."); - } - - return JsonSerializable.fromJsonNode(responseNode, MemoryBankIngestEventsOperation.class); - } - - public MemoryBankIngestEventsOperation privateIngestEvents( - String name, - String streamId, - IngestionDirectContentsSource directContentsSource, - Map scope, - MemoryGenerationTriggerConfig generationTriggerConfig, - IngestEventsConfig config) { - BuiltRequest builtRequest = - buildRequestForPrivateIngestEvents( - name, streamId, directContentsSource, scope, generationTriggerConfig, config); - - try (ApiResponse response = - this.apiClient.request( - "post", builtRequest.path(), builtRequest.body(), builtRequest.httpOptions())) { - return processResponseForPrivateIngestEvents(response, config); - } - } - - /** A shared buildRequest method for both sync and async methods. */ - BuiltRequest buildRequestForPrivateList(String name, ListAgentEngineMemoryConfig config) { + BuiltRequest buildRequestForPrivateList(String name, ListMemoriesConfig config) { - ListAgentEngineMemoryRequestParameters.Builder parameterBuilder = - ListAgentEngineMemoryRequestParameters.builder(); + ListMemoriesRequestParameters.Builder parameterBuilder = + ListMemoriesRequestParameters.builder(); if (!Common.isZero(name)) { parameterBuilder.name(name); @@ -1242,7 +1019,7 @@ BuiltRequest buildRequestForPrivateList(String name, ListAgentEngineMemoryConfig ObjectNode body; String path; if (this.apiClient.vertexAI()) { - body = listAgentEngineMemoryRequestParametersToVertex(parameterNode, null); + body = listMemoriesRequestParametersToVertex(parameterNode, null); path = Common.formatMap("{name}/memories", body.get("_url")); } else { throw new UnsupportedOperationException( @@ -1267,8 +1044,8 @@ BuiltRequest buildRequestForPrivateList(String name, ListAgentEngineMemoryConfig } /** A shared processResponse function for both sync and async methods. */ - ListReasoningEnginesMemoriesResponse processResponseForPrivateList( - ApiResponse response, ListAgentEngineMemoryConfig config) { + ListMemoriesResponse processResponseForPrivateList( + ApiResponse response, ListMemoriesConfig config) { ResponseBody responseBody = response.getBody(); String responseString; try { @@ -1285,11 +1062,10 @@ ListReasoningEnginesMemoriesResponse processResponseForPrivateList( + " Developer API mode."); } - return JsonSerializable.fromJsonNode(responseNode, ListReasoningEnginesMemoriesResponse.class); + return JsonSerializable.fromJsonNode(responseNode, ListMemoriesResponse.class); } - public ListReasoningEnginesMemoriesResponse privateList( - String name, ListAgentEngineMemoryConfig config) { + public ListMemoriesResponse privateList(String name, ListMemoriesConfig config) { BuiltRequest builtRequest = buildRequestForPrivateList(name, config); try (ApiResponse response = @@ -1301,10 +1077,9 @@ public ListReasoningEnginesMemoriesResponse privateList( /** A shared buildRequest method for both sync and async methods. */ BuiltRequest buildRequestForPrivateGetMemoryOperation( - String operationName, GetAgentEngineOperationConfig config) { + String operationName, GetMemoryBankOperationConfig config) { - GetAgentEngineMemoryOperationParameters.Builder parameterBuilder = - GetAgentEngineMemoryOperationParameters.builder(); + GetMemoryOperationParameters.Builder parameterBuilder = GetMemoryOperationParameters.builder(); if (!Common.isZero(operationName)) { parameterBuilder.operationName(operationName); @@ -1317,7 +1092,7 @@ BuiltRequest buildRequestForPrivateGetMemoryOperation( ObjectNode body; String path; if (this.apiClient.vertexAI()) { - body = getAgentEngineMemoryOperationParametersToVertex(parameterNode, null); + body = getMemoryOperationParametersToVertex(parameterNode, null); path = Common.formatMap("{operationName}", body.get("_url")); } else { throw new UnsupportedOperationException( @@ -1342,8 +1117,8 @@ BuiltRequest buildRequestForPrivateGetMemoryOperation( } /** A shared processResponse function for both sync and async methods. */ - AgentEngineMemoryOperation processResponseForPrivateGetMemoryOperation( - ApiResponse response, GetAgentEngineOperationConfig config) { + MemoryOperation processResponseForPrivateGetMemoryOperation( + ApiResponse response, GetMemoryBankOperationConfig config) { ResponseBody responseBody = response.getBody(); String responseString; try { @@ -1360,11 +1135,11 @@ AgentEngineMemoryOperation processResponseForPrivateGetMemoryOperation( + " Developer API mode."); } - return JsonSerializable.fromJsonNode(responseNode, AgentEngineMemoryOperation.class); + return JsonSerializable.fromJsonNode(responseNode, MemoryOperation.class); } - public AgentEngineMemoryOperation privateGetMemoryOperation( - String operationName, GetAgentEngineOperationConfig config) { + public MemoryOperation privateGetMemoryOperation( + String operationName, GetMemoryBankOperationConfig config) { BuiltRequest builtRequest = buildRequestForPrivateGetMemoryOperation(operationName, config); try (ApiResponse response = @@ -1376,10 +1151,10 @@ public AgentEngineMemoryOperation privateGetMemoryOperation( /** A shared buildRequest method for both sync and async methods. */ BuiltRequest buildRequestForPrivateGetGenerateMemoriesOperation( - String operationName, GetAgentEngineOperationConfig config) { + String operationName, GetMemoryBankOperationConfig config) { - GetAgentEngineGenerateMemoriesOperationParameters.Builder parameterBuilder = - GetAgentEngineGenerateMemoriesOperationParameters.builder(); + GetGenerateMemoriesOperationParameters.Builder parameterBuilder = + GetGenerateMemoriesOperationParameters.builder(); if (!Common.isZero(operationName)) { parameterBuilder.operationName(operationName); @@ -1392,7 +1167,7 @@ BuiltRequest buildRequestForPrivateGetGenerateMemoriesOperation( ObjectNode body; String path; if (this.apiClient.vertexAI()) { - body = getAgentEngineGenerateMemoriesOperationParametersToVertex(parameterNode, null); + body = getGenerateMemoriesOperationParametersToVertex(parameterNode, null); path = Common.formatMap("{operationName}", body.get("_url")); } else { throw new UnsupportedOperationException( @@ -1417,8 +1192,8 @@ BuiltRequest buildRequestForPrivateGetGenerateMemoriesOperation( } /** A shared processResponse function for both sync and async methods. */ - AgentEngineGenerateMemoriesOperation processResponseForPrivateGetGenerateMemoriesOperation( - ApiResponse response, GetAgentEngineOperationConfig config) { + GenerateMemoriesOperation processResponseForPrivateGetGenerateMemoriesOperation( + ApiResponse response, GetMemoryBankOperationConfig config) { ResponseBody responseBody = response.getBody(); String responseString; try { @@ -1435,11 +1210,11 @@ AgentEngineGenerateMemoriesOperation processResponseForPrivateGetGenerateMemorie + " Developer API mode."); } - return JsonSerializable.fromJsonNode(responseNode, AgentEngineGenerateMemoriesOperation.class); + return JsonSerializable.fromJsonNode(responseNode, GenerateMemoriesOperation.class); } - public AgentEngineGenerateMemoriesOperation privateGetGenerateMemoriesOperation( - String operationName, GetAgentEngineOperationConfig config) { + public GenerateMemoriesOperation privateGetGenerateMemoriesOperation( + String operationName, GetMemoryBankOperationConfig config) { BuiltRequest builtRequest = buildRequestForPrivateGetGenerateMemoriesOperation(operationName, config); @@ -1456,10 +1231,10 @@ BuiltRequest buildRequestForPrivateRetrieve( Map scope, RetrieveMemoriesRequestSimilaritySearchParams similaritySearchParams, RetrieveMemoriesRequestSimpleRetrievalParams simpleRetrievalParams, - RetrieveAgentEngineMemoriesConfig config) { + RetrieveMemoriesConfig config) { - RetrieveAgentEngineMemoriesRequestParameters.Builder parameterBuilder = - RetrieveAgentEngineMemoriesRequestParameters.builder(); + RetrieveMemoriesRequestParameters.Builder parameterBuilder = + RetrieveMemoriesRequestParameters.builder(); if (!Common.isZero(name)) { parameterBuilder.name(name); @@ -1481,7 +1256,7 @@ BuiltRequest buildRequestForPrivateRetrieve( ObjectNode body; String path; if (this.apiClient.vertexAI()) { - body = retrieveAgentEngineMemoriesRequestParametersToVertex(parameterNode, null); + body = retrieveMemoriesRequestParametersToVertex(parameterNode, null); path = Common.formatMap("{name}/memories:retrieve", body.get("_url")); } else { throw new UnsupportedOperationException( @@ -1507,7 +1282,7 @@ BuiltRequest buildRequestForPrivateRetrieve( /** A shared processResponse function for both sync and async methods. */ RetrieveMemoriesResponse processResponseForPrivateRetrieve( - ApiResponse response, RetrieveAgentEngineMemoriesConfig config) { + ApiResponse response, RetrieveMemoriesConfig config) { ResponseBody responseBody = response.getBody(); String responseString; try { @@ -1532,7 +1307,7 @@ public RetrieveMemoriesResponse privateRetrieve( Map scope, RetrieveMemoriesRequestSimilaritySearchParams similaritySearchParams, RetrieveMemoriesRequestSimpleRetrievalParams simpleRetrievalParams, - RetrieveAgentEngineMemoriesConfig config) { + RetrieveMemoriesConfig config) { BuiltRequest builtRequest = buildRequestForPrivateRetrieve( name, scope, similaritySearchParams, simpleRetrievalParams, config); @@ -1624,10 +1399,10 @@ public RetrieveProfilesResponse retrieveProfiles( /** A shared buildRequest method for both sync and async methods. */ BuiltRequest buildRequestForPrivateRollback( - String name, String targetRevisionId, RollbackAgentEngineMemoryConfig config) { + String name, String targetRevisionId, RollbackMemoryConfig config) { - RollbackAgentEngineMemoryRequestParameters.Builder parameterBuilder = - RollbackAgentEngineMemoryRequestParameters.builder(); + RollbackMemoryRequestParameters.Builder parameterBuilder = + RollbackMemoryRequestParameters.builder(); if (!Common.isZero(name)) { parameterBuilder.name(name); @@ -1643,7 +1418,7 @@ BuiltRequest buildRequestForPrivateRollback( ObjectNode body; String path; if (this.apiClient.vertexAI()) { - body = rollbackAgentEngineMemoryRequestParametersToVertex(parameterNode, null); + body = rollbackMemoryRequestParametersToVertex(parameterNode, null); path = Common.formatMap("{name}:rollback", body.get("_url")); } else { throw new UnsupportedOperationException( @@ -1668,8 +1443,8 @@ BuiltRequest buildRequestForPrivateRollback( } /** A shared processResponse function for both sync and async methods. */ - AgentEngineRollbackMemoryOperation processResponseForPrivateRollback( - ApiResponse response, RollbackAgentEngineMemoryConfig config) { + RollbackMemoryOperation processResponseForPrivateRollback( + ApiResponse response, RollbackMemoryConfig config) { ResponseBody responseBody = response.getBody(); String responseString; try { @@ -1686,11 +1461,11 @@ AgentEngineRollbackMemoryOperation processResponseForPrivateRollback( + " Developer API mode."); } - return JsonSerializable.fromJsonNode(responseNode, AgentEngineRollbackMemoryOperation.class); + return JsonSerializable.fromJsonNode(responseNode, RollbackMemoryOperation.class); } - public AgentEngineRollbackMemoryOperation privateRollback( - String name, String targetRevisionId, RollbackAgentEngineMemoryConfig config) { + public RollbackMemoryOperation privateRollback( + String name, String targetRevisionId, RollbackMemoryConfig config) { BuiltRequest builtRequest = buildRequestForPrivateRollback(name, targetRevisionId, config); try (ApiResponse response = @@ -1702,10 +1477,10 @@ public AgentEngineRollbackMemoryOperation privateRollback( /** A shared buildRequest method for both sync and async methods. */ BuiltRequest buildRequestForPrivateUpdate( - String name, String fact, Map scope, UpdateAgentEngineMemoryConfig config) { + String name, String fact, Map scope, UpdateMemoryConfig config) { - UpdateAgentEngineMemoryRequestParameters.Builder parameterBuilder = - UpdateAgentEngineMemoryRequestParameters.builder(); + UpdateMemoryRequestParameters.Builder parameterBuilder = + UpdateMemoryRequestParameters.builder(); if (!Common.isZero(name)) { parameterBuilder.name(name); @@ -1724,7 +1499,7 @@ BuiltRequest buildRequestForPrivateUpdate( ObjectNode body; String path; if (this.apiClient.vertexAI()) { - body = updateAgentEngineMemoryRequestParametersToVertex(parameterNode, null); + body = updateMemoryRequestParametersToVertex(parameterNode, null); path = Common.formatMap("{name}", body.get("_url")); } else { throw new UnsupportedOperationException( @@ -1749,8 +1524,7 @@ BuiltRequest buildRequestForPrivateUpdate( } /** A shared processResponse function for both sync and async methods. */ - AgentEngineMemoryOperation processResponseForPrivateUpdate( - ApiResponse response, UpdateAgentEngineMemoryConfig config) { + MemoryOperation processResponseForPrivateUpdate(ApiResponse response, UpdateMemoryConfig config) { ResponseBody responseBody = response.getBody(); String responseString; try { @@ -1767,11 +1541,11 @@ AgentEngineMemoryOperation processResponseForPrivateUpdate( + " Developer API mode."); } - return JsonSerializable.fromJsonNode(responseNode, AgentEngineMemoryOperation.class); + return JsonSerializable.fromJsonNode(responseNode, MemoryOperation.class); } - public AgentEngineMemoryOperation privateUpdate( - String name, String fact, Map scope, UpdateAgentEngineMemoryConfig config) { + public MemoryOperation privateUpdate( + String name, String fact, Map scope, UpdateMemoryConfig config) { BuiltRequest builtRequest = buildRequestForPrivateUpdate(name, fact, scope, config); try (ApiResponse response = @@ -1787,10 +1561,10 @@ BuiltRequest buildRequestForPrivatePurge( String filter, List filterGroups, boolean force, - PurgeAgentEngineMemoriesConfig config) { + PurgeMemoriesConfig config) { - PurgeAgentEngineMemoriesRequestParameters.Builder parameterBuilder = - PurgeAgentEngineMemoriesRequestParameters.builder(); + PurgeMemoriesRequestParameters.Builder parameterBuilder = + PurgeMemoriesRequestParameters.builder(); if (!Common.isZero(name)) { parameterBuilder.name(name); @@ -1812,7 +1586,7 @@ BuiltRequest buildRequestForPrivatePurge( ObjectNode body; String path; if (this.apiClient.vertexAI()) { - body = purgeAgentEngineMemoriesRequestParametersToVertex(parameterNode, null); + body = purgeMemoriesRequestParametersToVertex(parameterNode, null); path = Common.formatMap("{name}/memories:purge", body.get("_url")); } else { throw new UnsupportedOperationException( @@ -1837,8 +1611,8 @@ BuiltRequest buildRequestForPrivatePurge( } /** A shared processResponse function for both sync and async methods. */ - AgentEnginePurgeMemoriesOperation processResponseForPrivatePurge( - ApiResponse response, PurgeAgentEngineMemoriesConfig config) { + PurgeMemoriesOperation processResponseForPrivatePurge( + ApiResponse response, PurgeMemoriesConfig config) { ResponseBody responseBody = response.getBody(); String responseString; try { @@ -1855,15 +1629,15 @@ AgentEnginePurgeMemoriesOperation processResponseForPrivatePurge( + " Developer API mode."); } - return JsonSerializable.fromJsonNode(responseNode, AgentEnginePurgeMemoriesOperation.class); + return JsonSerializable.fromJsonNode(responseNode, PurgeMemoriesOperation.class); } - public AgentEnginePurgeMemoriesOperation privatePurge( + public PurgeMemoriesOperation privatePurge( String name, String filter, List filterGroups, boolean force, - PurgeAgentEngineMemoriesConfig config) { + PurgeMemoriesConfig config) { BuiltRequest builtRequest = buildRequestForPrivatePurge(name, filter, filterGroups, force, config); diff --git a/src/main/java/com/google/cloud/agentplatform/MemoryBanks.java b/src/main/java/com/google/cloud/agentplatform/MemoryBanks.java new file mode 100644 index 0000000..d5092a1 --- /dev/null +++ b/src/main/java/com/google/cloud/agentplatform/MemoryBanks.java @@ -0,0 +1,527 @@ +/* + * 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.fasterxml.jackson.databind.JsonNode; +import com.fasterxml.jackson.databind.node.ObjectNode; +import com.google.cloud.agentplatform.types.CreateMemoryBankConfig; +import com.google.cloud.agentplatform.types.CreateMemoryBankRequestParameters; +import com.google.cloud.agentplatform.types.DeleteMemoryBankConfig; +import com.google.cloud.agentplatform.types.DeleteMemoryBankOperation; +import com.google.cloud.agentplatform.types.DeleteMemoryBankRequestParameters; +import com.google.cloud.agentplatform.types.GetMemoryBankOperationConfig; +import com.google.cloud.agentplatform.types.GetMemoryBankOperationParameters; +import com.google.cloud.agentplatform.types.IngestEventsConfig; +import com.google.cloud.agentplatform.types.IngestEventsRequestParameters; +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; +import com.google.genai.Common.BuiltRequest; +import com.google.genai.JsonSerializable; +import com.google.genai.errors.GenAiIOException; +import com.google.genai.types.HttpOptions; +import java.io.IOException; +import java.util.Map; +import java.util.Optional; +import okhttp3.ResponseBody; + +public final class MemoryBanks { + public final Memories memories; + + final ApiClient apiClient; + + public MemoryBanks(ApiClient apiClient) { + this.apiClient = apiClient; + this.memories = new Memories(apiClient); + } + + @ExcludeFromGeneratedCoverageReport + ObjectNode createMemoryBankRequestParametersToVertex( + JsonNode fromObject, ObjectNode parentObject) { + ObjectNode toObject = JsonSerializable.objectMapper().createObjectNode(); + + return toObject; + } + + @ExcludeFromGeneratedCoverageReport + ObjectNode deleteMemoryBankRequestParametersToVertex( + JsonNode fromObject, ObjectNode parentObject) { + ObjectNode toObject = JsonSerializable.objectMapper().createObjectNode(); + if (Common.getValueByPath(fromObject, new String[] {"name"}) != null) { + Common.setValueByPath( + toObject, + new String[] {"_url", "name"}, + Common.getValueByPath(fromObject, new String[] {"name"})); + } + + if (Common.getValueByPath(fromObject, new String[] {"force"}) != null) { + Common.setValueByPath( + toObject, + new String[] {"force"}, + Common.getValueByPath(fromObject, new String[] {"force"})); + } + + return toObject; + } + + @ExcludeFromGeneratedCoverageReport + ObjectNode getMemoryBankOperationParametersToVertex( + JsonNode fromObject, ObjectNode parentObject) { + ObjectNode toObject = JsonSerializable.objectMapper().createObjectNode(); + if (Common.getValueByPath(fromObject, new String[] {"operationName"}) != null) { + Common.setValueByPath( + toObject, + new String[] {"_url", "operationName"}, + Common.getValueByPath(fromObject, new String[] {"operationName"})); + } + + return toObject; + } + + @ExcludeFromGeneratedCoverageReport + ObjectNode ingestEventsConfigToVertex(JsonNode fromObject, ObjectNode parentObject) { + ObjectNode toObject = JsonSerializable.objectMapper().createObjectNode(); + + if (Common.getValueByPath(fromObject, new String[] {"forceFlush"}) != null) { + Common.setValueByPath( + parentObject, + new String[] {"forceFlush"}, + Common.getValueByPath(fromObject, new String[] {"forceFlush"})); + } + + if (Common.getValueByPath(fromObject, new String[] {"revisionLabels"}) != null) { + Common.setValueByPath( + parentObject, + new String[] {"revisionLabels"}, + Common.getValueByPath(fromObject, new String[] {"revisionLabels"})); + } + + if (Common.getValueByPath(fromObject, new String[] {"revisionExpireTime"}) != null) { + Common.setValueByPath( + parentObject, + new String[] {"revisionExpireTime"}, + Common.getValueByPath(fromObject, new String[] {"revisionExpireTime"})); + } + + if (Common.getValueByPath(fromObject, new String[] {"revisionTtl"}) != null) { + Common.setValueByPath( + parentObject, + new String[] {"revisionTtl"}, + Common.getValueByPath(fromObject, new String[] {"revisionTtl"})); + } + + if (Common.getValueByPath(fromObject, new String[] {"disableMemoryRevisions"}) != null) { + Common.setValueByPath( + parentObject, + new String[] {"disableMemoryRevisions"}, + Common.getValueByPath(fromObject, new String[] {"disableMemoryRevisions"})); + } + + if (Common.getValueByPath(fromObject, new String[] {"metadata"}) != null) { + Common.setValueByPath( + parentObject, + new String[] {"metadata"}, + Common.getValueByPath(fromObject, new String[] {"metadata"})); + } + + if (Common.getValueByPath(fromObject, new String[] {"metadataMergeStrategy"}) != null) { + Common.setValueByPath( + parentObject, + new String[] {"metadataMergeStrategy"}, + Common.getValueByPath(fromObject, new String[] {"metadataMergeStrategy"})); + } + + return toObject; + } + + @ExcludeFromGeneratedCoverageReport + ObjectNode ingestEventsRequestParametersToVertex(JsonNode fromObject, ObjectNode parentObject) { + ObjectNode toObject = JsonSerializable.objectMapper().createObjectNode(); + if (Common.getValueByPath(fromObject, new String[] {"name"}) != null) { + Common.setValueByPath( + toObject, + new String[] {"_url", "name"}, + Common.getValueByPath(fromObject, new String[] {"name"})); + } + + if (Common.getValueByPath(fromObject, new String[] {"streamId"}) != null) { + Common.setValueByPath( + toObject, + new String[] {"streamId"}, + Common.getValueByPath(fromObject, new String[] {"streamId"})); + } + + if (Common.getValueByPath(fromObject, new String[] {"directContentsSource"}) != null) { + Common.setValueByPath( + toObject, + new String[] {"directContentsSource"}, + Common.getValueByPath(fromObject, new String[] {"directContentsSource"})); + } + + if (Common.getValueByPath(fromObject, new String[] {"scope"}) != null) { + Common.setValueByPath( + toObject, + new String[] {"scope"}, + Common.getValueByPath(fromObject, new String[] {"scope"})); + } + + if (Common.getValueByPath(fromObject, new String[] {"generationTriggerConfig"}) != null) { + Common.setValueByPath( + toObject, + new String[] {"generationTriggerConfig"}, + Common.getValueByPath(fromObject, new String[] {"generationTriggerConfig"})); + } + + if (Common.getValueByPath(fromObject, new String[] {"config"}) != null) { + JsonNode unused = + ingestEventsConfigToVertex( + JsonSerializable.toJsonNode( + Common.getValueByPath(fromObject, new String[] {"config"})), + toObject); + } + + return toObject; + } + + /** A shared buildRequest method for both sync and async methods. */ + BuiltRequest buildRequestForPrivateCreate(CreateMemoryBankConfig config) { + + CreateMemoryBankRequestParameters.Builder parameterBuilder = + CreateMemoryBankRequestParameters.builder(); + + if (!Common.isZero(config)) { + parameterBuilder.config(config); + } + JsonNode parameterNode = JsonSerializable.toJsonNode(parameterBuilder.build()); + + ObjectNode body; + String path; + if (this.apiClient.vertexAI()) { + body = createMemoryBankRequestParametersToVertex(parameterNode, null); + path = Common.formatMap("reasoningEngines", body.get("_url")); + } else { + throw new UnsupportedOperationException( + "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini" + + " Developer API mode."); + } + body.remove("_url"); + + JsonNode queryParams = body.get("_query"); + if (queryParams != null) { + body.remove("_query"); + path = String.format("%s?%s", path, Common.urlEncode((ObjectNode) queryParams)); + } + + // TODO: Remove the hack that removes config. + Optional requestHttpOptions = Optional.empty(); + if (config != null) { + requestHttpOptions = config.httpOptions(); + } + + return new BuiltRequest(path, JsonSerializable.toJsonString(body), requestHttpOptions); + } + + /** A shared processResponse function for both sync and async methods. */ + MemoryBankOperation processResponseForPrivateCreate( + ApiResponse response, CreateMemoryBankConfig config) { + ResponseBody responseBody = response.getBody(); + String responseString; + try { + responseString = responseBody.string(); + } catch (IOException e) { + throw new GenAiIOException("Failed to read HTTP response.", e); + } + + JsonNode responseNode = JsonSerializable.stringToJsonNode(responseString); + + if (!this.apiClient.vertexAI()) { + throw new UnsupportedOperationException( + "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini" + + " Developer API mode."); + } + + return JsonSerializable.fromJsonNode(responseNode, MemoryBankOperation.class); + } + + public MemoryBankOperation privateCreate(CreateMemoryBankConfig config) { + BuiltRequest builtRequest = buildRequestForPrivateCreate(config); + + try (ApiResponse response = + this.apiClient.request( + "post", builtRequest.path(), builtRequest.body(), builtRequest.httpOptions())) { + return processResponseForPrivateCreate(response, config); + } + } + + /** A shared buildRequest method for both sync and async methods. */ + BuiltRequest buildRequestForPrivateDelete( + String name, boolean force, DeleteMemoryBankConfig config) { + + DeleteMemoryBankRequestParameters.Builder parameterBuilder = + DeleteMemoryBankRequestParameters.builder(); + + if (!Common.isZero(name)) { + parameterBuilder.name(name); + } + if (!Common.isZero(force)) { + parameterBuilder.force(force); + } + if (!Common.isZero(config)) { + parameterBuilder.config(config); + } + JsonNode parameterNode = JsonSerializable.toJsonNode(parameterBuilder.build()); + + ObjectNode body; + String path; + if (this.apiClient.vertexAI()) { + body = deleteMemoryBankRequestParametersToVertex(parameterNode, null); + path = Common.formatMap("{name}", body.get("_url")); + } else { + throw new UnsupportedOperationException( + "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini" + + " Developer API mode."); + } + body.remove("_url"); + + JsonNode queryParams = body.get("_query"); + if (queryParams != null) { + body.remove("_query"); + path = String.format("%s?%s", path, Common.urlEncode((ObjectNode) queryParams)); + } + + // TODO: Remove the hack that removes config. + Optional requestHttpOptions = Optional.empty(); + if (config != null) { + requestHttpOptions = config.httpOptions(); + } + + return new BuiltRequest(path, JsonSerializable.toJsonString(body), requestHttpOptions); + } + + /** A shared processResponse function for both sync and async methods. */ + DeleteMemoryBankOperation processResponseForPrivateDelete( + ApiResponse response, DeleteMemoryBankConfig config) { + ResponseBody responseBody = response.getBody(); + String responseString; + try { + responseString = responseBody.string(); + } catch (IOException e) { + throw new GenAiIOException("Failed to read HTTP response.", e); + } + + JsonNode responseNode = JsonSerializable.stringToJsonNode(responseString); + + if (!this.apiClient.vertexAI()) { + throw new UnsupportedOperationException( + "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini" + + " Developer API mode."); + } + + return JsonSerializable.fromJsonNode(responseNode, DeleteMemoryBankOperation.class); + } + + public DeleteMemoryBankOperation privateDelete( + String name, boolean force, DeleteMemoryBankConfig config) { + BuiltRequest builtRequest = buildRequestForPrivateDelete(name, force, config); + + try (ApiResponse response = + this.apiClient.request( + "delete", builtRequest.path(), builtRequest.body(), builtRequest.httpOptions())) { + return processResponseForPrivateDelete(response, config); + } + } + + /** A shared buildRequest method for both sync and async methods. */ + BuiltRequest buildRequestForPrivateIngestEvents( + String name, + String streamId, + IngestionDirectContentsSource directContentsSource, + Map scope, + MemoryGenerationTriggerConfig generationTriggerConfig, + IngestEventsConfig config) { + + IngestEventsRequestParameters.Builder parameterBuilder = + IngestEventsRequestParameters.builder(); + + if (!Common.isZero(name)) { + parameterBuilder.name(name); + } + if (!Common.isZero(streamId)) { + parameterBuilder.streamId(streamId); + } + if (!Common.isZero(directContentsSource)) { + parameterBuilder.directContentsSource(directContentsSource); + } + if (!Common.isZero(scope)) { + parameterBuilder.scope(scope); + } + if (!Common.isZero(generationTriggerConfig)) { + parameterBuilder.generationTriggerConfig(generationTriggerConfig); + } + if (!Common.isZero(config)) { + parameterBuilder.config(config); + } + JsonNode parameterNode = JsonSerializable.toJsonNode(parameterBuilder.build()); + + ObjectNode body; + String path; + if (this.apiClient.vertexAI()) { + body = ingestEventsRequestParametersToVertex(parameterNode, null); + path = Common.formatMap("{name}/memories:ingestEvents", body.get("_url")); + } else { + throw new UnsupportedOperationException( + "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini" + + " Developer API mode."); + } + body.remove("_url"); + + JsonNode queryParams = body.get("_query"); + if (queryParams != null) { + body.remove("_query"); + path = String.format("%s?%s", path, Common.urlEncode((ObjectNode) queryParams)); + } + + // TODO: Remove the hack that removes config. + Optional requestHttpOptions = Optional.empty(); + if (config != null) { + requestHttpOptions = config.httpOptions(); + } + + return new BuiltRequest(path, JsonSerializable.toJsonString(body), requestHttpOptions); + } + + /** A shared processResponse function for both sync and async methods. */ + MemoryBankIngestEventsOperation processResponseForPrivateIngestEvents( + ApiResponse response, IngestEventsConfig config) { + ResponseBody responseBody = response.getBody(); + String responseString; + try { + responseString = responseBody.string(); + } catch (IOException e) { + throw new GenAiIOException("Failed to read HTTP response.", e); + } + + JsonNode responseNode = JsonSerializable.stringToJsonNode(responseString); + + if (!this.apiClient.vertexAI()) { + throw new UnsupportedOperationException( + "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini" + + " Developer API mode."); + } + + return JsonSerializable.fromJsonNode(responseNode, MemoryBankIngestEventsOperation.class); + } + + public MemoryBankIngestEventsOperation privateIngestEvents( + String name, + String streamId, + IngestionDirectContentsSource directContentsSource, + Map scope, + MemoryGenerationTriggerConfig generationTriggerConfig, + IngestEventsConfig config) { + BuiltRequest builtRequest = + buildRequestForPrivateIngestEvents( + name, streamId, directContentsSource, scope, generationTriggerConfig, config); + + try (ApiResponse response = + this.apiClient.request( + "post", builtRequest.path(), builtRequest.body(), builtRequest.httpOptions())) { + return processResponseForPrivateIngestEvents(response, config); + } + } + + /** A shared buildRequest method for both sync and async methods. */ + BuiltRequest buildRequestForPrivateGetMemoryBankOperation( + String operationName, GetMemoryBankOperationConfig config) { + + GetMemoryBankOperationParameters.Builder parameterBuilder = + GetMemoryBankOperationParameters.builder(); + + if (!Common.isZero(operationName)) { + parameterBuilder.operationName(operationName); + } + if (!Common.isZero(config)) { + parameterBuilder.config(config); + } + JsonNode parameterNode = JsonSerializable.toJsonNode(parameterBuilder.build()); + + ObjectNode body; + String path; + if (this.apiClient.vertexAI()) { + body = getMemoryBankOperationParametersToVertex(parameterNode, null); + path = Common.formatMap("{operationName}", body.get("_url")); + } else { + throw new UnsupportedOperationException( + "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini" + + " Developer API mode."); + } + body.remove("_url"); + + JsonNode queryParams = body.get("_query"); + if (queryParams != null) { + body.remove("_query"); + path = String.format("%s?%s", path, Common.urlEncode((ObjectNode) queryParams)); + } + + // TODO: Remove the hack that removes config. + Optional requestHttpOptions = Optional.empty(); + if (config != null) { + requestHttpOptions = config.httpOptions(); + } + + return new BuiltRequest(path, JsonSerializable.toJsonString(body), requestHttpOptions); + } + + /** A shared processResponse function for both sync and async methods. */ + MemoryBankOperation processResponseForPrivateGetMemoryBankOperation( + ApiResponse response, GetMemoryBankOperationConfig config) { + ResponseBody responseBody = response.getBody(); + String responseString; + try { + responseString = responseBody.string(); + } catch (IOException e) { + throw new GenAiIOException("Failed to read HTTP response.", e); + } + + JsonNode responseNode = JsonSerializable.stringToJsonNode(responseString); + + if (!this.apiClient.vertexAI()) { + throw new UnsupportedOperationException( + "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini" + + " Developer API mode."); + } + + return JsonSerializable.fromJsonNode(responseNode, MemoryBankOperation.class); + } + + public MemoryBankOperation privateGetMemoryBankOperation( + String operationName, GetMemoryBankOperationConfig config) { + BuiltRequest builtRequest = buildRequestForPrivateGetMemoryBankOperation(operationName, config); + + try (ApiResponse response = + this.apiClient.request( + "get", builtRequest.path(), builtRequest.body(), builtRequest.httpOptions())) { + return processResponseForPrivateGetMemoryBankOperation(response, config); + } + } +} diff --git a/src/main/java/com/google/cloud/agentplatform/MemoryRevisions.java b/src/main/java/com/google/cloud/agentplatform/MemoryRevisions.java index 243f499..5688734 100644 --- a/src/main/java/com/google/cloud/agentplatform/MemoryRevisions.java +++ b/src/main/java/com/google/cloud/agentplatform/MemoryRevisions.java @@ -20,11 +20,11 @@ import com.fasterxml.jackson.databind.JsonNode; import com.fasterxml.jackson.databind.node.ObjectNode; -import com.google.cloud.agentplatform.types.GetAgentEngineMemoryRevisionConfig; -import com.google.cloud.agentplatform.types.GetAgentEngineMemoryRevisionRequestParameters; -import com.google.cloud.agentplatform.types.ListAgentEngineMemoryRevisionsConfig; -import com.google.cloud.agentplatform.types.ListAgentEngineMemoryRevisionsRequestParameters; -import com.google.cloud.agentplatform.types.ListAgentEngineMemoryRevisionsResponse; +import com.google.cloud.agentplatform.types.GetMemoryRevisionConfig; +import com.google.cloud.agentplatform.types.GetMemoryRevisionRequestParameters; +import com.google.cloud.agentplatform.types.ListMemoryRevisionsConfig; +import com.google.cloud.agentplatform.types.ListMemoryRevisionsRequestParameters; +import com.google.cloud.agentplatform.types.ListMemoryRevisionsResponse; import com.google.cloud.agentplatform.types.MemoryRevision; import com.google.genai.ApiClient; import com.google.genai.ApiResponse; @@ -46,7 +46,7 @@ public MemoryRevisions(ApiClient apiClient) { } @ExcludeFromGeneratedCoverageReport - ObjectNode getAgentEngineMemoryRevisionRequestParametersToVertex( + ObjectNode getMemoryRevisionRequestParametersToVertex( JsonNode fromObject, ObjectNode parentObject) { ObjectNode toObject = JsonSerializable.objectMapper().createObjectNode(); if (Common.getValueByPath(fromObject, new String[] {"name"}) != null) { @@ -60,8 +60,7 @@ ObjectNode getAgentEngineMemoryRevisionRequestParametersToVertex( } @ExcludeFromGeneratedCoverageReport - ObjectNode listAgentEngineMemoryRevisionsConfigToVertex( - JsonNode fromObject, ObjectNode parentObject) { + ObjectNode listMemoryRevisionsConfigToVertex(JsonNode fromObject, ObjectNode parentObject) { ObjectNode toObject = JsonSerializable.objectMapper().createObjectNode(); if (Common.getValueByPath(fromObject, new String[] {"pageSize"}) != null) { @@ -89,7 +88,7 @@ ObjectNode listAgentEngineMemoryRevisionsConfigToVertex( } @ExcludeFromGeneratedCoverageReport - ObjectNode listAgentEngineMemoryRevisionsRequestParametersToVertex( + ObjectNode listMemoryRevisionsRequestParametersToVertex( JsonNode fromObject, ObjectNode parentObject) { ObjectNode toObject = JsonSerializable.objectMapper().createObjectNode(); if (Common.getValueByPath(fromObject, new String[] {"name"}) != null) { @@ -101,7 +100,7 @@ ObjectNode listAgentEngineMemoryRevisionsRequestParametersToVertex( if (Common.getValueByPath(fromObject, new String[] {"config"}) != null) { JsonNode unused = - listAgentEngineMemoryRevisionsConfigToVertex( + listMemoryRevisionsConfigToVertex( JsonSerializable.toJsonNode( Common.getValueByPath(fromObject, new String[] {"config"})), toObject); @@ -111,10 +110,10 @@ ObjectNode listAgentEngineMemoryRevisionsRequestParametersToVertex( } /** A shared buildRequest method for both sync and async methods. */ - BuiltRequest buildRequestForGet(String name, GetAgentEngineMemoryRevisionConfig config) { + BuiltRequest buildRequestForGet(String name, GetMemoryRevisionConfig config) { - GetAgentEngineMemoryRevisionRequestParameters.Builder parameterBuilder = - GetAgentEngineMemoryRevisionRequestParameters.builder(); + GetMemoryRevisionRequestParameters.Builder parameterBuilder = + GetMemoryRevisionRequestParameters.builder(); if (!Common.isZero(name)) { parameterBuilder.name(name); @@ -127,7 +126,7 @@ BuiltRequest buildRequestForGet(String name, GetAgentEngineMemoryRevisionConfig ObjectNode body; String path; if (this.apiClient.vertexAI()) { - body = getAgentEngineMemoryRevisionRequestParametersToVertex(parameterNode, null); + body = getMemoryRevisionRequestParametersToVertex(parameterNode, null); path = Common.formatMap("{name}", body.get("_url")); } else { throw new UnsupportedOperationException( @@ -152,8 +151,7 @@ BuiltRequest buildRequestForGet(String name, GetAgentEngineMemoryRevisionConfig } /** A shared processResponse function for both sync and async methods. */ - MemoryRevision processResponseForGet( - ApiResponse response, GetAgentEngineMemoryRevisionConfig config) { + MemoryRevision processResponseForGet(ApiResponse response, GetMemoryRevisionConfig config) { ResponseBody responseBody = response.getBody(); String responseString; try { @@ -173,7 +171,7 @@ MemoryRevision processResponseForGet( return JsonSerializable.fromJsonNode(responseNode, MemoryRevision.class); } - public MemoryRevision get(String name, GetAgentEngineMemoryRevisionConfig config) { + public MemoryRevision get(String name, GetMemoryRevisionConfig config) { BuiltRequest builtRequest = buildRequestForGet(name, config); try (ApiResponse response = @@ -184,11 +182,10 @@ public MemoryRevision get(String name, GetAgentEngineMemoryRevisionConfig config } /** A shared buildRequest method for both sync and async methods. */ - BuiltRequest buildRequestForPrivateList( - String name, ListAgentEngineMemoryRevisionsConfig config) { + BuiltRequest buildRequestForPrivateList(String name, ListMemoryRevisionsConfig config) { - ListAgentEngineMemoryRevisionsRequestParameters.Builder parameterBuilder = - ListAgentEngineMemoryRevisionsRequestParameters.builder(); + ListMemoryRevisionsRequestParameters.Builder parameterBuilder = + ListMemoryRevisionsRequestParameters.builder(); if (!Common.isZero(name)) { parameterBuilder.name(name); @@ -201,7 +198,7 @@ BuiltRequest buildRequestForPrivateList( ObjectNode body; String path; if (this.apiClient.vertexAI()) { - body = listAgentEngineMemoryRevisionsRequestParametersToVertex(parameterNode, null); + body = listMemoryRevisionsRequestParametersToVertex(parameterNode, null); path = Common.formatMap("{name}/revisions", body.get("_url")); } else { throw new UnsupportedOperationException( @@ -226,8 +223,8 @@ BuiltRequest buildRequestForPrivateList( } /** A shared processResponse function for both sync and async methods. */ - ListAgentEngineMemoryRevisionsResponse processResponseForPrivateList( - ApiResponse response, ListAgentEngineMemoryRevisionsConfig config) { + ListMemoryRevisionsResponse processResponseForPrivateList( + ApiResponse response, ListMemoryRevisionsConfig config) { ResponseBody responseBody = response.getBody(); String responseString; try { @@ -244,12 +241,10 @@ ListAgentEngineMemoryRevisionsResponse processResponseForPrivateList( + " Developer API mode."); } - return JsonSerializable.fromJsonNode( - responseNode, ListAgentEngineMemoryRevisionsResponse.class); + return JsonSerializable.fromJsonNode(responseNode, ListMemoryRevisionsResponse.class); } - public ListAgentEngineMemoryRevisionsResponse privateList( - String name, ListAgentEngineMemoryRevisionsConfig config) { + public ListMemoryRevisionsResponse privateList(String name, ListMemoryRevisionsConfig config) { BuiltRequest builtRequest = buildRequestForPrivateList(name, config); try (ApiResponse response = diff --git a/src/main/java/com/google/cloud/agentplatform/types/GetAgentEngineMemoryConfig.java b/src/main/java/com/google/cloud/agentplatform/types/CreateMemoryBankConfig.java similarity index 71% rename from src/main/java/com/google/cloud/agentplatform/types/GetAgentEngineMemoryConfig.java rename to src/main/java/com/google/cloud/agentplatform/types/CreateMemoryBankConfig.java index 6176b77..9d2f7f0 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/GetAgentEngineMemoryConfig.java +++ b/src/main/java/com/google/cloud/agentplatform/types/CreateMemoryBankConfig.java @@ -27,30 +27,30 @@ import com.google.genai.types.HttpOptions; import java.util.Optional; -/** Config for getting an Agent Engine Memory. */ +/** Config for create memory bank. */ @AutoValue -@JsonDeserialize(builder = GetAgentEngineMemoryConfig.Builder.class) -public abstract class GetAgentEngineMemoryConfig extends JsonSerializable { +@JsonDeserialize(builder = CreateMemoryBankConfig.Builder.class) +public abstract class CreateMemoryBankConfig extends JsonSerializable { /** Used to override HTTP request options. */ @JsonProperty("httpOptions") public abstract Optional httpOptions(); - /** Instantiates a builder for GetAgentEngineMemoryConfig. */ + /** Instantiates a builder for CreateMemoryBankConfig. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_GetAgentEngineMemoryConfig.Builder(); + return new AutoValue_CreateMemoryBankConfig.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for GetAgentEngineMemoryConfig. */ + /** Builder for CreateMemoryBankConfig. */ @AutoValue.Builder public abstract static class Builder { - /** For internal usage. Please use `GetAgentEngineMemoryConfig.builder()` for instantiation. */ + /** For internal usage. Please use `CreateMemoryBankConfig.builder()` for instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_GetAgentEngineMemoryConfig.Builder(); + return new AutoValue_CreateMemoryBankConfig.Builder(); } /** @@ -71,12 +71,12 @@ public Builder clearHttpOptions() { return httpOptions(Optional.empty()); } - public abstract GetAgentEngineMemoryConfig build(); + public abstract CreateMemoryBankConfig build(); } - /** Deserializes a JSON string to a GetAgentEngineMemoryConfig object. */ + /** Deserializes a JSON string to a CreateMemoryBankConfig object. */ @ExcludeFromGeneratedCoverageReport - public static GetAgentEngineMemoryConfig fromJson(String jsonString) { - return JsonSerializable.fromJsonString(jsonString, GetAgentEngineMemoryConfig.class); + public static CreateMemoryBankConfig fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, CreateMemoryBankConfig.class); } } diff --git a/src/main/java/com/google/cloud/agentplatform/types/CreateMemoryBankRequestParameters.java b/src/main/java/com/google/cloud/agentplatform/types/CreateMemoryBankRequestParameters.java new file mode 100644 index 0000000..3d2a7b3 --- /dev/null +++ b/src/main/java/com/google/cloud/agentplatform/types/CreateMemoryBankRequestParameters.java @@ -0,0 +1,96 @@ +/* + * 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.types; + +import com.fasterxml.jackson.annotation.JsonCreator; +import com.fasterxml.jackson.annotation.JsonProperty; +import com.fasterxml.jackson.databind.annotation.JsonDeserialize; +import com.google.api.core.InternalApi; +import com.google.auto.value.AutoValue; +import com.google.errorprone.annotations.CanIgnoreReturnValue; +import com.google.genai.JsonSerializable; +import java.util.Optional; + +/** Parameters for creating memory banks. */ +@AutoValue +@InternalApi +@JsonDeserialize(builder = CreateMemoryBankRequestParameters.Builder.class) +public abstract class CreateMemoryBankRequestParameters extends JsonSerializable { + /** */ + @JsonProperty("config") + public abstract Optional config(); + + /** Instantiates a builder for CreateMemoryBankRequestParameters. */ + @ExcludeFromGeneratedCoverageReport + public static Builder builder() { + return new AutoValue_CreateMemoryBankRequestParameters.Builder(); + } + + /** Creates a builder with the same values as this instance. */ + public abstract Builder toBuilder(); + + /** Builder for CreateMemoryBankRequestParameters. */ + @AutoValue.Builder + public abstract static class Builder { + /** + * For internal usage. Please use `CreateMemoryBankRequestParameters.builder()` for + * instantiation. + */ + @JsonCreator + private static Builder create() { + return new AutoValue_CreateMemoryBankRequestParameters.Builder(); + } + + /** + * Setter for config. + * + *

config: + */ + @JsonProperty("config") + public abstract Builder config(CreateMemoryBankConfig config); + + /** + * Setter for config builder. + * + *

config: + */ + @CanIgnoreReturnValue + public Builder config(CreateMemoryBankConfig.Builder configBuilder) { + return config(configBuilder.build()); + } + + @ExcludeFromGeneratedCoverageReport + abstract Builder config(Optional config); + + /** Clears the value of config field. */ + @ExcludeFromGeneratedCoverageReport + @CanIgnoreReturnValue + public Builder clearConfig() { + return config(Optional.empty()); + } + + public abstract CreateMemoryBankRequestParameters build(); + } + + /** Deserializes a JSON string to a CreateMemoryBankRequestParameters object. */ + @ExcludeFromGeneratedCoverageReport + public static CreateMemoryBankRequestParameters fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, CreateMemoryBankRequestParameters.class); + } +} diff --git a/src/main/java/com/google/cloud/agentplatform/types/CreateAgentEngineMemoryRequestParameters.java b/src/main/java/com/google/cloud/agentplatform/types/CreateMemoryRequestParameters.java similarity index 75% rename from src/main/java/com/google/cloud/agentplatform/types/CreateAgentEngineMemoryRequestParameters.java rename to src/main/java/com/google/cloud/agentplatform/types/CreateMemoryRequestParameters.java index 9e21daf..8dee92b 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/CreateAgentEngineMemoryRequestParameters.java +++ b/src/main/java/com/google/cloud/agentplatform/types/CreateMemoryRequestParameters.java @@ -28,12 +28,12 @@ import java.util.Map; import java.util.Optional; -/** Parameters for creating Agent Engine Memories. */ +/** Parameters for creating Memories. */ @AutoValue @InternalApi -@JsonDeserialize(builder = CreateAgentEngineMemoryRequestParameters.Builder.class) -public abstract class CreateAgentEngineMemoryRequestParameters extends JsonSerializable { - /** Name of the agent engine to create the memory under. */ +@JsonDeserialize(builder = CreateMemoryRequestParameters.Builder.class) +public abstract class CreateMemoryRequestParameters extends JsonSerializable { + /** Name of the memory bank to create the memory under. */ @JsonProperty("name") public abstract Optional name(); @@ -57,33 +57,32 @@ public abstract class CreateAgentEngineMemoryRequestParameters extends JsonSeria /** */ @JsonProperty("config") - public abstract Optional config(); + public abstract Optional config(); - /** Instantiates a builder for CreateAgentEngineMemoryRequestParameters. */ + /** Instantiates a builder for CreateMemoryRequestParameters. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_CreateAgentEngineMemoryRequestParameters.Builder(); + return new AutoValue_CreateMemoryRequestParameters.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for CreateAgentEngineMemoryRequestParameters. */ + /** Builder for CreateMemoryRequestParameters. */ @AutoValue.Builder public abstract static class Builder { /** - * For internal usage. Please use `CreateAgentEngineMemoryRequestParameters.builder()` for - * instantiation. + * For internal usage. Please use `CreateMemoryRequestParameters.builder()` for instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_CreateAgentEngineMemoryRequestParameters.Builder(); + return new AutoValue_CreateMemoryRequestParameters.Builder(); } /** * Setter for name. * - *

name: Name of the agent engine to create the memory under. + *

name: Name of the memory bank to create the memory under. */ @JsonProperty("name") public abstract Builder name(String name); @@ -146,7 +145,7 @@ public Builder clearScope() { *

config: */ @JsonProperty("config") - public abstract Builder config(AgentEngineMemoryConfig config); + public abstract Builder config(MemoryConfig config); /** * Setter for config builder. @@ -154,12 +153,12 @@ public Builder clearScope() { *

config: */ @CanIgnoreReturnValue - public Builder config(AgentEngineMemoryConfig.Builder configBuilder) { + public Builder config(MemoryConfig.Builder configBuilder) { return config(configBuilder.build()); } @ExcludeFromGeneratedCoverageReport - abstract Builder config(Optional config); + abstract Builder config(Optional config); /** Clears the value of config field. */ @ExcludeFromGeneratedCoverageReport @@ -168,13 +167,12 @@ public Builder clearConfig() { return config(Optional.empty()); } - public abstract CreateAgentEngineMemoryRequestParameters build(); + public abstract CreateMemoryRequestParameters build(); } - /** Deserializes a JSON string to a CreateAgentEngineMemoryRequestParameters object. */ + /** Deserializes a JSON string to a CreateMemoryRequestParameters object. */ @ExcludeFromGeneratedCoverageReport - public static CreateAgentEngineMemoryRequestParameters fromJson(String jsonString) { - return JsonSerializable.fromJsonString( - jsonString, CreateAgentEngineMemoryRequestParameters.class); + public static CreateMemoryRequestParameters fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, CreateMemoryRequestParameters.class); } } diff --git a/src/main/java/com/google/cloud/agentplatform/types/GetAgentEngineMemoryRevisionConfig.java b/src/main/java/com/google/cloud/agentplatform/types/DeleteMemoryBankConfig.java similarity index 69% rename from src/main/java/com/google/cloud/agentplatform/types/GetAgentEngineMemoryRevisionConfig.java rename to src/main/java/com/google/cloud/agentplatform/types/DeleteMemoryBankConfig.java index 3bcf577..3f9ba2b 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/GetAgentEngineMemoryRevisionConfig.java +++ b/src/main/java/com/google/cloud/agentplatform/types/DeleteMemoryBankConfig.java @@ -27,33 +27,30 @@ import com.google.genai.types.HttpOptions; import java.util.Optional; -/** Config for getting an Agent Engine Memory Revision. */ +/** Config for delete memory bank. */ @AutoValue -@JsonDeserialize(builder = GetAgentEngineMemoryRevisionConfig.Builder.class) -public abstract class GetAgentEngineMemoryRevisionConfig extends JsonSerializable { +@JsonDeserialize(builder = DeleteMemoryBankConfig.Builder.class) +public abstract class DeleteMemoryBankConfig extends JsonSerializable { /** Used to override HTTP request options. */ @JsonProperty("httpOptions") public abstract Optional httpOptions(); - /** Instantiates a builder for GetAgentEngineMemoryRevisionConfig. */ + /** Instantiates a builder for DeleteMemoryBankConfig. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_GetAgentEngineMemoryRevisionConfig.Builder(); + return new AutoValue_DeleteMemoryBankConfig.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for GetAgentEngineMemoryRevisionConfig. */ + /** Builder for DeleteMemoryBankConfig. */ @AutoValue.Builder public abstract static class Builder { - /** - * For internal usage. Please use `GetAgentEngineMemoryRevisionConfig.builder()` for - * instantiation. - */ + /** For internal usage. Please use `DeleteMemoryBankConfig.builder()` for instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_GetAgentEngineMemoryRevisionConfig.Builder(); + return new AutoValue_DeleteMemoryBankConfig.Builder(); } /** @@ -74,12 +71,12 @@ public Builder clearHttpOptions() { return httpOptions(Optional.empty()); } - public abstract GetAgentEngineMemoryRevisionConfig build(); + public abstract DeleteMemoryBankConfig build(); } - /** Deserializes a JSON string to a GetAgentEngineMemoryRevisionConfig object. */ + /** Deserializes a JSON string to a DeleteMemoryBankConfig object. */ @ExcludeFromGeneratedCoverageReport - public static GetAgentEngineMemoryRevisionConfig fromJson(String jsonString) { - return JsonSerializable.fromJsonString(jsonString, GetAgentEngineMemoryRevisionConfig.class); + public static DeleteMemoryBankConfig fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, DeleteMemoryBankConfig.class); } } diff --git a/src/main/java/com/google/cloud/agentplatform/types/DeleteAgentEngineMemoryOperation.java b/src/main/java/com/google/cloud/agentplatform/types/DeleteMemoryBankOperation.java similarity index 84% rename from src/main/java/com/google/cloud/agentplatform/types/DeleteAgentEngineMemoryOperation.java rename to src/main/java/com/google/cloud/agentplatform/types/DeleteMemoryBankOperation.java index f256ae9..335ae5f 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/DeleteAgentEngineMemoryOperation.java +++ b/src/main/java/com/google/cloud/agentplatform/types/DeleteMemoryBankOperation.java @@ -27,10 +27,10 @@ import java.util.Map; import java.util.Optional; -/** Operation for deleting agent engines. */ +/** Operation for deleting a memory bank. */ @AutoValue -@JsonDeserialize(builder = DeleteAgentEngineMemoryOperation.Builder.class) -public abstract class DeleteAgentEngineMemoryOperation extends JsonSerializable { +@JsonDeserialize(builder = DeleteMemoryBankOperation.Builder.class) +public abstract class DeleteMemoryBankOperation extends JsonSerializable { /** * The server-assigned name, which is only unique within the same service that originally returns * it. If you use the default HTTP mapping, the `name` should be a resource name ending with @@ -59,25 +59,22 @@ public abstract class DeleteAgentEngineMemoryOperation extends JsonSerializable @JsonProperty("error") public abstract Optional> error(); - /** Instantiates a builder for DeleteAgentEngineMemoryOperation. */ + /** Instantiates a builder for DeleteMemoryBankOperation. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_DeleteAgentEngineMemoryOperation.Builder(); + return new AutoValue_DeleteMemoryBankOperation.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for DeleteAgentEngineMemoryOperation. */ + /** Builder for DeleteMemoryBankOperation. */ @AutoValue.Builder public abstract static class Builder { - /** - * For internal usage. Please use `DeleteAgentEngineMemoryOperation.builder()` for - * instantiation. - */ + /** For internal usage. Please use `DeleteMemoryBankOperation.builder()` for instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_DeleteAgentEngineMemoryOperation.Builder(); + return new AutoValue_DeleteMemoryBankOperation.Builder(); } /** @@ -158,12 +155,12 @@ public Builder clearError() { return error(Optional.empty()); } - public abstract DeleteAgentEngineMemoryOperation build(); + public abstract DeleteMemoryBankOperation build(); } - /** Deserializes a JSON string to a DeleteAgentEngineMemoryOperation object. */ + /** Deserializes a JSON string to a DeleteMemoryBankOperation object. */ @ExcludeFromGeneratedCoverageReport - public static DeleteAgentEngineMemoryOperation fromJson(String jsonString) { - return JsonSerializable.fromJsonString(jsonString, DeleteAgentEngineMemoryOperation.class); + public static DeleteMemoryBankOperation fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, DeleteMemoryBankOperation.class); } } diff --git a/src/main/java/com/google/cloud/agentplatform/types/DeleteMemoryBankRequestParameters.java b/src/main/java/com/google/cloud/agentplatform/types/DeleteMemoryBankRequestParameters.java new file mode 100644 index 0000000..30386ac --- /dev/null +++ b/src/main/java/com/google/cloud/agentplatform/types/DeleteMemoryBankRequestParameters.java @@ -0,0 +1,140 @@ +/* + * 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.types; + +import com.fasterxml.jackson.annotation.JsonCreator; +import com.fasterxml.jackson.annotation.JsonProperty; +import com.fasterxml.jackson.databind.annotation.JsonDeserialize; +import com.google.api.core.InternalApi; +import com.google.auto.value.AutoValue; +import com.google.errorprone.annotations.CanIgnoreReturnValue; +import com.google.genai.JsonSerializable; +import java.util.Optional; + +/** Parameters for deleting a memory bank. */ +@AutoValue +@InternalApi +@JsonDeserialize(builder = DeleteMemoryBankRequestParameters.Builder.class) +public abstract class DeleteMemoryBankRequestParameters extends JsonSerializable { + /** Name of the memory bank. */ + @JsonProperty("name") + public abstract Optional name(); + + /** If set to true, any child resources will also be deleted. */ + @JsonProperty("force") + public abstract Optional force(); + + /** */ + @JsonProperty("config") + public abstract Optional config(); + + /** Instantiates a builder for DeleteMemoryBankRequestParameters. */ + @ExcludeFromGeneratedCoverageReport + public static Builder builder() { + return new AutoValue_DeleteMemoryBankRequestParameters.Builder(); + } + + /** Creates a builder with the same values as this instance. */ + public abstract Builder toBuilder(); + + /** Builder for DeleteMemoryBankRequestParameters. */ + @AutoValue.Builder + public abstract static class Builder { + /** + * For internal usage. Please use `DeleteMemoryBankRequestParameters.builder()` for + * instantiation. + */ + @JsonCreator + private static Builder create() { + return new AutoValue_DeleteMemoryBankRequestParameters.Builder(); + } + + /** + * Setter for name. + * + *

name: Name of the memory bank. + */ + @JsonProperty("name") + public abstract Builder name(String name); + + @ExcludeFromGeneratedCoverageReport + abstract Builder name(Optional name); + + /** Clears the value of name field. */ + @ExcludeFromGeneratedCoverageReport + @CanIgnoreReturnValue + public Builder clearName() { + return name(Optional.empty()); + } + + /** + * Setter for force. + * + *

force: If set to true, any child resources will also be deleted. + */ + @JsonProperty("force") + public abstract Builder force(boolean force); + + @ExcludeFromGeneratedCoverageReport + abstract Builder force(Optional force); + + /** Clears the value of force field. */ + @ExcludeFromGeneratedCoverageReport + @CanIgnoreReturnValue + public Builder clearForce() { + return force(Optional.empty()); + } + + /** + * Setter for config. + * + *

config: + */ + @JsonProperty("config") + public abstract Builder config(DeleteMemoryBankConfig config); + + /** + * Setter for config builder. + * + *

config: + */ + @CanIgnoreReturnValue + public Builder config(DeleteMemoryBankConfig.Builder configBuilder) { + return config(configBuilder.build()); + } + + @ExcludeFromGeneratedCoverageReport + abstract Builder config(Optional config); + + /** Clears the value of config field. */ + @ExcludeFromGeneratedCoverageReport + @CanIgnoreReturnValue + public Builder clearConfig() { + return config(Optional.empty()); + } + + public abstract DeleteMemoryBankRequestParameters build(); + } + + /** Deserializes a JSON string to a DeleteMemoryBankRequestParameters object. */ + @ExcludeFromGeneratedCoverageReport + public static DeleteMemoryBankRequestParameters fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, DeleteMemoryBankRequestParameters.class); + } +} diff --git a/src/main/java/com/google/cloud/agentplatform/types/DeleteMemoryConfig.java b/src/main/java/com/google/cloud/agentplatform/types/DeleteMemoryConfig.java new file mode 100644 index 0000000..c1e37da --- /dev/null +++ b/src/main/java/com/google/cloud/agentplatform/types/DeleteMemoryConfig.java @@ -0,0 +1,82 @@ +/* + * 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.types; + +import com.fasterxml.jackson.annotation.JsonCreator; +import com.fasterxml.jackson.annotation.JsonProperty; +import com.fasterxml.jackson.databind.annotation.JsonDeserialize; +import com.google.auto.value.AutoValue; +import com.google.errorprone.annotations.CanIgnoreReturnValue; +import com.google.genai.JsonSerializable; +import com.google.genai.types.HttpOptions; +import java.util.Optional; + +/** Config for deleting a Memory. */ +@AutoValue +@JsonDeserialize(builder = DeleteMemoryConfig.Builder.class) +public abstract class DeleteMemoryConfig extends JsonSerializable { + /** Used to override HTTP request options. */ + @JsonProperty("httpOptions") + public abstract Optional httpOptions(); + + /** Instantiates a builder for DeleteMemoryConfig. */ + @ExcludeFromGeneratedCoverageReport + public static Builder builder() { + return new AutoValue_DeleteMemoryConfig.Builder(); + } + + /** Creates a builder with the same values as this instance. */ + public abstract Builder toBuilder(); + + /** Builder for DeleteMemoryConfig. */ + @AutoValue.Builder + public abstract static class Builder { + /** For internal usage. Please use `DeleteMemoryConfig.builder()` for instantiation. */ + @JsonCreator + private static Builder create() { + return new AutoValue_DeleteMemoryConfig.Builder(); + } + + /** + * Setter for httpOptions. + * + *

httpOptions: Used to override HTTP request options. + */ + @JsonProperty("httpOptions") + public abstract Builder httpOptions(HttpOptions httpOptions); + + @ExcludeFromGeneratedCoverageReport + abstract Builder httpOptions(Optional httpOptions); + + /** Clears the value of httpOptions field. */ + @ExcludeFromGeneratedCoverageReport + @CanIgnoreReturnValue + public Builder clearHttpOptions() { + return httpOptions(Optional.empty()); + } + + public abstract DeleteMemoryConfig build(); + } + + /** Deserializes a JSON string to a DeleteMemoryConfig object. */ + @ExcludeFromGeneratedCoverageReport + public static DeleteMemoryConfig fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, DeleteMemoryConfig.class); + } +} diff --git a/src/main/java/com/google/cloud/agentplatform/types/DeleteMemoryOperation.java b/src/main/java/com/google/cloud/agentplatform/types/DeleteMemoryOperation.java new file mode 100644 index 0000000..d84afd6 --- /dev/null +++ b/src/main/java/com/google/cloud/agentplatform/types/DeleteMemoryOperation.java @@ -0,0 +1,166 @@ +/* + * 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.types; + +import com.fasterxml.jackson.annotation.JsonCreator; +import com.fasterxml.jackson.annotation.JsonProperty; +import com.fasterxml.jackson.databind.annotation.JsonDeserialize; +import com.google.auto.value.AutoValue; +import com.google.errorprone.annotations.CanIgnoreReturnValue; +import com.google.genai.JsonSerializable; +import java.util.Map; +import java.util.Optional; + +/** Operation for deleting memories. */ +@AutoValue +@JsonDeserialize(builder = DeleteMemoryOperation.Builder.class) +public abstract class DeleteMemoryOperation extends JsonSerializable { + /** + * The server-assigned name, which is only unique within the same service that originally returns + * it. If you use the default HTTP mapping, the `name` should be a resource name ending with + * `operations/{unique_id}`. + */ + @JsonProperty("name") + public abstract Optional name(); + + /** + * Service-specific metadata associated with the operation. It typically contains progress + * information and common metadata such as create time. Some services might not provide such + * metadata. Any method that returns a long-running operation should document the metadata type, + * if any. + */ + @JsonProperty("metadata") + public abstract Optional> metadata(); + + /** + * If the value is `false`, it means the operation is still in progress. If `true`, the operation + * is completed, and either `error` or `response` is available. + */ + @JsonProperty("done") + public abstract Optional done(); + + /** The error result of the operation in case of failure or cancellation. */ + @JsonProperty("error") + public abstract Optional> error(); + + /** Instantiates a builder for DeleteMemoryOperation. */ + @ExcludeFromGeneratedCoverageReport + public static Builder builder() { + return new AutoValue_DeleteMemoryOperation.Builder(); + } + + /** Creates a builder with the same values as this instance. */ + public abstract Builder toBuilder(); + + /** Builder for DeleteMemoryOperation. */ + @AutoValue.Builder + public abstract static class Builder { + /** For internal usage. Please use `DeleteMemoryOperation.builder()` for instantiation. */ + @JsonCreator + private static Builder create() { + return new AutoValue_DeleteMemoryOperation.Builder(); + } + + /** + * Setter for name. + * + *

name: The server-assigned name, which is only unique within the same service that + * originally returns it. If you use the default HTTP mapping, the `name` should be a resource + * name ending with `operations/{unique_id}`. + */ + @JsonProperty("name") + public abstract Builder name(String name); + + @ExcludeFromGeneratedCoverageReport + abstract Builder name(Optional name); + + /** Clears the value of name field. */ + @ExcludeFromGeneratedCoverageReport + @CanIgnoreReturnValue + public Builder clearName() { + return name(Optional.empty()); + } + + /** + * Setter for metadata. + * + *

metadata: Service-specific metadata associated with the operation. It typically contains + * progress information and common metadata such as create time. Some services might not provide + * such metadata. Any method that returns a long-running operation should document the metadata + * type, if any. + */ + @JsonProperty("metadata") + public abstract Builder metadata(Map metadata); + + @ExcludeFromGeneratedCoverageReport + abstract Builder metadata(Optional> metadata); + + /** Clears the value of metadata field. */ + @ExcludeFromGeneratedCoverageReport + @CanIgnoreReturnValue + public Builder clearMetadata() { + return metadata(Optional.empty()); + } + + /** + * Setter for done. + * + *

done: If the value is `false`, it means the operation is still in progress. If `true`, the + * operation is completed, and either `error` or `response` is available. + */ + @JsonProperty("done") + public abstract Builder done(boolean done); + + @ExcludeFromGeneratedCoverageReport + abstract Builder done(Optional done); + + /** Clears the value of done field. */ + @ExcludeFromGeneratedCoverageReport + @CanIgnoreReturnValue + public Builder clearDone() { + return done(Optional.empty()); + } + + /** + * Setter for error. + * + *

error: The error result of the operation in case of failure or cancellation. + */ + @JsonProperty("error") + public abstract Builder error(Map error); + + @ExcludeFromGeneratedCoverageReport + abstract Builder error(Optional> error); + + /** Clears the value of error field. */ + @ExcludeFromGeneratedCoverageReport + @CanIgnoreReturnValue + public Builder clearError() { + return error(Optional.empty()); + } + + public abstract DeleteMemoryOperation build(); + } + + /** Deserializes a JSON string to a DeleteMemoryOperation object. */ + @ExcludeFromGeneratedCoverageReport + public static DeleteMemoryOperation fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, DeleteMemoryOperation.class); + } +} diff --git a/src/main/java/com/google/cloud/agentplatform/types/DeleteAgentEngineMemoryRequestParameters.java b/src/main/java/com/google/cloud/agentplatform/types/DeleteMemoryRequestParameters.java similarity index 63% rename from src/main/java/com/google/cloud/agentplatform/types/DeleteAgentEngineMemoryRequestParameters.java rename to src/main/java/com/google/cloud/agentplatform/types/DeleteMemoryRequestParameters.java index 6866246..9749ec0 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/DeleteAgentEngineMemoryRequestParameters.java +++ b/src/main/java/com/google/cloud/agentplatform/types/DeleteMemoryRequestParameters.java @@ -27,44 +27,43 @@ import com.google.genai.JsonSerializable; import java.util.Optional; -/** Parameters for deleting agent engines. */ +/** Parameters for deleting memories. */ @AutoValue @InternalApi -@JsonDeserialize(builder = DeleteAgentEngineMemoryRequestParameters.Builder.class) -public abstract class DeleteAgentEngineMemoryRequestParameters extends JsonSerializable { - /** Name of the agent engine memory to delete. */ +@JsonDeserialize(builder = DeleteMemoryRequestParameters.Builder.class) +public abstract class DeleteMemoryRequestParameters extends JsonSerializable { + /** Name of the memory to delete. */ @JsonProperty("name") public abstract Optional name(); /** */ @JsonProperty("config") - public abstract Optional config(); + public abstract Optional config(); - /** Instantiates a builder for DeleteAgentEngineMemoryRequestParameters. */ + /** Instantiates a builder for DeleteMemoryRequestParameters. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_DeleteAgentEngineMemoryRequestParameters.Builder(); + return new AutoValue_DeleteMemoryRequestParameters.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for DeleteAgentEngineMemoryRequestParameters. */ + /** Builder for DeleteMemoryRequestParameters. */ @AutoValue.Builder public abstract static class Builder { /** - * For internal usage. Please use `DeleteAgentEngineMemoryRequestParameters.builder()` for - * instantiation. + * For internal usage. Please use `DeleteMemoryRequestParameters.builder()` for instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_DeleteAgentEngineMemoryRequestParameters.Builder(); + return new AutoValue_DeleteMemoryRequestParameters.Builder(); } /** * Setter for name. * - *

name: Name of the agent engine memory to delete. + *

name: Name of the memory to delete. */ @JsonProperty("name") public abstract Builder name(String name); @@ -85,7 +84,7 @@ public Builder clearName() { *

config: */ @JsonProperty("config") - public abstract Builder config(DeleteAgentEngineMemoryConfig config); + public abstract Builder config(DeleteMemoryConfig config); /** * Setter for config builder. @@ -93,12 +92,12 @@ public Builder clearName() { *

config: */ @CanIgnoreReturnValue - public Builder config(DeleteAgentEngineMemoryConfig.Builder configBuilder) { + public Builder config(DeleteMemoryConfig.Builder configBuilder) { return config(configBuilder.build()); } @ExcludeFromGeneratedCoverageReport - abstract Builder config(Optional config); + abstract Builder config(Optional config); /** Clears the value of config field. */ @ExcludeFromGeneratedCoverageReport @@ -107,13 +106,12 @@ public Builder clearConfig() { return config(Optional.empty()); } - public abstract DeleteAgentEngineMemoryRequestParameters build(); + public abstract DeleteMemoryRequestParameters build(); } - /** Deserializes a JSON string to a DeleteAgentEngineMemoryRequestParameters object. */ + /** Deserializes a JSON string to a DeleteMemoryRequestParameters object. */ @ExcludeFromGeneratedCoverageReport - public static DeleteAgentEngineMemoryRequestParameters fromJson(String jsonString) { - return JsonSerializable.fromJsonString( - jsonString, DeleteAgentEngineMemoryRequestParameters.class); + public static DeleteMemoryRequestParameters fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, DeleteMemoryRequestParameters.class); } } diff --git a/src/main/java/com/google/cloud/agentplatform/types/GenerateAgentEngineMemoriesConfig.java b/src/main/java/com/google/cloud/agentplatform/types/GenerateMemoriesConfig.java similarity index 93% rename from src/main/java/com/google/cloud/agentplatform/types/GenerateAgentEngineMemoriesConfig.java rename to src/main/java/com/google/cloud/agentplatform/types/GenerateMemoriesConfig.java index dc2624b..cab2481 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/GenerateAgentEngineMemoriesConfig.java +++ b/src/main/java/com/google/cloud/agentplatform/types/GenerateMemoriesConfig.java @@ -36,8 +36,8 @@ /** Config for generating memories. */ @AutoValue -@JsonDeserialize(builder = GenerateAgentEngineMemoriesConfig.Builder.class) -public abstract class GenerateAgentEngineMemoriesConfig extends JsonSerializable { +@JsonDeserialize(builder = GenerateMemoriesConfig.Builder.class) +public abstract class GenerateMemoriesConfig extends JsonSerializable { /** Used to override HTTP request options. */ @JsonProperty("httpOptions") public abstract Optional httpOptions(); @@ -93,25 +93,22 @@ public abstract class GenerateAgentEngineMemoriesConfig extends JsonSerializable @JsonProperty("allowedTopics") public abstract Optional> allowedTopics(); - /** Instantiates a builder for GenerateAgentEngineMemoriesConfig. */ + /** Instantiates a builder for GenerateMemoriesConfig. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_GenerateAgentEngineMemoriesConfig.Builder(); + return new AutoValue_GenerateMemoriesConfig.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for GenerateAgentEngineMemoriesConfig. */ + /** Builder for GenerateMemoriesConfig. */ @AutoValue.Builder public abstract static class Builder { - /** - * For internal usage. Please use `GenerateAgentEngineMemoriesConfig.builder()` for - * instantiation. - */ + /** For internal usage. Please use `GenerateMemoriesConfig.builder()` for instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_GenerateAgentEngineMemoriesConfig.Builder(); + return new AutoValue_GenerateMemoriesConfig.Builder(); } /** @@ -351,12 +348,12 @@ public Builder clearAllowedTopics() { return allowedTopics(Optional.empty()); } - public abstract GenerateAgentEngineMemoriesConfig build(); + public abstract GenerateMemoriesConfig build(); } - /** Deserializes a JSON string to a GenerateAgentEngineMemoriesConfig object. */ + /** Deserializes a JSON string to a GenerateMemoriesConfig object. */ @ExcludeFromGeneratedCoverageReport - public static GenerateAgentEngineMemoriesConfig fromJson(String jsonString) { - return JsonSerializable.fromJsonString(jsonString, GenerateAgentEngineMemoriesConfig.class); + public static GenerateMemoriesConfig fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, GenerateMemoriesConfig.class); } } diff --git a/src/main/java/com/google/cloud/agentplatform/types/AgentEngineGenerateMemoriesOperation.java b/src/main/java/com/google/cloud/agentplatform/types/GenerateMemoriesOperation.java similarity index 86% rename from src/main/java/com/google/cloud/agentplatform/types/AgentEngineGenerateMemoriesOperation.java rename to src/main/java/com/google/cloud/agentplatform/types/GenerateMemoriesOperation.java index be21ec1..cdd8e7e 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/AgentEngineGenerateMemoriesOperation.java +++ b/src/main/java/com/google/cloud/agentplatform/types/GenerateMemoriesOperation.java @@ -27,10 +27,10 @@ import java.util.Map; import java.util.Optional; -/** Operation that generates memories for an agent engine. */ +/** Operation that generates memories with a Memory Bank. */ @AutoValue -@JsonDeserialize(builder = AgentEngineGenerateMemoriesOperation.Builder.class) -public abstract class AgentEngineGenerateMemoriesOperation extends JsonSerializable { +@JsonDeserialize(builder = GenerateMemoriesOperation.Builder.class) +public abstract class GenerateMemoriesOperation extends JsonSerializable { /** * The server-assigned name, which is only unique within the same service that originally returns * it. If you use the default HTTP mapping, the `name` should be a resource name ending with @@ -63,25 +63,22 @@ public abstract class AgentEngineGenerateMemoriesOperation extends JsonSerializa @JsonProperty("response") public abstract Optional response(); - /** Instantiates a builder for AgentEngineGenerateMemoriesOperation. */ + /** Instantiates a builder for GenerateMemoriesOperation. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_AgentEngineGenerateMemoriesOperation.Builder(); + return new AutoValue_GenerateMemoriesOperation.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for AgentEngineGenerateMemoriesOperation. */ + /** Builder for GenerateMemoriesOperation. */ @AutoValue.Builder public abstract static class Builder { - /** - * For internal usage. Please use `AgentEngineGenerateMemoriesOperation.builder()` for - * instantiation. - */ + /** For internal usage. Please use `GenerateMemoriesOperation.builder()` for instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_AgentEngineGenerateMemoriesOperation.Builder(); + return new AutoValue_GenerateMemoriesOperation.Builder(); } /** @@ -190,12 +187,12 @@ public Builder clearResponse() { return response(Optional.empty()); } - public abstract AgentEngineGenerateMemoriesOperation build(); + public abstract GenerateMemoriesOperation build(); } - /** Deserializes a JSON string to a AgentEngineGenerateMemoriesOperation object. */ + /** Deserializes a JSON string to a GenerateMemoriesOperation object. */ @ExcludeFromGeneratedCoverageReport - public static AgentEngineGenerateMemoriesOperation fromJson(String jsonString) { - return JsonSerializable.fromJsonString(jsonString, AgentEngineGenerateMemoriesOperation.class); + public static GenerateMemoriesOperation fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, GenerateMemoriesOperation.class); } } diff --git a/src/main/java/com/google/cloud/agentplatform/types/GenerateAgentEngineMemoriesRequestParameters.java b/src/main/java/com/google/cloud/agentplatform/types/GenerateMemoriesRequestParameters.java similarity index 84% rename from src/main/java/com/google/cloud/agentplatform/types/GenerateAgentEngineMemoriesRequestParameters.java rename to src/main/java/com/google/cloud/agentplatform/types/GenerateMemoriesRequestParameters.java index f32c4de..91670d2 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/GenerateAgentEngineMemoriesRequestParameters.java +++ b/src/main/java/com/google/cloud/agentplatform/types/GenerateMemoriesRequestParameters.java @@ -28,12 +28,12 @@ import java.util.Map; import java.util.Optional; -/** Parameters for generating agent engine memories. */ +/** Parameters for generating memories. */ @AutoValue @InternalApi -@JsonDeserialize(builder = GenerateAgentEngineMemoriesRequestParameters.Builder.class) -public abstract class GenerateAgentEngineMemoriesRequestParameters extends JsonSerializable { - /** Name of the agent engine to generate memories for. */ +@JsonDeserialize(builder = GenerateMemoriesRequestParameters.Builder.class) +public abstract class GenerateMemoriesRequestParameters extends JsonSerializable { + /** Name of the Memory Bank to generate memories with. */ @JsonProperty("name") public abstract Optional name(); @@ -61,33 +61,33 @@ public abstract class GenerateAgentEngineMemoriesRequestParameters extends JsonS /** */ @JsonProperty("config") - public abstract Optional config(); + public abstract Optional config(); - /** Instantiates a builder for GenerateAgentEngineMemoriesRequestParameters. */ + /** Instantiates a builder for GenerateMemoriesRequestParameters. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_GenerateAgentEngineMemoriesRequestParameters.Builder(); + return new AutoValue_GenerateMemoriesRequestParameters.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for GenerateAgentEngineMemoriesRequestParameters. */ + /** Builder for GenerateMemoriesRequestParameters. */ @AutoValue.Builder public abstract static class Builder { /** - * For internal usage. Please use `GenerateAgentEngineMemoriesRequestParameters.builder()` for + * For internal usage. Please use `GenerateMemoriesRequestParameters.builder()` for * instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_GenerateAgentEngineMemoriesRequestParameters.Builder(); + return new AutoValue_GenerateMemoriesRequestParameters.Builder(); } /** * Setter for name. * - *

name: Name of the agent engine to generate memories for. + *

name: Name of the Memory Bank to generate memories with. */ @JsonProperty("name") public abstract Builder name(String name); @@ -223,7 +223,7 @@ public Builder clearScope() { *

config: */ @JsonProperty("config") - public abstract Builder config(GenerateAgentEngineMemoriesConfig config); + public abstract Builder config(GenerateMemoriesConfig config); /** * Setter for config builder. @@ -231,12 +231,12 @@ public Builder clearScope() { *

config: */ @CanIgnoreReturnValue - public Builder config(GenerateAgentEngineMemoriesConfig.Builder configBuilder) { + public Builder config(GenerateMemoriesConfig.Builder configBuilder) { return config(configBuilder.build()); } @ExcludeFromGeneratedCoverageReport - abstract Builder config(Optional config); + abstract Builder config(Optional config); /** Clears the value of config field. */ @ExcludeFromGeneratedCoverageReport @@ -245,13 +245,12 @@ public Builder clearConfig() { return config(Optional.empty()); } - public abstract GenerateAgentEngineMemoriesRequestParameters build(); + public abstract GenerateMemoriesRequestParameters build(); } - /** Deserializes a JSON string to a GenerateAgentEngineMemoriesRequestParameters object. */ + /** Deserializes a JSON string to a GenerateMemoriesRequestParameters object. */ @ExcludeFromGeneratedCoverageReport - public static GenerateAgentEngineMemoriesRequestParameters fromJson(String jsonString) { - return JsonSerializable.fromJsonString( - jsonString, GenerateAgentEngineMemoriesRequestParameters.class); + public static GenerateMemoriesRequestParameters fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, GenerateMemoriesRequestParameters.class); } } diff --git a/src/main/java/com/google/cloud/agentplatform/types/GetAgentEngineGenerateMemoriesOperationParameters.java b/src/main/java/com/google/cloud/agentplatform/types/GetGenerateMemoriesOperationParameters.java similarity index 69% rename from src/main/java/com/google/cloud/agentplatform/types/GetAgentEngineGenerateMemoriesOperationParameters.java rename to src/main/java/com/google/cloud/agentplatform/types/GetGenerateMemoriesOperationParameters.java index ccfa312..335f6bc 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/GetAgentEngineGenerateMemoriesOperationParameters.java +++ b/src/main/java/com/google/cloud/agentplatform/types/GetGenerateMemoriesOperationParameters.java @@ -30,35 +30,35 @@ /** Parameters for getting an operation with generated memories as a response. */ @AutoValue @InternalApi -@JsonDeserialize(builder = GetAgentEngineGenerateMemoriesOperationParameters.Builder.class) -public abstract class GetAgentEngineGenerateMemoriesOperationParameters extends JsonSerializable { +@JsonDeserialize(builder = GetGenerateMemoriesOperationParameters.Builder.class) +public abstract class GetGenerateMemoriesOperationParameters extends JsonSerializable { /** The server-assigned name for the operation. */ @JsonProperty("operationName") public abstract Optional operationName(); /** Used to override the default configuration. */ @JsonProperty("config") - public abstract Optional config(); + public abstract Optional config(); - /** Instantiates a builder for GetAgentEngineGenerateMemoriesOperationParameters. */ + /** Instantiates a builder for GetGenerateMemoriesOperationParameters. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_GetAgentEngineGenerateMemoriesOperationParameters.Builder(); + return new AutoValue_GetGenerateMemoriesOperationParameters.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for GetAgentEngineGenerateMemoriesOperationParameters. */ + /** Builder for GetGenerateMemoriesOperationParameters. */ @AutoValue.Builder public abstract static class Builder { /** - * For internal usage. Please use `GetAgentEngineGenerateMemoriesOperationParameters.builder()` - * for instantiation. + * For internal usage. Please use `GetGenerateMemoriesOperationParameters.builder()` for + * instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_GetAgentEngineGenerateMemoriesOperationParameters.Builder(); + return new AutoValue_GetGenerateMemoriesOperationParameters.Builder(); } /** @@ -85,7 +85,7 @@ public Builder clearOperationName() { *

config: Used to override the default configuration. */ @JsonProperty("config") - public abstract Builder config(GetAgentEngineOperationConfig config); + public abstract Builder config(GetMemoryBankOperationConfig config); /** * Setter for config builder. @@ -93,12 +93,12 @@ public Builder clearOperationName() { *

config: Used to override the default configuration. */ @CanIgnoreReturnValue - public Builder config(GetAgentEngineOperationConfig.Builder configBuilder) { + public Builder config(GetMemoryBankOperationConfig.Builder configBuilder) { return config(configBuilder.build()); } @ExcludeFromGeneratedCoverageReport - abstract Builder config(Optional config); + abstract Builder config(Optional config); /** Clears the value of config field. */ @ExcludeFromGeneratedCoverageReport @@ -107,13 +107,13 @@ public Builder clearConfig() { return config(Optional.empty()); } - public abstract GetAgentEngineGenerateMemoriesOperationParameters build(); + public abstract GetGenerateMemoriesOperationParameters build(); } - /** Deserializes a JSON string to a GetAgentEngineGenerateMemoriesOperationParameters object. */ + /** Deserializes a JSON string to a GetGenerateMemoriesOperationParameters object. */ @ExcludeFromGeneratedCoverageReport - public static GetAgentEngineGenerateMemoriesOperationParameters fromJson(String jsonString) { + public static GetGenerateMemoriesOperationParameters fromJson(String jsonString) { return JsonSerializable.fromJsonString( - jsonString, GetAgentEngineGenerateMemoriesOperationParameters.class); + jsonString, GetGenerateMemoriesOperationParameters.class); } } diff --git a/src/main/java/com/google/cloud/agentplatform/types/DeleteAgentEngineMemoryConfig.java b/src/main/java/com/google/cloud/agentplatform/types/GetMemoryBankOperationConfig.java similarity index 71% rename from src/main/java/com/google/cloud/agentplatform/types/DeleteAgentEngineMemoryConfig.java rename to src/main/java/com/google/cloud/agentplatform/types/GetMemoryBankOperationConfig.java index 07ae6d3..34321ac 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/DeleteAgentEngineMemoryConfig.java +++ b/src/main/java/com/google/cloud/agentplatform/types/GetMemoryBankOperationConfig.java @@ -27,32 +27,32 @@ import com.google.genai.types.HttpOptions; import java.util.Optional; -/** Config for deleting an Agent Engine Memory. */ +/** */ @AutoValue -@JsonDeserialize(builder = DeleteAgentEngineMemoryConfig.Builder.class) -public abstract class DeleteAgentEngineMemoryConfig extends JsonSerializable { +@JsonDeserialize(builder = GetMemoryBankOperationConfig.Builder.class) +public abstract class GetMemoryBankOperationConfig extends JsonSerializable { /** Used to override HTTP request options. */ @JsonProperty("httpOptions") public abstract Optional httpOptions(); - /** Instantiates a builder for DeleteAgentEngineMemoryConfig. */ + /** Instantiates a builder for GetMemoryBankOperationConfig. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_DeleteAgentEngineMemoryConfig.Builder(); + return new AutoValue_GetMemoryBankOperationConfig.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for DeleteAgentEngineMemoryConfig. */ + /** Builder for GetMemoryBankOperationConfig. */ @AutoValue.Builder public abstract static class Builder { /** - * For internal usage. Please use `DeleteAgentEngineMemoryConfig.builder()` for instantiation. + * For internal usage. Please use `GetMemoryBankOperationConfig.builder()` for instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_DeleteAgentEngineMemoryConfig.Builder(); + return new AutoValue_GetMemoryBankOperationConfig.Builder(); } /** @@ -73,12 +73,12 @@ public Builder clearHttpOptions() { return httpOptions(Optional.empty()); } - public abstract DeleteAgentEngineMemoryConfig build(); + public abstract GetMemoryBankOperationConfig build(); } - /** Deserializes a JSON string to a DeleteAgentEngineMemoryConfig object. */ + /** Deserializes a JSON string to a GetMemoryBankOperationConfig object. */ @ExcludeFromGeneratedCoverageReport - public static DeleteAgentEngineMemoryConfig fromJson(String jsonString) { - return JsonSerializable.fromJsonString(jsonString, DeleteAgentEngineMemoryConfig.class); + public static GetMemoryBankOperationConfig fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, GetMemoryBankOperationConfig.class); } } diff --git a/src/main/java/com/google/cloud/agentplatform/types/GetMemoryBankOperationParameters.java b/src/main/java/com/google/cloud/agentplatform/types/GetMemoryBankOperationParameters.java new file mode 100644 index 0000000..373395a --- /dev/null +++ b/src/main/java/com/google/cloud/agentplatform/types/GetMemoryBankOperationParameters.java @@ -0,0 +1,118 @@ +/* + * 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.types; + +import com.fasterxml.jackson.annotation.JsonCreator; +import com.fasterxml.jackson.annotation.JsonProperty; +import com.fasterxml.jackson.databind.annotation.JsonDeserialize; +import com.google.api.core.InternalApi; +import com.google.auto.value.AutoValue; +import com.google.errorprone.annotations.CanIgnoreReturnValue; +import com.google.genai.JsonSerializable; +import java.util.Optional; + +/** Parameters for getting an operation with a memory bank as a response. */ +@AutoValue +@InternalApi +@JsonDeserialize(builder = GetMemoryBankOperationParameters.Builder.class) +public abstract class GetMemoryBankOperationParameters extends JsonSerializable { + /** The server-assigned name for the operation. */ + @JsonProperty("operationName") + public abstract Optional operationName(); + + /** Used to override the default configuration. */ + @JsonProperty("config") + public abstract Optional config(); + + /** Instantiates a builder for GetMemoryBankOperationParameters. */ + @ExcludeFromGeneratedCoverageReport + public static Builder builder() { + return new AutoValue_GetMemoryBankOperationParameters.Builder(); + } + + /** Creates a builder with the same values as this instance. */ + public abstract Builder toBuilder(); + + /** Builder for GetMemoryBankOperationParameters. */ + @AutoValue.Builder + public abstract static class Builder { + /** + * For internal usage. Please use `GetMemoryBankOperationParameters.builder()` for + * instantiation. + */ + @JsonCreator + private static Builder create() { + return new AutoValue_GetMemoryBankOperationParameters.Builder(); + } + + /** + * Setter for operationName. + * + *

operationName: The server-assigned name for the operation. + */ + @JsonProperty("operationName") + public abstract Builder operationName(String operationName); + + @ExcludeFromGeneratedCoverageReport + abstract Builder operationName(Optional operationName); + + /** Clears the value of operationName field. */ + @ExcludeFromGeneratedCoverageReport + @CanIgnoreReturnValue + public Builder clearOperationName() { + return operationName(Optional.empty()); + } + + /** + * Setter for config. + * + *

config: Used to override the default configuration. + */ + @JsonProperty("config") + public abstract Builder config(GetMemoryBankOperationConfig config); + + /** + * Setter for config builder. + * + *

config: Used to override the default configuration. + */ + @CanIgnoreReturnValue + public Builder config(GetMemoryBankOperationConfig.Builder configBuilder) { + return config(configBuilder.build()); + } + + @ExcludeFromGeneratedCoverageReport + abstract Builder config(Optional config); + + /** Clears the value of config field. */ + @ExcludeFromGeneratedCoverageReport + @CanIgnoreReturnValue + public Builder clearConfig() { + return config(Optional.empty()); + } + + public abstract GetMemoryBankOperationParameters build(); + } + + /** Deserializes a JSON string to a GetMemoryBankOperationParameters object. */ + @ExcludeFromGeneratedCoverageReport + public static GetMemoryBankOperationParameters fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, GetMemoryBankOperationParameters.class); + } +} diff --git a/src/main/java/com/google/cloud/agentplatform/types/GetMemoryConfig.java b/src/main/java/com/google/cloud/agentplatform/types/GetMemoryConfig.java new file mode 100644 index 0000000..7c7c1a5 --- /dev/null +++ b/src/main/java/com/google/cloud/agentplatform/types/GetMemoryConfig.java @@ -0,0 +1,82 @@ +/* + * 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.types; + +import com.fasterxml.jackson.annotation.JsonCreator; +import com.fasterxml.jackson.annotation.JsonProperty; +import com.fasterxml.jackson.databind.annotation.JsonDeserialize; +import com.google.auto.value.AutoValue; +import com.google.errorprone.annotations.CanIgnoreReturnValue; +import com.google.genai.JsonSerializable; +import com.google.genai.types.HttpOptions; +import java.util.Optional; + +/** Config for getting a Memory. */ +@AutoValue +@JsonDeserialize(builder = GetMemoryConfig.Builder.class) +public abstract class GetMemoryConfig extends JsonSerializable { + /** Used to override HTTP request options. */ + @JsonProperty("httpOptions") + public abstract Optional httpOptions(); + + /** Instantiates a builder for GetMemoryConfig. */ + @ExcludeFromGeneratedCoverageReport + public static Builder builder() { + return new AutoValue_GetMemoryConfig.Builder(); + } + + /** Creates a builder with the same values as this instance. */ + public abstract Builder toBuilder(); + + /** Builder for GetMemoryConfig. */ + @AutoValue.Builder + public abstract static class Builder { + /** For internal usage. Please use `GetMemoryConfig.builder()` for instantiation. */ + @JsonCreator + private static Builder create() { + return new AutoValue_GetMemoryConfig.Builder(); + } + + /** + * Setter for httpOptions. + * + *

httpOptions: Used to override HTTP request options. + */ + @JsonProperty("httpOptions") + public abstract Builder httpOptions(HttpOptions httpOptions); + + @ExcludeFromGeneratedCoverageReport + abstract Builder httpOptions(Optional httpOptions); + + /** Clears the value of httpOptions field. */ + @ExcludeFromGeneratedCoverageReport + @CanIgnoreReturnValue + public Builder clearHttpOptions() { + return httpOptions(Optional.empty()); + } + + public abstract GetMemoryConfig build(); + } + + /** Deserializes a JSON string to a GetMemoryConfig object. */ + @ExcludeFromGeneratedCoverageReport + public static GetMemoryConfig fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, GetMemoryConfig.class); + } +} diff --git a/src/main/java/com/google/cloud/agentplatform/types/GetAgentEngineMemoryOperationParameters.java b/src/main/java/com/google/cloud/agentplatform/types/GetMemoryOperationParameters.java similarity index 69% rename from src/main/java/com/google/cloud/agentplatform/types/GetAgentEngineMemoryOperationParameters.java rename to src/main/java/com/google/cloud/agentplatform/types/GetMemoryOperationParameters.java index e2b15ba..538e697 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/GetAgentEngineMemoryOperationParameters.java +++ b/src/main/java/com/google/cloud/agentplatform/types/GetMemoryOperationParameters.java @@ -30,35 +30,34 @@ /** Parameters for getting an operation with a memory as a response. */ @AutoValue @InternalApi -@JsonDeserialize(builder = GetAgentEngineMemoryOperationParameters.Builder.class) -public abstract class GetAgentEngineMemoryOperationParameters extends JsonSerializable { +@JsonDeserialize(builder = GetMemoryOperationParameters.Builder.class) +public abstract class GetMemoryOperationParameters extends JsonSerializable { /** The server-assigned name for the operation. */ @JsonProperty("operationName") public abstract Optional operationName(); /** Used to override the default configuration. */ @JsonProperty("config") - public abstract Optional config(); + public abstract Optional config(); - /** Instantiates a builder for GetAgentEngineMemoryOperationParameters. */ + /** Instantiates a builder for GetMemoryOperationParameters. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_GetAgentEngineMemoryOperationParameters.Builder(); + return new AutoValue_GetMemoryOperationParameters.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for GetAgentEngineMemoryOperationParameters. */ + /** Builder for GetMemoryOperationParameters. */ @AutoValue.Builder public abstract static class Builder { /** - * For internal usage. Please use `GetAgentEngineMemoryOperationParameters.builder()` for - * instantiation. + * For internal usage. Please use `GetMemoryOperationParameters.builder()` for instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_GetAgentEngineMemoryOperationParameters.Builder(); + return new AutoValue_GetMemoryOperationParameters.Builder(); } /** @@ -85,7 +84,7 @@ public Builder clearOperationName() { *

config: Used to override the default configuration. */ @JsonProperty("config") - public abstract Builder config(GetAgentEngineOperationConfig config); + public abstract Builder config(GetMemoryBankOperationConfig config); /** * Setter for config builder. @@ -93,12 +92,12 @@ public Builder clearOperationName() { *

config: Used to override the default configuration. */ @CanIgnoreReturnValue - public Builder config(GetAgentEngineOperationConfig.Builder configBuilder) { + public Builder config(GetMemoryBankOperationConfig.Builder configBuilder) { return config(configBuilder.build()); } @ExcludeFromGeneratedCoverageReport - abstract Builder config(Optional config); + abstract Builder config(Optional config); /** Clears the value of config field. */ @ExcludeFromGeneratedCoverageReport @@ -107,13 +106,12 @@ public Builder clearConfig() { return config(Optional.empty()); } - public abstract GetAgentEngineMemoryOperationParameters build(); + public abstract GetMemoryOperationParameters build(); } - /** Deserializes a JSON string to a GetAgentEngineMemoryOperationParameters object. */ + /** Deserializes a JSON string to a GetMemoryOperationParameters object. */ @ExcludeFromGeneratedCoverageReport - public static GetAgentEngineMemoryOperationParameters fromJson(String jsonString) { - return JsonSerializable.fromJsonString( - jsonString, GetAgentEngineMemoryOperationParameters.class); + public static GetMemoryOperationParameters fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, GetMemoryOperationParameters.class); } } diff --git a/src/main/java/com/google/cloud/agentplatform/types/GetMemoryRequestParameters.java b/src/main/java/com/google/cloud/agentplatform/types/GetMemoryRequestParameters.java new file mode 100644 index 0000000..54fe00a --- /dev/null +++ b/src/main/java/com/google/cloud/agentplatform/types/GetMemoryRequestParameters.java @@ -0,0 +1,115 @@ +/* + * 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.types; + +import com.fasterxml.jackson.annotation.JsonCreator; +import com.fasterxml.jackson.annotation.JsonProperty; +import com.fasterxml.jackson.databind.annotation.JsonDeserialize; +import com.google.api.core.InternalApi; +import com.google.auto.value.AutoValue; +import com.google.errorprone.annotations.CanIgnoreReturnValue; +import com.google.genai.JsonSerializable; +import java.util.Optional; + +/** Parameters for getting a Memory. */ +@AutoValue +@InternalApi +@JsonDeserialize(builder = GetMemoryRequestParameters.Builder.class) +public abstract class GetMemoryRequestParameters extends JsonSerializable { + /** Name of the memory. */ + @JsonProperty("name") + public abstract Optional name(); + + /** */ + @JsonProperty("config") + public abstract Optional config(); + + /** Instantiates a builder for GetMemoryRequestParameters. */ + @ExcludeFromGeneratedCoverageReport + public static Builder builder() { + return new AutoValue_GetMemoryRequestParameters.Builder(); + } + + /** Creates a builder with the same values as this instance. */ + public abstract Builder toBuilder(); + + /** Builder for GetMemoryRequestParameters. */ + @AutoValue.Builder + public abstract static class Builder { + /** For internal usage. Please use `GetMemoryRequestParameters.builder()` for instantiation. */ + @JsonCreator + private static Builder create() { + return new AutoValue_GetMemoryRequestParameters.Builder(); + } + + /** + * Setter for name. + * + *

name: Name of the memory. + */ + @JsonProperty("name") + public abstract Builder name(String name); + + @ExcludeFromGeneratedCoverageReport + abstract Builder name(Optional name); + + /** Clears the value of name field. */ + @ExcludeFromGeneratedCoverageReport + @CanIgnoreReturnValue + public Builder clearName() { + return name(Optional.empty()); + } + + /** + * Setter for config. + * + *

config: + */ + @JsonProperty("config") + public abstract Builder config(GetMemoryConfig config); + + /** + * Setter for config builder. + * + *

config: + */ + @CanIgnoreReturnValue + public Builder config(GetMemoryConfig.Builder configBuilder) { + return config(configBuilder.build()); + } + + @ExcludeFromGeneratedCoverageReport + abstract Builder config(Optional config); + + /** Clears the value of config field. */ + @ExcludeFromGeneratedCoverageReport + @CanIgnoreReturnValue + public Builder clearConfig() { + return config(Optional.empty()); + } + + public abstract GetMemoryRequestParameters build(); + } + + /** Deserializes a JSON string to a GetMemoryRequestParameters object. */ + @ExcludeFromGeneratedCoverageReport + public static GetMemoryRequestParameters fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, GetMemoryRequestParameters.class); + } +} diff --git a/src/main/java/com/google/cloud/agentplatform/types/GetMemoryRevisionConfig.java b/src/main/java/com/google/cloud/agentplatform/types/GetMemoryRevisionConfig.java new file mode 100644 index 0000000..218c5d5 --- /dev/null +++ b/src/main/java/com/google/cloud/agentplatform/types/GetMemoryRevisionConfig.java @@ -0,0 +1,82 @@ +/* + * 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.types; + +import com.fasterxml.jackson.annotation.JsonCreator; +import com.fasterxml.jackson.annotation.JsonProperty; +import com.fasterxml.jackson.databind.annotation.JsonDeserialize; +import com.google.auto.value.AutoValue; +import com.google.errorprone.annotations.CanIgnoreReturnValue; +import com.google.genai.JsonSerializable; +import com.google.genai.types.HttpOptions; +import java.util.Optional; + +/** Config for getting a Memory Revision. */ +@AutoValue +@JsonDeserialize(builder = GetMemoryRevisionConfig.Builder.class) +public abstract class GetMemoryRevisionConfig extends JsonSerializable { + /** Used to override HTTP request options. */ + @JsonProperty("httpOptions") + public abstract Optional httpOptions(); + + /** Instantiates a builder for GetMemoryRevisionConfig. */ + @ExcludeFromGeneratedCoverageReport + public static Builder builder() { + return new AutoValue_GetMemoryRevisionConfig.Builder(); + } + + /** Creates a builder with the same values as this instance. */ + public abstract Builder toBuilder(); + + /** Builder for GetMemoryRevisionConfig. */ + @AutoValue.Builder + public abstract static class Builder { + /** For internal usage. Please use `GetMemoryRevisionConfig.builder()` for instantiation. */ + @JsonCreator + private static Builder create() { + return new AutoValue_GetMemoryRevisionConfig.Builder(); + } + + /** + * Setter for httpOptions. + * + *

httpOptions: Used to override HTTP request options. + */ + @JsonProperty("httpOptions") + public abstract Builder httpOptions(HttpOptions httpOptions); + + @ExcludeFromGeneratedCoverageReport + abstract Builder httpOptions(Optional httpOptions); + + /** Clears the value of httpOptions field. */ + @ExcludeFromGeneratedCoverageReport + @CanIgnoreReturnValue + public Builder clearHttpOptions() { + return httpOptions(Optional.empty()); + } + + public abstract GetMemoryRevisionConfig build(); + } + + /** Deserializes a JSON string to a GetMemoryRevisionConfig object. */ + @ExcludeFromGeneratedCoverageReport + public static GetMemoryRevisionConfig fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, GetMemoryRevisionConfig.class); + } +} diff --git a/src/main/java/com/google/cloud/agentplatform/types/GetAgentEngineMemoryRequestParameters.java b/src/main/java/com/google/cloud/agentplatform/types/GetMemoryRevisionRequestParameters.java similarity index 65% rename from src/main/java/com/google/cloud/agentplatform/types/GetAgentEngineMemoryRequestParameters.java rename to src/main/java/com/google/cloud/agentplatform/types/GetMemoryRevisionRequestParameters.java index 30a4f1b..ed0776b 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/GetAgentEngineMemoryRequestParameters.java +++ b/src/main/java/com/google/cloud/agentplatform/types/GetMemoryRevisionRequestParameters.java @@ -27,44 +27,44 @@ import com.google.genai.JsonSerializable; import java.util.Optional; -/** Parameters for getting an agent engine. */ +/** Parameters for getting a memory revision. */ @AutoValue @InternalApi -@JsonDeserialize(builder = GetAgentEngineMemoryRequestParameters.Builder.class) -public abstract class GetAgentEngineMemoryRequestParameters extends JsonSerializable { - /** Name of the agent engine. */ +@JsonDeserialize(builder = GetMemoryRevisionRequestParameters.Builder.class) +public abstract class GetMemoryRevisionRequestParameters extends JsonSerializable { + /** Name of the Memory Revision. */ @JsonProperty("name") public abstract Optional name(); /** */ @JsonProperty("config") - public abstract Optional config(); + public abstract Optional config(); - /** Instantiates a builder for GetAgentEngineMemoryRequestParameters. */ + /** Instantiates a builder for GetMemoryRevisionRequestParameters. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_GetAgentEngineMemoryRequestParameters.Builder(); + return new AutoValue_GetMemoryRevisionRequestParameters.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for GetAgentEngineMemoryRequestParameters. */ + /** Builder for GetMemoryRevisionRequestParameters. */ @AutoValue.Builder public abstract static class Builder { /** - * For internal usage. Please use `GetAgentEngineMemoryRequestParameters.builder()` for + * For internal usage. Please use `GetMemoryRevisionRequestParameters.builder()` for * instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_GetAgentEngineMemoryRequestParameters.Builder(); + return new AutoValue_GetMemoryRevisionRequestParameters.Builder(); } /** * Setter for name. * - *

name: Name of the agent engine. + *

name: Name of the Memory Revision. */ @JsonProperty("name") public abstract Builder name(String name); @@ -85,7 +85,7 @@ public Builder clearName() { *

config: */ @JsonProperty("config") - public abstract Builder config(GetAgentEngineMemoryConfig config); + public abstract Builder config(GetMemoryRevisionConfig config); /** * Setter for config builder. @@ -93,12 +93,12 @@ public Builder clearName() { *

config: */ @CanIgnoreReturnValue - public Builder config(GetAgentEngineMemoryConfig.Builder configBuilder) { + public Builder config(GetMemoryRevisionConfig.Builder configBuilder) { return config(configBuilder.build()); } @ExcludeFromGeneratedCoverageReport - abstract Builder config(Optional config); + abstract Builder config(Optional config); /** Clears the value of config field. */ @ExcludeFromGeneratedCoverageReport @@ -107,12 +107,12 @@ public Builder clearConfig() { return config(Optional.empty()); } - public abstract GetAgentEngineMemoryRequestParameters build(); + public abstract GetMemoryRevisionRequestParameters build(); } - /** Deserializes a JSON string to a GetAgentEngineMemoryRequestParameters object. */ + /** Deserializes a JSON string to a GetMemoryRevisionRequestParameters object. */ @ExcludeFromGeneratedCoverageReport - public static GetAgentEngineMemoryRequestParameters fromJson(String jsonString) { - return JsonSerializable.fromJsonString(jsonString, GetAgentEngineMemoryRequestParameters.class); + public static GetMemoryRevisionRequestParameters fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, GetMemoryRevisionRequestParameters.class); } } diff --git a/src/main/java/com/google/cloud/agentplatform/types/IngestEventsRequestParameters.java b/src/main/java/com/google/cloud/agentplatform/types/IngestEventsRequestParameters.java index 9a7582d..0a6060d 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/IngestEventsRequestParameters.java +++ b/src/main/java/com/google/cloud/agentplatform/types/IngestEventsRequestParameters.java @@ -28,12 +28,12 @@ import java.util.Map; import java.util.Optional; -/** Parameters for purging agent engine memories. */ +/** Parameters for ingesting events to Memory Bank. */ @AutoValue @InternalApi @JsonDeserialize(builder = IngestEventsRequestParameters.Builder.class) public abstract class IngestEventsRequestParameters extends JsonSerializable { - /** Name of the Agent Engine to ingest events into. */ + /** Name of the Memory Bank to ingest events into. */ @JsonProperty("name") public abstract Optional name(); @@ -85,7 +85,7 @@ private static Builder create() { /** * Setter for name. * - *

name: Name of the Agent Engine to ingest events into. + *

name: Name of the Memory Bank to ingest events into. */ @JsonProperty("name") public abstract Builder name(String name); diff --git a/src/main/java/com/google/cloud/agentplatform/types/ListAgentEngineMemoryRevisionsRequestParameters.java b/src/main/java/com/google/cloud/agentplatform/types/ListAgentEngineMemoryRevisionsRequestParameters.java deleted file mode 100644 index e0772bf..0000000 --- a/src/main/java/com/google/cloud/agentplatform/types/ListAgentEngineMemoryRevisionsRequestParameters.java +++ /dev/null @@ -1,119 +0,0 @@ -/* - * 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.types; - -import com.fasterxml.jackson.annotation.JsonCreator; -import com.fasterxml.jackson.annotation.JsonProperty; -import com.fasterxml.jackson.databind.annotation.JsonDeserialize; -import com.google.api.core.InternalApi; -import com.google.auto.value.AutoValue; -import com.google.errorprone.annotations.CanIgnoreReturnValue; -import com.google.genai.JsonSerializable; -import java.util.Optional; - -/** Parameters for listing Agent Engine memory revisions. */ -@AutoValue -@InternalApi -@JsonDeserialize(builder = ListAgentEngineMemoryRevisionsRequestParameters.Builder.class) -public abstract class ListAgentEngineMemoryRevisionsRequestParameters extends JsonSerializable { - /** Name of the Agent Engine memory */ - @JsonProperty("name") - public abstract Optional name(); - - /** */ - @JsonProperty("config") - public abstract Optional config(); - - /** Instantiates a builder for ListAgentEngineMemoryRevisionsRequestParameters. */ - @ExcludeFromGeneratedCoverageReport - public static Builder builder() { - return new AutoValue_ListAgentEngineMemoryRevisionsRequestParameters.Builder(); - } - - /** Creates a builder with the same values as this instance. */ - public abstract Builder toBuilder(); - - /** Builder for ListAgentEngineMemoryRevisionsRequestParameters. */ - @AutoValue.Builder - public abstract static class Builder { - /** - * For internal usage. Please use `ListAgentEngineMemoryRevisionsRequestParameters.builder()` - * for instantiation. - */ - @JsonCreator - private static Builder create() { - return new AutoValue_ListAgentEngineMemoryRevisionsRequestParameters.Builder(); - } - - /** - * Setter for name. - * - *

name: Name of the Agent Engine memory - */ - @JsonProperty("name") - public abstract Builder name(String name); - - @ExcludeFromGeneratedCoverageReport - abstract Builder name(Optional name); - - /** Clears the value of name field. */ - @ExcludeFromGeneratedCoverageReport - @CanIgnoreReturnValue - public Builder clearName() { - return name(Optional.empty()); - } - - /** - * Setter for config. - * - *

config: - */ - @JsonProperty("config") - public abstract Builder config(ListAgentEngineMemoryRevisionsConfig config); - - /** - * Setter for config builder. - * - *

config: - */ - @CanIgnoreReturnValue - public Builder config(ListAgentEngineMemoryRevisionsConfig.Builder configBuilder) { - return config(configBuilder.build()); - } - - @ExcludeFromGeneratedCoverageReport - abstract Builder config(Optional config); - - /** Clears the value of config field. */ - @ExcludeFromGeneratedCoverageReport - @CanIgnoreReturnValue - public Builder clearConfig() { - return config(Optional.empty()); - } - - public abstract ListAgentEngineMemoryRevisionsRequestParameters build(); - } - - /** Deserializes a JSON string to a ListAgentEngineMemoryRevisionsRequestParameters object. */ - @ExcludeFromGeneratedCoverageReport - public static ListAgentEngineMemoryRevisionsRequestParameters fromJson(String jsonString) { - return JsonSerializable.fromJsonString( - jsonString, ListAgentEngineMemoryRevisionsRequestParameters.class); - } -} diff --git a/src/main/java/com/google/cloud/agentplatform/types/ListAgentEngineMemoryConfig.java b/src/main/java/com/google/cloud/agentplatform/types/ListMemoriesConfig.java similarity index 85% rename from src/main/java/com/google/cloud/agentplatform/types/ListAgentEngineMemoryConfig.java rename to src/main/java/com/google/cloud/agentplatform/types/ListMemoriesConfig.java index 59c6c2e..1610335 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/ListAgentEngineMemoryConfig.java +++ b/src/main/java/com/google/cloud/agentplatform/types/ListMemoriesConfig.java @@ -27,10 +27,10 @@ import com.google.genai.types.HttpOptions; import java.util.Optional; -/** Config for listing agent engine memories. */ +/** Config for listing memories. */ @AutoValue -@JsonDeserialize(builder = ListAgentEngineMemoryConfig.Builder.class) -public abstract class ListAgentEngineMemoryConfig extends JsonSerializable { +@JsonDeserialize(builder = ListMemoriesConfig.Builder.class) +public abstract class ListMemoriesConfig extends JsonSerializable { /** Used to override HTTP request options. */ @JsonProperty("httpOptions") public abstract Optional httpOptions(); @@ -60,22 +60,22 @@ public abstract class ListAgentEngineMemoryConfig extends JsonSerializable { @JsonProperty("orderBy") public abstract Optional orderBy(); - /** Instantiates a builder for ListAgentEngineMemoryConfig. */ + /** Instantiates a builder for ListMemoriesConfig. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_ListAgentEngineMemoryConfig.Builder(); + return new AutoValue_ListMemoriesConfig.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for ListAgentEngineMemoryConfig. */ + /** Builder for ListMemoriesConfig. */ @AutoValue.Builder public abstract static class Builder { - /** For internal usage. Please use `ListAgentEngineMemoryConfig.builder()` for instantiation. */ + /** For internal usage. Please use `ListMemoriesConfig.builder()` for instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_ListAgentEngineMemoryConfig.Builder(); + return new AutoValue_ListMemoriesConfig.Builder(); } /** @@ -173,12 +173,12 @@ public Builder clearOrderBy() { return orderBy(Optional.empty()); } - public abstract ListAgentEngineMemoryConfig build(); + public abstract ListMemoriesConfig build(); } - /** Deserializes a JSON string to a ListAgentEngineMemoryConfig object. */ + /** Deserializes a JSON string to a ListMemoriesConfig object. */ @ExcludeFromGeneratedCoverageReport - public static ListAgentEngineMemoryConfig fromJson(String jsonString) { - return JsonSerializable.fromJsonString(jsonString, ListAgentEngineMemoryConfig.class); + public static ListMemoriesConfig fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, ListMemoriesConfig.class); } } diff --git a/src/main/java/com/google/cloud/agentplatform/types/GetAgentEngineMemoryRevisionRequestParameters.java b/src/main/java/com/google/cloud/agentplatform/types/ListMemoriesRequestParameters.java similarity index 62% rename from src/main/java/com/google/cloud/agentplatform/types/GetAgentEngineMemoryRevisionRequestParameters.java rename to src/main/java/com/google/cloud/agentplatform/types/ListMemoriesRequestParameters.java index 3535d68..ba0531a 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/GetAgentEngineMemoryRevisionRequestParameters.java +++ b/src/main/java/com/google/cloud/agentplatform/types/ListMemoriesRequestParameters.java @@ -27,44 +27,43 @@ import com.google.genai.JsonSerializable; import java.util.Optional; -/** Parameters for getting an Agent Engine memory revision. */ +/** Parameters for listing memories. */ @AutoValue @InternalApi -@JsonDeserialize(builder = GetAgentEngineMemoryRevisionRequestParameters.Builder.class) -public abstract class GetAgentEngineMemoryRevisionRequestParameters extends JsonSerializable { - /** Name of the agent engine. */ +@JsonDeserialize(builder = ListMemoriesRequestParameters.Builder.class) +public abstract class ListMemoriesRequestParameters extends JsonSerializable { + /** Name of the Memory Bank. */ @JsonProperty("name") public abstract Optional name(); /** */ @JsonProperty("config") - public abstract Optional config(); + public abstract Optional config(); - /** Instantiates a builder for GetAgentEngineMemoryRevisionRequestParameters. */ + /** Instantiates a builder for ListMemoriesRequestParameters. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_GetAgentEngineMemoryRevisionRequestParameters.Builder(); + return new AutoValue_ListMemoriesRequestParameters.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for GetAgentEngineMemoryRevisionRequestParameters. */ + /** Builder for ListMemoriesRequestParameters. */ @AutoValue.Builder public abstract static class Builder { /** - * For internal usage. Please use `GetAgentEngineMemoryRevisionRequestParameters.builder()` for - * instantiation. + * For internal usage. Please use `ListMemoriesRequestParameters.builder()` for instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_GetAgentEngineMemoryRevisionRequestParameters.Builder(); + return new AutoValue_ListMemoriesRequestParameters.Builder(); } /** * Setter for name. * - *

name: Name of the agent engine. + *

name: Name of the Memory Bank. */ @JsonProperty("name") public abstract Builder name(String name); @@ -85,7 +84,7 @@ public Builder clearName() { *

config: */ @JsonProperty("config") - public abstract Builder config(GetAgentEngineMemoryRevisionConfig config); + public abstract Builder config(ListMemoriesConfig config); /** * Setter for config builder. @@ -93,12 +92,12 @@ public Builder clearName() { *

config: */ @CanIgnoreReturnValue - public Builder config(GetAgentEngineMemoryRevisionConfig.Builder configBuilder) { + public Builder config(ListMemoriesConfig.Builder configBuilder) { return config(configBuilder.build()); } @ExcludeFromGeneratedCoverageReport - abstract Builder config(Optional config); + abstract Builder config(Optional config); /** Clears the value of config field. */ @ExcludeFromGeneratedCoverageReport @@ -107,13 +106,12 @@ public Builder clearConfig() { return config(Optional.empty()); } - public abstract GetAgentEngineMemoryRevisionRequestParameters build(); + public abstract ListMemoriesRequestParameters build(); } - /** Deserializes a JSON string to a GetAgentEngineMemoryRevisionRequestParameters object. */ + /** Deserializes a JSON string to a ListMemoriesRequestParameters object. */ @ExcludeFromGeneratedCoverageReport - public static GetAgentEngineMemoryRevisionRequestParameters fromJson(String jsonString) { - return JsonSerializable.fromJsonString( - jsonString, GetAgentEngineMemoryRevisionRequestParameters.class); + public static ListMemoriesRequestParameters fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, ListMemoriesRequestParameters.class); } } diff --git a/src/main/java/com/google/cloud/agentplatform/types/ListReasoningEnginesMemoriesResponse.java b/src/main/java/com/google/cloud/agentplatform/types/ListMemoriesResponse.java similarity index 77% rename from src/main/java/com/google/cloud/agentplatform/types/ListReasoningEnginesMemoriesResponse.java rename to src/main/java/com/google/cloud/agentplatform/types/ListMemoriesResponse.java index 3b7a026..671710e 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/ListReasoningEnginesMemoriesResponse.java +++ b/src/main/java/com/google/cloud/agentplatform/types/ListMemoriesResponse.java @@ -31,10 +31,10 @@ import java.util.List; import java.util.Optional; -/** Response for listing agent engine memories. */ +/** Response for listing memories. */ @AutoValue -@JsonDeserialize(builder = ListReasoningEnginesMemoriesResponse.Builder.class) -public abstract class ListReasoningEnginesMemoriesResponse extends JsonSerializable { +@JsonDeserialize(builder = ListMemoriesResponse.Builder.class) +public abstract class ListMemoriesResponse extends JsonSerializable { /** Used to retain the full HTTP response. */ @JsonProperty("sdkHttpResponse") public abstract Optional sdkHttpResponse(); @@ -43,29 +43,26 @@ public abstract class ListReasoningEnginesMemoriesResponse extends JsonSerializa @JsonProperty("nextPageToken") public abstract Optional nextPageToken(); - /** List of agent engine memories. */ + /** List of memories. */ @JsonProperty("memories") public abstract Optional> memories(); - /** Instantiates a builder for ListReasoningEnginesMemoriesResponse. */ + /** Instantiates a builder for ListMemoriesResponse. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_ListReasoningEnginesMemoriesResponse.Builder(); + return new AutoValue_ListMemoriesResponse.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for ListReasoningEnginesMemoriesResponse. */ + /** Builder for ListMemoriesResponse. */ @AutoValue.Builder public abstract static class Builder { - /** - * For internal usage. Please use `ListReasoningEnginesMemoriesResponse.builder()` for - * instantiation. - */ + /** For internal usage. Please use `ListMemoriesResponse.builder()` for instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_ListReasoningEnginesMemoriesResponse.Builder(); + return new AutoValue_ListMemoriesResponse.Builder(); } /** @@ -107,7 +104,7 @@ public Builder clearNextPageToken() { /** * Setter for memories. * - *

memories: List of agent engine memories. + *

memories: List of memories. */ @JsonProperty("memories") public abstract Builder memories(List memories); @@ -115,7 +112,7 @@ public Builder clearNextPageToken() { /** * Setter for memories. * - *

memories: List of agent engine memories. + *

memories: List of memories. */ @CanIgnoreReturnValue public Builder memories(Memory... memories) { @@ -125,7 +122,7 @@ public Builder memories(Memory... memories) { /** * Setter for memories builder. * - *

memories: List of agent engine memories. + *

memories: List of memories. */ @CanIgnoreReturnValue public Builder memories(Memory.Builder... memoriesBuilders) { @@ -145,12 +142,12 @@ public Builder clearMemories() { return memories(Optional.empty()); } - public abstract ListReasoningEnginesMemoriesResponse build(); + public abstract ListMemoriesResponse build(); } - /** Deserializes a JSON string to a ListReasoningEnginesMemoriesResponse object. */ + /** Deserializes a JSON string to a ListMemoriesResponse object. */ @ExcludeFromGeneratedCoverageReport - public static ListReasoningEnginesMemoriesResponse fromJson(String jsonString) { - return JsonSerializable.fromJsonString(jsonString, ListReasoningEnginesMemoriesResponse.class); + public static ListMemoriesResponse fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, ListMemoriesResponse.class); } } diff --git a/src/main/java/com/google/cloud/agentplatform/types/ListAgentEngineMemoryRevisionsConfig.java b/src/main/java/com/google/cloud/agentplatform/types/ListMemoryRevisionsConfig.java similarity index 80% rename from src/main/java/com/google/cloud/agentplatform/types/ListAgentEngineMemoryRevisionsConfig.java rename to src/main/java/com/google/cloud/agentplatform/types/ListMemoryRevisionsConfig.java index a853c00..d83c36b 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/ListAgentEngineMemoryRevisionsConfig.java +++ b/src/main/java/com/google/cloud/agentplatform/types/ListMemoryRevisionsConfig.java @@ -27,10 +27,10 @@ import com.google.genai.types.HttpOptions; import java.util.Optional; -/** Config for listing Agent Engine memory revisions. */ +/** Config for listing memory revisions. */ @AutoValue -@JsonDeserialize(builder = ListAgentEngineMemoryRevisionsConfig.Builder.class) -public abstract class ListAgentEngineMemoryRevisionsConfig extends JsonSerializable { +@JsonDeserialize(builder = ListMemoryRevisionsConfig.Builder.class) +public abstract class ListMemoryRevisionsConfig extends JsonSerializable { /** Used to override HTTP request options. */ @JsonProperty("httpOptions") public abstract Optional httpOptions(); @@ -50,25 +50,22 @@ public abstract class ListAgentEngineMemoryRevisionsConfig extends JsonSerializa @JsonProperty("filter") public abstract Optional filter(); - /** Instantiates a builder for ListAgentEngineMemoryRevisionsConfig. */ + /** Instantiates a builder for ListMemoryRevisionsConfig. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_ListAgentEngineMemoryRevisionsConfig.Builder(); + return new AutoValue_ListMemoryRevisionsConfig.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for ListAgentEngineMemoryRevisionsConfig. */ + /** Builder for ListMemoryRevisionsConfig. */ @AutoValue.Builder public abstract static class Builder { - /** - * For internal usage. Please use `ListAgentEngineMemoryRevisionsConfig.builder()` for - * instantiation. - */ + /** For internal usage. Please use `ListMemoryRevisionsConfig.builder()` for instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_ListAgentEngineMemoryRevisionsConfig.Builder(); + return new AutoValue_ListMemoryRevisionsConfig.Builder(); } /** @@ -144,12 +141,12 @@ public Builder clearFilter() { return filter(Optional.empty()); } - public abstract ListAgentEngineMemoryRevisionsConfig build(); + public abstract ListMemoryRevisionsConfig build(); } - /** Deserializes a JSON string to a ListAgentEngineMemoryRevisionsConfig object. */ + /** Deserializes a JSON string to a ListMemoryRevisionsConfig object. */ @ExcludeFromGeneratedCoverageReport - public static ListAgentEngineMemoryRevisionsConfig fromJson(String jsonString) { - return JsonSerializable.fromJsonString(jsonString, ListAgentEngineMemoryRevisionsConfig.class); + public static ListMemoryRevisionsConfig fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, ListMemoryRevisionsConfig.class); } } diff --git a/src/main/java/com/google/cloud/agentplatform/types/ListAgentEngineMemoryRequestParameters.java b/src/main/java/com/google/cloud/agentplatform/types/ListMemoryRevisionsRequestParameters.java similarity index 65% rename from src/main/java/com/google/cloud/agentplatform/types/ListAgentEngineMemoryRequestParameters.java rename to src/main/java/com/google/cloud/agentplatform/types/ListMemoryRevisionsRequestParameters.java index a39fd01..449c78f 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/ListAgentEngineMemoryRequestParameters.java +++ b/src/main/java/com/google/cloud/agentplatform/types/ListMemoryRevisionsRequestParameters.java @@ -27,44 +27,44 @@ import com.google.genai.JsonSerializable; import java.util.Optional; -/** Parameters for listing agent engines. */ +/** Parameters for listing memory revisions. */ @AutoValue @InternalApi -@JsonDeserialize(builder = ListAgentEngineMemoryRequestParameters.Builder.class) -public abstract class ListAgentEngineMemoryRequestParameters extends JsonSerializable { - /** Name of the agent engine. */ +@JsonDeserialize(builder = ListMemoryRevisionsRequestParameters.Builder.class) +public abstract class ListMemoryRevisionsRequestParameters extends JsonSerializable { + /** Name of the memory */ @JsonProperty("name") public abstract Optional name(); /** */ @JsonProperty("config") - public abstract Optional config(); + public abstract Optional config(); - /** Instantiates a builder for ListAgentEngineMemoryRequestParameters. */ + /** Instantiates a builder for ListMemoryRevisionsRequestParameters. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_ListAgentEngineMemoryRequestParameters.Builder(); + return new AutoValue_ListMemoryRevisionsRequestParameters.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for ListAgentEngineMemoryRequestParameters. */ + /** Builder for ListMemoryRevisionsRequestParameters. */ @AutoValue.Builder public abstract static class Builder { /** - * For internal usage. Please use `ListAgentEngineMemoryRequestParameters.builder()` for + * For internal usage. Please use `ListMemoryRevisionsRequestParameters.builder()` for * instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_ListAgentEngineMemoryRequestParameters.Builder(); + return new AutoValue_ListMemoryRevisionsRequestParameters.Builder(); } /** * Setter for name. * - *

name: Name of the agent engine. + *

name: Name of the memory */ @JsonProperty("name") public abstract Builder name(String name); @@ -85,7 +85,7 @@ public Builder clearName() { *

config: */ @JsonProperty("config") - public abstract Builder config(ListAgentEngineMemoryConfig config); + public abstract Builder config(ListMemoryRevisionsConfig config); /** * Setter for config builder. @@ -93,12 +93,12 @@ public Builder clearName() { *

config: */ @CanIgnoreReturnValue - public Builder config(ListAgentEngineMemoryConfig.Builder configBuilder) { + public Builder config(ListMemoryRevisionsConfig.Builder configBuilder) { return config(configBuilder.build()); } @ExcludeFromGeneratedCoverageReport - abstract Builder config(Optional config); + abstract Builder config(Optional config); /** Clears the value of config field. */ @ExcludeFromGeneratedCoverageReport @@ -107,13 +107,12 @@ public Builder clearConfig() { return config(Optional.empty()); } - public abstract ListAgentEngineMemoryRequestParameters build(); + public abstract ListMemoryRevisionsRequestParameters build(); } - /** Deserializes a JSON string to a ListAgentEngineMemoryRequestParameters object. */ + /** Deserializes a JSON string to a ListMemoryRevisionsRequestParameters object. */ @ExcludeFromGeneratedCoverageReport - public static ListAgentEngineMemoryRequestParameters fromJson(String jsonString) { - return JsonSerializable.fromJsonString( - jsonString, ListAgentEngineMemoryRequestParameters.class); + public static ListMemoryRevisionsRequestParameters fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, ListMemoryRevisionsRequestParameters.class); } } diff --git a/src/main/java/com/google/cloud/agentplatform/types/ListAgentEngineMemoryRevisionsResponse.java b/src/main/java/com/google/cloud/agentplatform/types/ListMemoryRevisionsResponse.java similarity index 81% rename from src/main/java/com/google/cloud/agentplatform/types/ListAgentEngineMemoryRevisionsResponse.java rename to src/main/java/com/google/cloud/agentplatform/types/ListMemoryRevisionsResponse.java index 3b824e4..9152a41 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/ListAgentEngineMemoryRevisionsResponse.java +++ b/src/main/java/com/google/cloud/agentplatform/types/ListMemoryRevisionsResponse.java @@ -31,10 +31,10 @@ import java.util.List; import java.util.Optional; -/** Response for listing agent engine memory revisions. */ +/** Response for listing memory revisions. */ @AutoValue -@JsonDeserialize(builder = ListAgentEngineMemoryRevisionsResponse.Builder.class) -public abstract class ListAgentEngineMemoryRevisionsResponse extends JsonSerializable { +@JsonDeserialize(builder = ListMemoryRevisionsResponse.Builder.class) +public abstract class ListMemoryRevisionsResponse extends JsonSerializable { /** Used to retain the full HTTP response. */ @JsonProperty("sdkHttpResponse") public abstract Optional sdkHttpResponse(); @@ -47,25 +47,22 @@ public abstract class ListAgentEngineMemoryRevisionsResponse extends JsonSeriali @JsonProperty("memoryRevisions") public abstract Optional> memoryRevisions(); - /** Instantiates a builder for ListAgentEngineMemoryRevisionsResponse. */ + /** Instantiates a builder for ListMemoryRevisionsResponse. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_ListAgentEngineMemoryRevisionsResponse.Builder(); + return new AutoValue_ListMemoryRevisionsResponse.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for ListAgentEngineMemoryRevisionsResponse. */ + /** Builder for ListMemoryRevisionsResponse. */ @AutoValue.Builder public abstract static class Builder { - /** - * For internal usage. Please use `ListAgentEngineMemoryRevisionsResponse.builder()` for - * instantiation. - */ + /** For internal usage. Please use `ListMemoryRevisionsResponse.builder()` for instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_ListAgentEngineMemoryRevisionsResponse.Builder(); + return new AutoValue_ListMemoryRevisionsResponse.Builder(); } /** @@ -145,13 +142,12 @@ public Builder clearMemoryRevisions() { return memoryRevisions(Optional.empty()); } - public abstract ListAgentEngineMemoryRevisionsResponse build(); + public abstract ListMemoryRevisionsResponse build(); } - /** Deserializes a JSON string to a ListAgentEngineMemoryRevisionsResponse object. */ + /** Deserializes a JSON string to a ListMemoryRevisionsResponse object. */ @ExcludeFromGeneratedCoverageReport - public static ListAgentEngineMemoryRevisionsResponse fromJson(String jsonString) { - return JsonSerializable.fromJsonString( - jsonString, ListAgentEngineMemoryRevisionsResponse.class); + public static ListMemoryRevisionsResponse fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, ListMemoryRevisionsResponse.class); } } diff --git a/src/main/java/com/google/cloud/agentplatform/types/MemoryBank.java b/src/main/java/com/google/cloud/agentplatform/types/MemoryBank.java new file mode 100644 index 0000000..3a2d6ee --- /dev/null +++ b/src/main/java/com/google/cloud/agentplatform/types/MemoryBank.java @@ -0,0 +1,85 @@ +/* + * 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.types; + +import com.fasterxml.jackson.annotation.JsonCreator; +import com.fasterxml.jackson.annotation.JsonProperty; +import com.fasterxml.jackson.databind.annotation.JsonDeserialize; +import com.google.auto.value.AutoValue; +import com.google.errorprone.annotations.CanIgnoreReturnValue; +import com.google.genai.JsonSerializable; +import java.util.Optional; + +/** A memory bank. */ +@AutoValue +@JsonDeserialize(builder = MemoryBank.Builder.class) +public abstract class MemoryBank extends JsonSerializable { + /** + * Required. Represents the ID of the schema. Must be 1-63 characters, start with a lowercase + * letter, and consist of lowercase letters, numbers, and hyphens. + */ + @JsonProperty("name") + public abstract Optional name(); + + /** Instantiates a builder for MemoryBank. */ + @ExcludeFromGeneratedCoverageReport + public static Builder builder() { + return new AutoValue_MemoryBank.Builder(); + } + + /** Creates a builder with the same values as this instance. */ + public abstract Builder toBuilder(); + + /** Builder for MemoryBank. */ + @AutoValue.Builder + public abstract static class Builder { + /** For internal usage. Please use `MemoryBank.builder()` for instantiation. */ + @JsonCreator + private static Builder create() { + return new AutoValue_MemoryBank.Builder(); + } + + /** + * Setter for name. + * + *

name: Required. Represents the ID of the schema. Must be 1-63 characters, start with a + * lowercase letter, and consist of lowercase letters, numbers, and hyphens. + */ + @JsonProperty("name") + public abstract Builder name(String name); + + @ExcludeFromGeneratedCoverageReport + abstract Builder name(Optional name); + + /** Clears the value of name field. */ + @ExcludeFromGeneratedCoverageReport + @CanIgnoreReturnValue + public Builder clearName() { + return name(Optional.empty()); + } + + public abstract MemoryBank build(); + } + + /** Deserializes a JSON string to a MemoryBank object. */ + @ExcludeFromGeneratedCoverageReport + public static MemoryBank fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, MemoryBank.class); + } +} diff --git a/src/main/java/com/google/cloud/agentplatform/types/MemoryBankOperation.java b/src/main/java/com/google/cloud/agentplatform/types/MemoryBankOperation.java new file mode 100644 index 0000000..0005768 --- /dev/null +++ b/src/main/java/com/google/cloud/agentplatform/types/MemoryBankOperation.java @@ -0,0 +1,198 @@ +/* + * 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.types; + +import com.fasterxml.jackson.annotation.JsonCreator; +import com.fasterxml.jackson.annotation.JsonProperty; +import com.fasterxml.jackson.databind.annotation.JsonDeserialize; +import com.google.auto.value.AutoValue; +import com.google.errorprone.annotations.CanIgnoreReturnValue; +import com.google.genai.JsonSerializable; +import java.util.Map; +import java.util.Optional; + +/** Operation that has an memory bank as a response. */ +@AutoValue +@JsonDeserialize(builder = MemoryBankOperation.Builder.class) +public abstract class MemoryBankOperation extends JsonSerializable { + /** + * The server-assigned name, which is only unique within the same service that originally returns + * it. If you use the default HTTP mapping, the `name` should be a resource name ending with + * `operations/{unique_id}`. + */ + @JsonProperty("name") + public abstract Optional name(); + + /** + * Service-specific metadata associated with the operation. It typically contains progress + * information and common metadata such as create time. Some services might not provide such + * metadata. Any method that returns a long-running operation should document the metadata type, + * if any. + */ + @JsonProperty("metadata") + public abstract Optional> metadata(); + + /** + * If the value is `false`, it means the operation is still in progress. If `true`, the operation + * is completed, and either `error` or `response` is available. + */ + @JsonProperty("done") + public abstract Optional done(); + + /** The error result of the operation in case of failure or cancellation. */ + @JsonProperty("error") + public abstract Optional> error(); + + /** The created Memory Bank. */ + @JsonProperty("response") + public abstract Optional response(); + + /** Instantiates a builder for MemoryBankOperation. */ + @ExcludeFromGeneratedCoverageReport + public static Builder builder() { + return new AutoValue_MemoryBankOperation.Builder(); + } + + /** Creates a builder with the same values as this instance. */ + public abstract Builder toBuilder(); + + /** Builder for MemoryBankOperation. */ + @AutoValue.Builder + public abstract static class Builder { + /** For internal usage. Please use `MemoryBankOperation.builder()` for instantiation. */ + @JsonCreator + private static Builder create() { + return new AutoValue_MemoryBankOperation.Builder(); + } + + /** + * Setter for name. + * + *

name: The server-assigned name, which is only unique within the same service that + * originally returns it. If you use the default HTTP mapping, the `name` should be a resource + * name ending with `operations/{unique_id}`. + */ + @JsonProperty("name") + public abstract Builder name(String name); + + @ExcludeFromGeneratedCoverageReport + abstract Builder name(Optional name); + + /** Clears the value of name field. */ + @ExcludeFromGeneratedCoverageReport + @CanIgnoreReturnValue + public Builder clearName() { + return name(Optional.empty()); + } + + /** + * Setter for metadata. + * + *

metadata: Service-specific metadata associated with the operation. It typically contains + * progress information and common metadata such as create time. Some services might not provide + * such metadata. Any method that returns a long-running operation should document the metadata + * type, if any. + */ + @JsonProperty("metadata") + public abstract Builder metadata(Map metadata); + + @ExcludeFromGeneratedCoverageReport + abstract Builder metadata(Optional> metadata); + + /** Clears the value of metadata field. */ + @ExcludeFromGeneratedCoverageReport + @CanIgnoreReturnValue + public Builder clearMetadata() { + return metadata(Optional.empty()); + } + + /** + * Setter for done. + * + *

done: If the value is `false`, it means the operation is still in progress. If `true`, the + * operation is completed, and either `error` or `response` is available. + */ + @JsonProperty("done") + public abstract Builder done(boolean done); + + @ExcludeFromGeneratedCoverageReport + abstract Builder done(Optional done); + + /** Clears the value of done field. */ + @ExcludeFromGeneratedCoverageReport + @CanIgnoreReturnValue + public Builder clearDone() { + return done(Optional.empty()); + } + + /** + * Setter for error. + * + *

error: The error result of the operation in case of failure or cancellation. + */ + @JsonProperty("error") + public abstract Builder error(Map error); + + @ExcludeFromGeneratedCoverageReport + abstract Builder error(Optional> error); + + /** Clears the value of error field. */ + @ExcludeFromGeneratedCoverageReport + @CanIgnoreReturnValue + public Builder clearError() { + return error(Optional.empty()); + } + + /** + * Setter for response. + * + *

response: The created Memory Bank. + */ + @JsonProperty("response") + public abstract Builder response(MemoryBank response); + + /** + * Setter for response builder. + * + *

response: The created Memory Bank. + */ + @CanIgnoreReturnValue + public Builder response(MemoryBank.Builder responseBuilder) { + return response(responseBuilder.build()); + } + + @ExcludeFromGeneratedCoverageReport + abstract Builder response(Optional response); + + /** Clears the value of response field. */ + @ExcludeFromGeneratedCoverageReport + @CanIgnoreReturnValue + public Builder clearResponse() { + return response(Optional.empty()); + } + + public abstract MemoryBankOperation build(); + } + + /** Deserializes a JSON string to a MemoryBankOperation object. */ + @ExcludeFromGeneratedCoverageReport + public static MemoryBankOperation fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, MemoryBankOperation.class); + } +} diff --git a/src/main/java/com/google/cloud/agentplatform/types/AgentEngineMemoryConfig.java b/src/main/java/com/google/cloud/agentplatform/types/MemoryConfig.java similarity index 94% rename from src/main/java/com/google/cloud/agentplatform/types/AgentEngineMemoryConfig.java rename to src/main/java/com/google/cloud/agentplatform/types/MemoryConfig.java index 1cf1d15..06c0b61 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/AgentEngineMemoryConfig.java +++ b/src/main/java/com/google/cloud/agentplatform/types/MemoryConfig.java @@ -36,8 +36,8 @@ /** Config for creating a Memory. */ @AutoValue -@JsonDeserialize(builder = AgentEngineMemoryConfig.Builder.class) -public abstract class AgentEngineMemoryConfig extends JsonSerializable { +@JsonDeserialize(builder = MemoryConfig.Builder.class) +public abstract class MemoryConfig extends JsonSerializable { /** Used to override HTTP request options. */ @JsonProperty("httpOptions") public abstract Optional httpOptions(); @@ -104,22 +104,22 @@ public abstract class AgentEngineMemoryConfig extends JsonSerializable { @JsonProperty("memoryId") public abstract Optional memoryId(); - /** Instantiates a builder for AgentEngineMemoryConfig. */ + /** Instantiates a builder for MemoryConfig. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_AgentEngineMemoryConfig.Builder(); + return new AutoValue_MemoryConfig.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for AgentEngineMemoryConfig. */ + /** Builder for MemoryConfig. */ @AutoValue.Builder public abstract static class Builder { - /** For internal usage. Please use `AgentEngineMemoryConfig.builder()` for instantiation. */ + /** For internal usage. Please use `MemoryConfig.builder()` for instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_AgentEngineMemoryConfig.Builder(); + return new AutoValue_MemoryConfig.Builder(); } /** @@ -371,12 +371,12 @@ public Builder clearMemoryId() { return memoryId(Optional.empty()); } - public abstract AgentEngineMemoryConfig build(); + public abstract MemoryConfig build(); } - /** Deserializes a JSON string to a AgentEngineMemoryConfig object. */ + /** Deserializes a JSON string to a MemoryConfig object. */ @ExcludeFromGeneratedCoverageReport - public static AgentEngineMemoryConfig fromJson(String jsonString) { - return JsonSerializable.fromJsonString(jsonString, AgentEngineMemoryConfig.class); + public static MemoryConfig fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, MemoryConfig.class); } } diff --git a/src/main/java/com/google/cloud/agentplatform/types/AgentEngineMemoryOperation.java b/src/main/java/com/google/cloud/agentplatform/types/MemoryOperation.java similarity index 85% rename from src/main/java/com/google/cloud/agentplatform/types/AgentEngineMemoryOperation.java rename to src/main/java/com/google/cloud/agentplatform/types/MemoryOperation.java index d2860bb..d0c1dd6 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/AgentEngineMemoryOperation.java +++ b/src/main/java/com/google/cloud/agentplatform/types/MemoryOperation.java @@ -27,10 +27,10 @@ import java.util.Map; import java.util.Optional; -/** Operation that has an agent engine memory as a response. */ +/** Operation that has a memory as a response. */ @AutoValue -@JsonDeserialize(builder = AgentEngineMemoryOperation.Builder.class) -public abstract class AgentEngineMemoryOperation extends JsonSerializable { +@JsonDeserialize(builder = MemoryOperation.Builder.class) +public abstract class MemoryOperation extends JsonSerializable { /** * The server-assigned name, which is only unique within the same service that originally returns * it. If you use the default HTTP mapping, the `name` should be a resource name ending with @@ -59,26 +59,26 @@ public abstract class AgentEngineMemoryOperation extends JsonSerializable { @JsonProperty("error") public abstract Optional> error(); - /** The Agent Engine Memory. */ + /** The Memory. */ @JsonProperty("response") public abstract Optional response(); - /** Instantiates a builder for AgentEngineMemoryOperation. */ + /** Instantiates a builder for MemoryOperation. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_AgentEngineMemoryOperation.Builder(); + return new AutoValue_MemoryOperation.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for AgentEngineMemoryOperation. */ + /** Builder for MemoryOperation. */ @AutoValue.Builder public abstract static class Builder { - /** For internal usage. Please use `AgentEngineMemoryOperation.builder()` for instantiation. */ + /** For internal usage. Please use `MemoryOperation.builder()` for instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_AgentEngineMemoryOperation.Builder(); + return new AutoValue_MemoryOperation.Builder(); } /** @@ -162,7 +162,7 @@ public Builder clearError() { /** * Setter for response. * - *

response: The Agent Engine Memory. + *

response: The Memory. */ @JsonProperty("response") public abstract Builder response(Memory response); @@ -170,7 +170,7 @@ public Builder clearError() { /** * Setter for response builder. * - *

response: The Agent Engine Memory. + *

response: The Memory. */ @CanIgnoreReturnValue public Builder response(Memory.Builder responseBuilder) { @@ -187,12 +187,12 @@ public Builder clearResponse() { return response(Optional.empty()); } - public abstract AgentEngineMemoryOperation build(); + public abstract MemoryOperation build(); } - /** Deserializes a JSON string to a AgentEngineMemoryOperation object. */ + /** Deserializes a JSON string to a MemoryOperation object. */ @ExcludeFromGeneratedCoverageReport - public static AgentEngineMemoryOperation fromJson(String jsonString) { - return JsonSerializable.fromJsonString(jsonString, AgentEngineMemoryOperation.class); + public static MemoryOperation fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, MemoryOperation.class); } } diff --git a/src/main/java/com/google/cloud/agentplatform/types/PurgeAgentEngineMemoriesConfig.java b/src/main/java/com/google/cloud/agentplatform/types/PurgeMemoriesConfig.java similarity index 77% rename from src/main/java/com/google/cloud/agentplatform/types/PurgeAgentEngineMemoriesConfig.java rename to src/main/java/com/google/cloud/agentplatform/types/PurgeMemoriesConfig.java index 8b88b39..17dbf58 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/PurgeAgentEngineMemoriesConfig.java +++ b/src/main/java/com/google/cloud/agentplatform/types/PurgeMemoriesConfig.java @@ -29,8 +29,8 @@ /** Config for purging memories. */ @AutoValue -@JsonDeserialize(builder = PurgeAgentEngineMemoriesConfig.Builder.class) -public abstract class PurgeAgentEngineMemoriesConfig extends JsonSerializable { +@JsonDeserialize(builder = PurgeMemoriesConfig.Builder.class) +public abstract class PurgeMemoriesConfig extends JsonSerializable { /** Used to override HTTP request options. */ @JsonProperty("httpOptions") public abstract Optional httpOptions(); @@ -39,24 +39,22 @@ public abstract class PurgeAgentEngineMemoriesConfig extends JsonSerializable { @JsonProperty("waitForCompletion") public abstract Optional waitForCompletion(); - /** Instantiates a builder for PurgeAgentEngineMemoriesConfig. */ + /** Instantiates a builder for PurgeMemoriesConfig. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_PurgeAgentEngineMemoriesConfig.Builder(); + return new AutoValue_PurgeMemoriesConfig.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for PurgeAgentEngineMemoriesConfig. */ + /** Builder for PurgeMemoriesConfig. */ @AutoValue.Builder public abstract static class Builder { - /** - * For internal usage. Please use `PurgeAgentEngineMemoriesConfig.builder()` for instantiation. - */ + /** For internal usage. Please use `PurgeMemoriesConfig.builder()` for instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_PurgeAgentEngineMemoriesConfig.Builder(); + return new AutoValue_PurgeMemoriesConfig.Builder(); } /** @@ -95,12 +93,12 @@ public Builder clearWaitForCompletion() { return waitForCompletion(Optional.empty()); } - public abstract PurgeAgentEngineMemoriesConfig build(); + public abstract PurgeMemoriesConfig build(); } - /** Deserializes a JSON string to a PurgeAgentEngineMemoriesConfig object. */ + /** Deserializes a JSON string to a PurgeMemoriesConfig object. */ @ExcludeFromGeneratedCoverageReport - public static PurgeAgentEngineMemoriesConfig fromJson(String jsonString) { - return JsonSerializable.fromJsonString(jsonString, PurgeAgentEngineMemoriesConfig.class); + public static PurgeMemoriesConfig fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, PurgeMemoriesConfig.class); } } diff --git a/src/main/java/com/google/cloud/agentplatform/types/AgentEnginePurgeMemoriesOperation.java b/src/main/java/com/google/cloud/agentplatform/types/PurgeMemoriesOperation.java similarity index 86% rename from src/main/java/com/google/cloud/agentplatform/types/AgentEnginePurgeMemoriesOperation.java rename to src/main/java/com/google/cloud/agentplatform/types/PurgeMemoriesOperation.java index a0a9324..7a164e4 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/AgentEnginePurgeMemoriesOperation.java +++ b/src/main/java/com/google/cloud/agentplatform/types/PurgeMemoriesOperation.java @@ -27,10 +27,10 @@ import java.util.Map; import java.util.Optional; -/** Operation that purges memories from an agent engine. */ +/** Operation that purges memories from a Memory Bank. */ @AutoValue -@JsonDeserialize(builder = AgentEnginePurgeMemoriesOperation.Builder.class) -public abstract class AgentEnginePurgeMemoriesOperation extends JsonSerializable { +@JsonDeserialize(builder = PurgeMemoriesOperation.Builder.class) +public abstract class PurgeMemoriesOperation extends JsonSerializable { /** * The server-assigned name, which is only unique within the same service that originally returns * it. If you use the default HTTP mapping, the `name` should be a resource name ending with @@ -63,25 +63,22 @@ public abstract class AgentEnginePurgeMemoriesOperation extends JsonSerializable @JsonProperty("response") public abstract Optional response(); - /** Instantiates a builder for AgentEnginePurgeMemoriesOperation. */ + /** Instantiates a builder for PurgeMemoriesOperation. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_AgentEnginePurgeMemoriesOperation.Builder(); + return new AutoValue_PurgeMemoriesOperation.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for AgentEnginePurgeMemoriesOperation. */ + /** Builder for PurgeMemoriesOperation. */ @AutoValue.Builder public abstract static class Builder { - /** - * For internal usage. Please use `AgentEnginePurgeMemoriesOperation.builder()` for - * instantiation. - */ + /** For internal usage. Please use `PurgeMemoriesOperation.builder()` for instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_AgentEnginePurgeMemoriesOperation.Builder(); + return new AutoValue_PurgeMemoriesOperation.Builder(); } /** @@ -190,12 +187,12 @@ public Builder clearResponse() { return response(Optional.empty()); } - public abstract AgentEnginePurgeMemoriesOperation build(); + public abstract PurgeMemoriesOperation build(); } - /** Deserializes a JSON string to a AgentEnginePurgeMemoriesOperation object. */ + /** Deserializes a JSON string to a PurgeMemoriesOperation object. */ @ExcludeFromGeneratedCoverageReport - public static AgentEnginePurgeMemoriesOperation fromJson(String jsonString) { - return JsonSerializable.fromJsonString(jsonString, AgentEnginePurgeMemoriesOperation.class); + public static PurgeMemoriesOperation fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, PurgeMemoriesOperation.class); } } diff --git a/src/main/java/com/google/cloud/agentplatform/types/PurgeAgentEngineMemoriesRequestParameters.java b/src/main/java/com/google/cloud/agentplatform/types/PurgeMemoriesRequestParameters.java similarity index 84% rename from src/main/java/com/google/cloud/agentplatform/types/PurgeAgentEngineMemoriesRequestParameters.java rename to src/main/java/com/google/cloud/agentplatform/types/PurgeMemoriesRequestParameters.java index 8ec52da..bf072b8 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/PurgeAgentEngineMemoriesRequestParameters.java +++ b/src/main/java/com/google/cloud/agentplatform/types/PurgeMemoriesRequestParameters.java @@ -31,12 +31,12 @@ import java.util.List; import java.util.Optional; -/** Parameters for purging agent engine memories. */ +/** Parameters for purging memories. */ @AutoValue @InternalApi -@JsonDeserialize(builder = PurgeAgentEngineMemoriesRequestParameters.Builder.class) -public abstract class PurgeAgentEngineMemoriesRequestParameters extends JsonSerializable { - /** Name of the Agent Engine to purge memories from. */ +@JsonDeserialize(builder = PurgeMemoriesRequestParameters.Builder.class) +public abstract class PurgeMemoriesRequestParameters extends JsonSerializable { + /** Name of the Memory Bank to purge memories from. */ @JsonProperty("name") public abstract Optional name(); @@ -70,33 +70,32 @@ public abstract class PurgeAgentEngineMemoriesRequestParameters extends JsonSeri /** */ @JsonProperty("config") - public abstract Optional config(); + public abstract Optional config(); - /** Instantiates a builder for PurgeAgentEngineMemoriesRequestParameters. */ + /** Instantiates a builder for PurgeMemoriesRequestParameters. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_PurgeAgentEngineMemoriesRequestParameters.Builder(); + return new AutoValue_PurgeMemoriesRequestParameters.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for PurgeAgentEngineMemoriesRequestParameters. */ + /** Builder for PurgeMemoriesRequestParameters. */ @AutoValue.Builder public abstract static class Builder { /** - * For internal usage. Please use `PurgeAgentEngineMemoriesRequestParameters.builder()` for - * instantiation. + * For internal usage. Please use `PurgeMemoriesRequestParameters.builder()` for instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_PurgeAgentEngineMemoriesRequestParameters.Builder(); + return new AutoValue_PurgeMemoriesRequestParameters.Builder(); } /** * Setter for name. * - *

name: Name of the Agent Engine to purge memories from. + *

name: Name of the Memory Bank to purge memories from. */ @JsonProperty("name") public abstract Builder name(String name); @@ -220,7 +219,7 @@ public Builder clearForce() { *

config: */ @JsonProperty("config") - public abstract Builder config(PurgeAgentEngineMemoriesConfig config); + public abstract Builder config(PurgeMemoriesConfig config); /** * Setter for config builder. @@ -228,12 +227,12 @@ public Builder clearForce() { *

config: */ @CanIgnoreReturnValue - public Builder config(PurgeAgentEngineMemoriesConfig.Builder configBuilder) { + public Builder config(PurgeMemoriesConfig.Builder configBuilder) { return config(configBuilder.build()); } @ExcludeFromGeneratedCoverageReport - abstract Builder config(Optional config); + abstract Builder config(Optional config); /** Clears the value of config field. */ @ExcludeFromGeneratedCoverageReport @@ -242,13 +241,12 @@ public Builder clearConfig() { return config(Optional.empty()); } - public abstract PurgeAgentEngineMemoriesRequestParameters build(); + public abstract PurgeMemoriesRequestParameters build(); } - /** Deserializes a JSON string to a PurgeAgentEngineMemoriesRequestParameters object. */ + /** Deserializes a JSON string to a PurgeMemoriesRequestParameters object. */ @ExcludeFromGeneratedCoverageReport - public static PurgeAgentEngineMemoriesRequestParameters fromJson(String jsonString) { - return JsonSerializable.fromJsonString( - jsonString, PurgeAgentEngineMemoriesRequestParameters.class); + public static PurgeMemoriesRequestParameters fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, PurgeMemoriesRequestParameters.class); } } diff --git a/src/main/java/com/google/cloud/agentplatform/types/RetrieveAgentEngineMemoriesConfig.java b/src/main/java/com/google/cloud/agentplatform/types/RetrieveMemoriesConfig.java similarity index 92% rename from src/main/java/com/google/cloud/agentplatform/types/RetrieveAgentEngineMemoriesConfig.java rename to src/main/java/com/google/cloud/agentplatform/types/RetrieveMemoriesConfig.java index 0e690d7..1288aa4 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/RetrieveAgentEngineMemoriesConfig.java +++ b/src/main/java/com/google/cloud/agentplatform/types/RetrieveMemoriesConfig.java @@ -34,8 +34,8 @@ /** Config for retrieving memories. */ @AutoValue -@JsonDeserialize(builder = RetrieveAgentEngineMemoriesConfig.Builder.class) -public abstract class RetrieveAgentEngineMemoriesConfig extends JsonSerializable { +@JsonDeserialize(builder = RetrieveMemoriesConfig.Builder.class) +public abstract class RetrieveMemoriesConfig extends JsonSerializable { /** Used to override HTTP request options. */ @JsonProperty("httpOptions") public abstract Optional httpOptions(); @@ -72,25 +72,22 @@ public abstract class RetrieveAgentEngineMemoriesConfig extends JsonSerializable @JsonProperty("memoryTypes") public abstract Optional> memoryTypes(); - /** Instantiates a builder for RetrieveAgentEngineMemoriesConfig. */ + /** Instantiates a builder for RetrieveMemoriesConfig. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_RetrieveAgentEngineMemoriesConfig.Builder(); + return new AutoValue_RetrieveMemoriesConfig.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for RetrieveAgentEngineMemoriesConfig. */ + /** Builder for RetrieveMemoriesConfig. */ @AutoValue.Builder public abstract static class Builder { - /** - * For internal usage. Please use `RetrieveAgentEngineMemoriesConfig.builder()` for - * instantiation. - */ + /** For internal usage. Please use `RetrieveMemoriesConfig.builder()` for instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_RetrieveAgentEngineMemoriesConfig.Builder(); + return new AutoValue_RetrieveMemoriesConfig.Builder(); } /** @@ -287,12 +284,12 @@ public Builder memoryTypesFromString(List memoryTypes) { return memoryTypes(listItems); } - public abstract RetrieveAgentEngineMemoriesConfig build(); + public abstract RetrieveMemoriesConfig build(); } - /** Deserializes a JSON string to a RetrieveAgentEngineMemoriesConfig object. */ + /** Deserializes a JSON string to a RetrieveMemoriesConfig object. */ @ExcludeFromGeneratedCoverageReport - public static RetrieveAgentEngineMemoriesConfig fromJson(String jsonString) { - return JsonSerializable.fromJsonString(jsonString, RetrieveAgentEngineMemoriesConfig.class); + public static RetrieveMemoriesConfig fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, RetrieveMemoriesConfig.class); } } diff --git a/src/main/java/com/google/cloud/agentplatform/types/RetrieveAgentEngineMemoriesRequestParameters.java b/src/main/java/com/google/cloud/agentplatform/types/RetrieveMemoriesRequestParameters.java similarity index 80% rename from src/main/java/com/google/cloud/agentplatform/types/RetrieveAgentEngineMemoriesRequestParameters.java rename to src/main/java/com/google/cloud/agentplatform/types/RetrieveMemoriesRequestParameters.java index abb7b54..b94432b 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/RetrieveAgentEngineMemoriesRequestParameters.java +++ b/src/main/java/com/google/cloud/agentplatform/types/RetrieveMemoriesRequestParameters.java @@ -28,12 +28,12 @@ import java.util.Map; import java.util.Optional; -/** Parameters for retrieving agent engine memories. */ +/** Parameters for retrieving memories. */ @AutoValue @InternalApi -@JsonDeserialize(builder = RetrieveAgentEngineMemoriesRequestParameters.Builder.class) -public abstract class RetrieveAgentEngineMemoriesRequestParameters extends JsonSerializable { - /** Name of the agent engine to retrieve memories from. */ +@JsonDeserialize(builder = RetrieveMemoriesRequestParameters.Builder.class) +public abstract class RetrieveMemoriesRequestParameters extends JsonSerializable { + /** Name of the Memory Bank to retrieve memories from. */ @JsonProperty("name") public abstract Optional name(); @@ -56,33 +56,33 @@ public abstract class RetrieveAgentEngineMemoriesRequestParameters extends JsonS /** */ @JsonProperty("config") - public abstract Optional config(); + public abstract Optional config(); - /** Instantiates a builder for RetrieveAgentEngineMemoriesRequestParameters. */ + /** Instantiates a builder for RetrieveMemoriesRequestParameters. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_RetrieveAgentEngineMemoriesRequestParameters.Builder(); + return new AutoValue_RetrieveMemoriesRequestParameters.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for RetrieveAgentEngineMemoriesRequestParameters. */ + /** Builder for RetrieveMemoriesRequestParameters. */ @AutoValue.Builder public abstract static class Builder { /** - * For internal usage. Please use `RetrieveAgentEngineMemoriesRequestParameters.builder()` for + * For internal usage. Please use `RetrieveMemoriesRequestParameters.builder()` for * instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_RetrieveAgentEngineMemoriesRequestParameters.Builder(); + return new AutoValue_RetrieveMemoriesRequestParameters.Builder(); } /** * Setter for name. * - *

name: Name of the agent engine to retrieve memories from. + *

name: Name of the Memory Bank to retrieve memories from. */ @JsonProperty("name") public abstract Builder name(String name); @@ -186,7 +186,7 @@ public Builder clearSimpleRetrievalParams() { *

config: */ @JsonProperty("config") - public abstract Builder config(RetrieveAgentEngineMemoriesConfig config); + public abstract Builder config(RetrieveMemoriesConfig config); /** * Setter for config builder. @@ -194,12 +194,12 @@ public Builder clearSimpleRetrievalParams() { *

config: */ @CanIgnoreReturnValue - public Builder config(RetrieveAgentEngineMemoriesConfig.Builder configBuilder) { + public Builder config(RetrieveMemoriesConfig.Builder configBuilder) { return config(configBuilder.build()); } @ExcludeFromGeneratedCoverageReport - abstract Builder config(Optional config); + abstract Builder config(Optional config); /** Clears the value of config field. */ @ExcludeFromGeneratedCoverageReport @@ -208,13 +208,12 @@ public Builder clearConfig() { return config(Optional.empty()); } - public abstract RetrieveAgentEngineMemoriesRequestParameters build(); + public abstract RetrieveMemoriesRequestParameters build(); } - /** Deserializes a JSON string to a RetrieveAgentEngineMemoriesRequestParameters object. */ + /** Deserializes a JSON string to a RetrieveMemoriesRequestParameters object. */ @ExcludeFromGeneratedCoverageReport - public static RetrieveAgentEngineMemoriesRequestParameters fromJson(String jsonString) { - return JsonSerializable.fromJsonString( - jsonString, RetrieveAgentEngineMemoriesRequestParameters.class); + public static RetrieveMemoriesRequestParameters fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, RetrieveMemoriesRequestParameters.class); } } diff --git a/src/main/java/com/google/cloud/agentplatform/types/RetrieveMemoryProfilesRequestParameters.java b/src/main/java/com/google/cloud/agentplatform/types/RetrieveMemoryProfilesRequestParameters.java index 94e72b9..7d66d85 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/RetrieveMemoryProfilesRequestParameters.java +++ b/src/main/java/com/google/cloud/agentplatform/types/RetrieveMemoryProfilesRequestParameters.java @@ -28,12 +28,12 @@ import java.util.Map; import java.util.Optional; -/** Parameters for retrieving agent engine memory profiles. */ +/** Parameters for retrieving memory profiles. */ @AutoValue @InternalApi @JsonDeserialize(builder = RetrieveMemoryProfilesRequestParameters.Builder.class) public abstract class RetrieveMemoryProfilesRequestParameters extends JsonSerializable { - /** Name of the agent engine to retrieve memory profiles from. */ + /** Name of the Memory Bank to retrieve memory profiles from. */ @JsonProperty("name") public abstract Optional name(); @@ -74,7 +74,7 @@ private static Builder create() { /** * Setter for name. * - *

name: Name of the agent engine to retrieve memory profiles from. + *

name: Name of the Memory Bank to retrieve memory profiles from. */ @JsonProperty("name") public abstract Builder name(String name); diff --git a/src/main/java/com/google/cloud/agentplatform/types/RollbackAgentEngineMemoryConfig.java b/src/main/java/com/google/cloud/agentplatform/types/RollbackMemoryConfig.java similarity index 77% rename from src/main/java/com/google/cloud/agentplatform/types/RollbackAgentEngineMemoryConfig.java rename to src/main/java/com/google/cloud/agentplatform/types/RollbackMemoryConfig.java index d8156fe..4b3c5ac 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/RollbackAgentEngineMemoryConfig.java +++ b/src/main/java/com/google/cloud/agentplatform/types/RollbackMemoryConfig.java @@ -29,8 +29,8 @@ /** Config for rolling back a memory. */ @AutoValue -@JsonDeserialize(builder = RollbackAgentEngineMemoryConfig.Builder.class) -public abstract class RollbackAgentEngineMemoryConfig extends JsonSerializable { +@JsonDeserialize(builder = RollbackMemoryConfig.Builder.class) +public abstract class RollbackMemoryConfig extends JsonSerializable { /** Used to override HTTP request options. */ @JsonProperty("httpOptions") public abstract Optional httpOptions(); @@ -39,24 +39,22 @@ public abstract class RollbackAgentEngineMemoryConfig extends JsonSerializable { @JsonProperty("waitForCompletion") public abstract Optional waitForCompletion(); - /** Instantiates a builder for RollbackAgentEngineMemoryConfig. */ + /** Instantiates a builder for RollbackMemoryConfig. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_RollbackAgentEngineMemoryConfig.Builder(); + return new AutoValue_RollbackMemoryConfig.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for RollbackAgentEngineMemoryConfig. */ + /** Builder for RollbackMemoryConfig. */ @AutoValue.Builder public abstract static class Builder { - /** - * For internal usage. Please use `RollbackAgentEngineMemoryConfig.builder()` for instantiation. - */ + /** For internal usage. Please use `RollbackMemoryConfig.builder()` for instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_RollbackAgentEngineMemoryConfig.Builder(); + return new AutoValue_RollbackMemoryConfig.Builder(); } /** @@ -95,12 +93,12 @@ public Builder clearWaitForCompletion() { return waitForCompletion(Optional.empty()); } - public abstract RollbackAgentEngineMemoryConfig build(); + public abstract RollbackMemoryConfig build(); } - /** Deserializes a JSON string to a RollbackAgentEngineMemoryConfig object. */ + /** Deserializes a JSON string to a RollbackMemoryConfig object. */ @ExcludeFromGeneratedCoverageReport - public static RollbackAgentEngineMemoryConfig fromJson(String jsonString) { - return JsonSerializable.fromJsonString(jsonString, RollbackAgentEngineMemoryConfig.class); + public static RollbackMemoryConfig fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, RollbackMemoryConfig.class); } } diff --git a/src/main/java/com/google/cloud/agentplatform/types/AgentEngineRollbackMemoryOperation.java b/src/main/java/com/google/cloud/agentplatform/types/RollbackMemoryOperation.java similarity index 85% rename from src/main/java/com/google/cloud/agentplatform/types/AgentEngineRollbackMemoryOperation.java rename to src/main/java/com/google/cloud/agentplatform/types/RollbackMemoryOperation.java index d776e16..bc41990 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/AgentEngineRollbackMemoryOperation.java +++ b/src/main/java/com/google/cloud/agentplatform/types/RollbackMemoryOperation.java @@ -29,8 +29,8 @@ /** Operation that rolls back a memory. */ @AutoValue -@JsonDeserialize(builder = AgentEngineRollbackMemoryOperation.Builder.class) -public abstract class AgentEngineRollbackMemoryOperation extends JsonSerializable { +@JsonDeserialize(builder = RollbackMemoryOperation.Builder.class) +public abstract class RollbackMemoryOperation extends JsonSerializable { /** * The server-assigned name, which is only unique within the same service that originally returns * it. If you use the default HTTP mapping, the `name` should be a resource name ending with @@ -59,25 +59,22 @@ public abstract class AgentEngineRollbackMemoryOperation extends JsonSerializabl @JsonProperty("error") public abstract Optional> error(); - /** Instantiates a builder for AgentEngineRollbackMemoryOperation. */ + /** Instantiates a builder for RollbackMemoryOperation. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_AgentEngineRollbackMemoryOperation.Builder(); + return new AutoValue_RollbackMemoryOperation.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for AgentEngineRollbackMemoryOperation. */ + /** Builder for RollbackMemoryOperation. */ @AutoValue.Builder public abstract static class Builder { - /** - * For internal usage. Please use `AgentEngineRollbackMemoryOperation.builder()` for - * instantiation. - */ + /** For internal usage. Please use `RollbackMemoryOperation.builder()` for instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_AgentEngineRollbackMemoryOperation.Builder(); + return new AutoValue_RollbackMemoryOperation.Builder(); } /** @@ -158,12 +155,12 @@ public Builder clearError() { return error(Optional.empty()); } - public abstract AgentEngineRollbackMemoryOperation build(); + public abstract RollbackMemoryOperation build(); } - /** Deserializes a JSON string to a AgentEngineRollbackMemoryOperation object. */ + /** Deserializes a JSON string to a RollbackMemoryOperation object. */ @ExcludeFromGeneratedCoverageReport - public static AgentEngineRollbackMemoryOperation fromJson(String jsonString) { - return JsonSerializable.fromJsonString(jsonString, AgentEngineRollbackMemoryOperation.class); + public static RollbackMemoryOperation fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, RollbackMemoryOperation.class); } } diff --git a/src/main/java/com/google/cloud/agentplatform/types/RollbackAgentEngineMemoryRequestParameters.java b/src/main/java/com/google/cloud/agentplatform/types/RollbackMemoryRequestParameters.java similarity index 68% rename from src/main/java/com/google/cloud/agentplatform/types/RollbackAgentEngineMemoryRequestParameters.java rename to src/main/java/com/google/cloud/agentplatform/types/RollbackMemoryRequestParameters.java index a0d973e..0c8ca27 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/RollbackAgentEngineMemoryRequestParameters.java +++ b/src/main/java/com/google/cloud/agentplatform/types/RollbackMemoryRequestParameters.java @@ -27,12 +27,12 @@ import com.google.genai.JsonSerializable; import java.util.Optional; -/** Parameters for generating agent engine memories. */ +/** Parameters for generating memories. */ @AutoValue @InternalApi -@JsonDeserialize(builder = RollbackAgentEngineMemoryRequestParameters.Builder.class) -public abstract class RollbackAgentEngineMemoryRequestParameters extends JsonSerializable { - /** Name of the agent engine memory to rollback. */ +@JsonDeserialize(builder = RollbackMemoryRequestParameters.Builder.class) +public abstract class RollbackMemoryRequestParameters extends JsonSerializable { + /** Name of the memory to rollback. */ @JsonProperty("name") public abstract Optional name(); @@ -42,33 +42,32 @@ public abstract class RollbackAgentEngineMemoryRequestParameters extends JsonSer /** */ @JsonProperty("config") - public abstract Optional config(); + public abstract Optional config(); - /** Instantiates a builder for RollbackAgentEngineMemoryRequestParameters. */ + /** Instantiates a builder for RollbackMemoryRequestParameters. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_RollbackAgentEngineMemoryRequestParameters.Builder(); + return new AutoValue_RollbackMemoryRequestParameters.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for RollbackAgentEngineMemoryRequestParameters. */ + /** Builder for RollbackMemoryRequestParameters. */ @AutoValue.Builder public abstract static class Builder { /** - * For internal usage. Please use `RollbackAgentEngineMemoryRequestParameters.builder()` for - * instantiation. + * For internal usage. Please use `RollbackMemoryRequestParameters.builder()` for instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_RollbackAgentEngineMemoryRequestParameters.Builder(); + return new AutoValue_RollbackMemoryRequestParameters.Builder(); } /** * Setter for name. * - *

name: Name of the agent engine memory to rollback. + *

name: Name of the memory to rollback. */ @JsonProperty("name") public abstract Builder name(String name); @@ -107,7 +106,7 @@ public Builder clearTargetRevisionId() { *

config: */ @JsonProperty("config") - public abstract Builder config(RollbackAgentEngineMemoryConfig config); + public abstract Builder config(RollbackMemoryConfig config); /** * Setter for config builder. @@ -115,12 +114,12 @@ public Builder clearTargetRevisionId() { *

config: */ @CanIgnoreReturnValue - public Builder config(RollbackAgentEngineMemoryConfig.Builder configBuilder) { + public Builder config(RollbackMemoryConfig.Builder configBuilder) { return config(configBuilder.build()); } @ExcludeFromGeneratedCoverageReport - abstract Builder config(Optional config); + abstract Builder config(Optional config); /** Clears the value of config field. */ @ExcludeFromGeneratedCoverageReport @@ -129,13 +128,12 @@ public Builder clearConfig() { return config(Optional.empty()); } - public abstract RollbackAgentEngineMemoryRequestParameters build(); + public abstract RollbackMemoryRequestParameters build(); } - /** Deserializes a JSON string to a RollbackAgentEngineMemoryRequestParameters object. */ + /** Deserializes a JSON string to a RollbackMemoryRequestParameters object. */ @ExcludeFromGeneratedCoverageReport - public static RollbackAgentEngineMemoryRequestParameters fromJson(String jsonString) { - return JsonSerializable.fromJsonString( - jsonString, RollbackAgentEngineMemoryRequestParameters.class); + public static RollbackMemoryRequestParameters fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, RollbackMemoryRequestParameters.class); } } diff --git a/src/main/java/com/google/cloud/agentplatform/types/UpdateAgentEngineMemoryConfig.java b/src/main/java/com/google/cloud/agentplatform/types/UpdateMemoryConfig.java similarity index 93% rename from src/main/java/com/google/cloud/agentplatform/types/UpdateAgentEngineMemoryConfig.java rename to src/main/java/com/google/cloud/agentplatform/types/UpdateMemoryConfig.java index 99c617d..22d150b 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/UpdateAgentEngineMemoryConfig.java +++ b/src/main/java/com/google/cloud/agentplatform/types/UpdateMemoryConfig.java @@ -34,10 +34,10 @@ import java.util.Map; import java.util.Optional; -/** Config for updating agent engine memory. */ +/** Config for updating a memory. */ @AutoValue -@JsonDeserialize(builder = UpdateAgentEngineMemoryConfig.Builder.class) -public abstract class UpdateAgentEngineMemoryConfig extends JsonSerializable { +@JsonDeserialize(builder = UpdateMemoryConfig.Builder.class) +public abstract class UpdateMemoryConfig extends JsonSerializable { /** Used to override HTTP request options. */ @JsonProperty("httpOptions") public abstract Optional httpOptions(); @@ -111,24 +111,22 @@ public abstract class UpdateAgentEngineMemoryConfig extends JsonSerializable { @JsonProperty("updateMask") public abstract Optional updateMask(); - /** Instantiates a builder for UpdateAgentEngineMemoryConfig. */ + /** Instantiates a builder for UpdateMemoryConfig. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_UpdateAgentEngineMemoryConfig.Builder(); + return new AutoValue_UpdateMemoryConfig.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for UpdateAgentEngineMemoryConfig. */ + /** Builder for UpdateMemoryConfig. */ @AutoValue.Builder public abstract static class Builder { - /** - * For internal usage. Please use `UpdateAgentEngineMemoryConfig.builder()` for instantiation. - */ + /** For internal usage. Please use `UpdateMemoryConfig.builder()` for instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_UpdateAgentEngineMemoryConfig.Builder(); + return new AutoValue_UpdateMemoryConfig.Builder(); } /** @@ -399,12 +397,12 @@ public Builder clearUpdateMask() { return updateMask(Optional.empty()); } - public abstract UpdateAgentEngineMemoryConfig build(); + public abstract UpdateMemoryConfig build(); } - /** Deserializes a JSON string to a UpdateAgentEngineMemoryConfig object. */ + /** Deserializes a JSON string to a UpdateMemoryConfig object. */ @ExcludeFromGeneratedCoverageReport - public static UpdateAgentEngineMemoryConfig fromJson(String jsonString) { - return JsonSerializable.fromJsonString(jsonString, UpdateAgentEngineMemoryConfig.class); + public static UpdateMemoryConfig fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, UpdateMemoryConfig.class); } } diff --git a/src/main/java/com/google/cloud/agentplatform/types/UpdateAgentEngineMemoryRequestParameters.java b/src/main/java/com/google/cloud/agentplatform/types/UpdateMemoryRequestParameters.java similarity index 75% rename from src/main/java/com/google/cloud/agentplatform/types/UpdateAgentEngineMemoryRequestParameters.java rename to src/main/java/com/google/cloud/agentplatform/types/UpdateMemoryRequestParameters.java index 6eac5df..806e1b6 100644 --- a/src/main/java/com/google/cloud/agentplatform/types/UpdateAgentEngineMemoryRequestParameters.java +++ b/src/main/java/com/google/cloud/agentplatform/types/UpdateMemoryRequestParameters.java @@ -28,12 +28,12 @@ import java.util.Map; import java.util.Optional; -/** Parameters for updating agent engine memories. */ +/** Parameters for updating memories. */ @AutoValue @InternalApi -@JsonDeserialize(builder = UpdateAgentEngineMemoryRequestParameters.Builder.class) -public abstract class UpdateAgentEngineMemoryRequestParameters extends JsonSerializable { - /** Name of the agent engine memory to update. */ +@JsonDeserialize(builder = UpdateMemoryRequestParameters.Builder.class) +public abstract class UpdateMemoryRequestParameters extends JsonSerializable { + /** Name of the memory to update. */ @JsonProperty("name") public abstract Optional name(); @@ -57,33 +57,32 @@ public abstract class UpdateAgentEngineMemoryRequestParameters extends JsonSeria /** */ @JsonProperty("config") - public abstract Optional config(); + public abstract Optional config(); - /** Instantiates a builder for UpdateAgentEngineMemoryRequestParameters. */ + /** Instantiates a builder for UpdateMemoryRequestParameters. */ @ExcludeFromGeneratedCoverageReport public static Builder builder() { - return new AutoValue_UpdateAgentEngineMemoryRequestParameters.Builder(); + return new AutoValue_UpdateMemoryRequestParameters.Builder(); } /** Creates a builder with the same values as this instance. */ public abstract Builder toBuilder(); - /** Builder for UpdateAgentEngineMemoryRequestParameters. */ + /** Builder for UpdateMemoryRequestParameters. */ @AutoValue.Builder public abstract static class Builder { /** - * For internal usage. Please use `UpdateAgentEngineMemoryRequestParameters.builder()` for - * instantiation. + * For internal usage. Please use `UpdateMemoryRequestParameters.builder()` for instantiation. */ @JsonCreator private static Builder create() { - return new AutoValue_UpdateAgentEngineMemoryRequestParameters.Builder(); + return new AutoValue_UpdateMemoryRequestParameters.Builder(); } /** * Setter for name. * - *

name: Name of the agent engine memory to update. + *

name: Name of the memory to update. */ @JsonProperty("name") public abstract Builder name(String name); @@ -146,7 +145,7 @@ public Builder clearScope() { *

config: */ @JsonProperty("config") - public abstract Builder config(UpdateAgentEngineMemoryConfig config); + public abstract Builder config(UpdateMemoryConfig config); /** * Setter for config builder. @@ -154,12 +153,12 @@ public Builder clearScope() { *

config: */ @CanIgnoreReturnValue - public Builder config(UpdateAgentEngineMemoryConfig.Builder configBuilder) { + public Builder config(UpdateMemoryConfig.Builder configBuilder) { return config(configBuilder.build()); } @ExcludeFromGeneratedCoverageReport - abstract Builder config(Optional config); + abstract Builder config(Optional config); /** Clears the value of config field. */ @ExcludeFromGeneratedCoverageReport @@ -168,13 +167,12 @@ public Builder clearConfig() { return config(Optional.empty()); } - public abstract UpdateAgentEngineMemoryRequestParameters build(); + public abstract UpdateMemoryRequestParameters build(); } - /** Deserializes a JSON string to a UpdateAgentEngineMemoryRequestParameters object. */ + /** Deserializes a JSON string to a UpdateMemoryRequestParameters object. */ @ExcludeFromGeneratedCoverageReport - public static UpdateAgentEngineMemoryRequestParameters fromJson(String jsonString) { - return JsonSerializable.fromJsonString( - jsonString, UpdateAgentEngineMemoryRequestParameters.class); + public static UpdateMemoryRequestParameters fromJson(String jsonString) { + return JsonSerializable.fromJsonString(jsonString, UpdateMemoryRequestParameters.class); } } diff --git a/src/test/java/com/google/cloud/agentplatform/AgentEnginesMemoriesRevisionsTest.java b/src/test/java/com/google/cloud/agentplatform/AgentEnginesMemoriesRevisionsTest.java deleted file mode 100644 index dcb2bfa..0000000 --- a/src/test/java/com/google/cloud/agentplatform/AgentEnginesMemoriesRevisionsTest.java +++ /dev/null @@ -1,64 +0,0 @@ -/* - * Copyright 2026 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. - */ - -package com.google.cloud.agentplatform; - -import static org.junit.jupiter.api.Assertions.assertEquals; - -import com.google.cloud.agentplatform.types.ListAgentEngineMemoryRevisionsResponse; -import com.google.cloud.agentplatform.types.MemoryRevision; -import org.junit.jupiter.api.Test; -import org.junit.jupiter.api.condition.EnabledIfEnvironmentVariable; -import org.junit.jupiter.api.extension.ExtendWith; - -@EnabledIfEnvironmentVariable( - named = "GOOGLE_GENAI_REPLAYS_DIRECTORY", - matches = ".*genai/replays.*") -@ExtendWith(EnvironmentVariablesMockingExtension.class) -public class AgentEnginesMemoriesRevisionsTest { - - @Test - public void testPrivateGetAgentEngineMemoryRevision() throws Exception { - Client client = - TestUtils.createClient( - true, "ae_memory_revisions_get/test_get_memory_revisions.vertex.json"); - - String name = - "projects/964831358985/locations/us-central1/reasoningEngines/2886612747586371584/memories/3858070028511346688/revisions/516064922187071488"; - - MemoryRevision revision = client.agentEngines.memories.revisions.get(name, null); - - assertEquals(revision.name().get(), name); - } - - @Test - public void testPrivateListAgentEngineMemoryRevisions() throws Exception { - Client client = - TestUtils.createClient( - true, - "ae_memory_revisions_private_list/test_private_list_memory_revisions.vertex.json"); - - String name = - "projects/964831358985/locations/us-central1/reasoningEngines/2886612747586371584/memories/3858070028511346688"; - - ListAgentEngineMemoryRevisionsResponse response = - client.agentEngines.memories.revisions.privateList(name, null); - - assertEquals( - response.memoryRevisions().get().get(0).name().get(), - name + "/revisions/516064922187071488"); - } -} diff --git a/src/test/java/com/google/cloud/agentplatform/AgentEnginesMemoriesTest.java b/src/test/java/com/google/cloud/agentplatform/AgentEnginesMemoriesTest.java deleted file mode 100644 index c0ebae7..0000000 --- a/src/test/java/com/google/cloud/agentplatform/AgentEnginesMemoriesTest.java +++ /dev/null @@ -1,213 +0,0 @@ -/* - * Copyright 2026 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. - */ - -package com.google.cloud.agentplatform; - -import static org.junit.jupiter.api.Assertions.assertNotNull; - -import com.google.cloud.agentplatform.types.AgentEngineGenerateMemoriesOperation; -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.DeleteAgentEngineMemoryOperation; -import com.google.cloud.agentplatform.types.GenerateMemoriesRequestVertexSessionSource; -import com.google.cloud.agentplatform.types.ListReasoningEnginesMemoriesResponse; -import com.google.cloud.agentplatform.types.Memory; -import com.google.cloud.agentplatform.types.RetrieveMemoriesResponse; -import com.google.common.collect.ImmutableMap; -import org.junit.jupiter.api.Test; -import org.junit.jupiter.api.condition.EnabledIfEnvironmentVariable; -import org.junit.jupiter.api.extension.ExtendWith; - -@EnabledIfEnvironmentVariable( - named = "GOOGLE_GENAI_REPLAYS_DIRECTORY", - matches = ".*genai/replays.*") -@ExtendWith(EnvironmentVariablesMockingExtension.class) -public class AgentEnginesMemoriesTest { - - @Test - public void testPrivateCreateAgentEngineMemory() throws Exception { - Client client = - TestUtils.createClient( - true, "ae_memories_private_create/test_private_create_memory.vertex.json"); - - AgentEngineMemoryOperation operation = - client.agentEngines.memories.privateCreate( - "projects/964831358985/locations/us-central1/reasoningEngines/2886612747586371584", - "memory_fact", - ImmutableMap.of("user_id", "123"), - null); - - assertNotNull(operation.name()); - } - - @Test - public void testDeleteAgentEngineMemory() throws Exception { - Client client = - TestUtils.createClient(true, "ae_memories_delete/test_delete_memory.vertex.json"); - - DeleteAgentEngineMemoryOperation operation = - client.agentEngines.memories.delete( - "projects/964831358985/locations/us-central1/reasoningEngines/2886612747586371584/memories/5605466683931099136", - null); - - assertNotNull(operation.name()); - } - - @Test - public void testPrivateGenerateAgentEngineMemory() throws Exception { - Client client = - TestUtils.createClient( - true, "ae_memories_private_generate/test_private_generate_memory.vertex.json"); - - String aeName = - "projects/964831358985/locations/us-central1/reasoningEngines/2886612747586371584"; - - AgentEngineGenerateMemoriesOperation response = - client.agentEngines.memories.privateGenerate( - aeName, - GenerateMemoriesRequestVertexSessionSource.builder() - .session( - "{PROJECT_AND_LOCATION_PATH}/reasoningEngines/2886612747586371584/sessions/6922431337672474624") - .build(), - null, - null, - null, - null); - - assertNotNull(response.name()); - } - - @Test - public void testPrivateGetAgentEngineMemory() throws Exception { - Client client = TestUtils.createClient(true, "ae_memories_get/test_get_memory.vertex.json"); - - Memory response = - client.agentEngines.memories.get( - "projects/964831358985/locations/us-central1/reasoningEngines/2886612747586371584/memories/3858070028511346688", - null); - - assertNotNull(response.name().get()); - } - - @Test - public void testPrivateGetAgentEngineMemoryOperation() throws Exception { - Client client = - TestUtils.createClient( - true, - "ae_memories_private_get_memory_operation/test_private_get_memory_operation.vertex.json"); - - AgentEngineMemoryOperation operation = - client.agentEngines.memories.privateGetMemoryOperation( - "projects/964831358985/locations/us-central1/reasoningEngines/2886612747586371584/memories/3858070028511346688/operations/1044963283964002304", - null); - - assertNotNull(operation.name()); - } - - @Test - public void testPrivateGetAgentEngineGenerateMemoryOperation() throws Exception { - Client client = - TestUtils.createClient( - true, - "ae_memories_private_get_generate_memories_operation/test_private_get_generate_memories_operation.vertex.json"); - - AgentEngineGenerateMemoriesOperation operation = - client.agentEngines.memories.privateGetGenerateMemoriesOperation( - "projects/964831358985/locations/us-central1/reasoningEngines/2886612747586371584/operations/5669315676343369728", - null); - - assertNotNull(operation.name()); - } - - @Test - public void testPrivateListAgentEngineMemory() throws Exception { - Client client = - TestUtils.createClient( - true, "ae_memories_private_list/test_private_list_memory.vertex.json"); - - ListReasoningEnginesMemoriesResponse response = - client.agentEngines.memories.privateList( - "projects/964831358985/locations/us-central1/reasoningEngines/2886612747586371584", - null); - - assertNotNull(response.memories().get().get(0).name().get()); - } - - @Test - public void testPrivateRetrieveAgentEngineMemory() throws Exception { - Client client = - TestUtils.createClient( - true, "ae_memories_private_retrieve/test_private_retrieve.vertex.json"); - - RetrieveMemoriesResponse response = - client.agentEngines.memories.privateRetrieve( - "projects/964831358985/locations/us-central1/reasoningEngines/2886612747586371584", - ImmutableMap.of("user_id", "123"), - null, - null, - null); - - assertNotNull(response.retrievedMemories().get().get(0).memory().get()); - } - - @Test - public void testPrivateRollbackAgentEngineMemory() throws Exception { - Client client = - TestUtils.createClient( - true, "ae_memories_private_rollback/test_private_rollback.vertex.json"); - - AgentEngineRollbackMemoryOperation operation = - client.agentEngines.memories.privateRollback( - "projects/964831358985/locations/us-central1/reasoningEngines/2886612747586371584/memories/3858070028511346688", - "3001207491565453312", - null); - - assertNotNull(operation.name()); - } - - @Test - public void testPrivateUpdateAgentEngineMemory() throws Exception { - Client client = - TestUtils.createClient( - true, "ae_memories_private_update/test_private_update_memory.vertex.json"); - - AgentEngineMemoryOperation operation = - client.agentEngines.memories.privateUpdate( - "projects/964831358985/locations/us-central1/reasoningEngines/2886612747586371584/memories/3858070028511346688", - "memory_fact_updated", - ImmutableMap.of("user_id", "123"), - null); - - assertNotNull(operation.name()); - } - - @Test - public void testPrivatePurgeAgentEngineMemory() throws Exception { - Client client = - TestUtils.createClient(true, "ae_memories_private_purge/test_private_purge.vertex.json"); - - AgentEnginePurgeMemoriesOperation operation = - client.agentEngines.memories.privatePurge( - "projects/964831358985/locations/us-central1/reasoningEngines/6086402690647064576", - "scope.user_id=123", - null, - false, - null); - - assertNotNull(operation.name()); - } -}