AgentShield for AutoGen

Monitor AutoGen multi-agent conversations. Track token costs per agent, replay conversations, set budget caps.

Quick setup

import agentshield
from autogen import AssistantAgent, UserProxyAgent

shield = agentshield.init(api_key="your-key")

with shield.session(agent_id="autogen-crew") as session:
    assistant = AssistantAgent("assistant", llm_config=llm_config)
    user_proxy = UserProxyAgent("user_proxy")
    user_proxy.initiate_chat(assistant, message=task)
    session.track(messages=assistant.chat_messages)