ragaai-catalyst 2.1.7.5b3__tar.gz → 2.1.7.5b4__tar.gz

{ragaai_catalyst-2.1.7.5b3/ragaai_catalyst.egg-info → ragaai_catalyst-2.1.7.5b4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ragaai_catalyst
-Version: 2.1.7.5b3
+Version: 2.1.7.5b4
 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>
 Requires-Python: <=3.13.2,>=3.10

{ragaai_catalyst-2.1.7.5b3 → ragaai_catalyst-2.1.7.5b4}/Quickstart.md

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

{ragaai_catalyst-2.1.7.5b3 → ragaai_catalyst-2.1.7.5b4}/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.1.7.5.beta.3"
+version = "2.1.7.5.beta.4"
 authors = [
     {name = "Kiran Scaria", email = "kiran.scaria@raga.ai"},
     {name = "Kedar Gaikwad", email = "kedar.gaikwad@raga.ai"},

{ragaai_catalyst-2.1.7.5b3 → ragaai_catalyst-2.1.7.5b4}/ragaai_catalyst/ragaai_catalyst.py

@@ -13,6 +13,7 @@ logging_level = (
 class RagaAICatalyst:
     BASE_URL = None
     TIMEOUT = 10  # Default timeout in seconds
+    TOKEN_EXPIRY_TIME = 6  # Default token expiration time (6 hours in hours)
 
     def __init__(
         self,
@@ -20,6 +21,7 @@ class RagaAICatalyst:
         secret_key,
         api_keys: Optional[Dict[str, str]] = None,
         base_url: Optional[str] = None,
+        token_expiry_time: Optional[float] = 6,
     ):
         """
         Initializes a new instance of the RagaAICatalyst class.
@@ -29,6 +31,7 @@ class RagaAICatalyst:
             secret_key (str): The secret key for the RagaAICatalyst.
             api_keys (Optional[Dict[str, str]]): A dictionary of API keys for different services. Defaults to None.
             base_url (Optional[str]): The base URL for the RagaAICatalyst API. Defaults to None.
+            token_expiry_time (Optional[float]): The time in hours before the token expires. Defaults to 0.1 hours.
 
         Raises:
             ValueError: If the RAGAAI_CATALYST_ACCESS_KEY and RAGAAI_CATALYST_SECRET_KEY environment variables are not set.
@@ -54,6 +57,9 @@ class RagaAICatalyst:
         self._token_expiry = None
         self._token_refresh_lock = threading.Lock()
         self._refresh_thread = None
+        
+        # Set token expiration time (convert hours to seconds)
+        RagaAICatalyst.TOKEN_EXPIRY_TIME = token_expiry_time * 60 * 60
 
         RagaAICatalyst.BASE_URL = (
             os.getenv("RAGAAI_CATALYST_BASE_URL")
@@ -149,9 +155,8 @@ class RagaAICatalyst:
         """Get the API key for a specific service."""
         return self.api_keys.get(service)
 
-    # Token expiration time in seconds (6 hours by default)
-    # TODO: Change this to 23 hours
-    TOKEN_EXPIRATION_TIME = 6 * 60 * 60
+    # Token expiration time is now configurable via the token_expiry_time parameter
+    # Default is 6 hours, but can be changed to 23 hours or any other value
 
     def _get_credentials(self) -> tuple[str, str]:
         """Get access key and secret key from instance or environment."""
@@ -165,6 +170,29 @@ class RagaAICatalyst:
             self.get_token(force_refresh=True)
         except Exception as e:
             logger.error(f"Background token refresh failed: {str(e)}")
+            
+    def _schedule_token_refresh(self):
+        """Schedule a token refresh to happen 20 seconds before expiration."""
+        if not self._token_expiry:
+            return
+            
+        # Calculate when to refresh (20 seconds before expiration)
+        current_time = time.time()
+        refresh_buffer = min(20, RagaAICatalyst.TOKEN_EXPIRY_TIME * 0.05)  # 20 seconds or 5% of expiry time, whichever is smaller
+        time_until_refresh = max(self._token_expiry - current_time - refresh_buffer, 1)  # At least 1 second
+        
+        def delayed_refresh():
+            # Sleep until it's time to refresh
+            time.sleep(time_until_refresh)
+            logger.debug(f"Scheduled token refresh triggered")
+            self._refresh_token_async()
+            
+        # Start a new thread for the delayed refresh
+        if not self._refresh_thread or not self._refresh_thread.is_alive():
+            self._refresh_thread = threading.Thread(target=delayed_refresh)
+            self._refresh_thread.daemon = True
+            self._refresh_thread.start()
+            logger.debug(f"Token refresh scheduled in {time_until_refresh:.1f} seconds")
 
     def get_token(self, force_refresh=False) -> Union[str, None]:
         """
@@ -226,8 +254,12 @@ class RagaAICatalyst:
             token = token_response.get("data", {}).get("token")
             if token:
                 os.environ["RAGAAI_CATALYST_TOKEN"] = token
-                self._token_expiry = time.time() + self.TOKEN_EXPIRATION_TIME
-                logger.info(f"Token refreshed successfully. Next refresh in {self.TOKEN_EXPIRATION_TIME/3600:.1f} hours")
+                self._token_expiry = time.time() + RagaAICatalyst.TOKEN_EXPIRY_TIME
+                logger.debug(f"Token refreshed successfully. Next refresh in {RagaAICatalyst.TOKEN_EXPIRY_TIME/3600:.1f} hours")
+                
+                # Schedule token refresh 20 seconds before expiration
+                self._schedule_token_refresh()
+                
                 return token
             else:
                 logger.error("Token(s) not set")
@@ -237,7 +269,7 @@ class RagaAICatalyst:
         """
         Ensures a valid token is available, with different handling for missing token vs expired token:
         - Missing token: Synchronous retrieval (fail fast)
-        - Expired token: Asynchronous refresh in background
+        - Expired token: Synchronous refresh (since token is needed immediately)
 
         Returns:
             - A string representing the valid token if successful.
@@ -250,13 +282,22 @@ class RagaAICatalyst:
         if not current_token:
             return self.get_token(force_refresh=True)
 
-        # Case 2: Token expired or about to expire - async refresh
+        # Case 2: Token expired - synchronous refresh (since we need a valid token now)
         if not self._token_expiry or current_time >= self._token_expiry:
+            logger.info("Token expired, refreshing synchronously")
+            return self.get_token(force_refresh=True)
+            
+        # Case 3: Token valid but approaching expiry (less than 10% of lifetime remaining)
+        # Start background refresh but return current token
+        token_remaining_time = self._token_expiry - current_time
+        if token_remaining_time < (RagaAICatalyst.TOKEN_EXPIRY_TIME * 0.1):
             if not self._refresh_thread or not self._refresh_thread.is_alive():
+                logger.info("Token approaching expiry, starting background refresh")
                 self._refresh_thread = threading.Thread(target=self._refresh_token_async)
                 self._refresh_thread.daemon = True
                 self._refresh_thread.start()
 
+        # Return current token (which is valid)
         return current_token
 
     def get_auth_header(self) -> Dict[str, str]:

{ragaai_catalyst-2.1.7.5b3 → ragaai_catalyst-2.1.7.5b4}/ragaai_catalyst/tracers/tracer.py

@@ -402,7 +402,50 @@ class Tracer(AgenticTracing):
         def recursive_mask_values(obj, parent_key=None):
             """Apply masking to all values in nested structure."""
             if isinstance(obj, dict):
-                return {k: recursive_mask_values(v, k) for k, v in obj.items()}
+                if self.tracer_type == "langchain":
+                    # Special handling for LangChain data
+                    if isinstance(obj, dict):
+                        if obj.get("name", "") == "retrieve_documents.langchain.workflow":
+                            prompt_structured_data = {
+                                "traceloop.entity.input": json.dumps({
+                                    "kwargs": {
+                                        "input": masking_func(json.loads(obj.get("attributes", {}).get("traceloop.entity.input", "")).get("kwargs", {}).get("input", "")),
+                                    }
+                                })
+                            }     
+                            prompt_data = {
+                                "name": "retrieve_documents.langchain.workflow",
+                                "attributes": prompt_structured_data,
+                            }
+                            return prompt_data
+                        elif obj.get("name", "") == "PromptTemplate.langchain.task":
+                            context_structured_data = {
+                                "traceloop.entity.input": json.dumps({
+                                    "kwargs": {
+                                        "context": masking_func(json.loads(obj.get("attributes", {}).get("traceloop.entity.input", "")).get("kwargs", {}).get("context", "")),
+                                    }
+                                }),
+                                "traceloop.entity.output": json.dumps({
+                                    "kwargs": {
+                                        "text": masking_func(json.loads(obj.get("attributes", {}).get("traceloop.entity.output", "")).get("kwargs", {}).get("text", "")),
+                                    }
+                                })
+                            }
+                            context_data = {
+                                "name": "PromptTemplate.langchain.task",
+                                "attributes": context_structured_data,
+                            }
+                            return context_data
+                        elif obj.get("name", "") == "ChatOpenAI.langchain.task":
+                            response_structured_data = {"gen_ai.completion.0.content": masking_func(obj.get("attributes", {}).get("gen_ai.completion.0.content", "")),
+                                                        "gen_ai.prompt.0.content": masking_func(obj.get("attributes", {}).get("gen_ai.prompt.0.content", ""))}
+                            response_data = {
+                                "name": "ChatOpenAI.langchain.task",
+                                "attributes" : response_structured_data
+                            }
+                            return response_data
+                else:
+                    return {k: recursive_mask_values(v, k) for k, v in obj.items()}
             elif isinstance(obj, list):
                 return [recursive_mask_values(item, parent_key) for item in obj]
             elif isinstance(obj, str):

{ragaai_catalyst-2.1.7.5b3 → ragaai_catalyst-2.1.7.5b4/ragaai_catalyst.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ragaai_catalyst
-Version: 2.1.7.5b3
+Version: 2.1.7.5b4
 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>
 Requires-Python: <=3.13.2,>=3.10