diff options
Diffstat (limited to 'pkg/aflow/llm_agent.go')
| -rw-r--r-- | pkg/aflow/llm_agent.go | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/pkg/aflow/llm_agent.go b/pkg/aflow/llm_agent.go index 76661add6..b897643c7 100644 --- a/pkg/aflow/llm_agent.go +++ b/pkg/aflow/llm_agent.go @@ -191,9 +191,9 @@ func (a *LLMAgent) parseResponse(resp *genai.GenerateContentResponse) ( err = fmt.Errorf("%v (%v)", candidate.FinishMessage, candidate.FinishReason) return } - // We don't expect to receive these now. - if candidate.GroundingMetadata != nil || candidate.CitationMetadata != nil || - candidate.LogprobsResult != nil { + // We don't expect to receive these fields now. + // Note: CitationMetadata may be present sometimes, but we don't have uses for it. + if candidate.GroundingMetadata != nil || candidate.LogprobsResult != nil { err = fmt.Errorf("unexpected reply fields (%+v)", *candidate) return } |