Browse Source

feat:修复纳米盒禁用智能体问题

- 修改标题:移除标题中的"(禁用智能体)"字样
- 调整输入框定位:使用更精确的XPath定位textarea元素
-保留其他功能不变
master
zzx 1 month ago
parent
commit
f3a7bd9539
  1. 10
      spiders/ai_seo/nanometer.py

10
spiders/ai_seo/nanometer.py

@ -35,8 +35,7 @@ class NanometerSpider(AbstractAiSeoSpider):
# 开始操作 # 开始操作
await self.browser_page.goto(self.get_home_url(), timeout=600000) await self.browser_page.goto(self.get_home_url(), timeout=600000)
#开启深度思考 #开启深度思考
# await self.browser_page.locator('//*[@id="NMAI_SIDEBAR_MENU"]/div/div[2]').click()
chat_input_element = self.browser_page.locator('//*[@id="NM-ASSISTANT_chat_input"]')
chat_input_element = self.browser_page.locator('//*[@id="NM-ASSISTANT_chat_input"]//textarea')
# 输入提问词 # 输入提问词
await chat_input_element.press_sequentially(self.prompt) await chat_input_element.press_sequentially(self.prompt)
await self.browser_page.keyboard.press('Enter') await self.browser_page.keyboard.press('Enter')
@ -68,6 +67,13 @@ class NanometerSpider(AbstractAiSeoSpider):
div_height = div_box['height'] if div_box else None div_height = div_box['height'] if div_box else None
logger.debug(f'answer_element: {div_height}') logger.debug(f'answer_element: {div_height}')
view_port_height = div_box['height']+ 500 view_port_height = div_box['height']+ 500
# 修改标题
title = iframe.locator("//h1[@id='message-prompt']")
title_text = await title.inner_text()
new_title = title_text.replace('(禁用智能体)', '')
await title.evaluate(f"node => node.innerHTML = '{new_title}'")
# 调整视口大小 # 调整视口大小
await self.browser_page.set_viewport_size({ await self.browser_page.set_viewport_size({
'width': 1920, 'width': 1920,

Loading…
Cancel
Save