trustgraph-cli 2.10.18__tar.gz → 2.11.1__tar.gz

{trustgraph_cli-2.10.18 → trustgraph_cli-2.11.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-cli
-Version: 2.10.18
+Version: 2.11.1
 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
@@ -8,7 +8,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
-Requires-Dist: trustgraph-base<2.11,>=2.10
+Requires-Dist: trustgraph-base<2.12,>=2.11
 Requires-Dist: requests
 Requires-Dist: pulsar-client
 Requires-Dist: aiohttp

{trustgraph_cli-2.10.18 → trustgraph_cli-2.11.1}/pyproject.toml

@@ -10,7 +10,7 @@ description = "TrustGraph provides a means to run a pipeline of flexible AI proc
 readme = "README.md"
 requires-python = ">=3.10"
 dependencies = [
-    "trustgraph-base>=2.10,<2.11",
+    "trustgraph-base>=2.11,<2.12",
     "requests",
     "pulsar-client",
     "aiohttp",
@@ -68,6 +68,7 @@ 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-field-extract = "trustgraph.cli.invoke_field_extraction: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.11.1/trustgraph/cli/invoke_field_extraction.py

@@ -0,0 +1,120 @@
+"""
+Invokes the field-extraction service. Sends text and field
+instructions to a span-extraction model and returns extracted spans.
+"""
+
+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, fields, token=None, workspace="default"):
+
+    api = Api(url=url, token=token, workspace=workspace)
+    socket = api.socket()
+    flow = socket.flow(flow_id)
+
+    try:
+        result = flow.field_extraction(state=state, fields=fields)
+
+        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, spans in answers.items():
+            print(f"\n{name}:")
+            for span in spans:
+                print(f"  {span}")
+
+    finally:
+        socket.close()
+
+def main():
+
+    parser = argparse.ArgumentParser(
+        prog='tg-invoke-field-extract',
+        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(
+        '-F', '--fields',
+        required=True,
+        help='Fields 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)
+        fields = load_json(args.fields)
+
+        query(
+            url=args.url,
+            flow_id=args.flow_id,
+            state=state,
+            fields=fields,
+            token=args.token,
+            workspace=args.workspace,
+        )
+
+    except Exception as e:
+
+        print("Exception:", e, flush=True)
+
+if __name__ == "__main__":
+    main()

trustgraph_cli-2.11.1/trustgraph/cli_version.py

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

{trustgraph_cli-2.10.18 → trustgraph_cli-2.11.1}/trustgraph_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-cli
-Version: 2.10.18
+Version: 2.11.1
 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
@@ -8,7 +8,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
-Requires-Dist: trustgraph-base<2.11,>=2.10
+Requires-Dist: trustgraph-base<2.12,>=2.11
 Requires-Dist: requests
 Requires-Dist: pulsar-client
 Requires-Dist: aiohttp

{trustgraph_cli-2.10.18 → trustgraph_cli-2.11.1}/trustgraph_cli.egg-info/SOURCES.txt

@@ -38,6 +38,7 @@ trustgraph/cli/invoke_decision_scoring.py
 trustgraph/cli/invoke_document_embeddings.py
 trustgraph/cli/invoke_document_rag.py
 trustgraph/cli/invoke_embeddings.py
+trustgraph/cli/invoke_field_extraction.py
 trustgraph/cli/invoke_graph_embeddings.py
 trustgraph/cli/invoke_graph_rag.py
 trustgraph/cli/invoke_llm.py

{trustgraph_cli-2.10.18 → trustgraph_cli-2.11.1}/trustgraph_cli.egg-info/entry_points.txt

@@ -33,6 +33,7 @@ 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
+tg-invoke-field-extract = trustgraph.cli.invoke_field_extraction:main
 tg-invoke-graph-embeddings = trustgraph.cli.invoke_graph_embeddings:main
 tg-invoke-graph-rag = trustgraph.cli.invoke_graph_rag:main
 tg-invoke-llm = trustgraph.cli.invoke_llm:main

{trustgraph_cli-2.10.18 → trustgraph_cli-2.11.1}/trustgraph_cli.egg-info/requires.txt

@@ -1,4 +1,4 @@
-trustgraph-base<2.11,>=2.10
+trustgraph-base<2.12,>=2.11
 requests
 pulsar-client
 aiohttp

trustgraph_cli-2.10.18/trustgraph/cli_version.py

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