feat(frontend): 更新模板选择功能
- 在模板选择变更时,获取所选模板的详细信息 - 创建 PromptTemplate 对象并获取输入变量列表 - 此更新为后续的模板编辑功能做准备
This commit is contained in:
parent
7ee751c88f
commit
8c35a38c47
@ -1,6 +1,7 @@
|
|||||||
import gradio as gr
|
import gradio as gr
|
||||||
import sys
|
import sys
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
from langchain.prompts import PromptTemplate
|
||||||
|
|
||||||
sys.path.append(str(Path(__file__).resolve().parent.parent))
|
sys.path.append(str(Path(__file__).resolve().parent.parent))
|
||||||
from global_var import get_docs, get_prompt_store
|
from global_var import get_docs, get_prompt_store
|
||||||
@ -45,6 +46,11 @@ def dataset_generate_page():
|
|||||||
|
|
||||||
def on_prompt_change(selected_prompt):
|
def on_prompt_change(selected_prompt):
|
||||||
print(f"模板选择已更改为: {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
|
return selected_prompt
|
||||||
|
|
||||||
doc_dropdown.change(on_doc_change, inputs=doc_dropdown, outputs=doc_state)
|
doc_dropdown.change(on_doc_change, inputs=doc_dropdown, outputs=doc_state)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user