pltr-cli 0.3.0__py3-none-any.whl → 0.5.0__py3-none-any.whl

{pltr_cli-0.3.0.dist-info → pltr_cli-0.5.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pltr-cli
-Version: 0.3.0
+Version: 0.5.0
 Summary: Command-line interface for Palantir Foundry APIs
 Project-URL: Homepage, https://github.com/anjor/pltr-cli
 Project-URL: Repository, https://github.com/anjor/pltr-cli
@@ -37,7 +37,7 @@ Description-Content-Type: text/markdown
 
 # pltr-cli
 
-A comprehensive command-line interface for Palantir Foundry APIs, providing 65+ commands for data analysis, ontology operations, SQL queries, and administrative tasks.
+A comprehensive command-line interface for Palantir Foundry APIs, providing 80+ commands for data analysis, dataset management, ontology operations, orchestration, SQL queries, folder management, and administrative tasks.
 
 ## Overview
 
@@ -46,8 +46,11 @@ A comprehensive command-line interface for Palantir Foundry APIs, providing 65+
 ## ✨ Key Features
 
 - 🔐 **Secure Authentication**: Token and OAuth2 support with encrypted credential storage
-- 📊 **Dataset Operations**: Get dataset information and create new datasets (RID-based API)
+- 📊 **Dataset Operations**: Complete dataset management with branches, files, transactions, and views (RID-based API)
+- 📁 **Folder Management**: Create, explore, and manage Foundry filesystem structure
 - 🎯 **Comprehensive Ontology Access**: 13 commands for objects, actions, and queries
+- 🏗️ **Orchestration Management**: Create, manage, and monitor builds, jobs, and schedules
+- 🎬 **MediaSets Operations**: Upload, download, and manage media content with transaction support
 - 📝 **Full SQL Support**: Execute, submit, monitor, and export query results
 - 👥 **Admin Operations**: User, group, role, and organization management (16 commands)
 - 💻 **Interactive Shell**: REPL mode with tab completion and command history
@@ -109,9 +112,23 @@ pltr admin user current
 # List available ontologies
 pltr ontology list
 
+# Search for builds
+pltr orchestration builds search
+
+# Create a new folder
+pltr folder create "My Project"
+
+# List root folder contents
+pltr folder list ri.compass.main.folder.0
+
 # Execute a simple SQL query
 pltr sql execute "SELECT 1 as test"
 
+# Explore dataset operations (requires dataset RID)
+pltr dataset get ri.foundry.main.dataset.abc123
+pltr dataset branches list ri.foundry.main.dataset.abc123
+pltr dataset files list ri.foundry.main.dataset.abc123
+
 # Start interactive mode for exploration
 pltr shell
 ```
@@ -133,7 +150,7 @@ pltr-cli provides comprehensive documentation to help you get the most out of th
 ### 📖 User Guides
 - **[Quick Start Guide](docs/user-guide/quick-start.md)** - Get up and running in 5 minutes
 - **[Authentication Setup](docs/user-guide/authentication.md)** - Complete guide to token and OAuth2 setup
-- **[Command Reference](docs/user-guide/commands.md)** - Complete reference for all 65+ commands
+- **[Command Reference](docs/user-guide/commands.md)** - Complete reference for all 70+ commands
 - **[Common Workflows](docs/user-guide/workflows.md)** - Real-world data analysis patterns
 - **[Troubleshooting](docs/user-guide/troubleshooting.md)** - Solutions to common issues
 
@@ -154,6 +171,16 @@ pltr sql execute "SELECT * FROM table"  # Run SQL queries
 pltr ontology list             # List ontologies
 pltr dataset get <rid>         # Get dataset info
 
+# Orchestration
+pltr orchestration builds search       # Search builds
+pltr orchestration jobs get <job-rid>  # Get job details
+pltr orchestration schedules create   # Create schedule
+
+# MediaSets
+pltr media-sets get <set-rid> <item-rid>  # Get media item info
+pltr media-sets upload <set-rid> file.jpg "/path/file.jpg" <txn-id>  # Upload media
+pltr media-sets download <set-rid> <item-rid> output.jpg  # Download media
+
 # Administrative
 pltr admin user current        # Current user info
 pltr admin user list          # List users
@@ -167,6 +194,113 @@ pltr completion install       # Enable tab completion
 
 For the complete command reference with examples, see **[Command Reference](docs/user-guide/commands.md)**.
 
+### 🏗️ Orchestration Commands
+
+pltr-cli provides comprehensive support for Foundry's Orchestration module:
+
+#### Build Management
+```bash
+# Search for builds
+pltr orchestration builds search
+
+# Get build details
+pltr orchestration builds get ri.orchestration.main.build.12345
+
+# Create a new build
+pltr orchestration builds create '{"dataset_rid": "ri.foundry.main.dataset.abc"}' --branch main
+
+# Cancel a running build
+pltr orchestration builds cancel ri.orchestration.main.build.12345
+
+# List jobs in a build
+pltr orchestration builds jobs ri.orchestration.main.build.12345
+```
+
+#### Job Management
+```bash
+# Get job details
+pltr orchestration jobs get ri.orchestration.main.job.12345
+
+# Get multiple jobs in batch
+pltr orchestration jobs get-batch "rid1,rid2,rid3"
+```
+
+#### Schedule Management
+```bash
+# Get schedule information
+pltr orchestration schedules get ri.orchestration.main.schedule.12345
+
+# Create a new schedule
+pltr orchestration schedules create '{"type": "BUILD", "target": "dataset-rid"}' \
+  --name "Daily Build" --description "Automated daily build"
+
+# Pause/unpause schedules
+pltr orchestration schedules pause ri.orchestration.main.schedule.12345
+pltr orchestration schedules unpause ri.orchestration.main.schedule.12345
+
+# Execute schedule immediately
+pltr orchestration schedules run ri.orchestration.main.schedule.12345
+
+# Delete a schedule
+pltr orchestration schedules delete ri.orchestration.main.schedule.12345 --yes
+```
+
+**All orchestration commands support:**
+- Multiple output formats (table, JSON, CSV)
+- File output (`--output filename`)
+- Profile selection (`--profile production`)
+- Preview mode for schedules (`--preview`)
+
+### 🎬 MediaSets Commands
+
+pltr-cli provides full support for Foundry's MediaSets module for managing media content:
+
+#### Media Item Operations
+```bash
+# Get media item information
+pltr media-sets get ri.mediasets.main.media-set.abc ri.mediasets.main.media-item.123
+
+# Get media item RID by path
+pltr media-sets get-by-path ri.mediasets.main.media-set.abc "/images/photo.jpg"
+
+# Get a reference for embedding
+pltr media-sets reference ri.mediasets.main.media-set.abc ri.mediasets.main.media-item.123
+```
+
+#### Transaction Management
+```bash
+# Create a new upload transaction
+pltr media-sets create ri.mediasets.main.media-set.abc --branch main
+
+# Commit transaction (makes uploads available)
+pltr media-sets commit ri.mediasets.main.media-set.abc transaction-id-12345
+
+# Abort transaction (deletes uploads)
+pltr media-sets abort ri.mediasets.main.media-set.abc transaction-id-12345 --yes
+```
+
+#### Upload and Download
+```bash
+# Upload a file to media set
+pltr media-sets upload ri.mediasets.main.media-set.abc \
+  /local/path/image.jpg "/media/images/image.jpg" transaction-id-12345
+
+# Download media item (processed version)
+pltr media-sets download ri.mediasets.main.media-set.abc \
+  ri.mediasets.main.media-item.123 /local/download/image.jpg
+
+# Download original version
+pltr media-sets download ri.mediasets.main.media-set.abc \
+  ri.mediasets.main.media-item.123 /local/download/original.jpg --original
+```
+
+**All MediaSets commands support:**
+- Multiple output formats (table, JSON, CSV)
+- File output (`--output filename`)
+- Profile selection (`--profile production`)
+- Preview mode (`--preview`)
+- Transaction-based upload workflow
+
 ## ⚙️ Configuration
 
 pltr-cli stores configuration securely using industry best practices:
@@ -237,7 +371,7 @@ See **[API Wrapper Documentation](docs/api/wrapper.md)** for detailed architectu
 
 pltr-cli is **production-ready** with comprehensive features:
 
-- ✅ **65+ Commands** across 8 command groups
+- ✅ **80+ Commands** across 10 command groups
 - ✅ **273 Unit Tests** with 67% code coverage
 - ✅ **Published on PyPI** with automated releases
 - ✅ **Cross-Platform** support (Windows, macOS, Linux)

{pltr_cli-0.3.0.dist-info → pltr_cli-0.5.0.dist-info}/RECORD

@@ -1,6 +1,6 @@
 pltr/__init__.py,sha256=kUR5RAFc7HCeiqdlX36dZOHkUI5wI6V_43RpEcD8b-0,22
 pltr/__main__.py,sha256=HWJ49UoAYBQCf8kjuySPmBTuUjTZrOx-y6PzMTyS1KE,879
-pltr/cli.py,sha256=Kb9237LDITTX1VU-X-O5eKn8dXKI0R28HokYXGGklNU,1879
+pltr/cli.py,sha256=EIW67XC8-AttN6S2OUxaxEC2trYDxaHVy8j18G92EzU,2191
 pltr/auth/__init__.py,sha256=G0V-Rh25FaJsH2nhrf146XQQG_ApdbyPJNuHJC25kgk,38
 pltr/auth/base.py,sha256=LvmCwS7A0q0CITcym8udPzdACL52_jSGusiaeCTOaE8,981
 pltr/auth/manager.py,sha256=ZqlGefr1a8MGx0g7kkQhpmiuVp0XTg3f43yMBCk-IRo,4305
@@ -12,8 +12,11 @@ pltr/commands/admin.py,sha256=foscSO-QuH6uggUR5Rmv9pTqGjEXTUzpmMFj2-8hEJs,17065
 pltr/commands/alias.py,sha256=r9xMsQNrGvaixSlspzoO2IXQ44LFXuZM4itt8vC0dRc,6862
 pltr/commands/completion.py,sha256=YTxaRL4-rDs5n7aXf3ogFsxbHVJUBo_HiBbd0fbBPZ0,10870
 pltr/commands/configure.py,sha256=oYj-VlOEj3MDwtB2RC4bYOYzI_sXTanPnz7y1GmMTqY,4800
-pltr/commands/dataset.py,sha256=BCYfaBpLji5wasOiH_jOqO-JC9ScfJhodox9kl9W2Cw,3609
+pltr/commands/dataset.py,sha256=3QWqClosXna7cpUxOQ93mNIYTAoHMReeePCdzY44lEk,14112
+pltr/commands/folder.py,sha256=IAPPA3Smk1IWqThneEtZ08Zp79vDKVUabSkL_nDvUWk,10679
+pltr/commands/mediasets.py,sha256=FXq7OtYU9wLgUxQFcS_fkA4i_CozGnsYKxh8GOSI0ok,15342
 pltr/commands/ontology.py,sha256=zUgSrmv8xi26SQK7GsM3qusgR9Wuka0GyzE7L8DkduE,18317
+pltr/commands/orchestration.py,sha256=4gq5nb43bU0Ub3iaKV-EgXT_ka8ilwdyxY_9M5iS84s,21958
 pltr/commands/shell.py,sha256=QLF7TEGpaau9i0A9s3VjnegvtHde-SLRqI4suJFT4WI,3622
 pltr/commands/sql.py,sha256=wol0Rlvi_RplCFbOg4LCa3VXsOqmRZdFFVv7V6iVkh8,12602
 pltr/commands/verify.py,sha256=n8LWhbfGieYa-a5_l3MxqkYbdpyVf8-i0FQIL__AaPA,6650
@@ -24,18 +27,19 @@ pltr/config/settings.py,sha256=bfIiosPqH_W73TOHS71DvgZdAHka4fJDopU1SvBRFuQ,2908
 pltr/services/__init__.py,sha256=zQpgrqPdAkZI-nobi33mctU2-iGNgazzvjBVY8YRbSQ,101
 pltr/services/admin.py,sha256=8FjExmDeIKeVqkAxM83SVvpp_pH9W-Q33cgVs6BHxLQ,9957
 pltr/services/base.py,sha256=R2G781FI-sXtjUyLd91bVnmLb4cYZI3G8U5ndR9NLA4,1593
-pltr/services/dataset.py,sha256=W3zoh-9YIJ6HBsDijejVEngKvpudFoZinYtHDmAXCOc,2785
-pltr/services/dataset_full.py,sha256=FyMiwOSyX1cUrYXaK0T_1iq5G_X0e5iZTibJHuEmMeE,9869
-pltr/services/dataset_v2.py,sha256=_uhcVJ91w_Y07glceqHceccAwPWr6q1TWSIqcP1FU8I,4259
+pltr/services/dataset.py,sha256=cjl3aUhFcmWz_4ED8_h-Fl4Ym8ZMuA0dolOh89M7ufc,15412
+pltr/services/folder.py,sha256=mWElyvn-wXPB5sv8Ik_dLeW5JM6jZg3g9KKBk6UcrlQ,5389
+pltr/services/mediasets.py,sha256=HgHNFWoG9r-5xupANVOxHg_h5EKsBDl6PsO8hwdbm28,9854
 pltr/services/ontology.py,sha256=iW7qRK8ptlw-u4eAwLNC-mdzLoLZzh7SRqJyok2c3GU,14883
+pltr/services/orchestration.py,sha256=t5RNC0BoKcqeYKJQ5AQAyyO_dIYCHxiHV177JVSszhM,14729
 pltr/services/sql.py,sha256=19cscjlzN8WE1s8_ctiRcrOvMzCfmWRj49vjJ8Gs5Q4,11286
 pltr/utils/__init__.py,sha256=DF7TigL1XbKVGM5VjgU8_8AGIszofkdO80oCzLGGnTE,38
 pltr/utils/alias_resolver.py,sha256=DIF7P1UnUU8kqocJfIDEWjYq4s8_0KfqRZBbECeZEh8,1539
 pltr/utils/completion.py,sha256=bjeqjleEfB2YcQFpcxvF0GoQ763F6KBbULSZC4FWY_g,4980
-pltr/utils/formatting.py,sha256=Ptv7obiB7uWNf5YTiY4rUAKmabO5UGnwLocSfLplrZ0,18552
+pltr/utils/formatting.py,sha256=uI8pkcWsO15kSH3xiOU9WxHCVfiWADEi-W1ldiXsCRY,40383
 pltr/utils/progress.py,sha256=BKYbiLO61uhQbibabU7pxvvbAWMRLRmqk4pZldBQK_g,9053
-pltr_cli-0.3.0.dist-info/METADATA,sha256=BVT_zd9ZrK811JeMXPR_IX1S3-ahE3lhk6q2QiNdbFA,8864
-pltr_cli-0.3.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-pltr_cli-0.3.0.dist-info/entry_points.txt,sha256=8tvEcW04kA_oAE2Dwwu-Og9efjl4ESJvs4AzlP2KBdQ,38
-pltr_cli-0.3.0.dist-info/licenses/LICENSE,sha256=6VUFd_ytnOBD2O1tmkKrA-smigi9QEhYr_tge4h4z8Y,1070
-pltr_cli-0.3.0.dist-info/RECORD,,
+pltr_cli-0.5.0.dist-info/METADATA,sha256=TNVMhHGMRkuRcWfds_6y4B9h6vDSkBRMr9Gt-IeYvks,13356
+pltr_cli-0.5.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+pltr_cli-0.5.0.dist-info/entry_points.txt,sha256=8tvEcW04kA_oAE2Dwwu-Og9efjl4ESJvs4AzlP2KBdQ,38
+pltr_cli-0.5.0.dist-info/licenses/LICENSE,sha256=6VUFd_ytnOBD2O1tmkKrA-smigi9QEhYr_tge4h4z8Y,1070
+pltr_cli-0.5.0.dist-info/RECORD,,

pltr/services/dataset_full.py

@@ -1,302 +0,0 @@
-"""
-Dataset service wrapper for Foundry SDK.
-"""
-
-from typing import Any, Optional, List, Dict, Union
-from pathlib import Path
-
-from .base import BaseService
-
-
-class DatasetService(BaseService):
-    """Service wrapper for Foundry dataset operations."""
-
-    def _get_service(self) -> Any:
-        """Get the Foundry datasets service."""
-        return self.client.datasets
-
-    # list_datasets method removed - not supported by foundry-platform-sdk v1.27.0
-
-    def get_dataset(self, dataset_rid: str) -> Dict[str, Any]:
-        """
-        Get information about a specific dataset.
-
-        Args:
-            dataset_rid: Dataset Resource Identifier
-
-        Returns:
-            Dataset information dictionary
-        """
-        try:
-            # Use the v2 API's Dataset.get method
-            dataset = self.service.Dataset.get(dataset_rid)
-            return self._format_dataset_info(dataset)
-        except Exception as e:
-            raise RuntimeError(f"Failed to get dataset {dataset_rid}: {e}")
-
-    def get_schema(self, dataset_rid: str) -> Dict[str, Any]:
-        """
-        Get dataset schema.
-
-        Args:
-            dataset_rid: Dataset Resource Identifier
-
-        Returns:
-            Schema information
-        """
-        try:
-            schema = self.service.Dataset.get_schema(dataset_rid)
-            return {
-                "dataset_rid": dataset_rid,
-                "schema": schema,
-                "type": str(type(schema)),
-                "status": "Schema retrieved successfully",
-            }
-        except Exception as e:
-            raise RuntimeError(f"Failed to get schema for dataset {dataset_rid}: {e}")
-
-    def create_dataset(
-        self, name: str, parent_folder_rid: Optional[str] = None
-    ) -> Dict[str, Any]:
-        """
-        Create a new dataset.
-
-        Args:
-            name: Dataset name
-            parent_folder_rid: Parent folder RID (optional)
-
-        Returns:
-            Created dataset information
-        """
-        try:
-            dataset = self.service.create_dataset(
-                name=name, parent_folder_rid=parent_folder_rid
-            )
-            return self._format_dataset_info(dataset)
-        except Exception as e:
-            raise RuntimeError(f"Failed to create dataset '{name}': {e}")
-
-    def delete_dataset(self, dataset_rid: str) -> bool:
-        """
-        Delete a dataset.
-
-        Args:
-            dataset_rid: Dataset Resource Identifier
-
-        Returns:
-            True if deletion was successful
-        """
-        try:
-            self.service.delete_dataset(dataset_rid)
-            return True
-        except Exception as e:
-            raise RuntimeError(f"Failed to delete dataset {dataset_rid}: {e}")
-
-    def upload_file(
-        self,
-        dataset_rid: str,
-        file_path: Union[str, Path],
-        branch: str = "master",
-        transaction_rid: Optional[str] = None,
-    ) -> Dict[str, Any]:
-        """
-        Upload a file to a dataset.
-
-        Args:
-            dataset_rid: Dataset Resource Identifier
-            file_path: Path to file to upload
-            branch: Dataset branch name
-            transaction_rid: Transaction RID (optional)
-
-        Returns:
-            Upload result information
-        """
-        file_path = Path(file_path)
-        if not file_path.exists():
-            raise FileNotFoundError(f"File not found: {file_path}")
-
-        try:
-            with open(file_path, "rb") as f:
-                result = self.service.upload_file(
-                    dataset_rid=dataset_rid,
-                    file_path=file_path.name,
-                    file_data=f,
-                    branch=branch,
-                    transaction_rid=transaction_rid,
-                )
-
-            return {
-                "dataset_rid": dataset_rid,
-                "file_path": str(file_path),
-                "branch": branch,
-                "size_bytes": file_path.stat().st_size,
-                "uploaded": True,
-                "transaction_rid": getattr(result, "transaction_rid", transaction_rid),
-            }
-        except Exception as e:
-            raise RuntimeError(
-                f"Failed to upload file {file_path} to dataset {dataset_rid}: {e}"
-            )
-
-    def download_file(
-        self,
-        dataset_rid: str,
-        file_path: str,
-        output_path: Union[str, Path],
-        branch: str = "master",
-    ) -> Dict[str, Any]:
-        """
-        Download a file from a dataset.
-
-        Args:
-            dataset_rid: Dataset Resource Identifier
-            file_path: Path of file within dataset
-            output_path: Local path to save the downloaded file
-            branch: Dataset branch name
-
-        Returns:
-            Download result information
-        """
-        output_path = Path(output_path)
-
-        try:
-            # Ensure output directory exists
-            output_path.parent.mkdir(parents=True, exist_ok=True)
-
-            file_content = self.service.download_file(
-                dataset_rid=dataset_rid, file_path=file_path, branch=branch
-            )
-
-            # Write file content to disk
-            with open(output_path, "wb") as f:
-                if hasattr(file_content, "read"):
-                    # If it's a stream
-                    f.write(file_content.read())
-                else:
-                    # If it's bytes
-                    f.write(file_content)
-
-            return {
-                "dataset_rid": dataset_rid,
-                "file_path": file_path,
-                "output_path": str(output_path),
-                "branch": branch,
-                "size_bytes": output_path.stat().st_size,
-                "downloaded": True,
-            }
-        except Exception as e:
-            raise RuntimeError(
-                f"Failed to download file {file_path} from dataset {dataset_rid}: {e}"
-            )
-
-    def list_files(
-        self, dataset_rid: str, branch: str = "master"
-    ) -> List[Dict[str, Any]]:
-        """
-        List files in a dataset.
-
-        Args:
-            dataset_rid: Dataset Resource Identifier
-            branch: Dataset branch name
-
-        Returns:
-            List of file information dictionaries
-        """
-        try:
-            files = self.service.list_files(dataset_rid=dataset_rid, branch=branch)
-
-            return [
-                {
-                    "path": file.path,
-                    "size_bytes": getattr(file, "size_bytes", None),
-                    "last_modified": getattr(file, "last_modified", None),
-                    "transaction_rid": getattr(file, "transaction_rid", None),
-                }
-                for file in files
-            ]
-        except Exception as e:
-            raise RuntimeError(f"Failed to list files in dataset {dataset_rid}: {e}")
-
-    def get_branches(self, dataset_rid: str) -> List[Dict[str, Any]]:
-        """
-        Get list of branches for a dataset.
-
-        Args:
-            dataset_rid: Dataset Resource Identifier
-
-        Returns:
-            List of branch information dictionaries
-        """
-        try:
-            branches = self.service.list_branches(dataset_rid=dataset_rid)
-
-            return [
-                {
-                    "name": branch.name,
-                    "transaction_rid": getattr(branch, "transaction_rid", None),
-                    "created_time": getattr(branch, "created_time", None),
-                    "created_by": getattr(branch, "created_by", None),
-                }
-                for branch in branches
-            ]
-        except Exception as e:
-            raise RuntimeError(f"Failed to get branches for dataset {dataset_rid}: {e}")
-
-    def create_branch(
-        self, dataset_rid: str, branch_name: str, parent_branch: str = "master"
-    ) -> Dict[str, Any]:
-        """
-        Create a new branch for a dataset.
-
-        Args:
-            dataset_rid: Dataset Resource Identifier
-            branch_name: Name for the new branch
-            parent_branch: Parent branch to branch from
-
-        Returns:
-            Created branch information
-        """
-        try:
-            branch = self.service.create_branch(
-                dataset_rid=dataset_rid,
-                branch_name=branch_name,
-                parent_branch=parent_branch,
-            )
-
-            return {
-                "name": branch.name,
-                "dataset_rid": dataset_rid,
-                "parent_branch": parent_branch,
-                "transaction_rid": getattr(branch, "transaction_rid", None),
-            }
-        except Exception as e:
-            raise RuntimeError(
-                f"Failed to create branch '{branch_name}' for dataset {dataset_rid}: {e}"
-            )
-
-    def _format_dataset_info(self, dataset: Any) -> Dict[str, Any]:
-        """
-        Format dataset information for consistent output.
-
-        Args:
-            dataset: Dataset object from Foundry SDK
-
-        Returns:
-            Formatted dataset information dictionary
-        """
-        # The v2 Dataset object has different attributes
-        return {
-            "rid": getattr(dataset, "rid", "unknown"),
-            "name": getattr(dataset, "name", "Unknown"),
-            "description": getattr(dataset, "description", ""),
-            "path": getattr(dataset, "path", None),
-            "created": getattr(dataset, "created", None),
-            "modified": getattr(dataset, "modified", None),
-            # Try to get additional attributes that might exist
-            "created_time": getattr(dataset, "created_time", None),
-            "created_by": getattr(dataset, "created_by", None),
-            "last_modified": getattr(dataset, "last_modified", None),
-            "size_bytes": getattr(dataset, "size_bytes", None),
-            "schema_id": getattr(dataset, "schema_id", None),
-            "parent_folder_rid": getattr(dataset, "parent_folder_rid", None),
-        }

pltr/services/dataset_v2.py

@@ -1,128 +0,0 @@
-"""
-Dataset service wrapper for Foundry SDK v2 API.
-This is a simplified version that works with the actual foundry_sdk API.
-"""
-
-from typing import Any, Optional, Dict
-
-from .base import BaseService
-
-
-class DatasetServiceV2(BaseService):
-    """Service wrapper for Foundry dataset operations using v2 API."""
-
-    def _get_service(self) -> Any:
-        """Get the Foundry datasets service."""
-        return self.client.datasets
-
-    def get_dataset(self, dataset_rid: str) -> Dict[str, Any]:
-        """
-        Get information about a specific dataset.
-
-        Args:
-            dataset_rid: Dataset Resource Identifier
-
-        Returns:
-            Dataset information dictionary
-        """
-        try:
-            # The v2 API returns a Dataset object
-            dataset = self.service.Dataset.get(dataset_rid)
-
-            # Format the response
-            return {
-                "rid": dataset_rid,
-                "name": getattr(dataset, "name", "Unknown"),
-                "description": getattr(dataset, "description", ""),
-                "path": getattr(dataset, "path", ""),
-                "created": getattr(dataset, "created_time", None),
-                "modified": getattr(dataset, "modified_time", None),
-                # The actual attributes available depend on the SDK version
-                "status": "Retrieved successfully",
-            }
-        except Exception as e:
-            raise RuntimeError(f"Failed to get dataset {dataset_rid}: {e}")
-
-    def create_dataset(
-        self, name: str, parent_folder_rid: Optional[str] = None
-    ) -> Dict[str, Any]:
-        """
-        Create a new dataset.
-
-        Args:
-            name: Dataset name
-            parent_folder_rid: Parent folder RID (optional)
-
-        Returns:
-            Created dataset information
-        """
-        try:
-            # The create method parameters depend on the SDK version
-            dataset = self.service.Dataset.create(
-                name=name, parent_folder_rid=parent_folder_rid
-            )
-
-            return {
-                "rid": getattr(dataset, "rid", "unknown"),
-                "name": name,
-                "parent_folder_rid": parent_folder_rid,
-                "status": "Created successfully",
-            }
-        except Exception as e:
-            raise RuntimeError(f"Failed to create dataset '{name}': {e}")
-
-    def read_table(self, dataset_rid: str, format: str = "arrow") -> Any:
-        """
-        Read dataset as a table.
-
-        Args:
-            dataset_rid: Dataset Resource Identifier
-            format: Output format (arrow, pandas, etc.)
-
-        Returns:
-            Table data in specified format
-        """
-        try:
-            return self.service.Dataset.read_table(dataset_rid, format=format)
-        except Exception as e:
-            raise RuntimeError(f"Failed to read dataset {dataset_rid}: {e}")
-
-    def get_schema(self, dataset_rid: str) -> Dict[str, Any]:
-        """
-        Get dataset schema.
-
-        Args:
-            dataset_rid: Dataset Resource Identifier
-
-        Returns:
-            Schema information
-        """
-        try:
-            schema = self.service.Dataset.get_schema(dataset_rid)
-
-            # Format schema for display
-            return {
-                "dataset_rid": dataset_rid,
-                "schema": schema,
-                "status": "Schema retrieved successfully",
-            }
-        except Exception as e:
-            raise RuntimeError(f"Failed to get schema for dataset {dataset_rid}: {e}")
-
-    def list_datasets(self, limit: Optional[int] = None) -> list:
-        """
-        List datasets - NOT SUPPORTED BY SDK v2.
-
-        The foundry_sdk v2 API doesn't provide a list_datasets method.
-        Dataset operations are RID-based. You need to know the dataset RID
-        to interact with it.
-
-        Raises:
-            NotImplementedError: This operation is not supported by the SDK
-        """
-        raise NotImplementedError(
-            "Dataset listing is not supported by foundry_sdk v2. "
-            "The SDK requires knowing dataset RIDs in advance. "
-            "Consider using the Foundry web interface to find dataset RIDs, "
-            "or contact your Foundry administrator for dataset information."
-        )