Update bot/chatgpt/chat_gpt_bot.py

bot/chatgpt/chat_gpt_bot.py

@@ -11,8 +11,8 @@ user_session = dict()
 # OpenAI对话模型API (可用)
 class ChatGPTBot(Bot):
     def __init__(self):
-        openai.api_key = conf().get('open_ai_api_key')
-        openai.api_base="https://apai.zyai.online/v1"
+        openai.api_base = "http://i-1.gpushare.com:16398/v1"
+        openai.api_key = "none"
 
     def reply(self, query, context=None):
         # acquire reply content
@@ -42,13 +42,8 @@ class ChatGPTBot(Bot):
     def reply_text(self, query, user_id, retry_count=0):
         try:
             response = openai.ChatCompletion.create(
-                model="gpt-3.5-turbo",  # 对话模型的名称
-                messages=query,
-                temperature=1,  # 值在[0,1]之间，越大表示回复越具有不确定性
-                max_tokens=600,  # 回复最大的字符数
-                top_p=1,
-                frequency_penalty=0,  # [-2,2]之间，该值越大则更倾向于产生不同的内容
-                presence_penalty=0,  # [-2,2]之间，该值越大则更倾向于产生不同的内容
+                model="chatglm2-6b",  # 对话模型的名称
+                messages=query
             )
             # res_content = response.choices[0]['text'].strip().replace('<|endoftext|>', '')
             logger.info(response.choices[0]['message']['content'])