encounter/adk-python
0
0
Fork 0
mirror of https://github.com/encounter/adk-python.git synced 2026-07-09 18:19:28 -07:00
Code Issues Packages Projects Releases Wiki Activity

feat: add citation_metadata to LlmResponse

Browse Source 
PiperOrigin-RevId: 811997009
This commit is contained in:
Google Team Member
2025-09-26 16:31:01 -07:00
committed by Copybara-Service
parent 7b707cebea
commit 3f28e30c6d
2 changed files with 86 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/google/adk/models/llm_response.py
+8
View File
@@ -126,6 +126,12 @@ class LlmResponse(BaseModel):
This field is automatically populated when context caching is enabled.
"""
citation_metadata: Optional[types.CitationMetadata] = None
"""Citation metadata for the response.
This field is automatically populated when citation is enabled.
"""
@staticmethod
def create(
generate_content_response: types.GenerateContentResponse,
@@ -148,6 +154,7 @@ class LlmResponse(BaseModel):
grounding_metadata=candidate.grounding_metadata,
usage_metadata=usage_metadata,
finish_reason=candidate.finish_reason,
citation_metadata=candidate.citation_metadata,
avg_logprobs=candidate.avg_logprobs,
logprobs_result=candidate.logprobs_result,
)
@@ -155,6 +162,7 @@ class LlmResponse(BaseModel):
return LlmResponse(
error_code=candidate.finish_reason,
error_message=candidate.finish_message,
citation_metadata=candidate.citation_metadata,
usage_metadata=usage_metadata,
finish_reason=candidate.finish_reason,
avg_logprobs=candidate.avg_logprobs,