ragaai-catalyst 2.2b0__tar.gz → 2.2b1__tar.gz

{ragaai_catalyst-2.2b0/ragaai_catalyst.egg-info → ragaai_catalyst-2.2b1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ragaai_catalyst
-Version: 2.2b0
+Version: 2.2b1
 Summary: RAGA AI CATALYST
 Author-email: Kiran Scaria <kiran.scaria@raga.ai>, Kedar Gaikwad <kedar.gaikwad@raga.ai>, Dushyant Mahajan <dushyant.mahajan@raga.ai>, Siddhartha Kosti <siddhartha.kosti@raga.ai>, Ritika Goel <ritika.goel@raga.ai>, Vijay Chaurasia <vijay.chaurasia@raga.ai>, Tushar Kumar <tushar.kumar@raga.ai>
 Requires-Python: <=3.13.2,>=3.10

{ragaai_catalyst-2.2b0 → ragaai_catalyst-2.2b1}/pyproject.toml

@@ -8,7 +8,7 @@ description = "RAGA AI CATALYST"
 readme = "README.md"
 requires-python = ">=3.10,<=3.13.2"
 # license = {file = "LICENSE"}
-version = "2.2.beta.0"
+version = "2.2.beta.1"
 authors = [
     {name = "Kiran Scaria", email = "kiran.scaria@raga.ai"},
     {name = "Kedar Gaikwad", email = "kedar.gaikwad@raga.ai"},

{ragaai_catalyst-2.2b0 → ragaai_catalyst-2.2b1}/ragaai_catalyst/tracers/tracer.py

@@ -404,66 +404,27 @@ class Tracer(AgenticTracing):
 
         def recursive_mask_values(obj, parent_key=None):
             """Apply masking to all values in nested structure."""
-            if isinstance(obj, dict):
-                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:
+            try:
+                if isinstance(obj, dict):
                     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):
-                # List of keys that should NOT be masked
-                excluded_keys = {
-                    'start_time', 'end_time', 'name', 'id', 
-                    'hash_id', 'parent_id', 'source_hash_id',
-                    'cost', 'type', 'feedback', 'error', 'ctx','telemetry.sdk.version',
-                    'telemetry.sdk.language','service.name'
-                }
-                # Apply masking only if the key is NOT in the excluded list
-                if parent_key and parent_key.lower() not in excluded_keys:
-                    return masking_func(obj)
-                return obj
-            else:
+                elif isinstance(obj, list):
+                    return [recursive_mask_values(item, parent_key) for item in obj]
+                elif isinstance(obj, str):
+                    # List of keys that should NOT be masked
+                    excluded_keys = {
+                        'start_time', 'end_time', 'name', 'id', 
+                        'hash_id', 'parent_id', 'source_hash_id',
+                        'cost', 'type', 'feedback', 'error', 'ctx','telemetry.sdk.version',
+                        'telemetry.sdk.language','service.name'
+                    }
+                    # Apply masking only if the key is NOT in the excluded list
+                    if parent_key and parent_key.lower() not in excluded_keys:
+                        return masking_func(obj)
+                    return obj
+                else:
+                    return obj
+            except Exception as e:
+                logger.error(f"Error masking value: {e}")
                 return obj
 
         def file_post_processor(original_trace_json_path: os.PathLike) -> os.PathLike:
@@ -953,4 +914,4 @@ class Tracer(AgenticTracing):
             self.dynamic_exporter.user_details = user_details
             self.metadata = user_metadata
         else:
-            logger.warning("metadata must be a dictionary")
+            logger.warning("metadata must be a dictionary")

{ragaai_catalyst-2.2b0 → ragaai_catalyst-2.2b1/ragaai_catalyst.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ragaai_catalyst
-Version: 2.2b0
+Version: 2.2b1
 Summary: RAGA AI CATALYST
 Author-email: Kiran Scaria <kiran.scaria@raga.ai>, Kedar Gaikwad <kedar.gaikwad@raga.ai>, Dushyant Mahajan <dushyant.mahajan@raga.ai>, Siddhartha Kosti <siddhartha.kosti@raga.ai>, Ritika Goel <ritika.goel@raga.ai>, Vijay Chaurasia <vijay.chaurasia@raga.ai>, Tushar Kumar <tushar.kumar@raga.ai>
 Requires-Python: <=3.13.2,>=3.10