apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-23 04:01:02 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Value assignment fixes

Browse source
This commit is contained in:
Cyber MacGeddon 2025-07-21 11:17:17 +01:00
parent f02a30a848
commit 0809a4a145
1 changed files with 40 additions and 31 deletions
Download patch file Download diff file Expand all files Collapse all files

71
trustgraph-flow/trustgraph/extract/kg/agent/extract.py
View file

@ -100,12 +100,16 @@ class Processor(FlowProcessor):
tpls = Triples() tpls = Triples()
tpls.metadata = metadata tpls.metadata = metadata
tpls.triples = triples tpls.triples = triples
print("triples: ", tpls, flush=True)
await pub.send(tpls) await pub.send(tpls)
async def emit_entity_contexts(self, pub, metadata, entity_contexts): async def emit_entity_contexts(self, pub, metadata, entity_contexts):
ecs = EntityContexts() ecs = EntityContexts()
ecs.metadata = metadata ecs.metadata = metadata
ecs.entities = entity_contexts ecs.entities = entity_contexts
print("entity contexts: ", ecs, flush=True)
await pub.send(ecs) await pub.send(ecs)
def parse_json(self, text): def parse_json(self, text):
@ -174,6 +178,12 @@ class Processor(FlowProcessor):
triples, entity_contexts = self.process_extraction_data( triples, entity_contexts = self.process_extraction_data(
extraction_data, v.metadata extraction_data, v.metadata
) )
for tpl in triples:
print("TRIPLE>>>", tpl.s, tpl.p, tpl.o)
for ec in entity_contexts:
print("ENTITY CONTEXT>>>", tpl)
# Emit outputs # Emit outputs
if triples: if triples:
@ -202,24 +212,24 @@ class Processor(FlowProcessor):
# Add entity label # Add entity label
triples.append(Triple( triples.append(Triple(
subject=Value(value=entity_uri, is_uri=True), s = Value(value=entity_uri, is_uri=True),
predicate=Value(value=RDF_LABEL, is_uri=True), p = Value(value=RDF_LABEL, is_uri=True),
object=Value(value=defn["entity"], is_uri=False), o = Value(value=defn["entity"], is_uri=False),
)) ))
# Add definition # Add definition
triples.append(Triple( triples.append(Triple(
subject=Value(value=entity_uri, is_uri=True), s = Value(value=entity_uri, is_uri=True),
predicate=Value(value=DEFINITION, is_uri=True), p = Value(value=DEFINITION, is_uri=True),
object=Value(value=defn["definition"], is_uri=False), o = Value(value=defn["definition"], is_uri=False),
)) ))
# Add subject-of relationship to document # Add subject-of relationship to document
if metadata.id: if metadata.id:
triples.append(Triple( triples.append(Triple(
subject=Value(value=entity_uri, is_uri=True), s = Value(value=entity_uri, is_uri=True),
predicate=Value(value=SUBJECT_OF, is_uri=True), p = Value(value=SUBJECT_OF, is_uri=True),
object=metadata.id, o = Value(value=metadata.id, is_uri=True),
)) ))
# Create entity context for embeddings # Create entity context for embeddings
@ -243,52 +253,51 @@ class Processor(FlowProcessor):
# Add subject and predicate labels # Add subject and predicate labels
triples.append(Triple( triples.append(Triple(
subject=subject_value, s = subject_value,
predicate=Value(value=RDF_LABEL, is_uri=True), p = Value(value=RDF_LABEL, is_uri=True),
object=Value(value=rel["subject"], is_uri=False), o = Value(value=rel["subject"], is_uri=False),
)) ))
triples.append(Triple( triples.append(Triple(
subject=predicate_value, s = predicate_value,
predicate=Value(value=RDF_LABEL, is_uri=True), p = Value(value=RDF_LABEL, is_uri=True),
object=Value(value=rel["predicate"], is_uri=False), o = Value(value=rel["predicate"], is_uri=False),
)) ))
# Handle object (entity vs literal) # Handle object (entity vs literal)
if rel.get("object-entity", True): if rel.get("object-entity", True):
object_uri = self.to_uri(rel["object"])
triples.append(Triple( triples.append(Triple(
subject=object_value, s = object_value,
predicate=Value(value=RDF_LABEL, is_uri=True), p = Value(value=RDF_LABEL, is_uri=True),
object=rel["object"], o = Value(value=rel["object"], is_uri=True),
)) ))
# Add the main relationship triple # Add the main relationship triple
triples.append(Triple( triples.append(Triple(
subject=subject_value, s = subject_value,
predicate=predicate_value, p = predicate_value,
object=object_value o = object_value
)) ))
# Add subject-of relationships to document # Add subject-of relationships to document
if metadata.id: if metadata.id:
triples.append(Triple( triples.append(Triple(
subject=subject_value, s = subject_value,
predicate=Value(value=SUBJECT_OF, is_uri=True), p = Value(value=SUBJECT_OF, is_uri=True),
object=metadata.id, o = Value(value=metadata.id, is_uri=True),
)) ))
triples.append(Triple( triples.append(Triple(
subject=predicate_value, s = predicate_value,
predicate=Value(SUBJECT_OF, is_uri=True), p = Value(value=SUBJECT_OF, is_uri=True),
object=metadata.id, o = Value(value=metadata.id, is_uri=True),
)) ))
if rel.get("object-entity", True): if rel.get("object-entity", True):
triples.append(Triple( triples.append(Triple(
subject=object_value, s = object_value,
predicate=Value(SUBJECT_OF, is_uri=True), p = Value(value=SUBJECT_OF, is_uri=True),
object=metadata.id, o = Value(value=metadata.id, is_uri=True),
)) ))
return triples, entity_contexts return triples, entity_contexts