anatools 6.0.1__py3-none-any.whl → 6.0.3__py3-none-any.whl

{anatools-6.0.1.data → anatools-6.0.3.data}/scripts/renderedai

@@ -167,6 +167,7 @@ def cmd_workspaces_get(args):
         workspaceId=args.workspaceid,
         organizationId=args.orgid,
         limit=args.limit,
+        cursor=args.cursor,
         fields=parse_list_arg(args.fields) if args.fields else None
     )
     output_json(result)
@@ -225,6 +226,7 @@ def cmd_organizations_get(args):
     result = client.get_organizations(
         organizationId=args.orgid,
         limit=args.limit,
+        cursor=args.cursor,
         fields=parse_list_arg(args.fields) if args.fields else None
     )
     output_json(result)
@@ -242,6 +244,7 @@ def cmd_members_get(args):
     result = client.get_organization_members(
         organizationId=org_id,
         limit=args.limit,
+        cursor=args.cursor,
         fields=parse_list_arg(args.fields) if args.fields else None
     )
     output_json(result)
@@ -260,6 +263,7 @@ def cmd_datasets_get(args):
         workspaceId=workspace_id,
         datasetId=args.datasetid,
         limit=args.limit,
+        cursor=args.cursor,
         fields=parse_list_arg(args.fields) if args.fields else None
     )
     output_json(result)
@@ -329,6 +333,8 @@ def cmd_datasets_cancel(args):
 
 def cmd_datasets_download(args):
     """Download a dataset or a single file from a dataset."""
+    import zipfile
+
     client = get_client()
     workspace_id = require_arg(args, 'workspaceid', 'Workspace ID')
     dataset_id = require_arg(args, 'datasetid', 'Dataset ID')
@@ -347,7 +353,15 @@ def cmd_datasets_download(args):
             datasetId=dataset_id,
             localDir=args.outputdir
         )
-    output_json({"downloadPath": result})
+
+    if args.extract and result and result.endswith('.zip') and os.path.isfile(result):
+        extract_dir = os.path.splitext(result)[0]
+        with zipfile.ZipFile(result, 'r') as zf:
+            zf.extractall(extract_dir)
+        os.remove(result)
+        output_json({"downloadPath": extract_dir, "extracted": True})
+    else:
+        output_json({"downloadPath": result})
 
 
 def cmd_datasets_upload(args):
@@ -406,7 +420,8 @@ def cmd_datasets_files(args):
         workspaceId=workspace_id,
         datasetId=dataset_id,
         path=args.path,
-        limit=args.limit
+        limit=args.limit,
+        cursor=args.cursor
     )
     output_json(result)
 
@@ -420,6 +435,7 @@ def cmd_datasets_jobs(args):
         organizationId=args.orgid,
         datasetId=args.datasetid,
         limit=args.limit,
+        cursor=args.cursor,
         fields=parse_list_arg(args.fields) if args.fields else None
     )
     output_json(result)
@@ -456,6 +472,7 @@ def cmd_volumes_get(args):
         workspaceId=args.workspaceid,
         organizationId=args.orgid,
         limit=args.limit,
+        cursor=args.cursor,
         fields=parse_list_arg(args.fields) if args.fields else None
     )
     output_json(result)
@@ -514,7 +531,8 @@ def cmd_volume_data_get(args):
         dir=args.dir,
         files=parse_list_arg(args.files) if args.files else None,
         recursive=args.recursive,
-        limit=args.limit
+        limit=args.limit,
+        cursor=args.cursor
     )
     output_json(result)
 
@@ -573,7 +591,8 @@ def cmd_volume_data_search(args):
         keywords=parse_list_arg(args.keywords) if args.keywords else None,
         fileformats=parse_list_arg(args.formats) if args.formats else None,
         filetypes=parse_list_arg(args.types) if args.types else None,
-        limit=args.limit
+        limit=args.limit,
+        cursor=args.cursor
     )
     output_json(result)
 
@@ -986,6 +1005,7 @@ def cmd_graphs_get(args):
         graphId=args.graphid,
         staged=args.staged,
         limit=args.limit,
+        cursor=args.cursor,
         fields=parse_list_arg(args.fields) if args.fields else None
     )
     output_json(result)
@@ -1085,46 +1105,271 @@ def cmd_graphs_stage(args):
     output_json({"graphId": result})
 
 
+# =============================================================================
+# DATASET-VIEWER
+# =============================================================================
+
+DATASET_VIEWER_TRIGGER_PATH = os.path.join(os.path.expanduser('~'), '.theia', 'dataset-viewer-open')
+DATASET_VIEWER_STATUS_PATH = os.path.join(os.path.expanduser('~'), '.theia', 'dataset-viewer-status.json')
+
+
+def _write_dataset_viewer_trigger(action: str, payload: dict):
+    """Write a trigger file for the dataset viewer extension to pick up."""
+    trigger_data = {"action": action, **payload}
+    trigger_dir = os.path.dirname(DATASET_VIEWER_TRIGGER_PATH)
+    os.makedirs(trigger_dir, exist_ok=True)
+    with open(DATASET_VIEWER_TRIGGER_PATH, 'w') as f:
+        json.dump(trigger_data, f, indent=2)
+    return trigger_data
+
+
+def cmd_dataset_viewer_open(args):
+    """Open a dataset folder in the Annotation Viewer."""
+    dataset_path = getattr(args, 'path', None)
+    if not dataset_path:
+        output_error("Dataset path is required. Use --path", "MISSING_PATH")
+        sys.exit(1)
+
+    dataset_path = os.path.abspath(dataset_path)
+    if not os.path.isdir(dataset_path):
+        output_error(f"Directory not found: {dataset_path}", "PATH_NOT_FOUND")
+        sys.exit(1)
+
+    images_dir = os.path.join(dataset_path, 'images')
+    if not os.path.isdir(images_dir):
+        output_error(f"No images/ directory found in {dataset_path}", "NO_IMAGES_DIR")
+        sys.exit(1)
+
+    image_index = getattr(args, 'index', None) or 0
+
+    trigger_data = _write_dataset_viewer_trigger('open', {
+        'datasetPath': dataset_path,
+        'imageIndex': int(image_index),
+    })
+
+    output_json({
+        "status": "ok",
+        "action": "open",
+        "datasetPath": dataset_path,
+        "imageIndex": int(image_index),
+        "triggerPath": DATASET_VIEWER_TRIGGER_PATH,
+    })
+
+
+def cmd_dataset_viewer_next(args):
+    """Navigate to the next image in the dataset viewer."""
+    trigger_data = _write_dataset_viewer_trigger('next', {})
+    output_json({"status": "ok", "action": "next", "triggerPath": DATASET_VIEWER_TRIGGER_PATH})
+
+
+def cmd_dataset_viewer_prev(args):
+    """Navigate to the previous image in the dataset viewer."""
+    trigger_data = _write_dataset_viewer_trigger('prev', {})
+    output_json({"status": "ok", "action": "prev", "triggerPath": DATASET_VIEWER_TRIGGER_PATH})
+
+
+def cmd_dataset_viewer_goto(args):
+    """Navigate to a specific image by index or name."""
+    index = getattr(args, 'index', None)
+    name = getattr(args, 'name', None)
+
+    if index is None and not name:
+        output_error("Specify --index or --name", "MISSING_TARGET")
+        sys.exit(1)
+
+    payload = {}
+    if index is not None:
+        payload['imageIndex'] = int(index)
+    if name:
+        payload['imageName'] = name
+
+    trigger_data = _write_dataset_viewer_trigger('goto', payload)
+    output_json({
+        "status": "ok",
+        "action": "goto",
+        **payload,
+        "triggerPath": DATASET_VIEWER_TRIGGER_PATH,
+    })
+
+
+def cmd_dataset_viewer_annotations(args):
+    """Set which annotation types are enabled in the viewer."""
+    types_str = getattr(args, 'types', None)
+    if not types_str:
+        output_error("Annotation types required. Use --types (comma-separated: bbox,bbox3d,segmentation,centroid,mask)", "MISSING_TYPES")
+        sys.exit(1)
+
+    valid_types = {'bbox', 'bbox3d', 'segmentation', 'centroid', 'mask'}
+    types = [t.strip() for t in types_str.split(',')]
+    invalid = [t for t in types if t not in valid_types]
+    if invalid:
+        output_error(f"Invalid annotation types: {', '.join(invalid)}. Valid: {', '.join(sorted(valid_types))}", "INVALID_TYPES")
+        sys.exit(1)
+
+    trigger_data = _write_dataset_viewer_trigger('setAnnotations', {'annotations': types})
+    output_json({
+        "status": "ok",
+        "action": "setAnnotations",
+        "annotations": types,
+        "triggerPath": DATASET_VIEWER_TRIGGER_PATH,
+    })
+
+
+def cmd_dataset_viewer_filter(args):
+    """Set object type filter in the viewer."""
+    types_str = getattr(args, 'types', None)
+
+    if types_str:
+        types = [t.strip() for t in types_str.split(',')]
+    else:
+        # No types = show all (clear filter)
+        types = []
+
+    trigger_data = _write_dataset_viewer_trigger('setFilter', {'objectTypes': types})
+    output_json({
+        "status": "ok",
+        "action": "setFilter",
+        "objectTypes": types,
+        "triggerPath": DATASET_VIEWER_TRIGGER_PATH,
+    })
+
+
+def cmd_dataset_viewer_status(args):
+    """Get the current status of the dataset annotation viewer."""
+    if not os.path.exists(DATASET_VIEWER_STATUS_PATH):
+        output_json({
+            "status": "no_status_file",
+            "message": "No dataset viewer status file found. The viewer may not have been opened yet.",
+            "statusPath": DATASET_VIEWER_STATUS_PATH,
+            "sessions": [],
+        })
+        return
+
+    try:
+        with open(DATASET_VIEWER_STATUS_PATH, 'r') as f:
+            status_data = json.load(f)
+    except json.JSONDecodeError as e:
+        output_error(f"Failed to parse status file: {e}", "PARSE_ERROR")
+        return
+    except IOError as e:
+        output_error(f"Failed to read status file: {e}", "READ_ERROR")
+        return
+
+    # Optionally filter by dataset path
+    filter_path = getattr(args, 'path', None)
+    if filter_path:
+        filter_path = os.path.abspath(filter_path)
+        sessions = status_data.get('sessions', {})
+        if isinstance(sessions, dict):
+            filtered = {k: v for k, v in sessions.items() if os.path.abspath(k) == filter_path}
+            status_data['sessions'] = filtered
+
+    output_json({
+        "status": "ok",
+        "statusPath": DATASET_VIEWER_STATUS_PATH,
+        **status_data,
+    })
+
+
 # =============================================================================
 # GRAPH-EDITOR
 # =============================================================================
 
 def cmd_graph_editor_open(args):
-    """Download a graph and its channel schema, then open in the graph editor."""
-    client = get_client()
-    workspace_id = require_arg(args, 'workspaceid', 'Workspace ID')
-    graph_id = require_arg(args, 'graphid', 'Graph ID')
-    directory = args.outputdir or os.getcwd()
+    """Open a graph in the graph editor.
 
-    # Get graph metadata to find channelId
-    graphs = client.get_graphs(workspaceId=workspace_id, graphId=graph_id)
-    if not graphs:
-        output_error(f"Graph {graph_id} not found", "GRAPH_NOT_FOUND")
+    Supports two modes:
+    1. Download from platform: --workspaceid and --graphid (downloads graph and schema)
+    2. Local files: --graphfile and --schemafile (uses existing local files)
+    """
+    # Get arguments
+    workspace_id = getattr(args, 'workspaceid', None)
+    graph_id = getattr(args, 'graphid', None)
+    graph_file = getattr(args, 'graphfile', None)
+    schema_file = getattr(args, 'schemafile', None)
+
+    # Determine mode and validate arguments
+    has_platform_args = workspace_id or graph_id
+    has_local_args = graph_file or schema_file
+
+    if has_platform_args and has_local_args:
+        output_error(
+            "Cannot use both platform arguments (--workspaceid/--graphid) and local file arguments (--graphfile/--schemafile)",
+            "INVALID_ARGS"
+        )
         return
 
-    graph_info = graphs[0]
-    channel_id = graph_info.get('channelId')
-    if not channel_id:
-        output_error("Graph has no associated channel", "NO_CHANNEL")
+    if not has_platform_args and not has_local_args:
+        output_error(
+            "Must provide either --workspaceid and --graphid (to download from platform) or --graphfile and --schemafile (to use local files)",
+            "MISSING_ARGS"
+        )
         return
 
-    # Create output directory if needed
-    os.makedirs(directory, exist_ok=True)
+    # Mode 1: Local files
+    if has_local_args:
+        if not graph_file:
+            output_error("--graphfile is required when using local files", "MISSING_GRAPHFILE")
+            return
+        if not schema_file:
+            output_error("--schemafile is required when using local files", "MISSING_SCHEMAFILE")
+            return
+
+        # Validate files exist
+        if not os.path.exists(graph_file):
+            output_error(f"Graph file not found: {graph_file}", "FILE_NOT_FOUND")
+            return
+        if not os.path.exists(schema_file):
+            output_error(f"Schema file not found: {schema_file}", "FILE_NOT_FOUND")
+            return
 
-    # Download graph
-    graph_name = graph_info.get('name', graph_id).replace(' ', '_')
-    graph_path = os.path.join(directory, f"{graph_name}.yaml")
-    client.download_graph(workspaceId=workspace_id, graphId=graph_id, filepath=graph_path)
+        graph_path = graph_file
+        schema_path = schema_file
+        graph_id = None
+        channel_id = None
+        graph_name = os.path.basename(graph_file)
 
-    # Download channel schema
-    schema = client.get_channel_nodes(channelId=channel_id)
-    if not schema:
-        output_error("Failed to fetch channel schema", "SCHEMA_ERROR")
-        return
+    # Mode 2: Download from platform
+    else:
+        if not workspace_id:
+            output_error("--workspaceid is required when downloading from platform", "MISSING_WORKSPACEID")
+            return
+        if not graph_id:
+            output_error("--graphid is required when downloading from platform", "MISSING_GRAPHID")
+            return
+
+        client = get_client()
+        directory = args.outputdir or os.getcwd()
+
+        # Get graph metadata to find channelId
+        graphs = client.get_graphs(workspaceId=workspace_id, graphId=graph_id)
+        if not graphs:
+            output_error(f"Graph {graph_id} not found", "GRAPH_NOT_FOUND")
+            return
 
-    schema_path = os.path.join(directory, f"{channel_id}_schema.json")
-    with open(schema_path, 'w') as f:
-        json.dump(schema, f, indent=2)
+        graph_info = graphs[0]
+        channel_id = graph_info.get('channelId')
+        if not channel_id:
+            output_error("Graph has no associated channel", "NO_CHANNEL")
+            return
+
+        # Create output directory if needed
+        os.makedirs(directory, exist_ok=True)
+
+        # Download graph
+        graph_name = graph_info.get('name', graph_id).replace(' ', '_')
+        graph_path = os.path.join(directory, f"{graph_name}.yaml")
+        client.download_graph(workspaceId=workspace_id, graphId=graph_id, filepath=graph_path)
+
+        # Download channel schema
+        schema = client.get_channel_nodes(channelId=channel_id)
+        if not schema:
+            output_error("Failed to fetch channel schema", "SCHEMA_ERROR")
+            return
+
+        schema_path = os.path.join(directory, f"{channel_id}_schema.json")
+        with open(schema_path, 'w') as f:
+            json.dump(schema, f, indent=2)
 
     # Write trigger file to open in graph editor
     trigger_path = os.path.join(os.path.expanduser('~'), '.theia', 'graph-editor-open')
@@ -1139,14 +1384,25 @@ def cmd_graph_editor_open(args):
     with open(trigger_path, 'w') as f:
         json.dump(trigger_data, f)
 
-    output_json({
+    # Build output
+    result = {
         "graphPath": os.path.abspath(graph_path),
         "schemaPath": os.path.abspath(schema_path),
-        "triggerPath": trigger_path,
-        "graphId": graph_id,
-        "channelId": channel_id,
-        "graphName": graph_info.get('name')
-    })
+        "triggerPath": trigger_path
+    }
+
+    # Add platform-specific fields if downloaded
+    if graph_id:
+        result["graphId"] = graph_id
+    if channel_id:
+        result["channelId"] = channel_id
+    if has_local_args:
+        result["mode"] = "local"
+    else:
+        result["mode"] = "platform"
+        result["graphName"] = graph_info.get('name')
+
+    output_json(result)
 
 
 def cmd_graph_editor_edit_node(args):
@@ -2003,6 +2259,7 @@ def cmd_channels_get(args):
         organizationId=args.orgid,
         channelId=args.channelid,
         limit=args.limit,
+        cursor=args.cursor,
         fields=parse_list_arg(args.fields) if args.fields else None
     )
     output_json(result)
@@ -2135,6 +2392,7 @@ def cmd_services_get(args):
         organizationId=args.orgid,
         serviceId=args.serviceid,
         limit=args.limit,
+        cursor=args.cursor,
         fields=parse_list_arg(args.fields) if args.fields else None
     )
     output_json(result)
@@ -2191,6 +2449,7 @@ def cmd_services_jobs(args):
         workspaceId=workspace_id,
         jobId=args.jobid,
         limit=args.limit,
+        cursor=args.cursor,
         fields=parse_list_arg(args.fields) if args.fields else None
     )
     output_json(result)
@@ -2387,6 +2646,8 @@ def cmd_annotations_formats(args):
 
 def cmd_annotations_download(args):
     """Download an annotation."""
+    import zipfile
+
     client = get_client()
     workspace_id = require_arg(args, 'workspaceid', 'Workspace ID')
     annotation_id = require_arg(args, 'annotationid', 'Annotation ID')
@@ -2395,7 +2656,15 @@ def cmd_annotations_download(args):
         workspaceId=workspace_id,
         annotationId=annotation_id
     )
-    output_json({"downloadPath": result})
+
+    if args.extract and result and result.endswith('.zip') and os.path.isfile(result):
+        extract_dir = os.path.splitext(result)[0]
+        with zipfile.ZipFile(result, 'r') as zf:
+            zf.extractall(extract_dir)
+        os.remove(result)
+        output_json({"downloadPath": extract_dir, "extracted": True})
+    else:
+        output_json({"downloadPath": result})
 
 
 def cmd_annotations_edit(args):
@@ -2588,6 +2857,7 @@ def cmd_gan_datasets_get(args):
         datasetId=args.datasetid,
         gandatasetId=args.gandatasetid,
         limit=args.limit,
+        cursor=args.cursor,
         fields=parse_list_arg(args.fields) if args.fields else None
     )
     output_json(result)
@@ -2633,6 +2903,7 @@ def cmd_gan_models_get(args):
         workspaceId=args.workspaceid,
         modelId=args.modelid,
         limit=args.limit,
+        cursor=args.cursor,
         fields=parse_list_arg(args.fields) if args.fields else None
     )
     output_json(result)
@@ -2681,6 +2952,7 @@ def cmd_umap_get(args):
         umapId=args.umapid,
         datasetId=args.datasetid,
         limit=args.limit,
+        cursor=args.cursor,
         fields=parse_list_arg(args.fields) if args.fields else None
     )
     output_json(result)
@@ -2732,6 +3004,7 @@ def cmd_servers_get(args):
         workspaceId=args.workspaceid,
         serverId=args.serverid,
         limit=args.limit,
+        cursor=args.cursor,
         fields=parse_list_arg(args.fields) if args.fields else None
     )
     output_json(result)
@@ -2801,6 +3074,7 @@ def cmd_ml_models_get(args):
         datasetId=args.datasetid,
         modelId=args.modelid,
         limit=args.limit,
+        cursor=args.cursor,
         fields=parse_list_arg(args.fields) if args.fields else None
     )
     output_json(result)
@@ -2852,6 +3126,7 @@ def cmd_ml_inferences_get(args):
         datasetId=args.datasetid,
         modelId=args.modelid,
         limit=args.limit,
+        cursor=args.cursor,
         fields=parse_list_arg(args.fields) if args.fields else None
     )
     output_json(result)
@@ -2887,6 +3162,7 @@ def cmd_inpaint_get(args):
         volumeId=volume_id,
         inpaintId=args.inpaintid,
         limit=args.limit,
+        cursor=args.cursor,
         fields=parse_list_arg(args.fields) if args.fields else None
     )
     output_json(result)
@@ -2947,12 +3223,40 @@ def cmd_preview_get(args):
     workspace_id = require_arg(args, 'workspaceid', 'Workspace ID')
     preview_id = require_arg(args, 'previewid', 'Preview ID')
 
-    result = client.get_preview(
-        workspaceId=workspace_id,
-        previewId=preview_id,
-        fields=parse_list_arg(args.fields) if args.fields else None
-    )
-    output_json(result)
+    if args.download:
+        # Fetch preview with thumbnail and status fields for download
+        fields = parse_list_arg(args.fields) if args.fields else None
+        if fields is not None:
+            for f in ['thumbnail', 'status']:
+                if f not in fields:
+                    fields.append(f)
+        result = client.get_preview(
+            workspaceId=workspace_id,
+            previewId=preview_id,
+            fields=fields
+        )
+        status = result.get('status') if isinstance(result, dict) else None
+        thumbnail = result.get('thumbnail') if isinstance(result, dict) else None
+        if status != 'success':
+            output_error(f"Preview is not complete (status: {status}). Cannot download.", "PREVIEW_NOT_READY")
+            sys.exit(1)
+        if not thumbnail:
+            output_error("Preview has no thumbnail URL available.", "NO_THUMBNAIL")
+            sys.exit(1)
+        from anatools.lib.download import download_file
+        # Derive filename from URL or use default
+        from urllib.parse import urlparse
+        url_path = urlparse(thumbnail).path
+        fname = os.path.basename(url_path) if os.path.basename(url_path) else 'preview.png'
+        download_path = download_file(url=thumbnail, fname=fname, localDir=args.outputdir)
+        output_json({"downloadPath": download_path})
+    else:
+        result = client.get_preview(
+            workspaceId=workspace_id,
+            previewId=preview_id,
+            fields=parse_list_arg(args.fields) if args.fields else None
+        )
+        output_json(result)
 
 
 def cmd_preview_create(args):
@@ -2968,6 +3272,20 @@ def cmd_preview_create(args):
     output_json({"previewId": result})
 
 
+def cmd_preview_log(args):
+    """Get preview job log."""
+    client = get_client()
+    workspace_id = require_arg(args, 'workspaceid', 'Workspace ID')
+    preview_id = require_arg(args, 'previewid', 'Preview ID')
+
+    result = client.get_preview_log(
+        workspaceId=workspace_id,
+        previewId=preview_id,
+        fields=parse_list_arg(args.fields) if args.fields else None
+    )
+    output_json(result)
+
+
 # =============================================================================
 # AGENTS
 # =============================================================================
@@ -3108,7 +3426,8 @@ Examples:
     ws_get = workspaces_sub.add_parser('get', help='Get workspaces')
     ws_get.add_argument('--workspaceid', help='Filter by workspace ID')
     ws_get.add_argument('--orgid', help='Filter by organization ID')
-    ws_get.add_argument('--limit', type=int, help='Maximum results')
+    ws_get.add_argument('--limit', type=int, default=50, help='Maximum results (default: 50)')
+    ws_get.add_argument('--cursor', help='Cursor for pagination (use last item ID from previous page)')
     ws_get.add_argument('--fields', help='Comma-separated fields to return')
     ws_get.set_defaults(func=cmd_workspaces_get)
 
@@ -3158,7 +3477,8 @@ Examples:
     # organizations get
     org_get = organizations_sub.add_parser('get', help='Get organizations')
     org_get.add_argument('--orgid', help='Filter by organization ID')
-    org_get.add_argument('--limit', type=int, help='Maximum results')
+    org_get.add_argument('--limit', type=int, default=50, help='Maximum results (default: 50)')
+    org_get.add_argument('--cursor', help='Cursor for pagination (use last item ID from previous page)')
     org_get.add_argument('--fields', help='Comma-separated fields to return')
     org_get.set_defaults(func=cmd_organizations_get)
 
@@ -3171,7 +3491,8 @@ Examples:
     # members get
     members_get = members_sub.add_parser('get', help='Get organization members')
     members_get.add_argument('--orgid', required=True, help='Organization ID')
-    members_get.add_argument('--limit', type=int, help='Maximum results')
+    members_get.add_argument('--limit', type=int, default=50, help='Maximum results (default: 50)')
+    members_get.add_argument('--cursor', help='Cursor for pagination (use last item ID from previous page)')
     members_get.add_argument('--fields', help='Comma-separated fields to return')
     members_get.set_defaults(func=cmd_members_get)
 
@@ -3185,7 +3506,8 @@ Examples:
     ds_get = datasets_sub.add_parser('get', help='Get datasets')
     ds_get.add_argument('--workspaceid', required=True, help='Workspace ID')
     ds_get.add_argument('--datasetid', help='Filter by dataset ID')
-    ds_get.add_argument('--limit', type=int, help='Maximum results')
+    ds_get.add_argument('--limit', type=int, default=50, help='Maximum results (default: 50)')
+    ds_get.add_argument('--cursor', help='Cursor for pagination (use last item ID from previous page)')
     ds_get.add_argument('--fields', help='Comma-separated fields to return')
     ds_get.set_defaults(func=cmd_datasets_get)
 
@@ -3230,6 +3552,7 @@ Examples:
     ds_download.add_argument('--datasetid', required=True, help='Dataset ID')
     ds_download.add_argument('--filepath', help='Relative path to a specific file within the dataset (e.g., "images/000000-1-image.png"). If not provided, downloads the entire dataset.')
     ds_download.add_argument('--outputdir', help='Output directory')
+    ds_download.add_argument('--extract', action='store_true', help='Extract the downloaded zip file and remove the archive')
     ds_download.set_defaults(func=cmd_datasets_download)
 
     # datasets upload
@@ -3261,7 +3584,8 @@ Examples:
     ds_files.add_argument('--workspaceid', required=True, help='Workspace ID')
     ds_files.add_argument('--datasetid', required=True, help='Dataset ID')
     ds_files.add_argument('--path', help='Path within dataset')
-    ds_files.add_argument('--limit', type=int, default=100, help='Maximum results')
+    ds_files.add_argument('--limit', type=int, default=50, help='Maximum results (default: 50)')
+    ds_files.add_argument('--cursor', help='Cursor for pagination (use last item ID from previous page)')
     ds_files.set_defaults(func=cmd_datasets_files)
 
     # datasets jobs
@@ -3269,7 +3593,8 @@ Examples:
     ds_jobs.add_argument('--workspaceid', help='Workspace ID')
     ds_jobs.add_argument('--orgid', help='Organization ID')
     ds_jobs.add_argument('--datasetid', help='Filter by dataset ID')
-    ds_jobs.add_argument('--limit', type=int, help='Maximum results')
+    ds_jobs.add_argument('--limit', type=int, default=50, help='Maximum results (default: 50)')
+    ds_jobs.add_argument('--cursor', help='Cursor for pagination (use last item ID from previous page)')
     ds_jobs.add_argument('--fields', help='Comma-separated fields to return')
     ds_jobs.set_defaults(func=cmd_datasets_jobs)
 
@@ -3294,7 +3619,8 @@ Examples:
     vol_get.add_argument('--volumeid', help='Filter by volume ID')
     vol_get.add_argument('--workspaceid', help='Filter by workspace ID')
     vol_get.add_argument('--orgid', help='Filter by organization ID')
-    vol_get.add_argument('--limit', type=int, help='Maximum results')
+    vol_get.add_argument('--limit', type=int, default=50, help='Maximum results (default: 50)')
+    vol_get.add_argument('--cursor', help='Cursor for pagination (use last item ID from previous page)')
     vol_get.add_argument('--fields', help='Comma-separated fields to return')
     vol_get.set_defaults(func=cmd_volumes_get)
 
@@ -3345,7 +3671,8 @@ Examples:
     vd_get.add_argument('--dir', help='Directory path')
     vd_get.add_argument('--files', help='Comma-separated file paths')
     vd_get.add_argument('--recursive', action='store_true', help='Recursive listing')
-    vd_get.add_argument('--limit', type=int, help='Maximum results')
+    vd_get.add_argument('--limit', type=int, default=50, help='Maximum results (default: 50)')
+    vd_get.add_argument('--cursor', help='Cursor for pagination (use last item ID from previous page)')
     vd_get.set_defaults(func=cmd_volume_data_get)
 
     # volume-data upload
@@ -3380,7 +3707,8 @@ Examples:
     vd_search.add_argument('--keywords', help='Comma-separated keywords')
     vd_search.add_argument('--formats', help='Comma-separated file formats (e.g., png,jpg)')
     vd_search.add_argument('--types', help='Comma-separated file types (e.g., Image,3D)')
-    vd_search.add_argument('--limit', type=int, help='Maximum results')
+    vd_search.add_argument('--limit', type=int, default=50, help='Maximum results (default: 50)')
+    vd_search.add_argument('--cursor', help='Cursor for pagination (use last item ID from previous page)')
     vd_search.set_defaults(func=cmd_volume_data_search)
 
     # -------------------------------------------------------------------------
@@ -3394,7 +3722,8 @@ Examples:
     gr_get.add_argument('--workspaceid', required=True, help='Workspace ID')
     gr_get.add_argument('--graphid', help='Filter by graph ID')
     gr_get.add_argument('--staged', action='store_true', help='Only staged graphs')
-    gr_get.add_argument('--limit', type=int, help='Maximum results')
+    gr_get.add_argument('--limit', type=int, default=50, help='Maximum results (default: 50)')
+    gr_get.add_argument('--cursor', help='Cursor for pagination (use last item ID from previous page)')
     gr_get.add_argument('--fields', help='Comma-separated fields to return')
     gr_get.set_defaults(func=cmd_graphs_get)
 
@@ -3444,10 +3773,14 @@ Examples:
     graph_editor_sub = graph_editor.add_subparsers(dest='action', help='Action')
 
     # graph-editor open
-    ge_open = graph_editor_sub.add_parser('open', help='Download graph and schema, open in graph editor')
-    ge_open.add_argument('--workspaceid', required=True, help='Workspace ID')
-    ge_open.add_argument('--graphid', required=True, help='Graph ID')
-    ge_open.add_argument('--outputdir', help='Output directory (default: current directory)')
+    ge_open = graph_editor_sub.add_parser('open', help='Open graph in editor (download from platform or use local files)')
+    # Option 1: Download from platform
+    ge_open.add_argument('--workspaceid', help='Workspace ID (use with --graphid to download from platform)')
+    ge_open.add_argument('--graphid', help='Graph ID (use with --workspaceid to download from platform)')
+    ge_open.add_argument('--outputdir', help='Output directory for downloaded files (default: current directory)')
+    # Option 2: Use local files
+    ge_open.add_argument('--graphfile', help='Path to local graph file (use with --schemafile)')
+    ge_open.add_argument('--schemafile', help='Path to local schema file (use with --graphfile)')
     ge_open.set_defaults(func=cmd_graph_editor_open)
 
     # graph-editor edit-node
@@ -3544,6 +3877,47 @@ Examples:
     ge_status.add_argument('--file', help='Filter to a specific graph file path')
     ge_status.set_defaults(func=cmd_graph_editor_status)
 
+    # -------------------------------------------------------------------------
+    # DATASET-VIEWER
+    # -------------------------------------------------------------------------
+    dataset_viewer = subparsers.add_parser('dataset-viewer', help='Dataset annotation viewer integration')
+    dv_sub = dataset_viewer.add_subparsers(dest='action', help='Action')
+
+    # dataset-viewer open
+    dv_open = dv_sub.add_parser('open', help='Open a dataset folder in the Annotation Viewer')
+    dv_open.add_argument('--path', required=True, help='Path to dataset directory (must contain images/ subdirectory)')
+    dv_open.add_argument('--index', type=int, default=0, help='Initial image index (default: 0)')
+    dv_open.set_defaults(func=cmd_dataset_viewer_open)
+
+    # dataset-viewer next
+    dv_next = dv_sub.add_parser('next', help='Navigate to the next image')
+    dv_next.set_defaults(func=cmd_dataset_viewer_next)
+
+    # dataset-viewer prev
+    dv_prev = dv_sub.add_parser('prev', help='Navigate to the previous image')
+    dv_prev.set_defaults(func=cmd_dataset_viewer_prev)
+
+    # dataset-viewer goto
+    dv_goto = dv_sub.add_parser('goto', help='Navigate to a specific image by index or name')
+    dv_goto.add_argument('--index', type=int, help='Image index (0-based)')
+    dv_goto.add_argument('--name', help='Image filename (or partial match)')
+    dv_goto.set_defaults(func=cmd_dataset_viewer_goto)
+
+    # dataset-viewer annotations
+    dv_annotations = dv_sub.add_parser('annotations', help='Set which annotation types are displayed')
+    dv_annotations.add_argument('--types', required=True, help='Comma-separated annotation types: bbox,bbox3d,segmentation,centroid,mask')
+    dv_annotations.set_defaults(func=cmd_dataset_viewer_annotations)
+
+    # dataset-viewer filter
+    dv_filter = dv_sub.add_parser('filter', help='Filter visible objects by type (omit --types to show all)')
+    dv_filter.add_argument('--types', help='Comma-separated object type names to show (omit to clear filter)')
+    dv_filter.set_defaults(func=cmd_dataset_viewer_filter)
+
+    # dataset-viewer status
+    dv_status = dv_sub.add_parser('status', help='Get dataset viewer status')
+    dv_status.add_argument('--path', help='Filter to a specific dataset path')
+    dv_status.set_defaults(func=cmd_dataset_viewer_status)
+
     # -------------------------------------------------------------------------
     # CHANNELS
     # -------------------------------------------------------------------------
@@ -3555,7 +3929,8 @@ Examples:
     ch_get.add_argument('--workspaceid', help='Filter by workspace ID')
     ch_get.add_argument('--orgid', help='Filter by organization ID')
     ch_get.add_argument('--channelid', help='Filter by channel ID')
-    ch_get.add_argument('--limit', type=int, help='Maximum results')
+    ch_get.add_argument('--limit', type=int, default=50, help='Maximum results (default: 50)')
+    ch_get.add_argument('--cursor', help='Cursor for pagination (use last item ID from previous page)')
     ch_get.add_argument('--fields', help='Comma-separated fields to return')
     ch_get.set_defaults(func=cmd_channels_get)
 
@@ -3605,7 +3980,8 @@ Examples:
     svc_get.add_argument('--workspaceid', help='Filter by workspace ID')
     svc_get.add_argument('--orgid', help='Filter by organization ID')
     svc_get.add_argument('--serviceid', help='Filter by service ID')
-    svc_get.add_argument('--limit', type=int, help='Maximum results')
+    svc_get.add_argument('--limit', type=int, default=50, help='Maximum results (default: 50)')
+    svc_get.add_argument('--cursor', help='Cursor for pagination (use last item ID from previous page)')
     svc_get.add_argument('--fields', help='Comma-separated fields to return')
     svc_get.set_defaults(func=cmd_services_get)
 
@@ -3645,7 +4021,8 @@ Examples:
     svc_jobs_get = service_jobs_sub.add_parser('get', help='Get service jobs')
     svc_jobs_get.add_argument('--workspaceid', required=True, help='Workspace ID')
     svc_jobs_get.add_argument('--jobid', help='Filter by job ID')
-    svc_jobs_get.add_argument('--limit', type=int, help='Maximum results')
+    svc_jobs_get.add_argument('--limit', type=int, default=50, help='Maximum results (default: 50)')
+    svc_jobs_get.add_argument('--cursor', help='Cursor for pagination (use last item ID from previous page)')
     svc_jobs_get.add_argument('--fields', help='Comma-separated fields to return')
     svc_jobs_get.set_defaults(func=cmd_services_jobs)
 
@@ -3747,6 +4124,7 @@ Examples:
     ann_download = annotations_sub.add_parser('download', help='Download an annotation')
     ann_download.add_argument('--workspaceid', required=True, help='Workspace ID')
     ann_download.add_argument('--annotationid', required=True, help='Annotation ID')
+    ann_download.add_argument('--extract', action='store_true', help='Extract the downloaded zip file and remove the archive')
     ann_download.set_defaults(func=cmd_annotations_download)
 
     # annotations formats
@@ -3827,7 +4205,8 @@ Examples:
     gan_models_get.add_argument('--orgid', help='Organization ID')
     gan_models_get.add_argument('--workspaceid', help='Workspace ID')
     gan_models_get.add_argument('--modelid', help='Model ID')
-    gan_models_get.add_argument('--limit', type=int, help='Maximum results')
+    gan_models_get.add_argument('--limit', type=int, default=50, help='Maximum results (default: 50)')
+    gan_models_get.add_argument('--cursor', help='Cursor for pagination (use last item ID from previous page)')
     gan_models_get.add_argument('--fields', help='Comma-separated fields')
     gan_models_get.set_defaults(func=cmd_gan_models_get)
 
@@ -3858,7 +4237,8 @@ Examples:
     gan_ds_get.add_argument('--workspaceid', required=True, help='Workspace ID')
     gan_ds_get.add_argument('--datasetid', help='Dataset ID')
     gan_ds_get.add_argument('--gandatasetid', help='GAN dataset ID')
-    gan_ds_get.add_argument('--limit', type=int, help='Maximum results')
+    gan_ds_get.add_argument('--limit', type=int, default=50, help='Maximum results (default: 50)')
+    gan_ds_get.add_argument('--cursor', help='Cursor for pagination (use last item ID from previous page)')
     gan_ds_get.add_argument('--fields', help='Comma-separated fields')
     gan_ds_get.set_defaults(func=cmd_gan_datasets_get)
 
@@ -3889,7 +4269,8 @@ Examples:
     umap_get.add_argument('--workspaceid', required=True, help='Workspace ID')
     umap_get.add_argument('--umapid', help='UMAP ID')
     umap_get.add_argument('--datasetid', help='Dataset ID')
-    umap_get.add_argument('--limit', type=int, help='Maximum results')
+    umap_get.add_argument('--limit', type=int, default=50, help='Maximum results (default: 50)')
+    umap_get.add_argument('--cursor', help='Cursor for pagination (use last item ID from previous page)')
     umap_get.add_argument('--fields', help='Comma-separated fields')
     umap_get.set_defaults(func=cmd_umap_get)
 
@@ -3921,7 +4302,8 @@ Examples:
     srv_get.add_argument('--orgid', help='Organization ID')
     srv_get.add_argument('--workspaceid', help='Workspace ID')
     srv_get.add_argument('--serverid', help='Server ID')
-    srv_get.add_argument('--limit', type=int, help='Maximum results')
+    srv_get.add_argument('--limit', type=int, default=50, help='Maximum results (default: 50)')
+    srv_get.add_argument('--cursor', help='Cursor for pagination (use last item ID from previous page)')
     srv_get.add_argument('--fields', help='Comma-separated fields')
     srv_get.set_defaults(func=cmd_servers_get)
 
@@ -3964,7 +4346,8 @@ Examples:
     ml_models_get.add_argument('--workspaceid', required=True, help='Workspace ID')
     ml_models_get.add_argument('--datasetid', help='Dataset ID')
     ml_models_get.add_argument('--modelid', help='Model ID')
-    ml_models_get.add_argument('--limit', type=int, help='Maximum results')
+    ml_models_get.add_argument('--limit', type=int, default=50, help='Maximum results (default: 50)')
+    ml_models_get.add_argument('--cursor', help='Cursor for pagination (use last item ID from previous page)')
     ml_models_get.add_argument('--fields', help='Comma-separated fields')
     ml_models_get.set_defaults(func=cmd_ml_models_get)
 
@@ -3999,7 +4382,8 @@ Examples:
     ml_inf_get.add_argument('--inferenceid', help='Inference ID')
     ml_inf_get.add_argument('--datasetid', help='Dataset ID')
     ml_inf_get.add_argument('--modelid', help='Model ID')
-    ml_inf_get.add_argument('--limit', type=int, help='Maximum results')
+    ml_inf_get.add_argument('--limit', type=int, default=50, help='Maximum results (default: 50)')
+    ml_inf_get.add_argument('--cursor', help='Cursor for pagination (use last item ID from previous page)')
     ml_inf_get.add_argument('--fields', help='Comma-separated fields')
     ml_inf_get.set_defaults(func=cmd_ml_inferences_get)
 
@@ -4022,7 +4406,8 @@ Examples:
     inp_get = inpaint_sub.add_parser('get', help='Get inpaint jobs')
     inp_get.add_argument('--volumeid', required=True, help='Volume ID')
     inp_get.add_argument('--inpaintid', help='Inpaint ID')
-    inp_get.add_argument('--limit', type=int, help='Maximum results')
+    inp_get.add_argument('--limit', type=int, default=50, help='Maximum results (default: 50)')
+    inp_get.add_argument('--cursor', help='Cursor for pagination (use last item ID from previous page)')
     inp_get.add_argument('--fields', help='Comma-separated fields')
     inp_get.set_defaults(func=cmd_inpaint_get)
 
@@ -4061,6 +4446,8 @@ Examples:
     prv_get.add_argument('--workspaceid', required=True, help='Workspace ID')
     prv_get.add_argument('--previewid', required=True, help='Preview ID')
     prv_get.add_argument('--fields', help='Comma-separated fields')
+    prv_get.add_argument('--download', action='store_true', help='Download the preview thumbnail image')
+    prv_get.add_argument('--outputdir', help='Output directory for downloaded preview (default: current directory)')
     prv_get.set_defaults(func=cmd_preview_get)
 
     # preview create
@@ -4069,6 +4456,13 @@ Examples:
     prv_create.add_argument('--graphid', required=True, help='Graph ID')
     prv_create.set_defaults(func=cmd_preview_create)
 
+    # preview log
+    prv_log = preview_sub.add_parser('log', help='Get preview job log')
+    prv_log.add_argument('--workspaceid', required=True, help='Workspace ID')
+    prv_log.add_argument('--previewid', required=True, help='Preview ID')
+    prv_log.add_argument('--fields', help='Comma-separated fields')
+    prv_log.set_defaults(func=cmd_preview_log)
+
     # -------------------------------------------------------------------------
     # AGENTS
     # -------------------------------------------------------------------------