fix: azure vendor

Author: nirga

packages/traceloop-sdk/src/lib/tracing/ai-sdk-transformations.ts

@@ -24,7 +24,7 @@ const AI_PROMPT_TOOLS = "ai.prompt.tools";
 // Uses prefixes to match AI SDK patterns like "openai.chat", "anthropic.messages", etc.
 const VENDOR_MAPPING: Record<string, string> = {
   openai: "OpenAI",
-  "azure-openai": "OpenAI",
+  "azure-openai": "Azure",
   anthropic: "Anthropic",
   cohere: "Cohere",
   mistral: "MistralAI",
@@ -36,7 +36,6 @@ const VENDOR_MAPPING: Record<string, string> = {
   perplexity: "Perplexity",
   "amazon-bedrock": "AWS",
   bedrock: "AWS",
-  azure: "Azure",
   google: "Google",
   vertex: "Google",
   ollama: "Ollama",

packages/traceloop-sdk/test/ai-sdk-transformations.test.ts

@@ -927,7 +927,6 @@ describe("AI SDK Transformations", () => {
         "openai.completions",
         "openai.embeddings",
         "openai",
-        "azure-openai.chat",
       ];
 
       openaiProviders.forEach((provider) => {
@@ -942,6 +941,23 @@ describe("AI SDK Transformations", () => {
       });
     });
 
+    it("should transform azure openai provider to Azure system", () => {
+      const openaiProviders = [
+        "azure-openai",
+      ];
+
+      openaiProviders.forEach((provider) => {
+        const attributes = {
+          "ai.model.provider": provider,
+        };
+
+        transformAiSdkAttributes(attributes);
+
+        assert.strictEqual(attributes[SpanAttributes.LLM_SYSTEM], "Azure");
+        assert.strictEqual(attributes["ai.model.provider"], undefined);
+      });
+    });
+
     it("should transform other providers to their value", () => {
       const attributes = {
         "ai.model.provider": "anthropic",
@@ -1093,7 +1109,7 @@ describe("AI SDK Transformations", () => {
       assert.strictEqual(attributes[SpanAttributes.LLM_USAGE_TOTAL_TOKENS], 15);
 
       // Check vendor transformation
-      assert.strictEqual(attributes[SpanAttributes.LLM_SYSTEM], "OpenAI");
+      assert.strictEqual(attributes[SpanAttributes.LLM_SYSTEM], "Azure");
 
       // Check original AI SDK attributes are removed
       assert.strictEqual(attributes["ai.response.object"], undefined);