Airflow Summit 2026 is coming August 31 - September 2 in Austin, TX. Register now to secure your spot!

tests.system.microsoft.azure.example_azure_ai_agents

System test for Azure AI Foundry Hosted agent operators.

Requires a real Azure AI Foundry project and a Hosted agent container image pushed to Azure Container Registry:

  • AIRFLOW_CONN_AZURE_AI_AGENTS_DEFAULT: Azure AI Agents connection URI.

  • AZURE_AI_AGENTS_ENDPOINT: Azure AI Foundry project endpoint.

  • AZURE_AI_AGENTS_CONTAINER_IMAGE: Hosted agent container image URI.

  • AZURE_AI_AGENTS_MODEL_DEPLOYMENT_NAME: Model deployment available in the project.

  • AZURE_AI_AGENTS_RUN_PROTOCOL: Optional runtime protocol to invoke (responses or invocations).

  • AZURE_AI_AGENTS_USE_MODEL: Optional, defaults to false for a deterministic smoke test.

  • AZURE_AI_AGENTS_USE_MOCKS: Optional, defaults to true so the sample agent does not need external service credentials.

Attributes

DAG_ID

ENV_ID

AZURE_AI_AGENTS_CONN_ID

AZURE_AI_AGENTS_CONN_URI

ENDPOINT

MODEL_DEPLOYMENT_NAME

CONTAINER_IMAGE

AGENT_NAME

RUN_AGENT_PROTOCOL

HOSTED_AGENT_PROTOCOL

run_agent

HOSTED_AGENT_DEFINITION

UPDATED_AGENT_DEFINITION

SMOKE_FAILURE_CONTEXT

RUN_AGENT_PROMPT

RUN_AGENT_INPUT

CONN_FILE_PATH

create_agent

test_run

Functions

create_connection_file(conn_id_name, conn_uri)

Module Contents

tests.system.microsoft.azure.example_azure_ai_agents.DAG_ID = 'example_azure_ai_agents'[source]
tests.system.microsoft.azure.example_azure_ai_agents.ENV_ID[source]
tests.system.microsoft.azure.example_azure_ai_agents.AZURE_AI_AGENTS_CONN_ID[source]
tests.system.microsoft.azure.example_azure_ai_agents.AZURE_AI_AGENTS_CONN_URI[source]
tests.system.microsoft.azure.example_azure_ai_agents.ENDPOINT[source]
tests.system.microsoft.azure.example_azure_ai_agents.MODEL_DEPLOYMENT_NAME[source]
tests.system.microsoft.azure.example_azure_ai_agents.CONTAINER_IMAGE[source]
tests.system.microsoft.azure.example_azure_ai_agents.AGENT_NAME[source]
tests.system.microsoft.azure.example_azure_ai_agents.RUN_AGENT_PROTOCOL[source]
tests.system.microsoft.azure.example_azure_ai_agents.HOSTED_AGENT_PROTOCOL[source]
tests.system.microsoft.azure.example_azure_ai_agents.run_agent: airflow.providers.microsoft.azure.operators.ai_agents.RunAzureAIAgentOperator | None[source]
tests.system.microsoft.azure.example_azure_ai_agents.HOSTED_AGENT_DEFINITION: dict[str, Any][source]
tests.system.microsoft.azure.example_azure_ai_agents.UPDATED_AGENT_DEFINITION: dict[str, Any][source]
tests.system.microsoft.azure.example_azure_ai_agents.SMOKE_FAILURE_CONTEXT[source]
tests.system.microsoft.azure.example_azure_ai_agents.RUN_AGENT_PROMPT[source]
tests.system.microsoft.azure.example_azure_ai_agents.RUN_AGENT_INPUT[source]
tests.system.microsoft.azure.example_azure_ai_agents.create_connection_file(conn_id_name, conn_uri)[source]
tests.system.microsoft.azure.example_azure_ai_agents.CONN_FILE_PATH = None[source]
tests.system.microsoft.azure.example_azure_ai_agents.create_agent[source]
tests.system.microsoft.azure.example_azure_ai_agents.test_run[source]

Was this entry helpful?