From 8c35a38c47d139319eae064bf5a144e075846fd6 Mon Sep 17 00:00:00 2001 From: carry Date: Tue, 15 Apr 2025 21:31:50 +0800 Subject: [PATCH] =?UTF-8?q?feat(frontend):=20=E6=9B=B4=E6=96=B0=E6=A8=A1?= =?UTF-8?q?=E6=9D=BF=E9=80=89=E6=8B=A9=E5=8A=9F=E8=83=BD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 在模板选择变更时，获取所选模板的详细信息 - 创建 PromptTemplate 对象并获取输入变量列表 - 此更新为后续的模板编辑功能做准备 --- frontend/dataset_generate_page.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/frontend/dataset_generate_page.py b/frontend/dataset_generate_page.py index 71f751e..a264b3c 100644 --- a/frontend/dataset_generate_page.py +++ b/frontend/dataset_generate_page.py @@ -1,6 +1,7 @@ import gradio as gr import sys from pathlib import Path +from langchain.prompts import PromptTemplate sys.path.append(str(Path(__file__).resolve().parent.parent)) from global_var import get_docs, get_prompt_store @@ -45,6 +46,11 @@ def dataset_generate_page(): def on_prompt_change(selected_prompt): print(f"模板选择已更改为: {selected_prompt}") + selected_prompt_id = int(selected_prompt.split(" ")[0]) + selected_prompt = get_prompt_store().get(doc_id=selected_prompt_id) + selected_prompt = selected_prompt["content"] + selected_prompt = PromptTemplate.from_template(selected_prompt) + input_variables_list = selected_prompt.input_variables return selected_prompt doc_dropdown.change(on_doc_change, inputs=doc_dropdown, outputs=doc_state)