trustgraph-cli 2.10.0__tar.gz → 2.10.2__tar.gz

{trustgraph_cli-2.10.0 → trustgraph_cli-2.10.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-cli
-Version: 2.10.0
+Version: 2.10.2
 Summary: TrustGraph provides a means to run a pipeline of flexible AI processing components in a flexible means to achieve a processing pipeline.
 Author-email: "trustgraph.ai" <security@trustgraph.ai>
 Project-URL: Homepage, https://github.com/trustgraph-ai/trustgraph

{trustgraph_cli-2.10.0 → trustgraph_cli-2.10.2}/trustgraph/cli/load_knowledge.py

@@ -9,6 +9,7 @@ import rdflib
 from typing import Iterator, Tuple
 
 from trustgraph.api import Api, Triple
+from trustgraph.api.types import Uri, Literal
 from trustgraph.log_level import LogLevel
 
 default_url = os.getenv("TRUSTGRAPH_URL", 'http://localhost:8888/')
@@ -26,6 +27,7 @@ class KnowledgeLoader:
             document_id,
             url=default_url,
             token=None, workspace="default",
+            graph="",
     ):
         self.files = files
         self.flow = flow
@@ -34,6 +36,7 @@ class KnowledgeLoader:
         self.url = url
         self.token = token
         self.workspace = workspace
+        self.graph = graph
 
     def load_triples_from_file(self, file) -> Iterator[Triple]:
         """Generator that yields Triple objects from a Turtle file"""
@@ -42,23 +45,19 @@ class KnowledgeLoader:
         g.parse(file, format="turtle")
 
         for e in g:
-            s_value = str(e[0])
-            p_value = str(e[1])
-            o_value = str(e[2])
-
-            o_datatype = ""
-            o_language = ""
+            s_value = Uri(str(e[0]))
+            p_value = Uri(str(e[1]))
 
             if isinstance(e[2], rdflib.term.Literal):
-                if e[2].language:
-                    o_language = str(e[2].language)
-                if e[2].datatype:
-                    o_datatype = str(e[2].datatype)
-
-            yield Triple(
-                s=s_value, p=p_value, o=o_value,
-                o_datatype=o_datatype, o_language=o_language,
-            )
+                o_value = Literal(
+                    str(e[2]),
+                    datatype=str(e[2].datatype) if e[2].datatype else None,
+                    language=str(e[2].language) if e[2].language else None,
+                )
+            else:
+                o_value = Uri(str(e[2]))
+
+            yield Triple(s=s_value, p=p_value, o=o_value, g=self.graph)
 
     def load_entity_contexts_from_file(self, file) -> Iterator[Tuple[str, str]]:
         """Generator that yields (entity, context) tuples from a Turtle file"""
@@ -184,6 +183,12 @@ def main():
         help=f'Collection ID (default: {default_collection})'
     )
 
+    parser.add_argument(
+        '-g', '--graph',
+        default="",
+        help='Named graph URI (e.g. urn:graph:policy). Default: default graph'
+    )
+
     parser.add_argument(
         'files', nargs='+',
         help=f'Turtle files to load'
@@ -202,6 +207,7 @@ def main():
                 files=args.files,
                 collection=args.collection,
                 workspace=args.workspace,
+                graph=args.graph,
             )
 
             loader.run()

{trustgraph_cli-2.10.0 → trustgraph_cli-2.10.2}/trustgraph/cli/load_turtle.py

@@ -9,6 +9,7 @@ import rdflib
 from typing import Iterator
 
 from trustgraph.api import Api, Triple
+from trustgraph.api.types import Uri, Literal
 from trustgraph.log_level import LogLevel
 
 default_url = os.getenv("TRUSTGRAPH_URL", 'http://localhost:8888/')
@@ -42,23 +43,19 @@ class Loader:
         g.parse(file, format="turtle")
 
         for e in g:
-            s_value = str(e[0])
-            p_value = str(e[1])
-            o_value = str(e[2])
-
-            o_datatype = ""
-            o_language = ""
+            s_value = Uri(str(e[0]))
+            p_value = Uri(str(e[1]))
 
             if isinstance(e[2], rdflib.term.Literal):
-                if e[2].language:
-                    o_language = str(e[2].language)
-                if e[2].datatype:
-                    o_datatype = str(e[2].datatype)
-
-            yield Triple(
-                s=s_value, p=p_value, o=o_value,
-                o_datatype=o_datatype, o_language=o_language,
-            )
+                o_value = Literal(
+                    str(e[2]),
+                    datatype=str(e[2].datatype) if e[2].datatype else None,
+                    language=str(e[2].language) if e[2].language else None,
+                )
+            else:
+                o_value = Uri(str(e[2]))
+
+            yield Triple(s=s_value, p=p_value, o=o_value)
 
     def run(self):
         """Load triples using Python API"""

{trustgraph_cli-2.10.0 → trustgraph_cli-2.10.2}/trustgraph/cli/nquads.py

@@ -13,7 +13,7 @@ import re
 import rdflib
 
 from trustgraph.schema import IRI, LITERAL
-from trustgraph.api.types import Triple
+from trustgraph.api.types import Triple, Uri, Literal
 
 # RDF-star quoted triples have no standard N-Quads encoding; they are
 # skipped with a count so callers can surface the omission.
@@ -150,22 +150,24 @@ def serialize_nquads(batches, graph_iri, out):
     return written, skipped
 
 
-def _term_to_triple_field(term):
-    """Convert an rdflib term to (str_value, datatype, language)."""
+def _rdflib_to_object(term):
+    """Convert an rdflib term to a Uri or Literal."""
     if isinstance(term, rdflib.Literal):
-        dt = str(term.datatype) if term.datatype else ""
-        lang = str(term.language) if term.language else ""
-        return str(term), dt, lang
-    return str(term), "", ""
+        return Literal(
+            str(term),
+            datatype=str(term.datatype) if term.datatype else None,
+            language=str(term.language) if term.language else None,
+        )
+    return Uri(str(term))
 
 
 def parse_nquads(data):
     """Parse N-Quads bytes back into api Triple values.
 
     Preserves datatype, language tag and IRI-vs-literal distinctions
-    via the Triple.o_datatype and Triple.o_language fields. The whole
-    member is materialized in memory (bundles are bounded by
-    --triples-limit at export); line-streaming is a possible follow-up.
+    via Literal's datatype and language attributes. The whole member is
+    materialized in memory (bundles are bounded by --triples-limit at
+    export); line-streaming is a possible follow-up.
 
     :param data: N-Quads bytes (one bundle member)
     :returns: list of Triple
@@ -174,9 +176,9 @@ def parse_nquads(data):
     ds.parse(data=data.decode("utf-8"), format="nquads")
     result = []
     for s, p, o, _g in ds.quads((None, None, None, None)):
-        o_val, o_dt, o_lang = _term_to_triple_field(o)
         result.append(Triple(
-            s=str(s), p=str(p), o=o_val,
-            o_datatype=o_dt, o_language=o_lang,
+            s=Uri(str(s)),
+            p=Uri(str(p)),
+            o=_rdflib_to_object(o),
         ))
     return result

trustgraph_cli-2.10.2/trustgraph/cli_version.py

@@ -0,0 +1 @@
+__version__ = "2.10.2"

{trustgraph_cli-2.10.0 → trustgraph_cli-2.10.2}/trustgraph_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-cli
-Version: 2.10.0
+Version: 2.10.2
 Summary: TrustGraph provides a means to run a pipeline of flexible AI processing components in a flexible means to achieve a processing pipeline.
 Author-email: "trustgraph.ai" <security@trustgraph.ai>
 Project-URL: Homepage, https://github.com/trustgraph-ai/trustgraph

trustgraph_cli-2.10.0/trustgraph/cli_version.py

@@ -1 +0,0 @@
-__version__ = "2.10.0"