Spaces:
Running
Running
Commit ·
aa4e0f4
1
Parent(s): 0107b5a
update graph path in graph_builder.py
Browse files
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 |
-
|
|
|
|
| 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 = {
|