86 lines
3.5 KiB
Python
86 lines
3.5 KiB
Python
|
"""
|
|||
|
|
探测元宝页面的生成状态特征
|
|||
|
|
运行后，请在元宝页面发送一条消息，观察输出
|
|||
|
|
"""
|
|||
|
|
import asyncio
|
|||
|
|
from playwright.async_api import async_playwright
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
YUANBAO_URL = "https://yuanbao.tencent.com/chat"
|
|||
|
|
|
|||
|
|
async def detect():
|
|||
|
|
async with async_playwright() as p:
|
|||
|
|
user_data = str(Path.home() / ".yuanbao_chat_profile")
|
|||
|
|
context = await p.chromium.launch_persistent_context(
|
|||
|
|
user_data, channel="chrome", headless=False,
|
|||
|
|
args=["--start-maximized"], no_viewport=True
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
page = context.pages[0] if context.pages else await context.new_page()
|
|||
|
|
await page.goto(YUANBAO_URL)
|
|||
|
|
|
|||
|
|
print("=" * 60)
|
|||
|
|
print("元宝生成状态探测器")
|
|||
|
|
print("=" * 60)
|
|||
|
|
print("\n1. 请在元宝页面登录并创建新对话")
|
|||
|
|
print("2. 在元宝页面发送一条消息（比如'你好'）")
|
|||
|
|
print("3. 观察生成过程中和生成完成后的输出差异")
|
|||
|
|
print("4. 按 Ctrl+C 退出")
|
|||
|
|
print("=" * 60)
|
|||
|
|
|
|||
|
|
input("\n准备好后按回车开始探测...")
|
|||
|
|
|
|||
|
|
# 持续探测
|
|||
|
|
prev_state = None
|
|||
|
|
counter = 0
|
|||
|
|
|
|||
|
|
while True:
|
|||
|
|
try:
|
|||
|
|
# 获取最后一条AI消息的完整信息
|
|||
|
|
result = await page.evaluate("""
|
|||
|
|
() => {
|
|||
|
|
const aiMessages = document.querySelectorAll(
|
|||
|
|
'.agent-chat__list__item--ai, [class*="agent-chat"][class*="ai"], [class*="message"][class*="ai"]'
|
|||
|
|
);
|
|||
|
|
if (aiMessages.length === 0) return null;
|
|||
|
|
|
|||
|
|
const lastMsg = aiMessages[aiMessages.length - 1];
|
|||
|
|
const textEl = lastMsg.querySelector('.agent-chat__speech-text, [class*="text"], [class*="content"]');
|
|||
|
|
|
|||
|
|
return {
|
|||
|
|
className: lastMsg.className,
|
|||
|
|
innerHTML: lastMsg.innerHTML.substring(0, 500),
|
|||
|
|
textContent: textEl ? textEl.innerText.substring(0, 200) : '',
|
|||
|
|
hasCursor: lastMsg.innerHTML.includes('cursor') || lastMsg.innerHTML.includes('blink'),
|
|||
|
|
childElements: Array.from(lastMsg.children).map(c => c.tagName + '.' + c.className).slice(0, 5)
|
|||
|
|
};
|
|||
|
|
}
|
|||
|
|
""")
|
|||
|
|
|
|||
|
|
if result:
|
|||
|
|
state_hash = f"{result['className']}_{result['hasCursor']}"
|
|||
|
|
|
|||
|
|
if state_hash != prev_state:
|
|||
|
|
prev_state = state_hash
|
|||
|
|
counter += 1
|
|||
|
|
print(f"\n--- 探测 #{counter} ---")
|
|||
|
|
print(f"Class: {result['className']}")
|
|||
|
|
print(f"Has Cursor: {result['hasCursor']}")
|
|||
|
|
print(f"Child Elements: {result['childElements']}")
|
|||
|
|
print(f"Text Preview: {result['textContent'][:100]}...")
|
|||
|
|
print(f"HTML Snippet: {result['innerHTML'][:200]}...")
|
|||
|
|
|
|||
|
|
await asyncio.sleep(0.5)
|
|||
|
|
|
|||
|
|
except KeyboardInterrupt:
|
|||
|
|
break
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"Error: {e}")
|
|||
|
|
await asyncio.sleep(1)
|
|||
|
|
|
|||
|
|
await context.close()
|
|||
|
|
print("\n探测结束")
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
asyncio.run(detect())
|