From 5a2744069cb2cb11444064ca74e1a6cd25f1bbcf Mon Sep 17 00:00:00 2001 From: cyncyw <47289405+taozhiwang@users.noreply.github.com> Date: Sat, 28 Sep 2024 23:43:11 -0400 Subject: [PATCH] fix: rag save file (#385) * debug vector base --- rdagent/scenarios/kaggle/developer/feedback.py | 2 +- rdagent/scenarios/kaggle/experiment/scenario.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/rdagent/scenarios/kaggle/developer/feedback.py b/rdagent/scenarios/kaggle/developer/feedback.py index 59a68830..ffab0baa 100644 --- a/rdagent/scenarios/kaggle/developer/feedback.py +++ b/rdagent/scenarios/kaggle/developer/feedback.py @@ -162,7 +162,7 @@ class KGHypothesisExperiment2Feedback(HypothesisExperiment2Feedback): if self.scen.if_using_vector_rag: self.scen.vector_base.add_experience_to_vector_base(experiment_feedback) - self.scen.vector_base.save() + self.scen.vector_base.dump() elif self.scen.if_using_graph_rag: trace.knowledge_base.add_document(experiment_feedback, self.scen) diff --git a/rdagent/scenarios/kaggle/experiment/scenario.py b/rdagent/scenarios/kaggle/experiment/scenario.py index 3cf0b6ac..67ffd93e 100644 --- a/rdagent/scenarios/kaggle/experiment/scenario.py +++ b/rdagent/scenarios/kaggle/experiment/scenario.py @@ -45,7 +45,7 @@ class KGScenario(Scenario): if self.if_using_vector_rag and KAGGLE_IMPLEMENT_SETTING.rag_path: self.vector_base = KaggleExperienceBase(KAGGLE_IMPLEMENT_SETTING.rag_path) self.vector_base.path = datetime.now(timezone.utc).strftime("%Y-%m-%d-%H-%M-%S") + "_kaggle_kb.pkl" - self.vector_base.save() + self.vector_base.dump() self._output_format = self.output_format self._interface = self.interface