Browse Source

refactor(ai_seo): 优化一言蜘蛛的思考模式选择逻辑

- 修改思考模式按钮的定位方式,使用更准确的文本匹配
- 点击思考模式后,直接选择第二个模式选项,无需检查是否已激活
master
zzx 1 month ago
parent
commit
ac93ca5e3b
  1. 7
      spiders/ai_seo/yiyan.py

7
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()

Loading…
Cancel
Save