From dfdc3f83a248c83deafe3f0a89a8e7c04a744115 Mon Sep 17 00:00:00 2001
From: touer angler <61647484+weiensong@users.noreply.github.com>
Date: Thu, 16 Nov 2023 21:57:12 +0800
Subject: [PATCH] fix: default role and yaml formatted with chatgpt

---
 config.yaml.template | 25 ++++++++++++-------------
 main.py              |  4 ++--
 2 files changed, 14 insertions(+), 15 deletions(-)

diff --git a/config.yaml.template b/config.yaml.template
index 1e94a5c..364cf3d 100644
--- a/config.yaml.template
+++ b/config.yaml.template
@@ -21,7 +21,7 @@ logging:
       level: INFO
       formatter: simple
       filename: wx_info.log
-      maxBytes: 10485760 # 10MB
+      maxBytes: 10485760  # 10MB
       backupCount: 20
       encoding: utf8
 
@@ -30,7 +30,7 @@ logging:
       level: ERROR
       formatter: error
       filename: wx_error.log
-      maxBytes: 10485760 # 10MB
+      maxBytes: 10485760  # 10MB
       backupCount: 20
       encoding: utf8
 
@@ -39,34 +39,33 @@ logging:
     handlers: [console, info_file_handler, error_file_handler]
 
 groups:
-  enable: [] # 允许响应的群 roomId，大概长这样：2xxxxxxxxx3@chatroom
+  enable: []  # 允许响应的群 roomId，大概长这样：2xxxxxxxxx3@chatroom
 
 news:
-  receivers: [] # 定时新闻接收人（roomid 或者 wxid）
+  receivers: []  # 定时新闻接收人（roomid 或者 wxid）
 
 report_reminder:
-  receivers: [] # 定时日报周报月报提醒（roomid 或者 wxid）
-
+  receivers: []  # 定时日报周报月报提醒（roomid 或者 wxid）
 
 chatgpt:  # -----chatgpt配置这行不填-----
   key:  # 填写你 ChatGPT 的 key
-  api: https://api.openai.com/v1 # 如果你不知道这是干嘛的，就不要改
-  proxy: # 如果你在国内，你可能需要魔法，大概长这样：http://域名或者IP地址:端口号
-  prompt: # 根据需要对角色进行设定 你是智能聊天机器人，你叫wcferry
+  api: https://api.openai.com/v1  # 如果你不知道这是干嘛的，就不要改
+  proxy:  # 如果你在国内，你可能需要魔法，大概长这样：http://域名或者IP地址:端口号
+  prompt: 你是智能聊天机器人，你叫wcferry  # 根据需要对角色进行设定
 
 chatglm:  # -----chatglm配置这行不填-----
   key:  # key 暂时没有
-  api: http://localhost:8000/v1 # 根据自己的chatglm地址修改
+  api: http://localhost:8000/v1  # 根据自己的chatglm地址修改
   proxy:  # 如果你在国内，你可能需要魔法，大概长这样：http://域名或者IP地址:端口号
-  prompt:  # 根据需要对角色进行设定 你是智能聊天机器人，你叫小薇
+  prompt: 你是智能聊天机器人，你叫小薇  # 根据需要对角色进行设定 
   file_path: F:/Pictures/temp  #设定生成图片和代码使用的文件夹路径
 
 tigerbot:  # -----tigerbot配置这行不填-----
   key:  # key
   model:  # tigerbot-7b-sft
 
-xinghuo_web: # -----讯飞星火web模式api配置这行不填 抓取方式详见文档：https://www.bilibili.com/read/cv27066577-----
+xinghuo_web:  # -----讯飞星火web模式api配置这行不填 抓取方式详见文档：https://www.bilibili.com/read/cv27066577-----
   cookie:  # cookie
   fd:  # fd
   GtToken:  # GtToken
-  prompt: # 根据需要对角色进行设定你是智能聊天机器人，你叫wcferry。请用这个角色回答我的问题
+  prompt: 你是智能聊天机器人，你叫wcferry。请用这个角色回答我的问题  # 根据需要对角色进行设定
diff --git a/main.py b/main.py
index 944a39d..0a21679 100644
--- a/main.py
+++ b/main.py
@@ -59,8 +59,8 @@ def main(chat_type: int):
     robot.keepRunningAndBlockProcess()
 
 
-parser = ArgumentParser()
-parser.add_argument('-c', type=int, default=0, help='选择模型参数: 1, 2, 3, 4')
 if __name__ == "__main__":
+    parser = ArgumentParser()
+    parser.add_argument('-c', type=int, default=0, help='选择模型参数: 1, 2, 3, 4')
     args = parser.parse_args().c
     main(args)