mirror of https://github.com/Mai-with-u/MaiBot.git
reformat: 格式化memory.py
parent
f3fef69968
commit
77df50e666
|
|
@ -27,6 +27,7 @@ logger = log_module.setup_logger(LogClassification.MEMORY)
|
||||||
|
|
||||||
logger.info("初始化记忆系统")
|
logger.info("初始化记忆系统")
|
||||||
|
|
||||||
|
|
||||||
class Memory_graph:
|
class Memory_graph:
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self.G = nx.Graph() # 使用 networkx 的图结构
|
self.G = nx.Graph() # 使用 networkx 的图结构
|
||||||
|
|
@ -210,7 +211,8 @@ class Hippocampus:
|
||||||
# 成功抽取短期消息样本
|
# 成功抽取短期消息样本
|
||||||
# 数据写回:增加记忆次数
|
# 数据写回:增加记忆次数
|
||||||
for message in messages:
|
for message in messages:
|
||||||
db.messages.update_one({"_id": message["_id"]}, {"$set": {"memorized_times": message["memorized_times"] + 1}})
|
db.messages.update_one({"_id": message["_id"]},
|
||||||
|
{"$set": {"memorized_times": message["memorized_times"] + 1}})
|
||||||
return messages
|
return messages
|
||||||
try_count += 1
|
try_count += 1
|
||||||
# 三次尝试均失败
|
# 三次尝试均失败
|
||||||
|
|
@ -943,6 +945,7 @@ def segment_text(text):
|
||||||
seg_text = list(jieba.cut(text))
|
seg_text = list(jieba.cut(text))
|
||||||
return seg_text
|
return seg_text
|
||||||
|
|
||||||
|
|
||||||
driver = get_driver()
|
driver = get_driver()
|
||||||
config = driver.config
|
config = driver.config
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue