From 08c734b242ab7c0e6eaacebc5e0bd117902846c0 Mon Sep 17 00:00:00 2001 From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com> Date: Tue, 18 Feb 2025 14:30:18 +0800 Subject: [PATCH] feat: Generate problem support for title variable (#2310) --- apps/dataset/task/generate.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/apps/dataset/task/generate.py b/apps/dataset/task/generate.py index bf9e53869..fdcd171af 100644 --- a/apps/dataset/task/generate.py +++ b/apps/dataset/task/generate.py @@ -29,7 +29,8 @@ def generate_problem_by_paragraph(paragraph, llm_model, prompt): try: ListenerManagement.update_status(QuerySet(Paragraph).filter(id=paragraph.id), TaskType.GENERATE_PROBLEM, State.STARTED) - res = llm_model.invoke([HumanMessage(content=prompt.replace('{data}', paragraph.content))]) + res = llm_model.invoke( + [HumanMessage(content=prompt.replace('{data}', paragraph.content).replace('{title}', paragraph.title))]) if (res.content is None) or (len(res.content) == 0): return problems = res.content.split('\n')