xwl2023/Yi.Framework
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 新增错误信息返回

Browse Source
This commit is contained in:
ccnetcore
2026-01-03 01:12:47 +08:00
parent 91533909c2
commit ea403fcae0
2 changed files with 17 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application.Contracts/Dtos/Chat/ImageTaskOutput.cs
View File

@@ -56,4 +56,9 @@ public class ImageTaskOutput
/// 创建时间 /// 创建时间
/// </summary> /// </summary>
public DateTime CreationTime { get; set; } public DateTime CreationTime { get; set; }
/// <summary>
/// 错误信息
/// </summary>
public string? ErrorInfo { get; set; }
} }

10
Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/Chat/AiImageService.cs
View File

@@ -36,6 +36,7 @@ public class AiImageService : ApplicationService
private readonly IWebHostEnvironment _webHostEnvironment; private readonly IWebHostEnvironment _webHostEnvironment;
private readonly TokenManager _tokenManager; private readonly TokenManager _tokenManager;
private readonly ISqlSugarRepository<AiModelEntity> _aiModelRepository; private readonly ISqlSugarRepository<AiModelEntity> _aiModelRepository;
public AiImageService( public AiImageService(
ISqlSugarRepository<ImageStoreTaskAggregateRoot> imageTaskRepository, ISqlSugarRepository<ImageStoreTaskAggregateRoot> imageTaskRepository,
IBackgroundJobManager backgroundJobManager, IBackgroundJobManager backgroundJobManager,
@@ -43,7 +44,8 @@ public class AiImageService : ApplicationService
PremiumPackageManager premiumPackageManager, PremiumPackageManager premiumPackageManager,
ModelManager modelManager, ModelManager modelManager,
IGuidGenerator guidGenerator, IGuidGenerator guidGenerator,
IWebHostEnvironment webHostEnvironment, TokenManager tokenManager, ISqlSugarRepository<AiModelEntity> aiModelRepository) IWebHostEnvironment webHostEnvironment, TokenManager tokenManager,
ISqlSugarRepository<AiModelEntity> aiModelRepository)
{ {
_imageTaskRepository = imageTaskRepository; _imageTaskRepository = imageTaskRepository;
_backgroundJobManager = backgroundJobManager; _backgroundJobManager = backgroundJobManager;
@@ -174,6 +176,7 @@ public class AiImageService : ApplicationService
{ {
mimeType = header.Split(':')[1].Split(';')[0]; mimeType = header.Split(':')[1].Split(';')[0];
} }
base64Content = parts[1]; base64Content = parts[1];
} }
} }
@@ -253,7 +256,10 @@ public class AiImageService : ApplicationService
// StoreBase64 = x.StoreBase64, // StoreBase64 = x.StoreBase64,
StoreUrl = x.StoreUrl, StoreUrl = x.StoreUrl,
TaskStatus = x.TaskStatus, TaskStatus = x.TaskStatus,
CreationTime = x.CreationTime PublishStatus = x.PublishStatus,
Categories = x.Categories,
CreationTime = x.CreationTime,
ErrorInfo = x.ErrorInfo
}) })
.ToListAsync(); .ToListAsync();