自动问答之白嫖通义千问大模型

声明

文章只用于学习探讨，请勿用于非法行为，由于服务版本随时可能更新，脚本可能失效！。

实现原理

利用selenium操作通义千问对话页面，实现自动化问答和获取问答结果。

实现细节

'''
通义千问客户端
'''
import time

from selenium.webdriver import ActionChains, Keys

import config

from selenium.webdriver.chrome.service import Service
from selenium.webdriver.chrome.options import Options
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

from log_factory import get_logger

# 获取日志记录器
logger = get_logger('tongyi_log')


class TONG_YI_AI:

    def __init__(self, driver=None):
        self.driver = driver
        if self.driver is None:
            self.init_driver()

    def init_driver(self):
        '''
        初始化driver
        :return:
        '''
        selenium_config = config.Selenium_config()

        # 设置 ChromeDriver 的服务
        service = Service(selenium_config.chrome_driver_path)

        # 配置Chrome选项
        options = webdriver.ChromeOptions()
        options.add_argument("--disable-blink-features=AutomationControlled")
        # options.add_argument("--headless")  # 无头模式
        options.add_argument(
            "--user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36")
        options.add_argument(f'--user-data-dir={selenium_config.user_data_dir}')
        # 初始化WebDriver
        driver = webdriver.Chrome(service=service, options=options)
        driver.execute_cdp_cmd("Page.addScriptToEvaluateOnNewDocument", {
            "source": """
                Object.defineProperty(navigator, 'webdriver', {
                    get: () => undefined
                })
            """
        })
        self.driver = driver

    def open_index_page(self):
        '''
        打开文心一言首页
        :return:
        '''
        self.driver.get('https://tongyi.aliyun.com/qianwen/?XXXXXXXXXXX')
        # 等待页面加载完毕
        WebDriverWait(self.driver, 60).until(lambda d: d.execute_script('return document.readyState') == 'complete')
        logger.info(f'首页加载完成.... : {self.driver.title}')

    def answer_is_finish(self):
        try:
            # 等待包含"重新生成"文本的元素出现
            element = WebDriverWait(self.driver, 300).until(
                EC.presence_of_element_located((By.CLASS_NAME, "tools--IJfoLgka"))
            )
            return True
        except:
            return False

    def talk_to_ai(self, question):

        question = question.replace('\n', '')
        logger.info(f'通义千问问答，输入:{question}')
        editable_div = self.driver.find_element(By.CLASS_NAME, 'chatTextarea--tMt0p9_a textarea')
        for char in question:
            editable_div.send_keys(char)
        #editable_div.send_keys(question)

        # 模拟按下Enter键,防止按钮不生效
        editable_div.send_keys(Keys.RETURN)  # 或者 Keys.ENTER

        # 输入完之后，需要停顿一下
        time.sleep(3)

        if self.answer_is_finish():
            elements = self.driver.find_elements(By.XPATH, "//div[@class='answerItem--U4_Uv3iw longTextAnswer--geBuxJAI']")
            text = elements[-1].text
            logger.info(f'通义千问回答:{text}')
            return text


if __name__ == "__main__":
    ai = TONG_YI_AI(driver=None)
    try:
        ai.open_index_page()
        answer = ai.talk_to_ai('你好')
        logger.info(answer )

        time.sleep(200)
    finally:
        ai.driver.quit()

总结

以上就是代码的实现细节，对于目前市面上的其他大模型都可以采用这种思路。在使用过程中，缺少安装包就先pip install 安装一下。仅供交流使用…