ragaai-catalyst 2.1.5.1b1__tar.gz → 2.1.5.1b3__tar.gz

{ragaai_catalyst-2.1.5.1b1/ragaai_catalyst.egg-info → ragaai_catalyst-2.1.5.1b3}/PKG-INFO

@@ -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 → ragaai_catalyst-2.1.5.1b3}/pyproject.toml

@@ -8,7 +8,7 @@ description = "RAGA AI CATALYST"
 readme = "README.md"
 requires-python = ">=3.9,<3.13"
 # license = {file = "LICENSE"}
-version = "2.1.5.1.b1"
+version = "2.1.5.1.b3"
 authors = [
     {name = "Kiran Scaria", email = "kiran.scaria@raga.ai"},
     {name = "Kedar Gaikwad", email = "kedar.gaikwad@raga.ai"},
@@ -36,7 +36,7 @@ dependencies = [
     "openai>=1.57.0",
     "pandas",
     "groq>=0.11.0",
-    "PyPDF2>=3.0.1",
+    "pypdf>=5.3.1",
     "google-generativeai>=0.8.2",
     "google-genai>=1.3.0",
     "Markdown>=3.7",  

{ragaai_catalyst-2.1.5.1b1 → ragaai_catalyst-2.1.5.1b3}/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-2.1.5.1b1 → ragaai_catalyst-2.1.5.1b3}/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 → ragaai_catalyst-2.1.5.1b3/ragaai_catalyst.egg-info}/PKG-INFO

@@ -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 → ragaai_catalyst-2.1.5.1b3}/ragaai_catalyst.egg-info/requires.txt

@@ -14,7 +14,7 @@ langchain>=0.2.11
 openai>=1.57.0
 pandas
 groq>=0.11.0
-PyPDF2>=3.0.1
+pypdf>=5.3.1
 google-generativeai>=0.8.2
 google-genai>=1.3.0
 Markdown>=3.7

{ragaai_catalyst-2.1.5.1b1 → ragaai_catalyst-2.1.5.1b3}/requirements.txt

@@ -68,7 +68,7 @@ pyasn1_modules==0.4.1
 pydantic==2.10.4
 pydantic_core==2.27.2
 pyparsing==3.2.0
-PyPDF2==3.0.1
+pypdf==5.3.1
 python-dateutil==2.9.0.post0
 python-dotenv==1.0.1
 pytz==2024.2