From 18555848cdc292ac1ebc97dfd12817c79a688f20 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mi=C5=82osz=20Sobczyk?= <miloszsobczyk@google.com>
Date: Tue, 31 Mar 2026 06:14:27 -0700
Subject: [PATCH] refactor: use new setter methods in place of old ones

PiperOrigin-RevId: 892296560
---
 .../adk/memory/InMemoryMemoryService.java     |  2 +-
 .../java/com/google/adk/runner/Runner.java    |  9 ++++---
 .../BigQueryAgentAnalyticsPluginTest.java     | 24 +++++++++----------
 .../com/google/adk/tools/ExampleToolTest.java |  2 +-
 4 files changed, 18 insertions(+), 19 deletions(-)

diff --git a/core/src/main/java/com/google/adk/memory/InMemoryMemoryService.java b/core/src/main/java/com/google/adk/memory/InMemoryMemoryService.java
index 200c6df55..ff2995c74 100644
--- a/core/src/main/java/com/google/adk/memory/InMemoryMemoryService.java
+++ b/core/src/main/java/com/google/adk/memory/InMemoryMemoryService.java
@@ -130,7 +130,7 @@ public Single<SearchMemoryResponse> searchMemory(String appName, String userId,
           }
 
           return SearchMemoryResponse.builder()
-              .setMemories(ImmutableList.copyOf(matchingMemories))
+              .memories(ImmutableList.copyOf(matchingMemories))
               .build();
         });
   }
diff --git a/core/src/main/java/com/google/adk/runner/Runner.java b/core/src/main/java/com/google/adk/runner/Runner.java
index 120465f31..d6ad38561 100644
--- a/core/src/main/java/com/google/adk/runner/Runner.java
+++ b/core/src/main/java/com/google/adk/runner/Runner.java
@@ -582,19 +582,18 @@ private InvocationContext newInvocationContextForLive(
     if (liveRequestQueue != null) {
       // Default to AUDIO modality if not specified.
       if (CollectionUtils.isNullOrEmpty(runConfig.responseModalities())) {
-        runConfigBuilder.setResponseModalities(
-            ImmutableList.of(new Modality(Modality.Known.AUDIO)));
+        runConfigBuilder.responseModalities(ImmutableList.of(new Modality(Modality.Known.AUDIO)));
         if (runConfig.outputAudioTranscription() == null) {
-          runConfigBuilder.setOutputAudioTranscription(AudioTranscriptionConfig.builder().build());
+          runConfigBuilder.outputAudioTranscription(AudioTranscriptionConfig.builder().build());
         }
       } else if (!runConfig.responseModalities().contains(new Modality(Modality.Known.TEXT))) {
         if (runConfig.outputAudioTranscription() == null) {
-          runConfigBuilder.setOutputAudioTranscription(AudioTranscriptionConfig.builder().build());
+          runConfigBuilder.outputAudioTranscription(AudioTranscriptionConfig.builder().build());
         }
       }
       // Need input transcription for agent transferring in live mode.
       if (runConfig.inputAudioTranscription() == null) {
-        runConfigBuilder.setInputAudioTranscription(AudioTranscriptionConfig.builder().build());
+        runConfigBuilder.inputAudioTranscription(AudioTranscriptionConfig.builder().build());
       }
     }
     InvocationContext.Builder builder =
diff --git a/core/src/test/java/com/google/adk/plugins/agentanalytics/BigQueryAgentAnalyticsPluginTest.java b/core/src/test/java/com/google/adk/plugins/agentanalytics/BigQueryAgentAnalyticsPluginTest.java
index 8147c5cc6..0822c2cae 100644
--- a/core/src/test/java/com/google/adk/plugins/agentanalytics/BigQueryAgentAnalyticsPluginTest.java
+++ b/core/src/test/java/com/google/adk/plugins/agentanalytics/BigQueryAgentAnalyticsPluginTest.java
@@ -96,15 +96,15 @@ public void setUp() throws Exception {
     fakeAgent = new FakeAgent("agent_name");
     config =
         BigQueryLoggerConfig.builder()
-            .setEnabled(true)
-            .setProjectId("project")
-            .setDatasetId("dataset")
-            .setTableName("table")
-            .setBatchSize(10)
-            .setBatchFlushInterval(Duration.ofSeconds(10))
-            .setAutoSchemaUpgrade(false)
-            .setCredentials(mock(Credentials.class))
-            .setCustomTags(ImmutableMap.of("global_tag", "global_value"))
+            .enabled(true)
+            .projectId("project")
+            .datasetId("dataset")
+            .tableName("table")
+            .batchSize(10)
+            .batchFlushInterval(Duration.ofSeconds(10))
+            .autoSchemaUpgrade(false)
+            .credentials(mock(Credentials.class))
+            .customTags(ImmutableMap.of("global_tag", "global_value"))
             .build();
 
     when(mockBigQuery.getOptions())
@@ -167,9 +167,9 @@ public void afterRunCallback_flushesAndAppends() throws Exception {
   public void getStreamName_returnsCorrectFormat() {
     BigQueryLoggerConfig config =
         BigQueryLoggerConfig.builder()
-            .setProjectId("test-project")
-            .setDatasetId("test-dataset")
-            .setTableName("test-table")
+            .projectId("test-project")
+            .datasetId("test-dataset")
+            .tableName("test-table")
             .build();
 
     String streamName = plugin.getStreamName(config);
diff --git a/core/src/test/java/com/google/adk/tools/ExampleToolTest.java b/core/src/test/java/com/google/adk/tools/ExampleToolTest.java
index 55e5d8f93..e56afe60b 100644
--- a/core/src/test/java/com/google/adk/tools/ExampleToolTest.java
+++ b/core/src/test/java/com/google/adk/tools/ExampleToolTest.java
@@ -73,7 +73,7 @@ public void processLlmRequest_withInlineExamples_appendsFewShot() {
 
   @Test
   public void processLlmRequest_withProvider_appendsFewShot() {
-    ExampleTool tool = ExampleTool.builder().setExampleProvider(ProviderHolder.EXAMPLES).build();
+    ExampleTool tool = ExampleTool.builder().exampleProvider(ProviderHolder.EXAMPLES).build();
 
     InvocationContext ctx = buildInvocationContext();
     LlmRequest.Builder builder = LlmRequest.builder().model("gemini-2.0-flash");