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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pltr-cli
-Version: 0.14.0
+Version: 0.14.1
 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

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

@@ -1,6 +1,6 @@
 [project]
 name = "pltr-cli"
-version = "0.14.0"
+version = "0.14.1"
 description = "Command-line interface for Palantir Foundry APIs"
 authors = [
     { name = "anjor", email = "anjor@umd.edu" },

pltr_cli-0.14.1/src/pltr/__init__.py

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

{pltr_cli-0.14.0 → pltr_cli-0.14.1}/src/pltr/services/language_models.py

@@ -76,29 +76,29 @@ class LanguageModelsService(BaseService):
                 }
             ]
 
-            # Build request parameters
-            request_params: Dict[str, Any] = {
+            # Build SDK kwargs
+            request_kwargs: Dict[str, Any] = {
                 "messages": messages,
-                "maxTokens": max_tokens,
+                "max_tokens": max_tokens,
+                "preview": preview,
             }
 
             # Add optional parameters if provided
             if system is not None:
-                request_params["system"] = [{"type": "text", "text": system}]
+                request_kwargs["system"] = [{"type": "text", "text": system}]
             if temperature is not None:
-                request_params["temperature"] = temperature
+                request_kwargs["temperature"] = temperature
             if stop_sequences is not None:
-                request_params["stopSequences"] = stop_sequences
+                request_kwargs["stop_sequences"] = stop_sequences
             if top_k is not None:
-                request_params["topK"] = top_k
+                request_kwargs["top_k"] = top_k
             if top_p is not None:
-                request_params["topP"] = top_p
+                request_kwargs["top_p"] = top_p
 
             # Call SDK method
             response = self.service.AnthropicModel.messages(
                 model_id,
-                request=request_params,
-                preview=preview,  # type: ignore
+                **request_kwargs,  # type: ignore[arg-type,call-overload]
             )
 
             return self._serialize_response(response)
@@ -173,35 +173,35 @@ class LanguageModelsService(BaseService):
             ... )
         """
         try:
-            # Build request parameters
-            request_params: Dict[str, Any] = {
+            # Build SDK kwargs
+            request_kwargs: Dict[str, Any] = {
                 "messages": messages,
-                "maxTokens": max_tokens,
+                "max_tokens": max_tokens,
+                "preview": preview,
             }
 
             # Add optional parameters if provided
             if system is not None:
-                request_params["system"] = system
+                request_kwargs["system"] = system
             if temperature is not None:
-                request_params["temperature"] = temperature
+                request_kwargs["temperature"] = temperature
             if thinking is not None:
-                request_params["thinking"] = thinking
+                request_kwargs["thinking"] = thinking
             if tools is not None:
-                request_params["tools"] = tools
+                request_kwargs["tools"] = tools
             if tool_choice is not None:
-                request_params["toolChoice"] = tool_choice
+                request_kwargs["tool_choice"] = tool_choice
             if stop_sequences is not None:
-                request_params["stopSequences"] = stop_sequences
+                request_kwargs["stop_sequences"] = stop_sequences
             if top_k is not None:
-                request_params["topK"] = top_k
+                request_kwargs["top_k"] = top_k
             if top_p is not None:
-                request_params["topP"] = top_p
+                request_kwargs["top_p"] = top_p
 
             # Call SDK method
             response = self.service.AnthropicModel.messages(
                 model_id,
-                request=request_params,
-                preview=preview,  # type: ignore
+                **request_kwargs,  # type: ignore[arg-type,call-overload]
             )
 
             return self._serialize_response(response)
@@ -256,22 +256,23 @@ class LanguageModelsService(BaseService):
             >>> embeddings = [item['embedding'] for item in response['data']]
         """
         try:
-            # Build request parameters
-            request_params: Dict[str, Any] = {
+            # Build SDK kwargs
+            request_kwargs: Dict[str, Any] = {
                 "input": input_texts,
+                "preview": preview,
             }
 
             # Add optional parameters if provided
             if dimensions is not None:
-                request_params["dimensions"] = dimensions
+                request_kwargs["dimensions"] = dimensions
             if encoding_format is not None:
-                request_params["encodingFormat"] = encoding_format
+                # CLI accepts "float"/"base64", while SDK expects uppercase literals.
+                request_kwargs["encoding_format"] = encoding_format.upper()
 
             # Call SDK method
             response = self.service.OpenAiModel.embeddings(
                 model_id,
-                request=request_params,
-                preview=preview,  # type: ignore
+                **request_kwargs,  # type: ignore[arg-type,call-overload]
             )
 
             return self._serialize_response(response)

{pltr_cli-0.14.0 → pltr_cli-0.14.1}/tests/test_services/test_language_models.py

@@ -55,12 +55,12 @@ class TestLanguageModelsService:
         assert call_args[0][0] == model_id
         assert call_args[1]["preview"] is False
 
-        # Check request structure
-        request = call_args[1]["request"]
-        assert request["maxTokens"] == 1024
-        assert len(request["messages"]) == 1
-        assert request["messages"][0]["role"] == "user"
-        assert request["messages"][0]["content"][0]["text"] == message
+        # Check SDK kwargs structure
+        assert "request" not in call_args[1]
+        assert call_args[1]["max_tokens"] == 1024
+        assert len(call_args[1]["messages"]) == 1
+        assert call_args[1]["messages"][0]["role"] == "user"
+        assert call_args[1]["messages"][0]["content"][0]["text"] == message
 
         # Check result
         assert result["role"] == "assistant"
@@ -81,10 +81,9 @@ class TestLanguageModelsService:
 
         # Assert
         call_args = mock_client.language_models.AnthropicModel.messages.call_args
-        request = call_args[1]["request"]
-        assert "system" in request
-        assert request["system"][0]["type"] == "text"
-        assert request["system"][0]["text"] == system
+        assert "system" in call_args[1]
+        assert call_args[1]["system"][0]["type"] == "text"
+        assert call_args[1]["system"][0]["text"] == system
 
     def test_send_message_with_all_parameters(self, service, mock_client):
         """Test sending message with all optional parameters."""
@@ -110,12 +109,12 @@ class TestLanguageModelsService:
 
         # Assert
         call_args = mock_client.language_models.AnthropicModel.messages.call_args
-        request = call_args[1]["request"]
-        assert request["maxTokens"] == 500
-        assert request["temperature"] == 0.7
-        assert request["stopSequences"] == ["STOP", "END"]
-        assert request["topK"] == 50
-        assert request["topP"] == 0.9
+        assert "request" not in call_args[1]
+        assert call_args[1]["max_tokens"] == 500
+        assert call_args[1]["temperature"] == 0.7
+        assert call_args[1]["stop_sequences"] == ["STOP", "END"]
+        assert call_args[1]["top_k"] == 50
+        assert call_args[1]["top_p"] == 0.9
         assert call_args[1]["preview"] is True
 
     def test_send_message_error(self, service, mock_client):
@@ -155,9 +154,10 @@ class TestLanguageModelsService:
 
         # Assert
         call_args = mock_client.language_models.AnthropicModel.messages.call_args
-        request = call_args[1]["request"]
-        assert request["messages"] == messages
-        assert request["maxTokens"] == max_tokens
+        assert "request" not in call_args[1]
+        assert call_args[1]["messages"] == messages
+        assert call_args[1]["max_tokens"] == max_tokens
+        assert call_args[1]["preview"] is False
         assert result["role"] == "assistant"
 
     def test_send_messages_advanced_with_thinking(self, service, mock_client):
@@ -177,8 +177,7 @@ class TestLanguageModelsService:
 
         # Assert
         call_args = mock_client.language_models.AnthropicModel.messages.call_args
-        request = call_args[1]["request"]
-        assert request["thinking"] == thinking
+        assert call_args[1]["thinking"] == thinking
 
     def test_send_messages_advanced_with_tools(self, service, mock_client):
         """Test sending messages with tool calling."""
@@ -210,9 +209,8 @@ class TestLanguageModelsService:
 
         # Assert
         call_args = mock_client.language_models.AnthropicModel.messages.call_args
-        request = call_args[1]["request"]
-        assert request["tools"] == tools
-        assert request["toolChoice"] == tool_choice
+        assert call_args[1]["tools"] == tools
+        assert call_args[1]["tool_choice"] == tool_choice
 
     def test_send_messages_advanced_error(self, service, mock_client):
         """Test error handling in send_messages_advanced."""
@@ -259,7 +257,8 @@ class TestLanguageModelsService:
         mock_client.language_models.OpenAiModel.embeddings.assert_called_once()
         call_args = mock_client.language_models.OpenAiModel.embeddings.call_args
         assert call_args[0][0] == model_id
-        assert call_args[1]["request"]["input"] == input_texts
+        assert "request" not in call_args[1]
+        assert call_args[1]["input"] == input_texts
         assert call_args[1]["preview"] is False
         assert len(result["data"]) == 1
         assert result["usage"]["totalTokens"] == 2
@@ -286,7 +285,8 @@ class TestLanguageModelsService:
 
         # Assert
         call_args = mock_client.language_models.OpenAiModel.embeddings.call_args
-        assert call_args[1]["request"]["input"] == input_texts
+        assert "request" not in call_args[1]
+        assert call_args[1]["input"] == input_texts
         assert len(result["data"]) == 3
 
     def test_generate_embeddings_with_dimensions(self, service, mock_client):
@@ -308,8 +308,7 @@ class TestLanguageModelsService:
 
         # Assert
         call_args = mock_client.language_models.OpenAiModel.embeddings.call_args
-        request = call_args[1]["request"]
-        assert request["dimensions"] == dimensions
+        assert call_args[1]["dimensions"] == dimensions
 
     def test_generate_embeddings_with_encoding_format(self, service, mock_client):
         """Test generating embeddings with base64 encoding."""
@@ -332,8 +331,7 @@ class TestLanguageModelsService:
 
         # Assert
         call_args = mock_client.language_models.OpenAiModel.embeddings.call_args
-        request = call_args[1]["request"]
-        assert request["encodingFormat"] == encoding_format
+        assert call_args[1]["encoding_format"] == "BASE64"
 
     def test_generate_embeddings_with_all_parameters(self, service, mock_client):
         """Test generating embeddings with all optional parameters."""
@@ -359,10 +357,9 @@ class TestLanguageModelsService:
 
         # Assert
         call_args = mock_client.language_models.OpenAiModel.embeddings.call_args
-        request = call_args[1]["request"]
-        assert request["input"] == input_texts
-        assert request["dimensions"] == 512
-        assert request["encodingFormat"] == "float"
+        assert call_args[1]["input"] == input_texts
+        assert call_args[1]["dimensions"] == 512
+        assert call_args[1]["encoding_format"] == "FLOAT"
         assert call_args[1]["preview"] is True
 
     def test_generate_embeddings_error(self, service, mock_client):

{pltr_cli-0.14.0 → pltr_cli-0.14.1}/uv.lock

@@ -990,7 +990,7 @@ wheels = [
 
 [[package]]
 name = "pltr-cli"
-version = "0.14.0"
+version = "0.14.1"
 source = { editable = "." }
 dependencies = [
     { name = "click-repl" },

pltr_cli-0.14.0/src/pltr/__init__.py

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