Pretty print

[plaguss]

Description

This PR adds a base method to allow easily printing the prompts of the tasks.

from distilabel.steps.tasks import URIAL

# only importable if placed at the root at the repo, any LLM will work
from tests.unit.conftest import DummyLLM

t = URIAL(llm=DummyLLM())
t.load()
t.print()
...
╭────────────────────────────────────────────────────────────────── Prompt: URIAL  ──────────────────────────────────────────────────────────────────╮
│ ╭───────────────────────────────────────────────────────────────── User Message ─────────────────────────────────────────────────────────────────╮ │
│ │ # Instruction                                                                                                                                  │ │
│ │                                                                                                                                                │ │
│ │ Below is a list of conversations between a human and an AI assistant (you).                                                                    │ │
│ │ Users place their queries under "# User:", and your responses are under  "# Assistant:".                                                       │ │
│ │ You are a helpful, respectful, and honest assistant.                                                                                           │ │
│ │ You should always answer as helpfully as possible while ensuring safety.                                                                       │ │
│ │ Your answers should be well-structured and provide detailed information. They should also have an engaging tone.                               │ │
│ │ Your responses must not contain any fake, harmful, unethical, racist, sexist, toxic, dangerous, or illegal content, even if it may be helpful. │ │
│ │ Your response must be socially responsible, and thus you can refuse to answer some controversial topics.                                       │ │
│ │                                                                                                                                                │ │
│ │                                                                                                                                                │ │
│ │ # User:                                                                                                                                        │ │
│ │                                                                                                                                                │ │
│ │ <PLACEHOLDER_INSTRUCTION>                                                                                                                      │ │
│ │                                                                                                                                                │ │
│ │ # Assistant:                                                                                                                                   │ │
│ ╰────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯ │
╰────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯

Example screenshot:

For new tasks, we should review the prompt, and if not rendering properly, override the _sample_inputs method to return the appropriate content.

Run the following to see all the prompts at once:

pytest tests/integration/test_prints.py -s -vv

[github-actions]

Documentation for this PR has been built. You can view it at: https://distilabel.argilla.io/pr-934/

[codspeed-hq]

CodSpeed Performance Report

Merging #934 will not alter performance

_{Comparing pretty-print (a213728) with develop (2ce44f0)}

Summary

✅ 1 untouched benchmarks