feat: LLM - Released ChatModel tuning to GA

Ark-kun · copybara-github · commit 7d667f9f8be2 · 2023-09-27T23:35:49.000-07:00
PiperOrigin-RevId: 569078680
diff --git a/tests/unit/aiplatform/test_language_models.py b/tests/unit/aiplatform/test_language_models.py
@@ -1557,7 +1557,7 @@ def test_tune_chat_model(
                 _CHAT_BISON_PUBLISHER_MODEL_DICT
             ),
         ):
-            model = preview_language_models.ChatModel.from_pretrained("chat-bison@001")
+            model = language_models.ChatModel.from_pretrained("chat-bison@001")
 
             default_context = "Default context"
             tuning_job = model.tune_model(
diff --git a/vertexai/language_models/_language_models.py b/vertexai/language_models/_language_models.py
@@ -1338,7 +1338,7 @@ def start_chat(
         )
 
 
-class ChatModel(_ChatModelBase):
+class ChatModel(_ChatModelBase, _TunableChatModelMixin):
     """ChatModel represents a language model that is capable of chat.
 
     Examples::

Original file line number	Diff line number	Diff line change
`@@ -1338,7 +1338,7 @@ def start_chat(`
`1338`	`1338`	`)`
`1339`	`1339`
`1340`	`1340`
`1341`		`-class ChatModel(_ChatModelBase):`
	`1341`	`+class ChatModel(_ChatModelBase, _TunableChatModelMixin):`
`1342`	`1342`	`"""ChatModel represents a language model that is capable of chat.`
`1343`	`1343`
`1344`	`1344`	`Examples::`