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

Fix extraction bugs

Browse source
This commit is contained in:
Cyber MacGeddon 2025-04-19 14:17:57 +01:00
parent c05ad5c215
commit a8408f2c74
2 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
trustgraph-flow/trustgraph/extract/kg/relationships/extract.py
View file

@ -150,7 +150,7 @@ class Processor(FlowProcessor):
o=Value(value=str(p), is_uri=False)
))
if rel["o_entity"]:
if rel["object-entity"]:
# Label for o
triples.append(Triple(
s=o_value,
@ -165,7 +165,7 @@ class Processor(FlowProcessor):
o=Value(value=v.metadata.id, is_uri=True)
))
if rel["o_entity"]:
if rel["object-entity"]:
# 'Subject of' for o
triples.append(Triple(
s=o_value,

4
trustgraph-flow/trustgraph/model/prompt/template/service.py
View file

@ -13,7 +13,7 @@ from .... schema import PromptRequest, PromptResponse, Error
from .... schema import TextCompletionRequest, TextCompletionResponse
from .... base import FlowProcessor
from .... base import ProducerSpec, ConsumerSpec, TextCompletionSpec
from .... base import ProducerSpec, ConsumerSpec, TextCompletionClientSpec
from . prompt_manager import PromptConfiguration, Prompt, PromptManager
@ -43,7 +43,7 @@ class Processor(FlowProcessor):
)
self.register_specification(
TextCompletionSpec(
TextCompletionClientSpec(
request_name = "text-completion-request",
response_name = "text-completion-response",
)