Skip to content

Commit 6204b7d

Browse files
matthew29tangcopybara-github
authored andcommitted
chore: Add Java VertexAI generator diff test presubmit
PiperOrigin-RevId: 891828391
1 parent 592f1c0 commit 6204b7d

6 files changed

Lines changed: 158 additions & 5 deletions

File tree

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

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -228,6 +228,10 @@ public abstract class AgentEngineConfig extends JsonSerializable {
228228
@JsonProperty("agentConfigSource")
229229
public abstract Optional<ReasoningEngineSpecSourceCodeSpecAgentConfigSource> agentConfigSource();
230230

231+
/** The container spec for the Agent Engine. */
232+
@JsonProperty("containerSpec")
233+
public abstract Optional<ReasoningEngineSpecContainerSpec> containerSpec();
234+
231235
/** Instantiates a builder for AgentEngineConfig. */
232236
@ExcludeFromGeneratedCoverageReport
233237
public static Builder builder() {
@@ -1017,6 +1021,34 @@ public Builder clearAgentConfigSource() {
10171021
return agentConfigSource(Optional.empty());
10181022
}
10191023

1024+
/**
1025+
* Setter for containerSpec.
1026+
*
1027+
* <p>containerSpec: The container spec for the Agent Engine.
1028+
*/
1029+
@JsonProperty("containerSpec")
1030+
public abstract Builder containerSpec(ReasoningEngineSpecContainerSpec containerSpec);
1031+
1032+
/**
1033+
* Setter for containerSpec builder.
1034+
*
1035+
* <p>containerSpec: The container spec for the Agent Engine.
1036+
*/
1037+
@CanIgnoreReturnValue
1038+
public Builder containerSpec(ReasoningEngineSpecContainerSpec.Builder containerSpecBuilder) {
1039+
return containerSpec(containerSpecBuilder.build());
1040+
}
1041+
1042+
@ExcludeFromGeneratedCoverageReport
1043+
abstract Builder containerSpec(Optional<ReasoningEngineSpecContainerSpec> containerSpec);
1044+
1045+
/** Clears the value of containerSpec field. */
1046+
@ExcludeFromGeneratedCoverageReport
1047+
@CanIgnoreReturnValue
1048+
public Builder clearContainerSpec() {
1049+
return containerSpec(Optional.empty());
1050+
}
1051+
10201052
public abstract AgentEngineConfig build();
10211053
}
10221054

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@
3030
import java.util.List;
3131
import java.util.Optional;
3232

33-
/** The PSC interface config. */
33+
/** Configuration for PSC-I. */
3434
@AutoValue
3535
@JsonDeserialize(builder = PscInterfaceConfig.Builder.class)
3636
public abstract class PscInterfaceConfig extends JsonSerializable {

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@
2626
import com.google.genai.JsonSerializable;
2727
import java.util.Optional;
2828

29-
/** The configuration for agent engine sub-resources to manage context. */
29+
/** Configuration for how Agent Engine sub-resources should manage context. */
3030
@AutoValue
3131
@JsonDeserialize(builder = ReasoningEngineContextSpec.Builder.class)
3232
public abstract class ReasoningEngineContextSpec extends JsonSerializable {

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

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -95,6 +95,10 @@ public abstract class ReasoningEngineSpec extends JsonSerializable {
9595
@JsonProperty("sourceCodeSpec")
9696
public abstract Optional<ReasoningEngineSpecSourceCodeSpec> sourceCodeSpec();
9797

98+
/** Deploy from a container image with a defined entrypoint and commands. */
99+
@JsonProperty("containerSpec")
100+
public abstract Optional<ReasoningEngineSpecContainerSpec> containerSpec();
101+
98102
/** Instantiates a builder for ReasoningEngineSpec. */
99103
@ExcludeFromGeneratedCoverageReport
100104
public static Builder builder() {
@@ -362,6 +366,34 @@ public Builder clearSourceCodeSpec() {
362366
return sourceCodeSpec(Optional.empty());
363367
}
364368

369+
/**
370+
* Setter for containerSpec.
371+
*
372+
* <p>containerSpec: Deploy from a container image with a defined entrypoint and commands.
373+
*/
374+
@JsonProperty("containerSpec")
375+
public abstract Builder containerSpec(ReasoningEngineSpecContainerSpec containerSpec);
376+
377+
/**
378+
* Setter for containerSpec builder.
379+
*
380+
* <p>containerSpec: Deploy from a container image with a defined entrypoint and commands.
381+
*/
382+
@CanIgnoreReturnValue
383+
public Builder containerSpec(ReasoningEngineSpecContainerSpec.Builder containerSpecBuilder) {
384+
return containerSpec(containerSpecBuilder.build());
385+
}
386+
387+
@ExcludeFromGeneratedCoverageReport
388+
abstract Builder containerSpec(Optional<ReasoningEngineSpecContainerSpec> containerSpec);
389+
390+
/** Clears the value of containerSpec field. */
391+
@ExcludeFromGeneratedCoverageReport
392+
@CanIgnoreReturnValue
393+
public Builder clearContainerSpec() {
394+
return containerSpec(Optional.empty());
395+
}
396+
365397
public abstract ReasoningEngineSpec build();
366398
}
367399

Lines changed: 90 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,90 @@
1+
/*
2+
* Copyright 2025 Google LLC
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* https://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
17+
// Auto-generated code. Do not edit.
18+
19+
package com.google.cloud.vertexai.genai.types;
20+
21+
import com.fasterxml.jackson.annotation.JsonCreator;
22+
import com.fasterxml.jackson.annotation.JsonProperty;
23+
import com.fasterxml.jackson.databind.annotation.JsonDeserialize;
24+
import com.google.auto.value.AutoValue;
25+
import com.google.errorprone.annotations.CanIgnoreReturnValue;
26+
import com.google.genai.JsonSerializable;
27+
import java.util.Optional;
28+
29+
/** Specification for deploying from a container image. */
30+
@AutoValue
31+
@JsonDeserialize(builder = ReasoningEngineSpecContainerSpec.Builder.class)
32+
public abstract class ReasoningEngineSpecContainerSpec extends JsonSerializable {
33+
/**
34+
* Required. The Artifact Registry Docker image URI (e.g.,
35+
* us-central1-docker.pkg.dev/my-project/my-repo/my-image:tag) of the container image that is to
36+
* be run on each worker replica.
37+
*/
38+
@JsonProperty("imageUri")
39+
public abstract Optional<String> imageUri();
40+
41+
/** Instantiates a builder for ReasoningEngineSpecContainerSpec. */
42+
@ExcludeFromGeneratedCoverageReport
43+
public static Builder builder() {
44+
return new AutoValue_ReasoningEngineSpecContainerSpec.Builder();
45+
}
46+
47+
/** Creates a builder with the same values as this instance. */
48+
public abstract Builder toBuilder();
49+
50+
/** Builder for ReasoningEngineSpecContainerSpec. */
51+
@AutoValue.Builder
52+
public abstract static class Builder {
53+
/**
54+
* For internal usage. Please use `ReasoningEngineSpecContainerSpec.builder()` for
55+
* instantiation.
56+
*/
57+
@JsonCreator
58+
private static Builder create() {
59+
return new AutoValue_ReasoningEngineSpecContainerSpec.Builder();
60+
}
61+
62+
/**
63+
* Setter for imageUri.
64+
*
65+
* <p>imageUri: Required. The Artifact Registry Docker image URI (e.g.,
66+
* us-central1-docker.pkg.dev/my-project/my-repo/my-image:tag) of the container image that is to
67+
* be run on each worker replica.
68+
*/
69+
@JsonProperty("imageUri")
70+
public abstract Builder imageUri(String imageUri);
71+
72+
@ExcludeFromGeneratedCoverageReport
73+
abstract Builder imageUri(Optional<String> imageUri);
74+
75+
/** Clears the value of imageUri field. */
76+
@ExcludeFromGeneratedCoverageReport
77+
@CanIgnoreReturnValue
78+
public Builder clearImageUri() {
79+
return imageUri(Optional.empty());
80+
}
81+
82+
public abstract ReasoningEngineSpecContainerSpec build();
83+
}
84+
85+
/** Deserializes a JSON string to a ReasoningEngineSpecContainerSpec object. */
86+
@ExcludeFromGeneratedCoverageReport
87+
public static ReasoningEngineSpecContainerSpec fromJson(String jsonString) {
88+
return JsonSerializable.fromJsonString(jsonString, ReasoningEngineSpecContainerSpec.class);
89+
}
90+
}

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

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -28,9 +28,8 @@
2828
import java.util.Optional;
2929

3030
/**
31-
* The image spec for building an image (within a single build step).
32-
*
33-
* <p>It is based on the config file (i.e. Dockerfile) in the source directory.
31+
* The image spec for building an image (within a single build step), based on the config file (i.e.
32+
* Dockerfile) in the source directory.
3433
*/
3534
@AutoValue
3635
@JsonDeserialize(builder = ReasoningEngineSpecSourceCodeSpecImageSpec.Builder.class)

0 commit comments

Comments
 (0)