trustgraph-cli 2.10.11__tar.gz → 2.10.13__tar.gz

{trustgraph_cli-2.10.11 → trustgraph_cli-2.10.13}/PKG-INFO

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

@@ -65,6 +65,9 @@ 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"
 tg-invoke-llm = "trustgraph.cli.invoke_llm:main"
+tg-list-cloze-models = "trustgraph.cli.list_cloze_models:main"
+tg-invoke-cloze = "trustgraph.cli.invoke_cloze:main"
+tg-invoke-decision-scoring = "trustgraph.cli.invoke_decision_scoring:main"
 tg-invoke-embeddings = "trustgraph.cli.invoke_embeddings:main"
 tg-invoke-graph-embeddings = "trustgraph.cli.invoke_graph_embeddings:main"
 tg-invoke-document-embeddings = "trustgraph.cli.invoke_document_embeddings:main"

trustgraph_cli-2.10.13/trustgraph/cli/invoke_cloze.py

@@ -0,0 +1,88 @@
+"""
+Invokes the cloze classification service. Sends a query to a trained
+cloze model instance and returns the predicted label with confidence.
+"""
+
+import argparse
+import os
+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 query(url, flow_id, model, query_text, token=None, workspace="default"):
+
+    api = Api(url=url, token=token, workspace=workspace)
+    socket = api.socket()
+    flow = socket.flow(flow_id)
+
+    try:
+        result = flow.cloze(model=model, query=query_text)
+        print(f"Label: {result.get('label', '')}")
+        print(f"Confidence: {result.get('confidence', 0):.2%}")
+
+    finally:
+        socket.close()
+
+def main():
+
+    parser = argparse.ArgumentParser(
+        prog='tg-invoke-cloze',
+        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=f'Flow ID (default: default)'
+    )
+
+    parser.add_argument(
+        '-m', '--model',
+        required=True,
+        help='Name of the trained cloze model instance',
+    )
+
+    parser.add_argument(
+        'query',
+        help='Query text to classify',
+    )
+
+    args = parser.parse_args()
+
+    try:
+
+        query(
+            url=args.url,
+            flow_id=args.flow_id,
+            model=args.model,
+            query_text=args.query,
+            token=args.token,
+            workspace=args.workspace,
+        )
+
+    except Exception as e:
+
+        print("Exception:", e, flush=True)
+
+if __name__ == "__main__":
+    main()

trustgraph_cli-2.10.13/trustgraph/cli/invoke_decision_scoring.py

@@ -0,0 +1,119 @@
+"""
+Invokes the decision-scoring service. Sends state and question
+specifications to a System 1 decision model and returns scored answers.
+"""
+
+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 query(url, flow_id, state, questions, token=None, workspace="default"):
+
+    api = Api(url=url, token=token, workspace=workspace)
+    socket = api.socket()
+    flow = socket.flow(flow_id)
+
+    try:
+        result = flow.decision_scoring(state=state, questions=questions)
+
+        error = result.get("error")
+        if error:
+            print(
+                f"Error: [{error.get('type', 'unknown')}] "
+                f"{error.get('message', 'Unknown error')}",
+                file=sys.stderr,
+            )
+            sys.exit(1)
+
+        model = result.get("model", "")
+        if model:
+            print(f"Model: {model}")
+
+        answers = result.get("answers", {})
+        for name, answer in answers.items():
+            print(f"\n{name}:")
+            print(json.dumps(answer, indent=2))
+
+    finally:
+        socket.close()
+
+def main():
+
+    parser = argparse.ArgumentParser(
+        prog='tg-invoke-decision-scoring',
+        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=f'Flow ID (default: default)'
+    )
+
+    parser.add_argument(
+        '-s', '--state',
+        required=True,
+        help='State as JSON string or @filename',
+    )
+
+    parser.add_argument(
+        '-q', '--questions',
+        required=True,
+        help='Questions specification as JSON string or @filename',
+    )
+
+    args = parser.parse_args()
+
+    try:
+
+        def load_json(value):
+            if value.startswith("@"):
+                path = value[1:]
+                if path == "-":
+                    return json.load(sys.stdin)
+                with open(path) as f:
+                    return json.load(f)
+            return json.loads(value)
+
+        state = load_json(args.state)
+        questions = load_json(args.questions)
+
+        query(
+            url=args.url,
+            flow_id=args.flow_id,
+            state=state,
+            questions=questions,
+            token=args.token,
+            workspace=args.workspace,
+        )
+
+    except Exception as e:
+
+        print("Exception:", e, flush=True)
+
+if __name__ == "__main__":
+    main()

trustgraph_cli-2.10.13/trustgraph/cli/list_cloze_models.py

@@ -0,0 +1,101 @@
+"""
+Lists cloze model instances in the current workspace, showing their
+state (ready/training), labels, accuracy, and dataset sizes.
+"""
+
+import argparse
+import json
+import os
+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 query(url, flow_id, token=None, workspace="default", as_json=False):
+
+    api = Api(url=url, token=token, workspace=workspace)
+    socket = api.socket()
+    flow = socket.flow(flow_id)
+
+    try:
+        result = flow.cloze_list_models()
+        models = result.get("models", [])
+
+        if as_json:
+            print(json.dumps(models, indent=2))
+            return
+
+        if not models:
+            print("No cloze models found.")
+            return
+
+        for m in models:
+            print(f"{m['name']}:")
+            print(f"  State: {m['state']}")
+            if m.get("labels"):
+                print(f"  Labels: {', '.join(m['labels'])}")
+            if m.get("accuracy") is not None:
+                print(f"  Accuracy: {m['accuracy']:.1%}")
+            if m.get("train_size") is not None:
+                print(f"  Train/Test: {m['train_size']}/{m.get('test_size', 0)}")
+
+    finally:
+        socket.close()
+
+def main():
+
+    parser = argparse.ArgumentParser(
+        prog='tg-list-cloze-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=f'Flow ID (default: default)'
+    )
+
+    parser.add_argument(
+        '-j', '--json',
+        action='store_true',
+        default=False,
+        help='Output as JSON',
+    )
+
+    args = parser.parse_args()
+
+    try:
+
+        query(
+            url=args.url,
+            flow_id=args.flow_id,
+            token=args.token,
+            workspace=args.workspace,
+            as_json=args.json,
+        )
+
+    except Exception as e:
+
+        print("Exception:", e, flush=True)
+
+if __name__ == "__main__":
+    main()

trustgraph_cli-2.10.13/trustgraph/cli_version.py

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

{trustgraph_cli-2.10.11 → trustgraph_cli-2.10.13}/trustgraph_cli.egg-info/PKG-INFO

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

@@ -33,6 +33,8 @@ trustgraph/cli/import_workspace.py
 trustgraph/cli/import_workspace_config.py
 trustgraph/cli/init_pulsar_manager.py
 trustgraph/cli/invoke_agent.py
+trustgraph/cli/invoke_cloze.py
+trustgraph/cli/invoke_decision_scoring.py
 trustgraph/cli/invoke_document_embeddings.py
 trustgraph/cli/invoke_document_rag.py
 trustgraph/cli/invoke_embeddings.py
@@ -48,6 +50,7 @@ trustgraph/cli/invoke_rows_query.py
 trustgraph/cli/invoke_sparql_query.py
 trustgraph/cli/invoke_structured_query.py
 trustgraph/cli/list_api_keys.py
+trustgraph/cli/list_cloze_models.py
 trustgraph/cli/list_collections.py
 trustgraph/cli/list_config_items.py
 trustgraph/cli/list_explain_traces.py

{trustgraph_cli-2.10.11 → trustgraph_cli-2.10.13}/trustgraph_cli.egg-info/entry_points.txt

@@ -28,6 +28,8 @@ tg-graph-to-turtle = trustgraph.cli.graph_to_turtle:main
 tg-import-workspace = trustgraph.cli.import_workspace:main
 tg-import-workspace-config = trustgraph.cli.import_workspace_config:main
 tg-invoke-agent = trustgraph.cli.invoke_agent:main
+tg-invoke-cloze = trustgraph.cli.invoke_cloze:main
+tg-invoke-decision-scoring = trustgraph.cli.invoke_decision_scoring:main
 tg-invoke-document-embeddings = trustgraph.cli.invoke_document_embeddings:main
 tg-invoke-document-rag = trustgraph.cli.invoke_document_rag:main
 tg-invoke-embeddings = trustgraph.cli.invoke_embeddings:main
@@ -43,6 +45,7 @@ tg-invoke-rows-query = trustgraph.cli.invoke_rows_query:main
 tg-invoke-sparql-query = trustgraph.cli.invoke_sparql_query:main
 tg-invoke-structured-query = trustgraph.cli.invoke_structured_query:main
 tg-list-api-keys = trustgraph.cli.list_api_keys:main
+tg-list-cloze-models = trustgraph.cli.list_cloze_models: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

trustgraph_cli-2.10.11/trustgraph/cli_version.py

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