update: docs

soumik12345 · Dec 20, 2024 · e27fc71 · e27fc71
1 parent b2505c7
commit e27fc71
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 15 deletions.
diff --git a/.streamlit/config.toml b/.streamlit/config.toml
diff --git a/README.md b/README.md
@@ -41,17 +41,21 @@ import weave
 
 from safeguards.guardrails import (
     GuardrailManager,
-    PromptInjectionProtectAIGuardrail,
-    PromptInjectionSurveyGuardrail,
+    PromptInjectionClassifierGuardrail,
+    PromptInjectionLLMGuardrail,
+    OpenAIPrivilegeEscalationGuardrail,
 )
 from safeguards.llm import OpenAIModel
 
 weave.init(project_name="safeguards")
 
 manager = GuardrailManager(
     guardrails=[
-        PromptInjectionSurveyGuardrail(llm_model=OpenAIModel(model_name="gpt-4o")),
-        PromptInjectionProtectAIGuardrail(),
+        PromptInjectionLLMGuardrail(llm_model=OpenAIModel(model_name="gpt-4o")),
+        PromptInjectionClassifierGuardrail(
+            model_name="ProtectAI/deberta-v3-base-prompt-injection-v2"
+        ),
+        OpenAIPrivilegeEscalationGuardrail(llm_model=OpenAIModel(model_name="gpt-4o")),
     ]
 )
 manager.guard(

diff --git a/docs/index.md b/docs/index.md
@@ -45,17 +45,21 @@ import weave
 
 from safeguards.guardrails import (
     GuardrailManager,
-    PromptInjectionProtectAIGuardrail,
-    PromptInjectionSurveyGuardrail,
+    PromptInjectionClassifierGuardrail,
+    PromptInjectionLLMGuardrail,
+    OpenAIPrivilegeEscalationGuardrail,
 )
 from safeguards.llm import OpenAIModel
 
 weave.init(project_name="safeguards")
 
 manager = GuardrailManager(
     guardrails=[
-        PromptInjectionSurveyGuardrail(llm_model=OpenAIModel(model_name="gpt-4o")),
-        PromptInjectionProtectAIGuardrail(),
+        PromptInjectionLLMGuardrail(llm_model=OpenAIModel(model_name="gpt-4o")),
+        PromptInjectionClassifierGuardrail(
+            model_name="ProtectAI/deberta-v3-base-prompt-injection-v2"
+        ),
+        OpenAIPrivilegeEscalationGuardrail(llm_model=OpenAIModel(model_name="gpt-4o")),
     ]
 )
 manager.guard(