Skip to content

Commit 6af5bc1

Browse files
nardosalemunardosalemu
committed
feat: add code samples for model optimizer
1 parent 6b3375e commit 6af5bc1

File tree

4 files changed

+325
-0
lines changed

4 files changed

+325
-0
lines changed

‎genai/pom.xml‎

Lines changed: 76 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,76 @@
1+
<?xml version="1.0" encoding="UTF-8"?>
2+
<!--
3+
Copyright 2025 Google Inc.
4+
5+
Licensed under the Apache License, Version 2.0 (the "License");
6+
you may not use this file except in compliance with the License.
7+
You may obtain a copy of the License at
8+
9+
http://www.apache.org/licenses/LICENSE-2.0
10+
11+
Unless required by applicable law or agreed to in writing, software
12+
distributed under the License is distributed on an "AS IS" BASIS,
13+
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14+
See the License for the specific language governing permissions and
15+
limitations under the License.
16+
-->
17+
<project xmlns="http://maven.apache.org/POM/4.0.0"
18+
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
19+
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
20+
<modelVersion>4.0.0</modelVersion>
21+
<packaging>jar</packaging>
22+
<groupId>com.example.genai</groupId>
23+
<artifactId>genai-snippets</artifactId>
24+
<name>Google Cloud Gen AI Snippets</name>
25+
<url>https://github.com/GoogleCloudPlatform/java-docs-samples/tree/main/vision</url>
26+
27+
<!--
28+
The parent pom defines common style checks and testing strategies for our samples.
29+
Removing or replacing it should not affect the execution of the samples in any way.
30+
-->
31+
<parent>
32+
<groupId>com.google.cloud.samples</groupId>
33+
<artifactId>shared-configuration</artifactId>
34+
<version>1.2.0</version>
35+
</parent>
36+
37+
<properties>
38+
<maven.compiler.target>11</maven.compiler.target>
39+
<maven.compiler.source>11</maven.compiler.source>
40+
<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
41+
</properties>
42+
43+
<dependencyManagement>
44+
<dependencies>
45+
<dependency>
46+
<groupId>com.google.cloud</groupId>
47+
<artifactId>libraries-bom</artifactId>
48+
<scope>import</scope>
49+
<type>pom</type>
50+
<version>26.43.0</version>
51+
</dependency>
52+
</dependencies>
53+
</dependencyManagement>
54+
55+
<dependencies>
56+
<dependency>
57+
<groupId>com.google.genai</groupId>
58+
<artifactId>google-genai</artifactId>
59+
<version>0.5.0</version>
60+
</dependency>
61+
62+
<!-- test dependencies -->
63+
<dependency>
64+
<groupId>junit</groupId>
65+
<artifactId>junit</artifactId>
66+
<version>4.13.2</version>
67+
<scope>test</scope>
68+
</dependency>
69+
<dependency>
70+
<groupId>com.google.truth</groupId>
71+
<artifactId>truth</artifactId>
72+
<version>1.4.0</version>
73+
<scope>test</scope>
74+
</dependency>
75+
</dependencies>
76+
</project>

‎genai/src/main/java/genai/GenerateContentWithRouting.java‎

Lines changed: 48 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,48 @@
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+
* http://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+
package genai;
18+
19+
// [START genai_generate_content_with_routing]
20+
import com.google.genai.Client;
21+
import com.google.genai.types.GenerateContentResponse;
22+
import com.google.genai.types.GenerateContentConfig;
23+
import com.google.genai.types.ModelSelectionConfig;
24+
import com.google.genai.types.HttpOptions;
25+
26+
27+
public class GenerateContentWithRouting {
28+
29+
public static void main(String[] args) throws Exception {
30+
31+
32+
String modelName = "model-optimizer-exp-04-09";
33+
34+
HttpOptions httpOptions = HttpOptions.builder().apiVersion("v1beta1").build();
35+
36+
Client client = Client.builder().httpOptions(httpOptions).vertexAI(true).build();
37+
38+
ModelSelectionConfig modelSelectionConfig = ModelSelectionConfig.builder().featureSelectionPreference("PRIORITIZE_COST").build();
39+
40+
GenerateContentConfig generateContentConfig = GenerateContentConfig.builder().modelSelectionConfig(modelSelectionConfig).build();
41+
42+
GenerateContentResponse response =
43+
client.models.generateContent(modelName, "Why do we have 365 days in a year?", generateContentConfig);
44+
45+
System.out.println("Response: " + response.text());
46+
}
47+
}
48+
// [END genai_generate_content_with_routing]

‎genai/src/main/java/genai/StreamGenerateContentWithRouting.java‎

Lines changed: 54 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,54 @@
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+
* http://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+
package genai;
18+
19+
// [START genai_stream_generate_content_with_routing]
20+
import com.google.genai.Client;
21+
import com.google.genai.types.GenerateContentResponse;
22+
import com.google.genai.types.GenerateContentConfig;
23+
import com.google.genai.types.ModelSelectionConfig;
24+
import com.google.genai.types.HttpOptions;
25+
import com.google.genai.ResponseStream;
26+
27+
28+
public class StreamGenerateContentWithRouting {
29+
30+
public static void main(String[] args) throws Exception {
31+
32+
33+
String modelName = "model-optimizer-exp-04-09";
34+
35+
HttpOptions httpOptions = HttpOptions.builder().apiVersion("v1beta1").build();
36+
37+
Client client = Client.builder().httpOptions(httpOptions).vertexAI(true).build();
38+
39+
ModelSelectionConfig modelSelectionConfig = ModelSelectionConfig.builder().featureSelectionPreference("BALANCED").build();
40+
41+
GenerateContentConfig generateContentConfig = GenerateContentConfig.builder().modelSelectionConfig(modelSelectionConfig).build();
42+
43+
ResponseStream<GenerateContentResponse> responseStream =
44+
client.models.generateContentStream(modelName, "Why do we have 365 days in a year?", generateContentConfig);
45+
46+
System.out.println("Streaming response: ");
47+
for (GenerateContentResponse res : responseStream) {
48+
System.out.print(res.text());
49+
}
50+
51+
responseStream.close();
52+
}
53+
}
54+
// [END genai_stream_generate_content_with_routing]

‎storage-transfer/bin/pom.xml‎

Lines changed: 147 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,147 @@
1+
<?xml version="1.0" encoding="UTF-8"?>
2+
<!--
3+
~ Copyright (c) 2015 Google Inc.
4+
~
5+
~ Licensed under the Apache License, Version 2.0 (the "License"); you
6+
~ may not use this file except in compliance with the License. You may
7+
~ obtain a copy of the License at
8+
~
9+
~ http://www.apache.org/licenses/LICENSE-2.0
10+
~
11+
~ Unless required by applicable law or agreed to in writing, software
12+
~ distributed under the License is distributed on an "AS IS" BASIS,
13+
~ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
14+
~ implied. See the License for the specific language governing
15+
~ permissions and limitations under the License.
16+
-->
17+
18+
<project xmlns="http://maven.apache.org/POM/4.0.0"
19+
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
20+
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
21+
<modelVersion>4.0.0</modelVersion>
22+
23+
<groupId>com.example.storagetransfer</groupId>
24+
<artifactId>storage-transfer-sample</artifactId>
25+
<version>0.1</version>
26+
<packaging>jar</packaging>
27+
<!--
28+
The parent pom defines common style checks and testing strategies for our samples.
29+
Removing or replacing it should not affect the execution of the samples in anyway.
30+
-->
31+
<parent>
32+
<groupId>com.google.cloud.samples</groupId>
33+
<artifactId>shared-configuration</artifactId>
34+
<version>1.2.0</version>
35+
</parent>
36+
37+
<properties>
38+
<maven.compiler.target>1.8</maven.compiler.target>
39+
<maven.compiler.source>1.8</maven.compiler.source>
40+
<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
41+
</properties>
42+
43+
<dependencyManagement>
44+
<dependencies>
45+
<dependency>
46+
<artifactId>libraries-bom</artifactId>
47+
<groupId>com.google.cloud</groupId>
48+
<scope>import</scope>
49+
<type>pom</type>
50+
<version>26.32.0</version>
51+
</dependency>
52+
</dependencies>
53+
</dependencyManagement>
54+
55+
<dependencies>
56+
<dependency>
57+
<groupId>com.google.apis</groupId>
58+
<artifactId>google-api-services-storagetransfer</artifactId>
59+
<version>v1-rev20240126-2.0.0</version>
60+
<exclusions>
61+
<exclusion> <!-- exclude an old version of Guava -->
62+
<groupId>com.google.guava</groupId>
63+
<artifactId>guava-jdk5</artifactId>
64+
</exclusion>
65+
</exclusions>
66+
</dependency>
67+
<dependency>
68+
<groupId>com.google.cloud</groupId>
69+
<artifactId>google-cloud-storage-transfer</artifactId>
70+
</dependency>
71+
<dependency>
72+
<groupId>com.google.auth</groupId>
73+
<artifactId>google-auth-library-oauth2-http</artifactId>
74+
</dependency>
75+
<dependency>
76+
<groupId>com.google.guava</groupId>
77+
<artifactId>guava</artifactId>
78+
</dependency>
79+
80+
<!-- Test Dependencies -->
81+
<dependency>
82+
<groupId>com.google.truth</groupId>
83+
<artifactId>truth</artifactId>
84+
<version>1.4.0</version>
85+
<scope>test</scope>
86+
</dependency>
87+
88+
<dependency>
89+
<groupId>junit</groupId>
90+
<artifactId>junit</artifactId>
91+
<version>4.13.2</version>
92+
<scope>test</scope>
93+
</dependency>
94+
95+
<dependency>
96+
<groupId>org.mockito</groupId>
97+
<artifactId>mockito-core</artifactId>
98+
<version>5.10.0</version>
99+
<scope>test</scope>
100+
</dependency>
101+
102+
<dependency>
103+
<groupId>com.google.cloud</groupId>
104+
<artifactId>google-cloud-storage</artifactId>
105+
<scope>test</scope>
106+
</dependency>
107+
108+
<dependency>
109+
<groupId>com.google.cloud</groupId>
110+
<artifactId>google-cloud-pubsub</artifactId>
111+
<scope>test</scope>
112+
</dependency>
113+
114+
<dependency>
115+
<groupId>com.amazonaws</groupId>
116+
<artifactId>aws-java-sdk-s3</artifactId>
117+
<version>1.12.657</version>
118+
<scope>test</scope>
119+
</dependency>
120+
121+
<dependency>
122+
<groupId>com.amazonaws</groupId>
123+
<artifactId>aws-java-sdk-sqs</artifactId>
124+
<version>1.12.657</version>
125+
<scope>test</scope>
126+
</dependency>
127+
128+
<dependency>
129+
<groupId>com.azure</groupId>
130+
<artifactId>azure-storage-blob</artifactId>
131+
<version>12.25.1</version>
132+
</dependency>
133+
134+
<dependency>
135+
<artifactId>slf4j-api</artifactId>
136+
<groupId>org.slf4j</groupId>
137+
<version>2.0.12</version>
138+
</dependency>
139+
140+
<dependency>
141+
<artifactId>slf4j-simple</artifactId>
142+
<groupId>org.slf4j</groupId>
143+
<version>2.0.12</version>
144+
</dependency>
145+
146+
</dependencies>
147+
</project>

0 commit comments

Comments
 (0)