Usage not included with Azure OpenAI in streaming mode

[lavisht22]

What Happened?

When in streaming mode and the provider is set to "azure-openai", the last chunk containing the usage is not returned, where as it is returned from the original API.

I tested the original API and via Portkey using same configuration

{
    "model": "gpt-4o",
    "messages": [
        {
            "role": "system",
            "content": "You are a helpful assistant."
        }
    ],
    "stream": true,
    "stream_options": {
        "include_usage": true
    }
}

What Should Have Happened?

The stream should include the last message containing the usage details.

Relevant Code Snippet

No response

Your Twitter/LinkedIn

lthakkar22