mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-10-06 02:48:22 +00:00
update reme.py for consistency
This commit is contained in:
parent
dfadfed067
commit
4ad977408e
1 changed files with 4 additions and 26 deletions
30
reme/reme.py
30
reme/reme.py
|
|
@ -287,20 +287,8 @@ class ReMe(Application):
|
|||
raise NotImplementedError
|
||||
|
||||
procedural_summarizer: BaseMemoryAgent
|
||||
if version in ["default", "v1"]:
|
||||
procedural_summarizer = ProceduralSummarizer(
|
||||
tools=[
|
||||
VectorRetrieveMemory(
|
||||
add_memory_type_target=False,
|
||||
enable_thinking_params=True,
|
||||
metadata_desc=None,
|
||||
top_k=5,
|
||||
),
|
||||
AddMemory(enable_thinking_params=True,add_when_to_use=True, metadata_desc=metadata_summary),
|
||||
DeleteMemory(enable_thinking_params=True),
|
||||
UpdateMemory(enable_thinking_params=True,add_when_to_use=True, metadata_desc=metadata_summary),
|
||||
],
|
||||
)
|
||||
if version in ["default", "v1", "v2", "halumem"]:
|
||||
procedural_summarizer = ProceduralSummarizer(tools=[])
|
||||
else:
|
||||
raise NotImplementedError
|
||||
|
||||
|
|
@ -478,18 +466,8 @@ class ReMe(Application):
|
|||
raise NotImplementedError
|
||||
|
||||
procedural_retriever: BaseMemoryAgent
|
||||
if version in ["default", "v1"]:
|
||||
procedural_retriever = ProceduralRetriever(
|
||||
tools=[
|
||||
VectorRetrieveMemory(
|
||||
add_memory_type_target=True,
|
||||
enable_thinking_params=True,
|
||||
metadata_desc=metadata_retrieve,
|
||||
top_k=top_k,
|
||||
),
|
||||
ReadHistory(enable_thinking_params=True),
|
||||
],
|
||||
)
|
||||
if version in ["default", "v1", "v2", "halumem"]:
|
||||
procedural_retriever = ProceduralRetriever(tools=[])
|
||||
else:
|
||||
raise NotImplementedError
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue