ragaai-catalyst 2.2.4.1b7__py3-none-any.whl → 2.2.4.1b9__py3-none-any.whl

ragaai_catalyst/tracers/agentic_tracing/upload/session_manager.py

@@ -6,6 +6,7 @@ from requests.adapters import HTTPAdapter
 from urllib3.util.retry import Retry
 from urllib3.exceptions import PoolError, MaxRetryError, NewConnectionError
 from requests.exceptions import ConnectionError, Timeout
+from http.client import RemoteDisconnected
 from ragaai_catalyst import RagaAICatalyst
 import requests
 
@@ -92,11 +93,11 @@ class SessionManager:
         for i in range(num_connections):
             try:
                 # Make a lightweight HEAD request to the healthcheck endpoint to warm up the connection
-                response = self._session.head(healthcheck_url, timeout=5)
+                response = self._session.head(healthcheck_url, timeout=10)
                 logger.info(f"Warmup connection {i+1}: Status {response.status_code}")
             except Exception as e:
-                logger.error(f"Warmup connection {i+1} failed (this is normal): {e}")
-                # Ignore failures during warmup as they're expected
+                logger.warning(f"Warmup connection {i+1} failed (this may be normal): {e}")
+                # Ignore other failures during warmup as they're expected
                 continue
 
         logger.info("Connection pool warmup completed")
@@ -118,6 +119,8 @@ class SessionManager:
             logger.error(f"Connection pool exhausted during {operation_name}: {e}")
         elif isinstance(e, NewConnectionError):
             logger.error(f"Failed to establish new connection during {operation_name}: {e}")
+        elif isinstance(e, RemoteDisconnected):
+            logger.error(f"Remote connection closed unexpectedly during {operation_name}: {e}")
         elif isinstance(e, ConnectionError):
             logger.error(f"Connection error during {operation_name}: {e}")
         elif isinstance(e, Timeout):
@@ -125,6 +128,27 @@ class SessionManager:
         else:
             logger.error(f"Unexpected error during {operation_name}: {e}")
 
+    def make_request_with_retry(self, method, url, **kwargs):
+        """
+        Make HTTP request with additional retry logic for RemoteDisconnected errors
+        that may not be caught by urllib3's retry mechanism.
+        """
+        max_retries = 3
+        for attempt in range(max_retries):
+            try:
+                response = self._session.request(method, url, **kwargs)
+                return response
+            except (RemoteDisconnected, ConnectionError) as e:
+                logger.warning(f"Connection error on attempt {attempt + 1}/{max_retries}: {e}")
+                if attempt == max_retries - 1:
+                    # Re-raise the exception on the last attempt
+                    raise
+                # Wait before retrying (exponential backoff)
+                import time
+                wait_time = 2 ** attempt
+                logger.info(f"Retrying in {wait_time} seconds...")
+                time.sleep(wait_time)
+
 
 # Global session manager instance
 logger.info("Creating global SessionManager instance")

ragaai_catalyst/tracers/agentic_tracing/upload/upload_agentic_traces.py

@@ -6,6 +6,7 @@ import time
 from urllib.parse import urlparse, urlunparse
 from urllib3.exceptions import PoolError, MaxRetryError, NewConnectionError
 from requests.exceptions import ConnectionError, Timeout, RequestException
+from http.client import RemoteDisconnected
 from .session_manager import session_manager
 
 import requests
@@ -51,7 +52,7 @@ class UploadAgenticTraces:
             start_time = time.time()
             endpoint = f"{self.base_url}/v1/llm/presigned-url"
             # Changed to POST from GET
-            response = session_manager.session.request(
+            response = session_manager.make_request_with_retry(
                 "POST", endpoint, headers=headers, data=payload, timeout=self.timeout
             )
             elapsed_ms = (time.time() - start_time) * 1000
@@ -65,7 +66,7 @@ class UploadAgenticTraces:
                 return presignedurl
             else:
                 # If POST fails, try GET
-                response = session_manager.session.request(
+                response = session_manager.make_request_with_retry(
                     "GET", endpoint, headers=headers, data=payload, timeout=self.timeout
                 )
                 elapsed_ms = (time.time() - start_time) * 1000
@@ -86,7 +87,7 @@ class UploadAgenticTraces:
                         "Authorization": f"Bearer {token}",
                         "X-Project-Name": self.project_name,
                     }
-                    response = session_manager.session.request(
+                    response = session_manager.make_request_with_retry(
                         "POST",
                         endpoint,
                         headers=headers,
@@ -113,7 +114,7 @@ class UploadAgenticTraces:
                         f"Error while getting presigned url: {response.json()['message']}"
                     )
                     return None
-        except (PoolError, MaxRetryError, NewConnectionError, ConnectionError, Timeout) as e:
+        except (PoolError, MaxRetryError, NewConnectionError, ConnectionError, Timeout, RemoteDisconnected) as e:
             session_manager.handle_request_exceptions(e, "getting presigned URL")
             return None
         except RequestException as e:
@@ -152,7 +153,7 @@ class UploadAgenticTraces:
             return False
         try:
             start_time = time.time()
-            response = session_manager.session.request(
+            response = session_manager.make_request_with_retry(
                 "PUT", presignedUrl, headers=headers, data=payload, timeout=self.timeout
             )
             elapsed_ms = (time.time() - start_time) * 1000
@@ -162,7 +163,7 @@ class UploadAgenticTraces:
             if response.status_code != 200 or response.status_code != 201:
                 return response, response.status_code
             return True
-        except (PoolError, MaxRetryError, NewConnectionError, ConnectionError, Timeout) as e:
+        except (PoolError, MaxRetryError, NewConnectionError, ConnectionError, Timeout, RemoteDisconnected) as e:
             session_manager.handle_request_exceptions(e, "uploading trace to presigned URL")
             return False
         except RequestException as e:
@@ -185,7 +186,7 @@ class UploadAgenticTraces:
         try:
             start_time = time.time()
             endpoint = f"{self.base_url}/v1/llm/insert/trace"
-            response = session_manager.session.request(
+            response = session_manager.make_request_with_retry(
                 "POST", endpoint, headers=headers, data=payload, timeout=self.timeout
             )
             elapsed_ms = (time.time() - start_time) * 1000
@@ -203,7 +204,7 @@ class UploadAgenticTraces:
                     "Content-Type": "application/json",
                     "X-Project-Name": self.project_name,
                 }
-                response = session_manager.session.request(
+                response = session_manager.make_request_with_retry(
                     "POST",
                     endpoint,
                     headers=headers,
@@ -223,7 +224,7 @@ class UploadAgenticTraces:
             else:
                 logger.error(f"Error while inserting traces: {response.json()['message']}")
                 return False
-        except (PoolError, MaxRetryError, NewConnectionError, ConnectionError, Timeout) as e:
+        except (PoolError, MaxRetryError, NewConnectionError, ConnectionError, Timeout, RemoteDisconnected) as e:
             session_manager.handle_request_exceptions(e, "inserting traces")
             return False
         except RequestException as e:

ragaai_catalyst/tracers/agentic_tracing/upload/upload_code.py

@@ -6,6 +6,7 @@ import json
 from urllib.parse import urlparse, urlunparse
 from urllib3.exceptions import PoolError, MaxRetryError, NewConnectionError
 from requests.exceptions import ConnectionError, Timeout, RequestException
+from http.client import RemoteDisconnected
 
 import requests
 
@@ -71,7 +72,7 @@ def _fetch_dataset_code_hashes(project_name, dataset_name, base_url=None, timeou
         url_base = base_url if base_url is not None else RagaAICatalyst.BASE_URL
         start_time = time.time()
         endpoint = f"{url_base}/v2/llm/dataset/code?datasetName={dataset_name}"
-        response = session_manager.session.request(
+        response = session_manager.make_request_with_retry(
             "GET", endpoint, headers=headers, data=payload, timeout=timeout
         )
         elapsed_ms = (time.time() - start_time) * 1000
@@ -88,7 +89,7 @@ def _fetch_dataset_code_hashes(project_name, dataset_name, base_url=None, timeou
                 "Authorization": f"Bearer {os.getenv('RAGAAI_CATALYST_TOKEN')}",
                 "X-Project-Name": project_name,
             }
-            response = session_manager.session.request(
+            response = session_manager.make_request_with_retry(
                 "GET", endpoint, headers=headers, data=payload, timeout=timeout
             )
             elapsed_ms = (time.time() - start_time) * 1000
@@ -101,7 +102,7 @@ def _fetch_dataset_code_hashes(project_name, dataset_name, base_url=None, timeou
         else:
             logger.error(f"Error while fetching dataset code hashes: {response.json()['message']}")
             return None
-    except (PoolError, MaxRetryError, NewConnectionError, ConnectionError, Timeout) as e:
+    except (PoolError, MaxRetryError, NewConnectionError, ConnectionError, Timeout, RemoteDisconnected) as e:
         session_manager.handle_request_exceptions(e, "fetching dataset code hashes")
         return None
     except RequestException as e:
@@ -213,7 +214,7 @@ def _put_zip_presigned_url(project_name, presignedUrl, filename, timeout=120):
             payload = f.read()
 
         start_time = time.time()
-        response = session_manager.session.request(
+        response = session_manager.make_request_with_retry(
             "PUT", presignedUrl, headers=headers, data=payload, timeout=timeout
         )
         elapsed_ms = (time.time() - start_time) * 1000
@@ -223,7 +224,7 @@ def _put_zip_presigned_url(project_name, presignedUrl, filename, timeout=120):
         if response.status_code not in [200, 201]:
             return response, response.status_code
         return True
-    except (PoolError, MaxRetryError, NewConnectionError, ConnectionError, Timeout) as e:
+    except (PoolError, MaxRetryError, NewConnectionError, ConnectionError, Timeout, RemoteDisconnected) as e:
         session_manager.handle_request_exceptions(e, "uploading zip to presigned URL")
         return False
     except RequestException as e:
@@ -251,7 +252,7 @@ def _insert_code(
         url_base = base_url if base_url is not None else RagaAICatalyst.BASE_URL
         start_time = time.time()
         endpoint = f"{url_base}/v2/llm/dataset/code"
-        response = session_manager.session.request(
+        response = session_manager.make_request_with_retry(
             "POST", endpoint, headers=headers, data=payload, timeout=timeout
         )
         elapsed_ms = (time.time() - start_time) * 1000
@@ -269,7 +270,7 @@ def _insert_code(
                 "Content-Type": "application/json",
                 "Authorization": f"Bearer {token}",
             }
-            response = session_manager.session.request(
+            response = session_manager.make_request_with_retry(
                 "POST", endpoint, headers=headers, data=payload, timeout=timeout
             )
             elapsed_ms = (time.time() - start_time) * 1000
@@ -285,7 +286,7 @@ def _insert_code(
         else:
             logger.error(f"Failed to insert code: {response.json()['message']}")
             return None
-    except (PoolError, MaxRetryError, NewConnectionError, ConnectionError, Timeout) as e:
+    except (PoolError, MaxRetryError, NewConnectionError, ConnectionError, Timeout, RemoteDisconnected) as e:
         session_manager.handle_request_exceptions(e, "inserting code")
         return None
     except RequestException as e:

{ragaai_catalyst-2.2.4.1b7.dist-info → ragaai_catalyst-2.2.4.1b9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ragaai_catalyst
-Version: 2.2.4.1b7
+Version: 2.2.4.1b9
 Summary: RAGA AI CATALYST
 Author-email: Kiran Scaria <kiran.scaria@raga.ai>, Kedar Gaikwad <kedar.gaikwad@raga.ai>, Dushyant Mahajan <dushyant.mahajan@raga.ai>, Siddhartha Kosti <siddhartha.kosti@raga.ai>, Ritika Goel <ritika.goel@raga.ai>, Vijay Chaurasia <vijay.chaurasia@raga.ai>, Tushar Kumar <tushar.kumar@raga.ai>, Rishabh Pandey <rishabh.pandey@raga.ai>, Jyotsana C G <jyotsana@raga.ai>
 Requires-Python: <=3.13.2,>=3.10

{ragaai_catalyst-2.2.4.1b7.dist-info → ragaai_catalyst-2.2.4.1b9.dist-info}/RECORD

@@ -54,10 +54,10 @@ ragaai_catalyst/tracers/agentic_tracing/tracers/network_tracer.py,sha256=m8CxYkl
 ragaai_catalyst/tracers/agentic_tracing/tracers/tool_tracer.py,sha256=xxrliKPfdfbIZRZqMnUewsaTD8_Hv0dbuoBivNZGD4U,21674
 ragaai_catalyst/tracers/agentic_tracing/tracers/user_interaction_tracer.py,sha256=bhSUhNQCuJXKjgJAXhjKEYjnHMpYN90FSZdR84fNIKU,4614
 ragaai_catalyst/tracers/agentic_tracing/upload/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ragaai_catalyst/tracers/agentic_tracing/upload/session_manager.py,sha256=QYHebplvrLSqrM6dUmsWLlaQZVRhCQFlTaPAN1QBAt4,5671
+ragaai_catalyst/tracers/agentic_tracing/upload/session_manager.py,sha256=sOlxeIYIP8tycaTtZC9xkZosi6EDJUxvDw0_rc_NLI8,6823
 ragaai_catalyst/tracers/agentic_tracing/upload/trace_uploader.py,sha256=Ujbu0KDl7oDr-cFtLwrQK_i7ghMuPV92mFnRfobJ1aI,24822
-ragaai_catalyst/tracers/agentic_tracing/upload/upload_agentic_traces.py,sha256=vH9at3012iNrNIA30TMr7qLyvWtKyZn9wpd5esmBg0A,12866
-ragaai_catalyst/tracers/agentic_tracing/upload/upload_code.py,sha256=bpzL-eNCUaqXimSg7nl5YB0KlHbS_2sNvJTfKnfRjek,12027
+ragaai_catalyst/tracers/agentic_tracing/upload/upload_agentic_traces.py,sha256=0u4GWgqtaBz9cnr_KuqVIWDvhHWkgTAOTtiy0w8RPuk,13017
+ragaai_catalyst/tracers/agentic_tracing/upload/upload_code.py,sha256=CbTx2vBAPIat5bdIClv9szOo4i33YL_1v04mkUjNG2c,12170
 ragaai_catalyst/tracers/agentic_tracing/upload/upload_local_metric.py,sha256=m1O8lKpxKwtHofXLW3fTHX5yfqDW5GxoveARlg5cTw4,2571
 ragaai_catalyst/tracers/agentic_tracing/utils/__init__.py,sha256=XdB3X_ufe4RVvGorxSqAiB9dYv4UD7Hvvuw3bsDUppY,60
 ragaai_catalyst/tracers/agentic_tracing/utils/api_utils.py,sha256=ZduFA7MmTnWfQ2FzSD0hxMAAfNNTgBs4CXcHZdXJv6k,749
@@ -89,8 +89,8 @@ ragaai_catalyst/tracers/utils/rag_extraction_logic_final.py,sha256=3ygkRT__lLDRf
 ragaai_catalyst/tracers/utils/rag_trace_json_converter.py,sha256=54IEZO-YRjUAahV5nw8KClXqTF1LhfDry_TsZ4KGow4,20467
 ragaai_catalyst/tracers/utils/trace_json_converter.py,sha256=-HZVmijeUFLO7e9OAvi1RJdWVTxPRUHPd1MkKQlCD54,11785
 ragaai_catalyst/tracers/utils/utils.py,sha256=o-p9n2ZuophdrV0wrixu-BqRHCkovup_klc3mS8mU8g,2374
-ragaai_catalyst-2.2.4.1b7.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-ragaai_catalyst-2.2.4.1b7.dist-info/METADATA,sha256=d-uuQOt-nxBw7Q4x2EzeM7a1HPmTieueet0-ZRsP40I,17681
-ragaai_catalyst-2.2.4.1b7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ragaai_catalyst-2.2.4.1b7.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
-ragaai_catalyst-2.2.4.1b7.dist-info/RECORD,,
+ragaai_catalyst-2.2.4.1b9.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+ragaai_catalyst-2.2.4.1b9.dist-info/METADATA,sha256=ox1pbQ-3DNbMms9uReLZjUKdze0ztGZ2SeDehc1TROI,17681
+ragaai_catalyst-2.2.4.1b9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ragaai_catalyst-2.2.4.1b9.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
+ragaai_catalyst-2.2.4.1b9.dist-info/RECORD,,