feat(frontend): 更新模板选择功能

- 在模板选择变更时,获取所选模板的详细信息
- 创建 PromptTemplate 对象并获取输入变量列表
- 此更新为后续的模板编辑功能做准备
This commit is contained in:
carry 2025-04-15 21:31:50 +08:00
parent 7ee751c88f
commit 8c35a38c47

View File

@ -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)