pltr-cli 0.14.0__tar.gz → 0.15.0__tar.gz

{pltr_cli-0.14.0 → pltr_cli-0.15.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pltr-cli
-Version: 0.14.0
+Version: 0.15.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
@@ -30,6 +30,7 @@ Requires-Dist: click-repl>=0.3.0
 Requires-Dist: foundry-platform-sdk<2.0.0,>=1.69.0
 Requires-Dist: keyring>=25.6.0
 Requires-Dist: pandas>=2.0.0
+Requires-Dist: pyarrow>=15.0.0
 Requires-Dist: python-dotenv>=1.1.1
 Requires-Dist: requests>=2.32.4
 Requires-Dist: rich>=14.1.0

{pltr_cli-0.14.0 → pltr_cli-0.15.0}/mypy.ini

@@ -10,3 +10,7 @@ disallow_untyped_decorators = False
 # Ignore missing imports for foundry SDK
 [mypy-foundry.*]
 ignore_missing_imports = True
+
+# pyarrow ships no py.typed marker
+[mypy-pyarrow.*]
+ignore_missing_imports = True

{pltr_cli-0.14.0 → pltr_cli-0.15.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "pltr-cli"
-version = "0.14.0"
+version = "0.15.0"
 description = "Command-line interface for Palantir Foundry APIs"
 authors = [
     { name = "anjor", email = "anjor@umd.edu" },
@@ -12,6 +12,7 @@ dependencies = [
     "foundry-platform-sdk>=1.69.0,<2.0.0",
     "keyring>=25.6.0",
     "pandas>=2.0.0",
+    "pyarrow>=15.0.0",
     "python-dotenv>=1.1.1",
     "requests>=2.32.4",
     "rich>=14.1.0",

pltr_cli-0.15.0/src/pltr/__init__.py

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

{pltr_cli-0.14.0 → pltr_cli-0.15.0}/src/pltr/commands/language_models.py

@@ -123,6 +123,148 @@ def display_openai_response(response: dict, format: str, output: Optional[str]):
         formatter.display(response, format)
 
 
+# ===== Shared Language Model Commands =====
+
+
+@app.command("list")
+def list_models(
+    profile: Optional[str] = typer.Option(
+        None,
+        "--profile",
+        "-p",
+        help="Profile name",
+        autocompletion=complete_profile,
+    ),
+    format: str = typer.Option(
+        "table",
+        "--format",
+        "-f",
+        help="Output format (table, json, csv)",
+        autocompletion=complete_output_format,
+    ),
+    output: Optional[str] = typer.Option(
+        None, "--output", "-o", help="Output file path"
+    ),
+):
+    """List available language models for the current enrollment."""
+    try:
+        from ..services.language_models import LanguageModelsService
+
+        service = LanguageModelsService(profile=profile)
+
+        with SpinnerProgressTracker().track_spinner("Fetching available models..."):
+            models = service.list_available_models()
+
+        formatter.format_table(
+            models,
+            columns=["model_rid", "status", "type", "display_name"],
+            format=format,
+            output=output,
+        )
+
+        if output:
+            formatter.print_success(f"Model list saved to {output}")
+
+    except (ProfileNotFoundError, MissingCredentialsError) as e:
+        formatter.print_error(f"Authentication error: {e}")
+        raise typer.Exit(1)
+    except Exception as e:
+        formatter.print_error(f"Operation failed: {e}")
+        raise typer.Exit(1)
+
+
+@app.command("status")
+def model_status(
+    model_id: str = typer.Argument(
+        ...,
+        help="Model Resource Identifier (ri.language-model-service..language-model.<id>)",
+    ),
+    profile: Optional[str] = typer.Option(
+        None,
+        "--profile",
+        "-p",
+        help="Profile name",
+        autocompletion=complete_profile,
+    ),
+    format: str = typer.Option(
+        "table",
+        "--format",
+        "-f",
+        help="Output format (table, json, csv)",
+        autocompletion=complete_output_format,
+    ),
+    output: Optional[str] = typer.Option(
+        None, "--output", "-o", help="Output file path"
+    ),
+):
+    """Check enrollment status for a language model via direct API fallback endpoints."""
+    try:
+        from ..services.language_models import LanguageModelsService
+
+        service = LanguageModelsService(profile=profile)
+
+        with SpinnerProgressTracker().track_spinner("Checking model status..."):
+            result = service.get_model_enrollment_status(model_id)
+
+        formatter.format_dict(result, format=format, output=output)
+
+        if output:
+            formatter.print_success(f"Model status saved to {output}")
+
+    except (ProfileNotFoundError, MissingCredentialsError) as e:
+        formatter.print_error(f"Authentication error: {e}")
+        raise typer.Exit(1)
+    except Exception as e:
+        formatter.print_error(f"Operation failed: {e}")
+        raise typer.Exit(1)
+
+
+@app.command("enroll")
+def enroll_model(
+    model_id: str = typer.Argument(
+        ...,
+        help="Model Resource Identifier (ri.language-model-service..language-model.<id>)",
+    ),
+    profile: Optional[str] = typer.Option(
+        None,
+        "--profile",
+        "-p",
+        help="Profile name",
+        autocompletion=complete_profile,
+    ),
+    format: str = typer.Option(
+        "table",
+        "--format",
+        "-f",
+        help="Output format (table, json, csv)",
+        autocompletion=complete_output_format,
+    ),
+    output: Optional[str] = typer.Option(
+        None, "--output", "-o", help="Output file path"
+    ),
+):
+    """Enroll/enable a language model via direct API fallback endpoints."""
+    try:
+        from ..services.language_models import LanguageModelsService
+
+        service = LanguageModelsService(profile=profile)
+
+        with SpinnerProgressTracker().track_spinner("Enrolling model..."):
+            result = service.enroll_model(model_id)
+
+        formatter.format_dict(result, format=format, output=output)
+
+        if output:
+            formatter.print_success(f"Enrollment result saved to {output}")
+
+    except (ProfileNotFoundError, MissingCredentialsError) as e:
+        formatter.print_error(f"Authentication error: {e}")
+        raise typer.Exit(1)
+    except Exception as e:
+        formatter.print_error(f"Operation failed: {e}")
+        raise typer.Exit(1)
+
+
 # ===== Anthropic Commands =====
 
 
@@ -323,9 +465,9 @@ def anthropic_messages_advanced(
         # conversation.json:
         # {
         #   "messages": [
-        #     {"role": "user", "content": [{"type": "text", "text": "Hi"}]},
-        #     {"role": "assistant", "content": [{"type": "text", "text": "Hello!"}]},
-        #     {"role": "user", "content": [{"type": "text", "text": "Help me"}]}
+        #     {"role": "USER", "content": [{"type": "text", "text": "Hi"}]},
+        #     {"role": "ASSISTANT", "content": [{"type": "text", "text": "Hello!"}]},
+        #     {"role": "USER", "content": [{"type": "text", "text": "Help me"}]}
         #   ],
         #   "maxTokens": 500
         # }
@@ -334,11 +476,11 @@ def anthropic_messages_advanced(
 
         # Inline JSON with system prompt
         pltr language-models anthropic messages-advanced ri.language-models.main.model.abc123 \\
-            --request '{"messages": [{"role": "user", "content": [{"type": "text", "text": "Hi"}]}], "maxTokens": 100, "system": [{"type": "text", "text": "Be concise"}]}'
+            --request '{"messages": [{"role": "USER", "content": [{"type": "text", "text": "Hi"}]}], "maxTokens": 100, "system": [{"type": "text", "text": "Be concise"}]}'
 
         # With extended thinking
         pltr language-models anthropic messages-advanced ri.language-models.main.model.abc123 \\
-            --request '{"messages": [{"role": "user", "content": [{"type": "text", "text": "Solve this problem"}]}], "maxTokens": 2000, "thinking": {"type": "enabled", "budget": 10000}}'
+            --request '{"messages": [{"role": "USER", "content": [{"type": "text", "text": "Solve this problem"}]}], "maxTokens": 2000, "thinking": {"type": "enabled", "budget": 10000}}'
     """
     try:
         # Parse request JSON

{pltr_cli-0.14.0 → pltr_cli-0.15.0}/src/pltr/commands/ontology.py

@@ -164,6 +164,112 @@ def get_object_type(
         raise typer.Exit(1)
 
 
+@app.command("object-type-create")
+def create_object_type(
+    ontology_rid: str = typer.Argument(..., help="Ontology Resource Identifier"),
+    api_name: str = typer.Option(..., "--api-name", help="Object type API name"),
+    display_name: str = typer.Option(
+        ..., "--display-name", help="Object type display name"
+    ),
+    primary_key: str = typer.Option(
+        ..., "--primary-key", help="Primary key property API name"
+    ),
+    backing_dataset: str = typer.Option(
+        ..., "--backing-dataset", help="Backing dataset RID"
+    ),
+    description: Optional[str] = typer.Option(
+        None, "--description", help="Object type description"
+    ),
+    profile: Optional[str] = typer.Option(None, "--profile", "-p", help="Profile name"),
+    format: str = typer.Option(
+        "table", "--format", "-f", help="Output format (table, json, csv)"
+    ),
+    output: Optional[str] = typer.Option(
+        None, "--output", "-o", help="Output file path"
+    ),
+):
+    """Create a new object type in an ontology."""
+    try:
+        service = ObjectTypeService(profile=profile)
+
+        with SpinnerProgressTracker().track_spinner(
+            f"Creating object type {api_name}..."
+        ):
+            result = service.create_object_type(
+                ontology_rid=ontology_rid,
+                api_name=api_name,
+                display_name=display_name,
+                primary_key=primary_key,
+                backing_dataset=backing_dataset,
+                description=description,
+            )
+
+        formatter.format_dict(result, format=format, output=output)
+
+        if output:
+            formatter.print_success(f"Object type creation result saved to {output}")
+
+    except (ProfileNotFoundError, MissingCredentialsError) as e:
+        formatter.print_error(f"Authentication error: {e}")
+        raise typer.Exit(1)
+    except Exception as e:
+        formatter.print_error(f"Failed to create object type: {e}")
+        raise typer.Exit(1)
+
+
+@app.command("link-type-create")
+def create_link_type(
+    ontology_rid: str = typer.Argument(..., help="Ontology Resource Identifier"),
+    api_name: str = typer.Option(..., "--api-name", help="Link type API name"),
+    from_object: str = typer.Option(..., "--from", help="Source object type API name"),
+    to_object: str = typer.Option(..., "--to", help="Target object type API name"),
+    display_name: Optional[str] = typer.Option(
+        None, "--display-name", help="Link type display name"
+    ),
+    description: Optional[str] = typer.Option(
+        None, "--description", help="Link type description"
+    ),
+    reverse_api_name: Optional[str] = typer.Option(
+        None, "--reverse-api-name", help="Reverse direction link type API name"
+    ),
+    profile: Optional[str] = typer.Option(None, "--profile", "-p", help="Profile name"),
+    format: str = typer.Option(
+        "table", "--format", "-f", help="Output format (table, json, csv)"
+    ),
+    output: Optional[str] = typer.Option(
+        None, "--output", "-o", help="Output file path"
+    ),
+):
+    """Create a new link type in an ontology."""
+    try:
+        service = ObjectTypeService(profile=profile)
+
+        with SpinnerProgressTracker().track_spinner(
+            f"Creating link type {api_name}..."
+        ):
+            result = service.create_link_type(
+                ontology_rid=ontology_rid,
+                api_name=api_name,
+                from_object_type=from_object,
+                to_object_type=to_object,
+                display_name=display_name,
+                description=description,
+                reverse_api_name=reverse_api_name,
+            )
+
+        formatter.format_dict(result, format=format, output=output)
+
+        if output:
+            formatter.print_success(f"Link type creation result saved to {output}")
+
+    except (ProfileNotFoundError, MissingCredentialsError) as e:
+        formatter.print_error(f"Authentication error: {e}")
+        raise typer.Exit(1)
+    except Exception as e:
+        formatter.print_error(f"Failed to create link type: {e}")
+        raise typer.Exit(1)
+
+
 # Object operations
 @app.command("object-list")
 def list_objects(