safentic 1.0.3__tar.gz → 1.0.4__tar.gz

{safentic-1.0.3/safentic.egg-info → safentic-1.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: safentic
-Version: 1.0.3
+Version: 1.0.4
 Summary: Safentic SDK for behavior analysis
 Home-page: https://safentic.com
 Author: Safentic

{safentic-1.0.3 → safentic-1.0.4}/safentic/__init__.py

@@ -5,4 +5,4 @@ __all__ = [
     "SafenticError",
 ]
 
-__version__ = "1.0.3"
+__version__ = "1.0.4"

{safentic-1.0.3 → safentic-1.0.4}/safentic/layer.py

@@ -6,23 +6,30 @@ class SafenticError(Exception):
     """Raised when Safentic blocks an action."""
     pass
 
+class InvalidAPIKeyError(Exception):
+    """Raised when an invalid API key is used."""
+    pass
 
-class SafetyLayer():
+class SafetyLayer:
     """
     Safentic runtime enforcement wrapper for agent actions.
-    First, insert your api key, then run an action.
-    Example:
-        safety.protect("send_email", {"body": "..."})
-        # Raises SafenticError if blocked
+    Requires a valid API key to function.
     """
 
-    def __init__(self, api_key="", agent_id="", enforcer: PolicyEnforcer = None, raise_on_block: bool = True):
+    def __init__(self, api_key: str = "", agent_id: str = "", enforcer: PolicyEnforcer = None, raise_on_block: bool = True):
+        if not api_key:
+            raise InvalidAPIKeyError("Missing API key")
+
+        validation_response = validate_api_key(api_key)
+        if not validation_response or validation_response.get("status") != "valid":
+            raise InvalidAPIKeyError("Invalid or unauthorized API key")
+
+        self.api_key = api_key
         self.agent_id = agent_id
         self.raise_on_block = raise_on_block
         self.logger = AuditLogger()
 
         self.enforcer = enforcer or PolicyEnforcer()
-        self.api_key = validate_api_key(api_key)
         self.enforcer.reset(agent_id)
 
     def protect(self, tool_name: str, tool_args: dict) -> dict:
@@ -30,7 +37,7 @@ class SafetyLayer():
         Checks whether a tool action is allowed.
         Raises SafenticError if blocked (default), or returns result if raise_on_block=False.
         """
-            
+
         result = self.enforcer.enforce(self.agent_id, tool_name, tool_args)
 
         # Log structured event
@@ -41,10 +48,9 @@ class SafetyLayer():
             reason=result["reason"] if not result["allowed"] else None
         )
 
-        # Raise or return based on outcome and config
         if not result["allowed"]:
             if self.raise_on_block:
                 raise SafenticError(result["reason"])
             return result
 
-        return result
+        return result

{safentic-1.0.3 → safentic-1.0.4/safentic.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: safentic
-Version: 1.0.3
+Version: 1.0.4
 Summary: Safentic SDK for behavior analysis
 Home-page: https://safentic.com
 Author: Safentic

{safentic-1.0.3 → safentic-1.0.4}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="safentic",
-    version="1.0.3",
+    version="1.0.4",
     packages=find_packages(),
     install_requires=[
         "requests",