trustgraph-cli 2.2.9__tar.gz → 2.2.11__tar.gz

{trustgraph_cli-2.2.9 → trustgraph_cli-2.2.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-cli
-Version: 2.2.9
+Version: 2.2.11
 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.2.9 → trustgraph_cli-2.2.11}/trustgraph/cli/dump_queues.py

@@ -19,43 +19,67 @@ import argparse
 from trustgraph.base.subscriber import Subscriber
 from trustgraph.base.pubsub import get_pubsub
 
+def decode_json_strings(obj):
+    """Recursively decode JSON-encoded string values within a dict/list."""
+    if isinstance(obj, dict):
+        return {k: decode_json_strings(v) for k, v in obj.items()}
+    if isinstance(obj, list):
+        return [decode_json_strings(v) for v in obj]
+    if isinstance(obj, str):
+        try:
+            parsed = json.loads(obj)
+            if isinstance(parsed, (dict, list)):
+                return decode_json_strings(parsed)
+        except (json.JSONDecodeError, TypeError):
+            pass
+    return obj
+
+
+def to_dict(value):
+    """Recursively convert a value to a JSON-serialisable structure."""
+
+    if value is None or isinstance(value, (bool, int, float)):
+        return value
+
+    if isinstance(value, bytes):
+        value = value.decode('utf-8')
+
+    if isinstance(value, str):
+        try:
+            return json.loads(value)
+        except (json.JSONDecodeError, TypeError):
+            return value
+
+    if isinstance(value, dict):
+        return {k: to_dict(v) for k, v in value.items()}
+
+    if isinstance(value, (list, tuple)):
+        return [to_dict(v) for v in value]
+
+    # Pulsar schema objects expose fields via __dict__
+    if hasattr(value, '__dict__'):
+        return {
+            k: to_dict(v) for k, v in value.__dict__.items()
+            if not k.startswith('_')
+        }
+
+    return str(value)
+
+
 def format_message(queue_name, msg):
     """Format a message with timestamp and queue name."""
     timestamp = datetime.now().isoformat()
 
-    # Try to parse as JSON and pretty-print
     try:
-        # Handle both Message objects and raw bytes
-        if hasattr(msg, 'value'):
-            # Message object with .value() method
-            value = msg.value()
-        else:
-            # Raw bytes from schema-less subscription
-            value = msg
-
-        # If it's bytes, decode it
-        if isinstance(value, bytes):
-            value = value.decode('utf-8')
+        value = msg.value() if hasattr(msg, 'value') else msg
+        parsed = to_dict(value)
 
-        # If it's a string, try to parse as JSON
-        if isinstance(value, str):
-            try:
-                parsed = json.loads(value)
-                body = json.dumps(parsed, indent=2)
-            except (json.JSONDecodeError, TypeError):
-                body = value
+        # Unwrap nested JSON strings (e.g. terms values)
+        if isinstance(parsed, (dict, list)):
+            parsed = decode_json_strings(parsed)
+            body = json.dumps(parsed, indent=2, default=str)
         else:
-            # Try to convert to dict for pretty printing
-            try:
-                # Pulsar schema objects have __dict__ or similar
-                if hasattr(value, '__dict__'):
-                    parsed = {k: v for k, v in value.__dict__.items()
-                             if not k.startswith('_')}
-                else:
-                    parsed = str(value)
-                body = json.dumps(parsed, indent=2, default=str)
-            except (TypeError, AttributeError):
-                body = str(value)
+            body = str(parsed)
 
     except Exception as e:
         body = f"<Error formatting message: {e}>\n{str(msg)}"

{trustgraph_cli-2.2.9 → trustgraph_cli-2.2.11}/trustgraph/cli/invoke_agent.py

@@ -13,6 +13,11 @@ from trustgraph.api import (
     Question,
     Analysis,
     Conclusion,
+    Decomposition,
+    Finding,
+    Plan,
+    StepResult,
+    Synthesis,
     AgentThought,
     AgentObservation,
     AgentAnswer,
@@ -209,6 +214,35 @@ def question_explainable(
                     if entity.observation:
                         print(f"    Observation: {entity.observation}", file=sys.stderr)
 
+                elif isinstance(entity, Decomposition):
+                    print(f"\n  [decompose] {prov_id}", file=sys.stderr)
+                    for i, goal in enumerate(entity.goals):
+                        print(f"    Thread {i}: {goal}", file=sys.stderr)
+
+                elif isinstance(entity, Finding):
+                    print(f"\n  [finding] {prov_id}", file=sys.stderr)
+                    if entity.goal:
+                        print(f"    Goal: {entity.goal}", file=sys.stderr)
+                    if entity.document:
+                        print(f"    Document: {entity.document}", file=sys.stderr)
+
+                elif isinstance(entity, Plan):
+                    print(f"\n  [plan] {prov_id}", file=sys.stderr)
+                    for i, step in enumerate(entity.steps):
+                        print(f"    Step {i}: {step}", file=sys.stderr)
+
+                elif isinstance(entity, StepResult):
+                    print(f"\n  [step-result] {prov_id}", file=sys.stderr)
+                    if entity.step:
+                        print(f"    Step: {entity.step}", file=sys.stderr)
+                    if entity.document:
+                        print(f"    Document: {entity.document}", file=sys.stderr)
+
+                elif isinstance(entity, Synthesis):
+                    print(f"\n  [synthesis] {prov_id}", file=sys.stderr)
+                    if entity.document:
+                        print(f"    Document: {entity.document}", file=sys.stderr)
+
                 elif isinstance(entity, Conclusion):
                     print(f"\n  [conclusion] {prov_id}", file=sys.stderr)
                     if entity.document:

{trustgraph_cli-2.2.9 → trustgraph_cli-2.2.11}/trustgraph/cli/monitor_prompts.py

@@ -24,7 +24,7 @@ from pulsar.schema import BytesSchema
 
 
 default_flow = "default"
-default_queue_type = "prompt"
+default_queue_type = "prompt-rag"
 default_max_lines = 3
 default_max_width = 80
 

trustgraph_cli-2.2.11/trustgraph/cli_version.py

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

{trustgraph_cli-2.2.9 → trustgraph_cli-2.2.11}/trustgraph_cli.egg-info/PKG-INFO

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

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