feat(node): Add tracing support for AzureOpenAI #18281
Conversation
![cursor[bot]](https://avatars.githubusercontent.com/in/1210556?s=60&v=4){kind=small}
There was a problem hiding this comment.
Pull Request Overview
This pull request adds tracing support for the Azure OpenAI client in addition to the existing OpenAI client support, resolving issue #18280.
Key Changes:
- Extended the OpenAI instrumentation to also patch
AzureOpenAIclient constructor - Added test coverage for Azure OpenAI client in both default and v6 test suites
Reviewed Changes
Copilot reviewed 5 out of 5 changed files in this pull request and generated 4 comments.
Show a summary per file
| File | Description |
|---|---|
packages/node/src/integrations/tracing/openai/instrumentation.ts |
Refactored patching logic to support both OpenAI and AzureOpenAI clients by extracting common wrapping logic into a reusable _patchClient method |
dev-packages/node-integration-tests/suites/tracing/openai/v6/test.ts |
Added test case for Azure OpenAI client with v6 OpenAI SDK |
dev-packages/node-integration-tests/suites/tracing/openai/v6/scenario-azure-openai.mjs |
Created test scenario that instantiates and uses AzureOpenAI client with mock server for v6 |
dev-packages/node-integration-tests/suites/tracing/openai/test.ts |
Added test case for Azure OpenAI client with default OpenAI SDK version |
dev-packages/node-integration-tests/suites/tracing/openai/scenario-azure-openai.mjs |
Created test scenario that instantiates and uses AzureOpenAI client with mock server |
💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.
packages/node/src/integrations/tracing/openai/instrumentation.ts
Outdated
Show resolved
Hide resolved
packages/node/src/integrations/tracing/openai/instrumentation.ts
Outdated
Show resolved
Hide resolved
RulaKhaled
left a comment
RulaKhaled
left a comment
There was a problem hiding this comment.
Very nice, thanks for contributing!
As an improvement, we soon will tweak this a little to allow users to differentiate between azure and vanilla openai, but this is fine for now.
I’m not sure how to move forward with this PR. There are what I believe to be random test failures. |
2 participants
This pull request adds the support to Azure OpenAI client in addition to the existing support of the vanilla OpenAI client.
Fixes issue #18280