|
|
@@ -101,8 +101,8 @@ def apply_sharegpt_template(item: dict) -> dict:
|
|
|
|
|
|
|
|
|
def apply_raw_template(item: dict) -> dict:
|
|
|
- """Raw 模板: 直接读取 prompt/text 和 completion/output 字段。"""
|
|
|
- prompt = item.get("prompt", item.get("text", item.get("input", item.get("question", item.get("query", "")))))
|
|
|
+ """Raw 模板: 直接读取 prompt/instruction/text 和 completion/output 字段。"""
|
|
|
+ prompt = item.get("prompt", item.get("instruction", item.get("text", item.get("input", item.get("question", item.get("query", ""))))))
|
|
|
completion = item.get("completion", item.get("output", item.get("target", item.get("answer", item.get("response", "")))))
|
|
|
return {"prompt": str(prompt), "completion": str(completion)}
|
|
|
|