【微信机器人】ChatGPT 的国内替代方案TigerBot

好多群友说，WeChatRobot 的 ChatGPT 用不了。原因林林总总，当然不是代码的问题：

• 没有互联网
• 没法注册账号
• 没法充钱
• ……

今天介绍一个替代品：TigerBot（请给我打广告费，谢谢！）

注册账号

访问 TigerBot^[1]，点击 申请内测，然后需要验证手机号，然后等待审核。我申请的时候，不到 5 分钟，就审核通过了。

创建 key

访问 API_KEY^[2] 创建 API_KEY。

验证

请求示例

import requests
url = "https://api.tigerbot.com/bot-service/ai_service/gpt"

headers = {
  "Authorization": "Bearer " + API_KEY
}
payload = {
  "text": "中国的首都",
  "modelVersion": "tigerbot-7b-sft"
}
response = requests.post(url, headers=headers, json=payload)
print(response.text)

响应示例

{
  "code": 200,
  "msg": "操作成功",
  "data": {
    "result": ["北京"]
  }
}

接入 WeChatRobot

后面会集成进去，这里简单说一下怎么集成。

创建新功能

新建 func_tigerbot.py：

#! /usr/bin/env python3
# -*- coding: utf-8 -*-
import logging
import requests
from random import randint
class TigerBot():
    def __init__(self, tbconf=None) -> None:
        self.loger = logging.getLogger(__file__)
        self.tburl = "https://api.tigerbot.com/bot-service/ai_service/gpt"
        self.tbheaders = {"Authorization": "Bearer " + tbconf["key"]}
        self.tbmodel = tbconf["model"]
        self.fallback = "滚！"  # 请求失败时候的默认回复
   def answer(self, msg: str, sender: str = None) -> str:
        if self.tburl is None:
            return "请配置 Tiger Bot 参数"
        payload = {
            "text": msg,
            "modelVersion": self.tbmodel
        }
        try:
            rsp = requests.post(self.tburl, headers=self.tbheaders, json=payload).json()["data"]["result"][0]
        except Exception as e:
            idx = randint(0, len(self.fallback))
            rsp = self.fallback[idx]
        return rsp
if __name__ == "__main__":
    from configuration import Config
    c = Config()
    tbot = TigerBot(c.TIGERBOT)
    rsp = tbot.answer("你还活着？")
    print(rsp)

添加配置信息

• 在 config.yaml 中添加配置信息：

tigerbot:
  key: 刚才创建的 key
  model: tigerbot-7b-sft

• 在 configuration.py 中添加 TigerBot 配置

    def reload(self) -> None:
        yconfig = self._load_config()
        logging.config.dictConfig(yconfig["logging"])
        self.GROUPS = yconfig["groups"]["enable"]
        self.CHATGPT = yconfig.get("chatgpt")
        self.HTTP = yconfig.get("http")
        self.NEWS = yconfig["news"]["receivers"]
        self.TIGERBOT = yconfig.get("tigerbot")  # 添加这个

修改机器人代码

在 robot.py 中，将

class Robot(Job):
    """个性化自己的机器人
    """
    def __init__(self, config: Config, wcf: Wcf) -> None:
        self.wcf = wcf
        self.config = config
        self.LOG = logging.getLogger("Robot")
        self.wxid = self.wcf.get_self_wxid()
        self.allContacts = self.getAllContacts()
        self.chat = None
        chatgpt = self.config.CHATGPT
        if chatgpt:
            self.chat = ChatGPT(chatgpt.get("key"), chatgpt.get("api"), chatgpt.get("proxy"), chatgpt.get("prompt"))

改成

好了，搞定！