Skip to content

Commit 2ff00c0

Browse files
committed
fix(genai): Migrate Model Optimiser sample and rollback SDK update for Video Generation
1 parent 54e7f0e commit 2ff00c0

File tree

4 files changed

+54
-53
lines changed

4 files changed

+54
-53
lines changed
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
google-genai==1.27.0
1+
google-genai==1.20.0
Lines changed: 46 additions & 46 deletions
Original file line numberDiff line numberDiff line change
@@ -1,49 +1,49 @@
1-
# Copyright 2025 Google LLC
1+
# # Copyright 2025 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.
214
#
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
615
#
7-
# https://www.apache.org/licenses/LICENSE-2.0
16+
# # TODO: Migrate model_optimizer samples to /model_optimizer
17+
# # and deprecate following sample
18+
# def generate_content() -> str:
19+
# # [START googlegenaisdk_model_optimizer_textgen_with_txt]
20+
# from google import genai
21+
# from google.genai.types import (
22+
# FeatureSelectionPreference,
23+
# GenerateContentConfig,
24+
# HttpOptions,
25+
# ModelSelectionConfig
26+
# )
827
#
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-
16-
# TODO: Migrate model_optimizer samples to /model_optimizer
17-
# and deprecate following sample
18-
def generate_content() -> str:
19-
# [START googlegenaisdk_model_optimizer_textgen_with_txt]
20-
from google import genai
21-
from google.genai.types import (
22-
FeatureSelectionPreference,
23-
GenerateContentConfig,
24-
HttpOptions,
25-
ModelSelectionConfig
26-
)
27-
28-
client = genai.Client(http_options=HttpOptions(api_version="v1beta1"))
29-
response = client.models.generate_content(
30-
model="model-optimizer-exp-04-09",
31-
contents="How does AI work?",
32-
config=GenerateContentConfig(
33-
model_selection_config=ModelSelectionConfig(
34-
feature_selection_preference=FeatureSelectionPreference.BALANCED # Options: PRIORITIZE_QUALITY, BALANCED, PRIORITIZE_COST
35-
),
36-
),
37-
)
38-
print(response.text)
39-
# Example response:
40-
# Okay, let's break down how AI works. It's a broad field, so I'll focus on the ...
41-
#
42-
# Here's a simplified overview:
43-
# ...
44-
# [END googlegenaisdk_model_optimizer_textgen_with_txt]
45-
return response.text
46-
47-
48-
if __name__ == "__main__":
49-
generate_content()
28+
# client = genai.Client(http_options=HttpOptions(api_version="v1beta1"))
29+
# response = client.models.generate_content(
30+
# model="model-optimizer-exp-04-09",
31+
# contents="How does AI work?",
32+
# config=GenerateContentConfig(
33+
# model_selection_config=ModelSelectionConfig(
34+
# feature_selection_preference=FeatureSelectionPreference.BALANCED # Options: PRIORITIZE_QUALITY, BALANCED, PRIORITIZE_COST
35+
# ),
36+
# ),
37+
# )
38+
# print(response.text)
39+
# # Example response:
40+
# # Okay, let's break down how AI works. It's a broad field, so I'll focus on the ...
41+
# #
42+
# # Here's a simplified overview:
43+
# # ...
44+
# # [END googlegenaisdk_model_optimizer_textgen_with_txt]
45+
# return response.text
46+
#
47+
#
48+
# if __name__ == "__main__":
49+
# generate_content()

genai/text_generation/test_text_generation_examples.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -137,8 +137,9 @@ def test_textgen_with_youtube_video() -> None:
137137
assert response
138138

139139

140-
def test_model_optimizer_textgen_with_txt() -> None:
141-
os.environ["GOOGLE_CLOUD_LOCATION"] = "us-central1"
142-
response = model_optimizer_textgen_with_txt.generate_content()
143-
os.environ["GOOGLE_CLOUD_LOCATION"] = "global" # "us-central1"
144-
assert response
140+
# Migrated to Model Optimser Folder
141+
# def test_model_optimizer_textgen_with_txt() -> None:
142+
# os.environ["GOOGLE_CLOUD_LOCATION"] = "us-central1"
143+
# response = model_optimizer_textgen_with_txt.generate_content()
144+
# os.environ["GOOGLE_CLOUD_LOCATION"] = "global" # "us-central1"
145+
# assert response
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
google-genai==1.27.0
1+
google-genai==1.20.0

0 commit comments

Comments
 (0)