fix: correct the rhel_ai_version_id source (#359)

Signed-off-by: Xiaoxue Wang <[email protected]>
RedHatInsights · Aug 21, 2024 · fa2b0fb · fa2b0fb
1 parent d8640e4
commit fa2b0fb
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/dev/test-archives/core-base/data/etc/os-release b/dev/test-archives/core-base/data/etc/os-release
@@ -18,4 +18,4 @@ REDHAT_SUPPORT_PRODUCT_VERSION="9.4"
 OSTREE_VERSION='9.20240630.0'
 VARIANT_ID=rhel_ai
 VARIANT="RHEL AI"
-BUILD_ID='v1.1.3'
+RHEL_AI_VERSION_ID='v1.1.3'
diff --git a/src/puptoo/process/profile.py b/src/puptoo/process/profile.py
@@ -755,7 +755,7 @@ def system_profile(
         if variant_id == 'rhel_ai':
             rhel_ai_profile = {
                 "variant": os_release_parser.get("VARIANT"),
-                "rhel_ai_version_id": os_release_parser.get("BUILD_ID"),
+                "rhel_ai_version_id": os_release_parser.get("RHEL_AI_VERSION_ID"),
             }
             if nvidia_smi_l:
                 rhel_ai_profile["nvidia_gpu_models"] = [gpu["model"] for gpu in nvidia_smi_l]

diff --git a/tests/test_rhel_ai.py b/tests/test_rhel_ai.py
@@ -24,7 +24,7 @@
 OSTREE_VERSION='9.20240630.0'
 VARIANT_ID=rhel_ai
 VARIANT="RHEL AI"
-BUILD_ID='v1.1.3'
+RHEL_AI_VERSION_ID='v1.1.3'
 """.strip()
 
 OS_RELEASE_RHEL = """