From 0f591c60a2e7a0f7bf08ab24372cd138ff6ffc20 Mon Sep 17 00:00:00 2001 From: matthewhyx Date: Thu, 28 Dec 2023 00:47:46 +0800 Subject: [PATCH] fix schema --- .../builder/job/state_and_indicator.py | 27 ++++++++++--------- 1 file changed, 14 insertions(+), 13 deletions(-) diff --git a/python/knext/knext/examples/financial/builder/job/state_and_indicator.py b/python/knext/knext/examples/financial/builder/job/state_and_indicator.py index 8e59988b..dac01c4f 100644 --- a/python/knext/knext/examples/financial/builder/job/state_and_indicator.py +++ b/python/knext/knext/examples/financial/builder/job/state_and_indicator.py @@ -1,11 +1,11 @@ # -*- coding: utf-8 -*- -from knext.client.model.builder_job import BuilderJob -from knext.api.component import CSVReader, LLMBasedExtractor, KGWriter, SubGraphMapping -from nn4k.invoker import LLMInvoker - from schema.financial_schema_helper import Financial +from knext.api.component import CSVReader, LLMBasedExtractor, KGWriter, SubGraphMapping +from knext.client.model.builder_job import BuilderJob +from nn4k.invoker import LLMInvoker + class StateAndIndicator(BuilderJob): def build(self): @@ -28,7 +28,7 @@ class StateAndIndicator(BuilderJob): SubGraphMapping(spg_type_name=Financial.State) .add_mapping_field("id", Financial.State.id) .add_mapping_field("name", Financial.State.name) - .add_mapping_field("causeOf", Financial.State.causeOf) + .add_mapping_field("causes", Financial.State.causes) .add_predicting_field(Financial.State.derivedFrom) ) @@ -36,7 +36,6 @@ class StateAndIndicator(BuilderJob): SubGraphMapping(spg_type_name=Financial.Indicator) .add_mapping_field("id", Financial.Indicator.id) .add_mapping_field("name", Financial.Indicator.name) - # .add_predicting_field(Financial.Indicator.isA) ) sink = KGWriter() @@ -45,14 +44,16 @@ class StateAndIndicator(BuilderJob): if __name__ == '__main__': - from knext.api.auto_prompt import EEPrompt - prompt = EEPrompt( - event_type_name=Financial.RegionalEconomicIndicatorEvent, + from knext.api.auto_prompt import REPrompt + prompt = REPrompt( + spg_type_name=Financial.Company, property_names=[ - Financial.RegionalEconomicIndicatorEvent.date, - Financial.RegionalEconomicIndicatorEvent.region, - Financial.RegionalEconomicIndicatorEvent.source, - Financial.RegionalEconomicIndicatorEvent.subject, + Financial.Company.orgCertNo, + Financial.Company.regArea, + Financial.Company.businessScope, + Financial.Company.establishDate, + Financial.Company.legalPerson, + Financial.Company.regCapital ], ) print(prompt.template)