From 7cbf5bf56c8e31991062bef72a995bd274561a24 Mon Sep 17 00:00:00 2001 From: Xu Yang Date: Mon, 28 Apr 2025 01:51:51 -0600 Subject: [PATCH] when restart with kb and different pkl path, we need to initialize the kb from json file (#835) --- rdagent/app/data_science/loop.py | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/rdagent/app/data_science/loop.py b/rdagent/app/data_science/loop.py index 79e282b1..d7e87e30 100644 --- a/rdagent/app/data_science/loop.py +++ b/rdagent/app/data_science/loop.py @@ -228,13 +228,10 @@ class DataScienceRDLoop(RDLoop): replace_timer: bool = True, ) -> "LoopBase": session = super().load(path, output_path, do_truncate, replace_timer) - if ( - DS_RD_SETTING.enable_knowledge_base - and DS_RD_SETTING.knowledge_base_version == "v1" - and Path(DS_RD_SETTING.knowledge_base_path).exists() - ): - knowledge_base = DSKnowledgeBase(path=DS_RD_SETTING.knowledge_base_path) - session.trace.knowledge_base = knowledge_base + if DS_RD_SETTING.enable_knowledge_base and DS_RD_SETTING.knowledge_base_version == "v1": + session.trace.knowledge_base = DSKnowledgeBase( + path=DS_RD_SETTING.knowledge_base_path, idea_pool_json_path=DS_RD_SETTING.idea_pool_json_path + ) return session def dump(self, path: str | Path) -> None: