Skip to content

[Feature]: Add Python Azure OpenAI support to traceai_openai #154

@VaishnavGunjari

Description

@VaishnavGunjari

The current OpenAIInstrumentor only patches openai.OpenAI. The AzureOpenAI client is not instrumented, so Azure OpenAI calls are silently untraced.
This is a significant gap since Azure OpenAI is widely used in enterprise environments.

Proposed solution:
Extend OpenAIInstrumentor to wrap AzureOpenAI with Azure-specific span attributes (deployment_name, api_version, endpoint, gen_ai.system = "azure_openai").

I'd like to contribute this if it aligns with the roadmap. Happy to discuss before opening a PR.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions