feat: 完成图片生成功能

Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Jobs/ImageGenerationJob.cs

@@ -30,32 +30,93 @@ public class ImageGenerationJob : AsyncBackgroundJob<ImageGenerationJobArgs>, IT
 
     public override async Task ExecuteAsync(ImageGenerationJobArgs args)
     {
-        _logger.LogInformation("开始执行图片生成任务，TaskId: {TaskId}, ModelId: {ModelId}, UserId: {UserId}",
-            args.TaskId, args.ModelId, args.UserId);
+        var task = await _imageStoreTaskRepository.GetFirstAsync(x => x.Id == args.TaskId);
+        if (task is null)
+        {
+            throw new UserFriendlyException($"{args.TaskId} 图片生成任务不存在");
+        }
 
+        _logger.LogInformation("开始执行图片生成任务，TaskId: {TaskId}, ModelId: {ModelId}, UserId: {UserId}",
+            task.Id, task.ModelId, task.UserId);
         try
         {
-            var request = JsonSerializer.Deserialize<JsonElement>(args.RequestJson);
+            // 构建 Gemini API 请求对象
+            var parts = new List<object>
+            {
+                new { text = task.Prompt }
+            };
 
+            // 添加参考图（如果有）
+            foreach (var prefixBase64 in task.ReferenceImagesPrefixBase64)
+            {
+                var (mimeType, base64Data) = ParsePrefixBase64(prefixBase64);
+                parts.Add(new
+                {
+                    inline_data = new
+                    {
+                        mime_type = mimeType,
+                        data = base64Data
+                    }
+                });
+            }
+
+            var requestObj = new
+            {
+                contents = new[]
+                {
+                    new { parts }
+                }
+            };
+
+            var request = JsonSerializer.Deserialize<JsonElement>(
+                JsonSerializer.Serialize(requestObj));
+
+            //里面生成成功已经包含扣款了
             await _aiGateWayManager.GeminiGenerateContentImageForStatisticsAsync(
-                args.TaskId,
-                args.ModelId,
+                task.Id,
+                task.ModelId,
                 request,
-                args.UserId);
+                task.UserId,
+                tokenId:task.TokenId);
 
+            
             _logger.LogInformation("图片生成任务完成，TaskId: {TaskId}", args.TaskId);
         }
         catch (Exception ex)
         {
             _logger.LogError(ex, "图片生成任务失败，TaskId: {TaskId}, Error: {Error}", args.TaskId, ex.Message);
 
-            // 更新任务状态为失败
-            var task = await _imageStoreTaskRepository.GetFirstAsync(x => x.Id == args.TaskId);
-            if (task != null)
-            {
-                task.TaskStatus = TaskStatusEnum.Fail;
-                await _imageStoreTaskRepository.UpdateAsync(task);
-            }
+            task.TaskStatus = TaskStatusEnum.Fail;
+            task.ErrorInfo = ex.Message;
+
+            await _imageStoreTaskRepository.UpdateAsync(task);
         }
     }
-}
+
+    /// <summary>
+    /// 解析带前缀的 Base64 字符串，提取 mimeType 和纯 base64 数据
+    /// </summary>
+    private static (string mimeType, string base64Data) ParsePrefixBase64(string prefixBase64)
+    {
+        // 默认值
+        var mimeType = "image/png";
+        var base64Data = prefixBase64;
+
+        if (prefixBase64.Contains(","))
+        {
+            var parts = prefixBase64.Split(',');
+            if (parts.Length == 2)
+            {
+                var header = parts[0];
+                if (header.Contains(":") && header.Contains(";"))
+                {
+                    mimeType = header.Split(':')[1].Split(';')[0];
+                }
+
+                base64Data = parts[1];
+            }
+        }
+
+        return (mimeType, base64Data);
+    }
+}