trustgraph-cli 2.1.5__tar.gz → 2.1.6__tar.gz

{trustgraph_cli-2.1.5 → trustgraph_cli-2.1.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-cli
-Version: 2.1.5
+Version: 2.1.6
 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.1.5 → trustgraph_cli-2.1.6}/trustgraph/cli/invoke_document_embeddings.py

@@ -26,12 +26,14 @@ def query(url, flow_id, query_text, user, collection, limit, token=None):
             limit=limit
         )
 
-        chunk_ids = result.get("chunk_ids", [])
-        if not chunk_ids:
+        chunks = result.get("chunks", [])
+        if not chunks:
             print("No matching chunks found.")
         else:
-            for i, chunk_id in enumerate(chunk_ids, 1):
-                print(f"{i}. {chunk_id}")
+            for i, chunk in enumerate(chunks, 1):
+                chunk_id = chunk.get("chunk_id", "")
+                score = chunk.get("score", 0.0)
+                print(f"{i}. {chunk_id} (score: {score:.4f})")
 
     finally:
         # Clean up socket connection

{trustgraph_cli-2.1.5 → trustgraph_cli-2.1.6}/trustgraph/cli/invoke_graph_embeddings.py

@@ -27,8 +27,23 @@ def query(url, flow_id, query_text, user, collection, limit, token=None):
         )
 
         entities = result.get("entities", [])
-        for entity in entities:
-            print(entity)
+        if not entities:
+            print("No matching entities found.")
+        else:
+            for i, match in enumerate(entities, 1):
+                entity = match.get("entity", {})
+                score = match.get("score", 0.0)
+                # Format entity based on type (wire format uses compact keys)
+                term_type = entity.get("t", "")
+                if term_type == "i":  # IRI
+                    entity_str = entity.get("i", "")
+                elif term_type == "l":  # Literal
+                    entity_str = f'"{entity.get("v", "")}"'
+                elif term_type == "b":  # Blank node
+                    entity_str = f'_:{entity.get("d", "")}'
+                else:
+                    entity_str = str(entity)
+                print(f"{i}. {entity_str} (score: {score:.4f})")
 
     finally:
         # Clean up socket connection

trustgraph_cli-2.1.6/trustgraph/cli_version.py

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

{trustgraph_cli-2.1.5 → trustgraph_cli-2.1.6}/trustgraph_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-cli
-Version: 2.1.5
+Version: 2.1.6
 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.1.5/trustgraph/cli_version.py

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