fix(components): Update base image for KFP lightweight component for VPC SC compliance

PiperOrigin-RevId: 600622671

Author: Googler

components/google-cloud/google_cloud_pipeline_components/_implementation/model_evaluation/endpoint_batch_predict/component.py

@@ -16,6 +16,7 @@
 from typing import Dict, List, NamedTuple, Optional, Union
 from google_cloud_pipeline_components import utils as gcpc_utils
 from google_cloud_pipeline_components._implementation.model_evaluation import utils
+from google_cloud_pipeline_components._implementation.model_evaluation import version
 from kfp import dsl
 from kfp.dsl import Artifact
 from kfp.dsl import container_component
@@ -26,7 +27,7 @@
 _IMAGE_URI = 'us-docker.pkg.dev/vertex-evaluation/public/llm:wjess-fishfooding'
 
 
-@dsl.component
+@dsl.component(base_image=version.LLM_EVAL_IMAGE_TAG)
 def add_json_escape_parameters(parameters: dict) -> str:
   if not parameters:
     return
@@ -36,7 +37,7 @@ def add_json_escape_parameters(parameters: dict) -> str:
   return json_escaped_parameters
 
 
-@dsl.component
+@dsl.component(base_image=version.LLM_EVAL_IMAGE_TAG)
 def add_json_escape_paths(paths: list) -> str:
   if not paths:
     return