fix type mismatch between raw_client and combined_raw_client

Author: jasonozuzu-cohere

src/cohere/client_v2.py

@@ -1,11 +1,12 @@
-from .client import Client, AsyncClient
-from .v2.client import V2Client, AsyncV2Client
-import typing
-from .environment import ClientEnvironment
 import os
-import httpx
+import typing
 from concurrent.futures import ThreadPoolExecutor
 
+import httpx
+from .client import AsyncClient, Client
+from .environment import ClientEnvironment
+from .v2.client import AsyncRawV2Client, AsyncV2Client, RawV2Client, V2Client
+
 
 class _CombinedRawClient:
     """Proxy that combines v1 and v2 raw clients.
@@ -57,7 +58,7 @@ def __init__(
             self,
             client_wrapper=self._client_wrapper
         )
-        self._raw_client = _CombinedRawClient(v1_raw, self._raw_client)
+        self._raw_client = typing.cast(RawV2Client, _CombinedRawClient(v1_raw, self._raw_client))
 
 
 class AsyncClientV2(AsyncV2Client, AsyncClient):  # type: ignore
@@ -90,4 +91,4 @@ def __init__(
             self,
             client_wrapper=self._client_wrapper
         )
-        self._raw_client = _CombinedRawClient(v1_raw, self._raw_client)
+        self._raw_client = typing.cast(AsyncRawV2Client, _CombinedRawClient(v1_raw, self._raw_client))