Skip to content

Commit cf692e3

Browse files
matthew29tangcopybara-github
authored andcommitted
feat: Add rawEvent field to SessionEvents
PiperOrigin-RevId: 889373976
1 parent 11ad542 commit cf692e3

3 files changed

Lines changed: 53 additions & 0 deletions

File tree

google-cloud-vertexai/src/main/java/com/google/cloud/vertexai/genai/SessionEvents.java

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -86,6 +86,13 @@ ObjectNode appendAgentEngineSessionEventConfigToVertex(
8686
Common.getValueByPath(fromObject, new String[] {"eventMetadata"}));
8787
}
8888

89+
if (Common.getValueByPath(fromObject, new String[] {"rawEvent"}) != null) {
90+
Common.setValueByPath(
91+
parentObject,
92+
new String[] {"rawEvent"},
93+
Common.getValueByPath(fromObject, new String[] {"rawEvent"}));
94+
}
95+
8996
return toObject;
9097
}
9198

google-cloud-vertexai/src/main/java/com/google/cloud/vertexai/genai/types/AppendAgentEngineSessionEventConfig.java

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@
2626
import com.google.genai.JsonSerializable;
2727
import com.google.genai.types.Content;
2828
import com.google.genai.types.HttpOptions;
29+
import java.util.Map;
2930
import java.util.Optional;
3031

3132
/** Config for appending agent engine session event. */
@@ -56,6 +57,10 @@ public abstract class AppendAgentEngineSessionEventConfig extends JsonSerializab
5657
@JsonProperty("eventMetadata")
5758
public abstract Optional<EventMetadata> eventMetadata();
5859

60+
/** Weakly typed raw event data in proto struct format. */
61+
@JsonProperty("rawEvent")
62+
public abstract Optional<Map<String, Object>> rawEvent();
63+
5964
/** Instantiates a builder for AppendAgentEngineSessionEventConfig. */
6065
@ExcludeFromGeneratedCoverageReport
6166
public static Builder builder() {
@@ -205,6 +210,24 @@ public Builder clearEventMetadata() {
205210
return eventMetadata(Optional.empty());
206211
}
207212

213+
/**
214+
* Setter for rawEvent.
215+
*
216+
* <p>rawEvent: Weakly typed raw event data in proto struct format.
217+
*/
218+
@JsonProperty("rawEvent")
219+
public abstract Builder rawEvent(Map<String, Object> rawEvent);
220+
221+
@ExcludeFromGeneratedCoverageReport
222+
abstract Builder rawEvent(Optional<Map<String, Object>> rawEvent);
223+
224+
/** Clears the value of rawEvent field. */
225+
@ExcludeFromGeneratedCoverageReport
226+
@CanIgnoreReturnValue
227+
public Builder clearRawEvent() {
228+
return rawEvent(Optional.empty());
229+
}
230+
208231
public abstract AppendAgentEngineSessionEventConfig build();
209232
}
210233

google-cloud-vertexai/src/main/java/com/google/cloud/vertexai/genai/types/SessionEvent.java

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@
2626
import com.google.genai.JsonSerializable;
2727
import com.google.genai.types.Content;
2828
import java.time.Instant;
29+
import java.util.Map;
2930
import java.util.Optional;
3031

3132
/** A session event. */
@@ -71,6 +72,10 @@ public abstract class SessionEvent extends JsonSerializable {
7172
@JsonProperty("timestamp")
7273
public abstract Optional<Instant> timestamp();
7374

75+
/** Optional. Weakly typed raw event data in proto struct format. */
76+
@JsonProperty("rawEvent")
77+
public abstract Optional<Map<String, Object>> rawEvent();
78+
7479
/** Instantiates a builder for SessionEvent. */
7580
@ExcludeFromGeneratedCoverageReport
7681
public static Builder builder() {
@@ -273,6 +278,24 @@ public Builder clearTimestamp() {
273278
return timestamp(Optional.empty());
274279
}
275280

281+
/**
282+
* Setter for rawEvent.
283+
*
284+
* <p>rawEvent: Optional. Weakly typed raw event data in proto struct format.
285+
*/
286+
@JsonProperty("rawEvent")
287+
public abstract Builder rawEvent(Map<String, Object> rawEvent);
288+
289+
@ExcludeFromGeneratedCoverageReport
290+
abstract Builder rawEvent(Optional<Map<String, Object>> rawEvent);
291+
292+
/** Clears the value of rawEvent field. */
293+
@ExcludeFromGeneratedCoverageReport
294+
@CanIgnoreReturnValue
295+
public Builder clearRawEvent() {
296+
return rawEvent(Optional.empty());
297+
}
298+
276299
public abstract SessionEvent build();
277300
}
278301

0 commit comments

Comments
 (0)