Merge branch 'main' into develop

argilla-io · Jan 17, 2025 · dd6eda5 · dd6eda5
2 parents 74cc09e + 27b5db2
commit dd6eda5
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 3 deletions.
diff --git a/docs/sections/how_to_guides/basic/task/generator_task.md b/docs/sections/how_to_guides/basic/task/generator_task.md
@@ -19,7 +19,7 @@ class MyCustomTask(GeneratorTask):
     instruction: str
 
     @override
-    def process(self, offset: int = 0) -> GeneratorOutput:
+    def process(self, offset: int = 0) -> GeneratorStepOutput:
         output = self.llm.generate(
             inputs=[
                 [
@@ -82,7 +82,7 @@ from distilabel.typing import ChatType
 
 class MyCustomTask(GeneratorTask):
     @override
-    def process(self, offset: int = 0) -> GeneratorOutput:
+    def process(self, offset: int = 0) -> GeneratorStepOutput:
         output = self.llm.generate(
             inputs=[
                 [{"role": "user", "content": "Tell me a joke."}],

diff --git a/src/distilabel/utils/mkdocs/components_gallery.py b/src/distilabel/utils/mkdocs/components_gallery.py
@@ -509,7 +509,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__