1

gpt.py

@@ -82,11 +82,11 @@ async def get_gpt_message(task):
         )
         # 拆分指令
         res_prompt = split_prompt(response, task)
-        logger.info(res_prompt)
-        logger.info(len(res_prompt))
-        for item in res_prompt:
-            logger.info(item)
-        return
+        # logger.info(res_prompt)
+        # logger.info(len(res_prompt))
+        # for item in res_prompt:
+        #     logger.info(item)
+        # return
         # 回调
         gpt_callback(res_prompt, task)
         logger.info('open_ai_结束')
@@ -104,19 +104,15 @@ if __name__ == '__main__':
         task_num = RedisCon.get_task_num(key)
         print(task_num, '个任务')
         nums = 10
-        if task_num and task_num > 0 or 1:
+        if task_num and task_num > 0:
             if task_num < nums:
                 process = task_num
             else:
                 # 最多同时10个任务
                 process = nums
             loop = asyncio.get_event_loop()
-            for i in range(1):
+            for i in range(process):
                 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:
                     # 存在任务--转对象
                     task = json.loads(task)