Skip to content

Commit 0ff4dba

Browse files
Poggeccicopybara-github
authored andcommitted
fix: function calls in Gemini runLive not working
PiperOrigin-RevId: 792789389
1 parent 4fc83f0 commit 0ff4dba

File tree

2 files changed

+14
-1
lines changed

2 files changed

+14
-1
lines changed

core/src/main/java/com/google/adk/models/GeminiLlmConnection.java

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -206,7 +206,10 @@ public Completable sendContent(Content content) {
206206
.build());
207207
} else {
208208
return sendClientContentInternal(
209-
LiveSendClientContentParameters.builder().turns(ImmutableList.of(content)).build());
209+
LiveSendClientContentParameters.builder()
210+
.turns(ImmutableList.of(content))
211+
.turnComplete(true)
212+
.build());
210213
}
211214
}
212215

core/src/main/java/com/google/adk/tools/BaseTool.java

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323
import com.google.errorprone.annotations.CanIgnoreReturnValue;
2424
import com.google.genai.types.FunctionDeclaration;
2525
import com.google.genai.types.GenerateContentConfig;
26+
import com.google.genai.types.LiveConnectConfig;
2627
import com.google.genai.types.Tool;
2728
import io.reactivex.rxjava3.core.Completable;
2829
import io.reactivex.rxjava3.core.Single;
@@ -122,7 +123,16 @@ public Completable processLlmRequest(
122123
.add(toolWithFunctionDeclarations)
123124
.build())
124125
.build();
126+
LiveConnectConfig liveConnectConfig =
127+
llmRequest.liveConnectConfig().toBuilder()
128+
.tools(
129+
new ImmutableList.Builder<Tool>()
130+
.addAll(toolsWithoutFunctionDeclarations)
131+
.add(toolWithFunctionDeclarations)
132+
.build())
133+
.build();
125134
llmRequestBuilder.config(generateContentConfig);
135+
llmRequestBuilder.liveConnectConfig(liveConnectConfig);
126136
return Completable.complete();
127137
}
128138

0 commit comments

Comments
 (0)