feat: add execution stage for new guardrails (#683)

dianapirvulescu · web-flow · commit 6e11d2281a7e · 2026-03-11T12:33:50.000+02:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "uipath-langchain"
-version = "0.8.14"
+version = "0.8.15"
 description = "Python SDK that enables developers to build and deploy LangGraph agents to the UiPath Cloud Platform"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"
diff --git a/src/uipath_langchain/agent/react/guardrails/guardrails_subgraph.py b/src/uipath_langchain/agent/react/guardrails/guardrails_subgraph.py
@@ -33,6 +33,8 @@
 _VALIDATOR_ALLOWED_STAGES = {
     "prompt_injection": {ExecutionStage.PRE_EXECUTION},
     "pii_detection": {ExecutionStage.PRE_EXECUTION, ExecutionStage.POST_EXECUTION},
+    "harmful_content": {ExecutionStage.PRE_EXECUTION, ExecutionStage.POST_EXECUTION},
+    "intellectual_property": {ExecutionStage.POST_EXECUTION},
 }
 
 
diff --git a/uv.lock b/uv.lock

Original file line number	Diff line number	Diff line change
`@@ -33,6 +33,8 @@`
`33`	`33`	`_VALIDATOR_ALLOWED_STAGES = {`
`34`	`34`	`"prompt_injection": {ExecutionStage.PRE_EXECUTION},`
`35`	`35`	`"pii_detection": {ExecutionStage.PRE_EXECUTION, ExecutionStage.POST_EXECUTION},`
	`36`	`+ "harmful_content": {ExecutionStage.PRE_EXECUTION, ExecutionStage.POST_EXECUTION},`
	`37`	`+ "intellectual_property": {ExecutionStage.POST_EXECUTION},`
`36`	`38`	`}`
`37`	`39`
`38`	`40`