scriptoriumadmin/ReMe
1
0
Fork 0
mirror of https://github.com/agentscope-ai/ReMe.git synced 2026-08-28 05:25:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 11

refactor(config): update default LLM model and optimize fs compactor logic

Browse source
This commit is contained in:
jinli.yl 2026-02-24 22:00:09 +08:00
parent b1160d3b8e
commit 77928ccb8d
4 changed files with 14 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
reme/agent/fs/fs_compactor.py
View file

@ -81,32 +81,28 @@ class FsCompactor(BaseOp):
messages_to_summarize = self._normalize_messages(messages_to_summarize)
turn_prefix_messages = self._normalize_messages(turn_prefix_messages)
# Build prompt messages once before branching
history_prompt_messages = self._build_history_prompt(messages_to_summarize, previous_summary)
turn_prefix_prompt_messages = self._build_turn_prefix_prompt(turn_prefix_messages)
if self.return_prompt:
result = {
"system": self.get_prompt("system_prompt"),
}
result = {"system": self.get_prompt("system_prompt")}
if messages_to_summarize:
history_prompt_messages = self._build_history_prompt(messages_to_summarize, previous_summary)
if len(history_prompt_messages) == 2:
result["history_user"] = history_prompt_messages[-1].content
if len(history_prompt_messages) == 2:
result["history_user"] = history_prompt_messages[-1].content
if turn_prefix_messages:
turn_prefix_prompt_messages = self._build_turn_prefix_prompt(turn_prefix_messages)
if len(turn_prefix_prompt_messages) == 2:
result["turn_prefix_user"] = turn_prefix_prompt_messages[-1].content
if len(turn_prefix_prompt_messages) == 2:
result["turn_prefix_user"] = turn_prefix_prompt_messages[-1].content
return result
else:
if messages_to_summarize:
history_prompt_messages = self._build_history_prompt(messages_to_summarize, previous_summary)
if history_prompt_messages:
history_summary = "**History Summary**:\n\n" + await self._generate_summary(history_prompt_messages)
else:
history_summary = ""
if turn_prefix_messages:
turn_prefix_prompt_messages = self._build_turn_prefix_prompt(turn_prefix_messages)
if turn_prefix_prompt_messages:
turn_prefix_summary = "**Turn Context**:\n\n" + await self._generate_summary(
turn_prefix_prompt_messages,
)

2
reme/agent/fs/fs_summarizer.py
View file

@ -39,7 +39,7 @@ class FsSummarizer(BaseReact):
content=f"<conversation>\n{conversation}\n</conversation>\n"
+ self.prompt_format(
"user_message_default",
conversation=format_messages(messages, add_index=False),
conversation=conversation,
working_dir=self.working_dir,
date=date_str,
memory_dir=self.memory_dir,

3
reme/config/cli.yaml
View file

@ -11,7 +11,8 @@ metadata:
llms:
default:
backend: openai
model_name: qwen3-235b-a22b-thinking-2507
# model_name: qwen3-235b-a22b-thinking-2507
model_name: qwen3.5-plus
request_interval: 1
embedding_models:

3
reme/reme_fs.py
View file

@ -2,8 +2,6 @@
from pathlib import Path
from loguru import logger
from .agent.fs import FsCompactor, FsContextChecker, FsSummarizer
from .config import ReMeConfigParser
from .core import Application
@ -83,7 +81,6 @@ class ReMeFs(Application):
self.service_config.metadata.setdefault("keep_recent_tokens", keep_recent_tokens)
self.service_config.metadata.setdefault("vector_weight", vector_weight)
self.service_config.metadata.setdefault("candidate_multiplier", candidate_multiplier)
logger.info(f"ReMe model_extra config: {self.service_config.metadata}")
async def context_check(self, messages: list[Message | dict]) -> dict:
"""Check if messages exceed context limits."""