volidator-python 0.1.0__py3-none-any.whl

volidator/__init__.py

@@ -0,0 +1,5 @@
+from .client import VolidatorClient
+from .compliance import VolidatorCompliance
+from .agent import VolidatorAgent
+
+__all__ = ["VolidatorClient", "VolidatorCompliance", "VolidatorAgent"]

volidator/agent.py

@@ -0,0 +1,137 @@
+from typing import Any
+
+
+class VolidatorAgent:
+    def __init__(self, client):
+        self.client = client
+
+    def _log_agent(
+        self,
+        action: str,
+        eu_ai_act: str,
+        nist_ai_rmf: str,
+        soc2_control: str,
+        iso_control: str,
+        payload: dict[str, Any]
+    ) -> bool:
+        core_fields = {
+            "actor",
+            "actorId",
+            "target",
+            "targetId",
+            "tenant",
+            "tenantId",
+            "metadata",
+            "context",
+            "telemetry",
+            "req",
+            "traceId",
+            "spanId",
+            "parentSpanId",
+        }
+        
+        # Enriched metadata includes any additional keyword args passed at root level
+        agent_data = {k: v for k, v in payload.items() if k not in core_fields}
+        metadata = payload.get("metadata") or {}
+
+        enriched_metadata = {
+            **metadata,
+            **agent_data,
+            "eu_ai_act": eu_ai_act,
+            "nist_ai_rmf": nist_ai_rmf,
+            "soc2_control": soc2_control,
+            "iso27001": iso_control,
+        }
+
+        # Build base payload structure
+        base_payload = {k: v for k, v in payload.items() if k in core_fields}
+        base_payload["action"] = action
+        base_payload["metadata"] = enriched_metadata
+
+        # AI logs have a larger 5MB metadata allowance
+        return self.client.log(base_payload, max_meta_override=5242880)
+
+    async def _log_agent_async(
+        self,
+        action: str,
+        eu_ai_act: str,
+        nist_ai_rmf: str,
+        soc2_control: str,
+        iso_control: str,
+        payload: dict[str, Any]
+    ) -> bool:
+        core_fields = {
+            "actor",
+            "actorId",
+            "target",
+            "targetId",
+            "tenant",
+            "tenantId",
+            "metadata",
+            "context",
+            "telemetry",
+            "req",
+            "traceId",
+            "spanId",
+            "parentSpanId",
+        }
+        
+        agent_data = {k: v for k, v in payload.items() if k not in core_fields}
+        metadata = payload.get("metadata") or {}
+
+        enriched_metadata = {
+            **metadata,
+            **agent_data,
+            "eu_ai_act": eu_ai_act,
+            "nist_ai_rmf": nist_ai_rmf,
+            "soc2_control": soc2_control,
+            "iso27001": iso_control,
+        }
+
+        base_payload = {k: v for k, v in payload.items() if k in core_fields}
+        base_payload["action"] = action
+        base_payload["metadata"] = enriched_metadata
+
+        return await self.client.log_async(base_payload, max_meta_override=5242880)
+
+    # 1. tool_call / tool_call_async
+    def tool_call(self, payload: dict[str, Any]) -> bool:
+        return self._log_agent("agent.tool_call", "Article 12", "MANAGE 2.2", "CC6.6", "A.12.4.1", payload)
+
+    async def tool_call_async(self, payload: dict[str, Any]) -> bool:
+        return await self._log_agent_async("agent.tool_call", "Article 12", "MANAGE 2.2", "CC6.6", "A.12.4.1", payload)
+
+    # 2. decision / decision_async
+    def decision(self, payload: dict[str, Any]) -> bool:
+        return self._log_agent("agent.decision", "Article 12 & 13", "GOVERN 1.7", "CC6.2", "A.18.1.3", payload)
+
+    async def decision_async(self, payload: dict[str, Any]) -> bool:
+        return await self._log_agent_async("agent.decision", "Article 12 & 13", "GOVERN 1.7", "CC6.2", "A.18.1.3", payload)
+
+    # 3. escalation / escalation_async
+    def escalation(self, payload: dict[str, Any]) -> bool:
+        return self._log_agent("agent.escalation", "Article 14", "GOVERN 5.1", "CC6.3", "A.6.1.2", payload)
+
+    async def escalation_async(self, payload: dict[str, Any]) -> bool:
+        return await self._log_agent_async("agent.escalation", "Article 14", "GOVERN 5.1", "CC6.3", "A.6.1.2", payload)
+
+    # 4. anomaly / anomaly_async
+    def anomaly(self, payload: dict[str, Any]) -> bool:
+        return self._log_agent("agent.anomaly", "Article 9", "MANAGE 2.4", "CC7.2", "A.16.1.2", payload)
+
+    async def anomaly_async(self, payload: dict[str, Any]) -> bool:
+        return await self._log_agent_async("agent.anomaly", "Article 9", "MANAGE 2.4", "CC7.2", "A.16.1.2", payload)
+
+    # 5. refusal / refusal_async
+    def refusal(self, payload: dict[str, Any]) -> bool:
+        return self._log_agent("agent.refusal", "Article 5", "GOVERN 1.1", "CC6.8", "A.18.1.3", payload)
+
+    async def refusal_async(self, payload: dict[str, Any]) -> bool:
+        return await self._log_agent_async("agent.refusal", "Article 5", "GOVERN 1.1", "CC6.8", "A.18.1.3", payload)
+
+    # 6. handoff / handoff_async
+    def handoff(self, payload: dict[str, Any]) -> bool:
+        return self._log_agent("agent.handoff", "Article 12", "MAP 1.6", "CC6.6", "A.12.4.1", payload)
+
+    async def handoff_async(self, payload: dict[str, Any]) -> bool:
+        return await self._log_agent_async("agent.handoff", "Article 12", "MAP 1.6", "CC6.6", "A.12.4.1", payload)