1

ai_pyppeteer.py

@@ -215,8 +215,8 @@ class ToWork:
                 'task_id': task['task_id'],
                 'prompt_id': task['prompt_id'],
             }
-            res = req.post(url, data=params).json()
-            logger.info(res.text)
+            res = req.post(url, data=params)
+            logger.info(res)
         except Exception as e:
             logger.error(e)
 

gpt.py

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