Add Legend to Component Gallery Icons (#1090)

argilla-io · Jan 14, 2025 · 27b5db2 · 27b5db2
1 parent 8ad4838
commit 27b5db2
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/src/distilabel/utils/mkdocs/components_gallery.py b/src/distilabel/utils/mkdocs/components_gallery.py
@@ -454,7 +454,7 @@ def _generate_embeddings_pages(self, src_dir: Path, embeddings: list) -> List[st
 
             paths.append(llm_path)
 
-        # Create the `components-gallery/llms/index.md` file
+        # Create the `components-gallery/embeddings/index.md` file
         content = _COMPONENTS_LIST_TEMPLATE.render(
             title="Embeddings Gallery",
             description="",

diff --git a/src/distilabel/utils/mkdocs/templates/components-gallery/index.md b/src/distilabel/utils/mkdocs/templates/components-gallery/index.md
@@ -5,6 +5,14 @@ hide:
 ---
 # Components Gallery
 
+??? info "Category Overview" 
+    | Icon                       | Category   | Description                                                       |
+    |----------------------------|------------|-------------------------------------------------------------------|
+    | :material-step-forward:    | Steps      | Steps are used for data manipulation.                            |
+    | :material-check-outline:   | Tasks      | Tasks allow performing data generation, annotation, and more.    |
+    | :material-brain:           | LLMs       | Explore all available Large Language Models integrated with distilabel. |
+    | :material-vector-line:     | Embeddings | Explore all available Embeddings Models integrated with distilabel.    |
+
 <div class="grid cards" markdown>
 
 -   :material-step-forward:{ .lg .middle } __Steps__