feat(api): manual updates

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
-configured_endpoints: 17
+configured_endpoints: 18
 openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/lemma%2Flemma-584f07ea035720c85ecac56802077189e3ef8e2970d3a7f9ffb75a605df61fef.yml
 openapi_spec_hash: 941d7b54bafb4c217116dc8fea329900
-config_hash: 2cf9a61cb7e01c536fb8672c6fce0d58
+config_hash: 9b31b57de762bba601e624c9104ca492

api.md

@@ -7,6 +7,7 @@ from lemma.types import (
     Dataset,
     DatasetListResponse,
     DatasetDeleteResponse,
+    DatasetCreateDownloadURLResponse,
     DatasetGenerateDatasetResponse,
     DatasetGenerateSchemaResponse,
     DatasetGenerateValidatorsResponse,
@@ -19,6 +20,7 @@ Methods:
 - <code title="get /datasets/{dataset_id}">client.datasets.<a href="./src/lemma/resources/datasets.py">retrieve</a>(dataset_id) -> <a href="./src/lemma/types/dataset.py">Dataset</a></code>
 - <code title="get /datasets">client.datasets.<a href="./src/lemma/resources/datasets.py">list</a>() -> <a href="./src/lemma/types/dataset_list_response.py">DatasetListResponse</a></code>
 - <code title="delete /datasets/{dataset_id}">client.datasets.<a href="./src/lemma/resources/datasets.py">delete</a>(dataset_id) -> <a href="./src/lemma/types/dataset_delete_response.py">DatasetDeleteResponse</a></code>
+- <code title="get /datasets/{dataset_id}/create-download-url">client.datasets.<a href="./src/lemma/resources/datasets.py">create_download_url</a>(dataset_id) -> str</code>
 - <code title="post /datasets/generate-dataset">client.datasets.<a href="./src/lemma/resources/datasets.py">generate_dataset</a>(\*\*<a href="src/lemma/types/dataset_generate_dataset_params.py">params</a>) -> <a href="./src/lemma/types/dataset_generate_dataset_response.py">DatasetGenerateDatasetResponse</a></code>
 - <code title="post /datasets/generate-schema">client.datasets.<a href="./src/lemma/resources/datasets.py">generate_schema</a>(\*\*<a href="src/lemma/types/dataset_generate_schema_params.py">params</a>) -> <a href="./src/lemma/types/dataset_generate_schema_response.py">DatasetGenerateSchemaResponse</a></code>
 - <code title="post /datasets/generate-validators">client.datasets.<a href="./src/lemma/resources/datasets.py">generate_validators</a>(\*\*<a href="src/lemma/types/dataset_generate_validators_params.py">params</a>) -> <a href="./src/lemma/types/dataset_generate_validators_response.py">DatasetGenerateValidatorsResponse</a></code>

src/lemma/resources/datasets.py

@@ -196,6 +196,43 @@ def delete(
             cast_to=DatasetDeleteResponse,
         )
 
+    def create_download_url(
+        self,
+        dataset_id: str,
+        *,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> str:
+        """
+        Create a download URL for a dataset.
+
+        Creates a secure, time-limited download URL that allows downloading the dataset
+        data. The URL can be used to access the dataset file without requiring
+        authentication.
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not dataset_id:
+            raise ValueError(f"Expected a non-empty value for `dataset_id` but received {dataset_id!r}")
+        return self._get(
+            f"/datasets/{dataset_id}/create-download-url",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=str,
+        )
+
     def generate_dataset(
         self,
         *,
@@ -514,6 +551,43 @@ async def delete(
             cast_to=DatasetDeleteResponse,
         )
 
+    async def create_download_url(
+        self,
+        dataset_id: str,
+        *,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> str:
+        """
+        Create a download URL for a dataset.
+
+        Creates a secure, time-limited download URL that allows downloading the dataset
+        data. The URL can be used to access the dataset file without requiring
+        authentication.
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not dataset_id:
+            raise ValueError(f"Expected a non-empty value for `dataset_id` but received {dataset_id!r}")
+        return await self._get(
+            f"/datasets/{dataset_id}/create-download-url",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=str,
+        )
+
     async def generate_dataset(
         self,
         *,
@@ -685,6 +759,9 @@ def __init__(self, datasets: DatasetsResource) -> None:
         self.delete = to_raw_response_wrapper(
             datasets.delete,
         )
+        self.create_download_url = to_raw_response_wrapper(
+            datasets.create_download_url,
+        )
         self.generate_dataset = to_raw_response_wrapper(
             datasets.generate_dataset,
         )
@@ -712,6 +789,9 @@ def __init__(self, datasets: AsyncDatasetsResource) -> None:
         self.delete = async_to_raw_response_wrapper(
             datasets.delete,
         )
+        self.create_download_url = async_to_raw_response_wrapper(
+            datasets.create_download_url,
+        )
         self.generate_dataset = async_to_raw_response_wrapper(
             datasets.generate_dataset,
         )
@@ -739,6 +819,9 @@ def __init__(self, datasets: DatasetsResource) -> None:
         self.delete = to_streamed_response_wrapper(
             datasets.delete,
         )
+        self.create_download_url = to_streamed_response_wrapper(
+            datasets.create_download_url,
+        )
         self.generate_dataset = to_streamed_response_wrapper(
             datasets.generate_dataset,
         )
@@ -766,6 +849,9 @@ def __init__(self, datasets: AsyncDatasetsResource) -> None:
         self.delete = async_to_streamed_response_wrapper(
             datasets.delete,
         )
+        self.create_download_url = async_to_streamed_response_wrapper(
+            datasets.create_download_url,
+        )
         self.generate_dataset = async_to_streamed_response_wrapper(
             datasets.generate_dataset,
         )

src/lemma/types/__init__.py

@@ -30,4 +30,5 @@
 from .prompt_create_iteration_response import PromptCreateIterationResponse as PromptCreateIterationResponse
 from .dataset_generate_dataset_response import DatasetGenerateDatasetResponse as DatasetGenerateDatasetResponse
 from .dataset_generate_validators_params import DatasetGenerateValidatorsParams as DatasetGenerateValidatorsParams
+from .dataset_create_download_url_response import DatasetCreateDownloadURLResponse as DatasetCreateDownloadURLResponse
 from .dataset_generate_validators_response import DatasetGenerateValidatorsResponse as DatasetGenerateValidatorsResponse

src/lemma/types/dataset_create_download_url_response.py

@@ -0,0 +1,7 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing_extensions import TypeAlias
+
+__all__ = ["DatasetCreateDownloadURLResponse"]
+
+DatasetCreateDownloadURLResponse: TypeAlias = str

tests/api_resources/test_datasets.py

@@ -179,6 +179,48 @@ def test_path_params_delete(self, client: Lemma) -> None:
                 "",
             )
 
+    @pytest.mark.skip(reason="Prism tests are disabled")
+    @parametrize
+    def test_method_create_download_url(self, client: Lemma) -> None:
+        dataset = client.datasets.create_download_url(
+            "dataset_id",
+        )
+        assert_matches_type(str, dataset, path=["response"])
+
+    @pytest.mark.skip(reason="Prism tests are disabled")
+    @parametrize
+    def test_raw_response_create_download_url(self, client: Lemma) -> None:
+        response = client.datasets.with_raw_response.create_download_url(
+            "dataset_id",
+        )
+
+        assert response.is_closed is True
+        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
+        dataset = response.parse()
+        assert_matches_type(str, dataset, path=["response"])
+
+    @pytest.mark.skip(reason="Prism tests are disabled")
+    @parametrize
+    def test_streaming_response_create_download_url(self, client: Lemma) -> None:
+        with client.datasets.with_streaming_response.create_download_url(
+            "dataset_id",
+        ) as response:
+            assert not response.is_closed
+            assert response.http_request.headers.get("X-Stainless-Lang") == "python"
+
+            dataset = response.parse()
+            assert_matches_type(str, dataset, path=["response"])
+
+        assert cast(Any, response.is_closed) is True
+
+    @pytest.mark.skip(reason="Prism tests are disabled")
+    @parametrize
+    def test_path_params_create_download_url(self, client: Lemma) -> None:
+        with pytest.raises(ValueError, match=r"Expected a non-empty value for `dataset_id` but received ''"):
+            client.datasets.with_raw_response.create_download_url(
+                "",
+            )
+
     @pytest.mark.skip(reason="Prism tests are disabled")
     @parametrize
     def test_method_generate_dataset(self, client: Lemma) -> None:
@@ -472,6 +514,48 @@ async def test_path_params_delete(self, async_client: AsyncLemma) -> None:
                 "",
             )
 
+    @pytest.mark.skip(reason="Prism tests are disabled")
+    @parametrize
+    async def test_method_create_download_url(self, async_client: AsyncLemma) -> None:
+        dataset = await async_client.datasets.create_download_url(
+            "dataset_id",
+        )
+        assert_matches_type(str, dataset, path=["response"])
+
+    @pytest.mark.skip(reason="Prism tests are disabled")
+    @parametrize
+    async def test_raw_response_create_download_url(self, async_client: AsyncLemma) -> None:
+        response = await async_client.datasets.with_raw_response.create_download_url(
+            "dataset_id",
+        )
+
+        assert response.is_closed is True
+        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
+        dataset = await response.parse()
+        assert_matches_type(str, dataset, path=["response"])
+
+    @pytest.mark.skip(reason="Prism tests are disabled")
+    @parametrize
+    async def test_streaming_response_create_download_url(self, async_client: AsyncLemma) -> None:
+        async with async_client.datasets.with_streaming_response.create_download_url(
+            "dataset_id",
+        ) as response:
+            assert not response.is_closed
+            assert response.http_request.headers.get("X-Stainless-Lang") == "python"
+
+            dataset = await response.parse()
+            assert_matches_type(str, dataset, path=["response"])
+
+        assert cast(Any, response.is_closed) is True
+
+    @pytest.mark.skip(reason="Prism tests are disabled")
+    @parametrize
+    async def test_path_params_create_download_url(self, async_client: AsyncLemma) -> None:
+        with pytest.raises(ValueError, match=r"Expected a non-empty value for `dataset_id` but received ''"):
+            await async_client.datasets.with_raw_response.create_download_url(
+                "",
+            )
+
     @pytest.mark.skip(reason="Prism tests are disabled")
     @parametrize
     async def test_method_generate_dataset(self, async_client: AsyncLemma) -> None: