gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-07-28 07:57:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: rag save file (#385)

Browse Source 
* debug vector base
This commit is contained in:
cyncyw
2024-09-28 23:43:11 -04:00
committed by GitHub
parent 4f8a9aca40
commit 5a2744069c
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/scenarios/kaggle/developer/feedback.py
+1 -1
View File
@@ -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)
rdagent/scenarios/kaggle/experiment/scenario.py
+1 -1
View File
@@ -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