Commit e1198cb2 by lei

1

parent e2a0445a
...@@ -82,11 +82,11 @@ async def get_gpt_message(task): ...@@ -82,11 +82,11 @@ async def get_gpt_message(task):
) )
# 拆分指令 # 拆分指令
res_prompt = split_prompt(response, task) res_prompt = split_prompt(response, task)
logger.info(res_prompt) # logger.info(res_prompt)
logger.info(len(res_prompt)) # logger.info(len(res_prompt))
for item in res_prompt: # for item in res_prompt:
logger.info(item) # logger.info(item)
return # return
# 回调 # 回调
gpt_callback(res_prompt, task) gpt_callback(res_prompt, task)
logger.info('open_ai_结束') logger.info('open_ai_结束')
...@@ -104,19 +104,15 @@ if __name__ == '__main__': ...@@ -104,19 +104,15 @@ if __name__ == '__main__':
task_num = RedisCon.get_task_num(key) task_num = RedisCon.get_task_num(key)
print(task_num, '个任务') print(task_num, '个任务')
nums = 10 nums = 10
if task_num and task_num > 0 or 1: if task_num and task_num > 0:
if task_num < nums: if task_num < nums:
process = task_num process = task_num
else: else:
# 最多同时10个任务 # 最多同时10个任务
process = nums process = nums
loop = asyncio.get_event_loop() loop = asyncio.get_event_loop()
for i in range(1): for i in range(process):
task = RedisCon.get_task(key) task = RedisCon.get_task(key)
task = json.dumps({'type': 2,
'prompt': 'illustration, futurism,NFT art, Fluorescent color, Laser color, A cute girlwho looks up and smiles, combinationof 2d and 3d, solid color background,cel shading,Tindal effect, Non- photorealistic rendering,Transparency,color Slant, animation, blender, geometry art, acrylic painting',
'prompt_img': None, 'user_id': 1, 'prompt_num': 5,
'callback': 'http://test.phpgpt.com/api/users/gpt/callback', 'task_id': 1})
if task: if task:
# 存在任务--转对象 # 存在任务--转对象
task = json.loads(task) task = json.loads(task)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment