ragaai-catalyst 2.2.3b2__tar.gz → 2.2.3b4__tar.gz

{ragaai_catalyst-2.2.3b2 → ragaai_catalyst-2.2.3b4}/.gitignore

@@ -188,10 +188,10 @@ legal_rag.py
 load_test_streaming.py
 load_test_v1.py
 load_test_v2.py
-locustfile.py
+# locustfile.py
 masking_example.py
 model_prices_and_context_window_backup.json
-multi_llm_test.py
+# multi_llm_test.py
 rag_agent_traces.json
 rag_agent_traces_with_streaming.json
 rag_agent_traces_wo_streaming.json
@@ -202,4 +202,11 @@ scifi_writer.py
 stress_test_langchain.py
 test.ipynb
 test_token_refresh.py
-uplaod_numerical_col_csv.py
+uplaod_numerical_col_csv.py
+dh_full_test.py
+model_gemini_15_flash_002_w_stream.json
+model_gemini_15_flash_002_wo_stream.json
+model_gpt_4o_mini_w_stream.json
+model_gpt_4o_mini_wo_stream.json
+model_gpt_41_w_stream.json
+model_gpt_41_wo_stream.json

{ragaai_catalyst-2.2.3b2/ragaai_catalyst.egg-info → ragaai_catalyst-2.2.3b4}/PKG-INFO

@@ -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 → ragaai_catalyst-2.2.3b4}/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.3.beta.2"
+version = "2.2.3.beta.4"
 authors = [
     {name = "Kiran Scaria", email = "kiran.scaria@raga.ai"},
     {name = "Kedar Gaikwad", email = "kedar.gaikwad@raga.ai"},

{ragaai_catalyst-2.2.3b2 → ragaai_catalyst-2.2.3b4}/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-2.2.3b2 → ragaai_catalyst-2.2.3b4}/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-2.2.3b2 → ragaai_catalyst-2.2.3b4}/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 → ragaai_catalyst-2.2.3b4/ragaai_catalyst.egg-info}/PKG-INFO

@@ -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