ragaai-catalyst 2.2.3b1__py3-none-any.whl → 2.2.3b3__py3-none-any.whl

ragaai_catalyst/evaluation.py

@@ -506,13 +506,6 @@ class Evaluation:
         response_text = parse_response()
         if response_text:
             df = pd.read_csv(io.StringIO(response_text))
-
-            column_list = df.columns.to_list()
-            # Remove unwanted columns
-            column_list = [col for col in column_list if not col.startswith('_')]
-            column_list = [col for col in column_list if '.' not in col]
-            # Remove _claims_ columns
-            column_list = [col for col in column_list if '_claims_' not in col]
-            return df[column_list]
+            return df
         else:
             return pd.DataFrame()

ragaai_catalyst/tracers/tracer.py

@@ -365,20 +365,21 @@ class Tracer(AgenticTracing):
                 "output_cost_per_million_token": 2.40
             })
         """
-        if not isinstance(cost_config, dict):
-            logger.error("cost_config must be a dictionary")
-
-        required_keys = {"model_name", "input_cost_per_million_token", "output_cost_per_million_token"}
-        if not all(key in cost_config for key in required_keys):
-            logger.error(f"cost_config must contain all required keys: {required_keys}")
-
-        model_name = cost_config["model_name"]
-        self.model_custom_cost[model_name] = {
-            "input_cost_per_token": float(cost_config["input_cost_per_million_token"])/ 1000000,
-            "output_cost_per_token": float(cost_config["output_cost_per_million_token"]) /1000000
-        }
-        self.dynamic_exporter.custom_model_cost = self.model_custom_cost
-        logger.info(f"Updated custom model cost for {model_name}: {self.model_custom_cost[model_name]}")
+        # if not isinstance(cost_config, dict):
+        #     logger.error("cost_config must be a dictionary")
+
+        # required_keys = {"model_name", "input_cost_per_million_token", "output_cost_per_million_token"}
+        # if not all(key in cost_config for key in required_keys):
+        #     logger.error(f"cost_config must contain all required keys: {required_keys}")
+
+        # model_name = cost_config["model_name"]
+        # self.model_custom_cost[model_name] = {
+        #     "input_cost_per_token": float(cost_config["input_cost_per_million_token"])/ 1000000,
+        #     "output_cost_per_token": float(cost_config["output_cost_per_million_token"]) /1000000
+        # }
+        # self.dynamic_exporter.custom_model_cost = self.model_custom_cost
+        # logger.info(f"Updated custom model cost for {model_name}: {self.model_custom_cost[model_name]}")
+        return None
         
 
     def register_masking_function(self, masking_func):