From a059005506afd82fee4c881b46a88f0f69ae1812 Mon Sep 17 00:00:00 2001
From: zhayujie <yjzha1996@163.com>
Date: Sat, 10 Dec 2022 13:15:31 +0800
Subject: [PATCH] feat: chatGPT bot

---
 .../__pycache__/baidu_unit_bot.cpython-36.pyc | Bin 1537 -> 1458 bytes
 bot/baidu/baidu_unit_bot.py                   |   5 ++-
 bot/bot_factory.py                            |   3 ++
 bot/chatgpt/chat_gpt_bot.py                   |  30 ++++++++++++++++++
 bridge/bridge.py                              |   2 +-
 .../__pycache__/wechat_channel.cpython-36.pyc | Bin 1552 -> 1552 bytes
 channel/wechat/wechat_channel.py              |   4 +--
 7 files changed, 37 insertions(+), 7 deletions(-)
 create mode 100644 bot/chatgpt/chat_gpt_bot.py

diff --git a/bot/baidu/__pycache__/baidu_unit_bot.cpython-36.pyc b/bot/baidu/__pycache__/baidu_unit_bot.cpython-36.pyc
index 9a1fff91fc1d8789e0cc130ecc9e5c53a58aaa33..d453432f692a24cfa7981e02c1d3daa8f1ab2141 100644
GIT binary patch
delta 113
zcmZqV*~G1E%*)H=n#qv7lZAodF#{3+@);a}xY%H#vOZf1V=#jz>&5~;RvAs!B0iva
v5jT*CVo%C1(F5WlCXf*G<N#K2$y*#Yx%nxjIjMHcApLAWf`gHRk%t)oI+znc

delta 152
zcmdnQ-N>VC%*)GF{P2AeKPv;nV+JI^0%SV?ak0llWqr9+))dxkrXrVAwp5m8hAj3J
zwr0j=Mn;Ad#$X0b_KgvItZKJ7;^T7@GxOr(OGSbDp6p%ld{)czj*U;3?tM06{gd4d
nPj+@_a!k%(mE)EK$`^5g2(igctZIB9CI=%2BM-BHC<h||`|~D+

diff --git a/bot/baidu/baidu_unit_bot.py b/bot/baidu/baidu_unit_bot.py
index 284eff5..e58b6a4 100644
--- a/bot/baidu/baidu_unit_bot.py
+++ b/bot/baidu/baidu_unit_bot.py
@@ -1,6 +1,5 @@
 # encoding:utf-8
 
-import json
 import requests
 from bot.bot import Bot
 
@@ -17,8 +16,8 @@ class BaiduUnitBot(Bot):
             return response.json()['result']['context']['SYS_PRESUMED_HIST'][1]
 
     def get_token(self):
-        access_key = '${YOUR_ACCESS_KEY}'
-        secret_key = '${YOUR_SECRET_KEY}'
+        access_key = 'wdNwpvzh4fBr8xzTuPePOWhU'
+        secret_key = 'EUtLXyrogfTQV8nnjncnqNEfeyoHDUqS'
         host = 'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=' + access_key + '&client_secret=' + secret_key
         response = requests.get(host)
         if response:
diff --git a/bot/bot_factory.py b/bot/bot_factory.py
index 919215f..eacf996 100644
--- a/bot/bot_factory.py
+++ b/bot/bot_factory.py
@@ -3,6 +3,7 @@ channel factory
 """
 
 from bot.baidu.baidu_unit_bot import BaiduUnitBot
+from bot.chatgpt.chat_gpt_bot import ChatGPTBot
 
 
 def create_bot(bot_type):
@@ -13,4 +14,6 @@ def create_bot(bot_type):
     """
     if bot_type == 'baidu':
         return BaiduUnitBot()
+    elif bot_type == 'chatGPT':
+        return ChatGPTBot()
     raise RuntimeError
\ No newline at end of file
diff --git a/bot/chatgpt/chat_gpt_bot.py b/bot/chatgpt/chat_gpt_bot.py
new file mode 100644
index 0000000..bd5615b
--- /dev/null
+++ b/bot/chatgpt/chat_gpt_bot.py
@@ -0,0 +1,30 @@
+import time
+from bot.bot import Bot
+from revChatGPT.revChatGPT import Chatbot
+
+config = {
+    "Authorization": "<Your Bearer Token Here>",  # This is optional
+    "session_token": "eyJhbGciOiJkaXIiLCJlbmMiOiJBMjU2R0NNIn0.._BniuRSwMqdGn8t2.Hv_C4nsWF1FWnVGxNs19YZOGLDIWFlyC3FeLlTE9HUEgV1M8IFmhz1aH0j40vutVTzoEv3JDSEZN8je4kzewXE5tcKqhQF_Jh0SSHNL52OdRe_tqu_aE61L39NkvxEkmMRP2teXmWtiBIGtrPX3B77IbHIyqx9Kbk17awwl_WJokr39KlWhHE-Mpf9AwPvOW4UPi2csIxVNsVFW6MjxffRL8QZAqhXaeVlAnpjlpfJrK1B9TRoH7wkj0Q4nQ4MMhenvKXQHhOaa2tfTC79BeOBr8d60b2HPGCiOZuYc4vCqehXRl4VA7_JF9EKEmhgBQAPp0J1LFTC-3i0ZVAUX4zud8UEAsAwkjknQx7BHx9_QUOGil5I4U84WWzsGKxNNj1Jhi3z877V6d90CrTQax_yuuNzV9wb7j_RYx5EbO1MKmOPvL1iksa7xm0SYWsK2fYUID2vW2cE0nAavnGyi7g2-Lf83ARyWAmU8puuEweuEbsS_-1-IP6b3rtapOoZWfWmW74Mm6mmdxBIldrKHTDK8rJanjbp5Dho4YdJaTlAo7Mlg6fwpFk5i-2_QPr-s2E141hLCAhbKFQ6OTKfv3yugF1OoOG0IsXy49yuq3ezznDhtArYgZ6mj-DfHaQakuYrw9aHsfHebiVZ6tuwr_lZr73AuwneD0wGc1sk5nTYqFGn-E9ZC3kgng5Sa_CrOg2RhuNPsuDDsDuJ22ANilQprmDR0N8J-2ASlqkyf_af23QWwaEmjj8pcDrg_FWs7-6Qdju4pSFcLbtbAFh9ORi5SD463LxGxHsRPNCL9LfUZFrVFYVfTKk3mGtb7XQ80p7L1O7_ix6UkiastENknqH-cwk6HCW8VKf2ASp28NoEhKCjW0CHfGT4eHQxxzh2wrb2FWmw7zAXd8_c48o1LY6GOgqBWnp4npMYRwA2yki0cuKR5eVrKHlZZU4g5hSQ5F_Pq0qDZsCrKFCUJ19C3ndqG0XZF70jUfJtTMdUgFh4viDVpVS4FvL35VDTygYnTz7DCRHc9uLlsG8e-7wfCHRcyL8U4HL1KTB2JIZnNMOAHaeiTxAc9PeLxhqSHuG7LttdHPWE0OBUZmMbArR9BzjHp4L3mL-oK8xGil9vuw43TWxpPpcbP_EzuZIBgWlTB4LZg19WmrRq_Ll1eLjXKeCEzoxisyicIdQ8q01iPKZOfVQZCISmgG8Ey6qm8DAk84tJ44p7IWXeE5Ho-SIdW1aR9H7U02ycvtBM24aRs30o_xwqfKfDInrAUsGeJAkA4iKkmUtzaHHYlBZVAITG7-FlsFBgDJQStG7w_4d-JqfqltO6YrbZG2A5EYX8RV5SFgNzEluTp_ZY0A65kThVLvVnldWGb0gkBiC8r-UzYKGgQQJwG-WqnyOvQ5oNMqUErqWaqYg8Z1u7nvZlFG_WeqwfF711huA2eMLIyV3gvdhxSVO9j2IfNsxQoMOIeUcXF_d64q95RgM2Cv5bw-kwRSqumejPhGghjDwv01et6zeHnseoEItHkyfKHw443rqBvitMlERQcvIfF07j0Pgm2-7fBiGFAstS-6swFnydmlKev9ypAKE-vFgH_R4s5R0cy_aYQr16rpmX0pRaDfEA-kDrq4u1PFUijuRQiOG0UN3Ih4E7HPZtKvCqQreUbceihgWsh0_u-Ygh6dpN7-ALf4En6e--FyRePScJ1RUHByJZ_r5R9tPgdFDmC2N38ybXA9CUjHh7uVZtlC5Gb78He5kRqTCKGV-lVqBzubaI-VO-oQfEyui81I3e1XpAAQySITT6d1HtM9HyEk0_CMQ83N5mpUAKaCWT-Pqzby-GmKfet-ct07NiH-8yvvgThMEAl78bk-W8MBjQB6fagHjR93OWMr6usow4noL5sf1gVa1izd1_OLbroUY8kFHN7BCryHsy-WmFnfUzNb6qS-7HsX4qsoBAHmQMlYRj-B9RMa7QC_53f8laQPMnT4c1M1d1yITarmtL9n6cL0Ku096w3c-4735yk-mYpds0KaAclU9_7NjcznEcY3NHRr0LxwxoV6-ClHFO6Ccqgxyk-Kng0J3oGYwgY4QBPuSCphvf9PmbhABeLILx0i-_eeI3gOfAKbZUGDweftoG1Lw7qgB14KOX1fig69PFxp8dGApMOTJvQeXWN7ISPlDZRfmRkSbO5siDsTS4WS01qcgdc_2osmN8aMyB_bP-FG4v3CF8z6Cpp6x-iIyXdHS_riNGydbf5-Et1e5nlYTRAVW4Cgf-P_AAZH82PGIpWsPWO3Lsisli1IdVSHi5Vxfj8OzzE0aC5CuXzvCv18KwCyDgHZ2AZeqGkBHPyptmXyIPspaSlRtq7H7j5w57deaOmJVjHM.iOPoBiz9LmbJxxXkoKyzfQ"
+}
+
+last_replay_time = time.time()
+chatbot = Chatbot(config)
+
+
+class ChatGPTBot(Bot):
+    def reply(self, query):
+        global last_replay_time
+        now = time.time()
+        if now - last_replay_time > 60 * 10:
+            chatbot.refresh_session()
+        last_replay_time = now
+
+        res = chatbot.get_chat_response(query, output="text")
+        # print(res)
+        return res['message']
+
+
+if __name__ == '__main__':
+
+    bot = ChatGPTBot()
+    print(bot.reply('讲一个故事'))
diff --git a/bridge/bridge.py b/bridge/bridge.py
index 3e5af00..97fecc5 100644
--- a/bridge/bridge.py
+++ b/bridge/bridge.py
@@ -6,4 +6,4 @@ class Bridge(object):
         pass
 
     def fetch_reply_content(self, query):
-        return bot_factory.BaiduUnitBot().reply(query)
+        return bot_factory.create_bot("chatGPT").reply(query)
diff --git a/channel/wechat/__pycache__/wechat_channel.cpython-36.pyc b/channel/wechat/__pycache__/wechat_channel.cpython-36.pyc
index 1efc97cdabf7425e83637443f128f9fddcef2f65..835416f2198f9373f9a3361039462c7cf9924286 100644
GIT binary patch
delta 36
ucmV+<0NekN43G>7GYt(2;DqvGoUsY*0|5b(_XFYqB$L_%mH`NpaRr~}#th{E

delta 36
ucmV+<0NekN43G>7GYt(2_T%qj>ahv!0|5e)_XFYqCX?C(mH`TraRs0HrVXk9

diff --git a/channel/wechat/wechat_channel.py b/channel/wechat/wechat_channel.py
index bcae08e..02624a8 100644
--- a/channel/wechat/wechat_channel.py
+++ b/channel/wechat/wechat_channel.py
@@ -2,8 +2,6 @@
 wechat channel
 """
 import itchat
-import time
-import random
 import json
 from itchat.content import *
 from channel.channel import Channel
@@ -34,6 +32,6 @@ class WechatChannel(Channel):
             self.send(super().build_reply_content(msg['Text']), from_user_id)
 
     def send(self, msg, receiver):
-        time.sleep(random.randint(1, 3))
+        # time.sleep(random.randint(1, 3))
         print(msg, receiver)
         itchat.send(msg + " [bot]", toUserName=receiver)