ragaai-catalyst 2.2.5b5__py3-none-any.whl → 2.2.6b1__py3-none-any.whl

ragaai_catalyst/prompt_manager.py

@@ -1,9 +1,16 @@
 import os
-import requests
+from ragaai_catalyst.session_manager import session_manager
 import json
 import re
 from .ragaai_catalyst import RagaAICatalyst
 import copy
+import logging
+import time
+from urllib3.exceptions import PoolError, MaxRetryError, NewConnectionError
+from requests.exceptions import ConnectionError, Timeout, RequestException
+from http.client import RemoteDisconnected
+
+logger = logging.getLogger(__name__)
 
 class PromptManager:
     NUM_PROJECTS = 100
@@ -17,38 +24,80 @@ class PromptManager:
             project_name (str): The name of the project.
 
         Raises:
-            requests.RequestException: If there's an error with the API request.
             ValueError: If the project is not found.
         """
         self.project_name = project_name
         self.base_url = f"{RagaAICatalyst.BASE_URL}/playground/prompt"
         self.timeout = 10
         self.size = 99999 #Number of projects to fetch
+        self.project_id = None
+        self.headers = {}
 
         try:
-            response = requests.get(
+            start_time = time.time()
+            response = session_manager.make_request_with_retry(
+                "GET",
                 f"{RagaAICatalyst.BASE_URL}/v2/llm/projects?size={self.size}",
                 headers={
                     "Authorization": f'Bearer {os.getenv("RAGAAI_CATALYST_TOKEN")}',
                 },
                 timeout=self.timeout,
             )
-            response.raise_for_status()
-            # logger.debug("Projects list retrieved successfully")
-
-            project_list = [
-                project["name"] for project in response.json()["data"]["content"]
-            ]
-            self.project_id = [
-            project["id"] for project in response.json()["data"]["content"] if project["name"]==project_name
-            ][0]
-
+            elapsed_ms = (time.time() - start_time) * 1000
+            logger.debug(f"API Call: [GET] /v2/llm/projects | Status: {response.status_code} | Time: {elapsed_ms:.2f}ms")
+
+            if response.status_code in [200, 201]:
+                # logger.debug("Projects list retrieved successfully")
+                project_list = [
+                    project["name"] for project in response.json()["data"]["content"]
+                ]
+                self.project_id = [
+                project["id"] for project in response.json()["data"]["content"] if project["name"]==project_name
+                ][0]
+            elif response.status_code == 401:
+                logger.warning("Received 401 error during fetching project list. Attempting to refresh token.")
+                token = RagaAICatalyst.get_token(force_refresh=True)
+                headers = {
+                    "Authorization": f"Bearer {token}",
+                }
+                start_time = time.time()
+                response = session_manager.make_request_with_retry(
+                    "GET", f"{RagaAICatalyst.BASE_URL}/v2/llm/projects?size={self.size}", 
+                    headers=headers, timeout=self.timeout
+                )
+                elapsed_ms = (time.time() - start_time) * 1000
+                logger.debug(f"API Call: [GET] /v2/llm/projects (retry) | Status: {response.status_code} | Time: {elapsed_ms:.2f}ms")
+
+                if response.status_code in [200, 201]:
+                    logger.info("Project list fetched successfully after 401 token refresh")
+                    project_list = [
+                        project["name"] for project in response.json()["data"]["content"]
+                    ]
+                    self.project_id = [
+                    project["id"] for project in response.json()["data"]["content"] if project["name"]==project_name
+                    ][0]
+                else:
+                    logger.error("Failed to fetch project list after 401 token refresh")
+                    return
+            else:
+                logger.error(f"HTTP {response.status_code} error when fetching project list")
+                return
+
+        except (PoolError, MaxRetryError, NewConnectionError, ConnectionError, Timeout, RemoteDisconnected) as e:
+            session_manager.handle_request_exceptions(e, "fetching project list")
+            logger.error(f"Failed to fetch project list, PromptManager will have limited functionality")
+            return
+        except RequestException as e:
+            logger.error(f"Error while fetching project list: {e}")
+            logger.error(f"PromptManager will have limited functionality")
+            return
         except (KeyError, json.JSONDecodeError) as e:
-            raise ValueError(f"Error parsing project list: {str(e)}")
+            logger.error(f"Error parsing project list: {str(e)}")
+            return
 
         if self.project_name not in project_list:
-            raise ValueError("Project not found. Please enter a valid project name")
-
+            logger.error("Project not found. Please enter a valid project name")
+            return
 
         self.headers = {
                 "Authorization": f'Bearer {os.getenv("RAGAAI_CATALYST_TOKEN")}',
@@ -61,17 +110,19 @@ class PromptManager:
         List all available prompts.
 
         Returns:
-            list: A list of prompt names.
-
-        Raises:
-            requests.RequestException: If there's an error with the API request.
+            list: A list of prompt names, or empty list if error occurs.
         """
+        if not self.project_id:
+            logger.error("PromptManager not properly initialized, cannot list prompts")
+            return []
+            
         prompt = Prompt()
         try:
             prompt_list = prompt.list_prompts(self.base_url, self.headers, self.timeout)
             return prompt_list
-        except requests.RequestException as e:
-            raise requests.RequestException(f"Error listing prompts: {str(e)}")
+        except Exception as e:
+            logger.error(f"Error listing prompts: {str(e)}")
+            return []
     
     def get_prompt(self, prompt_name, version=None):
         """
@@ -82,34 +133,39 @@ class PromptManager:
             version (str, optional): The version of the prompt. Defaults to None.
 
         Returns:
-            PromptObject: An object representing the prompt.
-
-        Raises:
-            ValueError: If the prompt or version is not found.
-            requests.RequestException: If there's an error with the API request.
+            PromptObject: An object representing the prompt, or None if error occurs.
         """
+        if not self.project_id:
+            logger.error("PromptManager not properly initialized, cannot get prompt")
+            return None
+            
         try:
             prompt_list = self.list_prompts()
-        except requests.RequestException as e:
-            raise requests.RequestException(f"Error fetching prompt list: {str(e)}")
+        except Exception as e:
+            logger.error(f"Error fetching prompt list: {str(e)}")
+            return None
 
         if prompt_name not in prompt_list:
-            raise ValueError("Prompt not found. Please enter a valid prompt name")
+            logger.error("Prompt not found. Please enter a valid prompt name")
+            return None
 
         try:
             prompt_versions = self.list_prompt_versions(prompt_name)
-        except requests.RequestException as e:
-            raise requests.RequestException(f"Error fetching prompt versions: {str(e)}")
+        except Exception as e:
+            logger.error(f"Error fetching prompt versions: {str(e)}")
+            return None
 
         if version and version not in prompt_versions.keys():
-            raise ValueError("Version not found. Please enter a valid version name")
+            logger.error("Version not found. Please enter a valid version name")
+            return None
 
         prompt = Prompt()
         try:
             prompt_object = prompt.get_prompt(self.base_url, self.headers, self.timeout, prompt_name, version)
             return prompt_object
-        except requests.RequestException as e:
-            raise requests.RequestException(f"Error fetching prompt: {str(e)}")
+        except Exception as e:
+            logger.error(f"Error fetching prompt: {str(e)}")
+            return None
 
     def list_prompt_versions(self, prompt_name):
         """
@@ -119,26 +175,29 @@ class PromptManager:
             prompt_name (str): The name of the prompt.
 
         Returns:
-            dict: A dictionary mapping version names to prompt texts.
-
-        Raises:
-            ValueError: If the prompt is not found.
-            requests.RequestException: If there's an error with the API request.
+            dict: A dictionary mapping version names to prompt texts, or empty dict if error occurs.
         """
+        if not self.project_id:
+            logger.error("PromptManager not properly initialized, cannot list prompt versions")
+            return {}
+            
         try:
             prompt_list = self.list_prompts()
-        except requests.RequestException as e:
-            raise requests.RequestException(f"Error fetching prompt list: {str(e)}")
+        except Exception as e:
+            logger.error(f"Error fetching prompt list: {str(e)}")
+            return {}
 
         if prompt_name not in prompt_list:
-            raise ValueError("Prompt not found. Please enter a valid prompt name")
+            logger.error("Prompt not found. Please enter a valid prompt name")
+            return {}
         
         prompt = Prompt()
         try:
             prompt_versions = prompt.list_prompt_versions(self.base_url, self.headers, self.timeout, prompt_name)
             return prompt_versions
-        except requests.RequestException as e:
-            raise requests.RequestException(f"Error fetching prompt versions: {str(e)}")
+        except Exception as e:
+            logger.error(f"Error fetching prompt versions: {str(e)}")
+            return {}
 
 
 class Prompt:
@@ -158,21 +217,48 @@ class Prompt:
             timeout (int): The timeout for the request.
 
         Returns:
-            list: A list of prompt names.
-
-        Raises:
-            requests.RequestException: If there's an error with the API request.
-            ValueError: If there's an error parsing the prompt list.
+            list: A list of prompt names, or empty list if error occurs.
         """
         try:
-            response = requests.get(url, headers=headers, timeout=timeout)
-            response.raise_for_status()
-            prompt_list = [prompt["name"] for prompt in response.json()["data"]]                        
-            return prompt_list
-        except requests.RequestException as e:
-            raise requests.RequestException(f"Error listing prompts: {str(e)}")
+            start_time = time.time()
+            response = session_manager.make_request_with_retry("GET", url, headers=headers, timeout=timeout)
+            elapsed_ms = (time.time() - start_time) * 1000
+            logger.debug(f"API Call: [GET] {url} | Status: {response.status_code} | Time: {elapsed_ms:.2f}ms")
+
+            if response.status_code in [200, 201]:
+                prompt_list = [prompt["name"] for prompt in response.json()["data"]]
+                return prompt_list
+            elif response.status_code == 401:
+                logger.warning("Received 401 error during listing prompts. Attempting to refresh token.")
+                token = RagaAICatalyst.get_token(force_refresh=True)
+                new_headers = headers.copy()
+                new_headers["Authorization"] = f"Bearer {token}"
+
+                start_time = time.time()
+                response = session_manager.make_request_with_retry("GET", url, headers=new_headers, timeout=timeout)
+                elapsed_ms = (time.time() - start_time) * 1000
+                logger.debug(f"API Call: [GET] {url} (retry) | Status: {response.status_code} | Time: {elapsed_ms:.2f}ms")
+
+                if response.status_code in [200, 201]:
+                    logger.info("Prompts listed successfully after 401 token refresh")
+                    prompt_list = [prompt["name"] for prompt in response.json()["data"]]
+                    return prompt_list
+                else:
+                    logger.error("Failed to list prompts after 401 token refresh")
+                    return []
+            else:
+                logger.error(f"HTTP {response.status_code} error when listing prompts")
+                return []
+
+        except (PoolError, MaxRetryError, NewConnectionError, ConnectionError, Timeout, RemoteDisconnected) as e:
+            session_manager.handle_request_exceptions(e, "listing prompts")
+            return []
+        except RequestException as e:
+            logger.error(f"Error while listing prompts: {e}")
+            return []
         except (KeyError, json.JSONDecodeError) as e:
-            raise ValueError(f"Error parsing prompt list: {str(e)}")
+            logger.error(f"Error parsing prompt list: {str(e)}")
+            return []
 
     def _get_response_by_version(self, base_url, headers, timeout, prompt_name, version):
         """
@@ -186,21 +272,47 @@ class Prompt:
             version (str): The version of the prompt.
 
         Returns:
-            response: The response object containing the prompt version data.
-
-        Raises:
-            requests.RequestException: If there's an error with the API request.
-            ValueError: If there's an error parsing the prompt version.
+            response: The response object containing the prompt version data, or None if error occurs.
         """
         try:
-            response = requests.get(f"{base_url}/version/{prompt_name}?version={version}",
-                                    headers=headers, timeout=timeout)
-            response.raise_for_status()
-        except requests.RequestException as e:
-            raise requests.RequestException(f"Error fetching prompt version: {str(e)}")
+            url = f"{base_url}/version/{prompt_name}?version={version}"
+            start_time = time.time()
+            response = session_manager.make_request_with_retry("GET", url, headers=headers, timeout=timeout)
+            elapsed_ms = (time.time() - start_time) * 1000
+            logger.debug(f"API Call: [GET] {url} | Status: {response.status_code} | Time: {elapsed_ms:.2f}ms")
+
+            if response.status_code in [200, 201]:
+                return response
+            elif response.status_code == 401:
+                logger.warning(f"Received 401 error during fetching prompt version {version} for {prompt_name}. Attempting to refresh token.")
+                token = RagaAICatalyst.get_token(force_refresh=True)
+                new_headers = headers.copy()
+                new_headers["Authorization"] = f"Bearer {token}"
+
+                start_time = time.time()
+                response = session_manager.make_request_with_retry("GET", url, headers=new_headers, timeout=timeout)
+                elapsed_ms = (time.time() - start_time) * 1000
+                logger.debug(f"API Call: [GET] {url} (retry) | Status: {response.status_code} | Time: {elapsed_ms:.2f}ms")
+
+                if response.status_code in [200, 201]:
+                    logger.info(f"Prompt version {version} for {prompt_name} fetched successfully after 401 token refresh")
+                    return response
+                else:
+                    logger.error(f"Failed to fetch prompt version {version} for {prompt_name} after 401 token refresh")
+                    return None
+            else:
+                logger.error(f"HTTP {response.status_code} error when fetching prompt version {version} for {prompt_name}")
+                return None
+
+        except (PoolError, MaxRetryError, NewConnectionError, ConnectionError, Timeout, RemoteDisconnected) as e:
+            session_manager.handle_request_exceptions(e, f"fetching prompt version {version} for {prompt_name}")
+            return None
+        except RequestException as e:
+            logger.error(f"Error while fetching prompt version {version} for {prompt_name}: {e}")
+            return None
         except (KeyError, json.JSONDecodeError, IndexError) as e:
-            raise ValueError(f"Error parsing prompt version: {str(e)}")
-        return response
+            logger.error(f"Error parsing prompt version: {str(e)}")
+            return None
 
     def _get_response(self, base_url, headers, timeout, prompt_name):
         """
@@ -213,21 +325,47 @@ class Prompt:
             prompt_name (str): The name of the prompt.
 
         Returns:
-            response: The response object containing the latest prompt version data.
-
-        Raises:
-            requests.RequestException: If there's an error with the API request.
-            ValueError: If there's an error parsing the prompt version.
+            response: The response object containing the latest prompt version data, or None if error occurs.
         """
         try:
-            response = requests.get(f"{base_url}/version/{prompt_name}",
-                                headers=headers, timeout=timeout)
-            response.raise_for_status()
-        except requests.RequestException as e:
-            raise requests.RequestException(f"Error fetching prompt version: {str(e)}")
+            url = f"{base_url}/version/{prompt_name}"
+            start_time = time.time()
+            response = session_manager.make_request_with_retry("GET", url, headers=headers, timeout=timeout)
+            elapsed_ms = (time.time() - start_time) * 1000
+            logger.debug(f"API Call: [GET] {url} | Status: {response.status_code} | Time: {elapsed_ms:.2f}ms")
+
+            if response.status_code in [200, 201]:
+                return response
+            elif response.status_code == 401:
+                logger.warning(f"Received 401 error during fetching latest prompt version for {prompt_name}. Attempting to refresh token.")
+                token = RagaAICatalyst.get_token(force_refresh=True)
+                new_headers = headers.copy()
+                new_headers["Authorization"] = f"Bearer {token}"
+
+                start_time = time.time()
+                response = session_manager.make_request_with_retry("GET", url, headers=new_headers, timeout=timeout)
+                elapsed_ms = (time.time() - start_time) * 1000
+                logger.debug(f"API Call: [GET] {url} (retry) | Status: {response.status_code} | Time: {elapsed_ms:.2f}ms")
+
+                if response.status_code in [200, 201]:
+                    logger.info(f"Latest prompt version for {prompt_name} fetched successfully after 401 token refresh")
+                    return response
+                else:
+                    logger.error(f"Failed to fetch latest prompt version for {prompt_name} after 401 token refresh")
+                    return None
+            else:
+                logger.error(f"HTTP {response.status_code} error when fetching latest prompt version for {prompt_name}")
+                return None
+
+        except (PoolError, MaxRetryError, NewConnectionError, ConnectionError, Timeout, RemoteDisconnected) as e:
+            session_manager.handle_request_exceptions(e, f"fetching latest prompt version for {prompt_name}")
+            return None
+        except RequestException as e:
+            logger.error(f"Error while fetching latest prompt version for {prompt_name}: {e}")
+            return None
         except (KeyError, json.JSONDecodeError, IndexError) as e:
-            raise ValueError(f"Error parsing prompt version: {str(e)}")
-        return response
+            logger.error(f"Error parsing prompt version: {str(e)}")
+            return None
 
     def _get_prompt_by_version(self, base_url, headers, timeout, prompt_name, version):
         """
@@ -241,14 +379,17 @@ class Prompt:
             version (str): The version of the prompt.
 
         Returns:
-            str: The text of the prompt.
-
-        Raises:
-            requests.RequestException: If there's an error with the API request.
+            str: The text of the prompt, or empty string if error occurs.
         """
         response = self._get_response_by_version(base_url, headers, timeout, prompt_name, version)
-        prompt_text = response.json()["data"]["docs"][0]["textFields"]
-        return prompt_text
+        if response is None:
+            return ""
+        try:
+            prompt_text = response.json()["data"]["docs"][0]["textFields"]
+            return prompt_text
+        except (KeyError, json.JSONDecodeError, IndexError) as e:
+            logger.error(f"Error parsing prompt text: {str(e)}")
+            return ""
 
     def get_prompt(self, base_url, headers, timeout, prompt_name, version=None):
         """
@@ -262,22 +403,24 @@ class Prompt:
             version (str, optional): The version of the prompt. Defaults to None.
 
         Returns:
-            PromptObject: An object representing the prompt.
-
-        Raises:
-            requests.RequestException: If there's an error with the API request.
+            PromptObject: An object representing the prompt, or None if error occurs.
         """
         if version:
             response = self._get_response_by_version(base_url, headers, timeout, prompt_name, version)
-            prompt_text = response.json()["data"]["docs"][0]["textFields"]
-            prompt_parameters = response.json()["data"]["docs"][0]["modelSpecs"]["parameters"]
-            model = response.json()["data"]["docs"][0]["modelSpecs"]["model"]
         else:
             response = self._get_response(base_url, headers, timeout, prompt_name)
+            
+        if response is None:
+            return None
+
+        try:
             prompt_text = response.json()["data"]["docs"][0]["textFields"]
             prompt_parameters = response.json()["data"]["docs"][0]["modelSpecs"]["parameters"]
             model = response.json()["data"]["docs"][0]["modelSpecs"]["model"]
-        return PromptObject(prompt_text, prompt_parameters, model)
+            return PromptObject(prompt_text, prompt_parameters, model)
+        except (KeyError, json.JSONDecodeError, IndexError) as e:
+            logger.error(f"Error parsing prompt data: {str(e)}")
+            return None
 
 
     def list_prompt_versions(self, base_url, headers, timeout, prompt_name):
@@ -291,25 +434,55 @@ class Prompt:
             prompt_name (str): The name of the prompt.
 
         Returns:
-            dict: A dictionary mapping version names to prompt texts.
-
-        Raises:
-            requests.RequestException: If there's an error with the API request.
-            ValueError: If there's an error parsing the prompt versions.
+            dict: A dictionary mapping version names to prompt texts, or empty dict if error occurs.
         """
         try:
-            response = requests.get(f"{base_url}/{prompt_name}/version",
-                                    headers=headers, timeout=timeout)
-            response.raise_for_status()
-            version_names = [version["name"] for version in response.json()["data"]]
-            prompt_versions = {}
-            for version in version_names:
-                prompt_versions[version] = self._get_prompt_by_version(base_url, headers, timeout, prompt_name, version)
-            return prompt_versions
-        except requests.RequestException as e:
-            raise requests.RequestException(f"Error listing prompt versions: {str(e)}")
+            url = f"{base_url}/{prompt_name}/version"
+            start_time = time.time()
+            response = session_manager.make_request_with_retry("GET", url, headers=headers, timeout=timeout)
+            elapsed_ms = (time.time() - start_time) * 1000
+            logger.debug(f"API Call: [GET] {url} | Status: {response.status_code} | Time: {elapsed_ms:.2f}ms")
+
+            if response.status_code in [200, 201]:
+                version_names = [version["name"] for version in response.json()["data"]]
+                prompt_versions = {}
+                for version in version_names:
+                    prompt_versions[version] = self._get_prompt_by_version(base_url, headers, timeout, prompt_name, version)
+                return prompt_versions
+            elif response.status_code == 401:
+                logger.warning(f"Received 401 error during listing prompt versions for {prompt_name}. Attempting to refresh token.")
+                token = RagaAICatalyst.get_token(force_refresh=True)
+                new_headers = headers.copy()
+                new_headers["Authorization"] = f"Bearer {token}"
+
+                start_time = time.time()
+                response = session_manager.make_request_with_retry("GET", url, headers=new_headers, timeout=timeout)
+                elapsed_ms = (time.time() - start_time) * 1000
+                logger.debug(f"API Call: [GET] {url} (retry) | Status: {response.status_code} | Time: {elapsed_ms:.2f}ms")
+
+                if response.status_code in [200, 201]:
+                    logger.info(f"Prompt versions for {prompt_name} listed successfully after 401 token refresh")
+                    version_names = [version["name"] for version in response.json()["data"]]
+                    prompt_versions = {}
+                    for version in version_names:
+                        prompt_versions[version] = self._get_prompt_by_version(base_url, new_headers, timeout, prompt_name, version)
+                    return prompt_versions
+                else:
+                    logger.error(f"Failed to list prompt versions for {prompt_name} after 401 token refresh")
+                    return {}
+            else:
+                logger.error(f"HTTP {response.status_code} error when listing prompt versions for {prompt_name}")
+                return {}
+
+        except (PoolError, MaxRetryError, NewConnectionError, ConnectionError, Timeout, RemoteDisconnected) as e:
+            session_manager.handle_request_exceptions(e, f"listing prompt versions for {prompt_name}")
+            return {}
+        except RequestException as e:
+            logger.error(f"Error while listing prompt versions for {prompt_name}: {e}")
+            return {}
         except (KeyError, json.JSONDecodeError) as e:
-            raise ValueError(f"Error parsing prompt versions: {str(e)}")
+            logger.error(f"Error parsing prompt versions: {str(e)}")
+            return {}
 
 
 class PromptObject:
@@ -325,7 +498,7 @@ class PromptObject:
         self.text = text
         self.parameters = parameters
         self.model = model
-    
+
     def _extract_variable_from_content(self, content):
         """
         Extract variables from the content.

ragaai_catalyst/tracers/agentic_tracing/upload/trace_uploader.py

@@ -3,22 +3,16 @@ trace_uploader.py - A dedicated process for handling trace uploads
 """
 
 import os
-import sys
 import json
 import time
-import signal
 import logging
 import argparse
 import tempfile
-from pathlib import Path
-import multiprocessing
-import queue
 from datetime import datetime
 import atexit
-import glob
 from logging.handlers import RotatingFileHandler
 import concurrent.futures
-from typing import Dict, Any, Optional
+from typing import Dict, Any
 import threading
 import uuid
 
@@ -50,7 +44,7 @@ try:
     from ragaai_catalyst.tracers.agentic_tracing.upload.upload_code import upload_code
     # from ragaai_catalyst.tracers.agentic_tracing.upload.upload_trace_metric import upload_trace_metric
     from ragaai_catalyst.tracers.agentic_tracing.utils.create_dataset_schema import create_dataset_schema_with_trace
-    from ragaai_catalyst.tracers.agentic_tracing.upload.session_manager import session_manager
+    from ragaai_catalyst.session_manager import session_manager
     from ragaai_catalyst import RagaAICatalyst
     IMPORTS_AVAILABLE = True
 except ImportError:
@@ -481,7 +475,8 @@ def submit_upload_task(filepath, hash_id, zip_path, project_name, project_id, da
         
         return task_id
     except RuntimeError as e:
-        if "cannot schedule new futures after shutdown" in str(e):
+        if any(msg in str(e) for msg in
+               ("cannot schedule new futures after shutdown", "cannot schedule new futures after interpreter shutdown")):
             logger.warning(f"Executor already shut down, falling back to synchronous processing: {e}")
             return do_sync_processing()
         else:

ragaai_catalyst/tracers/agentic_tracing/upload/upload_agentic_traces.py

@@ -7,7 +7,7 @@ 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
+from ragaai_catalyst.session_manager import session_manager
 
 logger = logging.getLogger(__name__)
 

ragaai_catalyst/tracers/agentic_tracing/upload/upload_code.py

@@ -9,7 +9,7 @@ from requests.exceptions import ConnectionError, Timeout, RequestException
 from http.client import RemoteDisconnected
 
 from ragaai_catalyst.ragaai_catalyst import RagaAICatalyst
-from .session_manager import session_manager
+from ragaai_catalyst.session_manager import session_manager
 
 logger = logging.getLogger(__name__)
 

ragaai_catalyst/tracers/agentic_tracing/utils/create_dataset_schema.py

@@ -1,25 +1,36 @@
 import os
 import json
-import re
+import requests
 import logging
 import time
+from typing import Optional
 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
+from ragaai_catalyst.session_manager import session_manager
 
+IGNORED_KEYS = {"log_source", "recorded_on"}
 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):
+def create_dataset_schema_with_trace(
+        project_name: str,
+        dataset_name: str,
+        base_url: Optional[str] = None,
+        user_details: Optional[dict] = None,
+        timeout: int = 120) -> requests.Response:
+    schema_mapping = {}
+
+    metadata = (
+        user_details.get("trace_user_detail", {}).get("metadata", {})
+        if user_details else {}
+    )
+    if isinstance(metadata, dict):
         for key, value in metadata.items():
-            if key in ["log_source", "recorded_on"]:
+            if key in IGNORED_KEYS:
                 continue
-            SCHEMA_MAPPING[key] = {"columnType": "metadata"}
+            schema_mapping[key] = {"columnType": "metadata"}
 
     headers = {
         "Content-Type": "application/json",
@@ -27,11 +38,11 @@ def create_dataset_schema_with_trace(project_name, dataset_name, base_url=None,
         "X-Project-Name": project_name,
     }
 
-    if SCHEMA_MAPPING:
+    if schema_mapping:
         payload = json.dumps({
             "datasetName": dataset_name,
             "traceFolderUrl": None,
-            "schemaMapping": SCHEMA_MAPPING
+            "schemaMapping": schema_mapping
         })
     else:
         payload = json.dumps({

ragaai_catalyst/tracers/agentic_tracing/utils/zip_list_of_unique_files.py

@@ -7,6 +7,8 @@ import re
 import ast
 import importlib.util
 import json
+from typing import List, Optional, Tuple
+
 import ipynbname
 from copy import deepcopy
 
@@ -460,8 +462,14 @@ class TraceDependencyTracker:
         logger.debug(f"Zip file created successfully at: {zip_filename}")
         return hash_id, zip_filename
 
-def zip_list_of_unique_files(filepaths, output_dir=None):
+def zip_list_of_unique_files(
+        filepaths: List[str],
+        output_dir: Optional[str] = None
+) -> Tuple[str, str]:
     """Create a zip file containing all unique files and their dependencies."""
+    if not filepaths:
+        logger.warning("The filepaths list is empty. Proceeding with an empty ZIP archive.")
+        filepaths = []
     if output_dir is None:
         # Set default output directory based on environment
         if JupyterNotebookHandler.is_running_in_colab():

ragaai_catalyst/tracers/exporters/dynamic_trace_exporter.py

@@ -4,17 +4,34 @@ Dynamic Trace Exporter - A wrapper for RAGATraceExporter that allows dynamic upd
 import logging
 from opentelemetry.sdk.trace.export import SpanExporter, SpanExportResult
 from ragaai_catalyst.tracers.exporters.ragaai_trace_exporter import RAGATraceExporter
+from typing import Optional, List, Dict, Callable
 
 logger = logging.getLogger("RagaAICatalyst")
 
+
 class DynamicTraceExporter(SpanExporter):
     """
     A wrapper around RAGATraceExporter that allows dynamic updates to properties.
     This exporter forwards all calls to the underlying RAGATraceExporter but allows
     certain properties to be updated dynamically during execution.
     """
-    
-    def __init__(self, tracer_type, files_to_zip, project_name, project_id, dataset_name, user_details, base_url, custom_model_cost, timeout=120, post_processor = None, max_upload_workers = 30,user_context = None, user_gt = None, external_id=None):
+    def __init__(
+            self,
+            project_name: str,
+            dataset_name: str,
+            base_url: str,
+            tracer_type: str,
+            files_to_zip: Optional[List[str]] = None,
+            project_id: Optional[str] = None,
+            user_details: Optional[Dict] = None,
+            custom_model_cost: Optional[dict] = None,
+            timeout: int = 120,
+            post_processor: Optional[Callable] = None,
+            max_upload_workers: int = 30,
+            user_context: Optional[str] = None,
+            user_gt: Optional[str] = None,
+            external_id: Optional[str] = None
+    ):
         """
         Initialize the DynamicTraceExporter.
         
@@ -30,22 +47,22 @@ class DynamicTraceExporter(SpanExporter):
             max_upload_workers: Maximum number of upload workers
         """
         self._exporter = RAGATraceExporter(
+            project_name=project_name,
+            dataset_name=dataset_name,
+            base_url=base_url,
             tracer_type=tracer_type,
             files_to_zip=files_to_zip,
-            project_name=project_name,
             project_id=project_id,
-            dataset_name=dataset_name,
             user_details=user_details,
-            base_url=base_url,
             custom_model_cost=custom_model_cost,
             timeout=timeout,
-            post_processor= post_processor,
-            max_upload_workers = max_upload_workers,
-            user_context = user_context,
-            user_gt = user_gt,
+            post_processor=post_processor,
+            max_upload_workers=max_upload_workers,
+            user_context=user_context,
+            user_gt=user_gt,
             external_id=external_id
         )
-        
+
         # Store the initial values
         self._files_to_zip = files_to_zip
         self._project_name = project_name
@@ -60,7 +77,6 @@ class DynamicTraceExporter(SpanExporter):
         self._user_gt = user_gt
         self._external_id = external_id
 
-    
     def export(self, spans):
         """
         Export spans by forwarding to the underlying exporter.
@@ -84,8 +100,6 @@ class DynamicTraceExporter(SpanExporter):
             return result
         except Exception as e:
             logger.error(f"Error exporting trace: {e}")
-            
-
 
     def shutdown(self):
         """
@@ -103,7 +117,7 @@ class DynamicTraceExporter(SpanExporter):
             return self._exporter.shutdown()
         except Exception as e:
             logger.error(f"Error shutting down exporter: {e}")
-    
+
     def _update_exporter_properties(self):
         """
         Update the underlying exporter's properties with the current values.
@@ -118,55 +132,55 @@ class DynamicTraceExporter(SpanExporter):
         self._exporter.post_processor = self._post_processor
         self._exporter.max_upload_workers = self._max_upload_workers
         self._exporter.user_context = self._user_context
-        self._exporter.user_gt = self._user_gt    
+        self._exporter.user_gt = self._user_gt
         self._exporter.external_id = self._external_id
-    
+
     # Getter and setter methods for dynamic properties
-    
+
     @property
     def files_to_zip(self):
         return self._files_to_zip
-    
+
     @files_to_zip.setter
     def files_to_zip(self, value):
         self._files_to_zip = value
-    
+
     @property
     def project_name(self):
         return self._project_name
-    
+
     @project_name.setter
     def project_name(self, value):
         self._project_name = value
-    
+
     @property
     def project_id(self):
         return self._project_id
-    
+
     @project_id.setter
     def project_id(self, value):
         self._project_id = value
-    
+
     @property
     def dataset_name(self):
         return self._dataset_name
-    
+
     @dataset_name.setter
     def dataset_name(self, value):
         self._dataset_name = value
-    
+
     @property
     def user_details(self):
         return self._user_details
-    
+
     @user_details.setter
     def user_details(self, value):
         self._user_details = value
-    
+
     @property
     def base_url(self):
         return self._base_url
-    
+
     @base_url.setter
     def base_url(self, value):
         self._base_url = value
@@ -174,15 +188,15 @@ class DynamicTraceExporter(SpanExporter):
     @property
     def custom_model_cost(self):
         return self._custom_model_cost
-    
+
     @custom_model_cost.setter
     def custom_model_cost(self, value):
         self._custom_model_cost = value
-    
+
     @property
     def max_upload_workers(self):
         return self._max_upload_workers
-    
+
     @max_upload_workers.setter
     def max_upload_workers(self, value):
         self._max_upload_workers = value
@@ -190,7 +204,7 @@ class DynamicTraceExporter(SpanExporter):
     @property
     def user_context(self):
         return self._user_context
-    
+
     @user_context.setter
     def user_context(self, value):
         self._user_context = value
@@ -198,7 +212,7 @@ class DynamicTraceExporter(SpanExporter):
     @property
     def user_gt(self):
         return self._user_gt
-    
+
     @user_gt.setter
     def user_gt(self, value):
         self._user_gt = value
@@ -206,8 +220,7 @@ class DynamicTraceExporter(SpanExporter):
     @property
     def external_id(self):
         return self._external_id
-    
+
     @external_id.setter
     def external_id(self, value):
         self._external_id = value
-    

ragaai_catalyst/tracers/exporters/ragaai_trace_exporter.py

@@ -3,6 +3,7 @@ import logging
 import os
 import tempfile
 from dataclasses import asdict
+from typing import Optional, Callable, Dict, List
 
 from opentelemetry.sdk.trace.export import SpanExporter, SpanExportResult
 
@@ -23,6 +24,7 @@ logging_level = (
     logger.setLevel(logging.DEBUG) if os.getenv("DEBUG") == "1" else logging.INFO
 )
 
+
 class TracerJSONEncoder(json.JSONEncoder):
     def default(self, obj):
         if isinstance(obj, datetime):
@@ -47,8 +49,25 @@ class TracerJSONEncoder(json.JSONEncoder):
         except:
             return None  # Last resort: return None instead of failing
 
+
 class RAGATraceExporter(SpanExporter):
-    def __init__(self, tracer_type, files_to_zip, project_name, project_id, dataset_name, user_details, base_url, custom_model_cost, timeout=120, post_processor = None, max_upload_workers = 30,user_context = None, user_gt = None, external_id=None):
+    def __init__(
+            self,
+            project_name: str,
+            dataset_name: str,
+            base_url: str,
+            tracer_type: str,
+            files_to_zip: Optional[List[str]] = None,
+            project_id: Optional[str] = None,
+            user_details: Optional[Dict] = None,
+            custom_model_cost: Optional[dict] = None,
+            timeout: int = 120,
+            post_processor: Optional[Callable] = None,
+            max_upload_workers: int = 30,
+            user_context: Optional[str] = None,
+            user_gt: Optional[str] = None,
+            external_id: Optional[str] = None
+    ):
         self.trace_spans = dict()
         self.tmp_dir = tempfile.gettempdir()
         self.tracer_type = tracer_type
@@ -77,7 +96,7 @@ class RAGATraceExporter(SpanExporter):
 
                 if trace_id not in self.trace_spans:
                     self.trace_spans[trace_id] = list()
-                
+
                 if span_json.get("attributes").get("openinference.span.kind", None) is None:
                     span_json["attributes"]["openinference.span.kind"] = "UNKNOWN"
 
@@ -118,10 +137,10 @@ class RAGATraceExporter(SpanExporter):
         if ragaai_trace_details is None:
             logger.error(f"Cannot upload trace {trace_id}: conversion failed and returned None")
             return  # Exit early if conversion failed
-            
+
         # Upload the trace if upload_trace function is provided
         try:
-            if self.post_processor!=None:
+            if self.post_processor != None:
                 ragaai_trace_details['trace_file_path'] = self.post_processor(ragaai_trace_details['trace_file_path'])
             self.upload_trace(ragaai_trace_details, trace_id)
         except Exception as e:
@@ -130,13 +149,14 @@ class RAGATraceExporter(SpanExporter):
     def prepare_trace(self, spans, trace_id):
         try:
             try:
-                ragaai_trace = convert_json_format(spans, self.custom_model_cost, self.user_context, self.user_gt,self.external_id)   
+                ragaai_trace = convert_json_format(spans, self.custom_model_cost, self.user_context, self.user_gt,
+                                                   self.external_id)
             except Exception as e:
                 print(f"Error in convert_json_format function: {trace_id}: {e}")
                 return None
-            
+
             try:
-                interactions = format_interactions(ragaai_trace)         
+                interactions = format_interactions(ragaai_trace)
                 ragaai_trace["workflow"] = interactions['workflow']
             except Exception as e:
                 print(f"Error in format_interactions function: {trace_id}: {e}")
@@ -183,18 +203,26 @@ class RAGATraceExporter(SpanExporter):
             except Exception as e:
                 print(f"Error in adding tracer type: {trace_id}: {e}")
                 return None
-            
-            #Add user passed metadata to the trace
+
+            # Add user passed metadata to the trace
             try:
-                if self.user_details.get("trace_user_detail").get("metadata") and isinstance(self.user_details.get("trace_user_detail").get("metadata"), dict):
-                    for key, value in self.user_details.get("trace_user_detail").get("metadata").items():
-                        if key in ["log_source", "recorded_on"]:
-                            continue
-                        ragaai_trace["metadata"][key] = value
+                logger.debug("Started adding user passed metadata")
+
+                metadata = (
+                    self.user_details.get("trace_user_detail", {}).get("metadata", {})
+                    if self.user_details else {}
+                )
+
+                if isinstance(metadata, dict):
+                    for key, value in metadata.items():
+                        if key not in {"log_source", "recorded_on"}:
+                            ragaai_trace.setdefault("metadata", {})[key] = value
+
+                logger.debug("Completed adding user passed metadata")
             except Exception as e:
                 print(f"Error in adding metadata: {trace_id}: {e}")
                 return None
-            
+
             try:
                 # Save the trace_json 
                 trace_file_path = os.path.join(self.tmp_dir, f"{trace_id}.json")
@@ -220,16 +248,16 @@ class RAGATraceExporter(SpanExporter):
         hash_id = ragaai_trace_details['hash_id']
         zip_path = ragaai_trace_details['code_zip_path']
         self.upload_task_id = submit_upload_task(
-                filepath=filepath,
-                hash_id=hash_id,
-                zip_path=zip_path,
-                project_name=self.project_name,
-                project_id=self.project_id,
-                dataset_name=self.dataset_name,
-                user_details=self.user_details,
-                base_url=self.base_url,
-                tracer_type=self.tracer_type,
-                timeout=self.timeout
-            )
-
-        logger.info(f"Submitted upload task with ID: {self.upload_task_id}")
+            filepath=filepath,
+            hash_id=hash_id,
+            zip_path=zip_path,
+            project_name=self.project_name,
+            project_id=self.project_id,
+            dataset_name=self.dataset_name,
+            user_details=self.user_details,
+            base_url=self.base_url,
+            tracer_type=self.tracer_type,
+            timeout=self.timeout
+        )
+
+        logger.info(f"Submitted upload task with ID: {self.upload_task_id}")

ragaai_catalyst/tracers/tracer.py

@@ -600,13 +600,13 @@ class Tracer(AgenticTracing):
 
         # Create a dynamic exporter that allows property updates
         self.dynamic_exporter = DynamicTraceExporter(
+            project_name=self.project_name,
+            dataset_name=self.dataset_name,
+            base_url=self.base_url,
             tracer_type=self.tracer_type,
             files_to_zip=list_of_unique_files,
-            project_name=self.project_name,
             project_id=self.project_id,
-            dataset_name=self.dataset_name,
             user_details=self.user_details,
-            base_url=self.base_url,
             custom_model_cost=self.model_custom_cost,
             timeout = self.timeout,
             post_processor= self.post_processor,

ragaai_catalyst/tracers/utils/trace_json_converter.py

@@ -188,7 +188,13 @@ def convert_json_format(
                 ]
                 model_name = next((name for name in reversed(model_names) if name), "")
                 if not model_name and span["attributes"].get("openinference.span.kind")=="LLM":
-                    model_name = json.loads(span["attributes"].get("metadata", "")).get("ls_model_name", "")
+                    try:
+                        metadata = span["attributes"].get("metadata") or span["attributes"].get("aiq.metadata")
+                        metadata = json.loads(metadata)
+                        model_name = metadata.get("ls_model_name", "")
+                    except Exception as e:
+                        model_name = ""
+                        logger.error(f"Failed to parse metadata: {e}", exc_info=True)
                 if model_name and span["attributes"].get("openinference.span.kind") == "LLM":
                     try:
                         model_costs = get_model_cost()

{ragaai_catalyst-2.2.5b5.dist-info → ragaai_catalyst-2.2.6b1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ragaai_catalyst
-Version: 2.2.5b5
+Version: 2.2.6b1
 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.5b5.dist-info → ragaai_catalyst-2.2.6b1.dist-info}/RECORD

@@ -5,10 +5,11 @@ ragaai_catalyst/evaluation.py,sha256=8P2zUMSMsQGmKdv7_dZ8F0iXWYddvappgKPN5oJXWuY
 ragaai_catalyst/guard_executor.py,sha256=VLqVO1gAEBaovoQQUjdNkivH1dXACFTsuXQ1nzAALQQ,14008
 ragaai_catalyst/guardrails_manager.py,sha256=_VrARJ1udmCF8TklNKy7XTQUaM8ATDhTOAGDonBkFro,14245
 ragaai_catalyst/internal_api_completion.py,sha256=DdICI5yfEudiOAIC8L4oxH0Qz7kX-BZCdo9IWsi2gNo,2965
-ragaai_catalyst/prompt_manager.py,sha256=W8ypramzOprrJ7-22d5vkBXIuIQ8v9XAzKDGxKsTK28,16550
+ragaai_catalyst/prompt_manager.py,sha256=9IoCTxjCY9FKopq6BoB0fVb3FtuSj3BBsNyD9AeH7ds,26265
 ragaai_catalyst/proxy_call.py,sha256=CHxldeceZUaLU-to_hs_Kf1z_b2vHMssLS_cOBedu78,5499
 ragaai_catalyst/ragaai_catalyst.py,sha256=ZlcpOgJA9lVRi51YFy4dVfsxU0I79LJu0MnVI5BIL-c,25201
 ragaai_catalyst/redteaming_old.py,sha256=W2d89Ok8W-C8g7TBM3fDIFLof3q9FuYSr0jcryH2XQo,7097
+ragaai_catalyst/session_manager.py,sha256=sOlxeIYIP8tycaTtZC9xkZosi6EDJUxvDw0_rc_NLI8,6823
 ragaai_catalyst/synthetic_data_generation.py,sha256=AumjIzKk-Uvn7RQGatpx7TPvlI4NjU-rUiVFockoGNg,37969
 ragaai_catalyst/utils.py,sha256=TlhEFwLyRU690HvANbyoRycR3nQ67lxVUQoUOfTPYQ0,3772
 ragaai_catalyst/redteaming/__init__.py,sha256=TJdvZpaZGFsg9qKONdjTosSVLZGadYFpHG6KE0xapKU,155
@@ -27,7 +28,7 @@ ragaai_catalyst/redteaming/utils/issue_description.py,sha256=iB0XbeOjdqHTPrikCKS
 ragaai_catalyst/redteaming/utils/rt.png,sha256=HzVC8bz_4UgwafKXuMe8RJVI6CyK_UmSgo53ceAOQK8,282154
 ragaai_catalyst/tracers/__init__.py,sha256=LfgTes-nHpazssbGKnn8kyLZNr49kIPrlkrqqoTFTfc,301
 ragaai_catalyst/tracers/distributed.py,sha256=CGPuOh4CsgEk428PPibieLaAG2Tt3BVygF6ZlmbXxg4,10009
-ragaai_catalyst/tracers/tracer.py,sha256=5jM-AJozLoc95OIR-K3K2fs0OJ2Oo47Lgtca_6ToFzk,33649
+ragaai_catalyst/tracers/tracer.py,sha256=uLSrN4HZT19YmEG5dTPx_aauuNU8UfflNTn3xjshfmI,33649
 ragaai_catalyst/tracers/agentic_tracing/README.md,sha256=X4QwLb7-Jg7GQMIXj-SerZIgDETfw-7VgYlczOR8ZeQ,4508
 ragaai_catalyst/tracers/agentic_tracing/__init__.py,sha256=yf6SKvOPSpH-9LiKaoLKXwqj5sez8F_5wkOb91yp0oE,260
 ragaai_catalyst/tracers/agentic_tracing/data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -35,31 +36,30 @@ ragaai_catalyst/tracers/agentic_tracing/data/data_structure.py,sha256=icAtNzKN_I
 ragaai_catalyst/tracers/agentic_tracing/tracers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ragaai_catalyst/tracers/agentic_tracing/tracers/main_tracer.py,sha256=Wq4LFclPlLy47LyXvbaLeYiSMQABj7VYS3J87xyea_E,4159
 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=NuetePZOdDmwRgN3aAsQrDIytXFicNylaAqORdz8C2o,25051
-ragaai_catalyst/tracers/agentic_tracing/upload/upload_agentic_traces.py,sha256=yLzYiyNk_XtwwlUFbq5uaRahDKKF-eFBoI6xqHUNedw,14526
-ragaai_catalyst/tracers/agentic_tracing/upload/upload_code.py,sha256=IAhNFS-nbV_ImNz8Xp98qU4r-2naj49qg9q08x53TFE,12521
+ragaai_catalyst/tracers/agentic_tracing/upload/trace_uploader.py,sha256=mDpLqTE6j7gnxTS1Siev-gW_mnUJ086UB5l1bBoA9vU,25007
+ragaai_catalyst/tracers/agentic_tracing/upload/upload_agentic_traces.py,sha256=oAiP6k3wYc6LTi4tHQctTMhTzpk492x5JTt5wEKcwC4,14541
+ragaai_catalyst/tracers/agentic_tracing/upload/upload_code.py,sha256=a3pqyVvu5OJ9oDesHO6TeeCPHG-V0RSu1ZLKB-rYxCw,12536
 ragaai_catalyst/tracers/agentic_tracing/utils/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-ragaai_catalyst/tracers/agentic_tracing/utils/create_dataset_schema.py,sha256=1c6HA_Alm5yofF-eifqwdwHboxr25mvW3xxVYBktfjA,3667
+ragaai_catalyst/tracers/agentic_tracing/utils/create_dataset_schema.py,sha256=GJex8B7mgzFDQgxzFIolce0RUWGYRYog_ijkC9nNt1E,3856
 ragaai_catalyst/tracers/agentic_tracing/utils/file_name_tracker.py,sha256=YG601l1a29ov9VPu9Vl4RXxgL7l16k54_WWnoTNoG58,2064
 ragaai_catalyst/tracers/agentic_tracing/utils/llm_utils.py,sha256=PiyXvEj_qu0EnJFjk4GfGyWFZbwlvQQh0hdQ_lm0p8E,22976
 ragaai_catalyst/tracers/agentic_tracing/utils/model_costs.json,sha256=2tzGw_cKCTPcfjEm7iGvFE6pTw7gMTPzeBov_MTaXNY,321336
 ragaai_catalyst/tracers/agentic_tracing/utils/system_monitor.py,sha256=H8WNsk4v_5T6OUw4TFOzlDLjQhJwjh1nAMyMAoqMEi4,6946
 ragaai_catalyst/tracers/agentic_tracing/utils/trace_utils.py,sha256=W7Nw-IpugejIoHbCtQiN4Sn4ughLocQ9AUCjuAtOhOo,17258
 ragaai_catalyst/tracers/agentic_tracing/utils/unique_decorator.py,sha256=G027toV-Km20JjKrc-Y_PilQ8ABEKrBvvzgLTnqVg7I,5819
-ragaai_catalyst/tracers/agentic_tracing/utils/zip_list_of_unique_files.py,sha256=4TeCGsFF26249fV6dJHLTZDrRa93SG9oer4rudoF8Y4,19443
+ragaai_catalyst/tracers/agentic_tracing/utils/zip_list_of_unique_files.py,sha256=Q3KXaqpvCPl7CL4bzwKvE6xhY5JNvfxar6QUXXXz3w0,19688
 ragaai_catalyst/tracers/exporters/__init__.py,sha256=qA3vx7z9CQ5kTGCn9LIDtIFvW9fJHQLkvF9-xBQUm94,237
-ragaai_catalyst/tracers/exporters/dynamic_trace_exporter.py,sha256=Rm-QaLv1qMAKpHKcFOcK_HWaKHwFBoUH45_4QYipE-g,6843
+ragaai_catalyst/tracers/exporters/dynamic_trace_exporter.py,sha256=cQvBGFOgSNLoyPIvkTrHottlpmg1vcgFgej-J2PwGtY,7149
 ragaai_catalyst/tracers/exporters/file_span_exporter.py,sha256=_icciSCktK6c86KB2HV3GZMFHvUitgKJ8x_IdPmgi1M,6363
-ragaai_catalyst/tracers/exporters/ragaai_trace_exporter.py,sha256=M3Bj311o_MpjY1gO9SGpfIIT2O0yz76uCgw6mKUwHdo,10032
+ragaai_catalyst/tracers/exporters/ragaai_trace_exporter.py,sha256=aUooYU7SlT29pYK_p9WO5udlFO-yOEZNeR3RSB-DOBo,10549
 ragaai_catalyst/tracers/instrumentators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ragaai_catalyst/tracers/utils/__init__.py,sha256=KeMaZtYaTojilpLv65qH08QmpYclfpacDA0U3wg6Ybw,64
 ragaai_catalyst/tracers/utils/model_prices_and_context_window_backup.json,sha256=WlZCZeOQ54aMVjYS8BAeka2uaFC3ftBTMZ8zzzA8TAI,495947
 ragaai_catalyst/tracers/utils/rag_extraction_logic_final.py,sha256=3ygkRT__lLDRflRttjzPu28tIA8cTCiGQVMQjqMItqQ,11309
-ragaai_catalyst/tracers/utils/trace_json_converter.py,sha256=-HZVmijeUFLO7e9OAvi1RJdWVTxPRUHPd1MkKQlCD54,11785
+ragaai_catalyst/tracers/utils/trace_json_converter.py,sha256=NPsxU04u6MCOMqisrgiAIv1bXFjWNwlrUn-LScC8f-s,12109
 ragaai_catalyst/tracers/utils/utils.py,sha256=o-p9n2ZuophdrV0wrixu-BqRHCkovup_klc3mS8mU8g,2374
-ragaai_catalyst-2.2.5b5.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-ragaai_catalyst-2.2.5b5.dist-info/METADATA,sha256=Oho91NQ2Otn-agTiKESdvBL1iZwLyQLeBtWTgu5nfc8,17735
-ragaai_catalyst-2.2.5b5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ragaai_catalyst-2.2.5b5.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
-ragaai_catalyst-2.2.5b5.dist-info/RECORD,,
+ragaai_catalyst-2.2.6b1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+ragaai_catalyst-2.2.6b1.dist-info/METADATA,sha256=qNKQxUD_sI5zPNeZFouH9xRsPdo1Wb0IxrLwAKB-ldg,17735
+ragaai_catalyst-2.2.6b1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ragaai_catalyst-2.2.6b1.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
+ragaai_catalyst-2.2.6b1.dist-info/RECORD,,