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

ragaai_catalyst/guard_executor.py

@@ -23,14 +23,15 @@ class GuardExecutor:
                 self.output_deployment_details = self.guard_manager.get_deployment(output_deployment_id)
             if input_deployment_id and output_deployment_id:
                 # check if 2 deployments are mapped to same dataset
-                pass
+                if self.input_deployment_details['data']['datasetId'] != self.output_deployment_details['data']['datasetId']:
+                    raise ValueError('Input deployment and output deployment should be mapped to same dataset')
             for guardrail in self.input_deployment_details['data']['guardrailsResponse']:
                 maps = guardrail['metricSpec']['config']['mappings']
                 for _map in maps:
                     if _map['schemaName']=='Response':
-                        raise ValueError('Response field mapping found in input guardrails')
+                        raise ValueError('Response field should be mapped only in output guardrails')
         except Exception as e:
-            raise ValueError('Error in fetching deployment details')
+            raise ValueError(str(e))
         self.base_url = guard_manager.base_url
         for key in field_map.keys():
             if key not in ['prompt','context','response','instruction']:
@@ -40,7 +41,8 @@ class GuardExecutor:
 
     def execute_deployment(self, deployment_id, payload):
         api = self.base_url + f'/guardrail/deployment/{deployment_id}/ingest'
-
+        if self.current_trace_id:
+            payload['traceId'] = self.current_trace_id
         payload = json.dumps(payload)
         headers = {
             'x-project-id': str(self.guard_manager.project_id),
@@ -182,6 +184,7 @@ class GuardExecutor:
             doc = self.set_variables(prompt,prompt_params)
         deployment_response = self.execute_deployment(self.output_deployment_id,doc)
         del self.id_2_doc[self.current_trace_id]
+        self.current_trace_id = None
         if deployment_response and deployment_response['data']['status'].lower() == 'fail':
             return deployment_response['data']['alternateResponse'], deployment_response
         elif deployment_response:

ragaai_catalyst/synthetic_data_generation.py

@@ -3,7 +3,7 @@ import ast
 import csv
 import json
 import random
-import PyPDF2
+import pypdf
 import markdown
 import pandas as pd
 from tqdm import tqdm
@@ -105,6 +105,9 @@ class SyntheticDataGeneration:
                     raise Exception(f"{e}")
 
                 else:
+                    if "'utf-8' codec can't encode characters" in str(e):
+                        print('Encountered non utf charactes, retrying with processed text')
+                        text = str(text.encode('utf-8',errors='ignore'))
                     print(f"Retrying...")
                     continue
         
@@ -469,7 +472,7 @@ class SyntheticDataGeneration:
         """
         text = ""
         with open(file_path, 'rb') as file:
-            pdf_reader = PyPDF2.PdfReader(file)
+            pdf_reader = pypdf.PdfReader(file)
             for page in pdf_reader.pages:
                 text += page.extract_text()
         return text

{ragaai_catalyst-2.1.5.1b1.dist-info → ragaai_catalyst-2.1.5.1b3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ragaai_catalyst
-Version: 2.1.5.1b1
+Version: 2.1.5.1b3
 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,>=3.9
@@ -22,7 +22,7 @@ Requires-Dist: langchain>=0.2.11
 Requires-Dist: openai>=1.57.0
 Requires-Dist: pandas
 Requires-Dist: groq>=0.11.0
-Requires-Dist: PyPDF2>=3.0.1
+Requires-Dist: pypdf>=5.3.1
 Requires-Dist: google-generativeai>=0.8.2
 Requires-Dist: google-genai>=1.3.0
 Requires-Dist: Markdown>=3.7

{ragaai_catalyst-2.1.5.1b1.dist-info → ragaai_catalyst-2.1.5.1b3.dist-info}/RECORD

@@ -3,14 +3,14 @@ ragaai_catalyst/_version.py,sha256=JKt9KaVNOMVeGs8ojO6LvIZr7ZkMzNN-gCcvryy4x8E,4
 ragaai_catalyst/dataset.py,sha256=YCj8Ovu6y38KEw-1HCe4xQWkmYPgfNTtMa8Q0g6B62o,29401
 ragaai_catalyst/evaluation.py,sha256=O96CydYVPh3duUmXjY6REIXMOR-tOPixSG-Qhrf636A,22955
 ragaai_catalyst/experiment.py,sha256=8yQo1phCHlpnJ-4CqCaIbLXg_1ZlAuLGI9kqGBl-OTE,18859
-ragaai_catalyst/guard_executor.py,sha256=WyosKLgEsAERJTtzs6wA4F1uDl8wKoiNZtTdyTPvOAw,13651
+ragaai_catalyst/guard_executor.py,sha256=rSdgf_3PB1Eaeoxz0cz6jbAoGvHXAeEfmM1jFUwL8cI,13970
 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=1FaeK_VZpJLQ1ZqEWpMyI8J8M2MI0abLLLDFWY9W-4A,19580
 ragaai_catalyst/redteaming_old.py,sha256=W2d89Ok8W-C8g7TBM3fDIFLof3q9FuYSr0jcryH2XQo,7097
-ragaai_catalyst/synthetic_data_generation.py,sha256=oWxV36wewwrggxXgvOZtzf-M5IOYalP1h5XjF6IOyY8,37597
+ragaai_catalyst/synthetic_data_generation.py,sha256=r7YKhVApkvc36uqeztggGH4DkPk9YblpAOpJO8LlW8k,37835
 ragaai_catalyst/utils.py,sha256=TlhEFwLyRU690HvANbyoRycR3nQ67lxVUQoUOfTPYQ0,3772
 ragaai_catalyst/redteaming/__init__.py,sha256=TJdvZpaZGFsg9qKONdjTosSVLZGadYFpHG6KE0xapKU,155
 ragaai_catalyst/redteaming/evaluator.py,sha256=C50SAc3RsR7PZnz-VQ7wQfDpiVEb7T3W3KV4Lj0tWYE,4599
@@ -90,8 +90,8 @@ ragaai_catalyst/tracers/utils/langchain_tracer_extraction_logic.py,sha256=XS2_x2
 ragaai_catalyst/tracers/utils/model_prices_and_context_window_backup.json,sha256=C3uwkibJ08C9sOX-54kulZYmJlIpZ-SQpfE6HNGrjbM,343502
 ragaai_catalyst/tracers/utils/trace_json_converter.py,sha256=qXSYKr4JMUpGQsB3mnr9_2qH6FqzUhCynNqlDp1IWTs,12440
 ragaai_catalyst/tracers/utils/utils.py,sha256=ViygfJ7vZ7U0CTSA1lbxVloHp4NSlmfDzBRNCJuMhis,2374
-ragaai_catalyst-2.1.5.1b1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-ragaai_catalyst-2.1.5.1b1.dist-info/METADATA,sha256=e0NQYAkSe9ivCqBqAFPv0zjz5IuS_PwOTFZUjB7BVKM,22057
-ragaai_catalyst-2.1.5.1b1.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
-ragaai_catalyst-2.1.5.1b1.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
-ragaai_catalyst-2.1.5.1b1.dist-info/RECORD,,
+ragaai_catalyst-2.1.5.1b3.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+ragaai_catalyst-2.1.5.1b3.dist-info/METADATA,sha256=-LuYUeg0C3QaUS8xS4K68nvwd4o9tXXx_ncJkxKEFC4,22056
+ragaai_catalyst-2.1.5.1b3.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
+ragaai_catalyst-2.1.5.1b3.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
+ragaai_catalyst-2.1.5.1b3.dist-info/RECORD,,