AutoGen Integration

# For AutoGen 0.4+ (recommended)
pip install sentrial autogen-agentchat>=0.4.0

# For legacy pyautogen
pip install sentrial pyautogen>=0.2.0

from autogen_agentchat.agents import AssistantAgent
from autogen_agentchat.teams import RoundRobinGroupChat
from autogen_ext.models.openai import OpenAIChatCompletionClient
from sentrial import SentrialClient
from sentrial.autogen import SentrialAutogenHandler

# Initialize
client = SentrialClient(api_key="sentrial_live_xxx")
session_id = client.create_session(
    name="Code Review Team",
    agent_name="autogen-review",
    user_id="user_123"
)

# Create handler
handler = SentrialAutogenHandler(client, session_id, verbose=True)

# Create model client
model_client = OpenAIChatCompletionClient(model="gpt-4o")

# Create agents
reviewer = AssistantAgent(
    name="CodeReviewer",
    model_client=model_client,
    system_message="You review code for bugs.",
)

security = AssistantAgent(
    name="SecurityExpert",
    model_client=model_client,
    system_message="You analyze security vulnerabilities.",
)

# Create team
team = RoundRobinGroupChat(
    participants=[reviewer, security],
    max_turns=4,
)

# Run with tracking
result = await handler.run_team(team, task="Review this code...")

# Finish
handler.finish(success=True)

from autogen import AssistantAgent, UserProxyAgent
from sentrial import SentrialClient
from sentrial.autogen import SentrialAutogenHandler

client = SentrialClient(api_key="sentrial_live_xxx")
session_id = client.create_session(
    name="AutoGen Chat",
    agent_name="autogen-chat",
    user_id="user_123"
)

handler = SentrialAutogenHandler(client, session_id)

# Create agents
assistant = AssistantAgent("assistant", llm_config=llm_config)
user_proxy = UserProxyAgent("user_proxy")

# Register for tracking
handler.register_agents([assistant, user_proxy])

# Chat
user_proxy.initiate_chat(assistant, message="Hello!")

handler.finish(success=True)