snap updated.

nodeloc · Jan 20, 2024 · 0d9d93f · 0d9d93f
1 parent 140c3e2
commit 0d9d93f
Show file tree

Hide file tree

Showing 6 changed files with 428 additions and 3 deletions.
diff --git a/COOKIES-NodeLoc b/COOKIES-NodeLoc
diff --git a/COOKIES-糊里糊涂 b/COOKIES-糊里糊涂
diff --git a/README.md b/README.md
@@ -11,6 +11,8 @@ V2.0 增加了多用户的支持以及多ChatGPT key的轮询。
 ```
 prompt请修改一下，以避免你和别人的答案类似，不同的prompt会产生截然不同的回答。
 
+2024年1月20日 兼容了第三方ChatGPT接口
+
 
 2024年1月17日 修复了不能T楼的小bug
 

diff --git a/config.py b/config.py
@@ -4,6 +4,9 @@
     {'username': 'username', 'password': '123456'},
     # 添加更多的用户名和密码组合
 ]
+chatgpt_api_url = 'https://api.openai.com'
+chatgpt_model = 'gpt-3.5-turbo'
+chatgpt_prompt = '你是一个常年混迹hostloc的人，帮助回答一些问题.'
 
 chatgpt_keys = [
     'chatgpt_key1',

diff --git a/discuz.py b/discuz.py
@@ -70,15 +70,15 @@ def get_reply_tid(self):
             sys.exit()
 
     def chat_with_gpt(self, prompt):
-        url = "https://api.openai.com/v1/chat/completions"
+        url = config.chatgpt_api_url + "/v1/chat/completions"
         headers = {
             "Content-Type": "application/json",
             "Authorization": "Bearer " + self.chatgpt_key,
         }
         data = {
-            "model": "gpt-3.5-turbo",
+            "model": config.chatgpt_model,
             "messages": [
-                {"role": "system", "content": "你是一个常年混迹hostloc的人，帮助回答一些问题."},
+                {"role": "system", "content": config.chatgpt_prompt},
                 {"role": "user", "content": prompt}
             ],
             "max_tokens": 500,
-Original file line number
+Diff line change
@@ Expand Up @@
     ```
     prompt请修改一下，以避免你和别人的答案类似，不同的prompt会产生截然不同的回答。
+年1月20日 兼容了第三方ChatGPT接口
 年1月17日 修复了不能T楼的小bug
@@ Expand Down @@