new feature: mcp pre-validation

agent_framework_oci/libs/agent_framework/src/agent_framework/mcp/tool_policy.py

@@ -13,6 +13,14 @@ class WorkflowExecutionPolicy(BaseModel):
     version: int | Literal["active"] = "active"
 
 
+class ToolPreValidationPolicy(BaseModel):
+    """Optional MCP business pre-validation executed before user confirmation."""
+
+    enabled: bool = False
+    tool: str | None = None
+    fail_open: bool = False
+
+
 class ToolPolicy(BaseModel):
     """Política de execução aplicada antes da chamada MCP ou workflow."""
 
@@ -20,6 +28,7 @@ class ToolPolicy(BaseModel):
     require_confirmation: bool = False
     requires: list[str] = Field(default_factory=list)
     execution: WorkflowExecutionPolicy = Field(default_factory=WorkflowExecutionPolicy)
+    pre_validation: ToolPreValidationPolicy = Field(default_factory=ToolPreValidationPolicy)
 
 
 class ToolPolicyRegistry:
@@ -55,11 +64,18 @@ class ToolPolicyRegistry:
         execution_raw = raw.get("execution") or {}
         base_execution = base.execution.model_dump()
         base_execution.update(execution_raw)
+        pre_validation_raw = raw.get("pre_validation") or {}
+        base_pre_validation = base.pre_validation.model_dump()
+        if isinstance(pre_validation_raw, bool):
+            base_pre_validation["enabled"] = pre_validation_raw
+        elif isinstance(pre_validation_raw, dict):
+            base_pre_validation.update(pre_validation_raw)
         return ToolPolicy(
             operation_type=operation_type,
             require_confirmation=bool(confirmation),
             requires=list(raw.get("requires", base.requires) or []),
             execution=WorkflowExecutionPolicy.model_validate(base_execution),
+            pre_validation=ToolPreValidationPolicy.model_validate(base_pre_validation),
         )
 
     def get(self, tool_name: str) -> ToolPolicy | None:

agent_framework_oci/libs/agent_framework/src/agent_framework/mcp/tool_router.py

@@ -83,12 +83,14 @@ class MCPToolRouter:
             required.extend(explicit.requires)
             source = "tool_policies.yaml"
         execution = explicit.execution.model_dump() if explicit is not None else {"mode": "direct_tool", "workflow": None, "version": "active"}
+        pre_validation = explicit.pre_validation.model_dump() if explicit is not None else {"enabled": False, "tool": None, "fail_open": False}
         return {
             "operation_type": operation_type,
             "require_confirmation": confirmation_required,
             "requires": list(dict.fromkeys(required)),
             "policy_source": source,
             "execution": execution,
+            "pre_validation": pre_validation,
         }
 
     def validate_execution_policy(

agent_framework_oci/libs/agent_framework/src/agent_framework/runtime/agent_runtime.py

@@ -713,6 +713,83 @@ class AgentRuntimeMixin:
             "policy_source": "tools.yaml",
         }
 
+    async def _run_transaction_pre_validation(
+        self,
+        state: dict[str, Any],
+        *,
+        tool_name: str,
+        arguments: dict[str, Any],
+        policy: dict[str, Any],
+        emit_events: bool = True,
+    ) -> dict[str, Any] | None:
+        """Execute an optional domain-owned MCP pre-validation before confirmation.
+
+        The framework knows only the generic contract ``eligible``. Business rules
+        remain in the configured MCP validator tool. No LLM is used here.
+        """
+        cfg = policy.get("pre_validation") if isinstance(policy, dict) else None
+        if not isinstance(cfg, dict) or not cfg.get("enabled"):
+            return None
+        validator = str(cfg.get("tool") or "").strip()
+        if not validator:
+            return None
+        validation_args = dict(arguments or {})
+        validation_args.pop("confirmed", None)
+        validation_args["target_tool"] = tool_name
+        if emit_events:
+            await self._emit_ic(
+                "IC.TRANSACTION_PREVALIDATION_REQUESTED",
+                state,
+                {"tool_name": tool_name, "validator_tool": validator},
+                component="agent_runtime.tool_policy",
+            )
+        result = await self._call_mcp_tool(validator, validation_args, state)
+        payload = result.get("result") if isinstance(result, dict) and isinstance(result.get("result"), dict) else result
+        eligible = payload.get("eligible") if isinstance(payload, dict) else None
+        if eligible is True:
+            state["transaction_pre_validation"] = {
+                "tool_name": tool_name, "validator_tool": validator, "eligible": True, "result": result
+            }
+            if emit_events:
+                await self._emit_ic(
+                    "IC.TRANSACTION_PREVALIDATION_PASSED", state,
+                    {"tool_name": tool_name, "validator_tool": validator},
+                    component="agent_runtime.tool_policy",
+                )
+            return None
+        transport_failed = isinstance(result, dict) and result.get("ok") is False and eligible is None
+        if transport_failed and bool(cfg.get("fail_open")):
+            return None
+        status = str((payload or {}).get("status") or ("PREVALIDATION_ERROR" if transport_failed else "OUT_OF_SCOPE"))
+        state["transaction_pre_validation"] = {
+            "tool_name": tool_name,
+            "validator_tool": validator,
+            "eligible": False,
+            "status": status,
+            "error": (payload or {}).get("error") if isinstance(payload, dict) else None,
+            "terminal": True,
+            "result": result,
+        }
+        # A rejeição da pré-validação encerra o latch transacional imediatamente.
+        # A regra de negócio permanece no MCP; o framework apenas materializa o
+        # resultado genérico de elegibilidade e garante que o próximo turno volte
+        # ao roteamento normal, sem herdar COLLECTING_/WAITING_.
+        self._finish_active_transaction(state, "OUT_OF_SCOPE", result=result)
+        state["next_state"] = None
+        state["confirmation_required"] = False
+        state["confirmation_received"] = False
+        if emit_events:
+            await self._emit_ic(
+                "IC.TRANSACTION_PREVALIDATION_REJECTED", state,
+                {"tool_name": tool_name, "validator_tool": validator, "status": status, "error": (payload or {}).get("error")},
+                component="agent_runtime.tool_policy",
+            )
+        enriched = dict(result or {})
+        enriched["pre_validation"] = True
+        enriched["target_tool"] = tool_name
+        enriched["transaction_status"] = "OUT_OF_SCOPE"
+        return enriched
+
     def _validate_tool_execution_policy(self, tool_name: str, arguments: dict[str, Any]) -> tuple[bool, str | None]:
         """Aplica a mesma política central usada pelo MCPToolRouter."""
         router = getattr(self, "tool_router", None)
@@ -1608,6 +1685,7 @@ class AgentRuntimeMixin:
             "tool_policy_result", "missing_parameters", "next_state", "pending_domain_workflow", "pending_tool_clarification",
             "business_workflows_executed", "active_transaction", "last_transaction",
             "transaction_evidence", "last_transaction_evidence", "relevant_transaction_evidence",
+            "transaction_pre_validation",
         )
         return {key: state.get(key) for key in keys if key in state}
 
@@ -2017,6 +2095,12 @@ class AgentRuntimeMixin:
                     state, tool_name=tool_name, arguments=arguments, status="COLLECTING_PARAMETERS"
                 )
                 state["missing_parameters"] = []
+                pre_validation_result = await self._run_transaction_pre_validation(
+                    state, tool_name=tool_name, arguments=arguments, policy=policy, emit_events=emit_events
+                )
+                if pre_validation_result is not None:
+                    return [pre_validation_result]
+
                 if policy.get("require_confirmation"):
                     waiting_state = self._waiting_state_name(state)
                     state.update({
@@ -2190,6 +2274,13 @@ class AgentRuntimeMixin:
             })
             return results
 
+        pre_validation_result = await self._run_transaction_pre_validation(
+            state, tool_name=selected_action, arguments=action_args, policy=policy, emit_events=emit_events
+        )
+        if pre_validation_result is not None:
+            results.append(pre_validation_result)
+            return results
+
         if policy.get("require_confirmation"):
             state.update({
                 "pending_tool_call": selected,