Email and identity
for AutoGen agents.

# Create one identity per agent (or do this in your bootstrap script)
curl -X POST https://api.loomal.ai/v0/identities \
  -H "Authorization: Bearer $LOOMAL_ROOT_KEY" \
  -d '{"name": "researcher"}'
# Repeat for writer, reviewer, etc.

pip install autogen-agentchat autogen-ext[openai,mcp]

import os
from autogen_agentchat.agents import AssistantAgent
from autogen_ext.tools.mcp import McpWorkbench, StdioServerParams
from autogen_ext.models.openai import OpenAIChatCompletionClient

def loomal_workbench(api_key: str) -> McpWorkbench:
    return McpWorkbench(StdioServerParams(
        command="npx",
        args=["-y", "@loomal/mcp"],
        env={"LOOMAL_API_KEY": api_key},
    ))

model = OpenAIChatCompletionClient(model="gpt-4o-mini")

researcher = AssistantAgent(
    "researcher",
    model_client=model,
    workbench=loomal_workbench(os.environ["LOOMAL_RESEARCHER_KEY"]),
    system_message="Find prospects, then email a brief to writer-y@loomal.ai.",
)

writer = AssistantAgent(
    "writer",
    model_client=model,
    workbench=loomal_workbench(os.environ["LOOMAL_WRITER_KEY"]),
    system_message="Read mail from researcher-x@loomal.ai. Draft outreach for each prospect.",
)

import asyncio
from autogen_agentchat.teams import RoundRobinGroupChat
from autogen_agentchat.conditions import MaxMessageTermination

async def main():
    async with researcher.workbench, writer.workbench:
        team = RoundRobinGroupChat(
            [researcher, writer],
            termination_condition=MaxMessageTermination(8),
        )
        await team.run(task="Find 3 SaaS companies hiring ML engineers and draft outreach.")

asyncio.run(main())

from autogen_agentchat.agents import UserProxyAgent

proxy = UserProxyAgent(
    "proxy",
    workbench=loomal_workbench(os.environ["LOOMAL_PROXY_KEY"]),
)

# In a tool the proxy calls, retrieve the key from the vault:
#   key = vault.get(label="github-token")
# The model only sees that the tool succeeded, never the key value.