chore(components): Use new module for looking up ReFINED and AutoSxS …

…image tags

PiperOrigin-RevId: 603560938

components/google-cloud/google_cloud_pipeline_components/_implementation/llm/env.py

@@ -12,17 +12,18 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """A collection of constants shared across components and pipelines."""
+
 import os
 
-_DEFAULT_AUTOSXS_IMAGE_TAG = '20240123_0507_RC00'
+from google_cloud_pipeline_components._implementation.llm.generated import refined_image_versions
 
 
 def get_private_image_tag() -> str:
-  return os.getenv('PRIVATE_IMAGE_TAG') or '20240124_0507_RC00'
+  return os.getenv('PRIVATE_IMAGE_TAG') or refined_image_versions.IMAGE_TAG
 
 
 def get_autosxs_image_tag() -> str:
-  return os.getenv('PRIVATE_IMAGE_TAG') or _DEFAULT_AUTOSXS_IMAGE_TAG
+  return os.getenv('PRIVATE_IMAGE_TAG') or refined_image_versions.IMAGE_TAG
 
 
 def get_use_test_machine_spec() -> bool: