windy/streamlit-testGenius
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

format code

Browse Source
This commit is contained in:
VincentXiuyuanZhao 2024-06-07 18:55:36 +08:00
parent 6a4d7d858d
commit 4f754ec092
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
testGrDownload.py
View File

@ -1,16 +1,17 @@
import gradio as gr import gradio as gr
import pandas as pd import pandas as pd
import tempfile import tempfile
import csv
from http import HTTPStatus from http import HTTPStatus
import dashscope import dashscope
from dashscope import Generation from dashscope import Generation
import os import os
from testAny import check_df_english, check_df_tags from testAny import check_df_english, check_df_tags
dashscope.api_key = os.getenv("DASHSCOPE_API_KEY") # Vincent's API key dashscope.api_key = os.getenv("DASHSCOPE_API_KEY") # Vincent's API key
# todo: delete instruction part or make it optional # todo: delete instruction part or make it optional
# todo: add a checkbox to choose whether to use instruction or not
def response(prompt, instruction=None): def response(prompt, instruction=None):
messages = [{'role': 'user', 'content': prompt}] messages = [{'role': 'user', 'content': prompt}]
if instruction is not None: # 如果提供了指令，则添加到messages中 if instruction is not None: # 如果提供了指令，则添加到messages中
@ -38,12 +39,13 @@ def response(prompt, instruction=None):
def format_full_prompt(df, introduction): def format_full_prompt(df, introduction):
# 为每个 row 创建 context # 为每个 row 创建 context，拼接RAG1和2
df['context'] = df.apply(lambda row: f"{row['RAG1']}-{row['RAG2']}", axis=1) df['context'] = df.apply(lambda row: f"{row['RAG1']}-{row['RAG2']}", axis=1)
# 准备用于 format 的字典 # 准备用于 format 的字典
format_dict = df[['business_use_mark', 'context', 'question']].apply(lambda x: dict(zip(x.index, x)), axis=1) column_list = df.drop('full_prompt', axis=1).columns.tolist() # 去除full_prompt列，其他的都为参数
if len(introduction) >= 100: format_dict = df[column_list].apply(lambda x: dict(zip(x.index, x)), axis=1)
if len(introduction) >= 200:
df['full_prompt'] = introduction df['full_prompt'] = introduction
# 使用 apply() 和 lambda 函数格式化 full_prompt 列 # 使用 apply() 和 lambda 函数格式化 full_prompt 列
df['full_prompt'] = df.apply(lambda row: row['full_prompt'].format(**format_dict[row.name]), axis=1) df['full_prompt'] = df.apply(lambda row: row['full_prompt'].format(**format_dict[row.name]), axis=1)