Fix bug where API Key authorization doesn't work in WebGL

uezo · uezo · commit 887ee79d6ed3 · 2025-04-28T20:55:20.000+09:00
diff --git a/Plugins/AIAvatarKitServiceWebGL.jslib b/Plugins/AIAvatarKitServiceWebGL.jslib
@@ -1,19 +1,22 @@
 mergeInto(LibraryManager.library, {
-    StartAIAvatarKitMessageStreamJS: function(targetObjectNamePtr, sessionIdPtr, urlPtr, aakStreamRequestPtr) {
+    StartAIAvatarKitMessageStreamJS: function(targetObjectNamePtr, sessionIdPtr, urlPtr, aakStreamRequestPtr, aakHeadersPtr) {
         let targetObjectName = UTF8ToString(targetObjectNamePtr);
         let sessionId = UTF8ToString(sessionIdPtr);
         let url = UTF8ToString(urlPtr);
         let aakStreamRequest = UTF8ToString(aakStreamRequestPtr);
+        let aakHeaders = JSON.parse(UTF8ToString(aakHeadersPtr));
         let decoder = new TextDecoder("utf-8");
 
         if (document.aakAbortController == null) {
             document.aakAbortController = new AbortController();
         }
 
+        let headers = { "Content-Type": "application/json" };
+        for (let key in aakHeaders) {
+            headers[key] = aakHeaders[key];
+        }
         fetch(url, {
-            headers: {
-                "Content-Type": "application/json"
-            },
+            headers: headers,
             method: "POST",
             body: aakStreamRequest,
             signal: document.aakAbortController.signal
diff --git a/Scripts/LLM/AIAvatarKit/AIAvatarKitServiceWebGL.cs b/Scripts/LLM/AIAvatarKit/AIAvatarKitServiceWebGL.cs
@@ -24,7 +24,7 @@ public override bool IsEnabled
 
 #if UNITY_WEBGL
         [DllImport("__Internal")]
-        protected static extern void StartAIAvatarKitMessageStreamJS(string targetObjectName, string sessionId, string url, string chatCompletionRequest);
+        protected static extern void StartAIAvatarKitMessageStreamJS(string targetObjectName, string sessionId, string url, string chatCompletionRequest, string aakHeaders);
         [DllImport("__Internal")]
         protected static extern void AbortAIAvatarKitMessageStreamJS();
 
@@ -83,6 +83,12 @@ public override async UniTask StartStreamingAsync(AIAvatarKitSession aakSession,
                 Debug.LogWarning("Custom headers for AIAvatarKit on WebGL is not supported for now.");
             }
 
+            var authHeader = new Dictionary<string, string>();
+            if (!string.IsNullOrEmpty(ApiKey))
+            {
+                authHeader["Authorization"] = $"Bearer {ApiKey}";
+            }
+
             var serializedData = JsonConvert.SerializeObject(data);
 
             if (DebugMode)
@@ -96,7 +102,8 @@ public override async UniTask StartStreamingAsync(AIAvatarKitSession aakSession,
                 gameObject.name,
                 sessionId,
                 BaseUrl + "/chat",
-                serializedData
+                serializedData,
+                JsonConvert.SerializeObject(authHeader)
             );
 
             // Preprocessing response
