ragaai-catalyst 2.2.4.1b10__py3-none-any.whl → 2.2.4.2b2__py3-none-any.whl

ragaai_catalyst/tracers/agentic_tracing/upload/trace_uploader.py

@@ -240,12 +240,14 @@ def process_upload(task_id: str, filepath: str, hash_id: str, zip_path: str,
                     user_details=user_details,
                     timeout=timeout
                 )
-                logger.info(f"Dataset schema created: {response}")
-
-                # Cache the response only if status code is 200
-                if response and hasattr(response, 'status_code') and response.status_code in [200, 201]:
+                if response is None:
+                    logger.error(f"Dataset schema creation failed for {dataset_name} - received None response")
+                elif hasattr(response, 'status_code') and response.status_code in [200, 201]:
+                    logger.info(f"Dataset schema created successfully: {response.status_code}")
                     _cache_dataset_creation(cache_key, response)
                     logger.info(f"Response cached successfully for dataset: {dataset_name} and key: {cache_key}")
+                else:
+                    logger.warning(f"Dataset schema creation returned unexpected response: {response}")
 
             except Exception as e:
                 logger.error(f"Error creating dataset schema: {e}")

ragaai_catalyst/tracers/agentic_tracing/upload/upload_agentic_traces.py

@@ -64,8 +64,33 @@ class UploadAgenticTraces:
                 presignedURLs = response.json()["data"]["presignedUrls"][0]
                 presignedurl = self.update_presigned_url(presignedURLs, self.base_url)
                 return presignedurl
+            elif response.status_code == 401:
+                logger.warning("Received 401 error. Attempting to refresh token.")
+                token = RagaAICatalyst.get_token(force_refresh=True)
+                headers = {
+                    "Content-Type": "application/json",
+                    "Authorization": f"Bearer {token}",
+                    "X-Project-Name": self.project_name,
+                }
+                response = session_manager.make_request_with_retry(
+                    "POST", endpoint, headers=headers, data=payload, timeout=self.timeout
+                )
+                elapsed_ms = (time.time() - start_time) * 1000
+                logger.debug(
+                    f"API Call: [POST] {endpoint} | Status: {response.status_code} | Time: {elapsed_ms:.2f}ms"
+                )
+                if response.status_code == 200:
+                    presignedURLs = response.json()["data"]["presignedUrls"][0]
+                    presignedurl = self.update_presigned_url(presignedURLs, self.base_url)
+                    return presignedurl
+                else:
+                    logger.error(
+                        f"Error while getting presigned url after token refresh: {response.json()['message']}"
+                    )
+                    return None
             else:
                 # If POST fails, try GET
+                logger.warning(f"POST request failed for presign url with status {response.status_code}. Falling back to GET request.")
                 response = session_manager.make_request_with_retry(
                     "GET", endpoint, headers=headers, data=payload, timeout=self.timeout
                 )

ragaai_catalyst/tracers/agentic_tracing/utils/create_dataset_schema.py

@@ -1,8 +1,16 @@
 import os
 import json
 import re
-import requests
-from ragaai_catalyst.tracers.agentic_tracing.tracers.base import RagaAICatalyst
+import logging
+import time
+from urllib3.exceptions import PoolError, MaxRetryError, NewConnectionError
+from requests.exceptions import ConnectionError, Timeout, RequestException
+from http.client import RemoteDisconnected
+
+from ragaai_catalyst import RagaAICatalyst
+from ragaai_catalyst.tracers.agentic_tracing.upload.session_manager import session_manager
+
+logger = logging.getLogger(__name__)
 
 def create_dataset_schema_with_trace(project_name, dataset_name, base_url=None, user_details=None, timeout=120):
     SCHEMA_MAPPING = {}
@@ -13,31 +21,69 @@ def create_dataset_schema_with_trace(project_name, dataset_name, base_url=None,
                 continue
             SCHEMA_MAPPING[key] = {"columnType": "metadata"}
 
-    def make_request():
-        headers = {
-            "Content-Type": "application/json",
-            "Authorization": f"Bearer {os.getenv('RAGAAI_CATALYST_TOKEN')}",
-            "X-Project-Name": project_name,
-        }
-        if SCHEMA_MAPPING:
-            payload = json.dumps({
-                "datasetName": dataset_name,
-                "traceFolderUrl": None,
-                "schemaMapping": SCHEMA_MAPPING
-            })
-        else:
-            payload = json.dumps({
-                "datasetName": dataset_name,
-                "traceFolderUrl": None,
-            })
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {os.getenv('RAGAAI_CATALYST_TOKEN')}",
+        "X-Project-Name": project_name,
+    }
+    logger.info(f"Header for dataset logs api {headers}")
+    if SCHEMA_MAPPING:
+        payload = json.dumps({
+            "datasetName": dataset_name,
+            "traceFolderUrl": None,
+            "schemaMapping": SCHEMA_MAPPING
+        })
+    else:
+        payload = json.dumps({
+            "datasetName": dataset_name,
+            "traceFolderUrl": None,
+        })
+
+    try:
         # Use provided base_url or fall back to default
         url_base = base_url if base_url is not None else RagaAICatalyst.BASE_URL
-        response = requests.request("POST",
-            f"{url_base}/v1/llm/dataset/logs",
-            headers=headers,
-            data=payload,
-            timeout=timeout
+        start_time = time.time()
+        endpoint = f"{url_base}/v1/llm/dataset/logs"
+
+        response = session_manager.make_request_with_retry(
+            "POST", endpoint, headers=headers, data=payload, timeout=timeout
         )
-        return response
-    response = make_request()
-    return response
+
+        elapsed_ms = (time.time() - start_time) * 1000
+        logger.debug(
+            f"API Call: [POST] {endpoint} | Status: {response.status_code} | Time: {elapsed_ms:.2f}ms"
+        )
+
+        if response.status_code in [200, 201]:
+            logger.info(f"Dataset schema created successfully: {response.status_code}")
+            return response
+        elif response.status_code == 401:
+            logger.warning("Received 401 error during dataset schema creation. Attempting to refresh token.")
+            RagaAICatalyst.get_token(force_refresh=True)
+            headers = {
+                "Content-Type": "application/json",
+                "Authorization": f"Bearer {os.getenv('RAGAAI_CATALYST_TOKEN')}",
+                "X-Project-Name": project_name,
+            }
+            response = session_manager.make_request_with_retry(
+                "POST", endpoint, headers=headers, data=payload, timeout=timeout
+            )
+            elapsed_ms = (time.time() - start_time) * 1000
+            logger.debug(
+                f"API Call: [POST] {endpoint} | Status: {response.status_code} | Time: {elapsed_ms:.2f}ms"
+            )
+            if response.status_code in [200, 201]:
+                logger.info(f"Dataset schema created successfully after 401: {response.status_code}")
+                return response
+            else:
+                logger.error(f"Failed to create dataset schema after 401: {response.status_code}")
+                return None
+        else:
+            logger.error(f"Failed to create dataset schema: {response.status_code}")
+            return None
+    except (PoolError, MaxRetryError, NewConnectionError, ConnectionError, Timeout, RemoteDisconnected) as e:
+        session_manager.handle_request_exceptions(e, "creating dataset schema")
+        return None
+    except RequestException as e:
+        logger.error(f"Failed to create dataset schema: {e}")
+        return None

{ragaai_catalyst-2.2.4.1b10.dist-info → ragaai_catalyst-2.2.4.2b2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ragaai_catalyst
-Version: 2.2.4.1b10
+Version: 2.2.4.2b2
 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.1b10.dist-info → ragaai_catalyst-2.2.4.2b2.dist-info}/RECORD

@@ -55,13 +55,13 @@ ragaai_catalyst/tracers/agentic_tracing/tracers/tool_tracer.py,sha256=xxrliKPfdf
 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=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=0u4GWgqtaBz9cnr_KuqVIWDvhHWkgTAOTtiy0w8RPuk,13017
+ragaai_catalyst/tracers/agentic_tracing/upload/trace_uploader.py,sha256=U3CV_52kAMGO2q0dRbLcYr8hLpYkzwPeKCWioevqxh0,25050
+ragaai_catalyst/tracers/agentic_tracing/upload/upload_agentic_traces.py,sha256=E4A-BX5y14pcvbchvH41EXTtwsVe7tvh8qBFHeCiIxc,14444
 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
-ragaai_catalyst/tracers/agentic_tracing/utils/create_dataset_schema.py,sha256=KxxIuuB7-TIfXdwaJsAgKsG90S2tOTS021mimCOczwc,1543
+ragaai_catalyst/tracers/agentic_tracing/utils/create_dataset_schema.py,sha256=KhvLaREsNYuLntxoEl68PFcI-vrkNxaXflDewCEA8yM,3724
 ragaai_catalyst/tracers/agentic_tracing/utils/file_name_tracker.py,sha256=YG601l1a29ov9VPu9Vl4RXxgL7l16k54_WWnoTNoG58,2064
 ragaai_catalyst/tracers/agentic_tracing/utils/generic.py,sha256=WwXT01xmp8MSr7KinuDCSK9a1ifpLcT7ajFkvYviG_A,1190
 ragaai_catalyst/tracers/agentic_tracing/utils/get_user_trace_metrics.py,sha256=vPZ4dn4EHFW0kqd1GyRpsYXbfrRrd0DXCmh-pzsDBNE,1109
@@ -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.1b10.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-ragaai_catalyst-2.2.4.1b10.dist-info/METADATA,sha256=yyj5kktIZM9ouj2kyLHqEaYDsXPX2dOOGYh2aafgDuk,17682
-ragaai_catalyst-2.2.4.1b10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ragaai_catalyst-2.2.4.1b10.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
-ragaai_catalyst-2.2.4.1b10.dist-info/RECORD,,
+ragaai_catalyst-2.2.4.2b2.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+ragaai_catalyst-2.2.4.2b2.dist-info/METADATA,sha256=zbeWbluwBmIS4QXiu3RvPqjOmAlOIfMuuOajXL9VLUs,17681
+ragaai_catalyst-2.2.4.2b2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ragaai_catalyst-2.2.4.2b2.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
+ragaai_catalyst-2.2.4.2b2.dist-info/RECORD,,