aivil-python 1.0.0__py3-none-any.whl

aivil/__init__.py

@@ -0,0 +1,112 @@
+import requests
+from typing import Optional, Dict
+
+AIVIL_API = "https://api.aivildev.com"
+
+class AIVILError(Exception):
+    def __init__(self, message, code=None, status=None):
+        super().__init__(message)
+        self.code = code
+        self.status = status
+
+class AIVILAuthError(AIVILError):
+    def __init__(self, message):
+        super().__init__(message, "AUTH_ERROR", 401)
+
+class AIVILPlanError(AIVILError):
+    def __init__(self, message, upgrade_url=None):
+        super().__init__(message, "PLAN_LIMIT", 403)
+        self.upgrade_url = upgrade_url or "https://aivildev.com/pricing"
+
+class AIVILTimeoutError(AIVILError):
+    def __init__(self, message):
+        super().__init__(message, "TIMEOUT", 408)
+
+class AIVIL:
+    def __init__(self, api_key, api_url=AIVIL_API, timeout=5, fallback_behavior="escalate", retries=2, debug=False):
+        if not api_key:
+            raise AIVILError("api_key is required. Sign up at aivildev.com/signup")
+        self.api_key = api_key
+        self.api_url = api_url
+        self.timeout = timeout
+        self.fallback_behavior = fallback_behavior
+        self.retries = retries
+        self.debug = debug
+
+    def _log(self, msg):
+        if self.debug:
+            print(f"[AIVIL] {msg}")
+
+    def _request(self, method, path, body=None, attempt=0):
+        try:
+            r = requests.request(method, f"{self.api_url}{path}",
+                headers={"Content-Type": "application/json", "Authorization": f"Bearer {self.api_key}"},
+                json=body, timeout=self.timeout)
+            data = r.json()
+            if r.status_code == 401: raise AIVILAuthError(data.get("error", "Invalid API key"))
+            if r.status_code == 403: raise AIVILPlanError(data.get("error", "Plan limit"), data.get("upgrade_url"))
+            if not r.ok: raise AIVILError(data.get("error", f"Error {r.status_code}"))
+            return data
+        except requests.Timeout:
+            if attempt < self.retries:
+                import time; time.sleep(1)
+                return self._request(method, path, body, attempt + 1)
+            raise AIVILTimeoutError(f"Timed out after {self.timeout}s")
+        except requests.ConnectionError as e:
+            if attempt < self.retries:
+                import time; time.sleep(1)
+                return self._request(method, path, body, attempt + 1)
+            raise AIVILError(str(e), "CONNECTION_ERROR")
+
+    def create_agent(self, config):
+        for f in ["name","role","owner","jurisdiction"]:
+            if f not in config: raise AIVILError(f"create_agent requires '{f}'")
+        return self._request("POST", "/agents", config)
+
+    def audit(self, agent_id, action, fallback=None):
+        fallback = fallback or self.fallback_behavior
+        try:
+            return self._request("POST", f"/agents/{agent_id}/audit", {"action": action})["verdict"]
+        except (AIVILTimeoutError, AIVILError) as e:
+            if e.code in ("TIMEOUT", "CONNECTION_ERROR"):
+                import warnings
+                warnings.warn(f"[AIVIL] Unreachable. Fallback: {fallback}")
+                return {"status": fallback.upper(), "reason": "AIVIL unreachable", "flags": ["AIVIL_FALLBACK"], "agent_id": agent_id, "fallback": True}
+            raise
+
+    def list_agents(self):
+        return self._request("GET", "/agents")["agents"]
+
+    def get_agent(self, agent_id):
+        return self._request("GET", f"/agents/{agent_id}")["agent"]
+
+    def find_agent(self, name):
+        return next((a for a in self.list_agents() if a["name"] == name), None)
+
+    def get_or_create(self, config):
+        existing = self.find_agent(config["name"])
+        if existing:
+            return {"agent": existing, "created": False}
+        data = self.create_agent(config)
+        return {**data, "created": True}
+
+    def update_policy(self, agent_id, policy, reason=""):
+        return self._request("PATCH", f"/agents/{agent_id}/policy", {"policy": policy, "change_reason": reason})
+
+    def suspend(self, agent_id, reason=""):
+        return self._request("POST", f"/agents/{agent_id}/suspend", {"reason": reason})
+
+    def reactivate(self, agent_id, reason=""):
+        return self._request("POST", f"/agents/{agent_id}/reactivate", {"reason": reason})
+
+    def retire(self, agent_id, reason=""):
+        return self._request("POST", f"/agents/{agent_id}/retire", {"reason": reason})
+
+    def get_audit_log(self, agent_id, limit=50):
+        return self._request("GET", f"/agents/{agent_id}/audit?limit={limit}")["logs"]
+
+    def verify(self, agent_id):
+        return requests.get(f"{self.api_url}/verify/{agent_id}", timeout=self.timeout).json()
+
+    def get_stats(self):
+        return self._request("GET", "/stats")

aivil_python-1.0.0.dist-info/METADATA

@@ -0,0 +1,12 @@
+Metadata-Version: 2.1
+Name: aivil-python
+Version: 1.0.0
+Summary: The AI Vital Identity Layer — civil registry for AI agents
+Home-page: https://aivildev.com
+Author: AIVIL
+Author-email: ihimanshu882@gmail.com
+Keywords: ai,agents,identity,policy,governance
+Requires-Python: >=3.8
+Requires-Dist: requests >=2.28.0
+
+Full docs at https://aivildev.com/docs

aivil_python-1.0.0.dist-info/RECORD

@@ -0,0 +1,5 @@
+aivil/__init__.py,sha256=TPHNwqxk0XNHt5_MsL1SKmgHMxMfupFQNnmxCXM5UhE,4707
+aivil_python-1.0.0.dist-info/METADATA,sha256=lTYbA51-t-4JXiDUm9gHgPVgC2-TE_c6WVfUZezsr70,352
+aivil_python-1.0.0.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+aivil_python-1.0.0.dist-info/top_level.txt,sha256=dMhostO7-swcoTACfzni6e_z-WnK4pvOTo3ExRxhX48,6
+aivil_python-1.0.0.dist-info/RECORD,,

aivil_python-1.0.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: bdist_wheel (0.42.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

aivil_python-1.0.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+aivil