Add missing files

Author: jhamon

pinecone/openapi_support/rest_aiohttp.py

@@ -30,15 +30,18 @@ def __init__(self, configuration: Configuration) -> None:
         else:
             self._session = aiohttp.ClientSession(connector=conn)
 
-        jitter_retry = JitterRetry(
-            attempts=5,
-            start_timeout=0.1,
-            max_timeout=3.0,
-            statuses={500, 502, 503, 504},
-            methods=None,  # retry on all methods
-            exceptions={aiohttp.ClientError, aiohttp.ServerDisconnectedError},
-        )
-        self._retry_client = RetryClient(client_session=self._session, retry_options=jitter_retry)
+        if configuration.retries is not None:
+            retry_options = configuration.retries
+        else:
+            retry_options = JitterRetry(
+                attempts=5,
+                start_timeout=0.1,
+                max_timeout=3.0,
+                statuses={500, 502, 503, 504},
+                methods=None,  # retry on all methods
+                exceptions={aiohttp.ClientError, aiohttp.ServerDisconnectedError},
+            )
+        self._retry_client = RetryClient(client_session=self._session, retry_options=retry_options)
 
     async def close(self):
         await self._retry_client.close()

pinecone/openapi_support/retry_aiohttp.py

@@ -0,0 +1,46 @@
+import random
+from typing import Optional
+from aiohttp_retry import RetryOptionsBase, EvaluateResponseCallbackType, ClientResponse
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+class JitterRetry(RetryOptionsBase):
+    """https://github.com/inyutin/aiohttp_retry/issues/44."""
+
+    def __init__(
+        self,
+        attempts: Optional[int] = 3,  # How many times we should retry
+        start_timeout: Optional[float] = 0.1,  # Base timeout time, then it exponentially grow
+        max_timeout: Optional[float] = 5.0,  # Max possible timeout between tries
+        statuses: Optional[set[int]] = None,  # On which statuses we should retry
+        exceptions: Optional[set[type[Exception]]] = None,  # On which exceptions we should retry
+        methods: Optional[set[str]] = None,  # On which HTTP methods we should retry
+        random_interval_size: Optional[float] = 2.0,  # size of interval for random component
+        retry_all_server_errors: bool = True,
+        evaluate_response_callback: Optional[EvaluateResponseCallbackType] = None,
+    ) -> None:
+        super().__init__(
+            attempts=attempts,
+            statuses=statuses,
+            exceptions=exceptions,
+            methods=methods,
+            retry_all_server_errors=retry_all_server_errors,
+            evaluate_response_callback=evaluate_response_callback,
+        )
+
+        self._start_timeout: float = start_timeout
+        self._max_timeout: float = max_timeout
+        self._random_interval_size = random_interval_size
+
+    def get_timeout(
+        self,
+        attempt: int,
+        response: Optional[ClientResponse] = None,  # noqa: ARG002
+    ) -> float:
+        logger.debug(f"JitterRetry get_timeout: attempt={attempt}, response={response}")
+        """Return timeout with exponential backoff."""
+        jitter = random.uniform(0, 0.1)
+        timeout = self._start_timeout * (2 ** (attempt - 1))
+        return min(timeout + jitter, self._max_timeout)

pinecone/openapi_support/retry_urllib3.py

@@ -0,0 +1,21 @@
+import random
+from urllib3.util.retry import Retry
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+class JitterRetry(Retry):
+    """
+    Retry with exponential back‑off with jitter.
+
+    The Retry class is being extended as built-in support for jitter was added only in urllib3 2.0.0.
+    Jitter logic is following the official implementation with a constant jitter factor: https://github.com/urllib3/urllib3/blob/main/src/urllib3/util/retry.py
+    """
+
+    def get_backoff_time(self) -> float:
+        backoff_value = super().get_backoff_time()
+        jitter = random.random() * 0.25
+        backoff_value += jitter
+        logger.debug(f"Calculating retry backoff: {backoff_value} (jitter: {jitter})")
+        return backoff_value

scripts/test-async-retry.py

@@ -0,0 +1,16 @@
+import dotenv
+import asyncio
+import logging
+from pinecone import PineconeAsyncio
+
+dotenv.load_dotenv()
+
+logging.basicConfig(level=logging.DEBUG)
+
+
+async def main():
+    async with PineconeAsyncio(host="http://localhost:8000") as pc:
+        await pc.db.index.list()
+
+
+asyncio.run(main())