From 7407253245480760165c489f753b1c9d7463a8b6 Mon Sep 17 00:00:00 2001 From: seehi <6580@pm.me> Date: Sun, 29 Sep 2024 21:41:16 +0800 Subject: [PATCH] remove llmranker --- metagpt/memory/role_zero_memory.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/metagpt/memory/role_zero_memory.py b/metagpt/memory/role_zero_memory.py index a635d586c..5c7fb94a2 100644 --- a/metagpt/memory/role_zero_memory.py +++ b/metagpt/memory/role_zero_memory.py @@ -49,14 +49,14 @@ class RoleZeroLongTermMemory(Memory): try: from metagpt.rag.engines import SimpleEngine - from metagpt.rag.schema import ChromaRetrieverConfig, LLMRankerConfig + from metagpt.rag.schema import ChromaRetrieverConfig except ImportError: raise ImportError("To use the RoleZeroMemory, you need to install the rag module.") retriever_configs = [ ChromaRetrieverConfig(persist_path=self.persist_path, collection_name=self.collection_name) ] - ranker_configs = [LLMRankerConfig()] + ranker_configs = [] rag_engine = SimpleEngine.from_objs(retriever_configs=retriever_configs, ranker_configs=ranker_configs)