fix(sockets): handle non-portable socket flags (#935)

* Not all socket constants are available on every platform

* Match the style of base code

* Add the change of the socket constant to async client

* Fix lint errors

* Fix formatting

---------

Co-authored-by: Kumar Saurabh <saurabh.kumar@aparavi.com>
Co-authored-by: Robert Craigie <robert@craigie.dev>

Author: 3 people

src/anthropic/_base_client.py

@@ -17,7 +17,9 @@
     TYPE_CHECKING,
     Any,
     Dict,
+    List,
     Type,
+    Tuple,
     Union,
     Generic,
     Mapping,
@@ -806,11 +808,20 @@ def __init__(self, **kwargs: Any) -> None:
         kwargs.setdefault("follow_redirects", True)
 
         if "transport" not in kwargs:
-            socket_options = [
-                (socket.SOL_SOCKET, socket.SO_KEEPALIVE, True),
-                (socket.IPPROTO_TCP, socket.TCP_KEEPINTVL, 60),
-                (socket.IPPROTO_TCP, socket.TCP_KEEPCNT, 5),
-            ]
+            socket_options: List[Tuple[int, int, Union[int, bool]]] = [(socket.SOL_SOCKET, socket.SO_KEEPALIVE, True)]
+
+            TCP_KEEPINTVL = getattr(socket, "TCP_KEEPINTVL", None)
+
+            if TCP_KEEPINTVL is not None:
+                socket_options.append((socket.IPPROTO_TCP, TCP_KEEPINTVL, 60))
+            elif sys.platform == "darwin":
+                TCP_KEEPALIVE = getattr(socket, "TCP_KEEPALIVE", 0x10)
+                socket_options.append((socket.IPPROTO_TCP, TCP_KEEPALIVE, 60))
+
+            TCP_KEEPCNT = getattr(socket, "TCP_KEEPCNT", None)
+            if TCP_KEEPCNT is not None:
+                socket_options.append((socket.IPPROTO_TCP, TCP_KEEPCNT, 5))
+
             TCP_KEEPIDLE = getattr(socket, "TCP_KEEPIDLE", None)
             if TCP_KEEPIDLE is not None:
                 socket_options.append((socket.IPPROTO_TCP, TCP_KEEPIDLE, 60))
@@ -1335,11 +1346,20 @@ def __init__(self, **kwargs: Any) -> None:
         kwargs.setdefault("follow_redirects", True)
 
         if "transport" not in kwargs:
-            socket_options = [
-                (socket.SOL_SOCKET, socket.SO_KEEPALIVE, True),
-                (socket.IPPROTO_TCP, socket.TCP_KEEPINTVL, 60),
-                (socket.IPPROTO_TCP, socket.TCP_KEEPCNT, 5),
-            ]
+            socket_options: List[Tuple[int, int, Union[int, bool]]] = [(socket.SOL_SOCKET, socket.SO_KEEPALIVE, True)]
+
+            TCP_KEEPINTVL = getattr(socket, "TCP_KEEPINTVL", None)
+
+            if TCP_KEEPINTVL is not None:
+                socket_options.append((socket.IPPROTO_TCP, TCP_KEEPINTVL, 60))
+            elif sys.platform == "darwin":
+                TCP_KEEPALIVE = getattr(socket, "TCP_KEEPALIVE", 0x10)
+                socket_options.append((socket.IPPROTO_TCP, TCP_KEEPALIVE, 60))
+
+            TCP_KEEPCNT = getattr(socket, "TCP_KEEPCNT", None)
+            if TCP_KEEPCNT is not None:
+                socket_options.append((socket.IPPROTO_TCP, TCP_KEEPCNT, 5))
+
             TCP_KEEPIDLE = getattr(socket, "TCP_KEEPIDLE", None)
             if TCP_KEEPIDLE is not None:
                 socket_options.append((socket.IPPROTO_TCP, TCP_KEEPIDLE, 60))