Skip to content

Commit 22763e6

Browse files
committed
add notes for model_name docs
1 parent 9d44b8e commit 22763e6

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

bigframes/ml/llm.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -919,25 +919,27 @@ def to_gbq(
919919

920920

921921
@typing_extensions.deprecated(
922-
"gemini-1.0-X and gemini-1.5-X are going to be deprecated. Use gemini-2.0-X (https://cloud.google.com/python/docs/reference/bigframes/latest/bigframes.ml.llm.GeminiTextGenerator) instead. ",
922+
"gemini-pro and gemini-1.5-X are going to be deprecated. Use gemini-2.0-X (https://cloud.google.com/python/docs/reference/bigframes/latest/bigframes.ml.llm.GeminiTextGenerator) instead. ",
923923
category=exceptions.ApiDeprecationWarning,
924924
)
925925
@log_adapter.class_logger
926926
class GeminiTextGenerator(base.RetriableRemotePredictor):
927927
"""Gemini text generator LLM model.
928928
929929
.. note::
930-
gemini-1.0-X and gemini-1.5-X are going to be deprecated. Use gemini-2.0-X (https://cloud.google.com/python/docs/reference/bigframes/latest/bigframes.ml.llm.GeminiTextGenerator) instead.
930+
gemini-pro and gemini-1.5-X are going to be deprecated. Use gemini-2.0-X (https://cloud.google.com/python/docs/reference/bigframes/latest/bigframes.ml.llm.GeminiTextGenerator) instead.
931931
932932
Args:
933933
model_name (str, Default to "gemini-pro"):
934934
The model for natural language tasks. Accepted values are "gemini-pro", "gemini-1.5-pro-preview-0514", "gemini-1.5-flash-preview-0514", "gemini-1.5-pro-001", "gemini-1.5-pro-002", "gemini-1.5-flash-001", "gemini-1.5-flash-002" and "gemini-2.0-flash-exp". Default to "gemini-pro".
935935
936936
.. note::
937+
"gemini-pro" is going to be deprecated. Will change to gemini-2.0-X for Bigframes 2.
937938
"gemini-2.0-flash-exp", "gemini-1.5-pro-preview-0514" and "gemini-1.5-flash-preview-0514" is subject to the "Pre-GA Offerings Terms" in the General Service Terms section of the
938939
Service Specific Terms(https://cloud.google.com/terms/service-terms#1). Pre-GA products and features are available "as is"
939940
and might have limited support. For more information, see the launch stage descriptions
940941
(https://cloud.google.com/products#product-launch-stages).
942+
"
941943
942944
session (bigframes.Session or None):
943945
BQ session to create the model. If None, use the global default session.

0 commit comments

Comments
 (0)