trustgraph-cli 1.4.6__tar.gz → 1.4.8__tar.gz

{trustgraph_cli-1.4.6 → trustgraph_cli-1.4.8}/PKG-INFO

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

@@ -86,6 +86,9 @@ 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*"]

trustgraph_cli-1.4.8/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.4.8/trustgraph/cli/list_collections.py

@@ -0,0 +1,86 @@
+"""
+List collections for a user
+"""
+
+import argparse
+import os
+import tabulate
+from trustgraph.api import Api
+import json
+
+default_url = os.getenv("TRUSTGRAPH_URL", 'http://localhost:8088/')
+default_user = "trustgraph"
+
+def list_collections(url, user, tag_filter):
+
+    api = Api(url).collection()
+
+    collections = api.list_collections(user=user, tag_filter=tag_filter)
+
+    # Handle None or empty collections
+    if not collections or len(collections) == 0:
+        print("No collections found.")
+        return
+
+    table = []
+    for collection in collections:
+        table.append([
+            collection.collection,
+            collection.name,
+            collection.description,
+            ", ".join(collection.tags),
+            collection.created_at,
+            collection.updated_at
+        ])
+
+    headers = ["Collection", "Name", "Description", "Tags", "Created", "Updated"]
+
+    print(tabulate.tabulate(
+        table,
+        headers=headers,
+        tablefmt="pretty",
+        stralign="left",
+        maxcolwidths=[20, 30, 50, 30, 19, 19],
+    ))
+
+def main():
+
+    parser = argparse.ArgumentParser(
+        prog='tg-list-collections',
+        description=__doc__,
+    )
+
+    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(
+        '-t', '--tag-filter',
+        action='append',
+        help='Filter by tags (can be specified multiple times)'
+    )
+
+    args = parser.parse_args()
+
+    try:
+
+        list_collections(
+            url = args.api_url,
+            user = args.user,
+            tag_filter = args.tag_filter
+        )
+
+    except Exception as e:
+
+        print("Exception:", e, flush=True)
+
+if __name__ == "__main__":
+    main()

trustgraph_cli-1.4.8/trustgraph/cli/set_collection.py

@@ -0,0 +1,103 @@
+"""
+Set collection metadata (creates if doesn't exist)
+"""
+
+import argparse
+import os
+import tabulate
+from trustgraph.api import Api
+
+default_url = os.getenv("TRUSTGRAPH_URL", 'http://localhost:8088/')
+default_user = "trustgraph"
+
+def set_collection(url, user, collection, name, description, tags):
+
+    api = Api(url).collection()
+
+    result = api.update_collection(
+        user=user,
+        collection=collection,
+        name=name,
+        description=description,
+        tags=tags
+    )
+
+    if result:
+        print(f"Collection '{collection}' set successfully.")
+
+        table = []
+        table.append(("Collection", result.collection))
+        table.append(("Name", result.name))
+        table.append(("Description", result.description))
+        table.append(("Tags", ", ".join(result.tags)))
+        table.append(("Updated", result.updated_at))
+
+        print(tabulate.tabulate(
+            table,
+            tablefmt="pretty",
+            stralign="left",
+            maxcolwidths=[None, 67],
+        ))
+    else:
+        print(f"Failed to set collection '{collection}'.")
+
+def main():
+
+    parser = argparse.ArgumentParser(
+        prog='tg-set-collection',
+        description=__doc__,
+    )
+
+    parser.add_argument(
+        'collection',
+        help='Collection ID to set'
+    )
+
+    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(
+        '-n', '--name',
+        help='Collection name'
+    )
+
+    parser.add_argument(
+        '-d', '--description',
+        help='Collection description'
+    )
+
+    parser.add_argument(
+        '-t', '--tag',
+        action='append',
+        dest='tags',
+        help='Collection tags (can be specified multiple times)'
+    )
+
+    args = parser.parse_args()
+
+    try:
+
+        set_collection(
+            url = args.api_url,
+            user = args.user,
+            collection = args.collection,
+            name = args.name,
+            description = args.description,
+            tags = args.tags
+        )
+
+    except Exception as e:
+
+        print("Exception:", e, flush=True)
+
+if __name__ == "__main__":
+    main()

trustgraph_cli-1.4.8/trustgraph/cli_version.py

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

{trustgraph_cli-1.4.6 → trustgraph_cli-1.4.8}/trustgraph_cli.egg-info/PKG-INFO

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

@@ -3,6 +3,7 @@ pyproject.toml
 trustgraph/cli_version.py
 trustgraph/cli/__init__.py
 trustgraph/cli/add_library_document.py
+trustgraph/cli/delete_collection.py
 trustgraph/cli/delete_config_item.py
 trustgraph/cli/delete_flow_class.py
 trustgraph/cli/delete_kg_core.py
@@ -24,6 +25,7 @@ trustgraph/cli/invoke_nlp_query.py
 trustgraph/cli/invoke_objects_query.py
 trustgraph/cli/invoke_prompt.py
 trustgraph/cli/invoke_structured_query.py
+trustgraph/cli/list_collections.py
 trustgraph/cli/list_config_items.py
 trustgraph/cli/load_doc_embeds.py
 trustgraph/cli/load_kg_core.py
@@ -38,6 +40,7 @@ trustgraph/cli/put_flow_class.py
 trustgraph/cli/put_kg_core.py
 trustgraph/cli/remove_library_document.py
 trustgraph/cli/save_doc_embeds.py
+trustgraph/cli/set_collection.py
 trustgraph/cli/set_mcp_tool.py
 trustgraph/cli/set_prompt.py
 trustgraph/cli/set_token_costs.py

{trustgraph_cli-1.4.6 → trustgraph_cli-1.4.8}/trustgraph_cli.egg-info/entry_points.txt

@@ -1,5 +1,6 @@
 [console_scripts]
 tg-add-library-document = trustgraph.cli.add_library_document:main
+tg-delete-collection = trustgraph.cli.delete_collection:main
 tg-delete-config-item = trustgraph.cli.delete_config_item:main
 tg-delete-flow-class = trustgraph.cli.delete_flow_class:main
 tg-delete-kg-core = trustgraph.cli.delete_kg_core:main
@@ -20,6 +21,7 @@ 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-list-collections = trustgraph.cli.list_collections:main
 tg-list-config-items = trustgraph.cli.list_config_items:main
 tg-load-doc-embeds = trustgraph.cli.load_doc_embeds:main
 tg-load-kg-core = trustgraph.cli.load_kg_core:main
@@ -34,6 +36,7 @@ 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-collection = trustgraph.cli.set_collection: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

trustgraph_cli-1.4.6/trustgraph/cli_version.py

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