Skip to content

Commit 052914c

Browse files
Google APIscopybara-github
authored andcommitted
feat: Add custom session id field to create session v1beta1
feat: Add custom memory id field to create memory v1beta1 feat: Add raw_event field to event proto v1beta1 PiperOrigin-RevId: 879826477
1 parent 67b1168 commit 052914c

File tree

3 files changed

+25
-1
lines changed

3 files changed

+25
-1
lines changed

google/cloud/aiplatform/v1beta1/memory_bank_service.proto

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
// Copyright 2025 Google LLC
1+
// Copyright 2026 Google LLC
22
//
33
// Licensed under the Apache License, Version 2.0 (the "License");
44
// you may not use this file except in compliance with the License.
@@ -52,6 +52,7 @@ service MemoryBankService {
5252
body: "memory"
5353
}
5454
};
55+
option (google.api.method_signature) = "parent,memory,memory_id";
5556
option (google.longrunning.operation_info) = {
5657
response_type: "Memory"
5758
metadata_type: "CreateMemoryOperationMetadata"
@@ -160,6 +161,15 @@ message CreateMemoryRequest {
160161

161162
// Required. The Memory to be created.
162163
Memory memory = 2 [(google.api.field_behavior) = REQUIRED];
164+
165+
// Optional. The user defined ID to use for memory, which will become the
166+
// final component of the memory resource name. If not provided, Vertex AI
167+
// will generate a value for this ID.
168+
//
169+
// This value may be up to 63 characters, and valid characters are
170+
// `[a-z0-9-]`. The first character must be a letter, and the last character
171+
// must be a letter or number.
172+
string memory_id = 3 [(google.api.field_behavior) = OPTIONAL];
163173
}
164174

165175
// Details of

google/cloud/aiplatform/v1beta1/session.proto

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -131,6 +131,10 @@ message SessionEvent {
131131

132132
// Optional. Metadata relating to this event.
133133
EventMetadata event_metadata = 11 [(google.api.field_behavior) = OPTIONAL];
134+
135+
// Optional. Weakly typed raw event data in proto struct format.
136+
google.protobuf.Struct raw_event = 12
137+
[(google.api.field_behavior) = OPTIONAL];
134138
}
135139

136140
// Metadata relating to a LLM response event.

google/cloud/aiplatform/v1beta1/session_service.proto

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,7 @@ service SessionService {
5252
}
5353
};
5454
option (google.api.method_signature) = "parent,session";
55+
option (google.api.method_signature) = "parent,session,session_id";
5556
option (google.longrunning.operation_info) = {
5657
response_type: "Session"
5758
metadata_type: "CreateSessionOperationMetadata"
@@ -152,6 +153,15 @@ message CreateSessionRequest {
152153

153154
// Required. The session to create.
154155
Session session = 2 [(google.api.field_behavior) = REQUIRED];
156+
157+
// Optional. The user defined ID to use for session, which will become the
158+
// final component of the session resource name. If not provided, Vertex AI
159+
// will generate a value for this ID.
160+
//
161+
// This value may be up to 63 characters, and valid characters are
162+
// `[a-z0-9-]`. The first character must be a letter, and the last character
163+
// must be a letter or number.
164+
string session_id = 3 [(google.api.field_behavior) = OPTIONAL];
155165
}
156166

157167
// Metadata associated with the

0 commit comments

Comments
 (0)