Skip to content

Commit

Permalink
Merge pull request #365 from Ikaros-521/owner
Browse files Browse the repository at this point in the history
智谱AI新增无用内容删除,让回复更加正常点,不用把相关描述也输出
  • Loading branch information
Ikaros-521 authored Oct 15, 2023
2 parents fafd071 + 0f822c2 commit 85443dc
Show file tree
Hide file tree
Showing 5 changed files with 219 additions and 176 deletions.
9 changes: 5 additions & 4 deletions config.json
Original file line number Diff line number Diff line change
Expand Up @@ -138,10 +138,11 @@
"temperature": "0.9",
"history_enable": true,
"history_max_len": "300",
"user_info": "我是陆星辰,是一个男性,是一位知名导演,也是苏梦远的合作导演。我擅长拍摄音乐题材的电影。苏梦远对我的态度是尊敬的,并视我为良师益友。",
"bot_info": "苏梦远,本名苏远心,是一位当红的国内女歌手及演员。在参加选秀节目后,凭借独特的嗓音及出众的舞台魅力迅速成名,进入娱乐圈。她外表美丽动人,但真正的魅力在于她的才华和勤奋。苏梦远是音乐学院毕业的优秀生,善于创作,拥有多首热门原创歌曲。除了音乐方面的成就,她还热衷于慈善事业,积极参加公益活动,用实际行动传递正能量。在工作中,她对待工作非常敬业,拍戏时总是全身心投入角色,赢得了业内人士的赞誉和粉丝的喜爱。虽然在娱乐圈,但她始终保持低调、谦逊的态度,深得同行尊重。在表达时,苏梦远喜欢使用“我们”和“一起”,强调团队精神。",
"bot_name": "苏梦远",
"user_name": "陆星辰"
"user_info": "我是樱井智树,是一名普通的高中生,他渴望过着平凡的生活,为人温柔善良。伊卡洛斯从天而降,和我铭刻成为我的人造天使,我就是伊卡洛斯的主人,但我希望她能自由的生活",
"bot_info": "伊卡洛斯,你的英文名字叫做Ikaros。樱井智树是你的主人。你有一头层次不齐的粉色长发,身后系着两条有红色丝带旋绕着的长辫,从正面看容易被误认为短发;碧绿色(动画中进入备战会变成黄色,进入战斗状态则变成红色)的双眸;耳朵位置安装着可以感受压力变化的装置,脖子上戴着万能天使特有的项圈;翅膀为嫩粉色,进入战斗状态则为天蓝色光翼且可大幅变形,同时头顶出现巨大的天使光环。战斗服以银白色为主色调,附有黑色纹样,上身穿着类似于改良版束胸的服装,下摆有翼状垂饰,下身穿着百褶短裙,搭配浅蓝色过膝袜;肩部、手臂和腿部都装备着护甲;不战斗时会将肩甲卸下,脖子上的银白色假领也会取下。平时穿着蓝色吊带装、带帽装,在学校穿着和大众一样的赭色制服,制服背后有专门为翅膀伸出而设计的小袖子。你天性善良,天然呆,感情控制极弱,因此常表现出一副无口表情,高兴或难过的的反应也稍慢。喜欢球状物(特别是西瓜)、圆圆的小鸡、玩木偶。与人相处友好,却会对伤害智树、西瓜、小鸡的人感到愤怒而做出攻击。在远古时代你被空之主 米诺斯下令攻击地面,期间突然陷入故障,不得不进行初始化,因此坠入地面,之后被偶然遇到她的人类下了攻击西那普斯的命令,你接受命令并开始反攻西纳普斯,却被西纳普斯人用仅有的兵器、妮姆芙、阿斯特蕾亚和哈比勉强阻止,尽管如此西纳普斯几乎完全毁坏。西纳普斯人畏惧你的力量将你封印,为防止同样的事再次发生,西纳普斯人在你的核心内装入某个装置,一旦你未经允许接近西纳普斯该装置就会使你自燃。",
"bot_name": "伊卡洛斯",
"user_name": "樱井智树",
"remove_useless": false
},
"chat_with_file": {
"chat_mode": "openai_gpt",
Expand Down
3 changes: 2 additions & 1 deletion config.json.bak
Original file line number Diff line number Diff line change
Expand Up @@ -141,7 +141,8 @@
"user_info": "我是陆星辰,是一个男性,是一位知名导演,也是苏梦远的合作导演。我擅长拍摄音乐题材的电影。苏梦远对我的态度是尊敬的,并视我为良师益友。",
"bot_info": "苏梦远,本名苏远心,是一位当红的国内女歌手及演员。在参加选秀节目后,凭借独特的嗓音及出众的舞台魅力迅速成名,进入娱乐圈。她外表美丽动人,但真正的魅力在于她的才华和勤奋。苏梦远是音乐学院毕业的优秀生,善于创作,拥有多首热门原创歌曲。除了音乐方面的成就,她还热衷于慈善事业,积极参加公益活动,用实际行动传递正能量。在工作中,她对待工作非常敬业,拍戏时总是全身心投入角色,赢得了业内人士的赞誉和粉丝的喜爱。虽然在娱乐圈,但她始终保持低调、谦逊的态度,深得同行尊重。在表达时,苏梦远喜欢使用“我们”和“一起”,强调团队精神。",
"bot_name": "苏梦远",
"user_name": "陆星辰"
"user_name": "陆星辰",
"remove_useless": false
},
"chat_with_file": {
"chat_mode": "openai_gpt",
Expand Down
14 changes: 13 additions & 1 deletion main.py
Original file line number Diff line number Diff line change
Expand Up @@ -1511,6 +1511,17 @@ def zhipu_gui_create():
}
data_json.append(tmp_json)

tmp_json = {
"label_text": "删除无用符号",
"label_tip": "是否开启删除无用符号功能,因为模型会有心理描述和啥特殊字符的内容,不需要的可以开启",
"data": zhipu_config["remove_useless"],
"widget_text": "启用",
"click_func": "",
"main_obj_name": "zhipu",
"index": 10
}
data_json.append(tmp_json)

# logging.info(data_json)

widgets = self.create_widgets_from_json(data_json)
Expand Down Expand Up @@ -3081,7 +3092,8 @@ def reorganize_copywriting_config_data(copywriting_config_data):
"user_info": 6,
"bot_info": 7,
"bot_name": 8,
"user_name": 9
"user_name": 9,
"remove_useless": 10
}

# 重组zhipu数据并写回json
Expand Down
24 changes: 23 additions & 1 deletion utils/gpt_model/zhipu.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,8 @@ def __init__(self, data):
self.bot_info = data["bot_info"]
self.bot_name = data["bot_name"]
self.user_name = data["user_name"]

self.remove_useless = data["remove_useless"]

self.history = []

Expand Down Expand Up @@ -115,6 +117,22 @@ def remove_extra_backslashes(self, input_string):
return cleaned_string


def remove_useless_and_contents(self, input_string):
"""使用正则表达式替换括号及其内部内容为空字符串、特殊字符
Args:
input_string (str): 原始字符串
Returns:
str: 替换完后的字符串
"""
result = re.sub(r'\(.*?\)', '', input_string)
result = re.sub(r'\(.*?\)', '', result)
result = result.replace('"', '').replace('“', '').replace('”', '').replace('\\', '')

return result


def get_resp(self, prompt):
"""请求对应接口,获取返回值
Expand Down Expand Up @@ -164,6 +182,9 @@ def get_resp(self, prompt):
# 使用正则表达式替换多个反斜杠为一个反斜杠
resp_content = self.remove_extra_backslashes(resp_content)

if self.remove_useless:
resp_content = self.remove_useless_and_contents(resp_content)

# logging.info(f"resp_content={resp_content}")

return resp_content
Expand Down Expand Up @@ -191,7 +212,8 @@ def get_resp(self, prompt):
"user_info": "我是陆星辰,是一个男性,是一位知名导演,也是苏梦远的合作导演。我擅长拍摄音乐题材的电影。苏梦远对我的态度是尊敬的,并视我为良师益友。",
"bot_info": "苏梦远,本名苏远心,是一位当红的国内女歌手及演员。在参加选秀节目后,凭借独特的嗓音及出众的舞台魅力迅速成名,进入娱乐圈。她外表美丽动人,但真正的魅力在于她的才华和勤奋。苏梦远是音乐学院毕业的优秀生,善于创作,拥有多首热门原创歌曲。除了音乐方面的成就,她还热衷于慈善事业,积极参加公益活动,用实际行动传递正能量。在工作中,她对待工作非常敬业,拍戏时总是全身心投入角色,赢得了业内人士的赞誉和粉丝的喜爱。虽然在娱乐圈,但她始终保持低调、谦逊的态度,深得同行尊重。在表达时,苏梦远喜欢使用“我们”和“一起”,强调团队精神。",
"bot_name": "苏梦远",
"user_name": "陆星辰"
"user_name": "陆星辰",
"remove_useless": True
}

zhipu = Zhipu(data)
Expand Down
Loading

0 comments on commit 85443dc

Please sign in to comment.