sort djl tables (#5629)

Signed-off-by: sirutBuasai <sirutbuasai27@outlook.com>

Author: sirutBuasai

docs/src/generate.py

@@ -15,6 +15,7 @@
 import logging
 from pathlib import Path
 
+import sorter as sorter_module
 from constants import (
     AVAILABLE_IMAGES_TABLE_HEADER,
     GLOBAL_CONFIG,
@@ -34,7 +35,6 @@
     sort_by_version,
 )
 from jinja2 import Template
-from sorter import accelerator_sorter, platform_sorter, repository_sorter
 from utils import (
     get_framework_order,
     load_jinja2,
@@ -44,6 +44,9 @@
     write_output,
 )
 
+DEFAULT_TIEBREAKERS = [sorter_module.platform_sorter, sorter_module.accelerator_sorter]
+
+
 LOGGER = logging.getLogger(__name__)
 
 
@@ -100,7 +103,11 @@ def _generate_framework_index(
     for version in sorted_versions:
         sorted_images = sort_by_version(
             images_by_version[version],
-            tiebreakers=[repository_sorter, platform_sorter, accelerator_sorter],
+            tiebreakers=[
+                sorter_module.repository_sorter,
+                sorter_module.platform_sorter,
+                sorter_module.accelerator_sorter,
+            ],
         )
 
         supported = [img for img in sorted_images if img.is_supported]
@@ -290,8 +297,14 @@ def generate_available_images(dry_run: bool = False) -> str:
         columns = table_config.get("columns", [])
         has_public_registry = check_public_registry(images, repository)
 
-        # Sort images by version desc, platform, accelerator
-        images = sort_by_version(images, tiebreakers=[platform_sorter, accelerator_sorter])
+        # Sort images by version desc with tiebreakers from config or defaults
+        tiebreaker_names = table_config.get("tiebreakers")
+        tiebreakers = (
+            [getattr(sorter_module, name) for name in tiebreaker_names]
+            if tiebreaker_names
+            else DEFAULT_TIEBREAKERS
+        )
+        images = sort_by_version(images, tiebreakers=tiebreakers)
 
         # Build table
         headers = [col["header"] for col in columns]

docs/src/sorter.py

@@ -12,7 +12,10 @@
 # language governing permissions and limitations under the License.
 """Sorting tiebreaker functions for image tables."""
 
+import re
+
 from constants import GLOBAL_CONFIG
+from utils import parse_version
 
 
 def repository_sorter(img) -> int:
@@ -32,3 +35,16 @@ def platform_sorter(img) -> int:
 def accelerator_sorter(img) -> int:
     """Accelerator order: GPU before NeuronX before CPU."""
     return {"gpu": 0, "neuronx": 1, "cpu": 2}.get(img.get("accelerator", "").lower(), 3)
+
+
+def engine_sorter(img) -> tuple:
+    """Engine order for DJL: LMI before TensorRT-LLM before None, then by version desc."""
+    engine = img.get("engine", "")
+    version_match = re.search(r"(\d+(?:\.\d+)*)", engine)
+    version = parse_version(version_match.group(1)) if version_match else parse_version("0")
+    engine_lower = engine.lower()
+    if "lmi" in engine_lower:
+        return (0, -version.major, -version.minor, -version.micro)
+    if "tensorrt" in engine_lower:
+        return (1, -version.major, -version.minor, -version.micro)
+    return (2, 0, 0, 0)

docs/src/tables/djl-inference.yml

@@ -12,3 +12,8 @@ columns:
     header: "Platform"
   - field: example_url
     header: "Example URL"
+
+tiebreakers:
+  - platform_sorter
+  - accelerator_sorter
+  - engine_sorter