Update outputs in docstrings

argilla-io · Jul 26, 2024 · 5d445c7 · 5d445c7
1 parent 5657ffd
commit 5d445c7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/distilabel/steps/tasks/magpie/base.py b/src/distilabel/steps/tasks/magpie/base.py
@@ -301,7 +301,8 @@ class Magpie(Task, MagpieBase):
     Output columns:
         - conversation (`ChatType`): the generated conversation which is a list of chat
             items with a role and a message. Only if `only_instruction=False`.
-        - instruction (`str`): the generated instructions if `only_instruction=True`.
+        - instruction (`str`): the generated instructions if `only_instruction=True` or `n_turns==1`.
+        - response (`str`): the generated response if `n_turns==1`.
         - model_name (`str`): The model name used to generate the `conversation` or `instruction`.
 
     Categories:

diff --git a/src/distilabel/steps/tasks/magpie/generator.py b/src/distilabel/steps/tasks/magpie/generator.py
@@ -78,6 +78,7 @@ class MagpieGenerator(GeneratorTask, MagpieBase):
         - conversation (`ChatType`): the generated conversation which is a list of chat
             items with a role and a message.
         - instruction (`str`): the generated instructions if `only_instruction=True`.
+        - response (`str`): the generated response if `n_turns==1`.
         - model_name (`str`): The model name used to generate the `conversation` or `instruction`.
 
     Categories: