add new proxy for the raw_client param

Author: Thomas Baker

src/cohere/client_v2.py

@@ -7,6 +7,26 @@
 from concurrent.futures import ThreadPoolExecutor
 
 
+class _CombinedRawClient:
+    """Proxy that combines v1 and v2 raw clients.
+
+    V2Client and Client both assign to self._raw_client in __init__,
+    causing a collision when combined in ClientV2/AsyncClientV2.
+    This proxy delegates to v2 first, falling back to v1 for
+    legacy methods like generate_stream.
+    """
+
+    def __init__(self, v1_raw_client: typing.Any, v2_raw_client: typing.Any):
+        self._v1 = v1_raw_client
+        self._v2 = v2_raw_client
+
+    def __getattr__(self, name: str) -> typing.Any:
+        try:
+            return getattr(self._v2, name)
+        except AttributeError:
+            return getattr(self._v1, name)
+
+
 class ClientV2(V2Client, Client):  # type: ignore
     def __init__(
         self,
@@ -32,10 +52,12 @@ def __init__(
             thread_pool_executor=thread_pool_executor,
             log_warning_experimental_features=log_warning_experimental_features,
         )
+        v1_raw = self._raw_client
         V2Client.__init__(
             self,
             client_wrapper=self._client_wrapper
         )
+        self._raw_client = _CombinedRawClient(v1_raw, self._raw_client)
 
 
 class AsyncClientV2(AsyncV2Client, AsyncClient):  # type: ignore
@@ -63,7 +85,9 @@ def __init__(
             thread_pool_executor=thread_pool_executor,
             log_warning_experimental_features=log_warning_experimental_features,
         )
+        v1_raw = self._raw_client
         AsyncV2Client.__init__(
             self,
             client_wrapper=self._client_wrapper
         )
+        self._raw_client = _CombinedRawClient(v1_raw, self._raw_client)