feat(api): api update

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
-configured_endpoints: 49
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/lemma%2Flemma-490b6c951dbe764981862a42c1067201cca1473d6ea2b5950d6b52246db9c455.yml
-openapi_spec_hash: 25f3fb0ed209b3ddd390b3b1df365f5f
+configured_endpoints: 48
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/lemma%2Flemma-dc1b1744bf9a13daa82065ecf5dc75e3870d611ab3fabf2b0f91a7b36815b2a4.yml
+openapi_spec_hash: ae1eb47bf32f7b2c225338a245298e02
 config_hash: 929863a9ffccfd7c8065b0deffba83c3

api.md

@@ -12,7 +12,6 @@ from lemma.types import (
     DatasetGenerateSchemaResponse,
     DatasetGenerateValidatorsResponse,
     DatasetInsertEntryResponse,
-    DatasetUploadJSONLResponse,
 )
 ```
 
@@ -27,8 +26,7 @@ Methods:
 - <code title="post /datasets/generate-dataset">client.datasets.<a href="./src/lemma/resources/datasets/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/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/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>
-- <code title="post /datasets/{dataset_id}/insert-entry">client.datasets.<a href="./src/lemma/resources/datasets/datasets.py">insert_entry</a>(dataset_id, \*\*<a href="src/lemma/types/dataset_insert_entry_params.py">params</a>) -> str</code>
-- <code title="post /datasets/{dataset_id}/upload-jsonl">client.datasets.<a href="./src/lemma/resources/datasets/datasets.py">upload_jsonl</a>(dataset_id, \*\*<a href="src/lemma/types/dataset_upload_jsonl_params.py">params</a>) -> <a href="./src/lemma/types/dataset_upload_jsonl_response.py">DatasetUploadJSONLResponse</a></code>
+- <code title="post /datasets/{dataset_id}/insert-entry">client.datasets.<a href="./src/lemma/resources/datasets/datasets.py">insert_entry</a>(dataset_id, \*\*<a href="src/lemma/types/dataset_insert_entry_params.py">params</a>) -> <a href="./src/lemma/types/dataset_insert_entry_response.py">DatasetInsertEntryResponse</a></code>
 
 ## Generations
 

src/lemma/resources/datasets/datasets.py

@@ -11,7 +11,6 @@
     dataset_create_params,
     dataset_update_params,
     dataset_insert_entry_params,
-    dataset_upload_jsonl_params,
     dataset_generate_schema_params,
     dataset_generate_dataset_params,
     dataset_generate_validators_params,
@@ -38,7 +37,7 @@
 from ...types.dataset import Dataset
 from ...types.dataset_list_response import DatasetListResponse
 from ...types.dataset_delete_response import DatasetDeleteResponse
-from ...types.dataset_upload_jsonl_response import DatasetUploadJSONLResponse
+from ...types.dataset_insert_entry_response import DatasetInsertEntryResponse
 from ...types.dataset_generate_schema_response import DatasetGenerateSchemaResponse
 from ...types.dataset_generate_dataset_response import DatasetGenerateDatasetResponse
 from ...types.dataset_generate_validators_response import DatasetGenerateValidatorsResponse
@@ -472,7 +471,7 @@ def insert_entry(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> str:
+    ) -> DatasetInsertEntryResponse:
         """
         Insert an entry into a dataset.
 
@@ -493,53 +492,7 @@ def insert_entry(
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=str,
-        )
-
-    def upload_jsonl(
-        self,
-        dataset_id: str,
-        *,
-        file: FileTypes,
-        # 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,
-    ) -> DatasetUploadJSONLResponse:
-        """
-        Upload a JSONL file to a dataset.
-
-        Accepts a JSONL (JSON Lines) file where each line is a valid JSON object. Each
-        entry is stored in S3 and a JSON schema is automatically generated from all
-        entries using GenSON. The dataset status will be updated to READY.
-
-        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}")
-        body = deepcopy_minimal({"file": file})
-        files = extract_files(cast(Mapping[str, object], body), paths=[["file"]])
-        # It should be noted that the actual Content-Type header that will be
-        # sent to the server will contain a `boundary` parameter, e.g.
-        # multipart/form-data; boundary=---abc--
-        extra_headers = {"Content-Type": "multipart/form-data", **(extra_headers or {})}
-        return self._post(
-            f"/datasets/{dataset_id}/upload-jsonl",
-            body=maybe_transform(body, dataset_upload_jsonl_params.DatasetUploadJSONLParams),
-            files=files,
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=DatasetUploadJSONLResponse,
+            cast_to=DatasetInsertEntryResponse,
         )
 
 
@@ -971,7 +924,7 @@ async def insert_entry(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> str:
+    ) -> DatasetInsertEntryResponse:
         """
         Insert an entry into a dataset.
 
@@ -992,53 +945,7 @@ async def insert_entry(
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=str,
-        )
-
-    async def upload_jsonl(
-        self,
-        dataset_id: str,
-        *,
-        file: FileTypes,
-        # 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,
-    ) -> DatasetUploadJSONLResponse:
-        """
-        Upload a JSONL file to a dataset.
-
-        Accepts a JSONL (JSON Lines) file where each line is a valid JSON object. Each
-        entry is stored in S3 and a JSON schema is automatically generated from all
-        entries using GenSON. The dataset status will be updated to READY.
-
-        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}")
-        body = deepcopy_minimal({"file": file})
-        files = extract_files(cast(Mapping[str, object], body), paths=[["file"]])
-        # It should be noted that the actual Content-Type header that will be
-        # sent to the server will contain a `boundary` parameter, e.g.
-        # multipart/form-data; boundary=---abc--
-        extra_headers = {"Content-Type": "multipart/form-data", **(extra_headers or {})}
-        return await self._post(
-            f"/datasets/{dataset_id}/upload-jsonl",
-            body=await async_maybe_transform(body, dataset_upload_jsonl_params.DatasetUploadJSONLParams),
-            files=files,
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=DatasetUploadJSONLResponse,
+            cast_to=DatasetInsertEntryResponse,
         )
 
 
@@ -1076,9 +983,6 @@ def __init__(self, datasets: DatasetsResource) -> None:
         self.insert_entry = to_raw_response_wrapper(
             datasets.insert_entry,
         )
-        self.upload_jsonl = to_raw_response_wrapper(
-            datasets.upload_jsonl,
-        )
 
     @cached_property
     def generations(self) -> GenerationsResourceWithRawResponse:
@@ -1119,9 +1023,6 @@ def __init__(self, datasets: AsyncDatasetsResource) -> None:
         self.insert_entry = async_to_raw_response_wrapper(
             datasets.insert_entry,
         )
-        self.upload_jsonl = async_to_raw_response_wrapper(
-            datasets.upload_jsonl,
-        )
 
     @cached_property
     def generations(self) -> AsyncGenerationsResourceWithRawResponse:
@@ -1162,9 +1063,6 @@ def __init__(self, datasets: DatasetsResource) -> None:
         self.insert_entry = to_streamed_response_wrapper(
             datasets.insert_entry,
         )
-        self.upload_jsonl = to_streamed_response_wrapper(
-            datasets.upload_jsonl,
-        )
 
     @cached_property
     def generations(self) -> GenerationsResourceWithStreamingResponse:
@@ -1205,9 +1103,6 @@ def __init__(self, datasets: AsyncDatasetsResource) -> None:
         self.insert_entry = async_to_streamed_response_wrapper(
             datasets.insert_entry,
         )
-        self.upload_jsonl = async_to_streamed_response_wrapper(
-            datasets.upload_jsonl,
-        )
 
     @cached_property
     def generations(self) -> AsyncGenerationsResourceWithStreamingResponse:

src/lemma/types/__init__.py

@@ -20,9 +20,7 @@
 from .dataset_delete_response import DatasetDeleteResponse as DatasetDeleteResponse
 from .project_delete_response import ProjectDeleteResponse as ProjectDeleteResponse
 from .dataset_insert_entry_params import DatasetInsertEntryParams as DatasetInsertEntryParams
-from .dataset_upload_jsonl_params import DatasetUploadJSONLParams as DatasetUploadJSONLParams
 from .dataset_insert_entry_response import DatasetInsertEntryResponse as DatasetInsertEntryResponse
-from .dataset_upload_jsonl_response import DatasetUploadJSONLResponse as DatasetUploadJSONLResponse
 from .dataset_generate_schema_params import DatasetGenerateSchemaParams as DatasetGenerateSchemaParams
 from .dataset_generate_dataset_params import DatasetGenerateDatasetParams as DatasetGenerateDatasetParams
 from .dataset_generate_schema_response import DatasetGenerateSchemaResponse as DatasetGenerateSchemaResponse

src/lemma/types/dataset_insert_entry_response.py

@@ -1,7 +1,13 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-from typing_extensions import TypeAlias
+from .._models import BaseModel
 
 __all__ = ["DatasetInsertEntryResponse"]
 
-DatasetInsertEntryResponse: TypeAlias = str
+
+class DatasetInsertEntryResponse(BaseModel):
+    entry_id: str
+    """The ID of the inserted entry"""
+
+    storage_bytes: int
+    """The storage bytes of the inserted entry"""