Skip to content

Commit b223fd7

Browse files
committed
pkg/aflow: expect CitationMetadata
CitationMetadata may be present in replies sometimes. CitationMetadata is a specific field in the Gemini API's response object that alerts you when the model has directly quoted or closely derived content from a specific source, such as a book, website, or open-source code repository. We've got the following error: syz-agent: unexpected reply fields ({Content:0xc0058eb4a0 CitationMetadata:0xc0094009a8 FinishMessage: TokenCount:0 FinishReason:STOP AvgLogprobs:0 GroundingMetadata:<nil> Index:0 LogprobsResult:<nil> SafetyRatings:[] URLContextMetadata:<nil>})
1 parent e30f059 commit b223fd7

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

pkg/aflow/llm_agent.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -191,9 +191,9 @@ func (a *LLMAgent) parseResponse(resp *genai.GenerateContentResponse) (
191191
err = fmt.Errorf("%v (%v)", candidate.FinishMessage, candidate.FinishReason)
192192
return
193193
}
194-
// We don't expect to receive these now.
195-
if candidate.GroundingMetadata != nil || candidate.CitationMetadata != nil ||
196-
candidate.LogprobsResult != nil {
194+
// We don't expect to receive these fields now.
195+
// Note: CitationMetadata may be present sometimes, but we don't have uses for it.
196+
if candidate.GroundingMetadata != nil || candidate.LogprobsResult != nil {
197197
err = fmt.Errorf("unexpected reply fields (%+v)", *candidate)
198198
return
199199
}

0 commit comments

Comments
 (0)