trustgraph-cli 0.21.0__tar.gz → 1.5.7__tar.gz

trustgraph_cli-1.5.7/PKG-INFO

@@ -0,0 +1,20 @@
+Metadata-Version: 2.4
+Name: trustgraph-cli
+Version: 1.5.7
+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
+Classifier: Programming Language :: Python :: 3
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: trustgraph-base<1.6,>=1.5
+Requires-Dist: requests
+Requires-Dist: pulsar-client
+Requires-Dist: aiohttp
+Requires-Dist: rdflib
+Requires-Dist: tabulate
+Requires-Dist: msgpack
+Requires-Dist: websockets
+
+See https://trustgraph.ai/

trustgraph_cli-1.5.7/pyproject.toml

@@ -0,0 +1,98 @@
+[build-system]
+requires = ["setuptools>=61.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "trustgraph-cli"
+dynamic = ["version"]
+authors = [{name = "trustgraph.ai", email = "security@trustgraph.ai"}]
+description = "TrustGraph provides a means to run a pipeline of flexible AI processing components in a flexible means to achieve a processing pipeline."
+readme = "README.md"
+requires-python = ">=3.8"
+dependencies = [
+    "trustgraph-base>=1.5,<1.6",
+    "requests",
+    "pulsar-client",
+    "aiohttp",
+    "rdflib",
+    "tabulate",
+    "msgpack",
+    "websockets",
+]
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "Operating System :: OS Independent",
+]
+
+[project.urls]
+Homepage = "https://github.com/trustgraph-ai/trustgraph"
+
+[project.scripts]
+tg-add-library-document = "trustgraph.cli.add_library_document:main"
+tg-delete-flow-class = "trustgraph.cli.delete_flow_class:main"
+tg-delete-mcp-tool = "trustgraph.cli.delete_mcp_tool:main"
+tg-delete-kg-core = "trustgraph.cli.delete_kg_core:main"
+tg-delete-tool = "trustgraph.cli.delete_tool:main"
+tg-dump-msgpack = "trustgraph.cli.dump_msgpack:main"
+tg-get-flow-class = "trustgraph.cli.get_flow_class:main"
+tg-get-kg-core = "trustgraph.cli.get_kg_core:main"
+tg-graph-to-turtle = "trustgraph.cli.graph_to_turtle:main"
+tg-init-trustgraph = "trustgraph.cli.init_trustgraph: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"
+tg-invoke-llm = "trustgraph.cli.invoke_llm:main"
+tg-invoke-mcp-tool = "trustgraph.cli.invoke_mcp_tool:main"
+tg-invoke-nlp-query = "trustgraph.cli.invoke_nlp_query:main"
+tg-invoke-objects-query = "trustgraph.cli.invoke_objects_query:main"
+tg-invoke-prompt = "trustgraph.cli.invoke_prompt:main"
+tg-invoke-structured-query = "trustgraph.cli.invoke_structured_query:main"
+tg-load-doc-embeds = "trustgraph.cli.load_doc_embeds:main"
+tg-load-kg-core = "trustgraph.cli.load_kg_core:main"
+tg-load-pdf = "trustgraph.cli.load_pdf:main"
+tg-load-sample-documents = "trustgraph.cli.load_sample_documents:main"
+tg-load-text = "trustgraph.cli.load_text:main"
+tg-load-turtle = "trustgraph.cli.load_turtle:main"
+tg-load-knowledge = "trustgraph.cli.load_knowledge:main"
+tg-load-structured-data = "trustgraph.cli.load_structured_data:main"
+tg-put-flow-class = "trustgraph.cli.put_flow_class:main"
+tg-put-kg-core = "trustgraph.cli.put_kg_core:main"
+tg-remove-library-document = "trustgraph.cli.remove_library_document:main"
+tg-save-doc-embeds = "trustgraph.cli.save_doc_embeds:main"
+tg-set-mcp-tool = "trustgraph.cli.set_mcp_tool:main"
+tg-set-prompt = "trustgraph.cli.set_prompt:main"
+tg-set-token-costs = "trustgraph.cli.set_token_costs:main"
+tg-set-tool = "trustgraph.cli.set_tool:main"
+tg-show-config = "trustgraph.cli.show_config:main"
+tg-show-flow-classes = "trustgraph.cli.show_flow_classes:main"
+tg-show-flow-state = "trustgraph.cli.show_flow_state:main"
+tg-show-flows = "trustgraph.cli.show_flows:main"
+tg-show-graph = "trustgraph.cli.show_graph:main"
+tg-show-kg-cores = "trustgraph.cli.show_kg_cores:main"
+tg-show-library-documents = "trustgraph.cli.show_library_documents:main"
+tg-show-library-processing = "trustgraph.cli.show_library_processing:main"
+tg-show-mcp-tools = "trustgraph.cli.show_mcp_tools:main"
+tg-show-parameter-types = "trustgraph.cli.show_parameter_types:main"
+tg-show-processor-state = "trustgraph.cli.show_processor_state:main"
+tg-show-prompts = "trustgraph.cli.show_prompts:main"
+tg-show-token-costs = "trustgraph.cli.show_token_costs:main"
+tg-show-token-rate = "trustgraph.cli.show_token_rate:main"
+tg-show-tools = "trustgraph.cli.show_tools:main"
+tg-start-flow = "trustgraph.cli.start_flow:main"
+tg-unload-kg-core = "trustgraph.cli.unload_kg_core:main"
+tg-start-library-processing = "trustgraph.cli.start_library_processing:main"
+tg-stop-flow = "trustgraph.cli.stop_flow:main"
+tg-stop-library-processing = "trustgraph.cli.stop_library_processing:main"
+tg-list-config-items = "trustgraph.cli.list_config_items:main"
+tg-get-config-item = "trustgraph.cli.get_config_item:main"
+tg-put-config-item = "trustgraph.cli.put_config_item:main"
+tg-delete-config-item = "trustgraph.cli.delete_config_item:main"
+tg-list-collections = "trustgraph.cli.list_collections:main"
+tg-set-collection = "trustgraph.cli.set_collection:main"
+tg-delete-collection = "trustgraph.cli.delete_collection:main"
+
+[tool.setuptools.packages.find]
+include = ["trustgraph*"]
+
+[tool.setuptools.dynamic]
+version = {attr = "trustgraph.cli_version.__version__"}

trustgraph_cli-1.5.7/trustgraph/cli/__init__.py

@@ -0,0 +1 @@
+# TrustGraph CLI modules

trustgraph_cli-1.5.7/trustgraph/cli/add_library_document.py

@@ -0,0 +1,204 @@
+"""
+Loads a document into the library
+"""
+
+import hashlib
+import argparse
+import os
+import time
+import uuid
+
+from trustgraph.api import Api
+from trustgraph.knowledge import hash, to_uri
+from trustgraph.knowledge import PREF_PUBEV, PREF_DOC, PREF_ORG
+from trustgraph.knowledge import Organization, PublicationEvent
+from trustgraph.knowledge import DigitalDocument
+
+default_url = os.getenv("TRUSTGRAPH_URL", 'http://localhost:8088/')
+default_user = 'trustgraph'
+
+class Loader:
+
+    def __init__(
+            self, id, url, user, metadata, title, comments, kind, tags
+    ):
+
+        self.api = Api(url).library()
+
+        self.user = user
+        self.metadata = metadata
+        self.title = title
+        self.comments = comments
+        self.kind = kind
+        self.identifier = id
+
+        if tags:
+            self.tags = tags.split(",")
+        else:
+            self.tags = []
+
+    def load(self, files):
+
+        for file in files:
+            self.load_file(file)
+
+    def load_file(self, file):
+
+        try:
+
+            path = file
+            data = open(path, "rb").read()
+
+            # Create a SHA256 hash from the data
+            if self.identifier:
+                id = self.identifier
+            else:
+                id = hash(data)
+                id = to_uri(PREF_DOC, id)
+                
+
+            self.metadata.id = id
+
+            self.api.add_document(
+                document=data, id=id, metadata=self.metadata, 
+                user=self.user, kind=self.kind, title=self.title,
+                comments=self.comments, tags=self.tags
+            )
+
+            print(f"{file}: Loaded successfully.")
+
+        except Exception as e:
+            print(f"{file}: Failed: {str(e)}", flush=True)
+            raise e
+
+def main():
+
+    parser = argparse.ArgumentParser(
+        prog='tg-add-library-document',
+        description=__doc__,
+    )
+
+    parser.add_argument(
+        '-u', '--url',
+        default=default_url,
+        help=f'API URL (default: {default_url})',
+    )
+
+    parser.add_argument(
+        '-U', '--user',
+        default=default_user,
+        help=f'User ID (default: {default_user})'
+    )
+
+    parser.add_argument(
+        '--name', help=f'Document name'
+    )
+
+    parser.add_argument(
+        '--description', help=f'Document description'
+    )
+
+    parser.add_argument(
+        '--copyright-notice', help=f'Copyright notice'
+    )
+
+    parser.add_argument(
+        '--copyright-holder', help=f'Copyright holder'
+    )
+
+    parser.add_argument(
+        '--copyright-year', help=f'Copyright year'
+    )
+
+    parser.add_argument(
+        '--license', help=f'Copyright license'
+    )
+
+    parser.add_argument(
+        '--publication-organization', help=f'Publication organization'
+    )
+
+    parser.add_argument(
+        '--publication-description', help=f'Publication description'
+    )
+
+    parser.add_argument(
+        '--publication-date', help=f'Publication date'
+    )
+
+    parser.add_argument(
+        '--document-url', help=f'Document URL'
+    )
+
+    parser.add_argument(
+        '--keyword', nargs='+', help=f'Keyword'
+    )
+
+    parser.add_argument(
+        '--identifier', '--id', help=f'Document ID'
+    )
+
+    parser.add_argument(
+        '-k', '--kind',
+        required=True,
+        help=f'Document MIME type'
+    )
+
+    parser.add_argument(
+        '--tags',
+        help=f'Tags, command separated'
+    )
+
+    parser.add_argument(
+        'files', nargs='+',
+        help=f'File to load'
+    )
+
+    args = parser.parse_args()
+
+    try:
+
+        document = DigitalDocument(
+            args.identifier,
+            name=args.name,
+            description=args.description,
+            copyright_notice=args.copyright_notice,
+            copyright_holder=args.copyright_holder,
+            copyright_year=args.copyright_year,
+            license=args.license,
+            url=args.document_url,
+            keywords=args.keyword,
+        )
+
+        if args.publication_organization:
+            org = Organization(
+                id=to_uri(PREF_ORG, hash(args.publication_organization)),
+                name=args.publication_organization,
+            )
+            document.publication = PublicationEvent(
+                id = to_uri(PREF_PUBEV, str(uuid.uuid4())),
+                organization=org,
+                description=args.publication_description,
+                start_date=args.publication_date,
+                end_date=args.publication_date,
+            )
+
+        p = Loader(
+            id=args.identifier,
+            url=args.url,
+            user=args.user,
+            metadata=document,
+            title=args.name,
+            comments=args.description,
+            kind=args.kind,
+            tags=args.tags,
+        )
+
+        p.load(args.files)
+
+    except Exception as e:
+
+        print("Exception:", e, flush=True)
+
+if __name__ == "__main__":
+    main()

trustgraph_cli-1.5.7/trustgraph/cli/delete_collection.py

@@ -0,0 +1,72 @@
+"""
+Delete a collection and all its data
+"""
+
+import argparse
+import os
+from trustgraph.api import Api
+
+default_url = os.getenv("TRUSTGRAPH_URL", 'http://localhost:8088/')
+default_user = "trustgraph"
+
+def delete_collection(url, user, collection, confirm):
+
+    if not confirm:
+        response = input(f"Are you sure you want to delete collection '{collection}' and all its data? (y/N): ")
+        if response.lower() not in ['y', 'yes']:
+            print("Operation cancelled.")
+            return
+
+    api = Api(url).collection()
+
+    api.delete_collection(user=user, collection=collection)
+
+    print(f"Collection '{collection}' deleted successfully.")
+
+def main():
+
+    parser = argparse.ArgumentParser(
+        prog='tg-delete-collection',
+        description=__doc__,
+    )
+
+    parser.add_argument(
+        'collection',
+        help='Collection ID to delete'
+    )
+
+    parser.add_argument(
+        '-u', '--api-url',
+        default=default_url,
+        help=f'API URL (default: {default_url})',
+    )
+
+    parser.add_argument(
+        '-U', '--user',
+        default=default_user,
+        help=f'User ID (default: {default_user})'
+    )
+
+    parser.add_argument(
+        '-y', '--yes',
+        action='store_true',
+        help='Skip confirmation prompt'
+    )
+
+    args = parser.parse_args()
+
+    try:
+
+        delete_collection(
+            url = args.api_url,
+            user = args.user,
+            collection = args.collection,
+            confirm = args.yes
+        )
+
+    except Exception as e:
+
+        print("Exception:", e, flush=True)
+
+if __name__ == "__main__":
+    main()

trustgraph_cli-1.5.7/trustgraph/cli/delete_config_item.py

@@ -0,0 +1,61 @@
+"""
+Deletes a configuration item
+"""
+
+import argparse
+import os
+from trustgraph.api import Api
+from trustgraph.api.types import ConfigKey
+
+default_url = os.getenv("TRUSTGRAPH_URL", 'http://localhost:8088/')
+
+def delete_config_item(url, config_type, key):
+
+    api = Api(url).config()
+
+    config_key = ConfigKey(type=config_type, key=key)
+    api.delete([config_key])
+
+    print(f"Configuration item deleted: {config_type}/{key}")
+
+def main():
+
+    parser = argparse.ArgumentParser(
+        prog='tg-delete-config-item',
+        description=__doc__,
+    )
+
+    parser.add_argument(
+        '--type',
+        required=True,
+        help='Configuration type',
+    )
+
+    parser.add_argument(
+        '--key',
+        required=True,
+        help='Configuration key',
+    )
+
+    parser.add_argument(
+        '-u', '--api-url',
+        default=default_url,
+        help=f'API URL (default: {default_url})',
+    )
+
+    args = parser.parse_args()
+
+    try:
+
+        delete_config_item(
+            url=args.api_url,
+            config_type=args.type,
+            key=args.key,
+        )
+
+    except Exception as e:
+
+        print("Exception:", e, flush=True)
+
+if __name__ == "__main__":
+    main()

trustgraph_cli-1.5.7/trustgraph/cli/delete_flow_class.py

@@ -0,0 +1,51 @@
+"""
+Deletes a flow class
+"""
+
+import argparse
+import os
+import tabulate
+from trustgraph.api import Api
+import json
+
+default_url = os.getenv("TRUSTGRAPH_URL", 'http://localhost:8088/')
+
+def delete_flow_class(url, class_name):
+
+    api = Api(url).flow()
+
+    class_names = api.delete_class(class_name)
+
+def main():
+
+    parser = argparse.ArgumentParser(
+        prog='tg-delete-flow-class',
+        description=__doc__,
+    )
+
+    parser.add_argument(
+        '-u', '--api-url',
+        default=default_url,
+        help=f'API URL (default: {default_url})',
+    )
+
+    parser.add_argument(
+        '-n', '--class-name',
+        help=f'Flow class name',
+    )
+
+    args = parser.parse_args()
+
+    try:
+
+        delete_flow_class(
+            url=args.api_url,
+            class_name=args.class_name,
+        )
+
+    except Exception as e:
+
+        print("Exception:", e, flush=True)
+
+if __name__ == "__main__":
+    main()

trustgraph_cli-1.5.7/trustgraph/cli/delete_kg_core.py

@@ -0,0 +1,59 @@
+"""
+Deletes a flow class
+"""
+
+import argparse
+import os
+import tabulate
+from trustgraph.api import Api
+import json
+
+default_url = os.getenv("TRUSTGRAPH_URL", 'http://localhost:8088/')
+
+def delete_kg_core(url, user, id):
+
+    api = Api(url).knowledge()
+
+    class_names = api.delete_kg_core(user = user, id = id)
+
+def main():
+
+    parser = argparse.ArgumentParser(
+        prog='tg-delete-kg-core',
+        description=__doc__,
+    )
+
+    parser.add_argument(
+        '-u', '--api-url',
+        default=default_url,
+        help=f'API URL (default: {default_url})',
+    )
+
+    parser.add_argument(
+        '-U', '--user',
+        default="trustgraph",
+        help='API URL (default: trustgraph)',
+    )
+
+    parser.add_argument(
+        '--id', '--identifier',
+        required=True,
+        help=f'Knowledge core ID',
+    )
+
+    args = parser.parse_args()
+
+    try:
+
+        delete_kg_core(
+            url=args.api_url,
+            user=args.user,
+            id=args.id,
+        )
+
+    except Exception as e:
+
+        print("Exception:", e, flush=True)
+
+if __name__ == "__main__":
+    main()

trustgraph_cli-1.5.7/trustgraph/cli/delete_mcp_tool.py

@@ -0,0 +1,93 @@
+"""
+Deletes MCP (Model Control Protocol) tools from the TrustGraph system.
+Removes MCP tool configurations by ID from the 'mcp' configuration group.
+"""
+
+import argparse
+import os
+from trustgraph.api import Api, ConfigKey
+import textwrap
+
+default_url = os.getenv("TRUSTGRAPH_URL", 'http://localhost:8088/')
+
+def delete_mcp_tool(
+        url : str,
+        id : str,
+):
+
+    api = Api(url).config()
+
+    # Check if the tool exists first
+    try:
+        values = api.get([
+            ConfigKey(type="mcp", key=id)
+        ])
+        
+        if not values or not values[0].value:
+            print(f"MCP tool '{id}' not found.")
+            return False
+            
+    except Exception as e:
+        print(f"MCP tool '{id}' not found.")
+        return False
+
+    # Delete the MCP tool configuration from the 'mcp' group
+    try:
+        api.delete([
+            ConfigKey(type="mcp", key=id)
+        ])
+        
+        print(f"MCP tool '{id}' deleted successfully.")
+        return True
+        
+    except Exception as e:
+        print(f"Error deleting MCP tool '{id}': {e}")
+        return False
+
+def main():
+
+    parser = argparse.ArgumentParser(
+        prog='tg-delete-mcp-tool',
+        description=__doc__,
+        epilog=textwrap.dedent('''
+            This utility removes MCP tool configurations from the TrustGraph system.
+            Once deleted, the tool will no longer be available for use.
+            
+            Examples:
+              %(prog)s --id weather
+              %(prog)s --id calculator
+              %(prog)s --api-url http://localhost:9000/ --id file-reader
+        ''').strip(),
+        formatter_class=argparse.RawDescriptionHelpFormatter
+    )
+
+    parser.add_argument(
+        '-u', '--api-url',
+        default=default_url,
+        help=f'API URL (default: {default_url})',
+    )
+
+    parser.add_argument(
+        '--id',
+        required=True,
+        help='MCP tool ID to delete',
+    )
+
+    args = parser.parse_args()
+
+    try:
+
+        if not args.id:
+            raise RuntimeError("Must specify --id for MCP tool to delete")
+
+        delete_mcp_tool(
+            url=args.api_url, 
+            id=args.id
+        )
+
+    except Exception as e:
+
+        print("Exception:", e, flush=True)
+
+if __name__ == "__main__":
+    main()