SDK regeneration

Author: fern-api[bot]

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "cohere-ai"
-version = "5.0.0"
+version = "5.0.0a0"
 description = ""
 readme = "README.md"
 authors = []

src/cohere/__init__.py

@@ -36,15 +36,22 @@
     CreateConnectorOAuth,
     CreateConnectorResponse,
     CreateConnectorServiceAuth,
+    CreateEmbedJobResponse,
     Dataset,
     DatasetPart,
+    DatasetType,
+    DatasetValidationStatus,
     DeleteConnectorResponse,
     DetectLanguageResponse,
     DetectLanguageResponseResultsItem,
     DetokenizeResponse,
     EmbedByTypeResponse,
     EmbedByTypeResponseEmbeddings,
     EmbedFloatsResponse,
+    EmbedInputType,
+    EmbedJob,
+    EmbedJobStatus,
+    EmbedJobTruncate,
     EmbedRequestTruncate,
     EmbedResponse,
     EmbedResponse_EmbeddingsByType,
@@ -66,6 +73,7 @@
     Generation,
     GetConnectorResponse,
     ListConnectorsResponse,
+    ListEmbedJobResponse,
     NonStreamedChatResponse,
     OAuthAuthorizeResponse,
     RerankRequestDocumentsItem,
@@ -93,12 +101,14 @@
 )
 from .errors import BadRequestError, ForbiddenError, InternalServerError, NotFoundError
 from .resources import (
+    CreateEmbedJobRequestTruncate,
     DatasetsCreateResponse,
     DatasetsGetResponse,
     DatasetsGetUsageResponse,
     DatasetsListResponse,
     connectors,
     datasets,
+    embed_jobs,
 )
 from .environment import ClientEnvironment
 
@@ -140,8 +150,12 @@
     "CreateConnectorOAuth",
     "CreateConnectorResponse",
     "CreateConnectorServiceAuth",
+    "CreateEmbedJobRequestTruncate",
+    "CreateEmbedJobResponse",
     "Dataset",
     "DatasetPart",
+    "DatasetType",
+    "DatasetValidationStatus",
     "DatasetsCreateResponse",
     "DatasetsGetResponse",
     "DatasetsGetUsageResponse",
@@ -153,6 +167,10 @@
     "EmbedByTypeResponse",
     "EmbedByTypeResponseEmbeddings",
     "EmbedFloatsResponse",
+    "EmbedInputType",
+    "EmbedJob",
+    "EmbedJobStatus",
+    "EmbedJobTruncate",
     "EmbedRequestTruncate",
     "EmbedResponse",
     "EmbedResponse_EmbeddingsByType",
@@ -176,6 +194,7 @@
     "GetConnectorResponse",
     "InternalServerError",
     "ListConnectorsResponse",
+    "ListEmbedJobResponse",
     "NonStreamedChatResponse",
     "NotFoundError",
     "OAuthAuthorizeResponse",
@@ -203,4 +222,5 @@
     "UpdateConnectorResponse",
     "connectors",
     "datasets",
+    "embed_jobs",
 ]

src/cohere/client.py

@@ -21,7 +21,7 @@ def get_headers(self) -> typing.Dict[str, str]:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "cohere-ai",
-            "X-Fern-SDK-Version": "5.0.0",
+            "X-Fern-SDK-Version": "5.0.0a0",
         }
         if self._client_name is not None:
             headers["X-Client-Name"] = self._client_name

src/cohere/core/client_wrapper.py

@@ -1,13 +1,16 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from . import connectors, datasets
+from . import connectors, datasets, embed_jobs
 from .datasets import DatasetsCreateResponse, DatasetsGetResponse, DatasetsGetUsageResponse, DatasetsListResponse
+from .embed_jobs import CreateEmbedJobRequestTruncate
 
 __all__ = [
+    "CreateEmbedJobRequestTruncate",
     "DatasetsCreateResponse",
     "DatasetsGetResponse",
     "DatasetsGetUsageResponse",
     "DatasetsListResponse",
     "connectors",
     "datasets",
+    "embed_jobs",
 ]

src/cohere/resources/__init__.py

@@ -263,12 +263,14 @@ def update(
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def o_auth_authorize(self, id: str) -> OAuthAuthorizeResponse:
+    def o_auth_authorize(self, id: str, *, after_token_redirect: typing.Optional[str] = None) -> OAuthAuthorizeResponse:
         """
         Authorize the connector with the given ID for the connector oauth app. See ['Connector Authentication'](https://docs.cohere.com/docs/connector-authentication) for more information.
 
         Parameters:
             - id: str. The ID of the connector to authorize.
+
+            - after_token_redirect: typing.Optional[str]. The URL to redirect to after the connector has been authorized.
         ---
         from cohere.client import Client
 
@@ -283,6 +285,7 @@ def o_auth_authorize(self, id: str) -> OAuthAuthorizeResponse:
         _response = self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"v1/connectors/{id}/oauth/authorize"),
+            params=remove_none_from_dict({"after_token_redirect": after_token_redirect}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -535,12 +538,16 @@ async def update(
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def o_auth_authorize(self, id: str) -> OAuthAuthorizeResponse:
+    async def o_auth_authorize(
+        self, id: str, *, after_token_redirect: typing.Optional[str] = None
+    ) -> OAuthAuthorizeResponse:
         """
         Authorize the connector with the given ID for the connector oauth app. See ['Connector Authentication'](https://docs.cohere.com/docs/connector-authentication) for more information.
 
         Parameters:
             - id: str. The ID of the connector to authorize.
+
+            - after_token_redirect: typing.Optional[str]. The URL to redirect to after the connector has been authorized.
         ---
         from cohere.client import AsyncClient
 
@@ -555,6 +562,7 @@ async def o_auth_authorize(self, id: str) -> OAuthAuthorizeResponse:
         _response = await self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"v1/connectors/{id}/oauth/authorize"),
+            params=remove_none_from_dict({"after_token_redirect": after_token_redirect}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )

src/cohere/resources/connectors/client.py

@@ -10,6 +10,7 @@
 from ...core.datetime_utils import serialize_datetime
 from ...core.jsonable_encoder import jsonable_encoder
 from ...core.remove_none_from_dict import remove_none_from_dict
+from ...types.dataset_type import DatasetType
 from .types.datasets_create_response import DatasetsCreateResponse
 from .types.datasets_get_response import DatasetsGetResponse
 from .types.datasets_get_usage_response import DatasetsGetUsageResponse
@@ -86,8 +87,11 @@ def create(
         self,
         *,
         name: typing.Optional[str] = None,
-        type: typing.Optional[str] = None,
+        type: typing.Optional[DatasetType] = None,
         keep_original_file: typing.Optional[bool] = None,
+        skip_malformed_input: typing.Optional[bool] = None,
+        keep_fields: typing.Optional[typing.Union[str, typing.List[str]]] = None,
+        optional_fields: typing.Optional[typing.Union[str, typing.List[str]]] = None,
         text_separator: typing.Optional[str] = None,
         csv_delimiter: typing.Optional[str] = None,
         data: typing.IO,
@@ -99,10 +103,16 @@ def create(
         Parameters:
             - name: typing.Optional[str]. The name of the uploaded dataset.
 
-            - type: typing.Optional[str]. The dataset type, which is used to validate the data.
+            - type: typing.Optional[DatasetType]. The dataset type, which is used to validate the data.
 
             - keep_original_file: typing.Optional[bool]. Indicates if the original file should be stored.
 
+            - skip_malformed_input: typing.Optional[bool]. Indicates whether rows with malformed input should be dropped (instead of failing the validation check). Dropped rows will be returned in the warnings field.
+
+            - keep_fields: typing.Optional[typing.Union[str, typing.List[str]]]. List of names of fields that will be persisted in the Dataset. By default the Dataset will retain only the required fields indicated in the [schema for the corresponding Dataset type](https://docs.cohere.com/docs/datasets#dataset-types). For example, datasets of type `embed-input` will drop all fields other than the required `text` field. If any of the fields in `keep_fields` are missing from the uploaded file, Dataset validation will fail.
+
+            - optional_fields: typing.Optional[typing.Union[str, typing.List[str]]]. List of names of fields that will be persisted in the Dataset. By default the Dataset will retain only the required fields indicated in the [schema for the corresponding Dataset type](https://docs.cohere.com/docs/datasets#dataset-types). For example, Datasets of type `embed-input` will drop all fields other than the required `text` field. If any of the fields in `optional_fields` are missing from the uploaded file, Dataset validation will pass.
+
             - text_separator: typing.Optional[str]. Raw .txt uploads will be split into entries using the text_separator value.
 
             - csv_delimiter: typing.Optional[str]. The delimiter used for .csv uploads.
@@ -119,6 +129,9 @@ def create(
                     "name": name,
                     "type": type,
                     "keep_original_file": keep_original_file,
+                    "skip_malformed_input": skip_malformed_input,
+                    "keep_fields": keep_fields,
+                    "optional_fields": optional_fields,
                     "text_separator": text_separator,
                     "csv_delimiter": csv_delimiter,
                 }
@@ -288,8 +301,11 @@ async def create(
         self,
         *,
         name: typing.Optional[str] = None,
-        type: typing.Optional[str] = None,
+        type: typing.Optional[DatasetType] = None,
         keep_original_file: typing.Optional[bool] = None,
+        skip_malformed_input: typing.Optional[bool] = None,
+        keep_fields: typing.Optional[typing.Union[str, typing.List[str]]] = None,
+        optional_fields: typing.Optional[typing.Union[str, typing.List[str]]] = None,
         text_separator: typing.Optional[str] = None,
         csv_delimiter: typing.Optional[str] = None,
         data: typing.IO,
@@ -301,10 +317,16 @@ async def create(
         Parameters:
             - name: typing.Optional[str]. The name of the uploaded dataset.
 
-            - type: typing.Optional[str]. The dataset type, which is used to validate the data.
+            - type: typing.Optional[DatasetType]. The dataset type, which is used to validate the data.
 
             - keep_original_file: typing.Optional[bool]. Indicates if the original file should be stored.
 
+            - skip_malformed_input: typing.Optional[bool]. Indicates whether rows with malformed input should be dropped (instead of failing the validation check). Dropped rows will be returned in the warnings field.
+
+            - keep_fields: typing.Optional[typing.Union[str, typing.List[str]]]. List of names of fields that will be persisted in the Dataset. By default the Dataset will retain only the required fields indicated in the [schema for the corresponding Dataset type](https://docs.cohere.com/docs/datasets#dataset-types). For example, datasets of type `embed-input` will drop all fields other than the required `text` field. If any of the fields in `keep_fields` are missing from the uploaded file, Dataset validation will fail.
+
+            - optional_fields: typing.Optional[typing.Union[str, typing.List[str]]]. List of names of fields that will be persisted in the Dataset. By default the Dataset will retain only the required fields indicated in the [schema for the corresponding Dataset type](https://docs.cohere.com/docs/datasets#dataset-types). For example, Datasets of type `embed-input` will drop all fields other than the required `text` field. If any of the fields in `optional_fields` are missing from the uploaded file, Dataset validation will pass.
+
             - text_separator: typing.Optional[str]. Raw .txt uploads will be split into entries using the text_separator value.
 
             - csv_delimiter: typing.Optional[str]. The delimiter used for .csv uploads.
@@ -321,6 +343,9 @@ async def create(
                     "name": name,
                     "type": type,
                     "keep_original_file": keep_original_file,
+                    "skip_malformed_input": skip_malformed_input,
+                    "keep_fields": keep_fields,
+                    "optional_fields": optional_fields,
                     "text_separator": text_separator,
                     "csv_delimiter": csv_delimiter,
                 }

src/cohere/resources/datasets/client.py

@@ -13,7 +13,7 @@
 
 class DatasetsGetUsageResponse(pydantic.BaseModel):
     organization_usage: typing.Optional[str] = pydantic.Field(
-        alias="organizationUsage", description="The total number of bytes used by the organization."
+        description="The total number of bytes used by the organization."
     )
 
     def json(self, **kwargs: typing.Any) -> str:
@@ -27,5 +27,4 @@ def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
     class Config:
         frozen = True
         smart_union = True
-        allow_population_by_field_name = True
         json_encoders = {dt.datetime: serialize_datetime}

src/cohere/resources/datasets/types/datasets_get_usage_response.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from .types import CreateEmbedJobRequestTruncate
+
+__all__ = ["CreateEmbedJobRequestTruncate"]