feat: Added AsUserMessage(this byte[] bytes, string mimeType) extensi…

…on to easy use vision.
tryAGI · Aug 31, 2024 · f0ac20f · f0ac20f
1 parent 11f4947
commit f0ac20f
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/src/libs/OpenAI/Extensions/StringExtensions.cs b/src/libs/OpenAI/Extensions/StringExtensions.cs
@@ -84,4 +84,31 @@ public static ChatCompletionRequestMessage AsRequestMessage(this ChatCompletionR
             ToolCalls = message.ToolCalls,
         };
     }
+
+    /// <summary>
+    /// 
+    /// </summary>
+    /// <param name="bytes"></param>
+    /// <param name="mimeType"></param>
+    /// <returns></returns>
+    /// <exception cref="ArgumentNullException"></exception>
+    public static ChatCompletionRequestMessage AsUserMessage(this byte[] bytes, string mimeType)
+    {
+        bytes = bytes ?? throw new ArgumentNullException(nameof(bytes));
+
+        return new ChatCompletionRequestUserMessage
+        {
+            Content = new ChatCompletionRequestUserMessageContentPart[] {
+                new ChatCompletionRequestMessageContentPartImage
+                {
+                    ImageUrl = new ChatCompletionRequestMessageContentPartImageImageUrl
+                    {
+                        Url = $"data:{mimeType};base64,{Convert.ToBase64String(bytes)}",
+                        Detail = ChatCompletionRequestMessageContentPartImageImageUrlDetail.Auto,
+                    },
+                    Type = ChatCompletionRequestMessageContentPartImageType.ImageUrl,
+                },
+            },
+        };
+    }
 }