ragaai-catalyst 2.0.6b1__py3-none-any.whl → 2.0.7b1__py3-none-any.whl

ragaai_catalyst/dataset.py

@@ -16,7 +16,7 @@ class Dataset:
 
     def __init__(self, project_name):
         self.project_name = project_name
-        self.num_projects = 100
+        self.num_projects = 99999
         Dataset.BASE_URL = (
             os.getenv("RAGAAI_CATALYST_BASE_URL")
             if os.getenv("RAGAAI_CATALYST_BASE_URL")

ragaai_catalyst/evaluation.py

@@ -16,7 +16,7 @@ class Evaluation:
         self.base_url = f"{RagaAICatalyst.BASE_URL}"
         self.timeout = 10
         self.jobId = None
-        self.num_projects=100
+        self.num_projects=99999
 
         try:
             response = requests.get(

ragaai_catalyst/guardrails_manager.py

@@ -13,7 +13,7 @@ class GuardrailsManager:
         """
         self.project_name = project_name
         self.timeout = 10
-        self.num_projects = 100
+        self.num_projects = 99999
         self.deployment_name = "NA"
         self.deployment_id = "NA"
         self.base_url = f"{RagaAICatalyst.BASE_URL}"
@@ -31,7 +31,7 @@ class GuardrailsManager:
         :return: A tuple containing a list of project names and a list of dictionaries with project IDs and names.
         """
         headers = {'Authorization': f'Bearer {os.getenv("RAGAAI_CATALYST_TOKEN")}'}
-        response = requests.request("GET", f"{self.base_url}/v2/llm/projects?size=12&page=0", headers=headers, timeout=self.timeout)
+        response = requests.request("GET", f"{self.base_url}/v2/llm/projects?size={self.num_projects}", headers=headers, timeout=self.timeout)
         project_content = response.json()["data"]["content"]
         list_project = [_["name"] for _ in project_content]
         project_name_with_id = [{"id": _["id"], "name": _["name"]} for _ in project_content]

ragaai_catalyst/internal_api_completion.py

@@ -36,7 +36,7 @@ def api_completion(messages, model_config, kwargs):
                     raise ValueError(response["error"]["message"])
                 else:
                     result=  response["choices"][0]["message"]["content"]
-                    response1 = result.replace('\n', '')
+                    response1 = result.replace('\n', '').replace('```json','').replace('```', '').strip()
                     try:
                         json_data = json.loads(response1)
                         df = pd.DataFrame(json_data)

ragaai_catalyst/prompt_manager.py

@@ -23,7 +23,7 @@ class PromptManager:
         self.project_name = project_name
         self.base_url = f"{RagaAICatalyst.BASE_URL}/playground/prompt"
         self.timeout = 10
-        self.size = 100 #Number of projects to fetch
+        self.size = 99999 #Number of projects to fetch
 
         try:
             response = requests.get(
@@ -432,7 +432,12 @@ class PromptObject:
         Returns:
             dict: A dictionary of parameters found in the prompt text.
         """
-        parameters = {param["name"]: self._convert_value(param["value"], param["type"]) for param in self.parameters}
+        parameters = {}
+        for param in self.parameters:
+            if "value" in param:
+                parameters[param["name"]] = self._convert_value(param["value"], param["type"])
+            else:
+                parameters[param["name"]] = ""
         parameters["model"] = self.model
         return parameters    
     

ragaai_catalyst/ragaai_catalyst.py

@@ -287,7 +287,7 @@ class RagaAICatalyst:
     def get_project_id(self, project_name):
         pass
 
-    def list_projects(self, num_projects=100):
+    def list_projects(self, num_projects=99999):
         """
         Retrieves a list of projects with the specified number of projects.
 

ragaai_catalyst/synthetic_data_generation.py

@@ -289,6 +289,7 @@ class SyntheticDataGeneration:
 
         # Extract the content from the response
         content = response.choices[0].message.content
+        content = content.replace('\n', '').replace('```json','').replace('```', '').strip()
 
         # Clean the response if needed (remove any prefix before the JSON list)
         list_start_index = content.find('[')

{ragaai_catalyst-2.0.6b1.dist-info → ragaai_catalyst-2.0.7b1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ragaai_catalyst
-Version: 2.0.6b1
+Version: 2.0.7b1
 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>
 Requires-Python: >=3.9
@@ -28,6 +28,7 @@ Requires-Dist: litellm==1.51.1
 Requires-Dist: tenacity==8.3.0
 Requires-Dist: tqdm>=4.66.5
 Requires-Dist: llama-index==0.10.0
+Requires-Dist: pyopenssl==24.2.1
 Provides-Extra: dev
 Requires-Dist: pytest; extra == "dev"
 Requires-Dist: pytest-cov; extra == "dev"
@@ -38,7 +39,7 @@ Requires-Dist: flake8; extra == "dev"
 
 # RagaAI Catalyst
 
-RagaAI Catalyst is a powerful tool for managing and optimizing LLM projects. It provides functionalities for project management, trace recording, and experiment management, allowing you to fine-tune and evaluate your LLM applications effectively.
+RagaAI Catalyst is a comprehensive platform designed to enhance the management and optimization of LLM projects. It offers a wide range of features, including project management, dataset management, evaluation management, trace management, prompt management, synthetic data generation, and guardrail management. These functionalities enable you to efficiently evaluate, and safeguard your LLM applications.
 
 ## Table of Contents
 
@@ -53,6 +54,7 @@ RagaAI Catalyst is a powerful tool for managing and optimizing LLM projects. It
     - [Trace Management](#trace-management)
     - [Prompt Management](#prompt-management)
     - [Synthetic Data Generation](#synthetic-data-generation)
+    - [Guardrail Management](#guardrail-management)
 
 ## Installation
 
@@ -144,6 +146,7 @@ evaluation = Evaluation(
 evaluation.list_metrics()
 
 # Add metrics to the experiment
+
 schema_mapping={
     'Query': 'prompt',
     'response': 'response',
@@ -201,8 +204,12 @@ tracer = Tracer(
 
 # Your code here
 
+
 # Stop the trace recording
 tracer.stop()
+
+# Get upload status
+tracer.get_upload_status()
 ```
 
 
@@ -237,7 +244,7 @@ print("variable:",variable)
 prompt_content = prompt.get_prompt_content()
 print("prompt_content:", prompt_content)
 
-# Compile a prompt with variables
+# Compile the prompt with variables
 compiled_prompt = prompt.compile(query="What's the weather?", context="sunny", llm_response="It's sunny today")
 print("Compiled prompt:", compiled_prompt)
 
@@ -280,7 +287,9 @@ sdg = SyntheticDataGeneration()
 text = sdg.process_document(input_data="file_path")
 
 # Generate results
-result = sdg.generate_qna(text, question_type ='simple',model_config={"provider":"openai","model":"gpt-4o-mini"},n=20)
+result = sdg.generate_qna(text, question_type ='complex',model_config={"provider":"openai","model":"openai/gpt-3.5-turbo"},n=5)
+
+print(result.head())
 
 # Get supported Q&A types
 sdg.get_supported_qna()
@@ -291,5 +300,87 @@ sdg.get_supported_providers()
 
 
 
+### Guardrail Management
+
+```py
+from ragaai_catalyst import GuardrailsManager
+
+# Initialize Guardrails Manager
+gdm = GuardrailsManager(project_name=project_name)
+
+# Get list of Guardrails available
+guardrails_list = gdm.list_guardrails()
+print('guardrails_list:', guardrails_list)
+
+# Get list of fail condition for guardrails
+fail_conditions = gdm.list_fail_condition()
+print('fail_conditions;', fail_conditions)
+
+#Get list of deployment ids
+deployment_list = gdm.list_deployment_ids()
+print('deployment_list:', deployment_list)
+
+# Get specific deployment id with guardrails information
+deployment_id_detail = gdm.get_deployment(17)
+print('deployment_id_detail:', deployment_id_detail)
+
+# Add guardrails to a deployment id
+guardrails_config = {"guardrailFailConditions": ["FAIL"],
+                     "deploymentFailCondition": "ALL_FAIL",
+                     "alternateResponse": "Your alternate response"}
+
+guardrails = [
+    {
+      "displayName": "Response_Evaluator",
+      "name": "Response Evaluator",
+      "config":{
+          "mappings": [{
+                        "schemaName": "Text",
+                        "variableName": "Response"
+                    }],
+          "params": {
+                    "isActive": {"value": False},
+                    "isHighRisk": {"value": True},
+                    "threshold": {"eq": 0},
+                    "competitors": {"value": ["Google","Amazon"]}
+                }
+      }
+    },
+    {
+      "displayName": "Regex_Check",
+      "name": "Regex Check",
+      "config":{
+          "mappings": [{
+                        "schemaName": "Text",
+                        "variableName": "Response"
+                    }],
+          "params":{
+              "isActive": {"value": False},
+              "isHighRisk": {"value": True},
+              "threshold": {"lt1": 1}
+          }
+      }
+    }
+]
+
+gdm.add_guardrails(deployment_id, guardrails, guardrails_config)
 
 
+# Import GuardExecutor
+from ragaai_catalyst import GuardExecutor
+
+# Initialise GuardExecutor with required params and Evaluate
+executor = GuardExecutor(deployment_id,gdm,field_map={'context':'document'})
+
+
+message={'role':'user',
+         'content':'What is the capital of France'
+        }
+prompt_params={'document':' France'}
+
+model_params = {'temperature':.7,'model':'gpt-4o-mini'}
+llm_caller = 'litellm'
+
+executor([message],prompt_params,model_params,llm_caller)
+
+```

{ragaai_catalyst-2.0.6b1.dist-info → ragaai_catalyst-2.0.7b1.dist-info}/RECORD

@@ -1,15 +1,15 @@
 ragaai_catalyst/__init__.py,sha256=BdIJ_UUre0uEnRTsLw_hE0C0muWk6XWNZqdVOel22R4,537
 ragaai_catalyst/_version.py,sha256=JKt9KaVNOMVeGs8ojO6LvIZr7ZkMzNN-gCcvryy4x8E,460
-ragaai_catalyst/dataset.py,sha256=bDNZkcji22sg-zJqMHEwueTO8A2f_GJu70WcEHESwQk,10729
-ragaai_catalyst/evaluation.py,sha256=ZS5G5RjmATjljQhAKYCrDXW2mUNXscpRRoL8cseDjAA,20283
+ragaai_catalyst/dataset.py,sha256=j_vu3Xkp_8qYW0J9Qnn53Uyh98MsugqYl5zxhOv9EOg,10731
+ragaai_catalyst/evaluation.py,sha256=EZGISQH6vGB7vP5OmUBRXmEqkqZPIYN_eEUUz1jTrJ8,20285
 ragaai_catalyst/experiment.py,sha256=8KvqgJg5JVnt9ghhGDJvdb4mN7ETBX_E5gNxBT0Nsn8,19010
 ragaai_catalyst/guard_executor.py,sha256=llPbE3DyVtrybojXknzBZj8-dtUrGBQwi9-ZiPJxGRo,3762
-ragaai_catalyst/guardrails_manager.py,sha256=EsuQuZOPraLzZMJD502B1-wb642Sm3vqxWM4Nv9Q8Jc,11892
-ragaai_catalyst/internal_api_completion.py,sha256=51YwXcas5NviC1wjr8EX5Y6BOyTbJ4FlKHM8gE46Wtk,2916
-ragaai_catalyst/prompt_manager.py,sha256=ZMIHrmsnPMq20YfeNxWXLtrxnJyMcxpeJ8Uya7S5dUA,16411
+ragaai_catalyst/guardrails_manager.py,sha256=DILMOAASK57FH9BLq_8yC1AQzRJ8McMFLwCXgYwNAd4,11904
+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=5Q1VCE7P33DtjaOtVGRUgBL8dpDL9kjisWGIkOyX4nE,17426
-ragaai_catalyst/synthetic_data_generation.py,sha256=957UYz58uX13i8vn24rzZief5FgtfOEnEH7S8VtXtVw,19157
+ragaai_catalyst/ragaai_catalyst.py,sha256=FdqMzwuQLqS2-3JJDsTQ8uh2itllOxfPrRUjb8Kwmn0,17428
+ragaai_catalyst/synthetic_data_generation.py,sha256=uDV9tNwto2xSkWg5XHXUvjErW-4P34CTrxaJpRfezyA,19250
 ragaai_catalyst/utils.py,sha256=TlhEFwLyRU690HvANbyoRycR3nQ67lxVUQoUOfTPYQ0,3772
 ragaai_catalyst/tracers/__init__.py,sha256=NppmJhD3sQ5R1q6teaZLS7rULj08Gb6JT8XiPRIe_B0,49
 ragaai_catalyst/tracers/llamaindex_callback.py,sha256=vPE7MieKjfwLrLUnnPs20Df0xNYqoCCj-Mt2NbiuiKU,14023
@@ -23,7 +23,7 @@ ragaai_catalyst/tracers/instrumentators/llamaindex.py,sha256=SMrRlR4xM7k9HK43hak
 ragaai_catalyst/tracers/instrumentators/openai.py,sha256=14R4KW9wQCR1xysLfsP_nxS7cqXrTPoD8En4MBAaZUU,379
 ragaai_catalyst/tracers/utils/__init__.py,sha256=KeMaZtYaTojilpLv65qH08QmpYclfpacDA0U3wg6Ybw,64
 ragaai_catalyst/tracers/utils/utils.py,sha256=ViygfJ7vZ7U0CTSA1lbxVloHp4NSlmfDzBRNCJuMhis,2374
-ragaai_catalyst-2.0.6b1.dist-info/METADATA,sha256=jDHkbahJ8JAcsVi0su6KtWbQoPgdZt07jxYxgT4LXDM,8525
-ragaai_catalyst-2.0.6b1.dist-info/WHEEL,sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY,91
-ragaai_catalyst-2.0.6b1.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
-ragaai_catalyst-2.0.6b1.dist-info/RECORD,,
+ragaai_catalyst-2.0.7b1.dist-info/METADATA,sha256=A_kASl-qN-h-7s3N9NxL8iZAuRKmwPjpAaVmIH0InpQ,11241
+ragaai_catalyst-2.0.7b1.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+ragaai_catalyst-2.0.7b1.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
+ragaai_catalyst-2.0.7b1.dist-info/RECORD,,

{ragaai_catalyst-2.0.6b1.dist-info → ragaai_catalyst-2.0.7b1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.5.0)
+Generator: setuptools (75.6.0)
 Root-Is-Purelib: true
 Tag: py3-none-any