trustgraph-cli 1.7.4__tar.gz → 1.8.12__tar.gz

{trustgraph_cli-1.7.4 → trustgraph_cli-1.8.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-cli
-Version: 1.7.4
+Version: 1.8.12
 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
@@ -8,7 +8,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
-Requires-Dist: trustgraph-base<1.8,>=1.7
+Requires-Dist: trustgraph-base<1.9,>=1.8
 Requires-Dist: requests
 Requires-Dist: pulsar-client
 Requires-Dist: aiohttp

{trustgraph_cli-1.7.4 → trustgraph_cli-1.8.12}/pyproject.toml

@@ -10,7 +10,7 @@ description = "TrustGraph provides a means to run a pipeline of flexible AI proc
 readme = "README.md"
 requires-python = ">=3.8"
 dependencies = [
-    "trustgraph-base>=1.7,<1.8",
+    "trustgraph-base>=1.8,<1.9",
     "requests",
     "pulsar-client",
     "aiohttp",

{trustgraph_cli-1.7.4 → trustgraph_cli-1.8.12}/trustgraph/cli/dump_queues.py

@@ -17,6 +17,7 @@ from datetime import datetime
 import argparse
 
 from trustgraph.base.subscriber import Subscriber
+from trustgraph.base.pubsub import get_pubsub
 
 def format_message(queue_name, msg):
     """Format a message with timestamp and queue name."""
@@ -167,11 +168,11 @@ async def async_main(queues, output_file, pulsar_host, listener_name, subscriber
     print(f"Mode: {'append' if append_mode else 'overwrite'}")
     print(f"Press Ctrl+C to stop\n")
 
-    # Connect to Pulsar
+    # Create backend connection
     try:
-        client = pulsar.Client(pulsar_host, listener_name=listener_name)
+        backend = get_pubsub(pulsar_host=pulsar_host, pulsar_listener=listener_name, pubsub_backend='pulsar')
     except Exception as e:
-        print(f"Error connecting to Pulsar at {pulsar_host}: {e}", file=sys.stderr)
+        print(f"Error connecting to backend at {pulsar_host}: {e}", file=sys.stderr)
         sys.exit(1)
 
     # Create Subscribers and central queue
@@ -181,7 +182,7 @@ async def async_main(queues, output_file, pulsar_host, listener_name, subscriber
     for queue_name in queues:
         try:
             sub = Subscriber(
-                client=client,
+                backend=backend,
                 topic=queue_name,
                 subscription=subscriber_name,
                 consumer_name=f"{subscriber_name}-{queue_name}",
@@ -195,7 +196,7 @@ async def async_main(queues, output_file, pulsar_host, listener_name, subscriber
 
     if not subscribers:
         print("\nNo subscribers created. Exiting.", file=sys.stderr)
-        client.close()
+        backend.close()
         sys.exit(1)
 
     print(f"\nListening for messages...\n")
@@ -256,7 +257,7 @@ async def async_main(queues, output_file, pulsar_host, listener_name, subscriber
         # Clean shutdown of Subscribers
         for _, sub in subscribers:
             await sub.stop()
-        client.close()
+        backend.close()
 
     print(f"\nMessages logged to: {output_file}")
 

{trustgraph_cli-1.7.4 → trustgraph_cli-1.8.12}/trustgraph/cli/invoke_agent.py

@@ -178,11 +178,22 @@ def question(
                 print()
 
         else:
-            # Non-streaming response
-            if "answer" in response:
-                print(response["answer"])
-            if "error" in response:
-                raise RuntimeError(response["error"])
+            # Non-streaming response - but agents use multipart messaging
+            # so we iterate through the chunks (which are complete messages, not text chunks)
+            for chunk in response:
+                # Display thoughts if verbose
+                if chunk.chunk_type == "thought" and verbose:
+                    output(wrap(chunk.content), "\U0001f914 ")
+                    print()
+
+                # Display observations if verbose
+                elif chunk.chunk_type == "observation" and verbose:
+                    output(wrap(chunk.content), "\U0001f4a1 ")
+                    print()
+
+                # Display answer
+                elif chunk.chunk_type == "final-answer" or chunk.chunk_type == "answer":
+                    print(chunk.content)
 
     finally:
         # Clean up socket connection

{trustgraph_cli-1.7.4 → trustgraph_cli-1.8.12}/trustgraph/cli/invoke_document_rag.py

@@ -24,7 +24,7 @@ def question(url, flow_id, question, user, collection, doc_limit, streaming=True
 
         try:
             response = flow.document_rag(
-                question=question,
+                query=question,
                 user=user,
                 collection=collection,
                 doc_limit=doc_limit,
@@ -42,7 +42,7 @@ def question(url, flow_id, question, user, collection, doc_limit, streaming=True
         # Use REST API for non-streaming
         flow = api.flow().id(flow_id)
         resp = flow.document_rag(
-            question=question,
+            query=question,
             user=user,
             collection=collection,
             doc_limit=doc_limit,

{trustgraph_cli-1.7.4 → trustgraph_cli-1.8.12}/trustgraph/cli/invoke_graph_rag.py

@@ -30,7 +30,7 @@ def question(
 
         try:
             response = flow.graph_rag(
-                question=question,
+                query=question,
                 user=user,
                 collection=collection,
                 entity_limit=entity_limit,
@@ -51,7 +51,7 @@ def question(
         # Use REST API for non-streaming
         flow = api.flow().id(flow_id)
         resp = flow.graph_rag(
-            question=question,
+            query=question,
             user=user,
             collection=collection,
             entity_limit=entity_limit,

{trustgraph_cli-1.7.4 → trustgraph_cli-1.8.12}/trustgraph/cli/list_collections.py

@@ -28,19 +28,17 @@ def list_collections(url, user, tag_filter):
             collection.collection,
             collection.name,
             collection.description,
-            ", ".join(collection.tags),
-            collection.created_at,
-            collection.updated_at
+            ", ".join(collection.tags)
         ])
 
-    headers = ["Collection", "Name", "Description", "Tags", "Created", "Updated"]
+    headers = ["Collection", "Name", "Description", "Tags"]
 
     print(tabulate.tabulate(
         table,
         headers=headers,
         tablefmt="pretty",
         stralign="left",
-        maxcolwidths=[20, 30, 50, 30, 19, 19],
+        maxcolwidths=[20, 30, 50, 30],
     ))
 
 def main():

{trustgraph_cli-1.7.4 → trustgraph_cli-1.8.12}/trustgraph/cli/set_collection.py

@@ -31,7 +31,6 @@ def set_collection(url, user, collection, name, description, tags, token=None):
         table.append(("Name", result.name))
         table.append(("Description", result.description))
         table.append(("Tags", ", ".join(result.tags)))
-        table.append(("Updated", result.updated_at))
 
         print(tabulate.tabulate(
             table,

trustgraph_cli-1.8.12/trustgraph/cli_version.py

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

{trustgraph_cli-1.7.4 → trustgraph_cli-1.8.12}/trustgraph_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-cli
-Version: 1.7.4
+Version: 1.8.12
 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
@@ -8,7 +8,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
-Requires-Dist: trustgraph-base<1.8,>=1.7
+Requires-Dist: trustgraph-base<1.9,>=1.8
 Requires-Dist: requests
 Requires-Dist: pulsar-client
 Requires-Dist: aiohttp

{trustgraph_cli-1.7.4 → trustgraph_cli-1.8.12}/trustgraph_cli.egg-info/requires.txt

@@ -1,4 +1,4 @@
-trustgraph-base<1.8,>=1.7
+trustgraph-base<1.9,>=1.8
 requests
 pulsar-client
 aiohttp

trustgraph_cli-1.7.4/trustgraph/cli_version.py

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