aditya-joshi-05 commited on
Commit
aa4e0f4
·
1 Parent(s): 0107b5a

update graph path in graph_builder.py

Browse files
Files changed (1) hide show
  1. retrieval/graph_builder.py +4 -1
retrieval/graph_builder.py CHANGED
@@ -55,9 +55,12 @@ from retrieval.relation_extractors import (
55
  build_extractor,
56
  )
57
 
 
 
58
  logger = logging.getLogger(__name__)
59
 
60
- _DEFAULT_GRAPH_PATH = Path("data/knowledge_graph.json")
 
61
 
62
  # spaCy entity types we care about for RAG
63
  _ENTITY_TYPES = {
 
55
  build_extractor,
56
  )
57
 
58
+ from config import get_settings
59
+
60
  logger = logging.getLogger(__name__)
61
 
62
+ cfg = get_settings()
63
+ _DEFAULT_GRAPH_PATH = Path(cfg.graph_path)
64
 
65
  # spaCy entity types we care about for RAG
66
  _ENTITY_TYPES = {