yujunjun/openspg
1
0
Fork 0
mirror of https://github.com/OpenSPG/openspg.git synced 2025-09-15 19:47:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix schema

Browse Source
This commit is contained in:
matthewhyx 2023-12-28 00:47:46 +08:00
parent dd751d4463
commit 0f591c60a2
1 changed files with 14 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
python/knext/knext/examples/financial/builder/job/state_and_indicator.py
View File

@ -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)