agentauthlayer 0.1.6__tar.gz → 0.1.8__tar.gz

{agentauthlayer-0.1.6 → agentauthlayer-0.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentauthlayer
-Version: 0.1.6
+Version: 0.1.8
 Summary: Library-first authentication and authorization SDK for AI agents
 Author: Vaibhav Ahluwalia
 License: MIT

{agentauthlayer-0.1.6 → agentauthlayer-0.1.8}/agent_auth/cli.py

@@ -41,6 +41,16 @@ def _bootstrap_status(base_url: str) -> bool | None:
 
 def login_command(args) -> int:
     base_url = (args.base_url or 'http://127.0.0.1:8002').rstrip('/')
+
+    if args.token:
+        save_credentials({
+            'base_url': base_url,
+            'token': args.token,
+            'email': args.email or 'token-auth',
+        })
+        print(f"Stored API token credentials for {base_url}")
+        return 0
+
     setup_status = _bootstrap_status(base_url)
 
     if setup_status is False:
@@ -71,6 +81,13 @@ def login_command(args) -> int:
                 'If this is your first time using this deployment, initialize it first in the UI, then try logging in again.',
                 file=sys.stderr,
             )
+        elif response.status_code == 401 and setup_status is True:
+            print(
+                f"{detail}\n\n"
+                'This local/project deployment is already initialized. Use the same password you set during bootstrap, '
+                'or reset the local state directory if you want a fresh demo environment.',
+                file=sys.stderr,
+            )
         else:
             print(detail, file=sys.stderr)
         return 1
@@ -92,7 +109,11 @@ def whoami_command(args) -> int:
         return 1
     client = AuthAPIClient()
     me = client.me()
-    print(json.dumps(me, indent=2))
+    print(json.dumps({
+        'auth_source': client.auth_source,
+        'base_url': client.base_url,
+        'user': me,
+    }, indent=2))
     return 0
 
 
@@ -198,6 +219,7 @@ def main():
     login_parser.add_argument('--base-url', default=DEFAULT_BASE_URL)
     login_parser.add_argument('--email')
     login_parser.add_argument('--password')
+    login_parser.add_argument('--token', help='Store an API token directly instead of using email/password login')
     login_parser.set_defaults(func=login_command)
 
     whoami_parser = subparsers.add_parser('whoami', help='Show the currently stored user context')

{agentauthlayer-0.1.6 → agentauthlayer-0.1.8}/agent_auth/client.py

@@ -25,6 +25,15 @@ class AuthAPIClient:
         self.token = token or os.getenv("AGENT_AUTH_TOKEN") or stored.get("token")
         self.timeout = timeout
 
+        if token:
+            self.auth_source = "explicit_token"
+        elif os.getenv("AGENT_AUTH_TOKEN"):
+            self.auth_source = "env_token"
+        elif stored.get("token"):
+            self.auth_source = "stored_credentials"
+        else:
+            self.auth_source = "none"
+
     def _headers(self) -> dict[str, str]:
         headers = {"Content-Type": "application/json"}
         if self.token:

{agentauthlayer-0.1.6 → agentauthlayer-0.1.8}/agent_auth/server_runtime.py

@@ -17,25 +17,36 @@ DEFAULT_PORT = 8002
 DEFAULT_BASE_URL = f"http://{DEFAULT_HOST}:{DEFAULT_PORT}"
 
 
-def local_data_dir() -> Path:
+def fallback_data_dir() -> Path:
     return Path.home() / ".agentauth"
 
 
-def _server_key(host: str, port: int) -> str:
-    safe_host = host.replace(":", "_").replace("/", "_")
-    return f"{safe_host}_{port}"
+def project_state_dir() -> Path:
+    return Path.cwd() / ".agentauth"
+
+
+def local_data_dir() -> Path:
+    explicit_state_dir = os.getenv("AGENTAUTH_STATE_DIR")
+    if explicit_state_dir:
+        return Path(explicit_state_dir).expanduser().resolve()
+
+    cwd = Path.cwd()
+    if cwd.exists() and cwd.is_dir():
+        return project_state_dir()
+
+    return fallback_data_dir()
 
 
-def local_db_path(host: str = DEFAULT_HOST, port: int = DEFAULT_PORT) -> Path:
-    return local_data_dir() / f"auth_{_server_key(host, port)}.db"
+def local_db_path() -> Path:
+    return local_data_dir() / "auth.db"
 
 
-def local_log_path(host: str = DEFAULT_HOST, port: int = DEFAULT_PORT) -> Path:
-    return local_data_dir() / f"server_{_server_key(host, port)}.log"
+def local_log_path() -> Path:
+    return local_data_dir() / "server.log"
 
 
-def local_pid_path(host: str = DEFAULT_HOST, port: int = DEFAULT_PORT) -> Path:
-    return local_data_dir() / f"server_{_server_key(host, port)}.pid"
+def local_pid_path() -> Path:
+    return local_data_dir() / "server.pid"
 
 
 def ensure_local_env(host: str = DEFAULT_HOST, port: int = DEFAULT_PORT) -> dict[str, str]:
@@ -44,9 +55,10 @@ def ensure_local_env(host: str = DEFAULT_HOST, port: int = DEFAULT_PORT) -> dict
 
     env = os.environ.copy()
     env.setdefault("STORAGE_BACKEND", "sqlite")
-    env.setdefault("DATABASE_URL", f"sqlite:///{local_db_path(host, port)}")
+    env.setdefault("DATABASE_URL", f"sqlite:///{local_db_path()}")
     env.setdefault("JWT_SECRET_KEY", f"agentauth_local_{secrets.token_urlsafe(24)}")
     env.setdefault("CORS_ORIGINS", "*")
+    env.setdefault("AGENTAUTH_STATE_DIR", str(data_dir))
     return env
 
 
@@ -59,7 +71,7 @@ def is_running(pid: int) -> bool:
 
 
 def read_pid(host: str = DEFAULT_HOST, port: int = DEFAULT_PORT) -> int | None:
-    pid_path = local_pid_path(host, port)
+    pid_path = local_pid_path()
     if not pid_path.exists():
         return None
     try:
@@ -69,7 +81,7 @@ def read_pid(host: str = DEFAULT_HOST, port: int = DEFAULT_PORT) -> int | None:
 
 
 def write_pid(pid: int, host: str = DEFAULT_HOST, port: int = DEFAULT_PORT) -> None:
-    local_pid_path(host, port).write_text(str(pid))
+    local_pid_path().write_text(str(pid))
 
 
 def server_url(host: str = DEFAULT_HOST, port: int = DEFAULT_PORT) -> str:
@@ -96,7 +108,7 @@ def start_local_server(host: str = DEFAULT_HOST, port: int = DEFAULT_PORT) -> tu
         return None, base_url
 
     env = ensure_local_env(host, port)
-    log_path = local_log_path(host, port)
+    log_path = local_log_path()
     log_path.parent.mkdir(parents=True, exist_ok=True)
 
     cmd = [

{agentauthlayer-0.1.6 → agentauthlayer-0.1.8}/agentauthlayer.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentauthlayer
-Version: 0.1.6
+Version: 0.1.8
 Summary: Library-first authentication and authorization SDK for AI agents
 Author: Vaibhav Ahluwalia
 License: MIT

{agentauthlayer-0.1.6 → agentauthlayer-0.1.8}/auth_app/repositories/sqlite_user_repo.py

@@ -50,9 +50,12 @@ class SQLiteUserRepository(UserRepositoryProtocol):
         self._conn.commit()
 
     def get_admin_hash(self) -> str | None:
-        cur = self._conn.execute("SELECT password_hash FROM users WHERE email = ?", ("admin@agent-auth.local",))
-        row = cur.fetchone()
-        return row[0] if row else None
+        for email in ("admin@agentauth.dev", "admin@agent-auth.local"):
+            cur = self._conn.execute("SELECT password_hash FROM users WHERE email = ?", (email,))
+            row = cur.fetchone()
+            if row:
+                return row[0]
+        return None
 
     def set_admin_hash(self, password_hash: str) -> None:
         user = self.get_user_by_email('admin@agent-auth.local')

{agentauthlayer-0.1.6 → agentauthlayer-0.1.8}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "agentauthlayer"
-version = "0.1.6"
+version = "0.1.8"
 description = "Library-first authentication and authorization SDK for AI agents"
 readme = "README.md"
 requires-python = ">=3.10"