Skip to content

Commit 99f0078

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
fix: Handle empty GenerateMemories response (no memories generated) as valid response.
PiperOrigin-RevId: 802295630
1 parent 74e52d9 commit 99f0078

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

vertexai/_genai/memories.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1284,10 +1284,8 @@ def generate(
12841284
operation_name=operation.name,
12851285
get_operation_fn=self._get_generate_memories_operation,
12861286
)
1287-
if not operation.response:
1288-
if operation.error:
1289-
raise RuntimeError(f"Failed to generate memory: {operation.error}")
1290-
raise RuntimeError(f"Error generating memory: {operation}")
1287+
if operation.error:
1288+
raise RuntimeError(f"Failed to generate memory: {operation.error}")
12911289
return operation
12921290

12931291
def list(

0 commit comments

Comments
 (0)