AOAI Token not returning tokens

Author: darjoo

src/System Application/App/AI/src/Azure OpenAI/AOAIToken.Codeunit.al

@@ -21,7 +21,7 @@ codeunit 7759 "AOAI Token"
     /// <returns>The token count.</returns>
     procedure GetGPT35TokenCount(Input: SecretText): Integer
     begin
-        AzureOpenAIImpl.GetTokenCount(Input, Encodingcl100kbaseLbl);
+        exit(AzureOpenAIImpl.GetTokenCount(Input, Encodingcl100kbaseLbl));
     end;
 
     /// <summary>
@@ -31,7 +31,7 @@ codeunit 7759 "AOAI Token"
     /// <returns>The token count.</returns>
     procedure GetGPT4TokenCount(Input: SecretText): Integer
     begin
-        AzureOpenAIImpl.GetTokenCount(Input, Encodingcl100kbaseLbl);
+        exit(AzureOpenAIImpl.GetTokenCount(Input, Encodingcl100kbaseLbl));
     end;
 
     /// <summary>
@@ -41,7 +41,7 @@ codeunit 7759 "AOAI Token"
     /// <returns>The token count.</returns>
     procedure GetAdaTokenCount(Input: SecretText): Integer
     begin
-        AzureOpenAIImpl.GetTokenCount(Input, Encodingcl100kbaseLbl);
+        exit(AzureOpenAIImpl.GetTokenCount(Input, Encodingcl100kbaseLbl));
     end;
 
     /// <summary>
@@ -51,6 +51,6 @@ codeunit 7759 "AOAI Token"
     /// <returns>The token count.</returns>
     procedure GetDavinciTokenCount(Input: SecretText): Integer
     begin
-        AzureOpenAIImpl.GetTokenCount(Input, Encodingp50kbaseLbl);
+        exit(AzureOpenAIImpl.GetTokenCount(Input, Encodingp50kbaseLbl));
     end;
 }

src/System Application/App/AI/src/Azure OpenAI/AzureOpenAIImpl.Codeunit.al

@@ -531,12 +531,11 @@ codeunit 7772 "Azure OpenAI Impl"
     end;
 #endif
 
-    [NonDebuggable]
     procedure GetTokenCount(Input: SecretText; Encoding: Text) TokenCount: Integer
     var
         ALCopilotFunctions: DotNet ALCopilotFunctions;
     begin
-        TokenCount := ALCopilotFunctions.GptTokenCount(Input.Unwrap(), Encoding);
+        TokenCount := ALCopilotFunctions.GptTokenCount(Input, Encoding);
     end;
 
 }