trustgraph-cli 2.10.2__tar.gz → 2.10.3__tar.gz

{trustgraph_cli-2.10.2 → trustgraph_cli-2.10.3}/PKG-INFO

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

@@ -44,6 +44,7 @@ tg-get-document-content = "trustgraph.cli.get_document_content:main"
 tg-graph-to-turtle = "trustgraph.cli.graph_to_turtle:main"
 tg-bootstrap-iam = "trustgraph.cli.bootstrap_iam:main"
 tg-login = "trustgraph.cli.login:main"
+tg-mint-token = "trustgraph.cli.mint_token:main"
 tg-create-user = "trustgraph.cli.create_user:main"
 tg-list-users = "trustgraph.cli.list_users:main"
 tg-whoami = "trustgraph.cli.whoami:main"

{trustgraph_cli-2.10.2 → trustgraph_cli-2.10.3}/trustgraph/cli/invoke_graph_rag.py

@@ -3,6 +3,7 @@ Uses the GraphRAG service to answer a question
 """
 
 import argparse
+import json
 import os
 import sys
 from trustgraph.api import (
@@ -48,6 +49,8 @@ def _question_explainable_api(
         max_subgraph_size, max_path_length, edge_score_limit=30,
         edge_limit=25, max_reranker_input=350, token=None, debug=False,
         workspace="default",
+        grounding_seeds=None, graph_seeds=None, languages=None,
+        traversal_instructions=None,
 ):
     """Execute graph RAG with explainability using the new API classes."""
     api = Api(url=url, token=token, workspace=workspace)
@@ -61,7 +64,7 @@ def _question_explainable_api(
         # Stream GraphRAG with explainability - process events as they arrive
         for item in flow.graph_rag_explain(
             query=question_text,
-                        collection=collection,
+            collection=collection,
             entity_limit=entity_limit,
             triple_limit=triple_limit,
             max_subgraph_size=max_subgraph_size,
@@ -69,6 +72,10 @@ def _question_explainable_api(
             edge_score_limit=edge_score_limit,
             edge_limit=edge_limit,
             max_reranker_input=max_reranker_input,
+            grounding_seeds=grounding_seeds,
+            graph_seeds=graph_seeds,
+            languages=languages,
+            traversal_instructions=traversal_instructions,
         ):
             if isinstance(item, RAGChunk):
                 # Print response content
@@ -164,6 +171,8 @@ def question(
         edge_limit=25, max_reranker_input=350, streaming=True, token=None,
         explainable=False, debug=False, show_usage=False,
         workspace="default",
+        grounding_seeds=None, graph_seeds=None, languages=None,
+        traversal_instructions=None,
 ):
 
     # Explainable mode uses the API to capture and process provenance events
@@ -172,7 +181,7 @@ def question(
             url=url,
             flow_id=flow_id,
             question_text=question,
-                        collection=collection,
+            collection=collection,
             entity_limit=entity_limit,
             triple_limit=triple_limit,
             max_subgraph_size=max_subgraph_size,
@@ -183,6 +192,10 @@ def question(
             token=token,
             debug=debug,
             workspace=workspace,
+            grounding_seeds=grounding_seeds,
+            graph_seeds=graph_seeds,
+            languages=languages,
+            traversal_instructions=traversal_instructions,
         )
         return
 
@@ -197,7 +210,7 @@ def question(
         try:
             response = flow.graph_rag(
                 query=question,
-                                collection=collection,
+                collection=collection,
                 entity_limit=entity_limit,
                 triple_limit=triple_limit,
                 max_subgraph_size=max_subgraph_size,
@@ -205,7 +218,11 @@ def question(
                 edge_score_limit=edge_score_limit,
                 edge_limit=edge_limit,
                 max_reranker_input=max_reranker_input,
-                streaming=True
+                streaming=True,
+                grounding_seeds=grounding_seeds,
+                graph_seeds=graph_seeds,
+                languages=languages,
+                traversal_instructions=traversal_instructions,
             )
 
             # Stream output
@@ -233,7 +250,7 @@ def question(
         flow = api.flow().id(flow_id)
         result = flow.graph_rag(
             query=question,
-                        collection=collection,
+            collection=collection,
             entity_limit=entity_limit,
             triple_limit=triple_limit,
             max_subgraph_size=max_subgraph_size,
@@ -241,6 +258,10 @@ def question(
             edge_score_limit=edge_score_limit,
             edge_limit=edge_limit,
             max_reranker_input=max_reranker_input,
+            grounding_seeds=grounding_seeds,
+            graph_seeds=graph_seeds,
+            languages=languages,
+            traversal_instructions=traversal_instructions,
         )
         print(result.text)
 
@@ -370,10 +391,55 @@ def main():
         help='Show token usage and model on stderr'
     )
 
+    parser.add_argument(
+        '--grounding-seeds',
+        type=str,
+        default=None,
+        help='JSON list of concept strings to bypass LLM grounding (e.g. \'["graphrag", "retrieval"]\')'
+    )
+
+    parser.add_argument(
+        '--graph-seeds',
+        type=str,
+        default=None,
+        help='JSON list of entity IRIs as traversal starting points (e.g. \'["http://example.org/Person/Jane"]\')'
+    )
+
+    parser.add_argument(
+        '--languages',
+        type=str,
+        default=None,
+        help='JSON list of language tags to filter by (e.g. \'["en", ""]\')'
+    )
+
+    parser.add_argument(
+        '--traversal-instructions',
+        type=str,
+        default=None,
+        help='JSON list of per-hop traversal restrictions (e.g. \'[{"types": ["http://example.org/Person"]}, {"relationships": ["http://example.org/worksFor"]}]\')'
+    )
+
     args = parser.parse_args()
 
     try:
 
+        grounding_seeds = (
+            json.loads(args.grounding_seeds)
+            if args.grounding_seeds else None
+        )
+        graph_seeds = (
+            json.loads(args.graph_seeds)
+            if args.graph_seeds else None
+        )
+        languages = (
+            json.loads(args.languages)
+            if args.languages else None
+        )
+        traversal_instructions = (
+            json.loads(args.traversal_instructions)
+            if args.traversal_instructions else None
+        )
+
         question(
             url=args.url,
             flow_id=args.flow_id,
@@ -392,6 +458,10 @@ def main():
             debug=args.debug,
             show_usage=args.show_usage,
             workspace=args.workspace,
+            grounding_seeds=grounding_seeds,
+            graph_seeds=graph_seeds,
+            languages=languages,
+            traversal_instructions=traversal_instructions,
         )
 
     except Exception as e:

trustgraph_cli-2.10.3/trustgraph/cli/mint_token.py

@@ -0,0 +1,70 @@
+"""
+Mint a JWT containing a caller-supplied UserContext.  Requires an
+admin auth token.  Prints the resulting JWT to stdout.
+"""
+
+import argparse
+import json
+import sys
+
+from ._iam import DEFAULT_URL, DEFAULT_TOKEN, call_auth, run_main
+
+
+def do_mint_token(args):
+    try:
+        user_context = json.loads(args.user_context)
+    except json.JSONDecodeError as e:
+        print(f"Invalid JSON for --user-context: {e}", file=sys.stderr)
+        sys.exit(1)
+
+    if not isinstance(user_context, dict):
+        print("--user-context must be a JSON object", file=sys.stderr)
+        sys.exit(1)
+
+    body = {
+        "user_id": args.user_id,
+        "workspace": args.workspace,
+        "user_context": user_context,
+    }
+
+    resp = call_auth(
+        args.api_url, "/api/v1/auth/mint-token", args.token, body,
+    )
+
+    jwt = resp.get("jwt", "")
+    expires = resp.get("jwt_expires", "")
+
+    if expires:
+        print(f"JWT expires: {expires}", file=sys.stderr)
+    print(jwt)
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        prog="tg-mint-token", description=__doc__,
+    )
+    parser.add_argument(
+        "-u", "--api-url", default=DEFAULT_URL,
+        help=f"API URL (default: {DEFAULT_URL})",
+    )
+    parser.add_argument(
+        "-t", "--token", default=DEFAULT_TOKEN,
+        help="Admin auth token (JWT or API key)",
+    )
+    parser.add_argument(
+        "--user-id", required=True,
+        help="User ID for the minted token's subject",
+    )
+    parser.add_argument(
+        "-w", "--workspace", required=True,
+        help="Target workspace for the minted token",
+    )
+    parser.add_argument(
+        "--user-context", required=True,
+        help="UserContext as a JSON string",
+    )
+    run_main(do_mint_token, parser)
+
+
+if __name__ == "__main__":
+    main()

trustgraph_cli-2.10.3/trustgraph/cli_version.py

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

{trustgraph_cli-2.10.2 → trustgraph_cli-2.10.3}/trustgraph_cli.egg-info/PKG-INFO

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

@@ -61,6 +61,7 @@ trustgraph/cli/load_sample_documents.py
 trustgraph/cli/load_structured_data.py
 trustgraph/cli/load_turtle.py
 trustgraph/cli/login.py
+trustgraph/cli/mint_token.py
 trustgraph/cli/monitor_prompts.py
 trustgraph/cli/nquads.py
 trustgraph/cli/put_config_item.py

{trustgraph_cli-2.10.2 → trustgraph_cli-2.10.3}/trustgraph_cli.egg-info/entry_points.txt

@@ -56,6 +56,7 @@ tg-load-sample-documents = trustgraph.cli.load_sample_documents:main
 tg-load-structured-data = trustgraph.cli.load_structured_data:main
 tg-load-turtle = trustgraph.cli.load_turtle:main
 tg-login = trustgraph.cli.login:main
+tg-mint-token = trustgraph.cli.mint_token:main
 tg-monitor-prompts = trustgraph.cli.monitor_prompts:main
 tg-put-config-item = trustgraph.cli.put_config_item:main
 tg-put-de-core = trustgraph.cli.put_de_core:main

trustgraph_cli-2.10.2/trustgraph/cli_version.py

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