Skip to content

fix(ai-sdk): Add model mapping to azure#892

Merged
nina-kollman merged 1 commit intomainfrom
nk/azure_vendor
Mar 4, 2026
Merged

fix(ai-sdk): Add model mapping to azure#892
nina-kollman merged 1 commit intomainfrom
nk/azure_vendor

Conversation

@nina-kollman
Copy link
Contributor

@nina-kollman nina-kollman commented Mar 2, 2026

Fixes: TLP-1863


Important

Add Azure to AI SDK vendor mapping and test its transformation to Azure system.

  • Behavior:
    • Add azure to VENDOR_MAPPING in ai-sdk-transformations.ts to map to Azure.
    • Add test case in ai-sdk-transformations.test.ts to verify transformation of azure provider to Azure system.
  • Tests:
    • Add test for azure provider transformation in ai-sdk-transformations.test.ts.
    • Ensure azure.chat, azure.completions, and azure are correctly transformed.

This description was created by Ellipsis for 4a1e238. You can customize this summary. It will automatically update as commits are pushed.

Summary by CodeRabbit

  • New Features
    • Added Azure vendor provider support. Azure-specific provider identifiers are now properly recognized and normalized in LLM tracing workflows. This enhancement improves data accuracy and service compatibility when instrumenting applications integrated with Azure-based AI and language model services.

Copy link
Contributor

@ellipsis-dev ellipsis-dev bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Important

Looks good to me! 👍

Reviewed everything up to 4a1e238 in 10 seconds. Click for details.
  • Reviewed 39 lines of code in 2 files
  • Skipped 0 files when reviewing.
  • Skipped posting 0 draft comments. View those below.
  • Modify your settings and rules to customize what types of comments Ellipsis leaves. And don't forget to react with 👍 or 👎 to teach Ellipsis.

Workflow ID: wflow_EmMUbYeyfJISrniR

You can customize Ellipsis by changing your verbosity settings, reacting with 👍 or 👎, replying to comments, or adding code review rules.

@coderabbitai
Copy link
Contributor

coderabbitai bot commented Mar 2, 2026

No actionable comments were generated in the recent review. 🎉

ℹ️ Recent review info

Configuration used: defaults

Review profile: CHILL

Plan: Pro

Disabled knowledge base sources:

  • Linear integration is disabled

You can enable these sources in your CodeRabbit configuration.

📥 Commits

Reviewing files that changed from the base of the PR and between ddb1241 and 4a1e238.

📒 Files selected for processing (2)
  • packages/traceloop-sdk/src/lib/tracing/ai-sdk-transformations.ts
  • packages/traceloop-sdk/test/ai-sdk/ai-sdk-transformations.test.ts

📝 Walkthrough

Walkthrough

This change adds Azure vendor prefix recognition to the AI SDK tracing transformations and extends test coverage to validate Azure-specific provider string transformations and normalization.

Changes

Cohort / File(s) Summary
Azure Vendor Mapping
packages/traceloop-sdk/src/lib/tracing/ai-sdk-transformations.ts
Added "azure" vendor prefix mapping to VENDOR_MAPPING, enabling recognition of Azure as a vendor in AI SDK tracing transformations.
Azure Provider Tests
packages/traceloop-sdk/test/ai-sdk/ai-sdk-transformations.test.ts
Added test case validating transformation of Azure provider strings ("azure.chat", "azure.completions", "azure") to provider name "azure" with system "Azure", confirming removal of legacy "ai.model.provider" attribute.

Estimated code review effort

🎯 1 (Trivial) | ⏱️ ~3 minutes

Poem

🐰 Azure skies now clearly seen,
A rabbit maps what's in between,
One vendor joins the parsed array,
Tests confirm it works today!
Hops of joy for vendor play! ✨

🚥 Pre-merge checks | ✅ 3
✅ Passed checks (3 passed)
Check name Status Explanation
Description Check ✅ Passed Check skipped - CodeRabbit’s high-level summary is enabled.
Title check ✅ Passed The title 'fix(ai-sdk): Add model mapping to azure' directly describes the main change—adding Azure vendor mapping in the AI SDK transformation logic.
Docstring Coverage ✅ Passed No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check.

✏️ Tip: You can configure your own custom pre-merge checks in the settings.

✨ Finishing Touches
  • 📝 Generate docstrings (stacked PR)
  • 📝 Generate docstrings (commit on current branch)
🧪 Generate unit tests (beta)
  • Create PR with unit tests
  • Post copyable unit tests in a comment
  • Commit unit tests in branch nk/azure_vendor

Tip

Try Coding Plans. Let us write the prompt for your AI agent so you can ship faster (with fewer bugs).
Share your feedback on Discord.


Comment @coderabbitai help to get the list of available commands and usage tips.

@nina-kollman nina-kollman merged commit e280ceb into main Mar 4, 2026
8 of 9 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants