SDK regeneration

Author: fern-api[bot]

.fern/metadata.json

@@ -1,9 +1,8 @@
 {
-  "cliVersion": "4.48.0",
+  "cliVersion": "4.62.0",
   "generatorName": "fernapi/fern-python-sdk",
-  "generatorVersion": "4.63.4",
+  "generatorVersion": "5.3.1",
   "generatorConfig": {
-    "pyproject_python_version": "^3.9",
     "inline_request_params": false,
     "extra_dependencies": {
       "fastavro": "^1.9.4",
@@ -79,6 +78,6 @@
       }
     ]
   },
-  "originGitCommit": "40f3fd11b9fd6b445b024a2fcf6f2e67f6181147",
-  "sdkVersion": "5.21.2"
+  "originGitCommit": "a424220a3cebe0f93c504c5dbc7d9ac58ab7a640",
+  "sdkVersion": "6.0.0"
 }

poetry.lock

@@ -4,7 +4,7 @@ dynamic = ["version"]
 
 [tool.poetry]
 name = "cohere"
-version = "5.21.2"
+version = "6.0.0"
 description = ""
 readme = "README.md"
 authors = []
@@ -14,7 +14,6 @@ classifiers = [
     "Intended Audience :: Developers",
     "Programming Language :: Python",
     "Programming Language :: Python :: 3",
-    "Programming Language :: Python :: 3.9",
     "Programming Language :: Python :: 3.10",
     "Programming Language :: Python :: 3.11",
     "Programming Language :: Python :: 3.12",
@@ -38,20 +37,22 @@ packages = [
 Repository = 'https://github.com/cohere-ai/cohere-python'
 
 [tool.poetry.dependencies]
-python = "^3.9"
+python = "^3.10"
+aiohttp = { version = ">=3.10.0,<4", optional = true}
 fastavro = "^1.9.4"
 httpx = ">=0.21.2"
+httpx-aiohttp = { version = "0.1.8", optional = true}
 pydantic = ">= 1.9.2"
-pydantic-core = ">=2.18.2"
+pydantic-core = ">=2.18.2,<2.44.0"
 requests = "^2.0.0"
 tokenizers = ">=0.15,<1"
 types-requests = "^2.0.0"
 typing_extensions = ">= 4.0.0"
 
 [tool.poetry.group.dev.dependencies]
 mypy = "==1.13.0"
-pytest = "^7.4.0"
-pytest-asyncio = "^0.23.5"
+pytest = "^8.2.0"
+pytest-asyncio = "^1.0.0"
 pytest-xdist = "^3.6.1"
 python-dateutil = "^2.9.0"
 types-python-dateutil = "^2.9.0.20240316"
@@ -60,6 +61,9 @@ ruff = "==0.11.5"
 [tool.pytest.ini_options]
 testpaths = [ "tests" ]
 asyncio_mode = "auto"
+markers = [
+    "aiohttp: tests that require httpx_aiohttp to be installed",
+]
 
 [tool.mypy]
 plugins = ["pydantic.mypy"]
@@ -91,3 +95,6 @@ section-order = ["future", "standard-library", "third-party", "first-party"]
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
+
+[tool.poetry.extras]
+aiohttp=["aiohttp", "httpx-aiohttp"]

pyproject.toml

@@ -1,7 +1,7 @@
 fastavro==1.9.4
 httpx>=0.21.2
 pydantic>= 1.9.2
-pydantic-core>=2.18.2
+pydantic-core>=2.18.2,<2.44.0
 requests==2.0.0
 tokenizers>=0.15,<1
 types-requests==2.0.0

requirements.txt

@@ -257,6 +257,7 @@
         UnprocessableEntityError,
     )
     from . import audio, batches, connectors, datasets, embed_jobs, finetuning, models, v2
+    from ._default_clients import DefaultAioHttpClient, DefaultAsyncHttpxClient
     from .aliases import (
         ChatResponse,
         ContentDeltaStreamedChatResponseV2,
@@ -448,6 +449,8 @@
     "DatasetsListResponse": ".datasets",
     "DebugStreamedChatResponse": ".types",
     "DebugV2ChatStreamResponse": ".v2",
+    "DefaultAioHttpClient": "._default_clients",
+    "DefaultAsyncHttpxClient": "._default_clients",
     "DeleteConnectorResponse": ".types",
     "DetokenizeResponse": ".types",
     "Document": ".types",
@@ -782,6 +785,8 @@ def __dir__():
     "DatasetsListResponse",
     "DebugStreamedChatResponse",
     "DebugV2ChatStreamResponse",
+    "DefaultAioHttpClient",
+    "DefaultAsyncHttpxClient",
     "DeleteConnectorResponse",
     "DetokenizeResponse",
     "Document",

src/cohere/__init__.py

@@ -0,0 +1,30 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import httpx
+
+SDK_DEFAULT_TIMEOUT = 60
+
+try:
+    import httpx_aiohttp  # type: ignore[import-not-found]
+except ImportError:
+
+    class DefaultAioHttpClient(httpx.AsyncClient):  # type: ignore
+        def __init__(self, **kwargs: typing.Any) -> None:
+            raise RuntimeError("To use the aiohttp client, install the aiohttp extra: pip install cohere[aiohttp]")
+
+else:
+
+    class DefaultAioHttpClient(httpx_aiohttp.HttpxAiohttpClient):  # type: ignore
+        def __init__(self, **kwargs: typing.Any) -> None:
+            kwargs.setdefault("timeout", SDK_DEFAULT_TIMEOUT)
+            kwargs.setdefault("follow_redirects", True)
+            super().__init__(**kwargs)
+
+
+class DefaultAsyncHttpxClient(httpx.AsyncClient):
+    def __init__(self, **kwargs: typing.Any) -> None:
+        kwargs.setdefault("timeout", SDK_DEFAULT_TIMEOUT)
+        kwargs.setdefault("follow_redirects", True)
+        super().__init__(**kwargs)

src/cohere/_default_clients.py

@@ -1581,6 +1581,24 @@ def audio(self):
         return self._audio
 
 
+def _make_default_async_client(
+    timeout: typing.Optional[float],
+    follow_redirects: typing.Optional[bool],
+) -> httpx.AsyncClient:
+    try:
+        import httpx_aiohttp  # type: ignore[import-not-found]
+    except ImportError:
+        pass
+    else:
+        if follow_redirects is not None:
+            return httpx_aiohttp.HttpxAiohttpClient(timeout=timeout, follow_redirects=follow_redirects)
+        return httpx_aiohttp.HttpxAiohttpClient(timeout=timeout)
+
+    if follow_redirects is not None:
+        return httpx.AsyncClient(timeout=timeout, follow_redirects=follow_redirects)
+    return httpx.AsyncClient(timeout=timeout)
+
+
 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 propagate to these functions.
@@ -1604,6 +1622,9 @@ class AsyncBaseCohere:
     headers : typing.Optional[typing.Dict[str, str]]
         Additional headers to send with every request.
 
+    async_token : typing.Optional[typing.Callable[[], typing.Awaitable[str]]]
+        An async callable that returns a bearer token. Use this when token acquisition involves async I/O (e.g., refreshing tokens via an async HTTP client). When provided, this is used instead of the synchronous token for async requests.
+
     timeout : typing.Optional[float]
         The timeout to be used, in seconds, for requests. By default the timeout is 300 seconds, unless a custom httpx client is used, in which case this default is not enforced.
 
@@ -1634,6 +1655,7 @@ def __init__(
         client_name: typing.Optional[str] = None,
         token: typing.Optional[typing.Union[str, typing.Callable[[], str]]] = os.getenv("CO_API_KEY"),
         headers: typing.Optional[typing.Dict[str, str]] = None,
+        async_token: typing.Optional[typing.Callable[[], typing.Awaitable[str]]] = None,
         timeout: typing.Optional[float] = None,
         follow_redirects: typing.Optional[bool] = True,
         httpx_client: typing.Optional[httpx.AsyncClient] = None,
@@ -1649,11 +1671,10 @@ def __init__(
             client_name=client_name,
             token=token,
             headers=headers,
+            async_token=async_token,
             httpx_client=httpx_client
             if httpx_client is not None
-            else httpx.AsyncClient(timeout=_defaulted_timeout, follow_redirects=follow_redirects)
-            if follow_redirects is not None
-            else httpx.AsyncClient(timeout=_defaulted_timeout),
+            else _make_default_async_client(timeout=_defaulted_timeout, follow_redirects=follow_redirects),
             timeout=_defaulted_timeout,
             logging=logging,
         )

src/cohere/base_client.py

@@ -3,3 +3,6 @@
 import typing
 
 CancelBatchResponse = typing.Dict[str, typing.Any]
+"""
+Response to a request to cancel a batch.
+"""

src/cohere/batches/types/cancel_batch_response.py

@@ -29,12 +29,12 @@ def get_headers(self) -> typing.Dict[str, str]:
         import platform
 
         headers: typing.Dict[str, str] = {
-            "User-Agent": "cohere/5.21.2",
+            "User-Agent": "cohere/6.0.0",
             "X-Fern-Language": "Python",
             "X-Fern-Runtime": f"python/{platform.python_version()}",
             "X-Fern-Platform": f"{platform.system().lower()}/{platform.release()}",
             "X-Fern-SDK-Name": "cohere",
-            "X-Fern-SDK-Version": "5.21.2",
+            "X-Fern-SDK-Version": "6.0.0",
             **(self.get_custom_headers() or {}),
         }
         if self._client_name is not None: