/revERNIEBot

文心一言(ERNIE Bot) Python API

Primary LanguagePythonThe UnlicenseUnlicense

revERNIEBot

文心一言 API

  • Selenium方法
  • requests调用协议方法

目前仅支持使用Selenium的方案

安装

安装Chrome

Windows系统

自行前往 https://www.google.cn/chrome/index.html 下载安装

Ubuntu/Debian系统

安装依赖软件

sudo apt install udev fonts-liberation libu2f-udev libvulkan1 xdg-utils -y

下载Chrome安装包

wget https://dl.google.com/linux/direct/google-chrome-stable_current_amd64.deb

安装Chrome

sudo dpkg -i google-chrome-stable_current_amd64.deb

检查Chrome版本

google-chrome -version

记住此版本号

CentOS/Rocky Linux系统

安装Chrome

yum install https://dl.google.com/linux/direct/google-chrome-stable_current_x86_64.rpm

下载ChromeDriver

点击此处展开

前往 https://registry.npmmirror.com/binary.html?path=chromedriver/

找到对应你Chrome版本号的目录,根据系统下载其中的文件,解压到任意目录

下载browsermob-proxy

点击此处展开

前往 https://github.com/lightbody/browsermob-proxy/releases/tag/browsermob-proxy-2.1.4

下载 browsermob-proxy-2.1.4-bin.zip,解压

使用

  1. 安装本依赖库
pip3 install --upgrade revERNIEBot
  1. 安装 Chrome/EdgeFirefox 上的Cookies Editor插件

  2. 前往yiyan.baidu.com

  3. 打开此插件,点击Export->Export as JSON,将复制的Cookies内容保存到文件cookies.json

  4. 代码调用

from revERNIEBot.selenium import ChatBot  # 目前仅支持Selenium的方案

chatbot = ChatBot(
    cookieFile="cookies.json"
    browsermobProxyPath="pathToBrowserMobProxyExecFile",  # browsermob-proxy的可执行文件路径,详见下方说明
    chromeDriverPath="pathToChromeDriver"  # ChromeDriver的路径,详见下方说明
)

print(chatbot.ask("你好"))  # 获取回复
print(chatbot.ask("你是谁?"))

chatbot.reset_session()  # 开启新对话

chatbot.quit()
  • browsermobProxyPath:填写刚刚下载的browsermobproxy的压缩包解压出来的bin/中的可执行文件路径,例如Linux系统下填写bin中的browsermob-proxy的路径,Windows填写browsermob-proxy.bat路径(Windows系统下需要将路径使用\分隔)
  • 系统上的java版本不能高于java8
  • chromeDriverPath:填写刚刚下载解压的ChromeDriver的可执行文件路径

注意

  • 有问题请在issue中发起讨论
  • 此仓库仅供学习