Skip to content

Commit 59df355

Browse files
committed
refactor: assign file_uri to local variable to avoid repeated attribute access
1 parent 65112bf commit 59df355

1 file changed

Lines changed: 5 additions & 4 deletions

File tree

src/google/adk/artifacts/gcs_artifact_service.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -231,17 +231,18 @@ def _save_artifact(
231231
content_type="text/plain",
232232
)
233233
elif artifact.file_data:
234-
if not artifact.file_data.file_uri:
234+
file_uri = artifact.file_data.file_uri
235+
if not file_uri:
235236
raise InputValidationError("Artifact file_data must have a file_uri.")
236237
if artifact_util.is_artifact_ref(artifact):
237-
if not artifact_util.parse_artifact_uri(artifact.file_data.file_uri):
238+
if not artifact_util.parse_artifact_uri(file_uri):
238239
raise InputValidationError(
239-
f"Invalid artifact reference URI: {artifact.file_data.file_uri}"
240+
f"Invalid artifact reference URI: {file_uri}"
240241
)
241242
# Store the URI as blob metadata; no content to upload.
242243
blob.metadata = {
243244
**(blob.metadata or {}),
244-
"file_uri": artifact.file_data.file_uri,
245+
"file_uri": file_uri,
245246
}
246247
blob.upload_from_string(
247248
b"",

0 commit comments

Comments
 (0)