🌿 Fern Regeneration -- April 8, 2026 (#751)

* SDK regeneration

* Update python version used in ci

* nit

* Add aiohttp test to ci

* fix custom tests

* nits

* fix

* nit

* nit

---------

Co-authored-by: fern-api[bot] <115122769+fern-api[bot]@users.noreply.github.com>
Co-authored-by: fern-support <126544928+fern-support@users.noreply.github.com>
Co-authored-by: jsklan <jsklan.development@gmail.com>

Author: 3 people

.fern/metadata.json

@@ -1,9 +1,8 @@
 {
-  "cliVersion": "4.48.0",
+  "cliVersion": "4.63.2",
   "generatorName": "fernapi/fern-python-sdk",
-  "generatorVersion": "4.63.4",
+  "generatorVersion": "5.3.3",
   "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": "f6bbd671944a203a435dd01ebe619dcb4e493503",
+  "sdkVersion": "6.0.0"
 }

.github/workflows/ci.yml

@@ -10,7 +10,7 @@ jobs:
       - name: Set up python
         uses: actions/setup-python@v4
         with:
-          python-version: 3.9
+          python-version: "3.10"
       - name: Bootstrap poetry
         uses: snok/install-poetry@v1
         with:
@@ -28,7 +28,7 @@ jobs:
       - name: Set up python
         uses: actions/setup-python@v4
         with:
-          python-version: 3.9
+          python-version: "3.10"
       - name: Bootstrap poetry
         uses: snok/install-poetry@v1
         with:
@@ -39,9 +39,17 @@ jobs:
       - name: Install aws deps
         run: poetry run pip install boto3 sagemaker botocore
       - name: Test
-        run: poetry run pytest .
+        run: poetry run pytest -rP -n auto .
         env:
           CO_API_KEY: ${{ secrets.COHERE_API_KEY }}
+      - name: Install aiohttp extra
+        run: poetry install --extras aiohttp
+
+      - name: Test (aiohttp)
+        run: poetry run pytest -rP -n auto -m aiohttp . || [ $? -eq 5 ]
+        env:
+          CO_API_KEY: ${{ secrets.COHERE_API_KEY }}
+
   publish:
     needs: [compile, test]
     if: github.event_name == 'push' && contains(github.ref, 'refs/tags/')
@@ -52,7 +60,7 @@ jobs:
       - name: Set up python
         uses: actions/setup-python@v4
         with:
-          python-version: 3.9
+          python-version: "3.10"
       - name: Bootstrap poetry
         run: |
           curl -sSL https://install.python-poetry.org | python - -y --version 1.5.1

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

@@ -2309,7 +2309,7 @@ Follow the [Migration Guide](https://docs.cohere.com/v2/docs/migrating-v1-to-v2)
 <dd>
 
 ```python
-from cohere import Client
+from cohere import Client, ChatMessageV2_User
 from cohere.environment import ClientEnvironment
 
 client = Client(
@@ -2320,10 +2320,9 @@ client = Client(
 client.v2.chat_stream(
     model="command-a-03-2025",
     messages=[
-        {
-            "role": "user",
-            "content": "Tell me about LLMs"
-        }
+        ChatMessageV2_User(
+            content="Tell me about LLMs",
+        )
     ],
 )
 
@@ -2608,7 +2607,7 @@ Follow the [Migration Guide](https://docs.cohere.com/v2/docs/migrating-v1-to-v2)
 <dd>
 
 ```python
-from cohere import Client
+from cohere import Client, ChatMessageV2_User
 from cohere.environment import ClientEnvironment
 
 client = Client(
@@ -2619,10 +2618,9 @@ client = Client(
 client.v2.chat_stream(
     model="command-a-03-2025",
     messages=[
-        {
-            "role": "user",
-            "content": "Tell me about LLMs"
-        }
+        ChatMessageV2_User(
+            content="Tell me about LLMs",
+        )
     ],
 )
 

reference.md

@@ -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

@@ -153,6 +153,7 @@
         Generation,
         GetConnectorResponse,
         GetModelResponse,
+        GetModelResponseSamplingDefaults,
         Image,
         ImageContent,
         ImageUrl,
@@ -257,6 +258,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 +450,8 @@
     "DatasetsListResponse": ".datasets",
     "DebugStreamedChatResponse": ".types",
     "DebugV2ChatStreamResponse": ".v2",
+    "DefaultAioHttpClient": "._default_clients",
+    "DefaultAsyncHttpxClient": "._default_clients",
     "DeleteConnectorResponse": ".types",
     "DetokenizeResponse": ".types",
     "Document": ".types",
@@ -490,6 +494,7 @@
     "GetBatchResponse": ".batches",
     "GetConnectorResponse": ".types",
     "GetModelResponse": ".types",
+    "GetModelResponseSamplingDefaults": ".types",
     "Image": ".types",
     "ImageContent": ".types",
     "ImageUrl": ".types",
@@ -782,6 +787,8 @@ def __dir__():
     "DatasetsListResponse",
     "DebugStreamedChatResponse",
     "DebugV2ChatStreamResponse",
+    "DefaultAioHttpClient",
+    "DefaultAsyncHttpxClient",
     "DeleteConnectorResponse",
     "DetokenizeResponse",
     "Document",
@@ -824,6 +831,7 @@ def __dir__():
     "GetBatchResponse",
     "GetConnectorResponse",
     "GetModelResponse",
+    "GetModelResponseSamplingDefaults",
     "Image",
     "ImageContent",
     "ImageUrl",

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.
+"""