ragaai-catalyst 1.0.7b3__py3-none-any.whl → 1.0.7b4__py3-none-any.whl

ragaai_catalyst/experiment.py

@@ -339,11 +339,51 @@ class Experiment:
         Parse the response to get the results
         """
         try:
-            parsed_response_df = pd.DataFrame(response["data"]["docs"])
-            return True, parsed_response_df
+            x = pd.DataFrame(response["data"]["docs"])
+
+            column_names_to_replace = [
+                {item["columnName"]: item["displayName"]}
+                for item in response["data"]["columns"]
+            ]
+
+            if column_names_to_replace:
+                for item in column_names_to_replace:
+                    x = x.rename(columns=item)
+
+                dict_cols = [
+                    col
+                    for col in x.columns
+                    if x[col].dtype == "object"
+                    and x[col].apply(lambda y: isinstance(y, dict)).any()
+                ]
+
+                for dict_col in dict_cols:
+                    x[f"{dict_col}_reason"] = x[dict_col].apply(
+                        lambda y: y.get("reason") if isinstance(y, dict) else None
+                    )
+                    x[f"{dict_col}_metric_config"] = x[dict_col].apply(
+                        lambda y: (
+                            y.get("metric_config") if isinstance(y, dict) else None
+                        )
+                    )
+                    x[f"{dict_col}_passed"] = x[dict_col].apply(
+                        lambda y: y.get("passed") if isinstance(y, dict) else None
+                    )
+                    x[f"{dict_col}_status"] = x[dict_col].apply(
+                        lambda y: y.get("status") if isinstance(y, dict) else None
+                    )
+
+                    x = x.drop(columns=[dict_col])
+
+            x.columns = x.columns.str.replace("_reason_reason", "_reason")
+            x.columns = x.columns.str.replace("_reason_metric_config", "_metric_config")
+            x.columns = x.columns.str.replace("_reason_passed", "_passed")
+            x.columns = x.columns.str.replace("_reason_status", "_status")
+            return True, x
+
         except Exception as e:
-            logger.error(f"Failed to parse response: {e}")
-            return False, ""
+            logger.error(f"Failed to parse response: {e}", exc_info=True)
+            return False, pd.DataFrame()
 
 
 class FailedToRetrieveResults(Exception):

{ragaai_catalyst-1.0.7b3.dist-info → ragaai_catalyst-1.0.7b4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ragaai_catalyst
-Version: 1.0.7b3
+Version: 1.0.7b4
 Summary: RAGA AI CATALYST
 Author-email: Kiran Scaria <kiran.scaria@raga.ai>, Kedar Gaikwad <kedar.gaikwad@raga.ai>, Dushyant Mahajan <dushyant.mahajan@raga.ai>
 Requires-Python: >=3.9

{ragaai_catalyst-1.0.7b3.dist-info → ragaai_catalyst-1.0.7b4.dist-info}/RECORD

@@ -1,7 +1,7 @@
 ragaai_catalyst/__init__.py,sha256=0hf_H1xkhrpGnUWTbjvYOXXQta1S4JO2yud1Eqydt-I,226
 ragaai_catalyst/_version.py,sha256=JKt9KaVNOMVeGs8ojO6LvIZr7ZkMzNN-gCcvryy4x8E,460
 ragaai_catalyst/dataset.py,sha256=g0YH0QGDpCiJ4VULB3tukGFNLpWGw0ZRwLMP07ay2IE,3560
-ragaai_catalyst/experiment.py,sha256=2TOyHknEcNuWbDDO6cgDVAWc6wuhSqGCGYfKDOndqts,13161
+ragaai_catalyst/experiment.py,sha256=7b81BGzZroUArS6UIsluNR6NWJSOcH8M4tFKiRRSoKc,14862
 ragaai_catalyst/ragaai_catalyst.py,sha256=DqEApmjxQufZZT4ap-IzoBmp3AX6RWMq7HXQ5kO0wsI,16468
 ragaai_catalyst/utils.py,sha256=TlhEFwLyRU690HvANbyoRycR3nQ67lxVUQoUOfTPYQ0,3772
 ragaai_catalyst/tracers/__init__.py,sha256=NppmJhD3sQ5R1q6teaZLS7rULj08Gb6JT8XiPRIe_B0,49
@@ -15,7 +15,7 @@ ragaai_catalyst/tracers/instrumentators/llamaindex.py,sha256=SMrRlR4xM7k9HK43hak
 ragaai_catalyst/tracers/instrumentators/openai.py,sha256=14R4KW9wQCR1xysLfsP_nxS7cqXrTPoD8En4MBAaZUU,379
 ragaai_catalyst/tracers/utils/__init__.py,sha256=KeMaZtYaTojilpLv65qH08QmpYclfpacDA0U3wg6Ybw,64
 ragaai_catalyst/tracers/utils/utils.py,sha256=ViygfJ7vZ7U0CTSA1lbxVloHp4NSlmfDzBRNCJuMhis,2374
-ragaai_catalyst-1.0.7b3.dist-info/METADATA,sha256=RBab5g5fSrrXqD4a-VWIuXdGQ33fZbE2qI4PUtLgTM8,5121
-ragaai_catalyst-1.0.7b3.dist-info/WHEEL,sha256=UvcQYKBHoFqaQd6LKyqHw9fxEolWLQnlzP0h_LgJAfI,91
-ragaai_catalyst-1.0.7b3.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
-ragaai_catalyst-1.0.7b3.dist-info/RECORD,,
+ragaai_catalyst-1.0.7b4.dist-info/METADATA,sha256=AcE3Fj9TqlB4if-kqn70Kd0NPCae_8I1_FiS0iSe0tw,5121
+ragaai_catalyst-1.0.7b4.dist-info/WHEEL,sha256=UvcQYKBHoFqaQd6LKyqHw9fxEolWLQnlzP0h_LgJAfI,91
+ragaai_catalyst-1.0.7b4.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
+ragaai_catalyst-1.0.7b4.dist-info/RECORD,,