diff --git a/studio/backend/utils/datasets/llm_assist.py b/studio/backend/utils/datasets/llm_assist.py index ded4c49a00..b5f8ba6b35 100644 --- a/studio/backend/utils/datasets/llm_assist.py +++ b/studio/backend/utils/datasets/llm_assist.py @@ -538,15 +538,15 @@ def _run_multi_pass_advisor( ), } - # ── Pass 2: Conversion strategy ── - print("🤖 Pass 2: Generating conversion strategy...", flush=True) + # ── Pass 2: Conversion templates ── + print("🤖 Pass 2: Generating conversion templates...", flush=True) t2 = time.monotonic() messages2 = [ { "role": "system", "content": ( "You are a dataset conversion specialist for LLM fine-tuning. " - "You design strategies to convert non-conversational datasets into " + "You design templates to convert non-conversational datasets into " "user/assistant conversation format. Respond with ONLY valid JSON." ), }, @@ -561,9 +561,7 @@ def _run_multi_pass_advisor( SAMPLE DATA: {samples_text} - Design a conversion strategy to turn this into conversation format for fine-tuning. - The strategy should create a user message template and an assistant message template. - Optionally include a system prompt ONLY if the task is ambiguous from the data alone. + Design user and assistant message templates for this dataset. RULES: - Use {{column_name}} placeholders in templates to reference column values. @@ -575,13 +573,9 @@ def _run_multi_pass_advisor( - The assistant template should produce the expected model output. - column_roles: mark columns used in the user template as "user", columns used in the assistant template as "assistant". - - system_prompt: set to null if the user/assistant templates alone - make the task clear. Only provide one when extra context is needed - (e.g. persona, domain expertise, output format constraints). Respond with a JSON object: {{ - "system_prompt": "", "user_template": "