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
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions