ragaai-catalyst 2.2.4.1b8__tar.gz → 2.2.4.1b9__tar.gz

{ragaai_catalyst-2.2.4.1b8 → ragaai_catalyst-2.2.4.1b9}/.gitignore

@@ -166,80 +166,4 @@ cython_debug/
 #  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
 #  and can be added to the global gitignore or merged into this file.  For a more nuclear
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
-#.idea/
-
-LOAD_TEST_README.md
-RAG_LOAD_TEST_README.md
-advanced_agent.py
-ai_document_061023_2.pdf
-catalyst_v2.ipynb
-chroma_db/
-clean_multi_dataset_example.py
-custom_cost_bug.ipynb
-custom_cost_bug.py
-data/
-deephealth_sample_50_data.csv
-dh_full_test.ipynb
-dh_full_test.py
-diagnosis_agent.py
-direct_anthropic.json
-direct_llm_call_test.py
-direct_llm_examples.py
-direct_opanai.json
-docker_gt_dataset1.csv
-ex_google_adk.py
-fastapi_requirements.txt
-fastapi_server.py
-fastapi_server_simple.py
-final_dataset_routing_demo.py
-google_adk_agent_ex.py
-google_adk_agent_user_data.py
-google_adk_storyflow_agent.py
-langchain_rag.py
-langchain_sample_trace.json
-legal_rag.py
-llamaindex_rag.py
-load_test_streaming.py
-load_test_v1.py
-load_test_v2.py
-locust_mini_endpoint.py
-locust_nano_endpoint.py
-locustfile.py
-locustfile_fixed.py
-locustfile_rag.py
-masking_example.py
-minio-file.json
-model_gemini_15_flash_002_w_stream.json
-model_gemini_15_flash_002_wo_stream.json
-model_gpt_41_w_stream.json
-model_gpt_41_wo_stream.json
-model_gpt_4o_mini_w_stream.json
-model_gpt_4o_mini_wo_stream.json
-model_prices_and_context_window_backup.json
-multi_llm_test.py
-multi_tool_agent/
-multi_tracer_test.py
-new_feature_test.ipynb
-pydantic_test.ipynb
-rag_agent_traces.json
-rag_agent_traces_adk_agent.json
-rag_agent_traces_with_streaming.json
-rag_agent_traces_wo_streaming.json
-rag_extraction_logic_final.py
-ragaai_catalyst/tracers/processors/__init__.py
-research_assistant.py
-run_rag_load_test.py
-sci_fi_story.md
-scifi_writer.py
-set_dataset_name_test.py
-simple_rag_app.py
-stress_test_langchain.py
-test.ipynb
-test_2.py
-test_dataset_routing.py
-test_fastapi.py
-test_simple_langchain.py
-test_token_refresh.py
-trace_upload_analysis.md
-updated_pydantic_test.py
-uplaod_numerical_col_csv.py
+#.idea/

{ragaai_catalyst-2.2.4.1b8/ragaai_catalyst.egg-info → ragaai_catalyst-2.2.4.1b9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ragaai_catalyst
-Version: 2.2.4.1b8
+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.1b8/quickstart.md → ragaai_catalyst-2.2.4.1b9/Quickstart.md

@@ -1,4 +1,4 @@
-# Quickstart
+# Quickstart | RagaAI Catalyst
 
 ## **1. Install RagaAI Catalyst**
 
@@ -17,8 +17,6 @@ pip install ragaai-catalyst
 2. Navigate to **Profile Settings** → **Authentication**.
 3. Click **Generate New Key** to obtain your **Access Key** and **Secret Key**.
 
-![How to generate authentication keys](docs/img/autheticate.gif)
-
 ### **Initialize the SDK**
 
 To begin using Catalyst, initialize it as follows:
@@ -33,6 +31,7 @@ catalyst = RagaAICatalyst(
 )
 ```
 
+
 ## **3. Create Your First Project**
 
 Create a new project and choose a use case from the available options:
@@ -47,8 +46,6 @@ project = catalyst.create_project(
 # List available use cases
 print(catalyst.project_use_cases())
 ```
-![Projects](docs/img/create_project.gif)
-
 
 ### **Add a Dataset**
 Initialize the dataset manager and create a dataset from a CSV file, DataFrame, or JSONl file.
@@ -74,7 +71,7 @@ dataset_manager.create_from_csv(
 # View dataset schema
 print(dataset_manager.get_schema_mapping())
 ```
-![Dataset](docs/img/dataset.gif)
+
 
 ## **4. Trace Your Application**
 
@@ -147,7 +144,7 @@ tracer.stop()
 # Verify data capture
 print(tracer.get_upload_status())
 ```
-![Tracing](docs/img/last_main.png)
+
 
 
 ## **5. Evaluation Framework**
@@ -192,4 +189,12 @@ evaluation.add_metrics(
 print(f"Status: {evaluation.get_status()}")
 print(f"Results: {evaluation.get_results()}")
 ```
-![Evaluation](docs/img/evaluation.gif)
+
+
+
+## **Next Steps**
+- **Explore the Dashboard:** Visualize metrics and insights in the RagaAI Web UI.
+
+
+
+**Version:** 1.0.0 | **Last Updated:** Mar 2025

{ragaai_catalyst-2.2.4.1b8 → ragaai_catalyst-2.2.4.1b9}/pyproject.toml

@@ -8,7 +8,7 @@ description = "RAGA AI CATALYST"
 readme = "README.md"
 requires-python = ">=3.10,<=3.13.2"
 # license = {file = "LICENSE"}
-version = "2.2.4.1.beta.8"
+version = "2.2.4.1.beta.9"
 authors = [
     {name = "Kiran Scaria", email = "kiran.scaria@raga.ai"},
     {name = "Kedar Gaikwad", email = "kedar.gaikwad@raga.ai"},

{ragaai_catalyst-2.2.4.1b8 → ragaai_catalyst-2.2.4.1b9}/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.warn(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-2.2.4.1b8 → ragaai_catalyst-2.2.4.1b9}/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-2.2.4.1b8 → ragaai_catalyst-2.2.4.1b9}/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.1b8 → ragaai_catalyst-2.2.4.1b9/ragaai_catalyst.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ragaai_catalyst
-Version: 2.2.4.1b8
+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.1b8 → ragaai_catalyst-2.2.4.1b9}/ragaai_catalyst.egg-info/SOURCES.txt

@@ -1,9 +1,9 @@
 .gitignore
 .gitmodules
 LICENSE
+Quickstart.md
 README.md
 pyproject.toml
-quickstart.md
 requirements.txt
 test_report_20250407_183101.txt
 tests_requirements.txt