enkryptai-sdk 1.0.2__py3-none-any.whl → 1.0.4__py3-none-any.whl

enkryptai_sdk/ai_proxy.py

@@ -54,7 +54,7 @@ class AIProxyClient(BaseClient):
             # Try to parse nested JSON error if it's a string
             if isinstance(error_message, str) and is_json_str:
                 import ast
-                import json
+                # import json
                 try:
                     # # Using ast.literal_eval to safely evaluate the string as a Python literal
                     # # As json.loads is not working with literal string representation of dict
@@ -65,7 +65,8 @@ class AIProxyClient(BaseClient):
                     response["error"] = parsed_error.get("error", parsed_error)
                     if "enkrypt_policy_detections" in parsed_error:
                         response["enkrypt_policy_detections"] = parsed_error["enkrypt_policy_detections"]
-                except json.JSONDecodeError:
+                # except json.JSONDecodeError:
+                except Exception:
                     # If parsing fails, keep the original error
                     pass
                     
@@ -78,9 +79,10 @@ class AIProxyClient(BaseClient):
                 try:
                     # print("Error in response: ", response)
                     return ChatCompletionErrorResponse.from_dict(response)
-                except (json.JSONDecodeError, TypeError, ValueError, SyntaxError):
+                # except (json.JSONDecodeError, TypeError, ValueError, SyntaxError):
+                except Exception:
                     # Fallback to direct error if error object can't be parsed
-                    print("Failed to parse error response: ", response)
+                    # print("Failed to parse error response: ", response)
                     return ChatCompletionDirectErrorResponse.from_dict(response)
             raise AIProxyClientError(response["error"])
 

enkryptai_sdk/evals.py

@@ -34,7 +34,7 @@ class EvalsClient:
             return response.json()
         
         except Exception as e:
-            print(e)
+            # print(e)
             return {"error": str(e)}
 
     # ----------------------------
@@ -59,7 +59,7 @@ class EvalsClient:
         try:
             return self._request("POST", "/guardrails/adherence", json=payload)
         except Exception as e:
-            print(e)
+            # print(e)
             return {"error": str(e)}
 
     def check_relevancy(self, question, llm_answer):
@@ -80,5 +80,5 @@ class EvalsClient:
         try:
             return self._request("POST", "/guardrails/relevancy", json=payload)
         except Exception as e:
-            print(e)
+            # print(e)
             return {"error": str(e)}

enkryptai_sdk/models.py

@@ -103,7 +103,7 @@ class ModelClient(BaseClient):
         """
         headers = {"X-Enkrypt-Model": model_saved_name}
         response = self._request("GET", "/models/get-model", headers=headers)
-        print(response)
+        # print(response)
         if response.get("error"):
             raise ModelClientError(response["error"])
         return ModelConfig.from_dict(response)

enkryptai_sdk/red_team.py

@@ -275,7 +275,7 @@ class RedTeamClient(BaseClient):
         response = self._request("GET", "/redteam/results/summary", headers=headers)
         if response.get("error"):
             raise RedTeamClientError(response["error"])
-        print(f"Response: {response}")
+        # print(f"Response: {response}")
         return RedTeamResultSummary.from_dict(response)
     
     def get_result_summary_test_type(self, task_id: str = None, test_name: str = None, test_type: str = None):
@@ -309,7 +309,7 @@ class RedTeamClient(BaseClient):
         response = self._request("GET", url, headers=headers)
         if response.get("error"):
             raise RedTeamClientError(response["error"])
-        print(f"Response: {response}")
+        # print(f"Response: {response}")
         return RedTeamResultSummary.from_dict(response)
 
     def get_result_details(self, task_id: str = None, test_name: str = None):

{enkryptai_sdk-1.0.2.dist-info → enkryptai_sdk-1.0.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: enkryptai-sdk
-Version: 1.0.2
+Version: 1.0.4
 Summary: A Python SDK with guardrails and red teaming functionality for API interactions
 Home-page: https://github.com/enkryptai/enkryptai-sdk
 Author: Enkrypt AI Team

{enkryptai_sdk-1.0.2.dist-info → enkryptai_sdk-1.0.4.dist-info}/RECORD

@@ -1,14 +1,14 @@
 enkryptai_sdk/__init__.py,sha256=rP6PtntJogJauj1lKWK8DkiBr3uYjireIUamr6aflu0,763
-enkryptai_sdk/ai_proxy.py,sha256=QnfRlyKKjUfDuJ19rTM8dtbWvBaHgLyb2Z9AOjK2PFA,3723
+enkryptai_sdk/ai_proxy.py,sha256=pD6kPmD9H0gttN28cezgV7_IVelLXAHNR5cPeXCM8Ew,3799
 enkryptai_sdk/base.py,sha256=MlEDcEIjXo35kat9XkGUu7VB2fIvJk38C94wAeO9bEw,1304
 enkryptai_sdk/config.py,sha256=IpB8_aO4zXdvv061v24oh83oyJ5Tp1QBQTzeuW4h9QY,8828
 enkryptai_sdk/datasets.py,sha256=xekcdY9wIniw2TnylaK6o1RNZ5DRoluNOGawBkVgaM0,4881
 enkryptai_sdk/deployments.py,sha256=qXoUQtLYRSqZsAAqPgxJAad1a3mgMAbPx09NiOyAdsw,4155
-enkryptai_sdk/evals.py,sha256=SAQ3GM2ppV-uPLWR1NdOa0fjiEWY5UqzdA8bXFZVl5I,2589
+enkryptai_sdk/evals.py,sha256=BywyEgIT7xdJ58svO_sDNOMVowdB0RTGoAZPEbCnDVo,2595
 enkryptai_sdk/guardrails.py,sha256=CtQwPBmpFFy71P22kG0gJbefYJflUT9aareYksPhNlQ,12096
 enkryptai_sdk/guardrails_old.py,sha256=SgzPZkTzbAPD9XfmYNG6M1-TrzbhDHpAkI3FjnVWS_s,6434
-enkryptai_sdk/models.py,sha256=3y6-7vaDrEmDZpc3zHmIsBs7DFjDprf_EA7PU2yPU0c,8976
-enkryptai_sdk/red_team.py,sha256=DbJRpWwZBHljR3o247PvqCzJm3_TV9UXaxISeUmEJ5w,14188
+enkryptai_sdk/models.py,sha256=FSdpfz-e1a0D7nuYCnQC859AdrOUVr60QbLR1mlZjp0,8978
+enkryptai_sdk/red_team.py,sha256=jlSf7oNPM7jJHXyv37ROQgkluuigxbSpeeAfvKqUiGI,14192
 enkryptai_sdk/response.py,sha256=43JRubzgGCpoVxYNzBZY0AlUgLbfcXD_AwD7wU3qY9o,4086
 enkryptai_sdk/dto/__init__.py,sha256=kKBw4rkfqMBuK8nXRDtD6Sd0_uqLKgbcHrqzuSGJpr0,2310
 enkryptai_sdk/dto/ai_proxy.py,sha256=clwMN4xdH8Zr55dnhilHbs-qaHRlCOrLPrij0Zd1Av0,11283
@@ -18,8 +18,8 @@ enkryptai_sdk/dto/deployments.py,sha256=lsKdG09C-rceIjGvEyYOBf5zBjrk7ma8NpPfgrAg
 enkryptai_sdk/dto/guardrails.py,sha256=XMFco-KlEqI4TYoJAyxxTrk-OFixtEcftBpG1SXFH4k,39583
 enkryptai_sdk/dto/models.py,sha256=O4gVhVTenlsytNJIvk2gO5530KZWMye6FCVCtF5IW-A,11700
 enkryptai_sdk/dto/red_team.py,sha256=BoOPYFjIONIC0XPuyJtkx_qLVYi2kLGEA4CzlySgbJA,13829
-enkryptai_sdk-1.0.2.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-enkryptai_sdk-1.0.2.dist-info/METADATA,sha256=erxYdcKCSpIFnjCvMQWFRETJk8xa7ErFuMlqv9dPA6I,43207
-enkryptai_sdk-1.0.2.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-enkryptai_sdk-1.0.2.dist-info/top_level.txt,sha256=s2X9UJJwvJamNmr6ZXWyyQe60sXtQGWFuaBYfhgHI_4,14
-enkryptai_sdk-1.0.2.dist-info/RECORD,,
+enkryptai_sdk-1.0.4.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+enkryptai_sdk-1.0.4.dist-info/METADATA,sha256=2lqT-652SgRDCyjpMhmRfyUiF6dK1uL-kl0i1RbKhsc,43207
+enkryptai_sdk-1.0.4.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+enkryptai_sdk-1.0.4.dist-info/top_level.txt,sha256=s2X9UJJwvJamNmr6ZXWyyQe60sXtQGWFuaBYfhgHI_4,14
+enkryptai_sdk-1.0.4.dist-info/RECORD,,