update

utils/google_ai_studio.py

@@ -11,11 +11,11 @@ from google.generativeai.types import (HarmBlockThreshold, HarmCategory)
 from loguru import logger
 
 
-
-CACHE_DIR = '/Users/z/Downloads/'
-PROXY_ADDR = 'http://localhost:1081'
-os.environ['http_proxy'] = PROXY_ADDR
-os.environ['https_proxy'] = PROXY_ADDR
+CACHE_DIR = '/app/cache/'
+# CACHE_DIR = '/Users/z/Downloads/'
+# PROXY_ADDR = 'http://localhost:1081'
+# os.environ['http_proxy'] = PROXY_ADDR
+# os.environ['https_proxy'] = PROXY_ADDR
 
 class GoogleAI(object):
     @classmethod

workers/consumption_work.py

@@ -20,7 +20,7 @@ class ConsumptionRecommend(object):
         logger.info(f"[处理] 开始获取redis数据")
         task = await RedisHelper().get_client().rpop(name = 'gong_ji_heng_ceng:scan_tasks')
         if not task:
-            logger.info('[ 改造 ] 无待执行的扫描任务')
+            logger.info('[处理] 无待执行的扫描任务')
             return
         task = orjson.loads(task)
         logger.info(f"[处理] 获取redis数据{task}")
@@ -92,6 +92,6 @@ async def run():
 
 
 if __name__ == '__main__':
-    asyncio.run(ConsumptionRecommend.run())
-    # loop = asyncio.get_event_loop()
-    # loop.run_until_complete(run())
+    # asyncio.run(ConsumptionRecommend.run())
+    loop = asyncio.get_event_loop()
+    loop.run_until_complete(run())

workers/select_work.py

@@ -25,6 +25,7 @@ class StartGetRecommend(object):
             current_time = datetime.datetime.now()
             formatted_time = current_time.strftime("%Y-%m-%d %H:%M:%S")
             for task in tasks:
+                time.sleep(0.5)
                 task = orjson.loads(task)
                 values = [[task['partition'], task['video_id'], task['channel'], task['time'], task["type"],formatted_time]]
                 Feishu.insert_columns("JY4esfYvShLbTkthHEqcTw5qnsh", "f53916", "ROWS", 1, 2)
@@ -49,6 +50,6 @@ async def run():
 
 
 if __name__ == '__main__':
-    asyncio.run(StartGetRecommend.run())
-    # loop = asyncio.get_event_loop()
-    # loop.run_until_complete(run())
+    # asyncio.run(StartGetRecommend.run())
+    loop = asyncio.get_event_loop()
+    loop.run_until_complete(run())