Argorix SDK for Python

SDK oficial de Python para integrar aplicaciones de IA con el Argorix Guardrails Runtime.

pip install argorix-guardrails

Rebranding. Este paquete se llamaba governanceai. El nombre governanceai sigue publicado como shim de compatibilidad (depende de argorix-guardrails y lo reexporta), pero ya no recibe features. Ver Migración desde governanceai.

API cubierta

Los guardrails de agentes (/v1/agent-guardrails/runtime/*) viven en el paquete argorix-guardrails-agent.

base_url acepta tanto https://api.argorix.com como https://api.argorix.com/v1: el sufijo /v1 se normaliza para no duplicar el prefijo.

Autenticación

• app_number en el body de cada request
• Authorization: Bearer <APP_API_KEY> en cada header

Ambos valores salen de Application Settings > API Keys en la consola.

Quick start

from argorix import ArgorixClient, ArgorixError

client = ArgorixClient(
    base_url="https://api.argorix.com",
    app_number=123456,
    app_api_key="ax_live_replace_me",
    timeout_seconds=10,
    max_retries=2,
)

try:
    state = client.install(mode="monitor", metadata={"environment": "production"})
    print(state.mode, state.selected_validators)

    decision = client.evaluate("Summarize this support ticket", stage="input")
    print(decision.allowed, decision.findings, decision.selected_validators)
except ArgorixError as exc:
    print(exc.status_code, exc)

Configuración por entorno

client = ArgorixClient.from_env()

Los argumentos explícitos siempre ganan sobre el entorno.

Flujo runtime clásico

decision_in = client.evaluate(user_prompt, stage="input")
if decision_in.blocked:
    raise RuntimeError(decision_in.findings)

model_reply = llm.invoke(user_prompt)

decision_out = client.evaluate(model_reply, stage="output")
if decision_out.blocked:
    raise RuntimeError(decision_out.findings)

Tool calls

decision = client.evaluate(
    "Open the customer export",
    stage="tool",
    tool_calls=[
        {"tool_name": "browser.fetch", "url": "https://example.com/private-report"}
    ],
)

Si el dominio no está permitido por la configuración de la aplicación, el backend puede responder bloqueando la operación.

Streaming (SSE)

evaluate_stream() consume POST /v1/guardrails/evaluate/stream y emite los eventos start, result, end — o error si la evaluación falla en el servidor.

for event in client.evaluate_stream(user_prompt, stage="input"):
    if event.event == "result":
        print("allowed:", event.data["allowed"])
    elif event.event == "error":
        print("guardrail error:", event.data["detail"])

Si solo te interesa la decisión final:

decision = client.evaluate_streamed_decision(user_prompt, stage="input")

Levanta ArgorixError si el servidor emite error o si el stream cierra sin result. El stream es perezoso: no se envía nada hasta que empiezas a iterar. Los reintentos cubren la conexión y el status inicial; una vez abierto el stream no se reintenta.

Modelos de respuesta

evaluate() devuelve GuardrailsDecision:

install() y heartbeat() devuelven GuardrailsState (application_id, app_number, repository, installation_connected, guardrails_config, effective_scope, selected_validators, mode, enabled, raw).

Ambos objetos aceptan acceso tipo diccionario (decision["repository"], state.get("app_number")) que lee directo de raw, así que campos nuevos del control plane quedan accesibles sin actualizar el SDK.

Telemetría

El cliente acumula requests_total, blocked_total y avg_latency_ms, y los adjunta a evaluate() y heartbeat() salvo que pases include_telemetry=False.

print(client.telemetry)
client.reset_telemetry()

Errores, timeout y retry

ArgorixClient expone timeout_seconds, max_retries, retry_backoff_seconds y retry_status_codes. Los errores levantan ArgorixError con status_code y response_body. Se reintentan errores de red y respuestas 408, 429, 500, 502, 503, 504 con backoff exponencial.

Cómo se refleja en la consola

Los eventos y evaluaciones de este SDK alimentan AI Applications, Guardrails Log, Risk & Governance y AI Compliance. En la consola:

• AI Applications: inventario resumido
• AI Application Profile: postura, evidencia y modelos/tools
• Application Settings: Installation, API Keys, Guardrails
• AI Workbench: validación de prompts y matriz TRUE/FALSE por control

AI-BOM CLI

argorix-bom --base-url https://api.argorix.com --cookie "argorix_session=..." generate --application-id travel-assistant --force
argorix-bom --base-url https://api.argorix.com --cookie "argorix_session=..." export --document-id bom-123 --format cyclonedx --output ai-bom.json
argorix-bom --base-url https://api.argorix.com --cookie "argorix_session=..." scan ./my-agent-repo --application-id my-agent-repo
argorix-bom --base-url https://api.argorix.com --cookie "argorix_session=..." scan ./my-agent-repo --format cyclonedx --output ai-bom-cyclonedx.json
argorix-bom --base-url https://api.argorix.com --cookie "argorix_session=..." scan ./my-agent-repo --watch --interval-seconds 3

Formatos: cyclonedx, spdx, sarif, markdown, html.

El flujo scan ejecuta POST /v1/scans/local, refresca el AI-BOM activo con POST /v1/ai-bom/applications/{application_id}/generate y opcionalmente exporta el documento. Sin --application-id, la CLI infiere un slug desde el nombre de la carpeta. En --watch no vuelve a pegarle al backend hasta detectar un cambio local.

Credencial: --cookie "argorix_session=...", ARGORIX_SESSION_COOKIE, o el legado GOVERNANCEAI_SESSION_COOKIE. Sigue siendo cookie de sesión del control-plane; migrarla a un token específico queda pendiente.

Migración desde governanceai

pip uninstall governanceai
pip install argorix-guardrails

Los nombres viejos siguen exportados desde argorix como alias, así que un cambio de import alcanza para arrancar. Cambios de comportamiento a revisar:

• install() y heartbeat() devuelven GuardrailsState en vez de dict. El acceso por clave sigue funcionando (state["application_id"]).
• decision.findings ahora son dataclasses GuardrailsFinding. finding["severity"] sigue funcionando; finding.severity es la forma recomendada.

Desarrollo

pip install -e ./sdk/python
python -m pytest ./sdk/python/tests
python -m build ./sdk/python

Semver y changelog

• Versión actual: 0.2.0
• Historial: CHANGELOG.md
• Licencia: LICENSE

Referencias

• sdk/README.md
• sdk/javascript/README.md
• sdk/python-agents/README.md
• example_usage.py