from pydantic import BaseModel, Field
from azure.ai.openai import AzureOpenAIAssistantsClient
from azure.identity import DefaultAzureCredential

cred = DefaultAzureCredential()
deployment = "my-deployment"

client = AzureOpenAIAssistantsClient(credential=cred, deployment_name=deployment)

class RouterOutput(BaseModel):
    next: str = Field(..., description="which agent to route to")
    instruction: str = Field(..., description="what instruction to give")

agent = client.create_agent(
    name="router",
    description="Routes tasks to agents",
    instructions="Route to correct agent based on message intent",
    tools=[],
    response_format=RouterOutput,
    chat_message_store_factory=None,
)

response = agent.run("high metrics alert in vm")
print(response.text)