Fix

Author: billytrend-cohere

.fernignore

@@ -15,4 +15,5 @@ src/cohere/bedrock_client.py
 src/cohere/aws_client.py
 src/cohere/sagemaker_client.py
 src/cohere/client_v2.py
-mypy.ini
+mypy.ini
+src/cohere/aliases.py

src/cohere/aliases.py

@@ -0,0 +1,24 @@
+from .v2 import (
+    ContentDeltaV2ChatStreamResponse,
+    ContentEndV2ChatStreamResponse,
+    ContentStartV2ChatStreamResponse,
+    MessageEndV2ChatStreamResponse,
+    MessageStartV2ChatStreamResponse,
+    ToolCallDeltaV2ChatStreamResponse,
+    ToolCallEndV2ChatStreamResponse,
+    ToolCallStartV2ChatStreamResponse,
+    V2ChatStreamResponse,
+    V2ChatResponse
+)
+
+# alias classes
+StreamedChatResponseV2 = V2ChatStreamResponse
+MessageStartStreamedChatResponseV2 = MessageStartV2ChatStreamResponse
+MessageEndStreamedChatResponseV2 = MessageEndV2ChatStreamResponse
+ContentStartStreamedChatResponseV2 = ContentStartV2ChatStreamResponse
+ContentDeltaStreamedChatResponseV2 = ContentDeltaV2ChatStreamResponse
+ContentEndStreamedChatResponseV2 = ContentEndV2ChatStreamResponse
+ToolCallStartStreamedChatResponseV2 = ToolCallStartV2ChatStreamResponse
+ToolCallDeltaStreamedChatResponseV2 = ToolCallDeltaV2ChatStreamResponse
+ToolCallEndStreamedChatResponseV2 = ToolCallEndV2ChatStreamResponse
+ChatResponse = V2ChatResponse

src/cohere/core/client_wrapper.py

@@ -22,10 +22,10 @@ def __init__(
 
     def get_headers(self) -> typing.Dict[str, str]:
         headers: typing.Dict[str, str] = {
-            "User-Agent": "cohere/6.0.0",
+            "User-Agent": "cohere/5.16.1",
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "cohere",
-            "X-Fern-SDK-Version": "6.0.0",
+            "X-Fern-SDK-Version": "5.16.1",
         }
         if self._client_name is not None:
             headers["X-Client-Name"] = self._client_name