feat(frontend): 实现了文档、提示和 API 提供商的获取逻辑
This commit is contained in:
parent
e16882953d
commit
314434951d
@ -98,6 +98,11 @@ def dataset_generate_page():
|
||||
return selected_prompt, dataframe_value
|
||||
|
||||
def on_generate_click(doc_state, prompt_state, api_state, variables_dataframe, rounds, progress=gr.Progress()):
|
||||
doc = [i for i in get_docs() if i.name == doc_state][0]
|
||||
prompt = [i for i in get_prompt_store().all() if i["id"] == int(prompt_state.split(" ")[0])][0]
|
||||
with Session(get_sql_engine()) as session:
|
||||
api_provider = session.exec(select(APIProvider).where(APIProvider.id == int(api_state.split(" ")[0]))).first()
|
||||
|
||||
variables_dict = {}
|
||||
# 正确遍历DataFrame的行数据
|
||||
for _, row in variables_dataframe.iterrows():
|
||||
@ -112,9 +117,6 @@ def dataset_generate_page():
|
||||
# 模拟每个步骤的工作负载
|
||||
time.sleep(0.5)
|
||||
|
||||
# 更新进度条
|
||||
# 第一个参数是当前的进度比例 (0.0 到 1.0)
|
||||
# desc 参数可以动态更新进度条旁边的描述文字
|
||||
current_progress = (i + 1) / total_steps
|
||||
progress(current_progress, desc=f"处理步骤 {i + 1}/{total_steps}")
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user