agentx-security-sdk 0.1.0__py3-none-any.whl

agentx_sdk/__init__.py

@@ -0,0 +1,4 @@
+from .decorators import agentx_protect
+from .client import AgentXClient
+
+__all__ = ["agentx_protect", "AgentXClient"]

agentx_sdk/client.py

@@ -0,0 +1,24 @@
+import os
+import requests
+
+class AgentXClient:
+    def __init__(self, api_key: str = None, gateway_url: str = "http://localhost:8000"):
+        # In the future, api_key authenticates them to our cloud. 
+        # For now, it connects to their local Sidecar Gateway.
+        self.api_key = api_key or os.environ.get("AGENTX_API_KEY")
+        self.gateway_url = gateway_url
+
+    def evaluate_intent(self, agent_id: str, query: str, chain_of_thought: str, receipt_id: str = None):
+        payload = {
+            "agent_id": agent_id,
+            "original_query": query,
+            "chain_of_thought": chain_of_thought,
+            "receipt_id": receipt_id
+        }
+        try:
+            response = requests.post(f"{self.gateway_url}/v1/evaluate_intent", json=payload)
+            response.raise_for_status()
+            return response.json()
+        except requests.exceptions.RequestException as e:
+            # If our gateway is down, we Fail Closed to protect the system.
+            return {"status": "ERROR", "message": f"AgentX Gateway unreachable: {str(e)}"}

agentx_sdk/decorators.py

@@ -0,0 +1,96 @@
+import functools
+import json
+import atexit
+import time
+from .client import AgentXClient
+
+_client = AgentXClient()
+
+# --- 1. THE ROI TRACKER ---
+_session_stats = {
+    "start_time": time.time(),
+    "total_calls": 0,
+    "intercepts": 0,
+    "critical_blocks": 0
+}
+
+def _print_roi_summary():
+    """Fires automatically when the developer's script ends or crashes."""
+    duration = round(time.time() - _session_stats["start_time"], 2)
+    
+    # Heuristic: Every intercept prevents the agent from burning ~1,500 tokens 
+    # going down a rabbit hole, plus saves the developer ~5 mins of debugging.
+    tokens_saved = _session_stats["intercepts"] * 1500
+    dollars_saved = (tokens_saved / 1000) * 0.015  # Approx GPT-4o cost
+    
+    print("\n" + "═"*50)
+    print(" 🛡️  AgentX Session Summary")
+    print("═"*50)
+    print(f" ⏱️  Uptime:                {duration} seconds")
+    print(f" 🛠️  Tools Monitored:       {_session_stats['total_calls']}")
+    print(f" 🛑 Socratic Intercepts:   {_session_stats['intercepts']}")
+    print(f" 💥 Catastrophic Blocks:   {_session_stats['critical_blocks']}")
+    print("─"*50)
+    print(f" 💰 Est. Tokens Saved:     ~{tokens_saved} (${dollars_saved:.4f})")
+    print(f" ⏳ Developer Time Saved:  ~{_session_stats['intercepts'] * 5} minutes")
+    print("═"*50 + "\n")
+
+# Register the hook to run on exit
+atexit.register(_print_roi_summary)
+# --------------------------
+
+
+def agentx_protect(agent_id: str, extract_query_func=None, extract_cot_func=None):
+    def decorator(func):
+        @functools.wraps(func)
+        def wrapper(*args, **kwargs):
+            # Track the call
+            _session_stats["total_calls"] += 1
+            
+            query = extract_query_func(*args, **kwargs) if extract_query_func else str(args)
+            chain_of_thought = extract_cot_func(*args, **kwargs) if extract_cot_func else "Implicit tool call"
+            receipt_id = kwargs.get("receipt_id", None)
+
+            print(f"\n🛡️ [AgentX SDK] Intercepting tool call to '{func.__name__}'...")
+
+            eval_res = _client.evaluate_intent(
+                agent_id=agent_id, 
+                query=query, 
+                chain_of_thought=chain_of_thought,
+                receipt_id=receipt_id
+            )
+
+            status = eval_res.get("status")
+            
+            if status in ["APPROVED", "COMPLIED"]:
+                print(f"✅ [AgentX SDK] Intent safe. Executing '{func.__name__}'.")
+                return func(*args, **kwargs)
+                
+            elif status == "CHALLENGED":
+                # Track the intercept
+                _session_stats["intercepts"] += 1
+                
+                challenge = eval_res.get('socratic_prompt', '')
+                policy_triggered = eval_res.get('policy_triggered', 'Unknown Policy')
+                
+                # --- THE NEW DETERMINISTIC CHECK ---
+                # We specifically track our highest-severity policy for the ROI dashboard
+                # Your engine is returning "Database Isolation", so we check for that!
+                if policy_triggered in ["Mass Destructive Intent", "Database Isolation"]:
+                    _session_stats["critical_blocks"] += 1
+                # -----------------------------------
+                    
+                new_receipt = eval_res.get('receipt_id')
+                print(f"🛑 [AgentX SDK] Policy '{policy_triggered}' violated. Returning challenge to Agent.")
+                
+                return json.dumps({
+                    "error": "AgentX Policy Violation",
+                    "challenge": challenge,
+                    "receipt_id": new_receipt,
+                    "instruction": "Revise your action to comply with the challenge and try again. Pass the receipt_id."
+                })
+            else:
+                return f"AgentX System Error: {eval_res.get('message')}"
+                
+        return wrapper
+    return decorator

agentx_security_sdk-0.1.0.dist-info/METADATA

@@ -0,0 +1,26 @@
+Metadata-Version: 2.4
+Name: agentx-security-sdk
+Version: 0.1.0
+Summary: The self-healing exception handler for autonomous AI agents.
+Home-page: https://github.com/vdalal/semantic-gateway
+Author: AgentX Team
+Author-email: founders@agentx.com
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: requests>=2.25.0
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+The self-healing exception handler for autonomous AI agents.

agentx_security_sdk-0.1.0.dist-info/RECORD

@@ -0,0 +1,7 @@
+agentx_sdk/__init__.py,sha256=F-qQjUNE3gVvsY1ZI3P8ZLNxNtQzC2a-hUZbQRELIWM,120
+agentx_sdk/client.py,sha256=qxtTM1GalxNTLd97Cf4DOa0TACBJjJmXK1RcQFRLw0g,1094
+agentx_sdk/decorators.py,sha256=8r0xFFi660qfP0QHQID-46nqXxsFQdp2lFlyKAwO-ks,4054
+agentx_security_sdk-0.1.0.dist-info/METADATA,sha256=CKB5R13fuVy0J9CvkbiUUWSg_nFu3pYSM6l-G1XNP6Q,855
+agentx_security_sdk-0.1.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+agentx_security_sdk-0.1.0.dist-info/top_level.txt,sha256=_sRdpWoXB1H2iyhAp9OiDsgz4HFAUfEiw_rb8gh1Lqg,11
+agentx_security_sdk-0.1.0.dist-info/RECORD,,

agentx_security_sdk-0.1.0.dist-info/WHEEL

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

agentx_security_sdk-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+agentx_sdk