Skip to content

Commit 02c2c24

Browse files
committed
use default module to simplify retriever config
1 parent 5599cdb commit 02c2c24

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

examples/question_answering/simple_rag_pipeline_config.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@
4141
}
4242
},
4343
"retriever": {
44-
"class_": "neo4j_graphrag.retrievers.VectorRetriever",
44+
"class_": "VectorRetriever",
4545
"params_": {
4646
"driver": {
4747
"resolver_": "CONFIG_KEY",

src/neo4j_graphrag/experimental/pipeline/config/template_pipeline/simple_rag_pipeline.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,8 @@ class RetrieverConfig(ObjectConfig[RetrieverWrapper]):
3434
# the result of _get_class is a Retriever
3535
# it is translated into a RetrieverWrapper (which is a Component)
3636
# in the 'parse' method below
37+
DEFAULT_MODULE = "neo4j_graphrag.retrievers"
38+
REQUIRED_PARAMS = ["driver"]
3739

3840
def parse(self, resolved_data: Optional[dict[str, Any]] = None) -> RetrieverWrapper:
3941
retriever = cast(Retriever, super().parse(resolved_data))

0 commit comments

Comments
 (0)