TealTiger Governance Integration#

pip install "ag2-tealtiger>=0.1.0"

from ag2_tealtiger import TealTigerGuard

# Observe everything, block nothing
guard = TealTigerGuard()
guard.attach(my_agent)

# After some tool calls...
for entry in guard.audit_trail:
    print(f"{entry.agent_id}: {entry.tool_name} → {entry.action} (cost: ${entry.cost_tracked:.4f})")

from ag2_tealtiger import TealTigerGuard, GovernanceMode
from tealtiger import TealEngine

engine = TealEngine(policies=[
    {"type": "cost_limit", "max_per_session": 5.00},
    {"type": "pii_block", "categories": ["ssn", "credit_card"]},
    {"type": "tool_allowlist", "allowed": ["web_search", "read_file"]},
])

guard = TealTigerGuard(engine=engine, mode=GovernanceMode.ENFORCE)
guard.attach(agent)

# Tool calls are now evaluated against policies
# DENY blocks with structured denial message visible in conversation
# ALLOW passes through transparently

from ag2_tealtiger import TealTigerAuditAgent, GovernanceMode

# Zero-config — governance built in
agent = TealTigerAuditAgent(name="coder")

# With policy engine and budget
agent = TealTigerAuditAgent(
    name="executor",
    engine=my_engine,
    mode=GovernanceMode.ENFORCE,
    budget_limit=5.0,
)

# Access governance state
print(agent.audit_trail)
print(agent.summary)

from ag2_tealtiger import GovernedGroupChat, TealTigerGuard, GovernanceMode

guard = TealTigerGuard(engine=engine, mode=GovernanceMode.ENFORCE)

for agent in [coder, reviewer, executor]:
    guard.attach(agent)

group_chat = GovernedGroupChat(agents=[coder, reviewer, executor], guard=guard)
speaker = group_chat.select_speaker(last_speaker=coder)

# Frozen agents skipped (no engine call)
# Over-budget agents skipped (no engine call)
# Policy-denied agents skipped to next candidate
# ALL_SPEAKERS_DENIED terminates the round safely

# Freeze mid-conversation — blocks ALL actions regardless of mode
guard.freeze("dangerous_agent")

# Other agents continue normally
speaker = group_chat.select_speaker(last_speaker=coder)

# Unfreeze to restore normal governance
guard.unfreeze("dangerous_agent")

guard.set_budget("expensive_agent", limit=5.0)

# After cumulative cost exceeds limit:
# - ENFORCE: blocks with BUDGET_EXCEEDED
# - 80% warning emitted at threshold
# - Reset: guard.reset_budget("expensive_agent")

# When TealEngine returns REFER, action is suspended (not terminated)
# GroupChat continues with remaining agents
# The escalation receipt contains full context for a human reviewer

# Resolve later:
guard.resolve_refer(decision_id, resolution="ALLOW", approval_id="reviewer-123")

# Messages between agents are also governed
# Frozen sender → blocked regardless of mode
# Policy-denied message → blocked in ENFORCE, logged in MONITOR
# Each message gets its own decision_id (independent from tool governance)

# Export full governance audit as JSONL
entries_written = guard.export_audit_trail("audit.jsonl")

# Each line contains: agent_id, tool_name, action, risk_score,
# reason_codes, evaluation_time_ms, TEEC correlation context