trustgraph-cli 2.10.7__tar.gz → 2.10.9__tar.gz

{trustgraph_cli-2.10.7 → trustgraph_cli-2.10.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-cli
-Version: 2.10.7
+Version: 2.10.9
 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.7 → trustgraph_cli-2.10.9}/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.9/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.9/trustgraph/cli_version.py

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

{trustgraph_cli-2.10.7 → trustgraph_cli-2.10.9}/trustgraph_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-cli
-Version: 2.10.7
+Version: 2.10.9
 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.7 → trustgraph_cli-2.10.9}/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.7 → trustgraph_cli-2.10.9}/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.7/trustgraph/cli_version.py

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