This is an automated email from the ASF dual-hosted git repository.
jin pushed a commit to branch text2gql
in repository https://gitbox.apache.org/repos/asf/incubator-hugegraph-ai.git
commit 643a3bb804ead24ffee91aab69424629a512713e
Author: Lriver <zj...@foxmail.com>
AuthorDate: Tue Sep 30 20:52:06 2025 +0800
feat: add configuration management module with dictionary paths and generation parameters
---
text2gremlin/AST_Text2Gremlin/base/Config.py | 66 ++++++++++++++++++++++++++++
1 file changed, 66 insertions(+)
diff --git a/text2gremlin/AST_Text2Gremlin/base/Config.py b/text2gremlin/AST_Text2Gremlin/base/Config.py
new file mode 100644
index 00000000..e64ebb18
--- /dev/null
+++ b/text2gremlin/AST_Text2Gremlin/base/Config.py
@@ -0,0 +1,66 @@
+
+"""
+项目配置管理模块。
+
+负责加载和管理项目配置文件，提供各模块所需的配置参数。
+"""
+
+import json
+import os
+import sys
+
+
+class Config:
+ def __init__(self, file_path):
+ self.file_path = file_path
+ self.config_data = self.load_config()
+ self.gen_query = self.config_data.get(
+ "genQuery"
+ ) # default genQuery，can be set as translate
+ self.db_id = self.config_data.get("db_id")
+
+ def load_config(self):
+ with open(self.file_path, "r") as file:
+ return json.load(file)
+
+ def get_input_query_path(self):
+ return self.config_data.get("input_query_path")
+
+ def get_input_query_template_path(self):
+ return self.config_data.get("input_query_template_path")
+
+ def get_input_corpus_dir_or_file(self):
+ return self.config_data.get("input_corpus_dir_or_path")
+
+ def set_input_corpus_dir_or_file(self, dir_or_file):
+ self.config_data["input_corpus_dir_or_path"] = dir_or_file
+
+ def get_output_path(self):
+ if self.gen_query:
+ dir_or_file = self.config_data.get("output_query_dir_or_file")
+ if os.path.isdir(dir_or_file):
+ output_path = os.path.join(dir_or_file, self.db_id + ".txt")
+ return output_path
+ else:
+ return dir_or_file
+ else:
+ return self.config_data.get("output_prompt_path")
+
+ def get_output_corpus(self):
+ return self.config_data.get("output_corpus_path")
+
+ def get_schema_dict_path(self):
+ return self.config_data.get("schema_dict_path")
+
+ def get_syn_dict_path(self):
+ return self.config_data.get("syn_dict_path")
+
+ def get_db_id(self):
+ return self.db_id
+
+ def get_schema_path(self, db_id):
+ schema_dict = self.config_data.get("db_schema_path")
+ return schema_dict[db_id] # todo error check
+
+ def get_config(self, module_name):
+ return self.config_data.get(module_name)