scriptoriumadmin/ReMe
1
0
Fork 0
mirror of https://github.com/agentscope-ai/ReMe.git synced 2026-09-06 08:16:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

language word

Browse source
This commit is contained in:
hs 2024-07-02 14:33:21 +08:00
parent fcdb258bdd
commit c0d9a22574
5 changed files with 6 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
memory_scope/memory/worker/summary/long_contra_repeat_worker.py
View file

@ -113,13 +113,7 @@ class LongContraRepeatWorker(MemoryBaseWorker):
self.logger.warning(f"idx={idx} is invalid!")
continue
long_contra_repeat_keep_flag = [
self.get_language_value(CONTRADICTORY_WORD),
self.get_language_value(INCLUDED_WORD),
self.get_language_value(NONE_WORD),
]
if keep_flag not in long_contra_repeat_keep_flag.values():
if keep_flag not in self.get_language_value([CONTRADICTORY_WORD, INCLUDED_WORD, NONE_WORD]):
self.logger.warning(f"keep_flag={keep_flag} is invalid!")
continue

5
memory_scope/memory/worker/summary/update_insight_prompt.yaml
View file

@ -61,7 +61,4 @@ user_query_prompt:
user_query:
cn: "句子：{content}"
insight_value:
cn: ["无", "重复"]
en:

4
memory_scope/memory/worker/summary/update_insight_worker.py
View file

@ -8,7 +8,7 @@ from memory_scope.constants.common_constants import (
from memory_scope.utils.tool_functions import prompt_to_msg
from memory_scope.scheme.memory_node import MemoryNode
from memory_scope.memory.worker.memory_base_worker import MemoryBaseWorker
from memory_scope.constants.language_constants import COMMA_WORD, COLON_WORD
from memory_scope.constants.language_constants import COMMA_WORD, COLON_WORD, NONE_WORD, REPEATED_WORD
class UpdateInsightWorker(MemoryBaseWorker):
@ -112,7 +112,7 @@ class UpdateInsightWorker(MemoryBaseWorker):
return insight_node
insight_value = profile_list[0]
if not insight_value or insight_value in self.prompt_handler.insight_value:
if not insight_value or insight_value in self.get_language_value([NONE_WORD, REPEATED_WORD]):
self.logger.info(f"insight_value={insight_value}, skip.")
return insight_node

3
memory_scope/memory/worker/summary/update_profile_prompt.yaml
View file

@ -118,6 +118,3 @@ update_unique_profile_user_query_prompt:
user_query:
cn: "句子：{content}"
en:
update_profile_key:
cn: ["无", "重复"]

4
memory_scope/memory/worker/summary/update_profile_worker.py
View file

@ -7,7 +7,7 @@ from memory_scope.scheme.memory_node import MemoryNode
from memory_scope.memory.worker.memory_base_worker import MemoryBaseWorker
from memory_scope.utils.global_context import GlobalContext
from memory_scope.utils.tool_functions import prompt_to_msg
from memory_scope.constants.language_constants import COMMA_WORD, COLON_WORD
from memory_scope.constants.language_constants import COMMA_WORD, COLON_WORD, NONE_WORD, REPEATED_WORD
class UpdateProfileWorker(MemoryBaseWorker):
@ -118,7 +118,7 @@ class UpdateProfileWorker(MemoryBaseWorker):
return user_attr
profile = profile_list[0]
if not profile or profile in self.prompt_handler.update_profile_key:
if not profile or profile in self.get_language_value([NONE_WORD, REPEATED_WORD]):
self.logger.info(f"profile={profile}, skip.")
return user_attr