com.openai.unity 5.2.2 (#121)

- Fixed Image Generation for Azure
RageAgainstThePixel · Nov 10, 2023 · 58cd5a7 · 58cd5a7
1 parent abdc30a
commit 58cd5a7
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 3 deletions.
diff --git a/Runtime/Chat/Content.cs b/Runtime/Chat/Content.cs
@@ -21,7 +21,7 @@ public Content(string text)
 
         [Preserve]
         public Content(Texture2D texture)
-            : this(ContentType.ImageUrl, $"data:image/jpeg;base64,{Convert.ToBase64String(texture.EncodeToPNG())}")
+            : this(ContentType.ImageUrl, $"data:image/png;base64,{Convert.ToBase64String(texture.EncodeToPNG())}")
         {
         }
 

diff --git a/Runtime/Images/ImagesEndpoint.cs b/Runtime/Images/ImagesEndpoint.cs
@@ -70,7 +70,8 @@ public async Task<IReadOnlyDictionary<string, Texture2D>> GenerateImageAsync(
         public async Task<IReadOnlyDictionary<string, Texture2D>> GenerateImageAsync(ImageGenerationRequest request, CancellationToken cancellationToken = default)
         {
             var payload = JsonConvert.SerializeObject(request, OpenAIClient.JsonSerializationOptions);
-            var response = await Rest.PostAsync(GetUrl("/generations"), payload, new RestParameters(client.DefaultRequestHeaders), cancellationToken);
+            var endpoint = GetUrl($"/generations{(client.Settings.Info.IsAzureDeployment ? ":submit" : string.Empty)}");
+            var response = await Rest.PostAsync(endpoint, payload, new RestParameters(client.DefaultRequestHeaders), cancellationToken);
             return await DeserializeResponseAsync(response, cancellationToken);
         }
 

diff --git a/Runtime/OpenAISettingsInfo.cs b/Runtime/OpenAISettingsInfo.cs
@@ -109,5 +109,7 @@ public OpenAISettingsInfo(string resourceName, string deploymentId, string apiVe
         public string BaseRequestUrlFormat { get; }
 
         public bool UseOAuthAuthentication { get; }
+
+        internal bool IsAzureDeployment => BaseRequestUrlFormat.Contains(AzureOpenAIDomain);
     }
 }
diff --git a/package.json b/package.json
@@ -3,7 +3,7 @@
   "displayName": "OpenAI",
   "description": "A OpenAI package for the Unity Game Engine to use GPT-4, GPT-3.5, GPT-3 and Dall-E though their RESTful API (currently in beta).\n\nIndependently developed, this is not an official library and I am not affiliated with OpenAI.\n\nAn OpenAI API account is required.",
   "keywords": [],
-  "version": "5.2.1",
+  "version": "5.2.2",
   "unity": "2021.3",
   "documentationUrl": "https://github.com/RageAgainstThePixel/com.openai.unity#documentation",
   "changelogUrl": "https://github.com/RageAgainstThePixel/com.openai.unity/releases",