diff --git a/spiders/ai_seo/yiyan.py b/spiders/ai_seo/yiyan.py index 344f5bc..bee44e7 100644 --- a/spiders/ai_seo/yiyan.py +++ b/spiders/ai_seo/yiyan.py @@ -35,10 +35,9 @@ class YiYanSpider(AbstractAiSeoSpider): # 检查登录状态 await self.check_login() if self.think: - think_btn = self.browser_page.locator("//span[text()='思考(X1 Turbo)']/parent::div") - clazz = await think_btn.get_attribute('class') - if 'active' not in clazz: - await think_btn.click() + think_btn = self.browser_page.locator("//span[text()='思考·自动']/parent::div") + await think_btn.click() + await self.browser_page.locator("//div[contains(@class, 'dtModeItem__')][2]").click() # 开始操作 chat_input_element = self.browser_page.locator("//div[@class='yc-editor']") await chat_input_element.click()