Skip to content

Commit fe585b8

Browse files
authored
refactor(genai/labels): replace vertexai.generative_models with genai (#14450)
* refactor(genai/labels): replace vertexai.generative_models with genai Update the codebase to use the new `genai` SDK replacement for all legacy `vertexai.generative_models` imports and usage. * Address gemini recomendation to assert presence of text attribute. * Removed main and updated year in files.
1 parent 13b64c1 commit fe585b8

5 files changed

Lines changed: 111 additions & 0 deletions

File tree

genai/labels/labels_example.py

Lines changed: 46 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,46 @@
1+
# Copyright 2026 Google LLC
2+
#
3+
# Licensed under the Apache License, Version 2.0 (the "License");
4+
# you may not use this file except in compliance with the License.
5+
# You may obtain a copy of the License at
6+
#
7+
# https://www.apache.org/licenses/LICENSE-2.0
8+
#
9+
# Unless required by applicable law or agreed to in writing, software
10+
# distributed under the License is distributed on an "AS IS" BASIS,
11+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
# See the License for the specific language governing permissions and
13+
# limitations under the License.
14+
15+
# [START aiplatform_genai_gemini_set_labels]
16+
17+
import os
18+
19+
from google import genai
20+
21+
PROJECT_ID = os.getenv("GOOGLE_CLOUD_PROJECT")
22+
LOCATION = os.getenv("LOCATION_ID")
23+
GEMINI_MODEL = os.getenv("GEMINI_MODEL_ID", "gemini-2.5-flash")
24+
25+
26+
def generate_content() -> genai.types.GenerateContentResponse:
27+
28+
genai_client = genai.Client(enterprise=True, project=PROJECT_ID, location=LOCATION)
29+
30+
labels = {
31+
"environment": "testing",
32+
"feature": "genai",
33+
"model": "gemini",
34+
}
35+
36+
config = genai.types.GenerateContentConfig(temperature=0.4, labels=labels)
37+
38+
response = genai_client.models.generate_content(
39+
model=GEMINI_MODEL, contents="What is Generative AI?", config=config
40+
)
41+
42+
print(response.text)
43+
return response
44+
45+
46+
# [END aiplatform_genai_gemini_set_labels]

genai/labels/noxfile_config.py

Lines changed: 42 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,42 @@
1+
# Copyright 2026 Google LLC
2+
#
3+
# Licensed under the Apache License, Version 2.0 (the "License");
4+
# you may not use this file except in compliance with the License.
5+
# You may obtain a copy of the License at
6+
#
7+
# http://www.apache.org/licenses/LICENSE-2.0
8+
#
9+
# Unless required by applicable law or agreed to in writing, software
10+
# distributed under the License is distributed on an "AS IS" BASIS,
11+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
# See the License for the specific language governing permissions and
13+
# limitations under the License.
14+
15+
# Default TEST_CONFIG_OVERRIDE for python repos.
16+
17+
# You can copy this file into your directory, then it will be imported from
18+
# the noxfile.py.
19+
20+
# The source of truth:
21+
# https://github.com/GoogleCloudPlatform/python-docs-samples/blob/main/noxfile_config.py
22+
23+
TEST_CONFIG_OVERRIDE = {
24+
# You can opt out from the test for specific Python versions.
25+
"ignored_versions": ["3.8", "3.9", "3.11", "3.12", "3.13"],
26+
# Old samples are opted out of enforcing Python type hints
27+
# All new samples should feature them
28+
"enforce_type_hints": True,
29+
# An envvar key for determining the project id to use. Change it
30+
# to 'BUILD_SPECIFIC_GCLOUD_PROJECT' if you want to opt in using a
31+
# build specific Cloud project. You can also use your own string
32+
# to use your own Cloud project.
33+
"gcloud_project_env": "GOOGLE_CLOUD_PROJECT",
34+
# 'gcloud_project_env': 'BUILD_SPECIFIC_GCLOUD_PROJECT',
35+
# If you need to use a specific version of pip,
36+
# change pip_version_override to the string representation
37+
# of the version number, for example, "20.2.4"
38+
"pip_version_override": None,
39+
# A dictionary you want to inject into your test. Don't put any
40+
# secrets here. These values will override predefined values.
41+
"envs": {},
42+
}

genai/labels/requirements-test.txt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
google-api-core==2.32.0
2+
pytest==9.0.3

genai/labels/requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
google-genai==2.12.1
Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,20 @@
1+
# Copyright 2026 Google LLC
2+
#
3+
# Licensed under the Apache License, Version 2.0 (the "License");
4+
# you may not use this file except in compliance with the License.
5+
# You may obtain a copy of the License at
6+
#
7+
# https://www.apache.org/licenses/LICENSE-2.0
8+
#
9+
# Unless required by applicable law or agreed to in writing, software
10+
# distributed under the License is distributed on an "AS IS" BASIS,
11+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
# See the License for the specific language governing permissions and
13+
# limitations under the License.
14+
15+
import labels_example
16+
17+
18+
def test_set_labels() -> None:
19+
response = labels_example.generate_content()
20+
assert response.text

0 commit comments

Comments
 (0)