trustgraph-cli 2.10.6__tar.gz → 2.10.8__tar.gz

{trustgraph_cli-2.10.6 → trustgraph_cli-2.10.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-cli
-Version: 2.10.6
+Version: 2.10.8
 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.6 → trustgraph_cli-2.10.8}/pyproject.toml

@@ -60,6 +60,7 @@ tg-revoke-api-key = "trustgraph.cli.revoke_api_key:main"
 tg-list-workspaces = "trustgraph.cli.list_workspaces:main"
 tg-list-my-workspaces = "trustgraph.cli.list_my_workspaces:main"
 tg-create-workspace = "trustgraph.cli.create_workspace:main"
+tg-list-models = "trustgraph.cli.list_models:main"
 tg-invoke-agent = "trustgraph.cli.invoke_agent:main"
 tg-invoke-document-rag = "trustgraph.cli.invoke_document_rag:main"
 tg-invoke-graph-rag = "trustgraph.cli.invoke_graph_rag:main"

{trustgraph_cli-2.10.6 → trustgraph_cli-2.10.8}/trustgraph/cli/invoke_graph_rag.py

@@ -50,7 +50,7 @@ def _question_explainable_api(
         edge_limit=25, max_reranker_input=350, token=None, debug=False,
         workspace="default",
         grounding_seeds=None, graph_seeds=None, languages=None,
-        traversal_instructions=None,
+        traversal_instructions=None, graph=None,
 ):
     """Execute graph RAG with explainability using the new API classes."""
     api = Api(url=url, token=token, workspace=workspace)
@@ -76,6 +76,7 @@ def _question_explainable_api(
             graph_seeds=graph_seeds,
             languages=languages,
             traversal_instructions=traversal_instructions,
+            graph=graph,
         ):
             if isinstance(item, RAGChunk):
                 # Print response content
@@ -172,7 +173,7 @@ def question(
         explainable=False, debug=False, show_usage=False,
         workspace="default",
         grounding_seeds=None, graph_seeds=None, languages=None,
-        traversal_instructions=None,
+        traversal_instructions=None, graph=None,
 ):
 
     # Explainable mode uses the API to capture and process provenance events
@@ -196,6 +197,7 @@ def question(
             graph_seeds=graph_seeds,
             languages=languages,
             traversal_instructions=traversal_instructions,
+            graph=graph,
         )
         return
 
@@ -223,6 +225,7 @@ def question(
                 graph_seeds=graph_seeds,
                 languages=languages,
                 traversal_instructions=traversal_instructions,
+                graph=graph,
             )
 
             # Stream output
@@ -262,6 +265,7 @@ def question(
             graph_seeds=graph_seeds,
             languages=languages,
             traversal_instructions=traversal_instructions,
+            graph=graph,
         )
         print(result.text)
 
@@ -419,6 +423,13 @@ def main():
         help='JSON list of per-hop traversal restrictions (e.g. \'[{"types": ["http://example.org/Person"]}, {"relationships": ["http://example.org/worksFor"]}]\')'
     )
 
+    parser.add_argument(
+        '-g', '--graph',
+        type=str,
+        default=None,
+        help='Named graph IRI to query (default: all graphs)'
+    )
+
     args = parser.parse_args()
 
     try:
@@ -462,6 +473,7 @@ def main():
             graph_seeds=graph_seeds,
             languages=languages,
             traversal_instructions=traversal_instructions,
+            graph=args.graph,
         )
 
     except Exception as e:

trustgraph_cli-2.10.8/trustgraph/cli/list_models.py

@@ -0,0 +1,112 @@
+"""
+Lists available models from the text-completion backend
+"""
+
+import argparse
+import json
+import os
+import sys
+from trustgraph.api import Api
+
+default_url = os.getenv("TRUSTGRAPH_URL", 'http://localhost:8888/')
+default_token = os.getenv("TRUSTGRAPH_TOKEN", None)
+default_workspace = os.getenv("TRUSTGRAPH_WORKSPACE", "default")
+
+
+def main():
+
+    parser = argparse.ArgumentParser(
+        prog='tg-list-models',
+        description=__doc__,
+    )
+
+    parser.add_argument(
+        '-u', '--url',
+        default=default_url,
+        help=f'API URL (default: {default_url})',
+    )
+
+    parser.add_argument(
+        '-t', '--token',
+        default=default_token,
+        help='Authentication token (default: $TRUSTGRAPH_TOKEN)',
+    )
+
+    parser.add_argument(
+        '-w', '--workspace',
+        default=default_workspace,
+        help=f'Workspace (default: {default_workspace})',
+    )
+
+    parser.add_argument(
+        '-f', '--flow-id',
+        default="default",
+        help='Flow ID (default: default)',
+    )
+
+    parser.add_argument(
+        '-j', '--json',
+        action='store_true',
+        help='Output as JSON',
+    )
+
+    args = parser.parse_args()
+
+    try:
+
+        api = Api(url=args.url, token=args.token, workspace=args.workspace)
+        socket = api.socket()
+        flow = socket.flow(args.flow_id)
+
+        try:
+            models = flow.list_models()
+        finally:
+            socket.close()
+
+        if args.json:
+            print(json.dumps(models, indent=2))
+        else:
+            if not models:
+                print("No models available (or backend does not support listing).", file=sys.stderr)
+                return
+
+            for m in models:
+                model_id = m.get('id', '?')
+                parts = [model_id]
+
+                name = m.get('name')
+                if name:
+                    parts.append(f"({name})")
+
+                owner = m.get('owned_by')
+                if owner:
+                    parts.append(f"[{owner}]")
+
+                ctx = m.get('context_length')
+                if ctx:
+                    parts.append(f"ctx={ctx}")
+
+                max_out = m.get('max_output_length')
+                if max_out:
+                    parts.append(f"max_out={max_out}")
+
+                family = m.get('family')
+                if family:
+                    parts.append(f"family={family}")
+
+                param = m.get('parameter_size')
+                if param:
+                    parts.append(f"params={param}")
+
+                quant = m.get('quantization')
+                if quant:
+                    parts.append(f"quant={quant}")
+
+                print("  ".join(parts))
+
+    except Exception as e:
+        print("Exception:", e, flush=True)
+
+
+if __name__ == "__main__":
+    main()

{trustgraph_cli-2.10.6 → trustgraph_cli-2.10.8}/trustgraph/cli/load_knowledge.py

@@ -62,7 +62,8 @@ class KnowledgeLoader:
             ds = rdflib.Dataset()
             ds.parse(file, format="trig")
             for s, p, o, g in ds.quads((None, None, None, None)):
-                graph = str(g.identifier) if g.identifier else ""
+                graph_id = g.identifier if hasattr(g, 'identifier') else g
+                graph = str(graph_id) if graph_id else ""
                 if graph == "urn:x-rdflib:default":
                     graph = ""
                 yield self._make_triple(s, p, o, graph)

trustgraph_cli-2.10.8/trustgraph/cli_version.py

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

{trustgraph_cli-2.10.6 → trustgraph_cli-2.10.8}/trustgraph_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-cli
-Version: 2.10.6
+Version: 2.10.8
 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.6 → trustgraph_cli-2.10.8}/trustgraph_cli.egg-info/SOURCES.txt

@@ -51,6 +51,7 @@ trustgraph/cli/list_api_keys.py
 trustgraph/cli/list_collections.py
 trustgraph/cli/list_config_items.py
 trustgraph/cli/list_explain_traces.py
+trustgraph/cli/list_models.py
 trustgraph/cli/list_my_workspaces.py
 trustgraph/cli/list_users.py
 trustgraph/cli/list_workspaces.py

{trustgraph_cli-2.10.6 → trustgraph_cli-2.10.8}/trustgraph_cli.egg-info/entry_points.txt

@@ -46,6 +46,7 @@ tg-list-api-keys = trustgraph.cli.list_api_keys:main
 tg-list-collections = trustgraph.cli.list_collections:main
 tg-list-config-items = trustgraph.cli.list_config_items:main
 tg-list-explain-traces = trustgraph.cli.list_explain_traces:main
+tg-list-models = trustgraph.cli.list_models:main
 tg-list-my-workspaces = trustgraph.cli.list_my_workspaces:main
 tg-list-users = trustgraph.cli.list_users:main
 tg-list-workspaces = trustgraph.cli.list_workspaces:main

trustgraph_cli-2.10.6/trustgraph/cli_version.py

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