ci(docs): add automatic doc generation to pre-commit

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

Author: nathan-weinberg

.pre-commit-config.yaml

@@ -57,3 +57,14 @@ repos:
         files: ^distribution/.*$
         additional_dependencies:
           - llama-stack==0.2.21
+
+      - id: doc-gen
+        name: Distribution Documentation
+        entry: ./scripts/gen_distro_docs.py
+        language: python
+        pass_filenames: false
+        require_serial: true
+        always_run: true
+        files: ^distribution/.*$
+        additional_dependencies:
+          - pyyaml==6.0.2

README.md

@@ -1,14 +1,20 @@
-# Red Hat Distribution Build Instructions
+# ODH Llama Stack Distribution
 
-This directory contains the necessary files to build a Red Hat compatible container image for the llama-stack.
+This directory contains the necessary files to build an Open Data Hub-compatible container image for [Llama Stack](https://github.com/llamastack/llama-stack).
 
-## Prerequisites
+To learn more about the distribution image content, see the [distribution directory](distribution).
+
+To learn about how to modify/build the image yourself, see below.
+
+## Build Instructions
+
+### Prerequisites
 
 - Python >=3.11
 - `llama` CLI tool installed: `pip install llama-stack`
 - Podman or Docker installed
 
-## Generating the Containerfile
+### Generating the Containerfile
 
 The Containerfile is auto-generated from a template. To generate it:
 
@@ -23,26 +29,26 @@ This will:
 - Generate dependencies using `llama stack build`
 - Create a new `Containerfile` with the required dependencies
 
-## Editing the Containerfile
+### Editing the Containerfile
 
 The Containerfile is auto-generated from a template. To edit it, you can modify the template in `distribution/Containerfile.in` and run the build script again.
 NEVER edit the generated `Containerfile` manually.
 
-## Building the Container Image
+### Building the Container Image
 
 Once the Containerfile is generated, you can build the image using either Podman or Docker:
 
-### Using Podman build image for x86_64
+#### Using Podman build image for x86_64
 
 ```bash
 podman build --platform linux/amd64 -f distribution/Containerfile -t rh .
 ```
 
-## Notes
+### Notes
 
 - The generated Containerfile should not be modified manually as it will be overwritten the next time you run the build script
 
-## Push the image to a registry
+### Push the image to a registry
 
 ```bash
 podman push <build-ID> quay.io/opendatahub/llama-stack:rh-distribution

distribution/README.md

@@ -0,0 +1,28 @@
+# Open Data Hub Llama Stack Distribution Image
+
+This image contains the official Open Data Hub Llama Stack distribution, with all the packages and configuration needed to run a Llama Stack server in a containerized environment.
+
+You can see an overview of the APIs and Providers the image ships with in the table below.
+
+| API | Provider |
+|-----|----------|
+| inference | remote::vllm |
+| inference | remote::bedrock |
+| inference | inline::sentence-transformers |
+| inference | remote::watsonx |
+| vector_io | inline::milvus |
+| vector_io | remote::milvus |
+| safety | remote::trustyai_fms |
+| agents | inline::meta-reference |
+| eval | remote::trustyai_lmeval |
+| datasetio | remote::huggingface |
+| datasetio | inline::localfs |
+| scoring | inline::basic |
+| scoring | inline::llm-as-judge |
+| scoring | inline::braintrust |
+| telemetry | inline::meta-reference |
+| tool_runtime | remote::brave-search |
+| tool_runtime | remote::tavily-search |
+| tool_runtime | inline::rag-runtime |
+| tool_runtime | remote::model-context-protocol |
+| files | inline::localfs |

scripts/gen_distro_docs.py

@@ -0,0 +1,72 @@
+#!/usr/bin/env python3
+
+import yaml
+from pathlib import Path
+
+
+REPO_ROOT = Path(__file__).parent.parent
+
+
+def gen_distro_table(providers_data):
+    # Start with table header
+    table_lines = ["| API | Provider |", "|-----|----------|"]
+
+    # Iterate through each API type and its providers
+    for api_name, provider_list in providers_data.items():
+        if isinstance(provider_list, list):
+            for provider in provider_list:
+                if isinstance(provider, dict) and "provider_type" in provider:
+                    provider_type = provider["provider_type"]
+                    table_lines.append(f"| {api_name} | {provider_type} |")
+
+    return "\n".join(table_lines)
+
+
+def gen_distro_docs():
+    # define distro run.yaml and README.md paths
+    run_yaml_path = REPO_ROOT / "distribution" / "run.yaml"
+    readme_path = REPO_ROOT / "distribution" / "README.md"
+
+    # Check if run.yaml exists
+    if not run_yaml_path.exists():
+        print(f"Error: {run_yaml_path} not found")
+        return 1
+
+    # header section
+    header = """# Open Data Hub Llama Stack Distribution Image
+
+This image contains the official Open Data Hub Llama Stack distribution, with all the packages and configuration needed to run a Llama Stack server in a containerized environment.
+
+You can see an overview of the APIs and Providers the image ships with in the table below.
+
+"""
+
+    try:
+        # Load the run.yaml data
+        with open(run_yaml_path, "r") as file:
+            run_yaml_data = yaml.safe_load(file)
+
+        # Extract providers section
+        providers = run_yaml_data.get("providers", {})
+
+        if not providers:
+            print("Error: No providers found in run.yaml")
+            return 1
+
+        # Generate the Markdown table
+        table_content = gen_distro_table(providers)
+
+        # Write to README.md
+        with open(readme_path, "w") as readme_file:
+            readme_file.write(header + table_content + "\n")
+
+        print(f"Successfully generated {readme_path}")
+        return 0
+
+    except Exception as e:
+        print(f"Error: {e}")
+        return 1
+
+
+if __name__ == "__main__":
+    exit(gen_distro_docs())