diff --git a/scripts/import_packages.py b/scripts/import_packages.py index 9d9241b3..1cfdfd1e 100644 --- a/scripts/import_packages.py +++ b/scripts/import_packages.py @@ -22,6 +22,7 @@ def __init__(self, jsonl_dir="data", vec_db_path="./sqlite_data/vectordb.db"): os.path.join(jsonl_dir, "malicious.jsonl"), ] self.conn = self._get_connection() + Config.load() # Load the configuration self.inference_engine = LlamaCppInferenceEngine() self.model_path = "./codegate_volume/models/all-minilm-L6-v2-q5_k_m.gguf"