feat: LLM - Released CodeGenerationModel tuning to GA

PiperOrigin-RevId: 577749202

Author: Ark-kun

tests/unit/aiplatform/test_language_models.py

@@ -1867,7 +1867,7 @@ def test_tune_code_generation_model(
                 _CODE_GENERATION_BISON_PUBLISHER_MODEL_DICT
             ),
         ):
-            model = preview_language_models.CodeGenerationModel.from_pretrained(
+            model = language_models.CodeGenerationModel.from_pretrained(
                 "code-bison@001"
             )
             # The tune_model call needs to be inside the PublisherModel mock

vertexai/language_models/_language_models.py

@@ -2405,7 +2405,7 @@ def send_message_streaming_async(
         )
 
 
-class CodeGenerationModel(_LanguageModel):
+class _CodeGenerationModel(_LanguageModel):
     """A language model that generates code.
 
     Examples:
@@ -2686,8 +2686,12 @@ def count_tokens(
         )
 
 
+class CodeGenerationModel(_CodeGenerationModel, _TunableTextModelMixin):
+    pass
+
+
 class _PreviewCodeGenerationModel(
-    CodeGenerationModel, _TunableModelMixin, _CountTokensCodeGenerationMixin
+    CodeGenerationModel, _CountTokensCodeGenerationMixin
 ):
     __name__ = "CodeGenerationModel"
     __module__ = "vertexai.preview.language_models"