diff --git a/trustgraph-flow/trustgraph/extract/kg/agent/extract.py b/trustgraph-flow/trustgraph/extract/kg/agent/extract.py index 59fec208..b7ef9259 100644 --- a/trustgraph-flow/trustgraph/extract/kg/agent/extract.py +++ b/trustgraph-flow/trustgraph/extract/kg/agent/extract.py @@ -27,13 +27,13 @@ class Processor(FlowProcessor): id = params.get("id") concurrency = params.get("concurrency", 1) - template_id = params.get("template-id", default_template_id) - config_key = params.get("config-type", default_config_type) + template_id = params.get("template_id", default_template_id) + config_key = params.get("config_type", default_config_type) super().__init__(**params | { "id": id, - "template-id": template_id, - "config-type": config_key, + "template_id": template_id, + "config_type": config_key, "concurrency": concurrency, }) diff --git a/trustgraph-flow/trustgraph/extract/kg/objects/processor.py b/trustgraph-flow/trustgraph/extract/kg/objects/processor.py index e925a349..b3483240 100644 --- a/trustgraph-flow/trustgraph/extract/kg/objects/processor.py +++ b/trustgraph-flow/trustgraph/extract/kg/objects/processor.py @@ -53,7 +53,7 @@ class Processor(FlowProcessor): super(Processor, self).__init__( **params | { "id": id, - "config-type": self.config_key, + "config_type": self.config_key, "concurrency": concurrency, } ) diff --git a/trustgraph-flow/trustgraph/prompt/template/service.py b/trustgraph-flow/trustgraph/prompt/template/service.py index 8ba49e3b..1b6822cc 100755 --- a/trustgraph-flow/trustgraph/prompt/template/service.py +++ b/trustgraph-flow/trustgraph/prompt/template/service.py @@ -37,7 +37,7 @@ class Processor(FlowProcessor): super(Processor, self).__init__( **params | { "id": id, - "config-type": self.config_key, + "config_type": self.config_key, "concurrency": concurrency, } ) diff --git a/trustgraph-flow/trustgraph/query/objects/cassandra/service.py b/trustgraph-flow/trustgraph/query/objects/cassandra/service.py index 8e446339..a4726d90 100644 --- a/trustgraph-flow/trustgraph/query/objects/cassandra/service.py +++ b/trustgraph-flow/trustgraph/query/objects/cassandra/service.py @@ -91,7 +91,7 @@ class Processor(FlowProcessor): super(Processor, self).__init__( **params | { "id": id, - "config-type": self.config_key, + "config_type": self.config_key, } ) diff --git a/trustgraph-flow/trustgraph/retrieval/nlp_query/service.py b/trustgraph-flow/trustgraph/retrieval/nlp_query/service.py index 67eaeaec..04dae978 100644 --- a/trustgraph-flow/trustgraph/retrieval/nlp_query/service.py +++ b/trustgraph-flow/trustgraph/retrieval/nlp_query/service.py @@ -36,7 +36,7 @@ class Processor(FlowProcessor): super(Processor, self).__init__( **params | { "id": id, - "config-type": self.config_key, + "config_type": self.config_key, } ) diff --git a/trustgraph-flow/trustgraph/storage/objects/cassandra/write.py b/trustgraph-flow/trustgraph/storage/objects/cassandra/write.py index 269171e4..8a6db66c 100644 --- a/trustgraph-flow/trustgraph/storage/objects/cassandra/write.py +++ b/trustgraph-flow/trustgraph/storage/objects/cassandra/write.py @@ -50,7 +50,7 @@ class Processor(FlowProcessor): super(Processor, self).__init__( **params | { "id": id, - "config-type": self.config_key, + "config_type": self.config_key, } )