85742ab165
Deploy Documentation / deploy (push) Has been cancelled
CPU Test / Test (Utilities, legacy, Python 3.10) (push) Has been cancelled
CPU Test / Test (LLM proxy, stable, Python 3.11) (push) Has been cancelled
CPU Test / Test (Others, stable, Python 3.11) (push) Has been cancelled
CPU Test / Test (Store, stable, Python 3.11) (push) Has been cancelled
CPU Test / Test (Utilities, stable, Python 3.11) (push) Has been cancelled
CPU Test / Test (Weave, stable, Python 3.11) (push) Has been cancelled
CPU Test / Test (AgentOps, stable, Python 3.12) (push) Has been cancelled
CPU Test / Test (LLM proxy, stable, Python 3.12) (push) Has been cancelled
CPU Test / Test (Others, stable, Python 3.12) (push) Has been cancelled
CPU Test / Test (Weave, latest, Python 3.13) (push) Has been cancelled
Dashboard / Chromatic (push) Has been cancelled
CPU Test / Lint - fast (push) Has been cancelled
CPU Test / Lint - next (push) Has been cancelled
CPU Test / Lint - slow (push) Has been cancelled
CPU Test / Lint - JavaScript (push) Has been cancelled
CPU Test / Build documentation (push) Has been cancelled
CPU Test / Test (AgentOps, legacy, Python 3.10) (push) Has been cancelled
CPU Test / Test (LLM proxy, legacy, Python 3.10) (push) Has been cancelled
CPU Test / Test (Others, legacy, Python 3.10) (push) Has been cancelled
CPU Test / Test (Store, legacy, Python 3.10) (push) Has been cancelled
CPU Test / Test (Weave, legacy, Python 3.10) (push) Has been cancelled
CPU Test / Test (AgentOps, stable, Python 3.11) (push) Has been cancelled
CPU Test / Test (Store, stable, Python 3.12) (push) Has been cancelled
CPU Test / Test (Utilities, stable, Python 3.12) (push) Has been cancelled
CPU Test / Test (Weave, stable, Python 3.12) (push) Has been cancelled
CPU Test / Test (AgentOps, latest, Python 3.13) (push) Has been cancelled
CPU Test / Test (LLM proxy, latest, Python 3.13) (push) Has been cancelled
CPU Test / Test (Others, latest, Python 3.13) (push) Has been cancelled
CPU Test / Test (Store, latest, Python 3.13) (push) Has been cancelled
CPU Test / Test (Utilities, latest, Python 3.13) (push) Has been cancelled
CPU Test / Test (JavaScript) (push) Has been cancelled
Agent-OS Integration for Agent-Lightning
Kernel-level safety during AI agent training.
Overview
Agent-OS provides deterministic governance for AI agents. This integration enables:
- 0% unpenalized policy violations — All unsafe actions are detected and penalized
- Policy violations → RL penalties — Agents learn to avoid unsafe behavior
- Complete audit trail — From training to production
Installation
pip install agentlightning agent-os
Quick Start
from agentlightning import Trainer
from agentlightning.contrib.runner.agentos import AgentOSRunner
from agentlightning.contrib.reward.agentos import PolicyReward
from agent_os import KernelSpace
from agent_os.policies import SQLPolicy
# Create governed kernel
kernel = KernelSpace(policy=SQLPolicy(
deny=["DROP", "DELETE"]
))
# Wrap in Agent-OS runner
runner = AgentOSRunner(kernel)
# Train with policy-aware rewards
trainer = Trainer(
runner=runner,
reward_fn=PolicyReward(kernel),
algorithm="GRPO"
)
trainer.train()
Components
AgentOSRunner
Wraps agent execution with kernel-level policy enforcement:
from agentlightning.contrib.runner.agentos import AgentOSRunner
runner = AgentOSRunner(
kernel,
fail_on_violation=False, # Continue but penalize
emit_violations=True, # Emit as spans
)
PolicyReward
Converts policy violations to negative RL rewards:
from agentlightning.contrib.reward.agentos import PolicyReward
reward_fn = PolicyReward(
kernel,
base_reward_fn=accuracy_reward,
critical_penalty=-100.0,
clean_bonus=5.0,
)
FlightRecorderAdapter
Imports Agent-OS audit logs to LightningStore:
from agentlightning.contrib.adapter.agentos import FlightRecorderAdapter
adapter = FlightRecorderAdapter(flight_recorder)
adapter.import_to_store(lightning_store)
Benchmarks
| Metric | Without Agent-OS | With Agent-OS |
|---|---|---|
| Undetected Policy Violations | 12.3% | 0.0% |
| Task Accuracy | 76.4% | 79.2% |
Note: "0% undetected violations" means all policy violations are caught and penalized, not that agents never attempt unsafe actions. Over training, agents learn to minimize violation attempts.
Documentation
- Agent-OS Documentation
- Integration guide: see project README or examples in this directory.
License
MIT