diff --git a/OllamaApiFacade/Extensions/KernelBuilderExtensions.cs b/OllamaApiFacade/Extensions/KernelBuilderExtensions.cs
index 112c424..27b3f0e 100644
--- a/OllamaApiFacade/Extensions/KernelBuilderExtensions.cs
+++ b/OllamaApiFacade/Extensions/KernelBuilderExtensions.cs
@@ -1,4 +1,5 @@
-using Microsoft.SemanticKernel;
+using System.ClientModel;
+using Microsoft.SemanticKernel;
 using OpenAI;
 
 namespace OllamaApiFacade.Extensions;
@@ -19,7 +20,8 @@ public static IKernelBuilder AddLmStudio(this IKernelBuilder builder, string mod
     {
         var uri = new Uri(endpoint);
         var openAiClientOptions = new OpenAIClientOptions { Endpoint = uri };
-        var openAiClient = new OpenAIClient("none", openAiClientOptions);
+        var apiKeyCredential = new ApiKeyCredential("none");
+        var openAiClient = new OpenAIClient(apiKeyCredential, openAiClientOptions);
 
         builder.AddOpenAIChatCompletion(model, openAiClient);
 
diff --git a/OllamaApiFacade/Extensions/SemanticKernelMapperExtensions.cs b/OllamaApiFacade/Extensions/SemanticKernelMapperExtensions.cs
index c58f40c..4853888 100644
--- a/OllamaApiFacade/Extensions/SemanticKernelMapperExtensions.cs
+++ b/OllamaApiFacade/Extensions/SemanticKernelMapperExtensions.cs
@@ -2,6 +2,7 @@
 using Microsoft.SemanticKernel.ChatCompletion;
 using OllamaApiFacade.DTOs;
 using OpenAI.Chat;
+using ChatMessageContent = Microsoft.SemanticKernel.ChatMessageContent;
 
 namespace OllamaApiFacade.Extensions;
 
@@ -65,7 +66,7 @@ public static ChatResponse ToChatResponse(this StreamingChatMessageContent messa
     }
 
     /// <summary>
-    /// Converts a <see cref="ChatMessageContent"/> object into a <see cref="OllamaApiFacade.DTOs.ChatResponse"/> object.
+    /// Converts a <see cref="Microsoft.SemanticKernel.ChatMessageContent"/> object into a <see cref="OllamaApiFacade.DTOs.ChatResponse"/> object.
     /// </summary>
     /// <param name="message">The chat message content to be converted.</param>
     /// <returns>A <see cref="OllamaApiFacade.DTOs.ChatResponse"/> object containing the converted message content.</returns>
@@ -117,9 +118,9 @@ public static CompletionResponse ToCompletionResponse(this ChatMessageContent ch
             ],
             Usage: new Usage
             {
-                PromptTokens = chatTokenUsage!.InputTokens,
-                CompletionTokens = chatTokenUsage.OutputTokens,
-                TotalTokens = chatTokenUsage.TotalTokens
+                PromptTokens = chatTokenUsage!.InputTokenCount,
+                CompletionTokens = chatTokenUsage.OutputTokenCount,
+                TotalTokens = chatTokenUsage.TotalTokenCount
             }
         );
     }