fix: fix a bug in kaggle conf (#459)

* fix a bug in kaggle conf

* fix a ci error
This commit is contained in:
WinstonLiyt
2024-10-29 15:13:18 +08:00
committed by GitHub
parent 762dd64239
commit 1d50ffd9ac
2 changed files with 5 additions and 3 deletions
+1 -3
View File
@@ -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"
+4
View File
@@ -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: