ragaai-catalyst 2.1.7.4b0__py3-none-any.whl → 2.1.7.4b2__py3-none-any.whl

ragaai_catalyst/tracers/tracer.py

@@ -859,3 +859,30 @@ class Tracer(AgenticTracing):
             self.user_context = context
         else:
             raise TypeError("context must be a string")
+    
+    def add_metadata(self, metadata):
+        """
+        Add metadata information to the trace. This method is only supported for 'langchain' and 'llamaindex' tracer types.
+
+        Args:
+            metadata: Additional metadata information to be added to the trace. Can be a dictionary.
+
+        Raises:
+            ValueError: If tracer_type is not 'langchain' or 'llamaindex'.
+        """
+        if self.tracer_type not in ["langchain", "llamaindex"]:
+            raise ValueError("add_metadata is only supported for 'langchain' and 'llamaindex' tracer types")
+        
+        # Convert string metadata to string if needed
+        user_details = self.user_details
+        user_metadata = user_details["trace_user_detail"]["metadata"]
+        if isinstance(metadata, dict):
+            for key, value in metadata.items():
+                if key in user_metadata:
+                    user_metadata[key] = value
+                else:
+                    raise ValueError(f"Key '{key}' not found in metadata")
+            self.dynamic_exporter.user_details = user_details
+            self.metadata = user_metadata
+        else:
+            raise TypeError("metadata must be a dictionary")

ragaai_catalyst/tracers/upload_traces.py

@@ -36,18 +36,22 @@ class UploadTraces:
             "model_name": {"columnType": "metadata"},
             "total_cost": {"columnType": "metadata", "dataType": "numerical"},
             "total_latency": {"columnType": "metadata", "dataType": "numerical"},
+            "error": {"columnType": "metadata"}
         }
 
         if additional_metadata_keys:
             for key in additional_metadata_keys:
                 if key == "model_name":
                     SCHEMA_MAPPING_NEW['response']["modelName"] = additional_metadata_keys[key]
+                elif key == "error":
+                    pass
                 else:
                     SCHEMA_MAPPING_NEW[key] = {"columnType": key, "parentColumn": "response"}
 
-        if "error" in additional_metadata_keys and additional_metadata_keys["error"]:
-            SCHEMA_MAPPING_NEW.pop("error", None)
-            SCHEMA_MAPPING_NEW["error"] = {"columnType": "metadata"}
+        if self.user_detail and self.user_detail["trace_user_detail"]["metadata"]:
+            for key in self.user_detail["trace_user_detail"]["metadata"]:
+                if key not in SCHEMA_MAPPING_NEW:
+                    SCHEMA_MAPPING_NEW[key] = {"columnType": "metadata"}
 
         if additional_pipeline_keys:
             for key in additional_pipeline_keys:

{ragaai_catalyst-2.1.7.4b0.dist-info → ragaai_catalyst-2.1.7.4b2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ragaai_catalyst
-Version: 2.1.7.4b0
+Version: 2.1.7.4b2
 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.4b0.dist-info → ragaai_catalyst-2.1.7.4b2.dist-info}/RECORD

@@ -31,8 +31,8 @@ ragaai_catalyst/tracers/distributed.py,sha256=MwlBwIxCAng-OI-7Ove_rkE1mTLeuW4Jw-
 ragaai_catalyst/tracers/langchain_callback.py,sha256=CB75zzG3-DkYTELj0vI1MOHQTY0MuQJfoHIXz9Cl8S8,34568
 ragaai_catalyst/tracers/llamaindex_callback.py,sha256=ZY0BJrrlz-P9Mg2dX-ZkVKG3gSvzwqBtk7JL_05MiYA,14028
 ragaai_catalyst/tracers/llamaindex_instrumentation.py,sha256=Ys_jLkvVqo12bKgXDmkp4TxJu9HkBATrFE8cIcTYxWw,14329
-ragaai_catalyst/tracers/tracer.py,sha256=LHRd35rTFgYovxsI4tyXHgsCCdccO1_O1txFLFyLUKc,39069
-ragaai_catalyst/tracers/upload_traces.py,sha256=appYayMqQ7av7UyY0WeAG-b9LCqKW1-vSbBCiJOpccs,6093
+ragaai_catalyst/tracers/tracer.py,sha256=M5pDK5I8smocgV_nAaLagB1ZpC-A8D8YW4peDf0ICNM,40265
+ragaai_catalyst/tracers/upload_traces.py,sha256=8T9j2FEUYEoMMU9hW8DbYRG7KtL-QAyCDucuD5gHUzA,6278
 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
@@ -88,8 +88,8 @@ ragaai_catalyst/tracers/utils/model_prices_and_context_window_backup.json,sha256
 ragaai_catalyst/tracers/utils/rag_trace_json_converter.py,sha256=adCKk7Nj8307XYYg2sB-QT-66OShOs2iTGwNVwqbHig,19373
 ragaai_catalyst/tracers/utils/trace_json_converter.py,sha256=E0_QfciQMMpCtQYrNB4l8HJhlaFalr5bkMqkVRgQahY,14073
 ragaai_catalyst/tracers/utils/utils.py,sha256=ViygfJ7vZ7U0CTSA1lbxVloHp4NSlmfDzBRNCJuMhis,2374
-ragaai_catalyst-2.1.7.4b0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-ragaai_catalyst-2.1.7.4b0.dist-info/METADATA,sha256=Eka_mwDBZZRkR9Nrum9EfD4cOZ3XTz9-DMmx-wi0g8E,17607
-ragaai_catalyst-2.1.7.4b0.dist-info/WHEEL,sha256=ooBFpIzZCPdw3uqIQsOo4qqbA4ZRPxHnOH7peeONza0,91
-ragaai_catalyst-2.1.7.4b0.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
-ragaai_catalyst-2.1.7.4b0.dist-info/RECORD,,
+ragaai_catalyst-2.1.7.4b2.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+ragaai_catalyst-2.1.7.4b2.dist-info/METADATA,sha256=ejPUvzAa3JiVbR6FCnOCbuvZbby17AanvPNW5xnLMaw,17607
+ragaai_catalyst-2.1.7.4b2.dist-info/WHEEL,sha256=ooBFpIzZCPdw3uqIQsOo4qqbA4ZRPxHnOH7peeONza0,91
+ragaai_catalyst-2.1.7.4b2.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
+ragaai_catalyst-2.1.7.4b2.dist-info/RECORD,,