[WIP]feat: add API level information to distro docs

Signed-off-by: Nathan Weinberg <nweinber@redhat.com>

Author: nathan-weinberg

.pre-commit-config.yaml

@@ -67,3 +67,4 @@ repos:
         files: ^distribution/run\.yaml$
         additional_dependencies:
           - pyyaml==6.0.2
+          - git+https://github.com/opendatahub-io/llama-stack.git@v0.3.0rc3+rhai0

scripts/gen_distro_docs.py

@@ -4,6 +4,7 @@
 import re
 import subprocess
 from pathlib import Path
+from llama_stack.core.library_client import LlamaStackAsLibraryClient
 
 
 REPO_ROOT = Path(__file__).parent.parent
@@ -32,6 +33,12 @@ def resolve_main_commit_hash(repo_url):
         print(f"Warning: Could not resolve commit hash from main: {e}")
         return None
 
+def extract_llama_stack_routes(distro_path: str):
+    """Extract Llama Stack route information"""
+    client = LlamaStackAsLibraryClient(str(distro_path))
+    routes = client.routes.list()
+    return routes
+
 
 def extract_llama_stack_version():
     """Extract Llama Stack version and repo owner from the Containerfile.
@@ -130,11 +137,11 @@ def load_external_providers_info():
         exit(1)
 
 
-def gen_distro_table(providers_data):
+def gen_distro_table(providers_data, routes):
     # Start with table header
     table_lines = [
-        "| API | Provider | External? | Enabled by default? | How to enable |",
-        "|-----|----------|-----------|---------------------|---------------|",
+        "| Provider API | Provider | REST APIs | External? | Enabled by default? | How to enable |",
+        "|--------------|----------|-----------|-----------|---------------------|---------------|",
     ]
 
     # Load external provider information from build.yaml
@@ -151,6 +158,17 @@ def gen_distro_table(providers_data):
                     provider_type = provider["provider_type"]
                     provider_id = provider.get("provider_id", "")
 
+                    rest_api_pairs = []
+                    for route in routes:
+                        if (provider_type != "inline::meta-reference") and \
+                           (provider_type in route.provider_types) and \
+                           (len(route.route.split("/")) < 4 ) and \
+                           (route.route not in rest_api_pairs):
+                            rest_api_pairs.append(
+                                route.route
+                            )
+                    rest_apis = "<br>".join(rest_api_pairs)
+
                     # Check if provider_id contains the conditional syntax ${<something>:+<something>}
                     # This regex matches the pattern ${...} containing :+
                     conditional_match = re.search(
@@ -176,6 +194,7 @@ def gen_distro_table(providers_data):
                         (
                             api_name,
                             provider_type,
+                            rest_apis,
                             external_status,
                             enabled_by_default,
                             how_to_enable,
@@ -189,12 +208,13 @@ def gen_distro_table(providers_data):
     for (
         api_name,
         provider_type,
+        rest_apis,
         external_status,
         enabled_by_default,
         how_to_enable,
     ) in api_provider_pairs:
         table_lines.append(
-            f"| {api_name} | {provider_type} | {external_status} | {enabled_by_default} | {how_to_enable} |"
+            f"| {api_name} | {provider_type} | {rest_apis} | {external_status} | {enabled_by_default} | {how_to_enable} |"
         )
 
     return "\n".join(table_lines)
@@ -253,13 +273,15 @@ def gen_distro_docs():
 
         # Extract providers section
         providers = run_yaml_data.get("providers", {})
-
         if not providers:
             print("Error: No providers found in run.yaml")
             return 1
 
+        # Extract routes info
+        routes = extract_llama_stack_routes(str(run_yaml_path))
+
         # Generate the Markdown table
-        table_content = gen_distro_table(providers)
+        table_content = gen_distro_table(providers, routes)
 
         # Write to README.md
         with open(readme_path, "w") as readme_file: