SDK regeneration (#509)

fern-api[bot] · web-flow · commit 457f5d7f2014 · 2024-05-23T11:29:02.000-05:00
Co-authored-by: fern-api &lt;115122769+fern-api[bot]@users.noreply.github.com&gt;
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "cohere"
-version = "5.5.2"
+version = "5.5.3"
 description = ""
 readme = "README.md"
 authors = []
diff --git a/src/cohere/base_client.py b/src/cohere/base_client.py
@@ -65,7 +65,7 @@
 
 class BaseCohere:
     """
-    Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propogate to these functions.
+    Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propagate to these functions.
 
     Parameters
     ----------
@@ -164,7 +164,7 @@ def chat_stream(
     ) -> typing.Iterator[StreamedChatResponse]:
         """
         Generates a text response to a user message.
-        To learn how to use Chat with Streaming and RAG follow [this guide](https://docs.cohere.com/docs/cochat-beta#various-ways-of-using-the-chat-endpoint).
+        To learn how to use the Chat API with Streaming and RAG follow our [Text Generation guides](https://docs.cohere.com/docs/chat-api).
 
         Parameters
         ----------
@@ -460,7 +460,7 @@ def chat_stream(
         for chunk in response:
             yield chunk
         """
-        _request: typing.Dict[str, typing.Any] = {"message": message}
+        _request: typing.Dict[str, typing.Any] = {"message": message, "stream": True}
         if model is not OMIT:
             _request["model"] = model
         if preamble is not OMIT:
@@ -583,7 +583,7 @@ def chat(
     ) -> NonStreamedChatResponse:
         """
         Generates a text response to a user message.
-        To learn how to use Chat with Streaming and RAG follow [this guide](https://docs.cohere.com/docs/cochat-beta#various-ways-of-using-the-chat-endpoint).
+        To learn how to use the Chat API with Streaming and RAG follow our [Text Generation guides](https://docs.cohere.com/docs/chat-api).
 
         Parameters
         ----------
@@ -803,7 +803,7 @@ def chat(
             temperature=0.3,
         )
         """
-        _request: typing.Dict[str, typing.Any] = {"message": message}
+        _request: typing.Dict[str, typing.Any] = {"message": message, "stream": False}
         if model is not OMIT:
             _request["model"] = model
         if preamble is not OMIT:
@@ -1038,7 +1038,7 @@ def generate_stream(
         for chunk in response:
             yield chunk
         """
-        _request: typing.Dict[str, typing.Any] = {"prompt": prompt}
+        _request: typing.Dict[str, typing.Any] = {"prompt": prompt, "stream": True}
         if model is not OMIT:
             _request["model"] = model
         if num_generations is not OMIT:
@@ -1253,7 +1253,7 @@ def generate(
             prompt="Please explain to me how LLMs work",
         )
         """
-        _request: typing.Dict[str, typing.Any] = {"prompt": prompt}
+        _request: typing.Dict[str, typing.Any] = {"prompt": prompt, "stream": False}
         if model is not OMIT:
             _request["model"] = model
         if num_generations is not OMIT:
@@ -2084,7 +2084,7 @@ def check_api_key(self, *, request_options: typing.Optional[RequestOptions] = No
 
 class AsyncBaseCohere:
     """
-    Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propogate to these functions.
+    Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propagate to these functions.
 
     Parameters
     ----------
@@ -2183,7 +2183,7 @@ async def chat_stream(
     ) -> typing.AsyncIterator[StreamedChatResponse]:
         """
         Generates a text response to a user message.
-        To learn how to use Chat with Streaming and RAG follow [this guide](https://docs.cohere.com/docs/cochat-beta#various-ways-of-using-the-chat-endpoint).
+        To learn how to use the Chat API with Streaming and RAG follow our [Text Generation guides](https://docs.cohere.com/docs/chat-api).
 
         Parameters
         ----------
@@ -2479,7 +2479,7 @@ async def chat_stream(
         async for chunk in response:
             yield chunk
         """
-        _request: typing.Dict[str, typing.Any] = {"message": message}
+        _request: typing.Dict[str, typing.Any] = {"message": message, "stream": True}
         if model is not OMIT:
             _request["model"] = model
         if preamble is not OMIT:
@@ -2602,7 +2602,7 @@ async def chat(
     ) -> NonStreamedChatResponse:
         """
         Generates a text response to a user message.
-        To learn how to use Chat with Streaming and RAG follow [this guide](https://docs.cohere.com/docs/cochat-beta#various-ways-of-using-the-chat-endpoint).
+        To learn how to use the Chat API with Streaming and RAG follow our [Text Generation guides](https://docs.cohere.com/docs/chat-api).
 
         Parameters
         ----------
@@ -2822,7 +2822,7 @@ async def chat(
             temperature=0.3,
         )
         """
-        _request: typing.Dict[str, typing.Any] = {"message": message}
+        _request: typing.Dict[str, typing.Any] = {"message": message, "stream": False}
         if model is not OMIT:
             _request["model"] = model
         if preamble is not OMIT:
@@ -3057,7 +3057,7 @@ async def generate_stream(
         async for chunk in response:
             yield chunk
         """
-        _request: typing.Dict[str, typing.Any] = {"prompt": prompt}
+        _request: typing.Dict[str, typing.Any] = {"prompt": prompt, "stream": True}
         if model is not OMIT:
             _request["model"] = model
         if num_generations is not OMIT:
@@ -3272,7 +3272,7 @@ async def generate(
             prompt="Please explain to me how LLMs work",
         )
         """
-        _request: typing.Dict[str, typing.Any] = {"prompt": prompt}
+        _request: typing.Dict[str, typing.Any] = {"prompt": prompt, "stream": False}
         if model is not OMIT:
             _request["model"] = model
         if num_generations is not OMIT:
diff --git a/src/cohere/connectors/client.py b/src/cohere/connectors/client.py
@@ -362,6 +362,9 @@ def delete(self, id: str, *, request_options: typing.Optional[RequestOptions] =
                     request_options.get("additional_query_parameters") if request_options is not None else None
                 )
             ),
+            json=jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))
+            if request_options is not None
+            else None,
             headers=jsonable_encoder(
                 remove_none_from_dict(
                     {
@@ -968,6 +971,9 @@ async def delete(
                     request_options.get("additional_query_parameters") if request_options is not None else None
                 )
             ),
+            json=jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))
+            if request_options is not None
+            else None,
             headers=jsonable_encoder(
                 remove_none_from_dict(
                     {
diff --git a/src/cohere/core/client_wrapper.py b/src/cohere/core/client_wrapper.py
@@ -25,7 +25,7 @@ def get_headers(self) -> typing.Dict[str, str]:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "cohere",
-            "X-Fern-SDK-Version": "5.5.2",
+            "X-Fern-SDK-Version": "5.5.3",
         }
         if self._client_name is not None:
             headers["X-Client-Name"] = self._client_name
diff --git a/src/cohere/core/unchecked_base_model.py b/src/cohere/core/unchecked_base_model.py
@@ -8,7 +8,7 @@
 import typing_extensions
 
 from .datetime_utils import serialize_datetime
-from .pydantic_utilities import pydantic_v1
+from .pydantic_utilities import IS_PYDANTIC_V2, pydantic_v1
 
 
 class UnionMetadata:
@@ -60,6 +60,7 @@ def construct(
                 else:
                     type_ = typing.cast(typing.Type, field.outer_type_)  # type: ignore
                     fields_values[name] = construct_type(object_=values[key], type_=type_)
+                _fields_set.add(name)
             elif not field.required:
                 default = field.get_default()
                 fields_values[name] = default
@@ -71,10 +72,19 @@ def construct(
                     _fields_set.add(key)
 
         # Add extras back in
+        _extra = {}
         for key, value in values.items():
-            if key not in cls.__fields__:
-                _fields_set.add(key)
-                fields_values[key] = value
+            if key not in _fields_set:
+                _extra[key] = value
+                # In v2 we'll need to exclude extra fields from fields_values
+                if not IS_PYDANTIC_V2:
+                    _fields_set.add(key)
+                    fields_values[key] = value
+
+        if IS_PYDANTIC_V2:
+            object.__setattr__(m, "__pydantic_private__", None)
+            object.__setattr__(m, "__pydantic_extra__", _extra)
+            object.__setattr__(m, "__pydantic_fields_set__", _fields_set)
 
         object.__setattr__(m, "__dict__", fields_values)
         object.__setattr__(m, "__fields_set__", _fields_set)
@@ -144,8 +154,12 @@ def construct_type(*, type_: typing.Type[typing.Any], object_: typing.Any) -> ty
         if not isinstance(object_, typing.Mapping):
             return object_
 
-        _, items_type = pydantic_v1.typing.get_args(type_)
-        return {key: construct_type(object_=item, type_=items_type) for key, item in object_.items()}
+        key_type, items_type = pydantic_v1.typing.get_args(type_)
+        d = {
+            construct_type(object_=key, type_=key_type): construct_type(object_=item, type_=items_type)
+            for key, item in object_.items()
+        }
+        return d
 
     if base_type == list:
         if not isinstance(object_, list):
@@ -190,4 +204,20 @@ def construct_type(*, type_: typing.Type[typing.Any], object_: typing.Any) -> ty
         except Exception:
             return object_
 
+    if base_type == int:
+        try:
+            return int(object_)
+        except Exception:
+            return object_
+
+    if base_type == bool:
+        try:
+            if isinstance(object_, str):
+                stringified_object = object_.lower()
+                return stringified_object == "true" or stringified_object == "1"
+
+            return bool(object_)
+        except Exception:
+            return object_
+
     return object_
diff --git a/src/cohere/datasets/client.py b/src/cohere/datasets/client.py
@@ -422,6 +422,9 @@ def delete(
                     request_options.get("additional_query_parameters") if request_options is not None else None
                 )
             ),
+            json=jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))
+            if request_options is not None
+            else None,
             headers=jsonable_encoder(
                 remove_none_from_dict(
                     {
@@ -844,6 +847,9 @@ async def delete(
                     request_options.get("additional_query_parameters") if request_options is not None else None
                 )
             ),
+            json=jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))
+            if request_options is not None
+            else None,
             headers=jsonable_encoder(
                 remove_none_from_dict(
                     {
diff --git a/src/cohere/finetuning/client.py b/src/cohere/finetuning/client.py
@@ -368,6 +368,9 @@ def delete_finetuned_model(
                     request_options.get("additional_query_parameters") if request_options is not None else None
                 )
             ),
+            json=jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))
+            if request_options is not None
+            else None,
             headers=jsonable_encoder(
                 remove_none_from_dict(
                     {
@@ -1127,6 +1130,9 @@ async def delete_finetuned_model(
                     request_options.get("additional_query_parameters") if request_options is not None else None
                 )
             ),
+            json=jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))
+            if request_options is not None
+            else None,
             headers=jsonable_encoder(
                 remove_none_from_dict(
                     {

Original file line number	Diff line number	Diff line change
`@@ -362,6 +362,9 @@ def delete(self, id: str, *, request_options: typing.Optional[RequestOptions] =`
`362`	`362`	`request_options.get("additional_query_parameters") if request_options is not None else None`
`363`	`363`	`)`
`364`	`364`	`),`
	`365`	`+ json=jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))`
	`366`	`+ if request_options is not None`
	`367`	`+ else None,`
`365`	`368`	`headers=jsonable_encoder(`
`366`	`369`	`remove_none_from_dict(`
`367`	`370`	`{`
`@@ -968,6 +971,9 @@ async def delete(`
`968`	`971`	`request_options.get("additional_query_parameters") if request_options is not None else None`
`969`	`972`	`)`
`970`	`973`	`),`
	`974`	`+ json=jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))`
	`975`	`+ if request_options is not None`
	`976`	`+ else None,`
`971`	`977`	`headers=jsonable_encoder(`
`972`	`978`	`remove_none_from_dict(`
`973`	`979`	`{`
Original file line number	Diff line number	Diff line change
`@@ -25,7 +25,7 @@ def get_headers(self) -> typing.Dict[str, str]:`
`25`	`25`	`headers: typing.Dict[str, str] = {`
`26`	`26`	`"X-Fern-Language": "Python",`
`27`	`27`	`"X-Fern-SDK-Name": "cohere",`
`28`		`- "X-Fern-SDK-Version": "5.5.2",`
	`28`	`+ "X-Fern-SDK-Version": "5.5.3",`
`29`	`29`	`}`
`30`	`30`	`if self._client_name is not None:`
`31`	`31`	`headers["X-Client-Name"] = self._client_name`
Original file line number	Diff line number	Diff line change
`@@ -422,6 +422,9 @@ def delete(`
`422`	`422`	`request_options.get("additional_query_parameters") if request_options is not None else None`
`423`	`423`	`)`
`424`	`424`	`),`
	`425`	`+ json=jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))`
	`426`	`+ if request_options is not None`
	`427`	`+ else None,`
`425`	`428`	`headers=jsonable_encoder(`
`426`	`429`	`remove_none_from_dict(`
`427`	`430`	`{`
`@@ -844,6 +847,9 @@ async def delete(`
`844`	`847`	`request_options.get("additional_query_parameters") if request_options is not None else None`
`845`	`848`	`)`
`846`	`849`	`),`
	`850`	`+ json=jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))`
	`851`	`+ if request_options is not None`
	`852`	`+ else None,`
`847`	`853`	`headers=jsonable_encoder(`
`848`	`854`	`remove_none_from_dict(`
`849`	`855`	`{`
Original file line number	Diff line number	Diff line change
`@@ -368,6 +368,9 @@ def delete_finetuned_model(`
`368`	`368`	`request_options.get("additional_query_parameters") if request_options is not None else None`
`369`	`369`	`)`
`370`	`370`	`),`
	`371`	`+ json=jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))`
	`372`	`+ if request_options is not None`
	`373`	`+ else None,`
`371`	`374`	`headers=jsonable_encoder(`
`372`	`375`	`remove_none_from_dict(`
`373`	`376`	`{`
`@@ -1127,6 +1130,9 @@ async def delete_finetuned_model(`
`1127`	`1130`	`request_options.get("additional_query_parameters") if request_options is not None else None`
`1128`	`1131`	`)`
`1129`	`1132`	`),`
	`1133`	`+ json=jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))`
	`1134`	`+ if request_options is not None`
	`1135`	`+ else None,`
`1130`	`1136`	`headers=jsonable_encoder(`
`1131`	`1137`	`remove_none_from_dict(`
`1132`	`1138`	`{`