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

anthropics · Mar 28, 2024 · 0d694e1 · 0d694e1
1 parent 30e8031
commit 0d694e1
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 8 deletions.
diff --git a/src/anthropic/__init__.py b/src/anthropic/__init__.py
@@ -17,7 +17,13 @@
 from ._models import BaseModel
 from ._version import __title__, __version__
 from ._response import APIResponse as APIResponse, AsyncAPIResponse as AsyncAPIResponse
-from ._constants import AI_PROMPT as AI_PROMPT, HUMAN_PROMPT as HUMAN_PROMPT
+from ._constants import (
+    AI_PROMPT as AI_PROMPT,
+    HUMAN_PROMPT as HUMAN_PROMPT,
+    DEFAULT_TIMEOUT,
+    DEFAULT_MAX_RETRIES,
+    DEFAULT_CONNECTION_LIMITS,
+)
 from ._exceptions import (
     APIError,
     ConflictError,
@@ -69,6 +75,9 @@
     "AsyncAnthropic",
     "file_from_path",
     "BaseModel",
+    "DEFAULT_TIMEOUT",
+    "DEFAULT_MAX_RETRIES",
+    "DEFAULT_CONNECTION_LIMITS",
     "HUMAN_PROMPT",
     "AI_PROMPT",
 ]

diff --git a/src/anthropic/_base_client.py b/src/anthropic/_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(

diff --git a/src/anthropic/_client.py b/src/anthropic/_client.py
@@ -34,8 +34,8 @@
     async_get_tokenizer,
 )
 from ._base_client import (
-    DEFAULT_LIMITS,
     DEFAULT_MAX_RETRIES,
+    DEFAULT_CONNECTION_LIMITS,
     SyncAPIClient,
     AsyncAPIClient,
     SyncHttpxClientWrapper,
@@ -239,7 +239,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
@@ -499,7 +499,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

diff --git a/src/anthropic/_constants.py b/src/anthropic/_constants.py
@@ -8,7 +8,7 @@
 # default timeout is 10 minutes
 DEFAULT_TIMEOUT = httpx.Timeout(timeout=600.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