feat(package): export default constants (#334)

Author: stainless-app[bot]

src/finch/__init__.py

@@ -7,6 +7,7 @@
 from ._models import BaseModel
 from ._version import __title__, __version__
 from ._response import APIResponse as APIResponse, AsyncAPIResponse as AsyncAPIResponse
+from ._constants import DEFAULT_TIMEOUT, DEFAULT_MAX_RETRIES, DEFAULT_CONNECTION_LIMITS
 from ._exceptions import (
     APIError,
     FinchError,
@@ -58,6 +59,9 @@
     "AsyncFinch",
     "file_from_path",
     "BaseModel",
+    "DEFAULT_TIMEOUT",
+    "DEFAULT_MAX_RETRIES",
+    "DEFAULT_CONNECTION_LIMITS",
 ]
 
 _setup_logging()

src/finch/_base_client.py

@@ -71,13 +71,13 @@
     extract_response_type,
 )
 from ._constants import (
-    DEFAULT_LIMITS,
     DEFAULT_TIMEOUT,
     MAX_RETRY_DELAY,
     DEFAULT_MAX_RETRIES,
     INITIAL_RETRY_DELAY,
     RAW_RESPONSE_HEADER,
     OVERRIDE_CAST_TO_HEADER,
+    DEFAULT_CONNECTION_LIMITS,
 )
 from ._streaming import Stream, SSEDecoder, AsyncStream, SSEBytesDecoder
 from ._exceptions import (
@@ -747,7 +747,7 @@ def __init__(
             if http_client is not None:
                 raise ValueError("The `http_client` argument is mutually exclusive with `connection_pool_limits`")
         else:
-            limits = DEFAULT_LIMITS
+            limits = DEFAULT_CONNECTION_LIMITS
 
         if transport is not None:
             warnings.warn(
@@ -1294,7 +1294,7 @@ def __init__(
             if http_client is not None:
                 raise ValueError("The `http_client` argument is mutually exclusive with `connection_pool_limits`")
         else:
-            limits = DEFAULT_LIMITS
+            limits = DEFAULT_CONNECTION_LIMITS
 
         if transport is not None:
             warnings.warn(

src/finch/_client.py

@@ -30,8 +30,8 @@
 from ._streaming import Stream as Stream, AsyncStream as AsyncStream
 from ._exceptions import APIStatusError
 from ._base_client import (
-    DEFAULT_LIMITS,
     DEFAULT_MAX_RETRIES,
+    DEFAULT_CONNECTION_LIMITS,
     SyncAPIClient,
     AsyncAPIClient,
     SyncHttpxClientWrapper,
@@ -273,7 +273,7 @@ def copy(
 
             http_client = None
         else:
-            if self._limits is not DEFAULT_LIMITS:
+            if self._limits is not DEFAULT_CONNECTION_LIMITS:
                 connection_pool_limits = self._limits
             else:
                 connection_pool_limits = None
@@ -621,7 +621,7 @@ def copy(
 
             http_client = None
         else:
-            if self._limits is not DEFAULT_LIMITS:
+            if self._limits is not DEFAULT_CONNECTION_LIMITS:
                 connection_pool_limits = self._limits
             else:
                 connection_pool_limits = None

src/finch/_constants.py

@@ -8,7 +8,7 @@
 # default timeout is 1 minute
 DEFAULT_TIMEOUT = httpx.Timeout(timeout=60.0, connect=5.0)
 DEFAULT_MAX_RETRIES = 2
-DEFAULT_LIMITS = httpx.Limits(max_connections=100, max_keepalive_connections=20)
+DEFAULT_CONNECTION_LIMITS = httpx.Limits(max_connections=100, max_keepalive_connections=20)
 
 INITIAL_RETRY_DELAY = 0.5
 MAX_RETRY_DELAY = 8.0