From 80e4a2aa79397d5fde95ace420afe7bd5cc1e7a5 Mon Sep 17 00:00:00 2001
From: WinstonLiyt <104308117+WinstonLiyt@users.noreply.github.com>
Date: Tue, 29 Oct 2024 15:13:18 +0800
Subject: [PATCH] fix: fix a bug in kaggle conf (#459)

* fix a bug in kaggle conf

* fix a ci error
---
 rdagent/app/kaggle/conf.py | 4 +---
 rdagent/app/kaggle/loop.py | 4 ++++
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/rdagent/app/kaggle/conf.py b/rdagent/app/kaggle/conf.py
index 06788caf..c9b1195c 100644
--- a/rdagent/app/kaggle/conf.py
+++ b/rdagent/app/kaggle/conf.py
@@ -63,9 +63,7 @@ class KaggleBasePropSetting(BasePropSetting):
     """Enable advanced graph-based RAG"""
 
     # Conditionally set the knowledge_base based on the use of graph RAG
-    knowledge_base: str = (
-        "rdagent.scenarios.kaggle.knowledge_management.graph.KGKnowledgeGraph" if if_using_graph_rag else ""
-    )
+    knowledge_base: str = ""
     """Knowledge base class, uses 'KGKnowledgeGraph' when advanced graph-based RAG is enabled, otherwise empty."""
 
     knowledge_base_path: str = "kg_graph.pkl"
diff --git a/rdagent/app/kaggle/loop.py b/rdagent/app/kaggle/loop.py
index e448366b..3839a4eb 100644
--- a/rdagent/app/kaggle/loop.py
+++ b/rdagent/app/kaggle/loop.py
@@ -129,6 +129,10 @@ def main(path=None, step_n=None, competition=None):
     if competition:
         KAGGLE_IMPLEMENT_SETTING.competition = competition
         download_data(competition=competition, local_path=KAGGLE_IMPLEMENT_SETTING.local_data_path)
+        if KAGGLE_IMPLEMENT_SETTING.if_using_graph_rag:
+            KAGGLE_IMPLEMENT_SETTING.knowledge_base = (
+                "rdagent.scenarios.kaggle.knowledge_management.graph.KGKnowledgeGraph"
+            )
     else:
         logger.error("Please specify competition name.")
     if path is None: