Using langchain with Azure Document Intelligence – no module ‘azure’

Just trying to run the demo code from langchain:

file_path = "TEST_EN.pptx"
endpoint = "my endpoint"
key = "my key"
analysis_features = ["ocrHighResolution"]
loader = AzureAIDocumentIntelligenceLoader(
    api_endpoint=endpoint,
    api_key=key,
    file_path=file_path,
    api_model="prebuilt-layout",
    analysis_features=analysis_features,
    #mode="page",
)

Gives me No module named ‘azure’.

I have installed:

azure-ai-formrecognizer
azure-ai-documentintelligence and even azure-core