2 Commits

  1. 8
      run.py
  2. 3
      spiders/ai_seo/kimi.py

8
run.py

@ -160,6 +160,14 @@ async def ai_seo_job(browser, platform_ids, time_range, job_id, type_name, run_i
# data.update(rest.get("data"))
# search_results.append(data)
# answer_data['search_result'] = search_results
if len(answer_data.get('answer', '') <= 20):
await AiSeoApis.update_task_status(task_id, 4)
logger.error(f"爬虫结果长度过短: id: {task_data['id']} platform_id: {task_data['platform_id']}")
logger.error(f"回滚任务状态: id: {task_id}")
logger.info(f"{answer_data.get('answer', '')}")
return
result = await AiSeoApis.submit_task(answer_data)
logger.info(f"任务提交成功: id: {task_data['id']}")
spider_pool.pop(run_id, None)

3
spiders/ai_seo/kimi.py

@ -47,6 +47,9 @@ class KimiSpider(AbstractAiSeoSpider):
confirm_btn = self.browser_page.locator('//button[text()="知道了"]')
if await confirm_btn.is_visible():
await confirm_btn.click()
confirm_btn = self.browser_page.locator('//button[text()="好的"]')
if await confirm_btn.is_visible():
await confirm_btn.click()
# if self.think:
# think_btn = self.browser_page.locator("span:text('长思考 (k1.5)')").locator('..')
# if await think_btn.is_visible():

Loading…
Cancel
Save