trustgraph-cli 1.1.1__tar.gz → 1.7.2__tar.gz

trustgraph_cli-1.7.2/PKG-INFO

@@ -0,0 +1,20 @@
+Metadata-Version: 2.4
+Name: trustgraph-cli
+Version: 1.7.2
+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.8,>=1.7
+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.7.2/pyproject.toml

@@ -0,0 +1,100 @@
+[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.7,<1.8",
+    "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-dump-queues = "trustgraph.cli.dump_queues: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-verify-system-status = "trustgraph.cli.verify_system_status: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.7.2/trustgraph/cli/__init__.py

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

trustgraph-cli-1.1.1/scripts/tg-add-library-document → trustgraph_cli-1.7.2/trustgraph/cli/add_library_document.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python3
-
 """
 Loads a document into the library
 """
@@ -202,5 +200,5 @@ def main():
 
         print("Exception:", e, flush=True)
 
-main()
-
+if __name__ == "__main__":
+    main()

trustgraph_cli-1.7.2/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.7.2/trustgraph/cli/delete_config_item.py

@@ -0,0 +1,69 @@
+"""
+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/')
+default_token = os.getenv("TRUSTGRAPH_TOKEN", None)
+
+def delete_config_item(url, config_type, key, token=None):
+
+    api = Api(url, token=token).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})',
+    )
+
+    parser.add_argument(
+        '-t', '--token',
+        default=default_token,
+        help='Authentication token (default: $TRUSTGRAPH_TOKEN)',
+    )
+
+    args = parser.parse_args()
+
+    try:
+
+        delete_config_item(
+            url=args.api_url,
+            config_type=args.type,
+            key=args.key,
+            token=args.token,
+        )
+
+    except Exception as e:
+
+        print("Exception:", e, flush=True)
+
+if __name__ == "__main__":
+    main()

trustgraph-cli-1.1.1/scripts/tg-delete-flow-class → trustgraph_cli-1.7.2/trustgraph/cli/delete_flow_class.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python3
-
 """
 Deletes a flow class
 """
@@ -49,5 +47,5 @@ def main():
 
         print("Exception:", e, flush=True)
 
-main()
-
+if __name__ == "__main__":
+    main()

trustgraph-cli-1.1.1/scripts/tg-delete-kg-core → trustgraph_cli-1.7.2/trustgraph/cli/delete_kg_core.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python3
-
 """
 Deletes a flow class
 """
@@ -21,7 +19,7 @@ def delete_kg_core(url, user, id):
 def main():
 
     parser = argparse.ArgumentParser(
-        prog='tg-delete-flow-class',
+        prog='tg-delete-kg-core',
         description=__doc__,
     )
 
@@ -57,5 +55,5 @@ def main():
 
         print("Exception:", e, flush=True)
 
-main()
-
+if __name__ == "__main__":
+    main()

trustgraph_cli-1.7.2/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()

trustgraph_cli-1.7.2/trustgraph/cli/delete_tool.py

@@ -0,0 +1,98 @@
+"""
+Deletes tools from the TrustGraph system.
+Removes tool configurations by ID from the agent configuration
+and updates the tool index accordingly.
+"""
+
+import argparse
+import os
+from trustgraph.api import Api, ConfigKey, ConfigValue
+import json
+import textwrap
+
+default_url = os.getenv("TRUSTGRAPH_URL", 'http://localhost:8088/')
+
+def delete_tool(
+        url : str,
+        id : str,
+):
+
+    api = Api(url).config()
+
+    # Check if the tool configuration exists
+    try:
+        tool_values = api.get([
+            ConfigKey(type="tool", key=id)
+        ])
+        
+        if not tool_values or not tool_values[0].value:
+            print(f"Tool configuration for '{id}' not found.")
+            return False
+            
+    except Exception as e:
+        print(f"Tool configuration for '{id}' not found.")
+        return False
+
+    # Delete the tool configuration and update the index
+    try:
+
+        # Delete the tool configuration
+        api.delete([
+            ConfigKey(type="tool", key=id)
+        ])
+        
+        print(f"Tool '{id}' deleted successfully.")
+        return True
+        
+    except Exception as e:
+        print(f"Error deleting tool '{id}': {e}")
+        return False
+
+def main():
+
+    parser = argparse.ArgumentParser(
+        prog='tg-delete-tool',
+        description=__doc__,
+        epilog=textwrap.dedent('''
+            This utility removes tool configurations from the TrustGraph system.
+            It removes the tool from both the tool index and deletes the tool
+            configuration. 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='Tool ID to delete',
+    )
+
+    args = parser.parse_args()
+
+    try:
+
+        if not args.id:
+            raise RuntimeError("Must specify --id for tool to delete")
+
+        delete_tool(
+            url=args.api_url, 
+            id=args.id
+        )
+
+    except Exception as e:
+
+        print("Exception:", e, flush=True)
+
+if __name__ == "__main__":
+    main()

trustgraph-cli-1.1.1/scripts/tg-dump-msgpack → trustgraph_cli-1.7.2/trustgraph/cli/dump_msgpack.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python3
-
 """
 This utility reads a knowledge core in msgpack format and outputs its
 contents in JSON form to standard output.  This is useful only as a
@@ -89,5 +87,5 @@ def main():
     else:
         dump(**vars(args))
 
-main()
-
+if __name__ == "__main__":
+    main()