ragaai-catalyst 2.2.3b2__py3-none-any.whl → 2.2.3b4__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/ragaai_catalyst.py

@@ -48,9 +48,6 @@ class RagaAICatalyst:
             logger.error(
                 "RAGAAI_CATALYST_ACCESS_KEY and RAGAAI_CATALYST_SECRET_KEY environment variables must be set"
             )
-            raise ValueError(
-                "RAGAAI_CATALYST_ACCESS_KEY and RAGAAI_CATALYST_SECRET_KEY environment variables must be set"
-            )
 
         RagaAICatalyst.access_key, RagaAICatalyst.secret_key = (
             self._set_access_key_secret_key(access_key, secret_key)
@@ -79,9 +76,7 @@ class RagaAICatalyst:
                 os.environ["RAGAAI_CATALYST_BASE_URL"] = RagaAICatalyst.BASE_URL
                 RagaAICatalyst.get_token(force_refresh=True)
             except requests.exceptions.RequestException:
-                raise ConnectionError(
-                    "The provided base_url is not accessible. Please re-check the base_url."
-                )
+                logger.error("The provided base_url is not accessible. Please re-check the base_url.")
         else:
             # Get the token from the server
             RagaAICatalyst.get_token(force_refresh=True)
@@ -264,7 +259,7 @@ class RagaAICatalyst:
             if response.status_code == 400:
                 token_response = response.json()
                 if token_response.get("message") == "Please enter valid credentials":
-                    raise Exception(
+                    logger.error(
                         "Authentication failed. Invalid credentials provided. Please check your Access key and Secret key. \nTo view or create new keys, navigate to Settings -> Authenticate in the RagaAI Catalyst dashboard."
                     )
 
@@ -379,13 +374,13 @@ class RagaAICatalyst:
         # Check if the project already exists
         existing_projects = self.list_projects()
         if project_name in existing_projects:
-            raise ValueError(
+            logger.error(
                 f"Project name '{project_name}' already exists. Please choose a different name."
             )
 
         usecase_list = self.project_use_cases()
         if usecase not in usecase_list:
-            raise ValueError(f"Select a valid usecase from {usecase_list}")
+            logger.error(f"Select a valid usecase from {usecase_list}")
 
         json_data = {"name": project_name, "type": type, "usecase": usecase}
         headers = {

ragaai_catalyst/tracers/tracer.py

@@ -365,20 +365,23 @@ 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]}")
+        logger.info("DEPRECATED: The set_model_cost method is deprecated and will be removed in a future version. Custom model costs can now be configured directly through the RagaAI Catalyst Platform")
+        print("DEPRECATED: The set_model_cost method is deprecated and will be removed in a future version. Custom model costs can now be configured directly through the RagaAI Catalyst Platform")
+        # 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):

{ragaai_catalyst-2.2.3b2.dist-info → ragaai_catalyst-2.2.3b4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ragaai_catalyst
-Version: 2.2.3b2
+Version: 2.2.3b4
 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>, Rishabh Pandey <rishabh.pandey@raga.ai>, Jyotsana C G <jyotsana@raga.ai>
 Requires-Python: <=3.13.2,>=3.10

{ragaai_catalyst-2.2.3b2.dist-info → ragaai_catalyst-2.2.3b4.dist-info}/RECORD

@@ -1,14 +1,14 @@
 ragaai_catalyst/__init__.py,sha256=2wfkucAbb3Bt_p2KHelkg9zBQp4yC4iZanltyieG18w,895
 ragaai_catalyst/_version.py,sha256=JKt9KaVNOMVeGs8ojO6LvIZr7ZkMzNN-gCcvryy4x8E,460
 ragaai_catalyst/dataset.py,sha256=H1zWw4nabiy8gsZ5msv6442VW5KBy4KTgxoqdH8r8J0,29368
-ragaai_catalyst/evaluation.py,sha256=tT86kBGE4HMMJoJO3kNZymFEdHv-Pyf0THplqOua0Xk,22940
+ragaai_catalyst/evaluation.py,sha256=8P2zUMSMsQGmKdv7_dZ8F0iXWYddvappgKPN5oJXWuY,22568
 ragaai_catalyst/experiment.py,sha256=8yQo1phCHlpnJ-4CqCaIbLXg_1ZlAuLGI9kqGBl-OTE,18859
 ragaai_catalyst/guard_executor.py,sha256=VLqVO1gAEBaovoQQUjdNkivH1dXACFTsuXQ1nzAALQQ,14008
 ragaai_catalyst/guardrails_manager.py,sha256=_VrARJ1udmCF8TklNKy7XTQUaM8ATDhTOAGDonBkFro,14245
 ragaai_catalyst/internal_api_completion.py,sha256=DdICI5yfEudiOAIC8L4oxH0Qz7kX-BZCdo9IWsi2gNo,2965
 ragaai_catalyst/prompt_manager.py,sha256=W8ypramzOprrJ7-22d5vkBXIuIQ8v9XAzKDGxKsTK28,16550
 ragaai_catalyst/proxy_call.py,sha256=CHxldeceZUaLU-to_hs_Kf1z_b2vHMssLS_cOBedu78,5499
-ragaai_catalyst/ragaai_catalyst.py,sha256=H7nhKuPi3pJH3QzO2dWshZ6FXgyZ5YRNOcB7MkXPRes,25413
+ragaai_catalyst/ragaai_catalyst.py,sha256=ZlcpOgJA9lVRi51YFy4dVfsxU0I79LJu0MnVI5BIL-c,25201
 ragaai_catalyst/redteaming_old.py,sha256=W2d89Ok8W-C8g7TBM3fDIFLof3q9FuYSr0jcryH2XQo,7097
 ragaai_catalyst/synthetic_data_generation.py,sha256=7lIWa3nwgW2-FlJrDaGxTN6OE4-dbbhLtKNOBQufhho,37952
 ragaai_catalyst/utils.py,sha256=TlhEFwLyRU690HvANbyoRycR3nQ67lxVUQoUOfTPYQ0,3772
@@ -31,7 +31,7 @@ ragaai_catalyst/tracers/distributed.py,sha256=CGPuOh4CsgEk428PPibieLaAG2Tt3BVygF
 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=NxGFTq6xnzb7opwcWNNYyyWtL9ausWUS0xe7p_ALbDk,42131
+ragaai_catalyst/tracers/tracer.py,sha256=3iC6eeuWRv3VXsCoeQrPSnEnNkuSH-wzqe4nGUecu1M,42575
 ragaai_catalyst/tracers/upload_traces.py,sha256=w1clGGfdOMpStUJX40NAlxe6dcFdN4pwcezyII0bGYA,6994
 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
@@ -88,8 +88,8 @@ ragaai_catalyst/tracers/utils/rag_extraction_logic_final.py,sha256=3ygkRT__lLDRf
 ragaai_catalyst/tracers/utils/rag_trace_json_converter.py,sha256=54IEZO-YRjUAahV5nw8KClXqTF1LhfDry_TsZ4KGow4,20467
 ragaai_catalyst/tracers/utils/trace_json_converter.py,sha256=pds1wuTv9VtwEXiMg-TWBtrK5-XReTcWQbJzWU9t3wE,11497
 ragaai_catalyst/tracers/utils/utils.py,sha256=o-p9n2ZuophdrV0wrixu-BqRHCkovup_klc3mS8mU8g,2374
-ragaai_catalyst-2.2.3b2.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-ragaai_catalyst-2.2.3b2.dist-info/METADATA,sha256=X2A7b0uKHOIub0AjPK6iciEhEn2fI-WDc-k9CwtJ1CY,17679
-ragaai_catalyst-2.2.3b2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ragaai_catalyst-2.2.3b2.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
-ragaai_catalyst-2.2.3b2.dist-info/RECORD,,
+ragaai_catalyst-2.2.3b4.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+ragaai_catalyst-2.2.3b4.dist-info/METADATA,sha256=ZdrndNumPtG0XijL_JgYJy5KncMlJfqAyBY4Kuf-fdg,17679
+ragaai_catalyst-2.2.3b4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ragaai_catalyst-2.2.3b4.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
+ragaai_catalyst-2.2.3b4.dist-info/RECORD,,