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

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

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

@@ -10,7 +10,7 @@ from trustgraph.api import Api
 default_url = os.getenv("TRUSTGRAPH_URL", 'http://localhost:8088/')
 default_token = os.getenv("TRUSTGRAPH_TOKEN", None)
 
-def query(url, flow_id, text, token=None):
+def query(url, flow_id, texts, token=None):
 
     # Create API client
     api = Api(url=url, token=token)
@@ -19,9 +19,14 @@ def query(url, flow_id, text, token=None):
 
     try:
         # Call embeddings service
-        result = flow.embeddings(text=text)
+        result = flow.embeddings(texts=texts)
         vectors = result.get("vectors", [])
-        print(vectors)
+        # Print each text's vectors
+        for i, vecs in enumerate(vectors):
+            if len(texts) > 1:
+                print(f"Text {i + 1}: {vecs}")
+            else:
+                print(vecs)
 
     finally:
         # Clean up socket connection
@@ -53,9 +58,9 @@ def main():
     )
 
     parser.add_argument(
-        'text',
-        nargs=1,
-        help='Text to convert to embedding vector',
+        'texts',
+        nargs='+',
+        help='Text(s) to convert to embedding vectors',
     )
 
     args = parser.parse_args()
@@ -65,7 +70,7 @@ def main():
         query(
             url=args.url,
             flow_id=args.flow_id,
-            text=args.text[0],
+            texts=args.texts,
             token=args.token,
         )
 

{trustgraph_cli-2.1.4 → 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.4 → 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.4
+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.4/trustgraph/cli_version.py

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