shen/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-02-15 16:25:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add min simularity to match role

Browse Source
This commit is contained in:
lanvent
2023-04-06 01:35:42 +08:00
parent d3758968d0
commit eaac3e3579
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
plugins/role/role.py
View File

@@ -52,7 +52,7 @@ class Role(Plugin):
logger.warn("[Role] init failed, ignore or see https://github.com/zhayujie/chatgpt-on-wechat/tree/master/plugins/role .")
raise e
def get_role(self, name, find_closest=True):
def get_role(self, name, find_closest=True, min_sim = 0.35):
name = name.lower()
found_role = None
if name in self.roles:
@@ -62,7 +62,7 @@ class Role(Plugin):
def str_simularity(a, b):
return difflib.SequenceMatcher(None, a, b).ratio()
max_sim = 0.0
max_sim = min_sim
max_role = None
for role in self.roles:
sim = str_simularity(name, role)