Update app.py
Browse files
app.py
CHANGED
@@ -187,68 +187,3 @@ def create_image_ep(content: Text = None):
|
|
187 |
headers=headers
|
188 |
)
|
189 |
return JSONResponse(content=result.json())
|
190 |
-
|
191 |
-
|
192 |
-
|
193 |
-
|
194 |
-
|
195 |
-
|
196 |
-
|
197 |
-
|
198 |
-
|
199 |
-
|
200 |
-
|
201 |
-
|
202 |
-
|
203 |
-
import requests
|
204 |
-
from concurrent.futures import ThreadPoolExecutor, as_completed
|
205 |
-
import openai
|
206 |
-
import json
|
207 |
-
import time
|
208 |
-
key = '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'
|
209 |
-
openai.api_key = key
|
210 |
-
headers = {
|
211 |
-
'Content-Type': 'application/json',
|
212 |
-
'Authorization': 'Bearer ' + key
|
213 |
-
}
|
214 |
-
|
215 |
-
# 初始化线程池
|
216 |
-
pool = ThreadPoolExecutor(max_workers=4)
|
217 |
-
url = 'https://api.openai.com/v1/chat/completions'
|
218 |
-
prompt = '根据下面的文章,生成的“问题和回答”QA对。要求:不少于3个,问题不能重复,,回答不是"未知",以一行一个json格式({“question”:"xxx","answer":"xxx"})生成:\n'
|
219 |
-
|
220 |
-
|
221 |
-
# 每个线程去调api,异常后 retry的次数 和 间隔时间
|
222 |
-
def task(content, retry=3, interval=2):
|
223 |
-
while retry > 0:
|
224 |
-
try:
|
225 |
-
messages = [{"role": "user", "content": prompt + content}]
|
226 |
-
data = {
|
227 |
-
"model": "gpt-3.5-turbo",
|
228 |
-
"messages": messages
|
229 |
-
}
|
230 |
-
print("messages = \n", messages)
|
231 |
-
result = requests.post(url=url,
|
232 |
-
data=json.dumps(data),
|
233 |
-
headers=headers
|
234 |
-
)
|
235 |
-
res = str(result.json()['choices'][0]['message']['content']).strip()
|
236 |
-
if res:
|
237 |
-
return res
|
238 |
-
else:
|
239 |
-
retry -= 1
|
240 |
-
except Exception as e:
|
241 |
-
print(e)
|
242 |
-
retry -= 1
|
243 |
-
time.sleep(interval)
|
244 |
-
|
245 |
-
|
246 |
-
doc_list = ['周深,1992年9月29日出生于湖南省邵阳市,成长于贵州省贵阳市,中国内地男歌手,毕业于乌克兰利沃夫国立谢科音乐科学院美声专业。2014年,参加浙江卫视歌唱选秀节目《中国好声音第三季》,从而正式出道。','2016年,获得东方风云榜最佳新人奖。同年,为动画电影《大鱼海棠》演唱印象曲《大鱼》,并凭借该单曲获得多个音乐奖项。2020年,作为首发歌手加盟《歌手当打之年》,其演唱的《达拉崩吧》以破亿播放量斩获云村播放量第一。','2017年3月14-15日,周深率领动感101爱心支教团奔赴家乡贵州,前往兴义市威舍镇阿依小学支教,助力偏远地区的孩子们实现音乐梦想,不仅为孩子们准备了礼物,还自创教学歌,与孩子们一起参加运动会。']
|
247 |
-
|
248 |
-
# 用线程池去调用api
|
249 |
-
all_task = [pool.submit(task, doc) for doc in doc_list]
|
250 |
-
# 等待所有线程完成
|
251 |
-
for future in as_completed(all_task):
|
252 |
-
print(future.result())
|
253 |
-
|
254 |
-
|
|
|
187 |
headers=headers
|
188 |
)
|
189 |
return JSONResponse(content=result.json())
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|