diff --git a/spiders/ai_seo/kimi.py b/spiders/ai_seo/kimi.py index 1131120..7547fef 100644 --- a/spiders/ai_seo/kimi.py +++ b/spiders/ai_seo/kimi.py @@ -9,6 +9,8 @@ from domain.ai_seo import AiAnswer, AiSearchResult from utils import create_logger from glom import glom, Coalesce +from utils.ai_seo_api_utils import AiSeoApis + logger = create_logger(__name__) @@ -52,6 +54,7 @@ class KimiSpider(AbstractAiSeoSpider): # 报错检查 if self.fail_status: + await AiSeoApis.update_spider_session(self.session_info['id'], 2) raise self.fail_exception # 关闭侧边栏 @@ -85,6 +88,12 @@ class KimiSpider(AbstractAiSeoSpider): if json_data['items']: logger.debug(json_data) detail = json_data['items'][-1] + if 'error' in detail: + logger.error(f"kimi回复错误: {detail['error']['detail']}") + self.fail_status = True + self.fail_exception = Exception(detail['error']['detail']) + self.completed_event.set() + return content = detail['content'] if self.think: self.ai_answer.search_result = self.get_search_list_enable_think(detail) diff --git a/spiders/ai_seo/metaso.py b/spiders/ai_seo/metaso.py index 28cd9a5..e68964e 100644 --- a/spiders/ai_seo/metaso.py +++ b/spiders/ai_seo/metaso.py @@ -55,7 +55,7 @@ class MetasoSpider(AbstractAiSeoSpider): # await self.completed_event.wait() # 等待指定元素 #//*[@id="search-content-container-8626530479804592128"]/div[2]/button - copy_button = await self.browser_page.wait_for_selector('//*[starts-with(@id, "search-content-container-")]/div[2]/button', timeout=600000) + copy_button = await self.browser_page.wait_for_selector('//*[starts-with(@id, "search-content-container-")]/div[2]/div[3]/button', timeout=600000) # 点击复制按钮 await copy_button.click() # 读取剪贴板