Building Agents with Azure OpenAI

pip install -U swarms

export AZURE_API_KEY="your-azure-openai-key"
export AZURE_API_BASE="https://your-resource-name.openai.azure.com/"
export AZURE_API_VERSION="2024-08-01-preview"

from swarms import Agent

agent = Agent(
    agent_name="Azure-Agent",
    model_name="azure/gpt-4.1",         # your Azure deployment name
    max_loops=1,
)

print(agent.run("Summarize the case for private cloud AI in three paragraphs."))

from swarms import Agent

def get_weather(city: str) -> str:
    """Return the current weather for a city."""
    return f"{city}: 21°C, partly cloudy"

agent = Agent(
    agent_name="Azure-Assistant",
    model_name="azure/gpt-4.1",
    tools=[get_weather],
    max_loops=3,
)

print(agent.run("What's the weather in Tokyo right now?"))

from swarms import Agent

agent = Agent(
    agent_name="Streaming-Azure",
    model_name="azure/gpt-4.1",
    streaming_on=True,
    max_loops=1,
)

agent.run("Walk me through how Azure Active Directory federates with Okta.")

from swarms import Agent

prod_agent = Agent(
    agent_name="Prod",
    model_name="azure/gpt-4-prod-50k",
    max_loops=1,
)

dev_agent = Agent(
    agent_name="Dev",
    model_name="azure/gpt-4-dev-5k",
    max_loops=1,
)

export AZURE_API_BASE="https://private-endpoint.privatelink.openai.azure.com/"

from swarms import Agent, SequentialWorkflow

triage = Agent(
    agent_name="Triage",
    model_name="azure/gpt-4o-mini",         # cheap deployment
    system_prompt="Classify and route the request.",
    max_loops=1,
)

analyst = Agent(
    agent_name="Analyst",
    model_name="azure/gpt-4.1",             # workhorse deployment
    system_prompt="Produce a detailed analysis.",
    max_loops=1,
)

pipeline = SequentialWorkflow(agents=[triage, analyst], max_loops=1)
print(pipeline.run("Evaluate whether we should migrate our auth provider from Okta to Entra ID."))

from swarms import Agent

agent = Agent(
    agent_name="Production-Azure",
    model_name="azure/gpt-4.1",
    max_loops=1,
    persistent_memory=True,
    context_compression=True,
    context_length=128_000,
    autosave=True,
    retry_attempts=3,
    print_on=False,
)