Skip to content

Commit cb95b56

Browse files
cornellgitcopybara-github
authored andcommitted
fix: remove debug logs from base llm flow to prevent accidentally logging user data
PiperOrigin-RevId: 786994128
1 parent f1c0602 commit cb95b56

File tree

1 file changed

+0
-11
lines changed

1 file changed

+0
-11
lines changed

core/src/main/java/com/google/adk/flows/llmflows/BaseLlmFlow.java

Lines changed: 0 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -164,11 +164,9 @@ protected Single<ResponseProcessingResult> postprocess(
164164
updatedResponse, Iterables.concat(eventIterables), Optional.empty()));
165165
}
166166

167-
logger.debug("Response after processors: {}", updatedResponse);
168167
Event modelResponseEvent =
169168
buildModelResponseEvent(baseEventForLlmResponse, llmRequest, updatedResponse);
170169
eventIterables.add(Collections.singleton(modelResponseEvent));
171-
logger.debug("Model response event: {}", modelResponseEvent.toJson());
172170

173171
Maybe<Event> maybeFunctionCallEvent =
174172
modelResponseEvent.functionCalls().isEmpty()
@@ -183,7 +181,6 @@ protected Single<ResponseProcessingResult> postprocess(
183181
Optional<String> transferToAgent = Optional.empty();
184182
if (functionCallEventOpt.isPresent()) {
185183
Event functionCallEvent = functionCallEventOpt.get();
186-
logger.debug("Function call event generated: {}", functionCallEvent);
187184
eventIterables.add(Collections.singleton(functionCallEvent));
188185
transferToAgent = functionCallEvent.actions().transferToAgent();
189186
}
@@ -332,7 +329,6 @@ private Flowable<Event> runOneStep(InvocationContext context) {
332329
LlmRequest llmRequestAfterPreprocess = preResult.updatedRequest();
333330
Iterable<Event> preEvents = preResult.events();
334331

335-
logger.debug("Pre-processing result: {}", preResult);
336332
if (context.endInvocation()) {
337333
logger.debug("End invocation requested during preprocessing.");
338334
return Flowable.fromIterable(preEvents);
@@ -353,16 +349,10 @@ private Flowable<Event> runOneStep(InvocationContext context) {
353349
.branch(context.branch())
354350
.build();
355351

356-
logger.debug("Starting LLM call with request: {}", llmRequestAfterPreprocess);
357352
Flowable<Event> restOfFlow =
358353
callLlm(context, llmRequestAfterPreprocess, mutableEventTemplate)
359354
.concatMap(
360355
llmResponse -> {
361-
logger.debug(
362-
"Processing LlmResponse with Event ID: {}",
363-
mutableEventTemplate.id());
364-
logger.debug("LLM response for current step: {}", llmResponse);
365-
366356
Single<ResponseProcessingResult> postResultSingle =
367357
postprocess(
368358
context,
@@ -385,7 +375,6 @@ private Flowable<Event> runOneStep(InvocationContext context) {
385375
})
386376
.concatMap(
387377
postResult -> {
388-
logger.debug("Post-processing result: {}", postResult);
389378
Flowable<Event> postProcessedEvents =
390379
Flowable.fromIterable(postResult.events());
391380
if (postResult.transferToAgent().isPresent()) {

0 commit comments

Comments
 (0)