AuthSec integration for AutoGen — secure AI delegation token retrieval
Project description
AuthSec AI — AutoGen Integration
Function-calling delegation — an AutoGen AssistantAgent directs a UserProxyAgent to invoke an AuthSec-secured Python function that retrieves confidential data from a protected backend.
Architecture
AutoGen Function (registered in main.py)
↓
authsec_helper.py — thin wrapper, MOCK fallback, debug logging
↓
authsec-langchain-sdk — official SDK (PyPI)
↓
AuthSec Delegation Backend
↓
Short-lived JWT (RS256, scoped, SPIFFE-identified)
↓
Protected API (with Authorization: Bearer header)
The official SDK owns the entire delegation-token lifecycle.
authsec_helper.py is a thin, framework-agnostic wrapper that adds MOCK fallback and structured console logging — it never reimplements the token exchange itself.
File Structure
| File | Role |
|---|---|
authsec_helper.py |
Thin wrapper around authsec-langchain-sdk; MOCK fallback, debug logging |
main.py |
Entry point; SecurityAssistant + UserProxy agent demo, fetch_secure_data() function |
requirements.txt |
Dependencies including authsec-langchain-sdk |
Prerequisites
| Requirement | Details |
|---|---|
| Python | 3.10 or later |
| AuthSec account | Registered AI agent client on the AuthSec dashboard |
| Trust delegation | Created with target type "Langchain AI agent" |
authsec-langchain-sdk |
pip install authsec-langchain-sdk |
Installation & Setup
cd integrations/autogen
pip install -r requirements.txt
requirements.txt pulls in pyautogen, requests, authsec-langchain-sdk, and supporting packages.
Running the Demo
MOCK Mode — zero setup
python main.py
No environment variables required. When AUTHSEC_BASE_URL and AUTHSEC_AGENT_CLIENT_ID are absent, the wrapper falls back to an offline mock mode with demo data. Useful for exploring the integration pattern without an AuthSec account.
LIVE Mode — against production
# Windows PowerShell
$env:AUTHSEC_BASE_URL = "https://prod.api.authsec.ai"
$env:AUTHSEC_AGENT_CLIENT_ID = "your-agent-client-uuid"
$env:OPENAI_API_KEY = "sk-..." # optional — enables real LLM agent
python main.py
# bash / zsh
export AUTHSEC_BASE_URL="https://prod.api.authsec.ai"
export AUTHSEC_AGENT_CLIENT_ID="your-agent-client-uuid"
export OPENAI_API_KEY="sk-..."
python main.py
When OPENAI_API_KEY is set, main.py launches a full AutoGen conversational loop between SecurityAssistant and UserProxy. Without it, the demo runs a high-fidelity simulation that still exercises the real AuthSec token flow.
Example Output (LIVE mode, verified against production)
[AuthSec SDK] -- Initializing AuthSec Client (AutoGen) ------------
[AuthSec SDK] [Mode] LIVE — using official authsec-langchain-sdk
|- Base URL : https://prod.api.authsec.ai
|- Client ID : fe6d5a81-58ac-4c4b-85fa-f84b6c9cb73d
[AuthSec SDK] [Init] Official authsec-langchain-sdk client initialized.
[AuthSec SDK] [Delegation] Requesting delegation token via official authsec-langchain-sdk...
[AuthSec SDK] [Success] LIVE delegation token acquired via official SDK.
|- Token : eyJhbGciOiJSUzI1NiIsInR5...
How the Integration Works
-
Agent setup —
SecurityAssistant(cognitive layer, GPT-4) andUserProxy(execution layer,human_input_mode="NEVER") are created using standard AutoGen configuration. -
Function registration —
fetch_secure_data()is registered viaautogen.register_function()so the assistant can propose calls and the proxy can execute them. -
SDK initialization — at module load,
AuthSecClient()readsAUTHSEC_BASE_URLandAUTHSEC_AGENT_CLIENT_IDand initializes anAuthsecClientfrom the officialauthsec-langchain-sdk. -
Conversational trigger — during the chat loop,
SecurityAssistantgenerates a function call. AutoGen routes execution toUserProxy, which invokesfetch_secure_data(). -
Delegation token retrieval — inside the function,
client.get_delegation_token()delegates to:GET /authsec/uflow/sdk/delegation-token?client_id=<uuid>The SDK handles the exact request contract required by the AuthSec backend (Trust Delegation target type: "Langchain AI agent").
-
Downstream API call — the JWT is passed as
Authorization: Bearer <token>to the protected endpoint viaclient.request_secure_api(). -
Agent analysis —
SecurityAssistantreceives the JSON payload, summarizes it, and terminates the session withTERMINATE.
Environment Variables Reference
| Variable | Required | Description |
|---|---|---|
AUTHSEC_BASE_URL |
Yes (LIVE) | AuthSec server root, e.g. https://prod.api.authsec.ai |
AUTHSEC_AGENT_CLIENT_ID |
Yes (LIVE) | Agent's client UUID from the AuthSec dashboard |
OPENAI_API_KEY |
Optional | Enables a real LLM-powered AutoGen agent; simulation runs without it |
Security Model
AuthSec replaces static, long-lived API keys with short-lived delegation tokens. Each token is a signed RS256 JWT scoped to the specific permissions granted through the trust delegation. Tokens expire automatically — the SDK refreshes them transparently — so there is no secret to rotate or leak into version control.
Every delegation token carries a SPIFFE identity (spiffe://authsec.dev/ns/default/sa/ai-agent) that cryptographically binds the token to a specific agent workload. Downstream APIs verify both the JWT signature and the SPIFFE subject, establishing a chain of trust from the AuthSec backend through the SDK to the protected resource.
This model means the AI agent never handles raw credentials for the target API. The AuthSec backend acts as a trust broker: it evaluates the delegation policy, mints a scoped token, and the agent simply presents that token. Revoking access is a single dashboard action — no credential rotation required.
Debug Logging
The wrapper emits structured, colour-coded console output under the [AuthSec SDK] prefix:
| Prefix | Meaning |
|---|---|
[Mode] LIVE |
SDK initialized, env vars detected |
[Mode] MOCK |
No env vars — running offline |
[Init] |
SDK client construction result |
[Delegation] |
Token request in progress |
[Success] |
Token acquired or API call succeeded |
[Error] |
SDK call failed — fallback triggered |
[Token] … [MOCK FALLBACK] |
Using locally generated mock JWT |
Troubleshooting
| Symptom | Cause | Fix |
|---|---|---|
404 Client not found |
client_id not registered, or trust delegation not created |
Verify the UUID on the AuthSec dashboard; ensure a trust delegation exists with target type "Langchain AI agent" |
ImportError: authsec_langchain |
SDK not installed | pip install authsec-langchain-sdk |
| SSL / connection errors | Wrong URL scheme or unreachable host | Confirm AUTHSEC_BASE_URL uses https:// for production |
400 Invalid client_id format |
Trailing whitespace or characters in UUID | Check for copy-paste artefacts in the env var |
A Note on secure-vault/… Endpoints
The
secure-vault/metrics,secure-vault/records, etc. endpoints used in this demo are illustrative mock resources. They are intentionally not hosted onprod.api.authsec.ai. The delegation token flow succeeds against production — you will see a valid JWT returned — but the subsequent call tosecure-vault/*will return a 404. This is expected behaviour, not a bug. In a real deployment you would replace these paths with your own protected API endpoints.
Further Reading
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file authsec_autogen-0.1.0.tar.gz.
File metadata
- Download URL: authsec_autogen-0.1.0.tar.gz
- Upload date:
- Size: 8.2 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.0.1 CPython/3.11.3
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
65cee01f4958cb7d961be50a6fd30f469570e7e64069ddbc1eeda44d7e35679c
|
|
| MD5 |
a7862c64741745aa8cb19584197694d9
|
|
| BLAKE2b-256 |
43345fbd54abeed54a268b3c0d2de988e82880d15a0829de44f56a08fa4151d8
|
File details
Details for the file authsec_autogen-0.1.0-py3-none-any.whl.
File metadata
- Download URL: authsec_autogen-0.1.0-py3-none-any.whl
- Upload date:
- Size: 8.2 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.0.1 CPython/3.11.3
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
96962af781c2fe4b8e4f38839ccd8246db7475dab74e74a5ee225592902d282f
|
|
| MD5 |
c63ffb535cba1d8690faaec8b738ef49
|
|
| BLAKE2b-256 |
5ab68b00a3fc659fa2a9c0aeb33dc8275f9749a8830ce33fe714ba68ee2e9edb
|