feat(api): api update

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
-configured_endpoints: 18
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/lemma%2Flemma-34406ba4cbe013a332aa89d2e6de85dd0f10426405476c8f37da3c3a4b57e0ce.yml
-openapi_spec_hash: 509d1233071637148a2b3298d7fc3de2
+configured_endpoints: 17
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/lemma%2Flemma-7d37defa0f907862d7d3c807dd33259df967ddaa70601f91d4d771c9c02e4dec.yml
+openapi_spec_hash: 357f9bb257a1f73b57bd4bd5f3018000
 config_hash: 3911423d0a218a9e0813f5fe6370ae95

README.md

@@ -32,7 +32,10 @@ from lemma import Lemma
 
 client = Lemma()
 
-datasets = client.datasets.list()
+dataset = client.datasets.create(
+    name="name",
+)
+print(dataset.id)
 ```
 
 While you can provide an `api_key` keyword argument,
@@ -52,7 +55,10 @@ client = AsyncLemma()
 
 
 async def main() -> None:
-    datasets = await client.datasets.list()
+    dataset = await client.datasets.create(
+        name="name",
+    )
+    print(dataset.id)
 
 
 asyncio.run(main())
@@ -83,7 +89,10 @@ async def main() -> None:
     async with AsyncLemma(
         http_client=DefaultAioHttpClient(),
     ) as client:
-        datasets = await client.datasets.list()
+        dataset = await client.datasets.create(
+            name="name",
+        )
+        print(dataset.id)
 
 
 asyncio.run(main())
@@ -131,7 +140,9 @@ from lemma import Lemma
 client = Lemma()
 
 try:
-    client.datasets.list()
+    client.datasets.create(
+        name="name",
+    )
 except lemma.APIConnectionError as e:
     print("The server could not be reached")
     print(e.__cause__)  # an underlying Exception, likely raised within httpx.
@@ -174,7 +185,9 @@ client = Lemma(
 )
 
 # Or, configure per-request:
-client.with_options(max_retries=5).datasets.list()
+client.with_options(max_retries=5).datasets.create(
+    name="name",
+)
 ```
 
 ### Timeouts
@@ -197,7 +210,9 @@ client = Lemma(
 )
 
 # Override per-request:
-client.with_options(timeout=5.0).datasets.list()
+client.with_options(timeout=5.0).datasets.create(
+    name="name",
+)
 ```
 
 On timeout, an `APITimeoutError` is thrown.
@@ -238,11 +253,13 @@ The "raw" Response object can be accessed by prefixing `.with_raw_response.` to
 from lemma import Lemma
 
 client = Lemma()
-response = client.datasets.with_raw_response.list()
+response = client.datasets.with_raw_response.create(
+    name="name",
+)
 print(response.headers.get('X-My-Header'))
 
-dataset = response.parse()  # get the object that `datasets.list()` would have returned
-print(dataset)
+dataset = response.parse()  # get the object that `datasets.create()` would have returned
+print(dataset.id)
 ```
 
 These methods return an [`APIResponse`](https://github.com/uselemma/sdk-python/tree/main/src/lemma/_response.py) object.
@@ -256,7 +273,9 @@ The above interface eagerly reads the full response body when you make the reque
 To stream the response body, use `.with_streaming_response` instead, which requires a context manager and only reads the response body once you call `.read()`, `.text()`, `.json()`, `.iter_bytes()`, `.iter_text()`, `.iter_lines()` or `.parse()`. In the async client, these are async methods.
 
 ```python
-with client.datasets.with_streaming_response.list() as response:
+with client.datasets.with_streaming_response.create(
+    name="name",
+) as response:
     print(response.headers.get("X-My-Header"))
 
     for line in response.iter_lines():

api.md

@@ -5,7 +5,6 @@ Types:
 ```python
 from lemma.types import (
     Dataset,
-    DatasetListResponse,
     DatasetDeleteResponse,
     DatasetCreateDownloadURLResponse,
     DatasetGenerateDatasetResponse,
@@ -18,7 +17,6 @@ Methods:
 
 - <code title="post /datasets">client.datasets.<a href="./src/lemma/resources/datasets.py">create</a>(\*\*<a href="src/lemma/types/dataset_create_params.py">params</a>) -> <a href="./src/lemma/types/dataset.py">Dataset</a></code>
 - <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>

src/lemma/resources/datasets.py

@@ -24,7 +24,6 @@
 )
 from .._base_client import make_request_options
 from ..types.dataset import Dataset
-from ..types.dataset_list_response import DatasetListResponse
 from ..types.dataset_delete_response import DatasetDeleteResponse
 from ..types.dataset_generate_schema_response import DatasetGenerateSchemaResponse
 from ..types.dataset_generate_dataset_response import DatasetGenerateDatasetResponse
@@ -136,30 +135,6 @@ def retrieve(
             cast_to=Dataset,
         )
 
-    def list(
-        self,
-        *,
-        # 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,
-    ) -> DatasetListResponse:
-        """
-        List all datasets for the current tenant.
-
-        Retrieves all datasets that belong to the current user's tenant. This includes
-        datasets created by any user within the same tenant.
-        """
-        return self._get(
-            "/datasets",
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=DatasetListResponse,
-        )
-
     def delete(
         self,
         dataset_id: str,
@@ -491,30 +466,6 @@ async def retrieve(
             cast_to=Dataset,
         )
 
-    async def list(
-        self,
-        *,
-        # 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,
-    ) -> DatasetListResponse:
-        """
-        List all datasets for the current tenant.
-
-        Retrieves all datasets that belong to the current user's tenant. This includes
-        datasets created by any user within the same tenant.
-        """
-        return await self._get(
-            "/datasets",
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=DatasetListResponse,
-        )
-
     async def delete(
         self,
         dataset_id: str,
@@ -753,9 +704,6 @@ def __init__(self, datasets: DatasetsResource) -> None:
         self.retrieve = to_raw_response_wrapper(
             datasets.retrieve,
         )
-        self.list = to_raw_response_wrapper(
-            datasets.list,
-        )
         self.delete = to_raw_response_wrapper(
             datasets.delete,
         )
@@ -783,9 +731,6 @@ def __init__(self, datasets: AsyncDatasetsResource) -> None:
         self.retrieve = async_to_raw_response_wrapper(
             datasets.retrieve,
         )
-        self.list = async_to_raw_response_wrapper(
-            datasets.list,
-        )
         self.delete = async_to_raw_response_wrapper(
             datasets.delete,
         )
@@ -813,9 +758,6 @@ def __init__(self, datasets: DatasetsResource) -> None:
         self.retrieve = to_streamed_response_wrapper(
             datasets.retrieve,
         )
-        self.list = to_streamed_response_wrapper(
-            datasets.list,
-        )
         self.delete = to_streamed_response_wrapper(
             datasets.delete,
         )
@@ -843,9 +785,6 @@ def __init__(self, datasets: AsyncDatasetsResource) -> None:
         self.retrieve = async_to_streamed_response_wrapper(
             datasets.retrieve,
         )
-        self.list = async_to_streamed_response_wrapper(
-            datasets.list,
-        )
         self.delete = async_to_streamed_response_wrapper(
             datasets.delete,
         )

src/lemma/types/__init__.py

@@ -10,7 +10,6 @@
 from .log_list_response import LogListResponse as LogListResponse
 from .prompt_create_params import PromptCreateParams as PromptCreateParams
 from .dataset_create_params import DatasetCreateParams as DatasetCreateParams
-from .dataset_list_response import DatasetListResponse as DatasetListResponse
 from .playground_run_params import PlaygroundRunParams as PlaygroundRunParams
 from .prompt_create_response import PromptCreateResponse as PromptCreateResponse
 from .prompt_delete_response import PromptDeleteResponse as PromptDeleteResponse

src/lemma/types/dataset_list_response.py

@@ -10,7 +10,6 @@
 from lemma import Lemma, AsyncLemma
 from lemma.types import (
     Dataset,
-    DatasetListResponse,
     DatasetDeleteResponse,
     DatasetGenerateSchemaResponse,
     DatasetGenerateDatasetResponse,
@@ -109,34 +108,6 @@ def test_path_params_retrieve(self, client: Lemma) -> None:
                 "",
             )
 
-    @pytest.mark.skip(reason="Prism tests are disabled")
-    @parametrize
-    def test_method_list(self, client: Lemma) -> None:
-        dataset = client.datasets.list()
-        assert_matches_type(DatasetListResponse, dataset, path=["response"])
-
-    @pytest.mark.skip(reason="Prism tests are disabled")
-    @parametrize
-    def test_raw_response_list(self, client: Lemma) -> None:
-        response = client.datasets.with_raw_response.list()
-
-        assert response.is_closed is True
-        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
-        dataset = response.parse()
-        assert_matches_type(DatasetListResponse, dataset, path=["response"])
-
-    @pytest.mark.skip(reason="Prism tests are disabled")
-    @parametrize
-    def test_streaming_response_list(self, client: Lemma) -> None:
-        with client.datasets.with_streaming_response.list() as response:
-            assert not response.is_closed
-            assert response.http_request.headers.get("X-Stainless-Lang") == "python"
-
-            dataset = response.parse()
-            assert_matches_type(DatasetListResponse, dataset, path=["response"])
-
-        assert cast(Any, response.is_closed) is True
-
     @pytest.mark.skip(reason="Prism tests are disabled")
     @parametrize
     def test_method_delete(self, client: Lemma) -> None:
@@ -444,34 +415,6 @@ async def test_path_params_retrieve(self, async_client: AsyncLemma) -> None:
                 "",
             )
 
-    @pytest.mark.skip(reason="Prism tests are disabled")
-    @parametrize
-    async def test_method_list(self, async_client: AsyncLemma) -> None:
-        dataset = await async_client.datasets.list()
-        assert_matches_type(DatasetListResponse, dataset, path=["response"])
-
-    @pytest.mark.skip(reason="Prism tests are disabled")
-    @parametrize
-    async def test_raw_response_list(self, async_client: AsyncLemma) -> None:
-        response = await async_client.datasets.with_raw_response.list()
-
-        assert response.is_closed is True
-        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
-        dataset = await response.parse()
-        assert_matches_type(DatasetListResponse, dataset, path=["response"])
-
-    @pytest.mark.skip(reason="Prism tests are disabled")
-    @parametrize
-    async def test_streaming_response_list(self, async_client: AsyncLemma) -> None:
-        async with async_client.datasets.with_streaming_response.list() as response:
-            assert not response.is_closed
-            assert response.http_request.headers.get("X-Stainless-Lang") == "python"
-
-            dataset = await response.parse()
-            assert_matches_type(DatasetListResponse, dataset, path=["response"])
-
-        assert cast(Any, response.is_closed) is True
-
     @pytest.mark.skip(reason="Prism tests are disabled")
     @parametrize
     async def test_method_delete(self, async_client: AsyncLemma) -> None: