Merge pull request #7 from ZenGuard-AI/baur/fix

[sdk] add secrets

Author: Baur

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "zenguard"
-version = "0.1.1"
+version = "0.1.2"
 description = "Plug-and-play production grade security for GenAI applications"
 authors = ["ZenGuard Team <hello@zenguard.ai>"]
 license = "MIT"

tests/e2e_test.py

@@ -43,6 +43,12 @@ def test_keywords(zenguard: ZenGuard):
     response = zenguard.detect(detectors=detectors, prompt=prompt)
     assert_successful_response_not_detected(response)
 
+def test_secrets(zenguard: ZenGuard):
+    prompt = "Simple secrets test"
+    detectors = [Detector.SECRETS]
+    response = zenguard.detect(detectors=detectors, prompt=prompt)
+    assert_successful_response_not_detected(response)
+
 
 if __name__ == "__main__":
     api_key = os.environ.get("ZEN_API_KEY")

zenguard/zenguard.py

@@ -24,6 +24,7 @@ class Detector(Enum):
     ALLOWED_TOPICS = "v1/detect/topics/allowed"
     BANNED_TOPICS = "v1/detect/topics/banned"
     KEYWORDS = "v1/detect/keywords"
+    SECRETS = "v1/detect/secrets"
 
 
 class ZenGuard: