fix: format of the input to GetPublisherModelRequest for Hugging Face models.

PiperOrigin-RevId: 738211073

Author: vertex-sdk-bot

tests/unit/vertexai/model_garden/test_model_garden.py

@@ -33,10 +33,16 @@
 _TEST_MODEL_FULL_RESOURCE_NAME = (
     "publishers/google/models/paligemma@paligemma-224-float32"
 )
+_TEST_HUGGING_FACE_MODEL_FULL_RESOURCE_NAME = (
+    "publishers/meta-llama/models/llama-3.3-70b-instruct"
+)
 _TEST_PUBLISHER_MODEL_NAME = "publishers/google/models/paligemma"
 _TEST_HUGGING_FACE_PUBLISHER_MODEL_NAME = "publishers/hf-google/models/gemma-2-2b"
 _TEST_MODEL_SIMPLIFIED_RESOURCE_NAME = "google/paligemma@paligemma-224-float32"
 _TEST_MODEL_HUGGING_FACE_ID = "meta-llama/Llama-3.3-70B-Instruct"
+_TEST_MODEL_HUGGING_FACE_RESOURCE_NAME = (
+    "publishers/hf-meta-llama/models/llama-3.3-70b-instruct"
+)
 # Note: The full resource name is in lower case.
 _TEST_MODEL_HUGGING_FACE_FULL_RESOURCE_NAME = (
     "publishers/hf-meta-llama/models/llama-3.3-70b-instruct@001"
@@ -120,6 +126,24 @@ def get_publisher_model_mock():
                     )
                 ),
             ),
+            types.PublisherModel(
+                name=_TEST_MODEL_HUGGING_FACE_RESOURCE_NAME,
+                supported_actions=types.PublisherModel.CallToAction(
+                    multi_deploy_vertex=types.PublisherModel.CallToAction.DeployVertex(
+                        multi_deploy_vertex=[
+                            types.PublisherModel.CallToAction.Deploy(
+                                dedicated_resources=types.DedicatedResources(
+                                    machine_spec=types.MachineSpec(
+                                        machine_type="g2-standard-16",
+                                        accelerator_type="NVIDIA_L4",
+                                        accelerator_count=1,
+                                    )
+                                )
+                            )
+                        ]
+                    )
+                ),
+            ),
         ]
         yield get_publisher_model_mock
 
@@ -758,6 +782,16 @@ def test_list_deploy_options(self, get_publisher_model_mock):
             )
         )
 
+        hf_model = model_garden.OpenModel(_TEST_MODEL_HUGGING_FACE_ID)
+        hf_model.list_deploy_options()
+        get_publisher_model_mock.assert_called_with(
+            types.GetPublisherModelRequest(
+                name=_TEST_HUGGING_FACE_MODEL_FULL_RESOURCE_NAME,
+                is_hugging_face_model=True,
+                include_equivalent_model_garden_model_deployment_configs=True,
+            )
+        )
+
     def test_list_deployable_models(self, list_publisher_models_mock):
         """Tests getting the supported deploy options for a model."""
         aiplatform.init(

vertexai/model_garden/_model_garden.py

@@ -147,8 +147,8 @@ def _reconcile_model_name(model_name: str) -> str:
                 )
             else:
                 return _get_publisher_model_resource_name(
-                    publisher="hf-" + simplified_name_match.group("publisher"),
-                    model=simplified_name_match.group("model") + "@001",
+                    publisher=simplified_name_match.group("publisher"),
+                    model=simplified_name_match.group("model"),
                 )
         else:
             raise ValueError(f"`{model_name}` is not a valid Open Model name")
@@ -594,7 +594,7 @@ def list_deploy_options(
         """Lists the verified deploy options for the model."""
         request = types.GetPublisherModelRequest(
             name=self._publisher_model_name,
-            is_hugging_face_model="@" not in self._publisher_model_name,
+            is_hugging_face_model=bool(self._is_hugging_face_model),
             include_equivalent_model_garden_model_deployment_configs=True,
         )
         response = self._us_central1_model_garden_client.get_publisher_model(request)