Skip to content

Commit 1855584

Browse files
MiloszSobczykcopybara-github
authored andcommitted
refactor: use new setter methods in place of old ones
PiperOrigin-RevId: 892296560
1 parent d0e1085 commit 1855584

File tree

4 files changed

+18
-19
lines changed

4 files changed

+18
-19
lines changed

core/src/main/java/com/google/adk/memory/InMemoryMemoryService.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,7 @@ public Single<SearchMemoryResponse> searchMemory(String appName, String userId,
130130
}
131131

132132
return SearchMemoryResponse.builder()
133-
.setMemories(ImmutableList.copyOf(matchingMemories))
133+
.memories(ImmutableList.copyOf(matchingMemories))
134134
.build();
135135
});
136136
}

core/src/main/java/com/google/adk/runner/Runner.java

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -582,19 +582,18 @@ private InvocationContext newInvocationContextForLive(
582582
if (liveRequestQueue != null) {
583583
// Default to AUDIO modality if not specified.
584584
if (CollectionUtils.isNullOrEmpty(runConfig.responseModalities())) {
585-
runConfigBuilder.setResponseModalities(
586-
ImmutableList.of(new Modality(Modality.Known.AUDIO)));
585+
runConfigBuilder.responseModalities(ImmutableList.of(new Modality(Modality.Known.AUDIO)));
587586
if (runConfig.outputAudioTranscription() == null) {
588-
runConfigBuilder.setOutputAudioTranscription(AudioTranscriptionConfig.builder().build());
587+
runConfigBuilder.outputAudioTranscription(AudioTranscriptionConfig.builder().build());
589588
}
590589
} else if (!runConfig.responseModalities().contains(new Modality(Modality.Known.TEXT))) {
591590
if (runConfig.outputAudioTranscription() == null) {
592-
runConfigBuilder.setOutputAudioTranscription(AudioTranscriptionConfig.builder().build());
591+
runConfigBuilder.outputAudioTranscription(AudioTranscriptionConfig.builder().build());
593592
}
594593
}
595594
// Need input transcription for agent transferring in live mode.
596595
if (runConfig.inputAudioTranscription() == null) {
597-
runConfigBuilder.setInputAudioTranscription(AudioTranscriptionConfig.builder().build());
596+
runConfigBuilder.inputAudioTranscription(AudioTranscriptionConfig.builder().build());
598597
}
599598
}
600599
InvocationContext.Builder builder =

core/src/test/java/com/google/adk/plugins/agentanalytics/BigQueryAgentAnalyticsPluginTest.java

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -96,15 +96,15 @@ public void setUp() throws Exception {
9696
fakeAgent = new FakeAgent("agent_name");
9797
config =
9898
BigQueryLoggerConfig.builder()
99-
.setEnabled(true)
100-
.setProjectId("project")
101-
.setDatasetId("dataset")
102-
.setTableName("table")
103-
.setBatchSize(10)
104-
.setBatchFlushInterval(Duration.ofSeconds(10))
105-
.setAutoSchemaUpgrade(false)
106-
.setCredentials(mock(Credentials.class))
107-
.setCustomTags(ImmutableMap.of("global_tag", "global_value"))
99+
.enabled(true)
100+
.projectId("project")
101+
.datasetId("dataset")
102+
.tableName("table")
103+
.batchSize(10)
104+
.batchFlushInterval(Duration.ofSeconds(10))
105+
.autoSchemaUpgrade(false)
106+
.credentials(mock(Credentials.class))
107+
.customTags(ImmutableMap.of("global_tag", "global_value"))
108108
.build();
109109

110110
when(mockBigQuery.getOptions())
@@ -167,9 +167,9 @@ public void afterRunCallback_flushesAndAppends() throws Exception {
167167
public void getStreamName_returnsCorrectFormat() {
168168
BigQueryLoggerConfig config =
169169
BigQueryLoggerConfig.builder()
170-
.setProjectId("test-project")
171-
.setDatasetId("test-dataset")
172-
.setTableName("test-table")
170+
.projectId("test-project")
171+
.datasetId("test-dataset")
172+
.tableName("test-table")
173173
.build();
174174

175175
String streamName = plugin.getStreamName(config);

core/src/test/java/com/google/adk/tools/ExampleToolTest.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ public void processLlmRequest_withInlineExamples_appendsFewShot() {
7373

7474
@Test
7575
public void processLlmRequest_withProvider_appendsFewShot() {
76-
ExampleTool tool = ExampleTool.builder().setExampleProvider(ProviderHolder.EXAMPLES).build();
76+
ExampleTool tool = ExampleTool.builder().exampleProvider(ProviderHolder.EXAMPLES).build();
7777

7878
InvocationContext ctx = buildInvocationContext();
7979
LlmRequest.Builder builder = LlmRequest.builder().model("gemini-2.0-flash");

0 commit comments

Comments
 (0)