ragaai-catalyst 2.2.4.1b9__tar.gz → 2.2.4.2b1__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ragaai_catalyst
-Version: 2.2.4.1b9
+Version: 2.2.4.2b1
 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.1b9 → ragaai_catalyst-2.2.4.2b1}/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.9"
+version = "2.2.4.2.beta.1"
 authors = [
     {name = "Kiran Scaria", email = "kiran.scaria@raga.ai"},
     {name = "Kedar Gaikwad", email = "kedar.gaikwad@raga.ai"},

{ragaai_catalyst-2.2.4.1b9 → ragaai_catalyst-2.2.4.2b1}/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-2.2.4.1b9 → ragaai_catalyst-2.2.4.2b1}/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-2.2.4.2b1/ragaai_catalyst/tracers/agentic_tracing/utils/create_dataset_schema.py

@@ -0,0 +1,89 @@
+import os
+import json
+import re
+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 = {}
+    metadata = user_details.get("trace_user_detail").get("metadata")
+    if metadata and isinstance(metadata, dict):
+        for key, value in metadata.items():
+            if key in ["log_source", "recorded_on"]:
+                continue
+            SCHEMA_MAPPING[key] = {"columnType": "metadata"}
+
+    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,
+        })
+
+    try:
+        # Use provided base_url or fall back to default
+        url_base = base_url if base_url is not None else RagaAICatalyst.BASE_URL
+        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
+        )
+
+        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.1b9 → ragaai_catalyst-2.2.4.2b1/ragaai_catalyst.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ragaai_catalyst
-Version: 2.2.4.1b9
+Version: 2.2.4.2b1
 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.1b9/ragaai_catalyst/tracers/agentic_tracing/utils/create_dataset_schema.py

@@ -1,43 +0,0 @@
-import os
-import json
-import re
-import requests
-from ragaai_catalyst.tracers.agentic_tracing.tracers.base import RagaAICatalyst
-
-def create_dataset_schema_with_trace(project_name, dataset_name, base_url=None, user_details=None, timeout=120):
-    SCHEMA_MAPPING = {}
-    metadata = user_details.get("trace_user_detail").get("metadata")
-    if metadata and isinstance(metadata, dict):
-        for key, value in metadata.items():
-            if key in ["log_source", "recorded_on"]:
-                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,
-            })
-        # 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
-        )
-        return response
-    response = make_request()
-    return response