diff --git a/api/src/main/java/org/apache/flink/agents/api/resource/ResourceName.java b/api/src/main/java/org/apache/flink/agents/api/resource/ResourceName.java
index 39da35aa..399c93b2 100644
--- a/api/src/main/java/org/apache/flink/agents/api/resource/ResourceName.java
+++ b/api/src/main/java/org/apache/flink/agents/api/resource/ResourceName.java
@@ -55,9 +55,9 @@ public static final class ChatModel {
 
         // Azure
         public static final String AZURE_CONNECTION =
-                "org.apache.flink.agents.integrations.chatmodels.anthropic.AzureAIChatModelConnection";
+                "org.apache.flink.agents.integrations.chatmodels.azureai.AzureAIChatModelConnection";
         public static final String AZURE_SETUP =
-                "org.apache.flink.agents.integrations.chatmodels.anthropic.AzureAIChatModelSetup";
+                "org.apache.flink.agents.integrations.chatmodels.azureai.AzureAIChatModelSetup";
 
         // Ollama
         public static final String OLLAMA_CONNECTION =
diff --git a/python/flink_agents/api/resource.py b/python/flink_agents/api/resource.py
index de5b2127..76ebf937 100644
--- a/python/flink_agents/api/resource.py
+++ b/python/flink_agents/api/resource.py
@@ -264,8 +264,8 @@ class Java:
             ANTHROPIC_SETUP = "org.apache.flink.agents.integrations.chatmodels.anthropic.AnthropicChatModelSetup"
 
             # Azure
-            AZURE_CONNECTION = "org.apache.flink.agents.integrations.chatmodels.anthropic.AzureAIChatModelConnection"
-            AZURE_SETUP = "org.apache.flink.agents.integrations.chatmodels.anthropic.AzureAIChatModelSetup"
+            AZURE_CONNECTION = "org.apache.flink.agents.integrations.chatmodels.azureai.AzureAIChatModelConnection"
+            AZURE_SETUP = "org.apache.flink.agents.integrations.chatmodels.azureai.AzureAIChatModelSetup"
 
             # Ollama
             OLLAMA_CONNECTION = "org.apache.flink.agents.integrations.chatmodels.ollama.OllamaChatModelConnection"