unsloth/studio/backend
Roland Tannous 202780c32c feat: Dataset Conversion Advisor — multi-pass LLM for non-conversational datasets
Non-conversational HF datasets (e.g. stanfordnlp/snli) were naively mapped
column→role, producing poor training results. The AI Assist button now runs
a 3-pass advisor using Qwen 7B that:
1. Fetches the HF dataset card/README to understand the dataset purpose
2. Classifies the dataset type and determines if conversion is needed
3. Generates a system prompt, user/assistant templates with {column}
   placeholders, and label mappings (e.g. 0→entailment)
4. Validates the conversion quality (score ≥7/10 required)

Architecture: advisor metadata flows as __-prefixed keys in
custom_format_mapping (e.g. __system_prompt, __user_template,
__assistant_template, __label_mapping). The existing _apply_user_mapping()
detects these keys and routes to template-based conversation construction.
No __ keys = existing simple mode (backwards compatible).

Backend: upgraded llm_assist.py (7B default, multi-pass advisor,
HF card fetching), extended API models, added _apply_template_mapping()
to dataset_utils.py.

Frontend: extended store with advisor state fields, wired AI Assist
to store templates/system prompt, inject __ metadata in training request,
show advisor notification banner in mapping card.
2026-03-10 15:39:56 +00:00
..
assets add trust_remote_code defaults to all model configs 2026-03-09 09:54:52 +00:00
auth Revert "Merge pull request #347 from unslothai/feature/studio-storage-roots" 2026-03-10 01:52:47 +00:00
core fix: download all GGUF shards for split models (e.g. 7B Q8_0) 2026-03-10 15:08:20 +00:00
loggers root studio folder 2026-02-02 09:13:49 +00:00
models feat: Dataset Conversion Advisor — multi-pass LLM for non-conversational datasets 2026-03-10 15:39:56 +00:00
plugins/data-designer-unstructured-seed Revert "Merge pull request #347 from unslothai/feature/studio-storage-roots" 2026-03-10 01:52:47 +00:00
requirements chore(data-recipe): bump data-designer to 0.5.2 and pin duckdb<1.5 2026-03-09 17:27:02 +01:00
routes feat: Dataset Conversion Advisor — multi-pass LLM for non-conversational datasets 2026-03-10 15:39:56 +00:00
state root studio folder 2026-02-02 09:13:49 +00:00
tests Add AGPL-3.0 SPDX headers to all source files 2026-03-09 20:17:45 +00:00
utils feat: Dataset Conversion Advisor — multi-pass LLM for non-conversational datasets 2026-03-10 15:39:56 +00:00
colab.py Add AGPL-3.0 SPDX headers to all source files 2026-03-09 20:17:45 +00:00
main.py feat: add LLM-assisted dataset detection using ephemeral GGUF helper 2026-03-10 09:20:45 +00:00
run.py Add AGPL-3.0 SPDX headers to all source files 2026-03-09 20:17:45 +00:00