ragaai-catalyst 2.2.5b6__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:

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

@@ -9,7 +9,7 @@ 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__)

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ragaai_catalyst
-Version: 2.2.5b6
+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.5b6.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
@@ -35,12 +36,11 @@ 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=3VZjuW3OnElYE-VjhhT9ILvFSG8Rh0sH9n2M_Wds_TU,25146
-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=tojXeuTigDoLCdPvBKRBWtTHwjtm2tFjTizM7NknSo4,3887
+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
@@ -58,8 +58,8 @@ ragaai_catalyst/tracers/utils/model_prices_and_context_window_backup.json,sha256
 ragaai_catalyst/tracers/utils/rag_extraction_logic_final.py,sha256=3ygkRT__lLDRflRttjzPu28tIA8cTCiGQVMQjqMItqQ,11309
 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.5b6.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-ragaai_catalyst-2.2.5b6.dist-info/METADATA,sha256=kHQKCIyv8lTxRSQ7OCFbG_5WSUF83HcVyOmDVcVMCyg,17735
-ragaai_catalyst-2.2.5b6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ragaai_catalyst-2.2.5b6.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
-ragaai_catalyst-2.2.5b6.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,,