mirror of
https://gitee.com/ccnetcore/Yi
synced 2026-04-01 14:46:37 +08:00
Compare commits
135 Commits
invitation
...
new-md
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
6f1efafd86 | ||
|
|
2714a507d9 | ||
|
|
9a9230786b | ||
|
|
4a8b58a65c | ||
|
|
7d81f88658 | ||
|
|
0ce3c0bbdd | ||
|
|
981235e6e9 | ||
|
|
d0ecb232a1 | ||
|
|
c7a52604e7 | ||
|
|
da81b2d8a3 | ||
|
|
7b14fdd8de | ||
|
|
1fc2734eb7 | ||
|
|
f3bef72ebb | ||
|
|
7e6d2e829b | ||
|
|
944626960b | ||
|
|
c073868989 | ||
|
|
d2981100fa | ||
|
|
ce4f7e5711 | ||
|
|
cc812ba2cb | ||
|
|
8a6e5abf48 | ||
|
|
8b191330b8 | ||
|
|
5ed79c6dd0 | ||
|
|
6e2ca8f1c3 | ||
|
|
a46a552097 | ||
|
|
53e56134d4 | ||
|
|
0d2f2cb826 | ||
|
|
f90105ebb4 | ||
|
|
67ed1ac1e3 | ||
|
|
69b84f6613 | ||
|
|
433d616b9b | ||
|
|
53aa575ad4 | ||
|
|
571df74c43 | ||
|
|
b7847c7e7d | ||
|
|
94eb41996e | ||
|
|
cefde6848d | ||
|
|
381b712b25 | ||
|
|
c319b0b4e4 | ||
|
|
1a32fa9e20 | ||
|
|
909406238c | ||
|
|
54a1d2a66f | ||
|
|
8dcbfcad33 | ||
|
|
f64fd43951 | ||
|
|
551597765c | ||
|
|
bfda33280a | ||
|
|
8d0411f1f4 | ||
|
|
3995d4acab | ||
|
|
6ff5727156 | ||
|
|
f654386dfe | ||
|
|
c03ef82643 | ||
|
|
525545329b | ||
|
|
755cb6f509 | ||
|
|
55469708f0 | ||
|
|
94c52c62fe | ||
|
|
37b4709d76 | ||
|
|
86555af6ce | ||
|
|
ddb00879f4 | ||
|
|
2d0ca08314 | ||
|
|
b78ecf27d5 | ||
|
|
02a5f69958 | ||
|
|
cf5bf746ef | ||
|
|
0a5e40ee25 | ||
|
|
51a266ef58 | ||
|
|
1f0901c90c | ||
|
|
a725c06396 | ||
|
|
54547f0d7c | ||
|
|
afe9c8bcae | ||
|
|
688d93e5c1 | ||
|
|
4c65b2398d | ||
|
|
41435f1aa3 | ||
|
|
20206bbc44 | ||
|
|
f2dc0d1825 | ||
|
|
51b4d1b072 | ||
|
|
9180799e4e | ||
|
|
9788b9182b | ||
|
|
260b9a4795 | ||
|
|
9380e3daa8 | ||
|
|
bb894e14a4 | ||
|
|
b492d82442 | ||
|
|
5eaffe2ec2 | ||
|
|
d7bcad9da7 | ||
|
|
04e11d15e2 | ||
|
|
97e3dc5eed | ||
|
|
695bd56a27 | ||
|
|
7919383be3 | ||
|
|
d6cc3c5d96 | ||
|
|
19f0d05a69 | ||
|
|
3c5e575e9b | ||
|
|
f875617de1 | ||
|
|
9976e8a6e2 | ||
|
|
38e8cbc5ca | ||
|
|
d95c14c903 | ||
|
|
ffb2f2fb4c | ||
|
|
4bd2fc357d | ||
|
|
da23d17af8 | ||
|
|
c1a6046107 | ||
|
|
2ec7b5f4fd | ||
|
|
d21f61646a | ||
|
|
eecdf442fb | ||
|
|
8e8338743d | ||
|
|
1b4c3cbb8d | ||
|
|
b7756e2112 | ||
|
|
cb49059e84 | ||
|
|
690cabfd96 | ||
|
|
4521212a90 | ||
|
|
771ecd9d81 | ||
|
|
94834f45c3 | ||
|
|
a9b2979a21 | ||
|
|
22ac150acd | ||
|
|
17337b8d78 | ||
|
|
09fb43ee14 | ||
|
|
477c0e3f2c | ||
|
|
5beef22269 | ||
|
|
933cbb91d8 | ||
|
|
cf137f6307 | ||
|
|
3e75792e43 | ||
|
|
108ba348f6 | ||
|
|
bcdcec40e0 | ||
|
|
2ce8baea42 | ||
|
|
609de29e71 | ||
|
|
2efed4f4a5 | ||
|
|
1cc5f2a14f | ||
|
|
d9997eeb28 | ||
|
|
06e77aa8fd | ||
|
|
e9e2228f6e | ||
|
|
d516a381d0 | ||
|
|
4e792ba976 | ||
|
|
f90d3871fa | ||
|
|
6005b9329d | ||
|
|
9d4b3e7d0c | ||
|
|
72795382a1 | ||
|
|
35cdff2afa | ||
|
|
dcf547f513 | ||
|
|
8660d45f36 | ||
|
|
63dd55e7a4 | ||
|
|
40cd89f90c |
3
.gitignore
vendored
3
.gitignore
vendored
@@ -265,6 +265,7 @@ src/Acme.BookStore.Blazor.Server.Tiered/Logs/*
|
|||||||
**/wwwroot/libs/*
|
**/wwwroot/libs/*
|
||||||
public
|
public
|
||||||
dist
|
dist
|
||||||
|
dist - 副本
|
||||||
.vscode
|
.vscode
|
||||||
/Yi.Abp.Net8/src/Yi.Abp.Web/appsettings.Development.json
|
/Yi.Abp.Net8/src/Yi.Abp.Web/appsettings.Development.json
|
||||||
/Yi.Abp.Net8/src/Yi.Abp.Web/appsettings.Production.json
|
/Yi.Abp.Net8/src/Yi.Abp.Web/appsettings.Production.json
|
||||||
@@ -277,3 +278,5 @@ database_backup
|
|||||||
/Yi.Abp.Net8/src/Yi.Abp.Web/yi-abp-dev.db
|
/Yi.Abp.Net8/src/Yi.Abp.Web/yi-abp-dev.db
|
||||||
|
|
||||||
package-lock.json
|
package-lock.json
|
||||||
|
|
||||||
|
.claude
|
||||||
|
|||||||
10
README.md
10
README.md
@@ -7,6 +7,12 @@
|
|||||||
[](https://gitee.com/ccnetcore/Yi)
|
[](https://gitee.com/ccnetcore/Yi)
|
||||||
[](https://gitee.com/ccnetcore/Yi)
|
[](https://gitee.com/ccnetcore/Yi)
|
||||||
|
|
||||||
|
本项目 CDN 加速及安全防护由 Tencent EdgeOne 赞助
|
||||||
|
|
||||||
|
[亚洲最佳CDN、边缘和安全解决方案 - Tencent EdgeOne](https://edgeone.ai/zh?from=github)
|
||||||
|
|
||||||
|
<img src="readme/edgeone.png"/>
|
||||||
|
|
||||||
[English](README-en.md) | 简体中文
|
[English](README-en.md) | 简体中文
|
||||||
****
|
****
|
||||||
## 🍍 简介:
|
## 🍍 简介:
|
||||||
@@ -60,9 +66,9 @@ bbs前端:`docker run -d --name yi.bbs -p 18001:18001 -v /home/Yi/Yi.Bbs.Vue3/
|
|||||||
|
|
||||||
Yi社区官网网址(Bbs社区正式):[ccnetcore.com](https://ccnetcore.com) (已上线,欢迎加入)
|
Yi社区官网网址(Bbs社区正式):[ccnetcore.com](https://ccnetcore.com) (已上线,欢迎加入)
|
||||||
|
|
||||||
Rbac后台演示地址:https://ccnetcore.com:1000 (用户cc、密码123456)
|
Rbac后台演示地址:https://data.ccnetcore.com:1000 (用户cc、密码123456)
|
||||||
|
|
||||||
Pure后台演示地址:https://ccnetcore.com:1001 (用户cc、密码123456)
|
Pure后台演示地址:https://data.ccnetcore.com:1001 (用户cc、密码123456)
|
||||||
|
|
||||||
## 🍏 支持:
|
## 🍏 支持:
|
||||||
|
|
||||||
|
|||||||
@@ -12,6 +12,7 @@
|
|||||||
</PropertyGroup>
|
</PropertyGroup>
|
||||||
|
|
||||||
<PropertyGroup>
|
<PropertyGroup>
|
||||||
|
<SatelliteResourceLanguages>en;zh-CN</SatelliteResourceLanguages>
|
||||||
<LangVersion>latest</LangVersion>
|
<LangVersion>latest</LangVersion>
|
||||||
<Version>1.0.0</Version>
|
<Version>1.0.0</Version>
|
||||||
<NoWarn>$(NoWarn);CS1591;CS8618;CS1998;CS8604;CS8620;CS8600;CS8602</NoWarn>
|
<NoWarn>$(NoWarn);CS1591;CS8618;CS1998;CS8604;CS8620;CS8600;CS8602</NoWarn>
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
// MIT 许可证
|
// MIT 许可证
|
||||||
//
|
//
|
||||||
// 版权 © 2020-present 百小僧, 百签科技(广东)有限公司 和所有贡献者
|
// 版权 © 2020-present 百小僧, 百签科技(广东)有限公司 和所有贡献者
|
||||||
//
|
//
|
||||||
@@ -17,25 +17,25 @@ using Microsoft.AspNetCore.Mvc;
|
|||||||
using Microsoft.AspNetCore.Mvc.Filters;
|
using Microsoft.AspNetCore.Mvc.Filters;
|
||||||
using Microsoft.Extensions.DependencyInjection;
|
using Microsoft.Extensions.DependencyInjection;
|
||||||
using Microsoft.Extensions.Logging;
|
using Microsoft.Extensions.Logging;
|
||||||
|
using Newtonsoft.Json;
|
||||||
using Volo.Abp.AspNetCore.Mvc;
|
using Volo.Abp.AspNetCore.Mvc;
|
||||||
using Volo.Abp.DependencyInjection;
|
using Volo.Abp.Validation;
|
||||||
using Yi.Framework.Core.Extensions;
|
using Yi.Framework.Core.Extensions;
|
||||||
|
|
||||||
namespace Yi.Framework.AspNetCore.UnifyResult.Fiters;
|
namespace Yi.Framework.AspNetCore.UnifyResult.Fiters;
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 友好异常拦截器
|
/// 友好异常拦截器
|
||||||
/// </summary>
|
/// </summary>
|
||||||
public sealed class FriendlyExceptionFilter : IAsyncExceptionFilter
|
public sealed class FriendlyExceptionFilter : IAsyncExceptionFilter
|
||||||
{
|
{
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 异常拦截
|
/// 异常拦截
|
||||||
/// </summary>
|
/// </summary>
|
||||||
/// <param name="context"></param>
|
/// <param name="context"></param>
|
||||||
/// <returns></returns>
|
/// <returns></returns>
|
||||||
public async Task OnExceptionAsync(ExceptionContext context)
|
public async Task OnExceptionAsync(ExceptionContext context)
|
||||||
{
|
{
|
||||||
|
|
||||||
// 排除 WebSocket 请求处理
|
// 排除 WebSocket 请求处理
|
||||||
if (context.HttpContext.IsWebSocketRequest()) return;
|
if (context.HttpContext.IsWebSocketRequest()) return;
|
||||||
|
|
||||||
@@ -44,20 +44,23 @@ public sealed class FriendlyExceptionFilter : IAsyncExceptionFilter
|
|||||||
|
|
||||||
// 解析异常信息
|
// 解析异常信息
|
||||||
var exceptionMetadata = GetExceptionMetadata(context);
|
var exceptionMetadata = GetExceptionMetadata(context);
|
||||||
|
var unifyResult = context.GetRequiredService<IUnifyResultProvider>();
|
||||||
IUnifyResultProvider unifyResult = context.GetRequiredService<IUnifyResultProvider>();
|
|
||||||
// 执行规范化异常处理
|
// 执行规范化异常处理
|
||||||
context.Result = unifyResult.OnException(context, exceptionMetadata);
|
context.Result = unifyResult.OnException(context, exceptionMetadata);
|
||||||
|
|
||||||
// 创建日志记录器
|
// 创建日志记录器
|
||||||
var logger = context.HttpContext.RequestServices.GetRequiredService<ILogger<FriendlyExceptionFilter>>();
|
var logger = context.HttpContext.RequestServices.GetRequiredService<ILogger<FriendlyExceptionFilter>>();
|
||||||
|
|
||||||
|
var errorMsg = "";
|
||||||
|
if (exceptionMetadata.Errors != null) errorMsg = "\n" + JsonConvert.SerializeObject(exceptionMetadata.Errors);
|
||||||
|
|
||||||
|
|
||||||
// 记录拦截日常
|
// 记录拦截日常
|
||||||
logger.LogError(context.Exception, context.Exception.Message);
|
logger.LogError(context.Exception, context.Exception.Message + errorMsg);
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 获取异常元数据
|
/// 获取异常元数据
|
||||||
/// </summary>
|
/// </summary>
|
||||||
/// <param name="context"></param>
|
/// <param name="context"></param>
|
||||||
/// <returns></returns>
|
/// <returns></returns>
|
||||||
@@ -74,22 +77,25 @@ public sealed class FriendlyExceptionFilter : IAsyncExceptionFilter
|
|||||||
// 判断是否是 ExceptionContext 或者 ActionExecutedContext
|
// 判断是否是 ExceptionContext 或者 ActionExecutedContext
|
||||||
var exception = context is ExceptionContext exContext
|
var exception = context is ExceptionContext exContext
|
||||||
? exContext.Exception
|
? exContext.Exception
|
||||||
: (
|
: context is ActionExecutedContext edContext
|
||||||
context is ActionExecutedContext edContext
|
? edContext.Exception
|
||||||
? edContext.Exception
|
: default;
|
||||||
: default
|
|
||||||
);
|
if (exception is AbpValidationException validationException)
|
||||||
|
{
|
||||||
|
errors = validationException.ValidationErrors;
|
||||||
|
isValidationException = true;
|
||||||
|
}
|
||||||
|
|
||||||
// 判断是否是友好异常
|
// 判断是否是友好异常
|
||||||
if (exception is UserFriendlyException friendlyException)
|
if (exception is UserFriendlyException friendlyException)
|
||||||
{
|
{
|
||||||
int statusCode2 = 500;
|
var statusCode2 = 500;
|
||||||
int.TryParse(friendlyException.Code, out statusCode2);
|
int.TryParse(friendlyException.Code, out statusCode2);
|
||||||
isFriendlyException = true;
|
isFriendlyException = true;
|
||||||
errorCode = friendlyException.Code;
|
errorCode = friendlyException.Code;
|
||||||
originErrorCode = friendlyException.Code;
|
originErrorCode = friendlyException.Code;
|
||||||
statusCode = statusCode2==0?403:statusCode2;
|
statusCode = statusCode2 == 0 ? 403 : statusCode2;
|
||||||
isValidationException = false;
|
|
||||||
errors = friendlyException.Message;
|
errors = friendlyException.Message;
|
||||||
data = friendlyException.Data;
|
data = friendlyException.Data;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,6 +2,7 @@
|
|||||||
using Microsoft.Extensions.DependencyInjection;
|
using Microsoft.Extensions.DependencyInjection;
|
||||||
using Swashbuckle.AspNetCore.SwaggerGen;
|
using Swashbuckle.AspNetCore.SwaggerGen;
|
||||||
using Volo.Abp.AspNetCore.Mvc.ExceptionHandling;
|
using Volo.Abp.AspNetCore.Mvc.ExceptionHandling;
|
||||||
|
using Volo.Abp.AspNetCore.Mvc.Response;
|
||||||
using Yi.Framework.AspNetCore.UnifyResult.Fiters;
|
using Yi.Framework.AspNetCore.UnifyResult.Fiters;
|
||||||
|
|
||||||
namespace Yi.Framework.AspNetCore.UnifyResult;
|
namespace Yi.Framework.AspNetCore.UnifyResult;
|
||||||
@@ -20,9 +21,10 @@ public static class UnifyResultExtensions
|
|||||||
services.AddTransient<FriendlyExceptionFilter>();
|
services.AddTransient<FriendlyExceptionFilter>();
|
||||||
services.AddMvc(options =>
|
services.AddMvc(options =>
|
||||||
{
|
{
|
||||||
|
options.Filters.RemoveAll(x => (x as ServiceFilterAttribute)?.ServiceType == typeof(AbpExceptionFilter));
|
||||||
|
options.Filters.RemoveAll(x => (x as ServiceFilterAttribute)?.ServiceType == typeof(AbpNoContentActionFilter));
|
||||||
options.Filters.AddService<SucceededUnifyResultFilter>(99);
|
options.Filters.AddService<SucceededUnifyResultFilter>(99);
|
||||||
options.Filters.AddService<FriendlyExceptionFilter>(100);
|
options.Filters.AddService<FriendlyExceptionFilter>(100);
|
||||||
options.Filters.RemoveAll(x => (x as ServiceFilterAttribute)?.ServiceType == typeof(AbpExceptionFilter));
|
|
||||||
});
|
});
|
||||||
return services;
|
return services;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -199,11 +199,11 @@ namespace Yi.Framework.Ddd.Application
|
|||||||
/// <summary>
|
/// <summary>
|
||||||
/// 批量删除实体
|
/// 批量删除实体
|
||||||
/// </summary>
|
/// </summary>
|
||||||
/// <param name="ids">实体ID集合</param>
|
/// <param name="id">实体ID集合</param>
|
||||||
[RemoteService(isEnabled: true)]
|
[RemoteService(isEnabled: true)]
|
||||||
public virtual async Task DeleteAsync(IEnumerable<TKey> ids)
|
public virtual async Task DeleteAsync(IEnumerable<TKey> id)
|
||||||
{
|
{
|
||||||
await Repository.DeleteManyAsync(ids);
|
await Repository.DeleteManyAsync(id);
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
|
|||||||
@@ -1,7 +1,8 @@
|
|||||||
using Microsoft.Extensions.DependencyInjection;
|
using Microsoft.Extensions.DependencyInjection;
|
||||||
using Volo.Abp.Modularity;
|
using Volo.Abp.Modularity;
|
||||||
using Volo.Abp.ObjectMapping;
|
using Volo.Abp.ObjectMapping;
|
||||||
using Yi.Framework.Core;
|
using Yi.Framework.Core;
|
||||||
|
using Mapster;
|
||||||
|
|
||||||
namespace Yi.Framework.Mapster
|
namespace Yi.Framework.Mapster
|
||||||
{
|
{
|
||||||
@@ -22,7 +23,8 @@ namespace Yi.Framework.Mapster
|
|||||||
public override void ConfigureServices(ServiceConfigurationContext context)
|
public override void ConfigureServices(ServiceConfigurationContext context)
|
||||||
{
|
{
|
||||||
var services = context.Services;
|
var services = context.Services;
|
||||||
|
// 扫描并注册所有映射配置
|
||||||
|
TypeAdapterConfig.GlobalSettings.Scan(AppDomain.CurrentDomain.GetAssemblies());
|
||||||
// 注册Mapster相关服务
|
// 注册Mapster相关服务
|
||||||
services.AddTransient<IAutoObjectMappingProvider, MapsterAutoObjectMappingProvider>();
|
services.AddTransient<IAutoObjectMappingProvider, MapsterAutoObjectMappingProvider>();
|
||||||
services.AddTransient<IObjectMapper, MapsterObjectMapper>();
|
services.AddTransient<IObjectMapper, MapsterObjectMapper>();
|
||||||
|
|||||||
@@ -60,8 +60,8 @@ namespace Yi.Framework.SqlSugarCore.Abstractions
|
|||||||
public bool EnabledSaasMultiTenancy { get; set; } = false;
|
public bool EnabledSaasMultiTenancy { get; set; } = false;
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 并发乐观锁异常,否则不处理
|
/// 是否开启更新并发乐观锁
|
||||||
/// </summary>
|
/// </summary>
|
||||||
public bool EnabledConcurrencyException { get; set; } = true;
|
public bool EnabledConcurrencyException { get;set; } = false;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -1,12 +1,7 @@
|
|||||||
using System.Linq;
|
|
||||||
using System.Linq.Expressions;
|
using System.Linq.Expressions;
|
||||||
using System.Text;
|
|
||||||
using Microsoft.Extensions.Logging;
|
|
||||||
using Microsoft.Extensions.Options;
|
using Microsoft.Extensions.Options;
|
||||||
using Nito.AsyncEx;
|
using Nito.AsyncEx;
|
||||||
using SqlSugar;
|
using SqlSugar;
|
||||||
using Volo.Abp;
|
|
||||||
using Volo.Abp.Auditing;
|
|
||||||
using Volo.Abp.Data;
|
using Volo.Abp.Data;
|
||||||
using Volo.Abp.DependencyInjection;
|
using Volo.Abp.DependencyInjection;
|
||||||
using Volo.Abp.Domain.Entities;
|
using Volo.Abp.Domain.Entities;
|
||||||
@@ -17,18 +12,19 @@ using Yi.Framework.SqlSugarCore.Abstractions;
|
|||||||
|
|
||||||
namespace Yi.Framework.SqlSugarCore.Repositories
|
namespace Yi.Framework.SqlSugarCore.Repositories
|
||||||
{
|
{
|
||||||
public class SqlSugarRepository<TEntity> : ISqlSugarRepository<TEntity>, IRepository<TEntity>
|
public class SqlSugarRepository<TEntity> : ISqlSugarRepository<TEntity>, IRepository<TEntity> where TEntity : class, IEntity, new()
|
||||||
where TEntity : class, IEntity, new()
|
|
||||||
{
|
{
|
||||||
|
[Obsolete("使用GetDbContextAsync()")]
|
||||||
public ISqlSugarClient _Db => AsyncContext.Run(async () => await GetDbContextAsync());
|
public ISqlSugarClient _Db => AsyncContext.Run(async () => await GetDbContextAsync());
|
||||||
|
|
||||||
|
[Obsolete("使用AsQueryable()")]
|
||||||
public ISugarQueryable<TEntity> _DbQueryable => _Db.Queryable<TEntity>();
|
public ISugarQueryable<TEntity> _DbQueryable => _Db.Queryable<TEntity>();
|
||||||
|
|
||||||
private readonly ISugarDbContextProvider<ISqlSugarDbContext> _dbContextProvider;
|
private readonly ISugarDbContextProvider<ISqlSugarDbContext> _dbContextProvider;
|
||||||
|
|
||||||
public IAbpLazyServiceProvider LazyServiceProvider { get; set; }
|
public IAbpLazyServiceProvider LazyServiceProvider { get; set; }
|
||||||
protected DbConnOptions? Options => LazyServiceProvider?.LazyGetService<IOptions<DbConnOptions>>().Value;
|
|
||||||
|
|
||||||
|
protected DbConnOptions? Options => LazyServiceProvider?.LazyGetService<IOptions<DbConnOptions>>().Value;
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 异步查询执行器
|
/// 异步查询执行器
|
||||||
/// </summary>
|
/// </summary>
|
||||||
@@ -64,26 +60,22 @@ namespace Yi.Framework.SqlSugarCore.Repositories
|
|||||||
|
|
||||||
#region Abp模块
|
#region Abp模块
|
||||||
|
|
||||||
public virtual async Task<TEntity?> FindAsync(Expression<Func<TEntity, bool>> predicate,
|
public virtual async Task<TEntity?> FindAsync(Expression<Func<TEntity, bool>> predicate, bool includeDetails = true, CancellationToken cancellationToken = default)
|
||||||
bool includeDetails = true, CancellationToken cancellationToken = default)
|
|
||||||
{
|
{
|
||||||
return await GetFirstAsync(predicate);
|
return await GetFirstAsync(predicate);
|
||||||
}
|
}
|
||||||
|
|
||||||
public virtual async Task<TEntity> GetAsync(Expression<Func<TEntity, bool>> predicate,
|
public virtual async Task<TEntity> GetAsync(Expression<Func<TEntity, bool>> predicate, bool includeDetails = true, CancellationToken cancellationToken = default)
|
||||||
bool includeDetails = true, CancellationToken cancellationToken = default)
|
|
||||||
{
|
{
|
||||||
return await GetFirstAsync(predicate);
|
return await GetFirstAsync(predicate);
|
||||||
}
|
}
|
||||||
|
|
||||||
public virtual async Task DeleteAsync(Expression<Func<TEntity, bool>> predicate, bool autoSave = false,
|
public virtual async Task DeleteAsync(Expression<Func<TEntity, bool>> predicate, bool autoSave = false, CancellationToken cancellationToken = default)
|
||||||
CancellationToken cancellationToken = default)
|
|
||||||
{
|
{
|
||||||
await this.DeleteAsync(predicate);
|
await this.DeleteAsync(predicate);
|
||||||
}
|
}
|
||||||
|
|
||||||
public virtual async Task DeleteDirectAsync(Expression<Func<TEntity, bool>> predicate,
|
public virtual async Task DeleteDirectAsync(Expression<Func<TEntity, bool>> predicate, CancellationToken cancellationToken = default)
|
||||||
CancellationToken cancellationToken = default)
|
|
||||||
{
|
{
|
||||||
await this.DeleteAsync(predicate);
|
await this.DeleteAsync(predicate);
|
||||||
}
|
}
|
||||||
@@ -113,71 +105,60 @@ namespace Yi.Framework.SqlSugarCore.Repositories
|
|||||||
throw new NotImplementedException();
|
throw new NotImplementedException();
|
||||||
}
|
}
|
||||||
|
|
||||||
public virtual async Task<List<TEntity>> GetListAsync(Expression<Func<TEntity, bool>> predicate,
|
public virtual async Task<List<TEntity>> GetListAsync(Expression<Func<TEntity, bool>> predicate, bool includeDetails = false, CancellationToken cancellationToken = default)
|
||||||
bool includeDetails = false, CancellationToken cancellationToken = default)
|
|
||||||
{
|
{
|
||||||
return await GetListAsync(predicate);
|
return await GetListAsync(predicate);
|
||||||
}
|
}
|
||||||
|
|
||||||
public virtual async Task<TEntity> InsertAsync(TEntity entity, bool autoSave = false,
|
public virtual async Task<TEntity> InsertAsync(TEntity entity, bool autoSave = false, CancellationToken cancellationToken = default)
|
||||||
CancellationToken cancellationToken = default)
|
|
||||||
{
|
{
|
||||||
return await InsertReturnEntityAsync(entity);
|
return await InsertReturnEntityAsync(entity);
|
||||||
}
|
}
|
||||||
|
|
||||||
public virtual async Task InsertManyAsync(IEnumerable<TEntity> entities, bool autoSave = false,
|
public virtual async Task InsertManyAsync(IEnumerable<TEntity> entities, bool autoSave = false, CancellationToken cancellationToken = default)
|
||||||
CancellationToken cancellationToken = default)
|
|
||||||
{
|
{
|
||||||
await InsertRangeAsync(entities.ToList());
|
await InsertRangeAsync(entities.ToList());
|
||||||
}
|
}
|
||||||
|
|
||||||
public virtual async Task<TEntity> UpdateAsync(TEntity entity, bool autoSave = false,
|
public virtual async Task<TEntity> UpdateAsync(TEntity entity, bool autoSave = false, CancellationToken cancellationToken = default)
|
||||||
CancellationToken cancellationToken = default)
|
|
||||||
{
|
{
|
||||||
await UpdateAsync(entity);
|
await UpdateAsync(entity);
|
||||||
return entity;
|
return entity;
|
||||||
}
|
}
|
||||||
|
|
||||||
public virtual async Task UpdateManyAsync(IEnumerable<TEntity> entities, bool autoSave = false,
|
public virtual async Task UpdateManyAsync(IEnumerable<TEntity> entities, bool autoSave = false, CancellationToken cancellationToken = default)
|
||||||
CancellationToken cancellationToken = default)
|
|
||||||
{
|
{
|
||||||
await UpdateRangeAsync(entities.ToList());
|
await UpdateRangeAsync(entities.ToList());
|
||||||
}
|
}
|
||||||
|
|
||||||
public virtual async Task DeleteAsync(TEntity entity, bool autoSave = false,
|
public virtual async Task DeleteAsync(TEntity entity, bool autoSave = false, CancellationToken cancellationToken = default)
|
||||||
CancellationToken cancellationToken = default)
|
|
||||||
{
|
{
|
||||||
await DeleteAsync(entity);
|
await DeleteAsync(entity);
|
||||||
}
|
}
|
||||||
|
|
||||||
public virtual async Task DeleteManyAsync(IEnumerable<TEntity> entities, bool autoSave = false,
|
public virtual async Task DeleteManyAsync(IEnumerable<TEntity> entities, bool autoSave = false, CancellationToken cancellationToken = default)
|
||||||
CancellationToken cancellationToken = default)
|
|
||||||
{
|
{
|
||||||
await DeleteAsync(entities.ToList());
|
await DeleteAsync(entities.ToList());
|
||||||
}
|
}
|
||||||
|
|
||||||
public virtual async Task<List<TEntity>> GetListAsync(bool includeDetails = false,
|
public virtual async Task<List<TEntity>> GetListAsync(bool includeDetails = false, CancellationToken cancellationToken = default)
|
||||||
CancellationToken cancellationToken = default)
|
|
||||||
{
|
{
|
||||||
return await GetListAsync();
|
return await GetListAsync();
|
||||||
}
|
}
|
||||||
|
|
||||||
public virtual async Task<long> GetCountAsync(CancellationToken cancellationToken = default)
|
public virtual async Task<long> GetCountAsync(CancellationToken cancellationToken = default)
|
||||||
{
|
{
|
||||||
return await this.CountAsync(_ => true);
|
return await this.CountAsync(_=>true);
|
||||||
}
|
}
|
||||||
|
|
||||||
public virtual async Task<List<TEntity>> GetPagedListAsync(int skipCount, int maxResultCount, string sorting,
|
public virtual async Task<List<TEntity>> GetPagedListAsync(int skipCount, int maxResultCount, string sorting, bool includeDetails = false, CancellationToken cancellationToken = default)
|
||||||
bool includeDetails = false, CancellationToken cancellationToken = default)
|
|
||||||
{
|
{
|
||||||
return await GetPageListAsync(_ => true, skipCount, maxResultCount);
|
return await GetPageListAsync(_ => true, skipCount, maxResultCount);
|
||||||
}
|
}
|
||||||
|
|
||||||
#endregion
|
#endregion
|
||||||
|
|
||||||
|
|
||||||
#region 内置DB快捷操作
|
#region 内置DB快捷操作
|
||||||
|
|
||||||
public virtual async Task<IDeleteable<TEntity>> AsDeleteable()
|
public virtual async Task<IDeleteable<TEntity>> AsDeleteable()
|
||||||
{
|
{
|
||||||
return (await GetDbSimpleClientAsync()).AsDeleteable();
|
return (await GetDbSimpleClientAsync()).AsDeleteable();
|
||||||
@@ -232,11 +213,9 @@ namespace Yi.Framework.SqlSugarCore.Repositories
|
|||||||
{
|
{
|
||||||
return (await GetDbSimpleClientAsync()).AsUpdateable(updateObjs);
|
return (await GetDbSimpleClientAsync()).AsUpdateable(updateObjs);
|
||||||
}
|
}
|
||||||
|
|
||||||
#endregion
|
#endregion
|
||||||
|
|
||||||
#region SimpleClient模块
|
#region SimpleClient模块
|
||||||
|
|
||||||
public virtual async Task<int> CountAsync(Expression<Func<TEntity, bool>> whereExpression)
|
public virtual async Task<int> CountAsync(Expression<Func<TEntity, bool>> whereExpression)
|
||||||
{
|
{
|
||||||
return await (await GetDbSimpleClientAsync()).CountAsync(whereExpression);
|
return await (await GetDbSimpleClientAsync()).CountAsync(whereExpression);
|
||||||
@@ -253,6 +232,7 @@ namespace Yi.Framework.SqlSugarCore.Repositories
|
|||||||
{
|
{
|
||||||
return await (await GetDbSimpleClientAsync()).DeleteAsync(deleteObj);
|
return await (await GetDbSimpleClientAsync()).DeleteAsync(deleteObj);
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
public virtual async Task<bool> DeleteAsync(List<TEntity> deleteObjs)
|
public virtual async Task<bool> DeleteAsync(List<TEntity> deleteObjs)
|
||||||
@@ -272,13 +252,13 @@ namespace Yi.Framework.SqlSugarCore.Repositories
|
|||||||
{
|
{
|
||||||
if (typeof(ISoftDelete).IsAssignableFrom(typeof(TEntity)))
|
if (typeof(ISoftDelete).IsAssignableFrom(typeof(TEntity)))
|
||||||
{
|
{
|
||||||
return await (await GetDbSimpleClientAsync()).AsUpdateable()
|
return await (await GetDbSimpleClientAsync()).AsUpdateable().SetColumns(nameof(ISoftDelete.IsDeleted), true).Where(whereExpression).ExecuteCommandAsync() > 0;
|
||||||
.SetColumns(nameof(ISoftDelete.IsDeleted), true).Where(whereExpression).ExecuteCommandAsync() > 0;
|
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
return await (await GetDbSimpleClientAsync()).DeleteAsync(whereExpression);
|
return await (await GetDbSimpleClientAsync()).DeleteAsync(whereExpression);
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
public virtual async Task<bool> DeleteByIdAsync(dynamic id)
|
public virtual async Task<bool> DeleteByIdAsync(dynamic id)
|
||||||
@@ -286,11 +266,7 @@ namespace Yi.Framework.SqlSugarCore.Repositories
|
|||||||
if (typeof(ISoftDelete).IsAssignableFrom(typeof(TEntity)))
|
if (typeof(ISoftDelete).IsAssignableFrom(typeof(TEntity)))
|
||||||
{
|
{
|
||||||
var entity = await GetByIdAsync(id);
|
var entity = await GetByIdAsync(id);
|
||||||
if (entity is null)
|
if (entity == null) return false;
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
//反射赋值
|
//反射赋值
|
||||||
ReflexHelper.SetModelValue(nameof(ISoftDelete.IsDeleted), true, entity);
|
ReflexHelper.SetModelValue(nameof(ISoftDelete.IsDeleted), true, entity);
|
||||||
return await UpdateAsync(entity);
|
return await UpdateAsync(entity);
|
||||||
@@ -311,7 +287,6 @@ namespace Yi.Framework.SqlSugarCore.Repositories
|
|||||||
{
|
{
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
//反射赋值
|
//反射赋值
|
||||||
entities.ForEach(e => ReflexHelper.SetModelValue(nameof(ISoftDelete.IsDeleted), true, e));
|
entities.ForEach(e => ReflexHelper.SetModelValue(nameof(ISoftDelete.IsDeleted), true, e));
|
||||||
return await UpdateRangeAsync(entities);
|
return await UpdateRangeAsync(entities);
|
||||||
@@ -320,6 +295,7 @@ namespace Yi.Framework.SqlSugarCore.Repositories
|
|||||||
{
|
{
|
||||||
return await (await GetDbSimpleClientAsync()).DeleteByIdAsync(ids);
|
return await (await GetDbSimpleClientAsync()).DeleteByIdAsync(ids);
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
public virtual async Task<TEntity> GetByIdAsync(dynamic id)
|
public virtual async Task<TEntity> GetByIdAsync(dynamic id)
|
||||||
@@ -328,6 +304,7 @@ namespace Yi.Framework.SqlSugarCore.Repositories
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
public virtual async Task<TEntity> GetFirstAsync(Expression<Func<TEntity, bool>> whereExpression)
|
public virtual async Task<TEntity> GetFirstAsync(Expression<Func<TEntity, bool>> whereExpression)
|
||||||
{
|
{
|
||||||
return await (await GetDbSimpleClientAsync()).GetFirstAsync(whereExpression);
|
return await (await GetDbSimpleClientAsync()).GetFirstAsync(whereExpression);
|
||||||
@@ -343,19 +320,14 @@ namespace Yi.Framework.SqlSugarCore.Repositories
|
|||||||
return await (await GetDbSimpleClientAsync()).GetListAsync(whereExpression);
|
return await (await GetDbSimpleClientAsync()).GetListAsync(whereExpression);
|
||||||
}
|
}
|
||||||
|
|
||||||
public virtual async Task<List<TEntity>> GetPageListAsync(Expression<Func<TEntity, bool>> whereExpression,
|
public virtual async Task<List<TEntity>> GetPageListAsync(Expression<Func<TEntity, bool>> whereExpression, int pageNum, int pageSize)
|
||||||
int pageNum, int pageSize)
|
|
||||||
{
|
{
|
||||||
return await (await GetDbSimpleClientAsync()).GetPageListAsync(whereExpression,
|
return await (await AsQueryable()).Where(whereExpression).ToPageListAsync(pageNum, pageSize);
|
||||||
new PageModel() { PageIndex = pageNum, PageSize = pageSize });
|
|
||||||
}
|
}
|
||||||
|
|
||||||
public virtual async Task<List<TEntity>> GetPageListAsync(Expression<Func<TEntity, bool>> whereExpression,
|
public virtual async Task<List<TEntity>> GetPageListAsync(Expression<Func<TEntity, bool>> whereExpression, int pageNum, int pageSize, Expression<Func<TEntity, object>>? orderByExpression = null, OrderByType orderByType = OrderByType.Asc)
|
||||||
int pageNum, int pageSize, Expression<Func<TEntity, object>>? orderByExpression = null,
|
|
||||||
OrderByType orderByType = OrderByType.Asc)
|
|
||||||
{
|
{
|
||||||
return await (await GetDbSimpleClientAsync()).GetPageListAsync(whereExpression,
|
return await (await AsQueryable()).Where(whereExpression) .OrderBy( orderByExpression,orderByType).ToPageListAsync(pageNum, pageSize);
|
||||||
new PageModel { PageIndex = pageNum, PageSize = pageSize }, orderByExpression, orderByType);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
public virtual async Task<TEntity> GetSingleAsync(Expression<Func<TEntity, bool>> whereExpression)
|
public virtual async Task<TEntity> GetSingleAsync(Expression<Func<TEntity, bool>> whereExpression)
|
||||||
@@ -410,9 +382,9 @@ namespace Yi.Framework.SqlSugarCore.Repositories
|
|||||||
|
|
||||||
public virtual async Task<bool> UpdateAsync(TEntity updateObj)
|
public virtual async Task<bool> UpdateAsync(TEntity updateObj)
|
||||||
{
|
{
|
||||||
if (typeof(TEntity).IsAssignableTo<IHasConcurrencyStamp>()) //带版本号乐观锁更新
|
if (Options is not null && Options.EnabledConcurrencyException)
|
||||||
{
|
{
|
||||||
if (Options is not null && Options.EnabledConcurrencyException)
|
if (typeof(TEntity).IsAssignableTo<IHasConcurrencyStamp>()) //带版本号乐观锁更新
|
||||||
{
|
{
|
||||||
try
|
try
|
||||||
{
|
{
|
||||||
@@ -426,24 +398,18 @@ namespace Yi.Framework.SqlSugarCore.Repositories
|
|||||||
$"{ex.Message}[更新失败:ConcurrencyStamp不是最新版本],entityInfo:{updateObj}", ex);
|
$"{ex.Message}[更新失败:ConcurrencyStamp不是最新版本],entityInfo:{updateObj}", ex);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
else
|
|
||||||
{
|
|
||||||
int num = await (await GetDbSimpleClientAsync())
|
|
||||||
.Context.Updateable(updateObj).ExecuteCommandAsync();
|
|
||||||
return num > 0;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return await (await GetDbSimpleClientAsync()).UpdateAsync(updateObj);
|
return await (await GetDbSimpleClientAsync()).UpdateAsync(updateObj);
|
||||||
}
|
}
|
||||||
|
|
||||||
public virtual async Task<bool> UpdateAsync(Expression<Func<TEntity, TEntity>> columns,
|
public virtual async Task<bool> UpdateAsync(Expression<Func<TEntity, TEntity>> columns, Expression<Func<TEntity, bool>> whereExpression)
|
||||||
Expression<Func<TEntity, bool>> whereExpression)
|
|
||||||
{
|
{
|
||||||
return await (await GetDbSimpleClientAsync()).UpdateAsync(columns, whereExpression);
|
return await (await GetDbSimpleClientAsync()).UpdateAsync(columns, whereExpression);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
public virtual async Task<bool> UpdateRangeAsync(List<TEntity> updateObjs)
|
public virtual async Task<bool> UpdateRangeAsync(List<TEntity> updateObjs)
|
||||||
{
|
{
|
||||||
return await (await GetDbSimpleClientAsync()).UpdateRangeAsync(updateObjs);
|
return await (await GetDbSimpleClientAsync()).UpdateRangeAsync(updateObjs);
|
||||||
@@ -452,34 +418,28 @@ namespace Yi.Framework.SqlSugarCore.Repositories
|
|||||||
#endregion
|
#endregion
|
||||||
}
|
}
|
||||||
|
|
||||||
public class SqlSugarRepository<TEntity, TKey> : SqlSugarRepository<TEntity>, ISqlSugarRepository<TEntity, TKey>,
|
public class SqlSugarRepository<TEntity, TKey> : SqlSugarRepository<TEntity>, ISqlSugarRepository<TEntity, TKey>, IRepository<TEntity, TKey> where TEntity : class, IEntity<TKey>, new()
|
||||||
IRepository<TEntity, TKey> where TEntity : class, IEntity<TKey>, new()
|
|
||||||
{
|
{
|
||||||
public SqlSugarRepository(ISugarDbContextProvider<ISqlSugarDbContext> dbContextProvider) : base(
|
public SqlSugarRepository(ISugarDbContextProvider<ISqlSugarDbContext> sugarDbContextProvider) : base(sugarDbContextProvider)
|
||||||
dbContextProvider)
|
|
||||||
{
|
{
|
||||||
}
|
}
|
||||||
|
|
||||||
public virtual async Task DeleteAsync(TKey id, bool autoSave = false,
|
public virtual async Task DeleteAsync(TKey id, bool autoSave = false, CancellationToken cancellationToken = default)
|
||||||
CancellationToken cancellationToken = default)
|
|
||||||
{
|
{
|
||||||
await DeleteByIdAsync(id);
|
await DeleteByIdAsync(id);
|
||||||
}
|
}
|
||||||
|
|
||||||
public virtual async Task DeleteManyAsync(IEnumerable<TKey> ids, bool autoSave = false,
|
public virtual async Task DeleteManyAsync(IEnumerable<TKey> ids, bool autoSave = false, CancellationToken cancellationToken = default)
|
||||||
CancellationToken cancellationToken = default)
|
|
||||||
{
|
{
|
||||||
await DeleteByIdsAsync(ids.Select(x => (object)x).ToArray());
|
await DeleteByIdsAsync(ids.Select(x => (object)x).ToArray());
|
||||||
}
|
}
|
||||||
|
|
||||||
public virtual async Task<TEntity?> FindAsync(TKey id, bool includeDetails = true,
|
public virtual async Task<TEntity?> FindAsync(TKey id, bool includeDetails = true, CancellationToken cancellationToken = default)
|
||||||
CancellationToken cancellationToken = default)
|
|
||||||
{
|
{
|
||||||
return await GetByIdAsync(id);
|
return await GetByIdAsync(id);
|
||||||
}
|
}
|
||||||
|
|
||||||
public virtual async Task<TEntity> GetAsync(TKey id, bool includeDetails = true,
|
public virtual async Task<TEntity> GetAsync(TKey id, bool includeDetails = true, CancellationToken cancellationToken = default)
|
||||||
CancellationToken cancellationToken = default)
|
|
||||||
{
|
{
|
||||||
return await GetByIdAsync(id);
|
return await GetByIdAsync(id);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -0,0 +1,22 @@
|
|||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.Announcement;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 公告缓存 DTO
|
||||||
|
/// </summary>
|
||||||
|
public class AnnouncementCacheDto
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 版本号
|
||||||
|
/// </summary>
|
||||||
|
public string Version { get; set; } = string.Empty;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 公告日志列表
|
||||||
|
/// </summary>
|
||||||
|
public List<AnnouncementLogDto> Logs { get; set; } = new List<AnnouncementLogDto>();
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 跳转链接
|
||||||
|
/// </summary>
|
||||||
|
public string? Url { get; set; }
|
||||||
|
}
|
||||||
@@ -0,0 +1,44 @@
|
|||||||
|
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.Announcement;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 公告日志 DTO
|
||||||
|
/// </summary>
|
||||||
|
public class AnnouncementLogDto
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 标题
|
||||||
|
/// </summary>
|
||||||
|
public string Title { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 内容列表
|
||||||
|
/// </summary>
|
||||||
|
public List<string> Content { get; set; } = new List<string>();
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 图片url
|
||||||
|
/// </summary>
|
||||||
|
public string? ImageUrl { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 开始时间(系统公告时间、活动开始时间)
|
||||||
|
/// </summary>
|
||||||
|
public DateTime StartTime { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 活动结束时间
|
||||||
|
/// </summary>
|
||||||
|
public DateTime? EndTime { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 公告类型(系统、活动)
|
||||||
|
/// </summary>
|
||||||
|
public AnnouncementTypeEnum Type{ get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 跳转链接
|
||||||
|
/// </summary>
|
||||||
|
public string? Url { get; set; }
|
||||||
|
}
|
||||||
@@ -0,0 +1,17 @@
|
|||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.Announcement;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 公告输出 DTO
|
||||||
|
/// </summary>
|
||||||
|
public class AnnouncementOutput
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 版本号
|
||||||
|
/// </summary>
|
||||||
|
public string Version { get; set; } = string.Empty;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 公告日志列表
|
||||||
|
/// </summary>
|
||||||
|
public List<AnnouncementLogDto> Logs { get; set; } = new List<AnnouncementLogDto>();
|
||||||
|
}
|
||||||
@@ -40,11 +40,6 @@ public class CardFlipStatusOutput
|
|||||||
/// </summary>
|
/// </summary>
|
||||||
public int InvitedCount { get; set; }
|
public int InvitedCount { get; set; }
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 是否已被邀请(被邀请后不可再提供邀请码)
|
|
||||||
/// </summary>
|
|
||||||
public bool IsInvited { get; set; }
|
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 翻牌记录
|
/// 翻牌记录
|
||||||
/// </summary>
|
/// </summary>
|
||||||
@@ -54,6 +49,11 @@ public class CardFlipStatusOutput
|
|||||||
/// 下次可翻牌提示
|
/// 下次可翻牌提示
|
||||||
/// </summary>
|
/// </summary>
|
||||||
public string? NextFlipTip { get; set; }
|
public string? NextFlipTip { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 当前用户是否已经填写过邀请码
|
||||||
|
/// </summary>
|
||||||
|
public bool IsFilledInviteCode { get; set; }
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
|
|||||||
@@ -0,0 +1,17 @@
|
|||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.Model;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// API类型选项
|
||||||
|
/// </summary>
|
||||||
|
public class ModelApiTypeOption
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 显示名称
|
||||||
|
/// </summary>
|
||||||
|
public string Label { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 枚举值
|
||||||
|
/// </summary>
|
||||||
|
public int Value { get; set; }
|
||||||
|
}
|
||||||
@@ -0,0 +1,79 @@
|
|||||||
|
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Extensions;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.Model;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型库展示数据
|
||||||
|
/// </summary>
|
||||||
|
public class ModelLibraryDto
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 模型ID
|
||||||
|
/// </summary>
|
||||||
|
public string ModelId { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型名称
|
||||||
|
/// </summary>
|
||||||
|
public string Name { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型描述
|
||||||
|
/// </summary>
|
||||||
|
public string? Description { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型类型
|
||||||
|
/// </summary>
|
||||||
|
public ModelTypeEnum ModelType { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型类型名称
|
||||||
|
/// </summary>
|
||||||
|
public string ModelTypeName => ModelType.GetDescription();
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型支持的API类型
|
||||||
|
/// </summary>
|
||||||
|
public List<ModelApiTypeOutput> ModelApiTypes { get; set; }
|
||||||
|
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型显示倍率
|
||||||
|
/// </summary>
|
||||||
|
public decimal MultiplierShow { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 供应商分组名称
|
||||||
|
/// </summary>
|
||||||
|
public string? ProviderName { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型图标URL
|
||||||
|
/// </summary>
|
||||||
|
public string? IconUrl { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 是否为尊享模型(PremiumChat类型)
|
||||||
|
/// </summary>
|
||||||
|
public bool IsPremium { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 排序
|
||||||
|
/// </summary>
|
||||||
|
public int OrderNum { get; set; }
|
||||||
|
}
|
||||||
|
|
||||||
|
public class ModelApiTypeOutput
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 模型类型
|
||||||
|
/// </summary>
|
||||||
|
public ModelApiTypeEnum ModelApiType { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型类型名称
|
||||||
|
/// </summary>
|
||||||
|
public string ModelApiTypeName => ModelApiType.GetDescription();
|
||||||
|
}
|
||||||
@@ -0,0 +1,35 @@
|
|||||||
|
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
||||||
|
using Yi.Framework.Ddd.Application.Contracts;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.Model;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取模型库列表查询参数
|
||||||
|
/// </summary>
|
||||||
|
public class ModelLibraryGetListInput : PagedAllResultRequestDto
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 搜索关键词(搜索模型名称、模型ID)
|
||||||
|
/// </summary>
|
||||||
|
public string? SearchKey { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 供应商名称筛选
|
||||||
|
/// </summary>
|
||||||
|
public List<string>? ProviderNames { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型类型筛选
|
||||||
|
/// </summary>
|
||||||
|
public List<ModelTypeEnum>? ModelTypes { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// API类型筛选
|
||||||
|
/// </summary>
|
||||||
|
public List<ModelApiTypeEnum>? ModelApiTypes { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 是否只显示尊享模型
|
||||||
|
/// </summary>
|
||||||
|
public bool? IsPremiumOnly { get; set; }
|
||||||
|
}
|
||||||
@@ -0,0 +1,17 @@
|
|||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.Model;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型类型选项
|
||||||
|
/// </summary>
|
||||||
|
public class ModelTypeOption
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 显示名称
|
||||||
|
/// </summary>
|
||||||
|
public string Label { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 枚举值
|
||||||
|
/// </summary>
|
||||||
|
public int Value { get; set; }
|
||||||
|
}
|
||||||
@@ -62,4 +62,9 @@ public class ModelGetListOutput
|
|||||||
/// 备注信息
|
/// 备注信息
|
||||||
/// </summary>
|
/// </summary>
|
||||||
public string? Remark { get; set; }
|
public string? Remark { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 是否为尊享包
|
||||||
|
/// </summary>
|
||||||
|
public bool IsPremiumPackage { get; set; }
|
||||||
}
|
}
|
||||||
@@ -0,0 +1,26 @@
|
|||||||
|
using System.ComponentModel.DataAnnotations;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.Token;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 创建Token输入
|
||||||
|
/// </summary>
|
||||||
|
public class TokenCreateInput
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 名称(同一用户不能重复)
|
||||||
|
/// </summary>
|
||||||
|
[Required(ErrorMessage = "名称不能为空")]
|
||||||
|
[StringLength(100, ErrorMessage = "名称长度不能超过100个字符")]
|
||||||
|
public string Name { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 过期时间(空为永不过期)
|
||||||
|
/// </summary>
|
||||||
|
public DateTime? ExpireTime { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 尊享包额度限制(空为不限制)
|
||||||
|
/// </summary>
|
||||||
|
public long? PremiumQuotaLimit { get; set; }
|
||||||
|
}
|
||||||
@@ -0,0 +1,47 @@
|
|||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.Token;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Token列表输出
|
||||||
|
/// </summary>
|
||||||
|
public class TokenGetListOutputDto
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// Token Id
|
||||||
|
/// </summary>
|
||||||
|
public Guid Id { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 名称
|
||||||
|
/// </summary>
|
||||||
|
public string Name { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Token密钥
|
||||||
|
/// </summary>
|
||||||
|
public string ApiKey { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 过期时间(空为永不过期)
|
||||||
|
/// </summary>
|
||||||
|
public DateTime? ExpireTime { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 尊享包额度限制(空为不限制)
|
||||||
|
/// </summary>
|
||||||
|
public long? PremiumQuotaLimit { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 尊享包已使用额度
|
||||||
|
/// </summary>
|
||||||
|
public long PremiumUsedQuota { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 是否禁用
|
||||||
|
/// </summary>
|
||||||
|
public bool IsDisabled { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 创建时间
|
||||||
|
/// </summary>
|
||||||
|
public DateTime CreationTime { get; set; }
|
||||||
|
}
|
||||||
@@ -0,0 +1,9 @@
|
|||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.Token;
|
||||||
|
|
||||||
|
public class TokenSelectListOutputDto
|
||||||
|
{
|
||||||
|
public Guid TokenId { get; set; }
|
||||||
|
public string Name { get; set; }
|
||||||
|
|
||||||
|
public bool IsDisabled { get; set; }
|
||||||
|
}
|
||||||
@@ -0,0 +1,32 @@
|
|||||||
|
using System.ComponentModel.DataAnnotations;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.Token;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 编辑Token输入
|
||||||
|
/// </summary>
|
||||||
|
public class TokenUpdateInput
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// Token Id
|
||||||
|
/// </summary>
|
||||||
|
[Required(ErrorMessage = "Id不能为空")]
|
||||||
|
public Guid Id { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 名称(同一用户不能重复)
|
||||||
|
/// </summary>
|
||||||
|
[Required(ErrorMessage = "名称不能为空")]
|
||||||
|
[StringLength(100, ErrorMessage = "名称长度不能超过100个字符")]
|
||||||
|
public string Name { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 过期时间(空为永不过期)
|
||||||
|
/// </summary>
|
||||||
|
public DateTime? ExpireTime { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 尊享包额度限制(空为不限制)
|
||||||
|
/// </summary>
|
||||||
|
public long? PremiumQuotaLimit { get; set; }
|
||||||
|
}
|
||||||
@@ -0,0 +1,12 @@
|
|||||||
|
using Volo.Abp.Application.Dtos;
|
||||||
|
using Yi.Framework.Ddd.Application.Contracts;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.UsageStatistics;
|
||||||
|
|
||||||
|
public class PremiumTokenUsageGetListInput : PagedAllResultRequestDto
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 是否免费
|
||||||
|
/// </summary>
|
||||||
|
public bool? IsFree { get; set; }
|
||||||
|
}
|
||||||
@@ -0,0 +1,57 @@
|
|||||||
|
using Volo.Abp.Application.Dtos;
|
||||||
|
using Yi.Framework.Ddd.Application.Contracts;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.UsageStatistics;
|
||||||
|
|
||||||
|
public class PremiumTokenUsageGetListOutput : CreationAuditedEntityDto
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// id
|
||||||
|
/// </summary>
|
||||||
|
public Guid Id { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 用户ID
|
||||||
|
/// </summary>
|
||||||
|
public Guid UserId { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 包名称
|
||||||
|
/// </summary>
|
||||||
|
public string PackageName { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 总用量(总token数)
|
||||||
|
/// </summary>
|
||||||
|
public long TotalTokens { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 剩余用量(剩余token数)
|
||||||
|
/// </summary>
|
||||||
|
public long RemainingTokens { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 已使用token数
|
||||||
|
/// </summary>
|
||||||
|
public long UsedTokens { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 到期时间
|
||||||
|
/// </summary>
|
||||||
|
public DateTime? ExpireDateTime { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 是否激活
|
||||||
|
/// </summary>
|
||||||
|
public bool IsActive { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 购买金额
|
||||||
|
/// </summary>
|
||||||
|
public decimal PurchaseAmount { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 备注
|
||||||
|
/// </summary>
|
||||||
|
public string? Remark { get; set; }
|
||||||
|
}
|
||||||
@@ -0,0 +1,27 @@
|
|||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.UsageStatistics;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 尊享包不同Token用量占比DTO(饼图)
|
||||||
|
/// </summary>
|
||||||
|
public class TokenPremiumUsageDto
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// Token Id
|
||||||
|
/// </summary>
|
||||||
|
public Guid TokenId { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Token名称
|
||||||
|
/// </summary>
|
||||||
|
public string TokenName { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Token消耗量
|
||||||
|
/// </summary>
|
||||||
|
public long Tokens { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 占比(百分比)
|
||||||
|
/// </summary>
|
||||||
|
public decimal Percentage { get; set; }
|
||||||
|
}
|
||||||
@@ -0,0 +1,9 @@
|
|||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.UsageStatistics;
|
||||||
|
|
||||||
|
public class UsageStatisticsGetInput
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// tokenId
|
||||||
|
/// </summary>
|
||||||
|
public Guid? TokenId { get; set; }
|
||||||
|
}
|
||||||
@@ -0,0 +1,15 @@
|
|||||||
|
using Yi.Framework.AiHub.Application.Contracts.Dtos.Announcement;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.IServices;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 公告服务接口
|
||||||
|
/// </summary>
|
||||||
|
public interface IAnnouncementService
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 获取公告信息
|
||||||
|
/// </summary>
|
||||||
|
/// <returns>公告信息</returns>
|
||||||
|
Task<List<AnnouncementLogDto>> GetAsync();
|
||||||
|
}
|
||||||
@@ -0,0 +1,35 @@
|
|||||||
|
using Volo.Abp.Application.Dtos;
|
||||||
|
using Yi.Framework.AiHub.Application.Contracts.Dtos.Model;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.IServices;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型服务接口
|
||||||
|
/// </summary>
|
||||||
|
public interface IModelService
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 获取模型库列表(公开接口,无需登录)
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="input">查询参数</param>
|
||||||
|
/// <returns>分页模型列表</returns>
|
||||||
|
Task<PagedResultDto<ModelLibraryDto>> GetListAsync(ModelLibraryGetListInput input);
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取供应商列表(公开接口,无需登录)
|
||||||
|
/// </summary>
|
||||||
|
/// <returns>供应商列表</returns>
|
||||||
|
Task<List<string>> GetProviderListAsync();
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取模型类型选项列表(公开接口,无需登录)
|
||||||
|
/// </summary>
|
||||||
|
/// <returns>模型类型选项</returns>
|
||||||
|
Task<List<ModelTypeOption>> GetModelTypeOptionsAsync();
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取API类型选项列表(公开接口,无需登录)
|
||||||
|
/// </summary>
|
||||||
|
/// <returns>API类型选项</returns>
|
||||||
|
Task<List<ModelApiTypeOption>> GetApiTypeOptionsAsync();
|
||||||
|
}
|
||||||
@@ -11,13 +11,13 @@ public interface IUsageStatisticsService
|
|||||||
/// 获取当前用户近7天的Token消耗统计
|
/// 获取当前用户近7天的Token消耗统计
|
||||||
/// </summary>
|
/// </summary>
|
||||||
/// <returns>每日Token使用量列表</returns>
|
/// <returns>每日Token使用量列表</returns>
|
||||||
Task<List<DailyTokenUsageDto>> GetLast7DaysTokenUsageAsync();
|
Task<List<DailyTokenUsageDto>> GetLast7DaysTokenUsageAsync(UsageStatisticsGetInput input);
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 获取当前用户各个模型的Token消耗量及占比
|
/// 获取当前用户各个模型的Token消耗量及占比
|
||||||
/// </summary>
|
/// </summary>
|
||||||
/// <returns>模型Token使用量列表</returns>
|
/// <returns>模型Token使用量列表</returns>
|
||||||
Task<List<ModelTokenUsageDto>> GetModelTokenUsageAsync();
|
Task<List<ModelTokenUsageDto>> GetModelTokenUsageAsync(UsageStatisticsGetInput input);
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 获取当前用户尊享服务Token用量统计
|
/// 获取当前用户尊享服务Token用量统计
|
||||||
|
|||||||
@@ -51,9 +51,8 @@ public class CardFlipService : ApplicationService, ICardFlipService
|
|||||||
// 统计本周邀请人数
|
// 统计本周邀请人数
|
||||||
var invitedCount = await _inviteCodeManager.GetWeeklyInvitationCountAsync(userId, weekStart);
|
var invitedCount = await _inviteCodeManager.GetWeeklyInvitationCountAsync(userId, weekStart);
|
||||||
|
|
||||||
// 检查用户是否已被邀请
|
//当前用户是否已填写过邀请码
|
||||||
var isInvited = await _inviteCodeManager.IsUserInvitedAsync(userId);
|
var isFilledInviteCode = await _inviteCodeManager.IsFilledInviteCodeAsync(userId);
|
||||||
|
|
||||||
var output = new CardFlipStatusOutput
|
var output = new CardFlipStatusOutput
|
||||||
{
|
{
|
||||||
TotalFlips = task?.TotalFlips ?? 0,
|
TotalFlips = task?.TotalFlips ?? 0,
|
||||||
@@ -63,8 +62,8 @@ public class CardFlipService : ApplicationService, ICardFlipService
|
|||||||
CanFlip = _cardFlipManager.CanFlipCard(task),
|
CanFlip = _cardFlipManager.CanFlipCard(task),
|
||||||
MyInviteCode = inviteCode?.Code,
|
MyInviteCode = inviteCode?.Code,
|
||||||
InvitedCount = invitedCount,
|
InvitedCount = invitedCount,
|
||||||
IsInvited = isInvited,
|
FlipRecords = BuildFlipRecords(task),
|
||||||
FlipRecords = BuildFlipRecords(task)
|
IsFilledInviteCode = isFilledInviteCode
|
||||||
};
|
};
|
||||||
|
|
||||||
// 生成提示信息
|
// 生成提示信息
|
||||||
@@ -87,7 +86,7 @@ public class CardFlipService : ApplicationService, ICardFlipService
|
|||||||
// 如果中奖,发放奖励
|
// 如果中奖,发放奖励
|
||||||
if (result.IsWin)
|
if (result.IsWin)
|
||||||
{
|
{
|
||||||
await GrantRewardAsync(userId, result.RewardAmount, $"翻牌活动第{input.FlipNumber}次中奖");
|
await GrantRewardAsync(userId, result.RewardAmount, $"翻牌活动-序号{input.FlipNumber}中奖");
|
||||||
}
|
}
|
||||||
|
|
||||||
// 构建输出
|
// 构建输出
|
||||||
@@ -147,7 +146,6 @@ public class CardFlipService : ApplicationService, ICardFlipService
|
|||||||
{
|
{
|
||||||
MyInviteCode = inviteCode?.Code,
|
MyInviteCode = inviteCode?.Code,
|
||||||
InvitedCount = invitedCount,
|
InvitedCount = invitedCount,
|
||||||
IsInvited = inviteCode?.IsUserInvited ?? false,
|
|
||||||
InvitationHistory = invitationHistory.Select(x => new InvitationHistoryItem
|
InvitationHistory = invitationHistory.Select(x => new InvitationHistoryItem
|
||||||
{
|
{
|
||||||
InvitedUserName = x.InvitedUserName,
|
InvitedUserName = x.InvitedUserName,
|
||||||
@@ -183,6 +181,9 @@ public class CardFlipService : ApplicationService, ICardFlipService
|
|||||||
var flippedOrder = task != null ? _cardFlipManager.GetFlippedOrder(task) : new List<int>();
|
var flippedOrder = task != null ? _cardFlipManager.GetFlippedOrder(task) : new List<int>();
|
||||||
var flippedNumbers = new HashSet<int>(flippedOrder);
|
var flippedNumbers = new HashSet<int>(flippedOrder);
|
||||||
|
|
||||||
|
// 获取中奖记录
|
||||||
|
var winRecords = task?.WinRecords ?? new Dictionary<int, long>();
|
||||||
|
|
||||||
// 构建记录,按照原始序号1-10排列
|
// 构建记录,按照原始序号1-10排列
|
||||||
for (int i = 1; i <= CardFlipManager.TOTAL_MAX_FLIPS; i++)
|
for (int i = 1; i <= CardFlipManager.TOTAL_MAX_FLIPS; i++)
|
||||||
{
|
{
|
||||||
@@ -202,23 +203,11 @@ public class CardFlipService : ApplicationService, ICardFlipService
|
|||||||
{
|
{
|
||||||
var flipOrderIndex = flippedOrder.IndexOf(i) + 1; // 第几次翻的(1-based)
|
var flipOrderIndex = flippedOrder.IndexOf(i) + 1; // 第几次翻的(1-based)
|
||||||
|
|
||||||
// 第8次翻的卡中奖
|
// 检查这次翻牌是否中奖
|
||||||
if (flipOrderIndex == 8)
|
if (winRecords.TryGetValue(flipOrderIndex, out var rewardAmount))
|
||||||
{
|
{
|
||||||
record.IsWin = true;
|
record.IsWin = true;
|
||||||
record.RewardAmount = task.EighthRewardAmount;
|
record.RewardAmount = rewardAmount;
|
||||||
}
|
|
||||||
// 第9次翻的卡中奖
|
|
||||||
else if (flipOrderIndex == 9)
|
|
||||||
{
|
|
||||||
record.IsWin = true;
|
|
||||||
record.RewardAmount = task.NinthRewardAmount;
|
|
||||||
}
|
|
||||||
// 第10次翻的卡中奖
|
|
||||||
else if (flipOrderIndex == 10)
|
|
||||||
{
|
|
||||||
record.IsWin = true;
|
|
||||||
record.RewardAmount = task.TenthRewardAmount;
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -246,10 +235,10 @@ public class CardFlipService : ApplicationService, ICardFlipService
|
|||||||
{
|
{
|
||||||
if (status.TotalFlips >= 7)
|
if (status.TotalFlips >= 7)
|
||||||
{
|
{
|
||||||
return $"本周使用他人邀请码可解锁{status.RemainingInviteFlips}次翻牌,且必中大奖!每次中奖最大额度将翻倍!";
|
return $"本周使用他人邀请码或他人使用你的邀请码,可解锁{status.RemainingInviteFlips}次翻牌,且必中大奖!每次中奖最大额度将翻倍!";
|
||||||
}
|
}
|
||||||
|
|
||||||
return $"本周使用他人邀请码可解锁{status.RemainingInviteFlips}次翻牌,必中大奖!每次中奖最大额度将翻倍!";
|
return $"本周使用他人邀请码或他人使用你的邀请码,可解锁{status.RemainingInviteFlips}次翻牌,必中大奖!每次中奖最大额度将翻倍!";
|
||||||
}
|
}
|
||||||
|
|
||||||
return "继续加油!";
|
return "继续加油!";
|
||||||
@@ -28,8 +28,8 @@ public class DailyTaskService : ApplicationService
|
|||||||
private readonly Dictionary<int, (long RequiredTokens, long RewardTokens, string Name, string Description)>
|
private readonly Dictionary<int, (long RequiredTokens, long RewardTokens, string Name, string Description)>
|
||||||
_taskConfigs = new()
|
_taskConfigs = new()
|
||||||
{
|
{
|
||||||
{ 1, (10000000, 2000000, "尊享包1000w token任务", "累积使用尊享包 1000w token") }, // 1000w消耗 -> 200w奖励
|
{ 1, (10000000, 1000000, "尊享包1000w token任务", "累积使用尊享包 1000w token") }, // 1000w消耗 -> 100w奖励
|
||||||
{ 2, (30000000, 4000000, "尊享包3000w token任务", "累积使用尊享包 3000w token") } // 3000w消耗 -> 600w奖励
|
{ 2, (30000000, 2000000, "尊享包3000w token任务", "累积使用尊享包 3000w token") } // 3000w消耗 -> 200w奖励
|
||||||
};
|
};
|
||||||
|
|
||||||
public DailyTaskService(
|
public DailyTaskService(
|
||||||
@@ -17,15 +17,18 @@ public class AiAccountService : ApplicationService
|
|||||||
private IAccountService _accountService;
|
private IAccountService _accountService;
|
||||||
private ISqlSugarRepository<AiUserExtraInfoEntity> _userRepository;
|
private ISqlSugarRepository<AiUserExtraInfoEntity> _userRepository;
|
||||||
private ISqlSugarRepository<AiRechargeAggregateRoot> _rechargeRepository;
|
private ISqlSugarRepository<AiRechargeAggregateRoot> _rechargeRepository;
|
||||||
|
private ISqlSugarRepository<PremiumPackageAggregateRoot> _premiumPackageRepository;
|
||||||
|
|
||||||
public AiAccountService(
|
public AiAccountService(
|
||||||
IAccountService accountService,
|
IAccountService accountService,
|
||||||
ISqlSugarRepository<AiUserExtraInfoEntity> userRepository,
|
ISqlSugarRepository<AiUserExtraInfoEntity> userRepository,
|
||||||
ISqlSugarRepository<AiRechargeAggregateRoot> rechargeRepository)
|
ISqlSugarRepository<AiRechargeAggregateRoot> rechargeRepository,
|
||||||
|
ISqlSugarRepository<PremiumPackageAggregateRoot> premiumPackageRepository)
|
||||||
{
|
{
|
||||||
_accountService = accountService;
|
_accountService = accountService;
|
||||||
_userRepository = userRepository;
|
_userRepository = userRepository;
|
||||||
_rechargeRepository = rechargeRepository;
|
_rechargeRepository = rechargeRepository;
|
||||||
|
_premiumPackageRepository = premiumPackageRepository;
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
@@ -72,4 +75,77 @@ public class AiAccountService : ApplicationService
|
|||||||
|
|
||||||
return output;
|
return output;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取利润统计数据
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="currentCost">当前成本(RMB)</param>
|
||||||
|
/// <returns></returns>
|
||||||
|
[Authorize]
|
||||||
|
[HttpGet("account/profit-statistics")]
|
||||||
|
public async Task<string> GetProfitStatisticsAsync([FromQuery] decimal currentCost)
|
||||||
|
{
|
||||||
|
if (CurrentUser.UserName != "Guo" && CurrentUser.UserName != "cc")
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("您暂无权限访问");
|
||||||
|
}
|
||||||
|
|
||||||
|
// 1. 获取尊享包总消耗和剩余库存
|
||||||
|
var premiumPackages = await _premiumPackageRepository._DbQueryable.ToListAsync();
|
||||||
|
long totalUsedTokens = premiumPackages.Sum(p => p.UsedTokens);
|
||||||
|
long totalRemainingTokens = premiumPackages.Sum(p => p.RemainingTokens);
|
||||||
|
|
||||||
|
// 2. 计算1亿Token成本
|
||||||
|
decimal costPerHundredMillion = totalUsedTokens > 0
|
||||||
|
? currentCost / (totalUsedTokens / 100000000m)
|
||||||
|
: 0;
|
||||||
|
|
||||||
|
// 3. 计算总成本(剩余+已使用的总成本)
|
||||||
|
long totalTokens = totalUsedTokens + totalRemainingTokens;
|
||||||
|
decimal totalCost = totalTokens > 0
|
||||||
|
? (totalTokens / 100000000m) * costPerHundredMillion
|
||||||
|
: 0;
|
||||||
|
|
||||||
|
// 4. 获取总收益(RechargeType=PremiumPackage的充值金额总和)
|
||||||
|
decimal totalRevenue = await _rechargeRepository._DbQueryable
|
||||||
|
.Where(x => x.RechargeType == Domain.Shared.Enums.RechargeTypeEnum.PremiumPackage)
|
||||||
|
.SumAsync(x => x.RechargeAmount);
|
||||||
|
|
||||||
|
// 5. 计算利润率
|
||||||
|
decimal profitRate = totalCost > 0
|
||||||
|
? (totalRevenue / totalCost - 1) * 100
|
||||||
|
: 0;
|
||||||
|
|
||||||
|
// 6. 按200售价计算成本
|
||||||
|
decimal costAt200Price = totalRevenue > 0
|
||||||
|
? (totalCost / totalRevenue) * 200
|
||||||
|
: 0;
|
||||||
|
|
||||||
|
// 7. 格式化输出
|
||||||
|
var today = DateTime.Now;
|
||||||
|
string dayOfWeek = today.ToString("dddd", new System.Globalization.CultureInfo("zh-CN"));
|
||||||
|
string weekDay = dayOfWeek switch
|
||||||
|
{
|
||||||
|
"星期一" => "周1",
|
||||||
|
"星期二" => "周2",
|
||||||
|
"星期三" => "周3",
|
||||||
|
"星期四" => "周4",
|
||||||
|
"星期五" => "周5",
|
||||||
|
"星期六" => "周6",
|
||||||
|
"星期日" => "周日",
|
||||||
|
_ => dayOfWeek
|
||||||
|
};
|
||||||
|
|
||||||
|
var result = $@"{today:M月d日} {weekDay}
|
||||||
|
尊享包已消耗({totalUsedTokens / 100000000m:F2}亿){totalUsedTokens}
|
||||||
|
尊享包剩余库存({totalRemainingTokens / 100000000m:F2}亿){totalRemainingTokens}
|
||||||
|
当前成本:{currentCost:F2}RMB
|
||||||
|
1亿Token成本:{costPerHundredMillion:F2} RMB=1亿 Token
|
||||||
|
总成本:{totalCost:F2} RMB
|
||||||
|
总收益:{totalRevenue:F2}RMB
|
||||||
|
利润率: {profitRate:F1}%
|
||||||
|
按200售价来算,成本在{costAt200Price:F2}";
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
@@ -0,0 +1,68 @@
|
|||||||
|
using Mapster;
|
||||||
|
using Microsoft.Extensions.Caching.Distributed;
|
||||||
|
using Microsoft.Extensions.Configuration;
|
||||||
|
using Volo.Abp.Application.Services;
|
||||||
|
using Volo.Abp.Caching;
|
||||||
|
using Yi.Framework.AiHub.Application.Contracts.Dtos.Announcement;
|
||||||
|
using Yi.Framework.AiHub.Application.Contracts.IServices;
|
||||||
|
using Yi.Framework.AiHub.Domain.Entities;
|
||||||
|
using Yi.Framework.SqlSugarCore.Abstractions;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Services;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 公告服务
|
||||||
|
/// </summary>
|
||||||
|
public class AnnouncementService : ApplicationService, IAnnouncementService
|
||||||
|
{
|
||||||
|
private readonly ISqlSugarRepository<AnnouncementAggregateRoot> _announcementRepository;
|
||||||
|
private readonly IConfiguration _configuration;
|
||||||
|
private readonly IDistributedCache<AnnouncementCacheDto> _announcementCache;
|
||||||
|
private const string AnnouncementCacheKey = "AiHub:Announcement";
|
||||||
|
|
||||||
|
public AnnouncementService(
|
||||||
|
ISqlSugarRepository<AnnouncementAggregateRoot> announcementRepository,
|
||||||
|
IConfiguration configuration,
|
||||||
|
IDistributedCache<AnnouncementCacheDto> announcementCache)
|
||||||
|
{
|
||||||
|
_announcementRepository = announcementRepository;
|
||||||
|
_configuration = configuration;
|
||||||
|
_announcementCache = announcementCache;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取公告信息
|
||||||
|
/// </summary>
|
||||||
|
public async Task<List<AnnouncementLogDto>> GetAsync()
|
||||||
|
{
|
||||||
|
// 使用 GetOrAddAsync 从缓存获取或添加数据,缓存1小时
|
||||||
|
var cacheData = await _announcementCache.GetOrAddAsync(
|
||||||
|
AnnouncementCacheKey,
|
||||||
|
async () => await LoadAnnouncementDataAsync(),
|
||||||
|
() => new DistributedCacheEntryOptions
|
||||||
|
{
|
||||||
|
AbsoluteExpirationRelativeToNow = TimeSpan.FromHours(1)
|
||||||
|
}
|
||||||
|
);
|
||||||
|
|
||||||
|
return cacheData?.Logs ?? new List<AnnouncementLogDto>();
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 从数据库加载公告数据
|
||||||
|
/// </summary>
|
||||||
|
private async Task<AnnouncementCacheDto> LoadAnnouncementDataAsync()
|
||||||
|
{
|
||||||
|
// 查询所有公告日志,按日期降序排列
|
||||||
|
var logs = await _announcementRepository._DbQueryable
|
||||||
|
.OrderByDescending(x => x.StartTime)
|
||||||
|
.ToListAsync();
|
||||||
|
|
||||||
|
// 转换为 DTO
|
||||||
|
var logDtos = logs.Adapt<List<AnnouncementLogDto>>();
|
||||||
|
return new AnnouncementCacheDto
|
||||||
|
{
|
||||||
|
Logs = logDtos
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -73,6 +73,7 @@ public class AiChatService : ApplicationService
|
|||||||
{
|
{
|
||||||
var output = await _aiModelRepository._DbQueryable
|
var output = await _aiModelRepository._DbQueryable
|
||||||
.Where(x => x.ModelType == ModelTypeEnum.Chat)
|
.Where(x => x.ModelType == ModelTypeEnum.Chat)
|
||||||
|
.Where(x => x.ModelApiType == ModelApiTypeEnum.OpenAi)
|
||||||
.OrderByDescending(x => x.OrderNum)
|
.OrderByDescending(x => x.OrderNum)
|
||||||
.Select(x => new ModelGetListOutput
|
.Select(x => new ModelGetListOutput
|
||||||
{
|
{
|
||||||
@@ -87,7 +88,8 @@ public class AiChatService : ApplicationService
|
|||||||
SystemPrompt = null,
|
SystemPrompt = null,
|
||||||
ApiHost = null,
|
ApiHost = null,
|
||||||
ApiKey = null,
|
ApiKey = null,
|
||||||
Remark = x.Description
|
Remark = x.Description,
|
||||||
|
IsPremiumPackage = PremiumPackageConst.ModeIds.Contains(x.ModelId)
|
||||||
}).ToListAsync();
|
}).ToListAsync();
|
||||||
return output;
|
return output;
|
||||||
}
|
}
|
||||||
@@ -134,7 +136,7 @@ public class AiChatService : ApplicationService
|
|||||||
|
|
||||||
//ai网关代理httpcontext
|
//ai网关代理httpcontext
|
||||||
await _aiGateWayManager.CompleteChatStreamForStatisticsAsync(_httpContextAccessor.HttpContext, input,
|
await _aiGateWayManager.CompleteChatStreamForStatisticsAsync(_httpContextAccessor.HttpContext, input,
|
||||||
CurrentUser.Id, sessionId, cancellationToken);
|
CurrentUser.Id, sessionId, null, cancellationToken);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@@ -171,6 +173,6 @@ public class AiChatService : ApplicationService
|
|||||||
|
|
||||||
//ai网关代理httpcontext
|
//ai网关代理httpcontext
|
||||||
await _aiGateWayManager.CompleteChatStreamForStatisticsAsync(_httpContextAccessor.HttpContext, input,
|
await _aiGateWayManager.CompleteChatStreamForStatisticsAsync(_httpContextAccessor.HttpContext, input,
|
||||||
CurrentUser.Id, null, cancellationToken);
|
CurrentUser.Id, null, null, cancellationToken);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -0,0 +1,118 @@
|
|||||||
|
using Mapster;
|
||||||
|
using SqlSugar;
|
||||||
|
using Volo.Abp.Application.Dtos;
|
||||||
|
using Volo.Abp.Application.Services;
|
||||||
|
using Yi.Framework.AiHub.Application.Contracts.Dtos.Model;
|
||||||
|
using Yi.Framework.AiHub.Application.Contracts.IServices;
|
||||||
|
using Yi.Framework.AiHub.Domain.Entities.Model;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Consts;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Extensions;
|
||||||
|
using Yi.Framework.SqlSugarCore.Abstractions;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Services.Chat;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型服务
|
||||||
|
/// </summary>
|
||||||
|
public class ModelService : ApplicationService, IModelService
|
||||||
|
{
|
||||||
|
private readonly ISqlSugarRepository<AiModelEntity, Guid> _modelRepository;
|
||||||
|
|
||||||
|
public ModelService(ISqlSugarRepository<AiModelEntity, Guid> modelRepository)
|
||||||
|
{
|
||||||
|
_modelRepository = modelRepository;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取模型库列表(公开接口,无需登录)
|
||||||
|
/// </summary>
|
||||||
|
public async Task<PagedResultDto<ModelLibraryDto>> GetListAsync(ModelLibraryGetListInput input)
|
||||||
|
{
|
||||||
|
RefAsync<int> total = 0;
|
||||||
|
|
||||||
|
// 查询所有未删除的模型,使用WhereIF动态添加筛选条件
|
||||||
|
var modelIds = (await _modelRepository._DbQueryable
|
||||||
|
.WhereIF(!string.IsNullOrWhiteSpace(input.SearchKey), x =>
|
||||||
|
x.Name.Contains(input.SearchKey) || x.ModelId.Contains(input.SearchKey))
|
||||||
|
.WhereIF(input.ProviderNames is not null, x =>
|
||||||
|
input.ProviderNames.Contains(x.ProviderName))
|
||||||
|
.WhereIF(input.ModelTypes is not null, x =>
|
||||||
|
input.ModelTypes.Contains(x.ModelType))
|
||||||
|
.WhereIF(input.ModelApiTypes is not null, x =>
|
||||||
|
input.ModelApiTypes.Contains(x.ModelApiType))
|
||||||
|
.WhereIF(input.IsPremiumOnly == true, x =>
|
||||||
|
PremiumPackageConst.ModeIds.Contains(x.ModelId))
|
||||||
|
.GroupBy(x => x.ModelId)
|
||||||
|
.Select(x => x.ModelId)
|
||||||
|
.ToPageListAsync(input.SkipCount, input.MaxResultCount, total));
|
||||||
|
|
||||||
|
var entities = await _modelRepository._DbQueryable.Where(x => modelIds.Contains(x.ModelId))
|
||||||
|
.OrderBy(x => x.OrderNum)
|
||||||
|
.OrderBy(x => x.Name).ToListAsync();
|
||||||
|
|
||||||
|
var output= entities.GroupBy(x => x.ModelId).Select(x => new ModelLibraryDto
|
||||||
|
{
|
||||||
|
ModelId = x.First().ModelId,
|
||||||
|
Name = x.First().Name,
|
||||||
|
Description = x.First().Description,
|
||||||
|
ModelType = x.First().ModelType,
|
||||||
|
ModelApiTypes = x.Select(y => new ModelApiTypeOutput { ModelApiType = y.ModelApiType }).ToList(),
|
||||||
|
MultiplierShow = x.First().MultiplierShow,
|
||||||
|
ProviderName = x.First().ProviderName,
|
||||||
|
IconUrl = x.First().IconUrl,
|
||||||
|
IsPremium = PremiumPackageConst.ModeIds.Contains(x.First().ModelId),
|
||||||
|
OrderNum = x.First().OrderNum
|
||||||
|
}).ToList();
|
||||||
|
|
||||||
|
return new PagedResultDto<ModelLibraryDto>(total, output);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取供应商列表(公开接口,无需登录)
|
||||||
|
/// </summary>
|
||||||
|
public async Task<List<string>> GetProviderListAsync()
|
||||||
|
{
|
||||||
|
var providers = await _modelRepository._DbQueryable
|
||||||
|
.Where(x => !x.IsDeleted)
|
||||||
|
.Where(x => !string.IsNullOrEmpty(x.ProviderName))
|
||||||
|
.GroupBy(x => x.ProviderName)
|
||||||
|
.OrderBy(x => x.ProviderName)
|
||||||
|
.Select(x => x.ProviderName)
|
||||||
|
.ToListAsync();
|
||||||
|
|
||||||
|
return providers;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取模型类型选项列表(公开接口,无需登录)
|
||||||
|
/// </summary>
|
||||||
|
public Task<List<ModelTypeOption>> GetModelTypeOptionsAsync()
|
||||||
|
{
|
||||||
|
var options = Enum.GetValues<ModelTypeEnum>()
|
||||||
|
.Select(e => new ModelTypeOption
|
||||||
|
{
|
||||||
|
Label = e.GetDescription(),
|
||||||
|
Value = (int)e
|
||||||
|
})
|
||||||
|
.ToList();
|
||||||
|
|
||||||
|
return Task.FromResult(options);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取API类型选项列表(公开接口,无需登录)
|
||||||
|
/// </summary>
|
||||||
|
public Task<List<ModelApiTypeOption>> GetApiTypeOptionsAsync()
|
||||||
|
{
|
||||||
|
var options = Enum.GetValues<ModelApiTypeEnum>()
|
||||||
|
.Select(e => new ModelApiTypeOption
|
||||||
|
{
|
||||||
|
Label = e.GetDescription(),
|
||||||
|
Value = (int)e
|
||||||
|
})
|
||||||
|
.ToList();
|
||||||
|
|
||||||
|
return Task.FromResult(options);
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,245 @@
|
|||||||
|
using Dm.util;
|
||||||
|
using Microsoft.AspNetCore.Authorization;
|
||||||
|
using Microsoft.AspNetCore.Mvc;
|
||||||
|
using SqlSugar;
|
||||||
|
using Volo.Abp.Application.Dtos;
|
||||||
|
using Volo.Abp.Application.Services;
|
||||||
|
using Volo.Abp.Users;
|
||||||
|
using Yi.Framework.AiHub.Application.Contracts.Dtos.Token;
|
||||||
|
using Yi.Framework.AiHub.Domain.Entities;
|
||||||
|
using Yi.Framework.AiHub.Domain.Entities.OpenApi;
|
||||||
|
using Yi.Framework.AiHub.Domain.Extensions;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Consts;
|
||||||
|
using Yi.Framework.Ddd.Application.Contracts;
|
||||||
|
using Yi.Framework.SqlSugarCore.Abstractions;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Services;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Token服务
|
||||||
|
/// </summary>
|
||||||
|
[Authorize]
|
||||||
|
public class TokenService : ApplicationService
|
||||||
|
{
|
||||||
|
private readonly ISqlSugarRepository<TokenAggregateRoot> _tokenRepository;
|
||||||
|
private readonly ISqlSugarRepository<UsageStatisticsAggregateRoot> _usageStatisticsRepository;
|
||||||
|
|
||||||
|
public TokenService(
|
||||||
|
ISqlSugarRepository<TokenAggregateRoot> tokenRepository,
|
||||||
|
ISqlSugarRepository<UsageStatisticsAggregateRoot> usageStatisticsRepository)
|
||||||
|
{
|
||||||
|
_tokenRepository = tokenRepository;
|
||||||
|
_usageStatisticsRepository = usageStatisticsRepository;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取当前用户的Token列表
|
||||||
|
/// </summary>
|
||||||
|
[HttpGet("token/list")]
|
||||||
|
public async Task<PagedResultDto<TokenGetListOutputDto>> GetListAsync([FromQuery] PagedAllResultRequestDto input)
|
||||||
|
{
|
||||||
|
RefAsync<int> total = 0;
|
||||||
|
var userId = CurrentUser.GetId();
|
||||||
|
|
||||||
|
var tokens = await _tokenRepository._DbQueryable
|
||||||
|
.Where(x => x.UserId == userId)
|
||||||
|
.OrderByDescending(x => x.CreationTime)
|
||||||
|
.ToPageListAsync(input.SkipCount, input.MaxResultCount, total);
|
||||||
|
|
||||||
|
if (!tokens.Any())
|
||||||
|
{
|
||||||
|
return new PagedResultDto<TokenGetListOutputDto>();
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取尊享包模型ID列表
|
||||||
|
var premiumModelIds = PremiumPackageConst.ModeIds;
|
||||||
|
|
||||||
|
// 批量查询所有Token的尊享包已使用额度
|
||||||
|
var tokenIds = tokens.Select(t => t.Id).ToList();
|
||||||
|
var usageStats = await _usageStatisticsRepository._DbQueryable
|
||||||
|
.Where(x => x.UserId == userId && tokenIds.Contains(x.TokenId) && premiumModelIds.Contains(x.ModelId))
|
||||||
|
.GroupBy(x => x.TokenId)
|
||||||
|
.Select(g => new
|
||||||
|
{
|
||||||
|
TokenId = g.TokenId,
|
||||||
|
UsedQuota = SqlFunc.AggregateSum(g.TotalTokenCount)
|
||||||
|
})
|
||||||
|
.ToListAsync();
|
||||||
|
|
||||||
|
var result = tokens.Select(t =>
|
||||||
|
{
|
||||||
|
var usedQuota = usageStats.FirstOrDefault(u => u.TokenId == t.Id)?.UsedQuota ?? 0;
|
||||||
|
return new TokenGetListOutputDto
|
||||||
|
{
|
||||||
|
Id = t.Id,
|
||||||
|
Name = t.Name,
|
||||||
|
ApiKey = t.Token,
|
||||||
|
ExpireTime = t.ExpireTime,
|
||||||
|
PremiumQuotaLimit = t.PremiumQuotaLimit,
|
||||||
|
PremiumUsedQuota = usedQuota,
|
||||||
|
IsDisabled = t.IsDisabled,
|
||||||
|
CreationTime = t.CreationTime
|
||||||
|
};
|
||||||
|
}).ToList();
|
||||||
|
|
||||||
|
return new PagedResultDto<TokenGetListOutputDto>(total, result);
|
||||||
|
}
|
||||||
|
|
||||||
|
[HttpGet("token/select-list")]
|
||||||
|
public async Task<List<TokenSelectListOutputDto>> GetSelectListAsync()
|
||||||
|
{
|
||||||
|
var userId = CurrentUser.GetId();
|
||||||
|
var tokens = await _tokenRepository._DbQueryable
|
||||||
|
.Where(x => x.UserId == userId)
|
||||||
|
.OrderBy(x => x.IsDisabled)
|
||||||
|
.OrderByDescending(x => x.CreationTime)
|
||||||
|
.Select(x => new TokenSelectListOutputDto
|
||||||
|
{
|
||||||
|
TokenId = x.Id,
|
||||||
|
Name = x.Name,
|
||||||
|
IsDisabled = x.IsDisabled
|
||||||
|
}).ToListAsync();
|
||||||
|
|
||||||
|
tokens.Insert(0,new TokenSelectListOutputDto
|
||||||
|
{
|
||||||
|
TokenId = Guid.Empty,
|
||||||
|
Name = "默认",
|
||||||
|
IsDisabled = false
|
||||||
|
});
|
||||||
|
return tokens;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 创建Token
|
||||||
|
/// </summary>
|
||||||
|
[HttpPost("token")]
|
||||||
|
public async Task<TokenGetListOutputDto> CreateAsync([FromBody] TokenCreateInput input)
|
||||||
|
{
|
||||||
|
var userId = CurrentUser.GetId();
|
||||||
|
|
||||||
|
// 检查用户是否为VIP
|
||||||
|
if (!CurrentUser.IsAiVip())
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("充值成为Vip,畅享第三方token服务");
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查名称是否重复
|
||||||
|
var exists = await _tokenRepository._DbQueryable
|
||||||
|
.AnyAsync(x => x.UserId == userId && x.Name == input.Name);
|
||||||
|
if (exists)
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException($"名称【{input.Name}】已存在,请使用其他名称");
|
||||||
|
}
|
||||||
|
|
||||||
|
var token = new TokenAggregateRoot(userId, input.Name)
|
||||||
|
{
|
||||||
|
ExpireTime = input.ExpireTime,
|
||||||
|
PremiumQuotaLimit = input.PremiumQuotaLimit
|
||||||
|
};
|
||||||
|
|
||||||
|
await _tokenRepository.InsertAsync(token);
|
||||||
|
|
||||||
|
return new TokenGetListOutputDto
|
||||||
|
{
|
||||||
|
Id = token.Id,
|
||||||
|
Name = token.Name,
|
||||||
|
ApiKey = token.Token,
|
||||||
|
ExpireTime = token.ExpireTime,
|
||||||
|
PremiumQuotaLimit = token.PremiumQuotaLimit,
|
||||||
|
PremiumUsedQuota = 0,
|
||||||
|
IsDisabled = token.IsDisabled,
|
||||||
|
CreationTime = token.CreationTime
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 编辑Token
|
||||||
|
/// </summary>
|
||||||
|
[HttpPut("token")]
|
||||||
|
public async Task UpdateAsync([FromBody] TokenUpdateInput input)
|
||||||
|
{
|
||||||
|
var userId = CurrentUser.GetId();
|
||||||
|
|
||||||
|
var token = await _tokenRepository._DbQueryable
|
||||||
|
.FirstAsync(x => x.Id == input.Id && x.UserId == userId);
|
||||||
|
|
||||||
|
if (token is null)
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("Token不存在或无权限操作");
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查名称是否重复(排除自己)
|
||||||
|
var exists = await _tokenRepository._DbQueryable
|
||||||
|
.AnyAsync(x => x.UserId == userId && x.Name == input.Name && x.Id != input.Id);
|
||||||
|
if (exists)
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException($"名称【{input.Name}】已存在,请使用其他名称");
|
||||||
|
}
|
||||||
|
|
||||||
|
token.Name = input.Name;
|
||||||
|
token.ExpireTime = input.ExpireTime;
|
||||||
|
token.PremiumQuotaLimit = input.PremiumQuotaLimit;
|
||||||
|
|
||||||
|
await _tokenRepository.UpdateAsync(token);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 删除Token
|
||||||
|
/// </summary>
|
||||||
|
[HttpDelete("token/{id}")]
|
||||||
|
public async Task DeleteAsync(Guid id)
|
||||||
|
{
|
||||||
|
var userId = CurrentUser.GetId();
|
||||||
|
|
||||||
|
var token = await _tokenRepository._DbQueryable
|
||||||
|
.FirstAsync(x => x.Id == id && x.UserId == userId);
|
||||||
|
|
||||||
|
if (token is null)
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("Token不存在或无权限操作");
|
||||||
|
}
|
||||||
|
|
||||||
|
await _tokenRepository.DeleteAsync(token);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 启用Token
|
||||||
|
/// </summary>
|
||||||
|
[HttpPost("token/{id}/enable")]
|
||||||
|
public async Task EnableAsync(Guid id)
|
||||||
|
{
|
||||||
|
var userId = CurrentUser.GetId();
|
||||||
|
|
||||||
|
var token = await _tokenRepository._DbQueryable
|
||||||
|
.FirstAsync(x => x.Id == id && x.UserId == userId);
|
||||||
|
|
||||||
|
if (token is null)
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("Token不存在或无权限操作");
|
||||||
|
}
|
||||||
|
|
||||||
|
token.Enable();
|
||||||
|
await _tokenRepository.UpdateAsync(token);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 禁用Token
|
||||||
|
/// </summary>
|
||||||
|
[HttpPost("token/{id}/disable")]
|
||||||
|
public async Task DisableAsync(Guid id)
|
||||||
|
{
|
||||||
|
var userId = CurrentUser.GetId();
|
||||||
|
|
||||||
|
var token = await _tokenRepository._DbQueryable
|
||||||
|
.FirstAsync(x => x.Id == id && x.UserId == userId);
|
||||||
|
|
||||||
|
if (token is null)
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("Token不存在或无权限操作");
|
||||||
|
}
|
||||||
|
|
||||||
|
token.Disable();
|
||||||
|
await _tokenRepository.UpdateAsync(token);
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -76,12 +76,12 @@ public class FileMasterService : ApplicationService
|
|||||||
if (input.Stream == true)
|
if (input.Stream == true)
|
||||||
{
|
{
|
||||||
await _aiGateWayManager.CompleteChatStreamForStatisticsAsync(_httpContextAccessor.HttpContext, input,
|
await _aiGateWayManager.CompleteChatStreamForStatisticsAsync(_httpContextAccessor.HttpContext, input,
|
||||||
userId, null, cancellationToken);
|
userId, null, null, cancellationToken);
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
await _aiGateWayManager.CompleteChatForStatisticsAsync(_httpContextAccessor.HttpContext, input, userId,
|
await _aiGateWayManager.CompleteChatForStatisticsAsync(_httpContextAccessor.HttpContext, input, userId,
|
||||||
null,
|
null, null,
|
||||||
cancellationToken);
|
cancellationToken);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -12,6 +12,7 @@ using Yi.Framework.AiHub.Domain.Shared.Dtos.Anthropic;
|
|||||||
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi;
|
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi;
|
||||||
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi.Embeddings;
|
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi.Embeddings;
|
||||||
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi.Images;
|
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi.Images;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi.Responses;
|
||||||
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
||||||
using Yi.Framework.Rbac.Application.Contracts.IServices;
|
using Yi.Framework.Rbac.Application.Contracts.IServices;
|
||||||
using Yi.Framework.SqlSugarCore.Abstractions;
|
using Yi.Framework.SqlSugarCore.Abstractions;
|
||||||
@@ -55,7 +56,9 @@ public class OpenApiService : ApplicationService
|
|||||||
{
|
{
|
||||||
//前面都是校验,后面才是真正的调用
|
//前面都是校验,后面才是真正的调用
|
||||||
var httpContext = this._httpContextAccessor.HttpContext;
|
var httpContext = this._httpContextAccessor.HttpContext;
|
||||||
var userId = await _tokenManager.GetUserIdAsync(GetTokenByHttpContext(httpContext));
|
var tokenValidation = await _tokenManager.ValidateTokenAsync(GetTokenByHttpContext(httpContext), input.Model);
|
||||||
|
var userId = tokenValidation.UserId;
|
||||||
|
var tokenId = tokenValidation.TokenId;
|
||||||
await _aiBlacklistManager.VerifiyAiBlacklist(userId);
|
await _aiBlacklistManager.VerifiyAiBlacklist(userId);
|
||||||
|
|
||||||
//如果是尊享包服务,需要校验是是否尊享包足够
|
//如果是尊享包服务,需要校验是是否尊享包足够
|
||||||
@@ -73,16 +76,17 @@ public class OpenApiService : ApplicationService
|
|||||||
if (input.Stream == true)
|
if (input.Stream == true)
|
||||||
{
|
{
|
||||||
await _aiGateWayManager.CompleteChatStreamForStatisticsAsync(_httpContextAccessor.HttpContext, input,
|
await _aiGateWayManager.CompleteChatStreamForStatisticsAsync(_httpContextAccessor.HttpContext, input,
|
||||||
userId, null, cancellationToken);
|
userId, null, tokenId, cancellationToken);
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
await _aiGateWayManager.CompleteChatForStatisticsAsync(_httpContextAccessor.HttpContext, input, userId,
|
await _aiGateWayManager.CompleteChatForStatisticsAsync(_httpContextAccessor.HttpContext, input, userId,
|
||||||
null,
|
null, tokenId,
|
||||||
cancellationToken);
|
cancellationToken);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 图片生成
|
/// 图片生成
|
||||||
/// </summary>
|
/// </summary>
|
||||||
@@ -93,11 +97,14 @@ public class OpenApiService : ApplicationService
|
|||||||
{
|
{
|
||||||
var httpContext = this._httpContextAccessor.HttpContext;
|
var httpContext = this._httpContextAccessor.HttpContext;
|
||||||
Intercept(httpContext);
|
Intercept(httpContext);
|
||||||
var userId = await _tokenManager.GetUserIdAsync(GetTokenByHttpContext(httpContext));
|
var tokenValidation = await _tokenManager.ValidateTokenAsync(GetTokenByHttpContext(httpContext), input.Model);
|
||||||
|
var userId = tokenValidation.UserId;
|
||||||
|
var tokenId = tokenValidation.TokenId;
|
||||||
await _aiBlacklistManager.VerifiyAiBlacklist(userId);
|
await _aiBlacklistManager.VerifiyAiBlacklist(userId);
|
||||||
await _aiGateWayManager.CreateImageForStatisticsAsync(httpContext, userId, null, input);
|
await _aiGateWayManager.CreateImageForStatisticsAsync(httpContext, userId, null, input, tokenId);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 向量生成
|
/// 向量生成
|
||||||
/// </summary>
|
/// </summary>
|
||||||
@@ -108,9 +115,11 @@ public class OpenApiService : ApplicationService
|
|||||||
{
|
{
|
||||||
var httpContext = this._httpContextAccessor.HttpContext;
|
var httpContext = this._httpContextAccessor.HttpContext;
|
||||||
Intercept(httpContext);
|
Intercept(httpContext);
|
||||||
var userId = await _tokenManager.GetUserIdAsync(GetTokenByHttpContext(httpContext));
|
var tokenValidation = await _tokenManager.ValidateTokenAsync(GetTokenByHttpContext(httpContext), input.Model);
|
||||||
|
var userId = tokenValidation.UserId;
|
||||||
|
var tokenId = tokenValidation.TokenId;
|
||||||
await _aiBlacklistManager.VerifiyAiBlacklist(userId);
|
await _aiBlacklistManager.VerifiyAiBlacklist(userId);
|
||||||
await _aiGateWayManager.EmbeddingForStatisticsAsync(httpContext, userId, null, input);
|
await _aiGateWayManager.EmbeddingForStatisticsAsync(httpContext, userId, null, input, tokenId);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@@ -151,7 +160,9 @@ public class OpenApiService : ApplicationService
|
|||||||
{
|
{
|
||||||
//前面都是校验,后面才是真正的调用
|
//前面都是校验,后面才是真正的调用
|
||||||
var httpContext = this._httpContextAccessor.HttpContext;
|
var httpContext = this._httpContextAccessor.HttpContext;
|
||||||
var userId = await _tokenManager.GetUserIdAsync(GetTokenByHttpContext(httpContext));
|
var tokenValidation = await _tokenManager.ValidateTokenAsync(GetTokenByHttpContext(httpContext), input.Model);
|
||||||
|
var userId = tokenValidation.UserId;
|
||||||
|
var tokenId = tokenValidation.TokenId;
|
||||||
await _aiBlacklistManager.VerifiyAiBlacklist(userId);
|
await _aiBlacklistManager.VerifiyAiBlacklist(userId);
|
||||||
|
|
||||||
// 验证用户是否为VIP
|
// 验证用户是否为VIP
|
||||||
@@ -177,13 +188,67 @@ public class OpenApiService : ApplicationService
|
|||||||
//ai网关代理httpcontext
|
//ai网关代理httpcontext
|
||||||
if (input.Stream)
|
if (input.Stream)
|
||||||
{
|
{
|
||||||
await _aiGateWayManager.AnthropicCompleteChatStreamForStatisticsAsync(_httpContextAccessor.HttpContext, input,
|
await _aiGateWayManager.AnthropicCompleteChatStreamForStatisticsAsync(_httpContextAccessor.HttpContext,
|
||||||
userId, null, cancellationToken);
|
input,
|
||||||
|
userId, null, tokenId, cancellationToken);
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
await _aiGateWayManager.AnthropicCompleteChatForStatisticsAsync(_httpContextAccessor.HttpContext, input, userId,
|
await _aiGateWayManager.AnthropicCompleteChatForStatisticsAsync(_httpContextAccessor.HttpContext, input,
|
||||||
null,
|
userId,
|
||||||
|
null, tokenId,
|
||||||
|
cancellationToken);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 响应-Openai新规范 (尊享服务专用)
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="input"></param>
|
||||||
|
/// <param name="cancellationToken"></param>
|
||||||
|
[HttpPost("openApi/v1/responses")]
|
||||||
|
public async Task ResponsesAsync([FromBody] OpenAiResponsesInput input, CancellationToken cancellationToken)
|
||||||
|
{
|
||||||
|
//前面都是校验,后面才是真正的调用
|
||||||
|
var httpContext = this._httpContextAccessor.HttpContext;
|
||||||
|
var tokenValidation = await _tokenManager.ValidateTokenAsync(GetTokenByHttpContext(httpContext), input.Model);
|
||||||
|
var userId = tokenValidation.UserId;
|
||||||
|
var tokenId = tokenValidation.TokenId;
|
||||||
|
await _aiBlacklistManager.VerifiyAiBlacklist(userId);
|
||||||
|
|
||||||
|
// 验证用户是否为VIP
|
||||||
|
var userInfo = await _accountService.GetAsync(null, null, userId);
|
||||||
|
if (userInfo == null)
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("用户信息不存在");
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查是否为VIP(使用RoleCodes判断)
|
||||||
|
if (!userInfo.RoleCodes.Contains(AiHubConst.VipRole) && userInfo.User.UserName != "cc")
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("该接口为尊享服务专用,需要VIP权限才能使用");
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查尊享token包用量
|
||||||
|
var availableTokens = await _premiumPackageManager.GetAvailableTokensAsync(userId);
|
||||||
|
if (availableTokens <= 0)
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("尊享token包用量不足,请先购买尊享token包");
|
||||||
|
}
|
||||||
|
|
||||||
|
//ai网关代理httpcontext
|
||||||
|
if (input.Stream == true)
|
||||||
|
{
|
||||||
|
await _aiGateWayManager.OpenAiResponsesStreamForStatisticsAsync(_httpContextAccessor.HttpContext,
|
||||||
|
input,
|
||||||
|
userId, null, tokenId, cancellationToken);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
await _aiGateWayManager.OpenAiResponsesAsyncForStatisticsAsync(_httpContextAccessor.HttpContext, input,
|
||||||
|
userId,
|
||||||
|
null, tokenId,
|
||||||
cancellationToken);
|
cancellationToken);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -202,7 +267,8 @@ public class OpenApiService : ApplicationService
|
|||||||
|
|
||||||
// 再检查 Authorization 头
|
// 再检查 Authorization 头
|
||||||
string authHeader = httpContext.Request.Headers["Authorization"];
|
string authHeader = httpContext.Request.Headers["Authorization"];
|
||||||
if (!string.IsNullOrWhiteSpace(authHeader) && authHeader.StartsWith("Bearer ", StringComparison.OrdinalIgnoreCase))
|
if (!string.IsNullOrWhiteSpace(authHeader) &&
|
||||||
|
authHeader.StartsWith("Bearer ", StringComparison.OrdinalIgnoreCase))
|
||||||
{
|
{
|
||||||
return authHeader.Substring("Bearer ".Length).Trim();
|
return authHeader.Substring("Bearer ".Length).Trim();
|
||||||
}
|
}
|
||||||
@@ -219,5 +285,4 @@ public class OpenApiService : ApplicationService
|
|||||||
}
|
}
|
||||||
|
|
||||||
#endregion
|
#endregion
|
||||||
|
|
||||||
}
|
}
|
||||||
@@ -1,55 +0,0 @@
|
|||||||
using Microsoft.AspNetCore.Authorization;
|
|
||||||
using Volo.Abp.Application.Services;
|
|
||||||
using Volo.Abp.Users;
|
|
||||||
using Yi.Framework.AiHub.Application.Contracts.Dtos.Token;
|
|
||||||
using Yi.Framework.AiHub.Domain.Entities.OpenApi;
|
|
||||||
using Yi.Framework.AiHub.Domain.Extensions;
|
|
||||||
using Yi.Framework.AiHub.Domain.Managers;
|
|
||||||
using Yi.Framework.SqlSugarCore.Abstractions;
|
|
||||||
|
|
||||||
namespace Yi.Framework.AiHub.Application.Services;
|
|
||||||
|
|
||||||
public class TokenService : ApplicationService
|
|
||||||
{
|
|
||||||
private readonly ISqlSugarRepository<TokenAggregateRoot> _tokenRepository;
|
|
||||||
private readonly TokenManager _tokenManager;
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 构造函数
|
|
||||||
/// </summary>
|
|
||||||
/// <param name="tokenRepository"></param>
|
|
||||||
/// <param name="tokenManager"></param>
|
|
||||||
public TokenService(ISqlSugarRepository<TokenAggregateRoot> tokenRepository, TokenManager tokenManager)
|
|
||||||
{
|
|
||||||
_tokenRepository = tokenRepository;
|
|
||||||
_tokenManager = tokenManager;
|
|
||||||
}
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 获取token
|
|
||||||
/// </summary>
|
|
||||||
/// <returns></returns>
|
|
||||||
[Authorize]
|
|
||||||
public async Task<TokenOutput> GetAsync()
|
|
||||||
{
|
|
||||||
return new TokenOutput
|
|
||||||
{
|
|
||||||
ApiKey = await _tokenManager.GetAsync(CurrentUser.GetId())
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 创建token
|
|
||||||
/// </summary>
|
|
||||||
/// <exception cref="UserFriendlyException"></exception>
|
|
||||||
[Authorize]
|
|
||||||
public async Task CreateAsync()
|
|
||||||
{
|
|
||||||
if (!CurrentUser.IsAiVip())
|
|
||||||
{
|
|
||||||
throw new UserFriendlyException("充值成为Vip,畅享第三方token服务");
|
|
||||||
}
|
|
||||||
|
|
||||||
await _tokenManager.CreateAsync(CurrentUser.GetId());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,12 +1,18 @@
|
|||||||
|
using Mapster;
|
||||||
using Microsoft.AspNetCore.Authorization;
|
using Microsoft.AspNetCore.Authorization;
|
||||||
|
using Microsoft.AspNetCore.Mvc;
|
||||||
using SqlSugar;
|
using SqlSugar;
|
||||||
|
using Volo.Abp.Application.Dtos;
|
||||||
using Volo.Abp.Application.Services;
|
using Volo.Abp.Application.Services;
|
||||||
using Volo.Abp.Users;
|
using Volo.Abp.Users;
|
||||||
using Yi.Framework.AiHub.Application.Contracts.Dtos.UsageStatistics;
|
using Yi.Framework.AiHub.Application.Contracts.Dtos.UsageStatistics;
|
||||||
using Yi.Framework.AiHub.Application.Contracts.IServices;
|
using Yi.Framework.AiHub.Application.Contracts.IServices;
|
||||||
using Yi.Framework.AiHub.Domain.Entities;
|
using Yi.Framework.AiHub.Domain.Entities;
|
||||||
using Yi.Framework.AiHub.Domain.Entities.Chat;
|
using Yi.Framework.AiHub.Domain.Entities.Chat;
|
||||||
|
using Yi.Framework.AiHub.Domain.Entities.OpenApi;
|
||||||
using Yi.Framework.AiHub.Domain.Extensions;
|
using Yi.Framework.AiHub.Domain.Extensions;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Consts;
|
||||||
|
using Yi.Framework.Ddd.Application.Contracts;
|
||||||
using Yi.Framework.SqlSugarCore.Abstractions;
|
using Yi.Framework.SqlSugarCore.Abstractions;
|
||||||
|
|
||||||
namespace Yi.Framework.AiHub.Application.Services;
|
namespace Yi.Framework.AiHub.Application.Services;
|
||||||
@@ -20,22 +26,25 @@ public class UsageStatisticsService : ApplicationService, IUsageStatisticsServic
|
|||||||
private readonly ISqlSugarRepository<MessageAggregateRoot> _messageRepository;
|
private readonly ISqlSugarRepository<MessageAggregateRoot> _messageRepository;
|
||||||
private readonly ISqlSugarRepository<UsageStatisticsAggregateRoot> _usageStatisticsRepository;
|
private readonly ISqlSugarRepository<UsageStatisticsAggregateRoot> _usageStatisticsRepository;
|
||||||
private readonly ISqlSugarRepository<PremiumPackageAggregateRoot> _premiumPackageRepository;
|
private readonly ISqlSugarRepository<PremiumPackageAggregateRoot> _premiumPackageRepository;
|
||||||
|
private readonly ISqlSugarRepository<TokenAggregateRoot> _tokenRepository;
|
||||||
|
|
||||||
public UsageStatisticsService(
|
public UsageStatisticsService(
|
||||||
ISqlSugarRepository<MessageAggregateRoot> messageRepository,
|
ISqlSugarRepository<MessageAggregateRoot> messageRepository,
|
||||||
ISqlSugarRepository<UsageStatisticsAggregateRoot> usageStatisticsRepository,
|
ISqlSugarRepository<UsageStatisticsAggregateRoot> usageStatisticsRepository,
|
||||||
ISqlSugarRepository<PremiumPackageAggregateRoot> premiumPackageRepository)
|
ISqlSugarRepository<PremiumPackageAggregateRoot> premiumPackageRepository,
|
||||||
|
ISqlSugarRepository<TokenAggregateRoot> tokenRepository)
|
||||||
{
|
{
|
||||||
_messageRepository = messageRepository;
|
_messageRepository = messageRepository;
|
||||||
_usageStatisticsRepository = usageStatisticsRepository;
|
_usageStatisticsRepository = usageStatisticsRepository;
|
||||||
_premiumPackageRepository = premiumPackageRepository;
|
_premiumPackageRepository = premiumPackageRepository;
|
||||||
|
_tokenRepository = tokenRepository;
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 获取当前用户近7天的Token消耗统计
|
/// 获取当前用户近7天的Token消耗统计
|
||||||
/// </summary>
|
/// </summary>
|
||||||
/// <returns>每日Token使用量列表</returns>
|
/// <returns>每日Token使用量列表</returns>
|
||||||
public async Task<List<DailyTokenUsageDto>> GetLast7DaysTokenUsageAsync()
|
public async Task<List<DailyTokenUsageDto>> GetLast7DaysTokenUsageAsync([FromQuery]UsageStatisticsGetInput input)
|
||||||
{
|
{
|
||||||
var userId = CurrentUser.GetId();
|
var userId = CurrentUser.GetId();
|
||||||
var endDate = DateTime.Today;
|
var endDate = DateTime.Today;
|
||||||
@@ -46,6 +55,7 @@ public class UsageStatisticsService : ApplicationService, IUsageStatisticsServic
|
|||||||
.Where(x => x.UserId == userId)
|
.Where(x => x.UserId == userId)
|
||||||
.Where(x => x.Role == "assistant" || x.Role == "system")
|
.Where(x => x.Role == "assistant" || x.Role == "system")
|
||||||
.Where(x => x.CreationTime >= startDate && x.CreationTime < endDate.AddDays(1))
|
.Where(x => x.CreationTime >= startDate && x.CreationTime < endDate.AddDays(1))
|
||||||
|
.WhereIF(input.TokenId.HasValue,x => x.TokenId == input.TokenId)
|
||||||
.GroupBy(x => x.CreationTime.Date)
|
.GroupBy(x => x.CreationTime.Date)
|
||||||
.Select(g => new
|
.Select(g => new
|
||||||
{
|
{
|
||||||
@@ -75,17 +85,19 @@ public class UsageStatisticsService : ApplicationService, IUsageStatisticsServic
|
|||||||
/// 获取当前用户各个模型的Token消耗量及占比
|
/// 获取当前用户各个模型的Token消耗量及占比
|
||||||
/// </summary>
|
/// </summary>
|
||||||
/// <returns>模型Token使用量列表</returns>
|
/// <returns>模型Token使用量列表</returns>
|
||||||
public async Task<List<ModelTokenUsageDto>> GetModelTokenUsageAsync()
|
public async Task<List<ModelTokenUsageDto>> GetModelTokenUsageAsync([FromQuery]UsageStatisticsGetInput input)
|
||||||
{
|
{
|
||||||
var userId = CurrentUser.GetId();
|
var userId = CurrentUser.GetId();
|
||||||
|
|
||||||
// 从UsageStatistics表获取各模型的token消耗统计
|
// 从UsageStatistics表获取各模型的token消耗统计(按ModelId聚合,因为同一模型可能有多个TokenId的记录)
|
||||||
var modelUsages = await _usageStatisticsRepository._DbQueryable
|
var modelUsages = await _usageStatisticsRepository._DbQueryable
|
||||||
.Where(x => x.UserId == userId)
|
.Where(x => x.UserId == userId)
|
||||||
|
.WhereIF(input.TokenId.HasValue,x => x.TokenId == input.TokenId)
|
||||||
|
.GroupBy(x => x.ModelId)
|
||||||
.Select(x => new
|
.Select(x => new
|
||||||
{
|
{
|
||||||
x.ModelId,
|
ModelId = x.ModelId,
|
||||||
x.TotalTokenCount
|
TotalTokenCount = SqlFunc.AggregateSum(x.TotalTokenCount)
|
||||||
})
|
})
|
||||||
.ToListAsync();
|
.ToListAsync();
|
||||||
|
|
||||||
@@ -125,9 +137,9 @@ public class UsageStatisticsService : ApplicationService, IUsageStatisticsServic
|
|||||||
|
|
||||||
if (premiumPackages.Any())
|
if (premiumPackages.Any())
|
||||||
{
|
{
|
||||||
// 过滤掉已过期的包
|
// 过滤掉已过期、禁用的包,不过滤用量负数的包
|
||||||
var validPackages = premiumPackages
|
var validPackages = premiumPackages
|
||||||
.Where(p => p.IsAvailable())
|
.Where(p => p.IsAvailable(false))
|
||||||
.ToList();
|
.ToList();
|
||||||
|
|
||||||
result.PremiumTotalTokens = validPackages.Sum(x => x.TotalTokens);
|
result.PremiumTotalTokens = validPackages.Sum(x => x.TotalTokens);
|
||||||
@@ -137,4 +149,77 @@ public class UsageStatisticsService : ApplicationService, IUsageStatisticsServic
|
|||||||
|
|
||||||
return result;
|
return result;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取当前用户尊享服务token用量统计列表
|
||||||
|
/// </summary>
|
||||||
|
/// <returns></returns>
|
||||||
|
[HttpGet("usage-statistics/premium-token-usage/list")]
|
||||||
|
public async Task<PagedResultDto<PremiumTokenUsageGetListOutput>> GetPremiumTokenUsageListAsync(
|
||||||
|
PremiumTokenUsageGetListInput input)
|
||||||
|
{
|
||||||
|
var userId = CurrentUser.GetId();
|
||||||
|
RefAsync<int> total = 0;
|
||||||
|
// 获取尊享包Token信息
|
||||||
|
var entities = await _premiumPackageRepository._DbQueryable
|
||||||
|
.Where(x => x.UserId == userId)
|
||||||
|
.WhereIF(input.IsFree == false, x => x.PurchaseAmount > 0)
|
||||||
|
.WhereIF(input.IsFree == true, x => x.PurchaseAmount == 0)
|
||||||
|
.WhereIF(input.StartTime is not null && input.EndTime is not null,
|
||||||
|
x => x.CreationTime >= input.StartTime && x.CreationTime <= input.EndTime)
|
||||||
|
.OrderByDescending(x => x.CreationTime)
|
||||||
|
.ToPageListAsync(input.SkipCount, input.MaxResultCount, total);
|
||||||
|
return new PagedResultDto<PremiumTokenUsageGetListOutput>(total,
|
||||||
|
entities.Adapt<List<PremiumTokenUsageGetListOutput>>());
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取当前用户尊享包不同Token用量占比(饼图)
|
||||||
|
/// </summary>
|
||||||
|
/// <returns>各Token的尊享模型用量及占比</returns>
|
||||||
|
[HttpGet("usage-statistics/premium-token-usage/by-token")]
|
||||||
|
public async Task<List<TokenPremiumUsageDto>> GetPremiumTokenUsageByTokenAsync()
|
||||||
|
{
|
||||||
|
var userId = CurrentUser.GetId();
|
||||||
|
var premiumModelIds = PremiumPackageConst.ModeIds;
|
||||||
|
|
||||||
|
// 从UsageStatistics表获取尊享模型的token消耗统计(按TokenId聚合)
|
||||||
|
var tokenUsages = await _usageStatisticsRepository._DbQueryable
|
||||||
|
.Where(x => x.UserId == userId && premiumModelIds.Contains(x.ModelId))
|
||||||
|
.GroupBy(x => x.TokenId)
|
||||||
|
.Select(x => new
|
||||||
|
{
|
||||||
|
TokenId = x.TokenId,
|
||||||
|
TotalTokenCount = SqlFunc.AggregateSum(x.TotalTokenCount)
|
||||||
|
})
|
||||||
|
.ToListAsync();
|
||||||
|
|
||||||
|
if (!tokenUsages.Any())
|
||||||
|
{
|
||||||
|
return new List<TokenPremiumUsageDto>();
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取用户的所有Token信息用于名称映射
|
||||||
|
var tokenIds = tokenUsages.Select(x => x.TokenId).ToList();
|
||||||
|
var tokens = await _tokenRepository._DbQueryable
|
||||||
|
.Where(x => x.UserId == userId && tokenIds.Contains(x.Id))
|
||||||
|
.Select(x => new { x.Id, x.Name })
|
||||||
|
.ToListAsync();
|
||||||
|
|
||||||
|
var tokenNameDict = tokens.ToDictionary(x => x.Id, x => x.Name);
|
||||||
|
|
||||||
|
// 计算总token数
|
||||||
|
var totalTokens = tokenUsages.Sum(x => x.TotalTokenCount);
|
||||||
|
|
||||||
|
// 计算各Token占比
|
||||||
|
var result = tokenUsages.Select(x => new TokenPremiumUsageDto
|
||||||
|
{
|
||||||
|
TokenId = x.TokenId,
|
||||||
|
TokenName = x.TokenId == Guid.Empty ? "默认" : (tokenNameDict.TryGetValue(x.TokenId, out var name) ? name : "其他"),
|
||||||
|
Tokens = x.TotalTokenCount,
|
||||||
|
Percentage = totalTokens > 0 ? Math.Round((decimal)x.TotalTokenCount / totalTokens * 100, 2) : 0
|
||||||
|
}).OrderByDescending(x => x.Tokens).ToList();
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
@@ -4,5 +4,13 @@ namespace Yi.Framework.AiHub.Domain.Shared.Consts;
|
|||||||
|
|
||||||
public class PremiumPackageConst
|
public class PremiumPackageConst
|
||||||
{
|
{
|
||||||
public static List<string> ModeIds = ["claude-sonnet-4-5-20250929"];
|
public static List<string> ModeIds =
|
||||||
|
[
|
||||||
|
"claude-sonnet-4-5-20250929",
|
||||||
|
"claude-haiku-4-5-20251001",
|
||||||
|
"claude-opus-4-5-20251101",
|
||||||
|
"gemini-3-pro-preview",
|
||||||
|
"gpt-5.1-codex-max",
|
||||||
|
"gpt-5.2"
|
||||||
|
];
|
||||||
}
|
}
|
||||||
@@ -56,4 +56,9 @@ public class AiModelDescribe
|
|||||||
/// 模型额外信息
|
/// 模型额外信息
|
||||||
/// </summary>
|
/// </summary>
|
||||||
public string? ModelExtraInfo { get; set; }
|
public string? ModelExtraInfo { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型倍率
|
||||||
|
/// </summary>
|
||||||
|
public decimal Multiplier { get; set; }
|
||||||
}
|
}
|
||||||
@@ -34,7 +34,7 @@ public class AnthropicStreamDto
|
|||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
public void SupplementalMultiplier(double multiplier)
|
public void SupplementalMultiplier(decimal multiplier)
|
||||||
{
|
{
|
||||||
if (this.Usage is not null)
|
if (this.Usage is not null)
|
||||||
{
|
{
|
||||||
@@ -130,7 +130,7 @@ public class AnthropicChatCompletionDto
|
|||||||
CompletionTokensDetails = null
|
CompletionTokensDetails = null
|
||||||
};
|
};
|
||||||
|
|
||||||
public void SupplementalMultiplier(double multiplier)
|
public void SupplementalMultiplier(decimal multiplier)
|
||||||
{
|
{
|
||||||
if (this.Usage is not null)
|
if (this.Usage is not null)
|
||||||
{
|
{
|
||||||
|
|||||||
@@ -109,8 +109,15 @@ public sealed class AnthropicInput
|
|||||||
public class AnthropicThinkingInput
|
public class AnthropicThinkingInput
|
||||||
{
|
{
|
||||||
[JsonPropertyName("type")] public string? Type { get; set; }
|
[JsonPropertyName("type")] public string? Type { get; set; }
|
||||||
|
|
||||||
[JsonPropertyName("budget_tokens")] public int? BudgetTokens { get; set; }
|
[JsonPropertyName("budget_tokens")] public int? BudgetTokens { get; set; }
|
||||||
|
|
||||||
|
[JsonPropertyName("signature")] public string? Signature { get; set; }
|
||||||
|
|
||||||
|
[JsonPropertyName("thinking")] public string? Thinking { get; set; }
|
||||||
|
|
||||||
|
[JsonPropertyName("data")] public string? Data { get; set; }
|
||||||
|
|
||||||
|
[JsonPropertyName("text")] public string? Text { get; set; }
|
||||||
}
|
}
|
||||||
|
|
||||||
public class AnthropicTooChoiceInput
|
public class AnthropicTooChoiceInput
|
||||||
|
|||||||
@@ -20,6 +20,8 @@ public class AnthropicMessageContent
|
|||||||
|
|
||||||
[JsonPropertyName("thinking")] public string? Thinking { get; set; }
|
[JsonPropertyName("thinking")] public string? Thinking { get; set; }
|
||||||
|
|
||||||
|
[JsonPropertyName("signature")] public string? Signature { get; set; }
|
||||||
|
|
||||||
[JsonPropertyName("input")] public object? Input { get; set; }
|
[JsonPropertyName("input")] public object? Input { get; set; }
|
||||||
|
|
||||||
[JsonPropertyName("content")]
|
[JsonPropertyName("content")]
|
||||||
|
|||||||
@@ -22,11 +22,15 @@ public class AnthropicMessageInput
|
|||||||
throw new ValidationException("Messages 中 Content 和 Contents 字段不能同时有值");
|
throw new ValidationException("Messages 中 Content 和 Contents 字段不能同时有值");
|
||||||
}
|
}
|
||||||
|
|
||||||
if (Content is not null)
|
if (!string.IsNullOrEmpty(Content))
|
||||||
{
|
{
|
||||||
return Content;
|
return Content;
|
||||||
}
|
}
|
||||||
|
// 如果 Contents 为空或 null,返回空字符串而不是 null
|
||||||
|
if (Contents == null || Contents.Count == 0)
|
||||||
|
{
|
||||||
|
return "_"; // 兼容客户端空值问题
|
||||||
|
}
|
||||||
return Contents!;
|
return Contents!;
|
||||||
}
|
}
|
||||||
set
|
set
|
||||||
|
|||||||
@@ -1,648 +0,0 @@
|
|||||||
using System.Text.Json;
|
|
||||||
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi;
|
|
||||||
|
|
||||||
namespace Yi.Framework.AiHub.Domain.Shared.Dtos.Anthropic;
|
|
||||||
|
|
||||||
public static class AnthropicToOpenAi
|
|
||||||
{
|
|
||||||
/// <summary>
|
|
||||||
/// 将AnthropicInput转换为ThorChatCompletionsRequest
|
|
||||||
/// </summary>
|
|
||||||
public static ThorChatCompletionsRequest ConvertAnthropicToOpenAi(AnthropicInput anthropicInput)
|
|
||||||
{
|
|
||||||
var openAiRequest = new ThorChatCompletionsRequest
|
|
||||||
{
|
|
||||||
Model = anthropicInput.Model,
|
|
||||||
MaxTokens = anthropicInput.MaxTokens,
|
|
||||||
Stream = anthropicInput.Stream,
|
|
||||||
Messages = new List<ThorChatMessage>(anthropicInput.Messages.Count)
|
|
||||||
};
|
|
||||||
|
|
||||||
// high medium minimal low
|
|
||||||
if (openAiRequest.Model.EndsWith("-high") ||
|
|
||||||
openAiRequest.Model.EndsWith("-medium") ||
|
|
||||||
openAiRequest.Model.EndsWith("-minimal") ||
|
|
||||||
openAiRequest.Model.EndsWith("-low"))
|
|
||||||
{
|
|
||||||
openAiRequest.ReasoningEffort = openAiRequest.Model switch
|
|
||||||
{
|
|
||||||
var model when model.EndsWith("-high") => "high",
|
|
||||||
var model when model.EndsWith("-medium") => "medium",
|
|
||||||
var model when model.EndsWith("-minimal") => "minimal",
|
|
||||||
var model when model.EndsWith("-low") => "low",
|
|
||||||
_ => "medium"
|
|
||||||
};
|
|
||||||
|
|
||||||
openAiRequest.Model = openAiRequest.Model.Replace("-high", "")
|
|
||||||
.Replace("-medium", "")
|
|
||||||
.Replace("-minimal", "")
|
|
||||||
.Replace("-low", "");
|
|
||||||
}
|
|
||||||
|
|
||||||
if (anthropicInput.Thinking != null &&
|
|
||||||
anthropicInput.Thinking.Type.Equals("enabled", StringComparison.OrdinalIgnoreCase))
|
|
||||||
{
|
|
||||||
openAiRequest.Thinking = new ThorChatClaudeThinking()
|
|
||||||
{
|
|
||||||
BudgetToken = anthropicInput.Thinking.BudgetTokens,
|
|
||||||
Type = "enabled",
|
|
||||||
};
|
|
||||||
openAiRequest.EnableThinking = true;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (openAiRequest.Model.EndsWith("-thinking"))
|
|
||||||
{
|
|
||||||
openAiRequest.EnableThinking = true;
|
|
||||||
openAiRequest.Model = openAiRequest.Model.Replace("-thinking", "");
|
|
||||||
}
|
|
||||||
|
|
||||||
if (openAiRequest.Stream == true)
|
|
||||||
{
|
|
||||||
openAiRequest.StreamOptions = new ThorStreamOptions()
|
|
||||||
{
|
|
||||||
IncludeUsage = true,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!string.IsNullOrEmpty(anthropicInput.System))
|
|
||||||
{
|
|
||||||
openAiRequest.Messages.Add(ThorChatMessage.CreateSystemMessage(anthropicInput.System));
|
|
||||||
}
|
|
||||||
|
|
||||||
if (anthropicInput.Systems?.Count > 0)
|
|
||||||
{
|
|
||||||
foreach (var systemContent in anthropicInput.Systems)
|
|
||||||
{
|
|
||||||
openAiRequest.Messages.Add(ThorChatMessage.CreateSystemMessage(systemContent.Text ?? string.Empty));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// 处理messages
|
|
||||||
if (anthropicInput.Messages != null)
|
|
||||||
{
|
|
||||||
foreach (var message in anthropicInput.Messages)
|
|
||||||
{
|
|
||||||
var thorMessages = ConvertAnthropicMessageToThor(message);
|
|
||||||
// 需要过滤 空消息
|
|
||||||
if (thorMessages.Count == 0)
|
|
||||||
{
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
openAiRequest.Messages.AddRange(thorMessages);
|
|
||||||
}
|
|
||||||
|
|
||||||
openAiRequest.Messages = openAiRequest.Messages
|
|
||||||
.Where(m => !string.IsNullOrEmpty(m.Content) || m.Contents?.Count > 0 || m.ToolCalls?.Count > 0 ||
|
|
||||||
!string.IsNullOrEmpty(m.ToolCallId))
|
|
||||||
.ToList();
|
|
||||||
}
|
|
||||||
|
|
||||||
// 处理tools
|
|
||||||
if (anthropicInput.Tools is { Count: > 0 })
|
|
||||||
{
|
|
||||||
openAiRequest.Tools = anthropicInput.Tools.Where(x => x.name != "web_search")
|
|
||||||
.Select(ConvertAnthropicToolToThor).ToList();
|
|
||||||
}
|
|
||||||
|
|
||||||
// 判断是否存在web_search
|
|
||||||
if (anthropicInput.Tools?.Any(x => x.name == "web_search") == true)
|
|
||||||
{
|
|
||||||
openAiRequest.WebSearchOptions = new ThorChatWebSearchOptions()
|
|
||||||
{
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
// 处理tool_choice
|
|
||||||
if (anthropicInput.ToolChoice != null)
|
|
||||||
{
|
|
||||||
openAiRequest.ToolChoice = ConvertAnthropicToolChoiceToThor(anthropicInput.ToolChoice);
|
|
||||||
}
|
|
||||||
|
|
||||||
return openAiRequest;
|
|
||||||
}
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 根据最后的内容块类型和OpenAI的完成原因确定Claude的停止原因
|
|
||||||
/// </summary>
|
|
||||||
public static string GetStopReasonByLastContentType(string? openAiFinishReason, string lastContentBlockType)
|
|
||||||
{
|
|
||||||
// 如果最后一个内容块是工具调用,优先返回tool_use
|
|
||||||
if (lastContentBlockType == "tool_use")
|
|
||||||
{
|
|
||||||
return "tool_use";
|
|
||||||
}
|
|
||||||
|
|
||||||
// 否则使用标准的转换逻辑
|
|
||||||
return GetClaudeStopReason(openAiFinishReason);
|
|
||||||
}
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 创建message_start事件
|
|
||||||
/// </summary>
|
|
||||||
public static AnthropicStreamDto CreateMessageStartEvent(string messageId, string model)
|
|
||||||
{
|
|
||||||
return new AnthropicStreamDto
|
|
||||||
{
|
|
||||||
Type = "message_start",
|
|
||||||
Message = new AnthropicChatCompletionDto
|
|
||||||
{
|
|
||||||
id = messageId,
|
|
||||||
type = "message",
|
|
||||||
role = "assistant",
|
|
||||||
model = model,
|
|
||||||
content = new AnthropicChatCompletionDtoContent[0],
|
|
||||||
Usage = new AnthropicCompletionDtoUsage
|
|
||||||
{
|
|
||||||
InputTokens = 0,
|
|
||||||
OutputTokens = 0,
|
|
||||||
CacheCreationInputTokens = 0,
|
|
||||||
CacheReadInputTokens = 0
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 创建content_block_start事件
|
|
||||||
/// </summary>
|
|
||||||
public static AnthropicStreamDto CreateContentBlockStartEvent()
|
|
||||||
{
|
|
||||||
return new AnthropicStreamDto
|
|
||||||
{
|
|
||||||
Type = "content_block_start",
|
|
||||||
Index = 0,
|
|
||||||
ContentBlock = new AnthropicChatCompletionDtoContentBlock
|
|
||||||
{
|
|
||||||
Type = "text",
|
|
||||||
Id = null,
|
|
||||||
Name = null
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 创建thinking block start事件
|
|
||||||
/// </summary>
|
|
||||||
public static AnthropicStreamDto CreateThinkingBlockStartEvent()
|
|
||||||
{
|
|
||||||
return new AnthropicStreamDto
|
|
||||||
{
|
|
||||||
Type = "content_block_start",
|
|
||||||
Index = 0,
|
|
||||||
ContentBlock = new AnthropicChatCompletionDtoContentBlock
|
|
||||||
{
|
|
||||||
Type = "thinking",
|
|
||||||
Id = null,
|
|
||||||
Name = null
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 创建content_block_delta事件
|
|
||||||
/// </summary>
|
|
||||||
public static AnthropicStreamDto CreateContentBlockDeltaEvent(string text)
|
|
||||||
{
|
|
||||||
return new AnthropicStreamDto
|
|
||||||
{
|
|
||||||
Type = "content_block_delta",
|
|
||||||
Index = 0,
|
|
||||||
Delta = new AnthropicChatCompletionDtoDelta
|
|
||||||
{
|
|
||||||
Type = "text_delta",
|
|
||||||
Text = text
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 创建thinking delta事件
|
|
||||||
/// </summary>
|
|
||||||
public static AnthropicStreamDto CreateThinkingBlockDeltaEvent(string thinking)
|
|
||||||
{
|
|
||||||
return new AnthropicStreamDto
|
|
||||||
{
|
|
||||||
Type = "content_block_delta",
|
|
||||||
Index = 0,
|
|
||||||
Delta = new AnthropicChatCompletionDtoDelta
|
|
||||||
{
|
|
||||||
Type = "thinking",
|
|
||||||
Thinking = thinking
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 创建content_block_stop事件
|
|
||||||
/// </summary>
|
|
||||||
public static AnthropicStreamDto CreateContentBlockStopEvent()
|
|
||||||
{
|
|
||||||
return new AnthropicStreamDto
|
|
||||||
{
|
|
||||||
Type = "content_block_stop",
|
|
||||||
Index = 0
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 创建message_delta事件
|
|
||||||
/// </summary>
|
|
||||||
public static AnthropicStreamDto CreateMessageDeltaEvent(string finishReason, AnthropicCompletionDtoUsage usage)
|
|
||||||
{
|
|
||||||
return new AnthropicStreamDto
|
|
||||||
{
|
|
||||||
Type = "message_delta",
|
|
||||||
Usage = usage,
|
|
||||||
Delta = new AnthropicChatCompletionDtoDelta
|
|
||||||
{
|
|
||||||
StopReason = finishReason
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 创建message_stop事件
|
|
||||||
/// </summary>
|
|
||||||
public static AnthropicStreamDto CreateMessageStopEvent()
|
|
||||||
{
|
|
||||||
return new AnthropicStreamDto
|
|
||||||
{
|
|
||||||
Type = "message_stop"
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 创建tool block start事件
|
|
||||||
/// </summary>
|
|
||||||
public static AnthropicStreamDto CreateToolBlockStartEvent(string? toolId, string? toolName)
|
|
||||||
{
|
|
||||||
return new AnthropicStreamDto
|
|
||||||
{
|
|
||||||
Type = "content_block_start",
|
|
||||||
Index = 0,
|
|
||||||
ContentBlock = new AnthropicChatCompletionDtoContentBlock
|
|
||||||
{
|
|
||||||
Type = "tool_use",
|
|
||||||
Id = toolId,
|
|
||||||
Name = toolName
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 创建tool delta事件
|
|
||||||
/// </summary>
|
|
||||||
public static AnthropicStreamDto CreateToolBlockDeltaEvent(string partialJson)
|
|
||||||
{
|
|
||||||
return new AnthropicStreamDto
|
|
||||||
{
|
|
||||||
Type = "content_block_delta",
|
|
||||||
Index = 0,
|
|
||||||
Delta = new AnthropicChatCompletionDtoDelta
|
|
||||||
{
|
|
||||||
Type = "input_json_delta",
|
|
||||||
PartialJson = partialJson
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 转换Anthropic消息为Thor消息列表
|
|
||||||
/// </summary>
|
|
||||||
public static List<ThorChatMessage> ConvertAnthropicMessageToThor(AnthropicMessageInput anthropicMessage)
|
|
||||||
{
|
|
||||||
var results = new List<ThorChatMessage>();
|
|
||||||
|
|
||||||
// 处理简单的字符串内容
|
|
||||||
if (anthropicMessage.Content != null)
|
|
||||||
{
|
|
||||||
var thorMessage = new ThorChatMessage
|
|
||||||
{
|
|
||||||
Role = anthropicMessage.Role,
|
|
||||||
Content = anthropicMessage.Content
|
|
||||||
};
|
|
||||||
results.Add(thorMessage);
|
|
||||||
return results;
|
|
||||||
}
|
|
||||||
|
|
||||||
// 处理多模态内容
|
|
||||||
if (anthropicMessage.Contents is { Count: > 0 })
|
|
||||||
{
|
|
||||||
var currentContents = new List<ThorChatMessageContent>();
|
|
||||||
var currentToolCalls = new List<ThorToolCall>();
|
|
||||||
|
|
||||||
foreach (var content in anthropicMessage.Contents)
|
|
||||||
{
|
|
||||||
switch (content.Type)
|
|
||||||
{
|
|
||||||
case "text":
|
|
||||||
currentContents.Add(ThorChatMessageContent.CreateTextContent(content.Text ?? string.Empty));
|
|
||||||
break;
|
|
||||||
case "thinking" when !string.IsNullOrEmpty(content.Thinking):
|
|
||||||
results.Add(new ThorChatMessage()
|
|
||||||
{
|
|
||||||
ReasoningContent = content.Thinking
|
|
||||||
});
|
|
||||||
break;
|
|
||||||
case "image":
|
|
||||||
{
|
|
||||||
if (content.Source != null)
|
|
||||||
{
|
|
||||||
var imageUrl = content.Source.Type == "base64"
|
|
||||||
? $"data:{content.Source.MediaType};base64,{content.Source.Data}"
|
|
||||||
: content.Source.Data;
|
|
||||||
currentContents.Add(ThorChatMessageContent.CreateImageUrlContent(imageUrl ?? string.Empty));
|
|
||||||
}
|
|
||||||
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
case "tool_use":
|
|
||||||
{
|
|
||||||
// 如果有普通内容,先创建内容消息
|
|
||||||
if (currentContents.Count > 0)
|
|
||||||
{
|
|
||||||
if (currentContents.Count == 1 && currentContents.Any(x => x.Type == "text"))
|
|
||||||
{
|
|
||||||
var contentMessage = new ThorChatMessage
|
|
||||||
{
|
|
||||||
Role = anthropicMessage.Role,
|
|
||||||
ContentCalculated = currentContents.FirstOrDefault()?.Text ?? string.Empty
|
|
||||||
};
|
|
||||||
results.Add(contentMessage);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
var contentMessage = new ThorChatMessage
|
|
||||||
{
|
|
||||||
Role = anthropicMessage.Role,
|
|
||||||
Contents = currentContents
|
|
||||||
};
|
|
||||||
results.Add(contentMessage);
|
|
||||||
}
|
|
||||||
|
|
||||||
currentContents = new List<ThorChatMessageContent>();
|
|
||||||
}
|
|
||||||
|
|
||||||
// 收集工具调用
|
|
||||||
var toolCall = new ThorToolCall
|
|
||||||
{
|
|
||||||
Id = content.Id,
|
|
||||||
Type = "function",
|
|
||||||
Function = new ThorChatMessageFunction
|
|
||||||
{
|
|
||||||
Name = content.Name,
|
|
||||||
Arguments = JsonSerializer.Serialize(content.Input)
|
|
||||||
}
|
|
||||||
};
|
|
||||||
currentToolCalls.Add(toolCall);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
case "tool_result":
|
|
||||||
{
|
|
||||||
// 如果有普通内容,先创建内容消息
|
|
||||||
if (currentContents.Count > 0)
|
|
||||||
{
|
|
||||||
var contentMessage = new ThorChatMessage
|
|
||||||
{
|
|
||||||
Role = anthropicMessage.Role,
|
|
||||||
Contents = currentContents
|
|
||||||
};
|
|
||||||
results.Add(contentMessage);
|
|
||||||
currentContents = [];
|
|
||||||
}
|
|
||||||
|
|
||||||
// 如果有工具调用,先创建工具调用消息
|
|
||||||
if (currentToolCalls.Count > 0)
|
|
||||||
{
|
|
||||||
var toolCallMessage = new ThorChatMessage
|
|
||||||
{
|
|
||||||
Role = anthropicMessage.Role,
|
|
||||||
ToolCalls = currentToolCalls
|
|
||||||
};
|
|
||||||
results.Add(toolCallMessage);
|
|
||||||
currentToolCalls = new List<ThorToolCall>();
|
|
||||||
}
|
|
||||||
|
|
||||||
// 创建工具结果消息
|
|
||||||
var toolMessage = new ThorChatMessage
|
|
||||||
{
|
|
||||||
Role = "tool",
|
|
||||||
ToolCallId = content.ToolUseId,
|
|
||||||
Content = content.Content?.ToString() ?? string.Empty
|
|
||||||
};
|
|
||||||
results.Add(toolMessage);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// 处理剩余的内容
|
|
||||||
if (currentContents.Count > 0)
|
|
||||||
{
|
|
||||||
var contentMessage = new ThorChatMessage
|
|
||||||
{
|
|
||||||
Role = anthropicMessage.Role,
|
|
||||||
Contents = currentContents
|
|
||||||
};
|
|
||||||
results.Add(contentMessage);
|
|
||||||
}
|
|
||||||
|
|
||||||
// 处理剩余的工具调用
|
|
||||||
if (currentToolCalls.Count > 0)
|
|
||||||
{
|
|
||||||
var toolCallMessage = new ThorChatMessage
|
|
||||||
{
|
|
||||||
Role = anthropicMessage.Role,
|
|
||||||
ToolCalls = currentToolCalls
|
|
||||||
};
|
|
||||||
results.Add(toolCallMessage);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// 如果没有任何内容,返回一个空的消息
|
|
||||||
if (results.Count == 0)
|
|
||||||
{
|
|
||||||
results.Add(new ThorChatMessage
|
|
||||||
{
|
|
||||||
Role = anthropicMessage.Role,
|
|
||||||
Content = string.Empty
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
// 如果只有一个text则使用content字段
|
|
||||||
if (results is [{ Contents.Count: 1 }] &&
|
|
||||||
results.FirstOrDefault()?.Contents?.FirstOrDefault()?.Type == "text" &&
|
|
||||||
!string.IsNullOrEmpty(results.FirstOrDefault()?.Contents?.FirstOrDefault()?.Text))
|
|
||||||
{
|
|
||||||
return
|
|
||||||
[
|
|
||||||
new ThorChatMessage
|
|
||||||
{
|
|
||||||
Role = results[0].Role,
|
|
||||||
Content = results.FirstOrDefault()?.Contents?.FirstOrDefault()?.Text ?? string.Empty
|
|
||||||
}
|
|
||||||
];
|
|
||||||
}
|
|
||||||
|
|
||||||
return results;
|
|
||||||
}
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 转换Anthropic工具为Thor工具
|
|
||||||
/// </summary>
|
|
||||||
public static ThorToolDefinition ConvertAnthropicToolToThor(AnthropicMessageTool anthropicTool)
|
|
||||||
{
|
|
||||||
IDictionary<string, ThorToolFunctionPropertyDefinition> values =
|
|
||||||
new Dictionary<string, ThorToolFunctionPropertyDefinition>();
|
|
||||||
|
|
||||||
if (anthropicTool.InputSchema?.Properties != null)
|
|
||||||
{
|
|
||||||
foreach (var property in anthropicTool.InputSchema.Properties)
|
|
||||||
{
|
|
||||||
if (property.Value?.description != null)
|
|
||||||
{
|
|
||||||
var definitionType = new ThorToolFunctionPropertyDefinition()
|
|
||||||
{
|
|
||||||
Description = property.Value.description,
|
|
||||||
Type = property.Value.type
|
|
||||||
};
|
|
||||||
if (property.Value?.items?.type != null)
|
|
||||||
{
|
|
||||||
definitionType.Items = new ThorToolFunctionPropertyDefinition()
|
|
||||||
{
|
|
||||||
Type = property.Value.items.type
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
values.Add(property.Key, definitionType);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
return new ThorToolDefinition
|
|
||||||
{
|
|
||||||
Type = "function",
|
|
||||||
Function = new ThorToolFunctionDefinition
|
|
||||||
{
|
|
||||||
Name = anthropicTool.name,
|
|
||||||
Description = anthropicTool.Description,
|
|
||||||
Parameters = new ThorToolFunctionPropertyDefinition
|
|
||||||
{
|
|
||||||
Type = anthropicTool.InputSchema?.Type ?? "object",
|
|
||||||
Properties = values,
|
|
||||||
Required = anthropicTool.InputSchema?.Required
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 将OpenAI的完成原因转换为Claude的停止原因
|
|
||||||
/// </summary>
|
|
||||||
public static string GetClaudeStopReason(string? openAIFinishReason)
|
|
||||||
{
|
|
||||||
return openAIFinishReason switch
|
|
||||||
{
|
|
||||||
"stop" => "end_turn",
|
|
||||||
"length" => "max_tokens",
|
|
||||||
"tool_calls" => "tool_use",
|
|
||||||
"content_filter" => "stop_sequence",
|
|
||||||
_ => "end_turn"
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 将OpenAI响应转换为Claude响应格式
|
|
||||||
/// </summary>
|
|
||||||
public static AnthropicChatCompletionDto ConvertOpenAIToClaude(ThorChatCompletionsResponse openAIResponse,
|
|
||||||
AnthropicInput originalRequest)
|
|
||||||
{
|
|
||||||
var claudeResponse = new AnthropicChatCompletionDto
|
|
||||||
{
|
|
||||||
id = openAIResponse.Id,
|
|
||||||
type = "message",
|
|
||||||
role = "assistant",
|
|
||||||
model = openAIResponse.Model ?? originalRequest.Model,
|
|
||||||
stop_reason = GetClaudeStopReason(openAIResponse.Choices?.FirstOrDefault()?.FinishReason),
|
|
||||||
stop_sequence = "",
|
|
||||||
content = []
|
|
||||||
};
|
|
||||||
|
|
||||||
if (openAIResponse.Choices is { Count: > 0 })
|
|
||||||
{
|
|
||||||
var choice = openAIResponse.Choices.First();
|
|
||||||
var contents = new List<AnthropicChatCompletionDtoContent>();
|
|
||||||
|
|
||||||
if (!string.IsNullOrEmpty(choice.Message.Content) && !string.IsNullOrEmpty(choice.Message.ReasoningContent))
|
|
||||||
{
|
|
||||||
contents.Add(new AnthropicChatCompletionDtoContent
|
|
||||||
{
|
|
||||||
type = "thinking",
|
|
||||||
Thinking = choice.Message.ReasoningContent
|
|
||||||
});
|
|
||||||
|
|
||||||
contents.Add(new AnthropicChatCompletionDtoContent
|
|
||||||
{
|
|
||||||
type = "text",
|
|
||||||
text = choice.Message.Content
|
|
||||||
});
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
// 处理思维内容
|
|
||||||
if (!string.IsNullOrEmpty(choice.Message.ReasoningContent))
|
|
||||||
contents.Add(new AnthropicChatCompletionDtoContent
|
|
||||||
{
|
|
||||||
type = "thinking",
|
|
||||||
Thinking = choice.Message.ReasoningContent
|
|
||||||
});
|
|
||||||
|
|
||||||
// 处理文本内容
|
|
||||||
if (!string.IsNullOrEmpty(choice.Message.Content))
|
|
||||||
contents.Add(new AnthropicChatCompletionDtoContent
|
|
||||||
{
|
|
||||||
type = "text",
|
|
||||||
text = choice.Message.Content
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
// 处理工具调用
|
|
||||||
if (choice.Message.ToolCalls is { Count: > 0 })
|
|
||||||
contents.AddRange(choice.Message.ToolCalls.Select(toolCall => new AnthropicChatCompletionDtoContent
|
|
||||||
{
|
|
||||||
type = "tool_use", id = toolCall.Id, name = toolCall.Function?.Name,
|
|
||||||
input = JsonSerializer.Deserialize<object>(toolCall.Function?.Arguments ?? "{}")
|
|
||||||
}));
|
|
||||||
|
|
||||||
claudeResponse.content = contents.ToArray();
|
|
||||||
}
|
|
||||||
|
|
||||||
// 处理使用情况统计 - 确保始终提供Usage信息
|
|
||||||
claudeResponse.Usage = new AnthropicCompletionDtoUsage
|
|
||||||
{
|
|
||||||
InputTokens = openAIResponse.Usage?.PromptTokens ?? 0,
|
|
||||||
OutputTokens = (int?)(openAIResponse.Usage?.CompletionTokens ?? 0),
|
|
||||||
CacheCreationInputTokens = openAIResponse.Usage?.PromptTokensDetails?.CachedTokens ?? 0,
|
|
||||||
CacheReadInputTokens = openAIResponse.Usage?.PromptTokensDetails?.CachedTokens ?? 0
|
|
||||||
};
|
|
||||||
|
|
||||||
return claudeResponse;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 转换Anthropic工具选择为Thor工具选择
|
|
||||||
/// </summary>
|
|
||||||
public static ThorToolChoice ConvertAnthropicToolChoiceToThor(AnthropicTooChoiceInput anthropicToolChoice)
|
|
||||||
{
|
|
||||||
return new ThorToolChoice
|
|
||||||
{
|
|
||||||
Type = anthropicToolChoice.Type ?? "auto",
|
|
||||||
Function = anthropicToolChoice.Name != null
|
|
||||||
? new ThorToolChoiceFunctionTool { Name = anthropicToolChoice.Name }
|
|
||||||
: null
|
|
||||||
};
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -0,0 +1,47 @@
|
|||||||
|
using System.Text.Json;
|
||||||
|
using System.Text.Json.Serialization;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi.Responses;
|
||||||
|
|
||||||
|
public class OpenAiResponsesInput
|
||||||
|
{
|
||||||
|
[JsonPropertyName("stream")] public bool? Stream { get; set; }
|
||||||
|
|
||||||
|
[JsonPropertyName("model")] public string Model { get; set; }
|
||||||
|
[JsonPropertyName("input")] public JsonElement Input { get; set; }
|
||||||
|
|
||||||
|
[JsonPropertyName("max_output_tokens")]
|
||||||
|
public int? MaxOutputTokens { get; set; }
|
||||||
|
|
||||||
|
[JsonPropertyName("max_tool_calls")] public JsonElement? MaxToolCalls { get; set; }
|
||||||
|
[JsonPropertyName("instructions")] public string? Instructions { get; set; }
|
||||||
|
[JsonPropertyName("metadata")] public JsonElement? Metadata { get; set; }
|
||||||
|
|
||||||
|
[JsonPropertyName("parallel_tool_calls")]
|
||||||
|
public bool? ParallelToolCalls { get; set; }
|
||||||
|
|
||||||
|
[JsonPropertyName("previous_response_id")]
|
||||||
|
public string? PreviousResponseId { get; set; }
|
||||||
|
|
||||||
|
[JsonPropertyName("prompt")] public JsonElement? Prompt { get; set; }
|
||||||
|
[JsonPropertyName("prompt_cache_key")] public string? PromptCacheKey { get; set; }
|
||||||
|
|
||||||
|
[JsonPropertyName("prompt_cache_retention")]
|
||||||
|
public string? PromptCacheRetention { get; set; }
|
||||||
|
|
||||||
|
[JsonPropertyName("reasoning")] public JsonElement? Reasoning { get; set; }
|
||||||
|
|
||||||
|
[JsonPropertyName("safety_identifier")]
|
||||||
|
public string? SafetyIdentifier { get; set; }
|
||||||
|
|
||||||
|
[JsonPropertyName("service_tier")] public string? ServiceTier { get; set; }
|
||||||
|
[JsonPropertyName("store")] public bool? Store { get; set; }
|
||||||
|
[JsonPropertyName("stream_options")] public JsonElement? StreamOptions { get; set; }
|
||||||
|
[JsonPropertyName("temperature")] public decimal? Temperature { get; set; }
|
||||||
|
[JsonPropertyName("text")] public JsonElement? Text { get; set; }
|
||||||
|
[JsonPropertyName("tool_choice")] public JsonElement? ToolChoice { get; set; }
|
||||||
|
[JsonPropertyName("tools")] public JsonElement? Tools { get; set; }
|
||||||
|
[JsonPropertyName("top_logprobs")] public int? TopLogprobs { get; set; }
|
||||||
|
[JsonPropertyName("top_p")] public decimal? TopP { get; set; }
|
||||||
|
[JsonPropertyName("truncation")] public string? Truncation { get; set; }
|
||||||
|
}
|
||||||
@@ -0,0 +1,91 @@
|
|||||||
|
using System.Text.Json.Serialization;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi.Responses;
|
||||||
|
|
||||||
|
public class OpenAiResponsesOutput
|
||||||
|
{
|
||||||
|
[JsonPropertyName("id")]
|
||||||
|
public string? Id { get; set; }
|
||||||
|
[JsonPropertyName("object")]
|
||||||
|
public string? Object { get; set; }
|
||||||
|
[JsonPropertyName("created_at")]
|
||||||
|
public long CreatedAt { get; set; }
|
||||||
|
[JsonPropertyName("status")]
|
||||||
|
public string? Status { get; set; }
|
||||||
|
[JsonPropertyName("error")]
|
||||||
|
public dynamic? Error { get; set; }
|
||||||
|
[JsonPropertyName("incomplete_details")]
|
||||||
|
public dynamic? IncompleteDetails { get; set; }
|
||||||
|
[JsonPropertyName("instructions")]
|
||||||
|
public dynamic? Instructions { get; set; }
|
||||||
|
[JsonPropertyName("max_output_tokens")]
|
||||||
|
public dynamic? MaxOutputTokens { get; set; }
|
||||||
|
[JsonPropertyName("model")]
|
||||||
|
public string? Model { get; set; }
|
||||||
|
// output 是复杂对象
|
||||||
|
[JsonPropertyName("output")]
|
||||||
|
public List<dynamic>? Output { get; set; }
|
||||||
|
[JsonPropertyName("parallel_tool_calls")]
|
||||||
|
public bool ParallelToolCalls { get; set; }
|
||||||
|
[JsonPropertyName("previous_response_id")]
|
||||||
|
public dynamic? PreviousResponseId { get; set; }
|
||||||
|
[JsonPropertyName("reasoning")]
|
||||||
|
public dynamic? Reasoning { get; set; }
|
||||||
|
[JsonPropertyName("store")]
|
||||||
|
public bool Store { get; set; }
|
||||||
|
[JsonPropertyName("temperature")]
|
||||||
|
public double Temperature { get; set; }
|
||||||
|
[JsonPropertyName("text")]
|
||||||
|
public dynamic? Text { get; set; }
|
||||||
|
[JsonPropertyName("tool_choice")]
|
||||||
|
public string? ToolChoice { get; set; }
|
||||||
|
[JsonPropertyName("tools")]
|
||||||
|
public List<dynamic>? Tools { get; set; }
|
||||||
|
[JsonPropertyName("top_p")]
|
||||||
|
public double TopP { get; set; }
|
||||||
|
[JsonPropertyName("truncation")]
|
||||||
|
public string? Truncation { get; set; }
|
||||||
|
// usage 为唯一强类型
|
||||||
|
[JsonPropertyName("usage")]
|
||||||
|
public OpenAiResponsesUsageOutput? Usage { get; set; }
|
||||||
|
[JsonPropertyName("user")]
|
||||||
|
public dynamic? User { get; set; }
|
||||||
|
[JsonPropertyName("metadata")]
|
||||||
|
public dynamic? Metadata { get; set; }
|
||||||
|
|
||||||
|
public void SupplementalMultiplier(decimal multiplier)
|
||||||
|
{
|
||||||
|
if (this.Usage is not null)
|
||||||
|
{
|
||||||
|
this.Usage.InputTokens =
|
||||||
|
(int)Math.Round((this.Usage?.InputTokens ?? 0) * multiplier);
|
||||||
|
|
||||||
|
this.Usage.OutputTokens =
|
||||||
|
(int)Math.Round((this.Usage?.OutputTokens ?? 0) * multiplier);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public class OpenAiResponsesUsageOutput
|
||||||
|
{
|
||||||
|
[JsonPropertyName("input_tokens")]
|
||||||
|
public int InputTokens { get; set; }
|
||||||
|
[JsonPropertyName("input_tokens_details")]
|
||||||
|
public OpenAiResponsesInputTokensDetails? InputTokensDetails { get; set; }
|
||||||
|
[JsonPropertyName("output_tokens")]
|
||||||
|
public int OutputTokens { get; set; }
|
||||||
|
[JsonPropertyName("output_tokens_details")]
|
||||||
|
public OpenAiResponsesOutputTokensDetails? OutputTokensDetails { get; set; }
|
||||||
|
[JsonPropertyName("total_tokens")]
|
||||||
|
public int TotalTokens { get; set; }
|
||||||
|
}
|
||||||
|
public class OpenAiResponsesInputTokensDetails
|
||||||
|
{
|
||||||
|
[JsonPropertyName("cached_tokens")]
|
||||||
|
public int CachedTokens { get; set; }
|
||||||
|
}
|
||||||
|
public class OpenAiResponsesOutputTokensDetails
|
||||||
|
{
|
||||||
|
[JsonPropertyName("reasoning_tokens")]
|
||||||
|
public int ReasoningTokens { get; set; }
|
||||||
|
}
|
||||||
@@ -61,7 +61,7 @@ public record ThorChatCompletionsResponse
|
|||||||
[JsonPropertyName("error")]
|
[JsonPropertyName("error")]
|
||||||
public ThorError? Error { get; set; }
|
public ThorError? Error { get; set; }
|
||||||
|
|
||||||
public void SupplementalMultiplier(double multiplier)
|
public void SupplementalMultiplier(decimal multiplier)
|
||||||
{
|
{
|
||||||
if (this.Usage is not null)
|
if (this.Usage is not null)
|
||||||
{
|
{
|
||||||
@@ -73,6 +73,9 @@ public record ThorChatCompletionsResponse
|
|||||||
(int)Math.Round((this.Usage.CompletionTokens ?? 0) * multiplier);
|
(int)Math.Round((this.Usage.CompletionTokens ?? 0) * multiplier);
|
||||||
this.Usage.PromptTokens =
|
this.Usage.PromptTokens =
|
||||||
(int)Math.Round((this.Usage.PromptTokens ?? 0) * multiplier);
|
(int)Math.Round((this.Usage.PromptTokens ?? 0) * multiplier);
|
||||||
|
|
||||||
|
this.Usage.TotalTokens =
|
||||||
|
(int)Math.Round((this.Usage.TotalTokens ?? 0) * multiplier);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -90,6 +90,28 @@ public class ThorChatMessage
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 用于数据存储
|
||||||
|
/// </summary>
|
||||||
|
[JsonIgnore]
|
||||||
|
public string MessagesStore
|
||||||
|
{
|
||||||
|
get
|
||||||
|
{
|
||||||
|
if (Content is not null)
|
||||||
|
{
|
||||||
|
return Content;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (Contents is not null && Contents.Any())
|
||||||
|
{
|
||||||
|
return JsonSerializer.Serialize(Contents);
|
||||||
|
}
|
||||||
|
return string.Empty;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 【可选】参与者的可选名称。提供模型信息以区分相同角色的参与者。
|
/// 【可选】参与者的可选名称。提供模型信息以区分相同角色的参与者。
|
||||||
/// </summary>
|
/// </summary>
|
||||||
|
|||||||
@@ -0,0 +1,7 @@
|
|||||||
|
namespace Yi.Framework.AiHub.Domain.Shared.Enums;
|
||||||
|
|
||||||
|
public enum AnnouncementTypeEnum
|
||||||
|
{
|
||||||
|
Activity=1,
|
||||||
|
System=2
|
||||||
|
}
|
||||||
@@ -102,11 +102,6 @@ public enum GoodsTypeEnum
|
|||||||
[Price(155.4, 6, 25.9)] [DisplayName("YiXinVip 6 month", "6个月", "年度热销")] [GoodsCategory(GoodsCategoryType.Vip)]
|
[Price(155.4, 6, 25.9)] [DisplayName("YiXinVip 6 month", "6个月", "年度热销")] [GoodsCategory(GoodsCategoryType.Vip)]
|
||||||
YiXinVip6 = 6,
|
YiXinVip6 = 6,
|
||||||
|
|
||||||
[Price(183.2, 8, 22.9)]
|
|
||||||
[DisplayName("YiXinVip 8 month", "8个月(推荐)", "限时活动,超高性价比")]
|
|
||||||
[GoodsCategory(GoodsCategoryType.Vip)]
|
|
||||||
YiXinVip8 = 8,
|
|
||||||
|
|
||||||
// 尊享包服务 - 需要VIP资格才能购买
|
// 尊享包服务 - 需要VIP资格才能购买
|
||||||
[Price(188.9, 0, 1750)]
|
[Price(188.9, 0, 1750)]
|
||||||
[DisplayName("YiXinPremiumPackage 5000W Tokens", "5000万Tokens", "简单尝试")]
|
[DisplayName("YiXinPremiumPackage 5000W Tokens", "5000万Tokens", "简单尝试")]
|
||||||
@@ -264,7 +259,7 @@ public static class GoodsTypeEnumExtensions
|
|||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 计算折扣金额(仅用于尊享包)
|
/// 计算折扣金额(仅用于尊享包)
|
||||||
/// 规则:每累加充值10元,减少2.5元,最多减少50元
|
/// 规则:每累加充值10元,减少10元,最多减少50元
|
||||||
/// </summary>
|
/// </summary>
|
||||||
/// <param name="goodsType">商品类型</param>
|
/// <param name="goodsType">商品类型</param>
|
||||||
/// <param name="totalRechargeAmount">用户累加充值金额</param>
|
/// <param name="totalRechargeAmount">用户累加充值金额</param>
|
||||||
@@ -276,11 +271,10 @@ public static class GoodsTypeEnumExtensions
|
|||||||
{
|
{
|
||||||
return 0m;
|
return 0m;
|
||||||
}
|
}
|
||||||
|
// 每满 10 元减 10 元
|
||||||
// 每10元减2.5元
|
var discountTimes = Math.Floor(totalRechargeAmount / 10m);
|
||||||
var discountAmount = Math.Floor(totalRechargeAmount / 2.5m);
|
var discountAmount = discountTimes * 10m;
|
||||||
|
// 最多减少 50 元
|
||||||
// 最多减少50元
|
|
||||||
return Math.Min(discountAmount, 50m);
|
return Math.Min(discountAmount, 50m);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -1,7 +1,15 @@
|
|||||||
namespace Yi.Framework.AiHub.Domain.Shared.Enums;
|
using System.ComponentModel;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Domain.Shared.Enums;
|
||||||
|
|
||||||
public enum ModelApiTypeEnum
|
public enum ModelApiTypeEnum
|
||||||
{
|
{
|
||||||
|
[Description("OpenAI")]
|
||||||
OpenAi,
|
OpenAi,
|
||||||
Claude
|
|
||||||
|
[Description("Claude")]
|
||||||
|
Claude,
|
||||||
|
|
||||||
|
[Description("Response")]
|
||||||
|
Response
|
||||||
}
|
}
|
||||||
@@ -1,9 +1,15 @@
|
|||||||
namespace Yi.Framework.AiHub.Domain.Shared.Enums;
|
using System.ComponentModel;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Domain.Shared.Enums;
|
||||||
|
|
||||||
public enum ModelTypeEnum
|
public enum ModelTypeEnum
|
||||||
{
|
{
|
||||||
|
[Description("聊天")]
|
||||||
Chat = 0,
|
Chat = 0,
|
||||||
|
|
||||||
|
[Description("图片")]
|
||||||
Image = 1,
|
Image = 1,
|
||||||
Embedding = 2,
|
|
||||||
PremiumChat = 3
|
[Description("嵌入")]
|
||||||
|
Embedding = 2
|
||||||
}
|
}
|
||||||
@@ -0,0 +1,27 @@
|
|||||||
|
using System.ComponentModel;
|
||||||
|
using System.Reflection;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Domain.Shared.Extensions;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 枚举扩展方法
|
||||||
|
/// </summary>
|
||||||
|
public static class EnumExtensions
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 获取枚举的Description特性值
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="value">枚举值</param>
|
||||||
|
/// <returns>Description特性值,如果没有则返回枚举名称</returns>
|
||||||
|
public static string GetDescription(this Enum value)
|
||||||
|
{
|
||||||
|
var field = value.GetType().GetField(value.ToString());
|
||||||
|
if (field == null)
|
||||||
|
{
|
||||||
|
return value.ToString();
|
||||||
|
}
|
||||||
|
|
||||||
|
var attribute = field.GetCustomAttribute<DescriptionAttribute>();
|
||||||
|
return attribute?.Description ?? value.ToString();
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,279 @@
|
|||||||
|
using System.Text.Json;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Domain.Shared.Extensions;
|
||||||
|
|
||||||
|
public static class JsonElementExtensions
|
||||||
|
{
|
||||||
|
#region 路径访问
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 链式获取深层属性,支持对象属性和数组索引
|
||||||
|
/// </summary>
|
||||||
|
/// <example>
|
||||||
|
/// root.GetPath("user", "addresses", 0, "city")
|
||||||
|
/// </example>
|
||||||
|
public static JsonElement? GetPath(this JsonElement element, params object[] path)
|
||||||
|
{
|
||||||
|
JsonElement current = element;
|
||||||
|
|
||||||
|
foreach (var key in path)
|
||||||
|
{
|
||||||
|
switch (key)
|
||||||
|
{
|
||||||
|
case string propertyName:
|
||||||
|
if (current.ValueKind != JsonValueKind.Object ||
|
||||||
|
!current.TryGetProperty(propertyName, out current))
|
||||||
|
return null;
|
||||||
|
break;
|
||||||
|
|
||||||
|
case int index:
|
||||||
|
if (current.ValueKind != JsonValueKind.Array ||
|
||||||
|
index < 0 || index >= current.GetArrayLength())
|
||||||
|
return null;
|
||||||
|
current = current[index];
|
||||||
|
break;
|
||||||
|
|
||||||
|
default:
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return current;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 安全获取对象属性
|
||||||
|
/// </summary>
|
||||||
|
public static JsonElement? Get(this JsonElement element, string propertyName)
|
||||||
|
{
|
||||||
|
if (element.ValueKind == JsonValueKind.Object &&
|
||||||
|
element.TryGetProperty(propertyName, out var value))
|
||||||
|
return value;
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 安全获取数组元素
|
||||||
|
/// </summary>
|
||||||
|
public static JsonElement? Get(this JsonElement element, int index)
|
||||||
|
{
|
||||||
|
if (element.ValueKind == JsonValueKind.Array &&
|
||||||
|
index >= 0 && index < element.GetArrayLength())
|
||||||
|
return element[index];
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 链式安全获取对象属性
|
||||||
|
/// </summary>
|
||||||
|
public static JsonElement? Get(this JsonElement? element, string propertyName)
|
||||||
|
=> element?.Get(propertyName);
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 链式安全获取数组元素
|
||||||
|
/// </summary>
|
||||||
|
public static JsonElement? Get(this JsonElement? element, int index)
|
||||||
|
=> element?.Get(index);
|
||||||
|
|
||||||
|
#endregion
|
||||||
|
|
||||||
|
#region 取值方法(带默认值)
|
||||||
|
|
||||||
|
public static string? GetString(this JsonElement? element, string? defaultValue = null)
|
||||||
|
=> element?.ValueKind == JsonValueKind.String ? element.Value.GetString() : defaultValue;
|
||||||
|
|
||||||
|
public static int GetInt(this JsonElement? element, int defaultValue = 0)
|
||||||
|
=> element?.ValueKind == JsonValueKind.Number ? element.Value.GetInt32() : defaultValue;
|
||||||
|
|
||||||
|
public static long GetLong(this JsonElement? element, long defaultValue = 0)
|
||||||
|
=> element?.ValueKind == JsonValueKind.Number ? element.Value.GetInt64() : defaultValue;
|
||||||
|
|
||||||
|
public static double GetDouble(this JsonElement? element, double defaultValue = 0)
|
||||||
|
=> element?.ValueKind == JsonValueKind.Number ? element.Value.GetDouble() : defaultValue;
|
||||||
|
|
||||||
|
public static decimal GetDecimal(this JsonElement? element, decimal defaultValue = 0)
|
||||||
|
=> element?.ValueKind == JsonValueKind.Number ? element.Value.GetDecimal() : defaultValue;
|
||||||
|
|
||||||
|
public static bool GetBool(this JsonElement? element, bool defaultValue = false)
|
||||||
|
=> element?.ValueKind is JsonValueKind.True or JsonValueKind.False
|
||||||
|
? element.Value.GetBoolean()
|
||||||
|
: defaultValue;
|
||||||
|
|
||||||
|
public static DateTime GetDateTime(this JsonElement? element, DateTime defaultValue = default)
|
||||||
|
=> element?.ValueKind == JsonValueKind.String && element.Value.TryGetDateTime(out var dt)
|
||||||
|
? dt
|
||||||
|
: defaultValue;
|
||||||
|
|
||||||
|
public static Guid GetGuid(this JsonElement? element, Guid defaultValue = default)
|
||||||
|
=> element?.ValueKind == JsonValueKind.String && element.Value.TryGetGuid(out var guid)
|
||||||
|
? guid
|
||||||
|
: defaultValue;
|
||||||
|
|
||||||
|
#endregion
|
||||||
|
|
||||||
|
#region 可空取值方法
|
||||||
|
|
||||||
|
public static int? GetIntOrNull(this JsonElement? element)
|
||||||
|
=> element?.ValueKind == JsonValueKind.Number ? element.Value.GetInt32() : null;
|
||||||
|
|
||||||
|
public static long? GetLongOrNull(this JsonElement? element)
|
||||||
|
=> element?.ValueKind == JsonValueKind.Number ? element.Value.GetInt64() : null;
|
||||||
|
|
||||||
|
public static double? GetDoubleOrNull(this JsonElement? element)
|
||||||
|
=> element?.ValueKind == JsonValueKind.Number ? element.Value.GetDouble() : null;
|
||||||
|
|
||||||
|
public static decimal? GetDecimalOrNull(this JsonElement? element)
|
||||||
|
=> element?.ValueKind == JsonValueKind.Number ? element.Value.GetDecimal() : null;
|
||||||
|
|
||||||
|
public static bool? GetBoolOrNull(this JsonElement? element)
|
||||||
|
=> element?.ValueKind is JsonValueKind.True or JsonValueKind.False
|
||||||
|
? element.Value.GetBoolean()
|
||||||
|
: null;
|
||||||
|
|
||||||
|
public static DateTime? GetDateTimeOrNull(this JsonElement? element)
|
||||||
|
=> element?.ValueKind == JsonValueKind.String && element.Value.TryGetDateTime(out var dt)
|
||||||
|
? dt
|
||||||
|
: null;
|
||||||
|
|
||||||
|
public static Guid? GetGuidOrNull(this JsonElement? element)
|
||||||
|
=> element?.ValueKind == JsonValueKind.String && element.Value.TryGetGuid(out var guid)
|
||||||
|
? guid
|
||||||
|
: null;
|
||||||
|
|
||||||
|
#endregion
|
||||||
|
|
||||||
|
#region 数组操作
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 安全获取数组,不存在返回空数组
|
||||||
|
/// </summary>
|
||||||
|
public static IEnumerable<JsonElement> GetArray(this JsonElement? element)
|
||||||
|
{
|
||||||
|
if (element?.ValueKind == JsonValueKind.Array)
|
||||||
|
{
|
||||||
|
foreach (var item in element.Value.EnumerateArray())
|
||||||
|
yield return item;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取数组长度
|
||||||
|
/// </summary>
|
||||||
|
public static int GetArrayLength(this JsonElement? element)
|
||||||
|
=> element?.ValueKind == JsonValueKind.Array ? element.Value.GetArrayLength() : 0;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 数组转 List
|
||||||
|
/// </summary>
|
||||||
|
public static List<string?> ToStringList(this JsonElement? element)
|
||||||
|
=> element.GetArray().Select(e => e.GetString()).ToList();
|
||||||
|
|
||||||
|
public static List<int> ToIntList(this JsonElement? element)
|
||||||
|
=> element.GetArray()
|
||||||
|
.Where(e => e.ValueKind == JsonValueKind.Number)
|
||||||
|
.Select(e => e.GetInt32())
|
||||||
|
.ToList();
|
||||||
|
|
||||||
|
#endregion
|
||||||
|
|
||||||
|
#region 对象操作
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 安全枚举对象属性
|
||||||
|
/// </summary>
|
||||||
|
public static IEnumerable<JsonProperty> GetProperties(this JsonElement? element)
|
||||||
|
{
|
||||||
|
if (element?.ValueKind == JsonValueKind.Object)
|
||||||
|
{
|
||||||
|
foreach (var prop in element.Value.EnumerateObject())
|
||||||
|
yield return prop;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取所有属性名
|
||||||
|
/// </summary>
|
||||||
|
public static IEnumerable<string> GetPropertyNames(this JsonElement? element)
|
||||||
|
=> element.GetProperties().Select(p => p.Name);
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 判断是否包含某属性
|
||||||
|
/// </summary>
|
||||||
|
public static bool HasProperty(this JsonElement? element, string propertyName)
|
||||||
|
=> element?.ValueKind == JsonValueKind.Object &&
|
||||||
|
element.Value.TryGetProperty(propertyName, out _);
|
||||||
|
|
||||||
|
#endregion
|
||||||
|
|
||||||
|
#region 类型判断
|
||||||
|
|
||||||
|
public static bool IsNull(this JsonElement? element)
|
||||||
|
=> element == null || element.Value.ValueKind == JsonValueKind.Null;
|
||||||
|
|
||||||
|
public static bool IsNullOrUndefined(this JsonElement? element)
|
||||||
|
=> element == null || element.Value.ValueKind is JsonValueKind.Null or JsonValueKind.Undefined;
|
||||||
|
|
||||||
|
public static bool IsObject(this JsonElement? element)
|
||||||
|
=> element?.ValueKind == JsonValueKind.Object;
|
||||||
|
|
||||||
|
public static bool IsArray(this JsonElement? element)
|
||||||
|
=> element?.ValueKind == JsonValueKind.Array;
|
||||||
|
|
||||||
|
public static bool IsString(this JsonElement? element)
|
||||||
|
=> element?.ValueKind == JsonValueKind.String;
|
||||||
|
|
||||||
|
public static bool IsNumber(this JsonElement? element)
|
||||||
|
=> element?.ValueKind == JsonValueKind.Number;
|
||||||
|
|
||||||
|
public static bool IsBool(this JsonElement? element)
|
||||||
|
=> element?.ValueKind is JsonValueKind.True or JsonValueKind.False;
|
||||||
|
|
||||||
|
public static bool Exists(this JsonElement? element)
|
||||||
|
=> element != null && element.Value.ValueKind != JsonValueKind.Undefined;
|
||||||
|
|
||||||
|
#endregion
|
||||||
|
|
||||||
|
#region 反序列化
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 反序列化为指定类型
|
||||||
|
/// </summary>
|
||||||
|
public static T? Deserialize<T>(this JsonElement? element, JsonSerializerOptions? options = null)
|
||||||
|
=> element.HasValue ? element.Value.Deserialize<T>(options) : default;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 反序列化为指定类型,带默认值
|
||||||
|
/// </summary>
|
||||||
|
public static T Deserialize<T>(this JsonElement? element, T defaultValue, JsonSerializerOptions? options = null)
|
||||||
|
=> element.HasValue ? element.Value.Deserialize<T>(options) ?? defaultValue : defaultValue;
|
||||||
|
|
||||||
|
#endregion
|
||||||
|
|
||||||
|
#region 转换为字典/动态类型
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 转换为 Dictionary
|
||||||
|
/// </summary>
|
||||||
|
public static Dictionary<string, JsonElement>? ToDictionary(this JsonElement? element)
|
||||||
|
{
|
||||||
|
if (element?.ValueKind != JsonValueKind.Object)
|
||||||
|
return null;
|
||||||
|
|
||||||
|
var dict = new Dictionary<string, JsonElement>();
|
||||||
|
foreach (var prop in element.Value.EnumerateObject())
|
||||||
|
dict[prop.Name] = prop.Value;
|
||||||
|
return dict;
|
||||||
|
}
|
||||||
|
|
||||||
|
#endregion
|
||||||
|
|
||||||
|
#region 原始值
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取原始 JSON 字符串
|
||||||
|
/// </summary>
|
||||||
|
public static string? GetRawText(this JsonElement? element)
|
||||||
|
=> element?.GetRawText();
|
||||||
|
|
||||||
|
#endregion
|
||||||
|
}
|
||||||
@@ -0,0 +1,30 @@
|
|||||||
|
using System.Text.Json;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Dtos;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi.Responses;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Domain.AiGateWay;
|
||||||
|
|
||||||
|
public interface IOpenAiResponseService
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 响应-流式
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="aiModelDescribe"></param>
|
||||||
|
/// <param name="input"></param>
|
||||||
|
/// <param name="cancellationToken"></param>
|
||||||
|
/// <returns></returns>
|
||||||
|
public IAsyncEnumerable<(string, JsonElement?)> ResponsesStreamAsync(AiModelDescribe aiModelDescribe,
|
||||||
|
OpenAiResponsesInput input,
|
||||||
|
CancellationToken cancellationToken);
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 响应-非流式
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="aiModelDescribe"></param>
|
||||||
|
/// <param name="input"></param>
|
||||||
|
/// <param name="cancellationToken"></param>
|
||||||
|
/// <returns></returns>
|
||||||
|
public Task<OpenAiResponsesOutput> ResponsesAsync(AiModelDescribe aiModelDescribe,
|
||||||
|
OpenAiResponsesInput input,
|
||||||
|
CancellationToken cancellationToken);
|
||||||
|
}
|
||||||
@@ -14,8 +14,6 @@ public class AnthropicChatCompletionsService(
|
|||||||
ILogger<AnthropicChatCompletionsService> logger)
|
ILogger<AnthropicChatCompletionsService> logger)
|
||||||
: IAnthropicChatCompletionService
|
: IAnthropicChatCompletionService
|
||||||
{
|
{
|
||||||
public const double ClaudeMultiplier = 1.3d;
|
|
||||||
|
|
||||||
public async Task<AnthropicChatCompletionDto> ChatCompletionsAsync(AiModelDescribe options, AnthropicInput input,
|
public async Task<AnthropicChatCompletionDto> ChatCompletionsAsync(AiModelDescribe options, AnthropicInput input,
|
||||||
CancellationToken cancellationToken = default)
|
CancellationToken cancellationToken = default)
|
||||||
{
|
{
|
||||||
@@ -86,7 +84,6 @@ public class AnthropicChatCompletionsService(
|
|||||||
var value =
|
var value =
|
||||||
await response.Content.ReadFromJsonAsync<AnthropicChatCompletionDto>(ThorJsonSerializer.DefaultOptions,
|
await response.Content.ReadFromJsonAsync<AnthropicChatCompletionDto>(ThorJsonSerializer.DefaultOptions,
|
||||||
cancellationToken: cancellationToken);
|
cancellationToken: cancellationToken);
|
||||||
value.SupplementalMultiplier(AnthropicChatCompletionsService.ClaudeMultiplier);
|
|
||||||
|
|
||||||
return value;
|
return value;
|
||||||
}
|
}
|
||||||
@@ -170,7 +167,6 @@ public class AnthropicChatCompletionsService(
|
|||||||
var result = JsonSerializer.Deserialize<AnthropicStreamDto>(data,
|
var result = JsonSerializer.Deserialize<AnthropicStreamDto>(data,
|
||||||
ThorJsonSerializer.DefaultOptions);
|
ThorJsonSerializer.DefaultOptions);
|
||||||
|
|
||||||
result.SupplementalMultiplier(AnthropicChatCompletionsService.ClaudeMultiplier);
|
|
||||||
yield return (eventType, result);
|
yield return (eventType, result);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -345,7 +345,7 @@ public sealed class ClaudiaChatCompletionsService(
|
|||||||
var response = await client.HttpRequestRaw(options.Endpoint.TrimEnd('/') + "/v1/messages", new
|
var response = await client.HttpRequestRaw(options.Endpoint.TrimEnd('/') + "/v1/messages", new
|
||||||
{
|
{
|
||||||
model = input.Model,
|
model = input.Model,
|
||||||
max_tokens = input.MaxTokens ?? 2048,
|
max_tokens = input.MaxTokens ?? 64000,
|
||||||
stream = true,
|
stream = true,
|
||||||
tool_choice,
|
tool_choice,
|
||||||
system = CreateMessage(input.Messages.Where(x => x.Role == "system").ToList(), options),
|
system = CreateMessage(input.Messages.Where(x => x.Role == "system").ToList(), options),
|
||||||
@@ -716,7 +716,7 @@ public sealed class ClaudiaChatCompletionsService(
|
|||||||
output.Usage.PromptTokens = output.Usage.InputTokens;
|
output.Usage.PromptTokens = output.Usage.InputTokens;
|
||||||
output.Usage.CompletionTokens = output.Usage.OutputTokens;
|
output.Usage.CompletionTokens = output.Usage.OutputTokens;
|
||||||
output.Usage.TotalTokens = output.Usage.InputTokens + output.Usage.OutputTokens;
|
output.Usage.TotalTokens = output.Usage.InputTokens + output.Usage.OutputTokens;
|
||||||
output.SupplementalMultiplier(AnthropicChatCompletionsService.ClaudeMultiplier);
|
|
||||||
yield return output;
|
yield return output;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -873,7 +873,6 @@ public sealed class ClaudiaChatCompletionsService(
|
|||||||
}
|
}
|
||||||
|
|
||||||
thor.Usage.TotalTokens = thor.Usage.InputTokens + thor.Usage.OutputTokens;
|
thor.Usage.TotalTokens = thor.Usage.InputTokens + thor.Usage.OutputTokens;
|
||||||
thor.SupplementalMultiplier(AnthropicChatCompletionsService.ClaudeMultiplier);
|
|
||||||
return thor;
|
return thor;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -1,313 +0,0 @@
|
|||||||
using Microsoft.Extensions.DependencyInjection;
|
|
||||||
using Microsoft.Extensions.Logging;
|
|
||||||
using Volo.Abp.DependencyInjection;
|
|
||||||
using Yi.Framework.AiHub.Domain.Shared.Dtos;
|
|
||||||
using Yi.Framework.AiHub.Domain.Shared.Dtos.Anthropic;
|
|
||||||
|
|
||||||
namespace Yi.Framework.AiHub.Domain.AiGateWay.Impl.ThorCustomOpenAI.Chats;
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// OpenAI到Claude适配器服务
|
|
||||||
/// 将Claude格式的请求转换为OpenAI格式,然后将OpenAI的响应转换为Claude格式
|
|
||||||
/// </summary>
|
|
||||||
public class CustomOpenAIAnthropicChatCompletionsService(
|
|
||||||
IAbpLazyServiceProvider serviceProvider,
|
|
||||||
ILogger<CustomOpenAIAnthropicChatCompletionsService> logger)
|
|
||||||
: IAnthropicChatCompletionService
|
|
||||||
{
|
|
||||||
private IChatCompletionService GetChatCompletionService()
|
|
||||||
{
|
|
||||||
return serviceProvider.GetRequiredKeyedService<IChatCompletionService>(nameof(OpenAiChatCompletionsService));
|
|
||||||
}
|
|
||||||
|
|
||||||
public async Task<AnthropicChatCompletionDto> ChatCompletionsAsync(AiModelDescribe aiModelDescribe,
|
|
||||||
AnthropicInput request,
|
|
||||||
CancellationToken cancellationToken = default)
|
|
||||||
{
|
|
||||||
// 转换请求格式:Claude -> OpenAI
|
|
||||||
var openAIRequest = AnthropicToOpenAi.ConvertAnthropicToOpenAi(request);
|
|
||||||
|
|
||||||
if (openAIRequest.Model.StartsWith("gpt-5"))
|
|
||||||
{
|
|
||||||
openAIRequest.MaxCompletionTokens = request.MaxTokens;
|
|
||||||
openAIRequest.MaxTokens = null;
|
|
||||||
}
|
|
||||||
else if (openAIRequest.Model.StartsWith("o3-mini") || openAIRequest.Model.StartsWith("o4-mini"))
|
|
||||||
{
|
|
||||||
openAIRequest.MaxCompletionTokens = request.MaxTokens;
|
|
||||||
openAIRequest.MaxTokens = null;
|
|
||||||
openAIRequest.Temperature = null;
|
|
||||||
}
|
|
||||||
|
|
||||||
// 调用OpenAI服务
|
|
||||||
var openAIResponse =
|
|
||||||
await GetChatCompletionService().CompleteChatAsync(aiModelDescribe,openAIRequest, cancellationToken);
|
|
||||||
|
|
||||||
// 转换响应格式:OpenAI -> Claude
|
|
||||||
var claudeResponse = AnthropicToOpenAi.ConvertOpenAIToClaude(openAIResponse, request);
|
|
||||||
|
|
||||||
return claudeResponse;
|
|
||||||
}
|
|
||||||
|
|
||||||
public async IAsyncEnumerable<(string, AnthropicStreamDto?)> StreamChatCompletionsAsync(AiModelDescribe aiModelDescribe,
|
|
||||||
AnthropicInput request,
|
|
||||||
CancellationToken cancellationToken = default)
|
|
||||||
{
|
|
||||||
var openAIRequest = AnthropicToOpenAi.ConvertAnthropicToOpenAi(request);
|
|
||||||
openAIRequest.Stream = true;
|
|
||||||
|
|
||||||
if (openAIRequest.Model.StartsWith("gpt-5"))
|
|
||||||
{
|
|
||||||
openAIRequest.MaxCompletionTokens = request.MaxTokens;
|
|
||||||
openAIRequest.MaxTokens = null;
|
|
||||||
}
|
|
||||||
else if (openAIRequest.Model.StartsWith("o3-mini") || openAIRequest.Model.StartsWith("o4-mini"))
|
|
||||||
{
|
|
||||||
openAIRequest.MaxCompletionTokens = request.MaxTokens;
|
|
||||||
openAIRequest.MaxTokens = null;
|
|
||||||
openAIRequest.Temperature = null;
|
|
||||||
}
|
|
||||||
|
|
||||||
var messageId = Guid.NewGuid().ToString();
|
|
||||||
var hasStarted = false;
|
|
||||||
var hasTextContentBlockStarted = false;
|
|
||||||
var hasThinkingContentBlockStarted = false;
|
|
||||||
var toolBlocksStarted = new Dictionary<int, bool>(); // 使用索引而不是ID
|
|
||||||
var toolCallIds = new Dictionary<int, string>(); // 存储每个索引对应的ID
|
|
||||||
var toolCallIndexToBlockIndex = new Dictionary<int, int>(); // 工具调用索引到块索引的映射
|
|
||||||
var accumulatedUsage = new AnthropicCompletionDtoUsage();
|
|
||||||
var isFinished = false;
|
|
||||||
var currentContentBlockType = ""; // 跟踪当前内容块类型
|
|
||||||
var currentBlockIndex = 0; // 跟踪当前块索引
|
|
||||||
var lastContentBlockType = ""; // 跟踪最后一个内容块类型,用于确定停止原因
|
|
||||||
|
|
||||||
await foreach (var openAIResponse in GetChatCompletionService().CompleteChatStreamAsync(aiModelDescribe,openAIRequest,
|
|
||||||
cancellationToken))
|
|
||||||
{
|
|
||||||
// 发送message_start事件
|
|
||||||
if (!hasStarted && openAIResponse.Choices?.Count > 0 &&
|
|
||||||
openAIResponse.Choices.Any(x => x.Delta.ToolCalls?.Count > 0) == false)
|
|
||||||
{
|
|
||||||
hasStarted = true;
|
|
||||||
var messageStartEvent = AnthropicToOpenAi.CreateMessageStartEvent(messageId, request.Model);
|
|
||||||
yield return ("message_start", messageStartEvent);
|
|
||||||
}
|
|
||||||
|
|
||||||
// 更新使用情况统计
|
|
||||||
if (openAIResponse.Usage != null)
|
|
||||||
{
|
|
||||||
// 使用最新的token计数(OpenAI通常在最后的响应中提供完整的统计)
|
|
||||||
if (openAIResponse.Usage.PromptTokens.HasValue)
|
|
||||||
{
|
|
||||||
accumulatedUsage.InputTokens = openAIResponse.Usage.PromptTokens.Value;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (openAIResponse.Usage.CompletionTokens.HasValue)
|
|
||||||
{
|
|
||||||
accumulatedUsage.OutputTokens = (int)openAIResponse.Usage.CompletionTokens.Value;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (openAIResponse.Usage.PromptTokensDetails?.CachedTokens.HasValue == true)
|
|
||||||
{
|
|
||||||
accumulatedUsage.CacheReadInputTokens =
|
|
||||||
openAIResponse.Usage.PromptTokensDetails.CachedTokens.Value;
|
|
||||||
}
|
|
||||||
|
|
||||||
// 记录调试信息
|
|
||||||
logger.LogDebug("OpenAI Usage更新: Input={InputTokens}, Output={OutputTokens}, CacheRead={CacheRead}",
|
|
||||||
accumulatedUsage.InputTokens, accumulatedUsage.OutputTokens,
|
|
||||||
accumulatedUsage.CacheReadInputTokens);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (openAIResponse.Choices is { Count: > 0 })
|
|
||||||
{
|
|
||||||
var choice = openAIResponse.Choices.First();
|
|
||||||
|
|
||||||
// 处理内容
|
|
||||||
if (!string.IsNullOrEmpty(choice.Delta?.Content))
|
|
||||||
{
|
|
||||||
// 如果当前有其他类型的内容块在运行,先结束它们
|
|
||||||
if (currentContentBlockType != "text" && !string.IsNullOrEmpty(currentContentBlockType))
|
|
||||||
{
|
|
||||||
var stopEvent = AnthropicToOpenAi.CreateContentBlockStopEvent();
|
|
||||||
stopEvent.Index = currentBlockIndex;
|
|
||||||
yield return ("content_block_stop", stopEvent);
|
|
||||||
currentBlockIndex++; // 切换内容块时增加索引
|
|
||||||
currentContentBlockType = "";
|
|
||||||
}
|
|
||||||
|
|
||||||
// 发送content_block_start事件(仅第一次)
|
|
||||||
if (!hasTextContentBlockStarted || currentContentBlockType != "text")
|
|
||||||
{
|
|
||||||
hasTextContentBlockStarted = true;
|
|
||||||
currentContentBlockType = "text";
|
|
||||||
lastContentBlockType = "text";
|
|
||||||
var contentBlockStartEvent = AnthropicToOpenAi.CreateContentBlockStartEvent();
|
|
||||||
contentBlockStartEvent.Index = currentBlockIndex;
|
|
||||||
yield return ("content_block_start",
|
|
||||||
contentBlockStartEvent);
|
|
||||||
}
|
|
||||||
|
|
||||||
// 发送content_block_delta事件
|
|
||||||
var contentDeltaEvent = AnthropicToOpenAi.CreateContentBlockDeltaEvent(choice.Delta.Content);
|
|
||||||
contentDeltaEvent.Index = currentBlockIndex;
|
|
||||||
yield return ("content_block_delta",
|
|
||||||
contentDeltaEvent);
|
|
||||||
}
|
|
||||||
|
|
||||||
// 处理工具调用
|
|
||||||
if (choice.Delta?.ToolCalls is { Count: > 0 })
|
|
||||||
{
|
|
||||||
foreach (var toolCall in choice.Delta.ToolCalls)
|
|
||||||
{
|
|
||||||
var toolCallIndex = toolCall.Index; // 使用索引来标识工具调用
|
|
||||||
|
|
||||||
// 发送tool_use content_block_start事件
|
|
||||||
if (toolBlocksStarted.TryAdd(toolCallIndex, true))
|
|
||||||
{
|
|
||||||
// 如果当前有文本或thinking内容块在运行,先结束它们
|
|
||||||
if (currentContentBlockType == "text" || currentContentBlockType == "thinking")
|
|
||||||
{
|
|
||||||
var stopEvent = AnthropicToOpenAi.CreateContentBlockStopEvent();
|
|
||||||
stopEvent.Index = currentBlockIndex;
|
|
||||||
yield return ("content_block_stop", stopEvent);
|
|
||||||
currentBlockIndex++; // 增加块索引
|
|
||||||
}
|
|
||||||
// 如果当前有其他工具调用在运行,也需要结束它们
|
|
||||||
else if (currentContentBlockType == "tool_use")
|
|
||||||
{
|
|
||||||
var stopEvent = AnthropicToOpenAi.CreateContentBlockStopEvent();
|
|
||||||
stopEvent.Index = currentBlockIndex;
|
|
||||||
yield return ("content_block_stop", stopEvent);
|
|
||||||
currentBlockIndex++; // 增加块索引
|
|
||||||
}
|
|
||||||
|
|
||||||
currentContentBlockType = "tool_use";
|
|
||||||
lastContentBlockType = "tool_use";
|
|
||||||
|
|
||||||
// 为此工具调用分配一个新的块索引
|
|
||||||
toolCallIndexToBlockIndex[toolCallIndex] = currentBlockIndex;
|
|
||||||
|
|
||||||
// 保存工具调用的ID(如果有的话)
|
|
||||||
if (!string.IsNullOrEmpty(toolCall.Id))
|
|
||||||
{
|
|
||||||
toolCallIds[toolCallIndex] = toolCall.Id;
|
|
||||||
}
|
|
||||||
else if (!toolCallIds.ContainsKey(toolCallIndex))
|
|
||||||
{
|
|
||||||
// 如果没有ID且之前也没有保存过,生成一个新的ID
|
|
||||||
toolCallIds[toolCallIndex] = Guid.NewGuid().ToString();
|
|
||||||
}
|
|
||||||
|
|
||||||
var toolBlockStartEvent = AnthropicToOpenAi.CreateToolBlockStartEvent(
|
|
||||||
toolCallIds[toolCallIndex],
|
|
||||||
toolCall.Function?.Name);
|
|
||||||
toolBlockStartEvent.Index = currentBlockIndex;
|
|
||||||
yield return ("content_block_start",
|
|
||||||
toolBlockStartEvent);
|
|
||||||
}
|
|
||||||
|
|
||||||
// 如果有增量的参数,发送content_block_delta事件
|
|
||||||
if (!string.IsNullOrEmpty(toolCall.Function?.Arguments))
|
|
||||||
{
|
|
||||||
var toolDeltaEvent =
|
|
||||||
AnthropicToOpenAi.CreateToolBlockDeltaEvent(toolCall.Function.Arguments);
|
|
||||||
// 使用该工具调用对应的块索引
|
|
||||||
toolDeltaEvent.Index = toolCallIndexToBlockIndex[toolCallIndex];
|
|
||||||
yield return ("content_block_delta",
|
|
||||||
toolDeltaEvent);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// 处理推理内容
|
|
||||||
if (!string.IsNullOrEmpty(choice.Delta?.ReasoningContent))
|
|
||||||
{
|
|
||||||
// 如果当前有其他类型的内容块在运行,先结束它们
|
|
||||||
if (currentContentBlockType != "thinking" && !string.IsNullOrEmpty(currentContentBlockType))
|
|
||||||
{
|
|
||||||
var stopEvent = AnthropicToOpenAi.CreateContentBlockStopEvent();
|
|
||||||
stopEvent.Index = currentBlockIndex;
|
|
||||||
yield return ("content_block_stop", stopEvent);
|
|
||||||
currentBlockIndex++; // 增加块索引
|
|
||||||
currentContentBlockType = "";
|
|
||||||
}
|
|
||||||
|
|
||||||
// 对于推理内容,也需要发送对应的事件
|
|
||||||
if (!hasThinkingContentBlockStarted || currentContentBlockType != "thinking")
|
|
||||||
{
|
|
||||||
hasThinkingContentBlockStarted = true;
|
|
||||||
currentContentBlockType = "thinking";
|
|
||||||
lastContentBlockType = "thinking";
|
|
||||||
var thinkingBlockStartEvent = AnthropicToOpenAi.CreateThinkingBlockStartEvent();
|
|
||||||
thinkingBlockStartEvent.Index = currentBlockIndex;
|
|
||||||
yield return ("content_block_start",
|
|
||||||
thinkingBlockStartEvent);
|
|
||||||
}
|
|
||||||
|
|
||||||
var thinkingDeltaEvent =
|
|
||||||
AnthropicToOpenAi.CreateThinkingBlockDeltaEvent(choice.Delta.ReasoningContent);
|
|
||||||
thinkingDeltaEvent.Index = currentBlockIndex;
|
|
||||||
yield return ("content_block_delta",
|
|
||||||
thinkingDeltaEvent);
|
|
||||||
}
|
|
||||||
|
|
||||||
// 处理结束
|
|
||||||
if (!string.IsNullOrEmpty(choice.FinishReason) && !isFinished)
|
|
||||||
{
|
|
||||||
isFinished = true;
|
|
||||||
|
|
||||||
// 发送content_block_stop事件(如果有活跃的内容块)
|
|
||||||
if (!string.IsNullOrEmpty(currentContentBlockType))
|
|
||||||
{
|
|
||||||
var contentBlockStopEvent = AnthropicToOpenAi.CreateContentBlockStopEvent();
|
|
||||||
contentBlockStopEvent.Index = currentBlockIndex;
|
|
||||||
yield return ("content_block_stop",
|
|
||||||
contentBlockStopEvent);
|
|
||||||
}
|
|
||||||
|
|
||||||
// 发送message_delta事件
|
|
||||||
var messageDeltaEvent = AnthropicToOpenAi.CreateMessageDeltaEvent(
|
|
||||||
AnthropicToOpenAi.GetStopReasonByLastContentType(choice.FinishReason, lastContentBlockType),
|
|
||||||
accumulatedUsage);
|
|
||||||
|
|
||||||
// 记录最终Usage统计
|
|
||||||
logger.LogDebug(
|
|
||||||
"流式响应结束,最终Usage: Input={InputTokens}, Output={OutputTokens}, CacheRead={CacheRead}",
|
|
||||||
accumulatedUsage.InputTokens, accumulatedUsage.OutputTokens,
|
|
||||||
accumulatedUsage.CacheReadInputTokens);
|
|
||||||
|
|
||||||
yield return ("message_delta",
|
|
||||||
messageDeltaEvent);
|
|
||||||
|
|
||||||
// 发送message_stop事件
|
|
||||||
var messageStopEvent = AnthropicToOpenAi.CreateMessageStopEvent();
|
|
||||||
yield return ("message_stop",
|
|
||||||
messageStopEvent);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// 确保流正确结束
|
|
||||||
if (!isFinished)
|
|
||||||
{
|
|
||||||
if (!string.IsNullOrEmpty(currentContentBlockType))
|
|
||||||
{
|
|
||||||
var contentBlockStopEvent = AnthropicToOpenAi.CreateContentBlockStopEvent();
|
|
||||||
contentBlockStopEvent.Index = currentBlockIndex;
|
|
||||||
yield return ("content_block_stop",
|
|
||||||
contentBlockStopEvent);
|
|
||||||
}
|
|
||||||
|
|
||||||
var messageDeltaEvent =
|
|
||||||
AnthropicToOpenAi.CreateMessageDeltaEvent(
|
|
||||||
AnthropicToOpenAi.GetStopReasonByLastContentType("end_turn", lastContentBlockType),
|
|
||||||
accumulatedUsage);
|
|
||||||
yield return ("message_delta", messageDeltaEvent);
|
|
||||||
|
|
||||||
var messageStopEvent = AnthropicToOpenAi.CreateMessageStopEvent();
|
|
||||||
yield return ("message_stop",
|
|
||||||
messageStopEvent);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -130,7 +130,7 @@ public sealed class OpenAiChatCompletionsService(ILogger<OpenAiChatCompletionsSe
|
|||||||
using var openai =
|
using var openai =
|
||||||
Activity.Current?.Source.StartActivity("OpenAI 对话补全");
|
Activity.Current?.Source.StartActivity("OpenAI 对话补全");
|
||||||
|
|
||||||
var response = await HttpClientFactory.GetHttpClient(options.Endpoint).PostJsonAsync(
|
var response = await httpClientFactory.CreateClient().PostJsonAsync(
|
||||||
options?.Endpoint.TrimEnd('/') + "/chat/completions",
|
options?.Endpoint.TrimEnd('/') + "/chat/completions",
|
||||||
chatCompletionCreate, options.ApiKey).ConfigureAwait(false);
|
chatCompletionCreate, options.ApiKey).ConfigureAwait(false);
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,123 @@
|
|||||||
|
using System.Diagnostics;
|
||||||
|
using System.Net;
|
||||||
|
using System.Net.Http.Json;
|
||||||
|
using System.Text.Json;
|
||||||
|
using Microsoft.Extensions.Logging;
|
||||||
|
using Yi.Framework.AiHub.Domain.AiGateWay.Exceptions;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Dtos;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi.Responses;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Domain.AiGateWay.Impl.ThorCustomOpenAI.Chats;
|
||||||
|
|
||||||
|
public class OpenAiResponseService(ILogger<OpenAiResponseService> logger,IHttpClientFactory httpClientFactory):IOpenAiResponseService
|
||||||
|
{
|
||||||
|
|
||||||
|
public async IAsyncEnumerable<(string, JsonElement?)> ResponsesStreamAsync(AiModelDescribe options, OpenAiResponsesInput input,
|
||||||
|
CancellationToken cancellationToken)
|
||||||
|
{
|
||||||
|
using var openai =
|
||||||
|
Activity.Current?.Source.StartActivity("OpenAi 响应");
|
||||||
|
|
||||||
|
|
||||||
|
var client = httpClientFactory.CreateClient();
|
||||||
|
|
||||||
|
var response = await client.HttpRequestRaw(options.Endpoint.TrimEnd('/') + "/responses", input, options.ApiKey);
|
||||||
|
|
||||||
|
openai?.SetTag("Model", input.Model);
|
||||||
|
openai?.SetTag("Response", response.StatusCode.ToString());
|
||||||
|
|
||||||
|
// 大于等于400的状态码都认为是异常
|
||||||
|
if (response.StatusCode >= HttpStatusCode.BadRequest)
|
||||||
|
{
|
||||||
|
var error = await response.Content.ReadAsStringAsync(cancellationToken).ConfigureAwait(false);
|
||||||
|
logger.LogError("OpenAI响应异常 请求地址:{Address}, StatusCode: {StatusCode} Response: {Response}",
|
||||||
|
options.Endpoint,
|
||||||
|
response.StatusCode, error);
|
||||||
|
|
||||||
|
throw new Exception("OpenAI响应异常" + response.StatusCode);
|
||||||
|
}
|
||||||
|
|
||||||
|
using var stream = new StreamReader(await response.Content.ReadAsStreamAsync(cancellationToken));
|
||||||
|
|
||||||
|
using StreamReader reader = new(await response.Content.ReadAsStreamAsync(cancellationToken));
|
||||||
|
string? line = string.Empty;
|
||||||
|
|
||||||
|
string? data = null;
|
||||||
|
string eventType = string.Empty;
|
||||||
|
|
||||||
|
while ((line = await reader.ReadLineAsync(cancellationToken).ConfigureAwait(false)) != null)
|
||||||
|
{
|
||||||
|
line += Environment.NewLine;
|
||||||
|
|
||||||
|
if (line.StartsWith('{'))
|
||||||
|
{
|
||||||
|
logger.LogInformation("OpenAI响应异常 , StatusCode: {StatusCode} Response: {Response}", response.StatusCode,
|
||||||
|
line);
|
||||||
|
|
||||||
|
throw new Exception("OpenAI响应异常" + line);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (string.IsNullOrWhiteSpace(line))
|
||||||
|
{
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (line.StartsWith("event:"))
|
||||||
|
{
|
||||||
|
eventType = line;
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!line.StartsWith(OpenAIConstant.Data)) continue;
|
||||||
|
|
||||||
|
data = line[OpenAIConstant.Data.Length..].Trim();
|
||||||
|
|
||||||
|
var result = JsonSerializer.Deserialize<JsonElement>(data,
|
||||||
|
ThorJsonSerializer.DefaultOptions);
|
||||||
|
|
||||||
|
yield return (eventType, result);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public async Task<OpenAiResponsesOutput> ResponsesAsync(AiModelDescribe options, OpenAiResponsesInput chatCompletionCreate,
|
||||||
|
CancellationToken cancellationToken)
|
||||||
|
{
|
||||||
|
using var openai =
|
||||||
|
Activity.Current?.Source.StartActivity("OpenAI 响应");
|
||||||
|
|
||||||
|
var response = await httpClientFactory.CreateClient().PostJsonAsync(
|
||||||
|
options?.Endpoint.TrimEnd('/') + "/responses",
|
||||||
|
chatCompletionCreate, options.ApiKey).ConfigureAwait(false);
|
||||||
|
|
||||||
|
openai?.SetTag("Model", chatCompletionCreate.Model);
|
||||||
|
openai?.SetTag("Response", response.StatusCode.ToString());
|
||||||
|
|
||||||
|
if (response.StatusCode == HttpStatusCode.Unauthorized)
|
||||||
|
{
|
||||||
|
throw new BusinessException("渠道未登录,请联系管理人员", "401");
|
||||||
|
}
|
||||||
|
|
||||||
|
// 如果限流则抛出限流异常
|
||||||
|
if (response.StatusCode == HttpStatusCode.TooManyRequests)
|
||||||
|
{
|
||||||
|
throw new ThorRateLimitException();
|
||||||
|
}
|
||||||
|
|
||||||
|
// 大于等于400的状态码都认为是异常
|
||||||
|
if (response.StatusCode >= HttpStatusCode.BadRequest)
|
||||||
|
{
|
||||||
|
var error = await response.Content.ReadAsStringAsync(cancellationToken).ConfigureAwait(false);
|
||||||
|
logger.LogError("OpenAI 响应异常 请求地址:{Address}, StatusCode: {StatusCode} Response: {Response}", options.Endpoint,
|
||||||
|
response.StatusCode, error);
|
||||||
|
|
||||||
|
throw new BusinessException("OpenAI响应异常", response.StatusCode.ToString());
|
||||||
|
}
|
||||||
|
|
||||||
|
var result =
|
||||||
|
await response.Content.ReadFromJsonAsync<OpenAiResponsesOutput>(
|
||||||
|
cancellationToken: cancellationToken).ConfigureAwait(false);
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,57 @@
|
|||||||
|
using SqlSugar;
|
||||||
|
using Volo.Abp.Domain.Entities.Auditing;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Domain.Entities;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 公告日志
|
||||||
|
/// </summary>
|
||||||
|
[SugarTable("Ai_Announcement")]
|
||||||
|
public class AnnouncementAggregateRoot : FullAuditedAggregateRoot<Guid>
|
||||||
|
{
|
||||||
|
public AnnouncementAggregateRoot()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 标题
|
||||||
|
/// </summary>
|
||||||
|
public string Title { get; set; } = string.Empty;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 内容列表(JSON格式存储)
|
||||||
|
/// </summary>
|
||||||
|
[SugarColumn(IsJson = true, IsNullable = false)]
|
||||||
|
public List<string> Content { get; set; } = new List<string>();
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 备注
|
||||||
|
/// </summary>
|
||||||
|
public string? Remark { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 图片url
|
||||||
|
/// </summary>
|
||||||
|
public string? ImageUrl { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 开始时间(系统公告时间、活动开始时间)
|
||||||
|
/// </summary>
|
||||||
|
public DateTime StartTime { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 活动结束时间
|
||||||
|
/// </summary>
|
||||||
|
public DateTime? EndTime { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 公告类型(系统、活动)
|
||||||
|
/// </summary>
|
||||||
|
public AnnouncementTypeEnum Type{ get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 跳转链接
|
||||||
|
/// </summary>
|
||||||
|
public string? Url { get; set; }
|
||||||
|
}
|
||||||
@@ -25,9 +25,8 @@ public class CardFlipTaskAggregateRoot : FullAuditedAggregateRoot<Guid>
|
|||||||
BonusFlipsUsed = 0;
|
BonusFlipsUsed = 0;
|
||||||
InviteFlipsUsed = 0;
|
InviteFlipsUsed = 0;
|
||||||
IsFirstFlipDone = false;
|
IsFirstFlipDone = false;
|
||||||
HasNinthReward = false;
|
|
||||||
HasTenthReward = false;
|
|
||||||
FlippedOrder = new List<int>();
|
FlippedOrder = new List<int>();
|
||||||
|
WinRecords = new Dictionary<int, long>();
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
@@ -66,34 +65,10 @@ public class CardFlipTaskAggregateRoot : FullAuditedAggregateRoot<Guid>
|
|||||||
public bool IsFirstFlipDone { get; set; }
|
public bool IsFirstFlipDone { get; set; }
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 是否已获得第8次奖励
|
/// 中奖记录(以翻牌顺序为key,例如第1次翻牌中奖则key为1,奖励金额为value)
|
||||||
/// </summary>
|
/// </summary>
|
||||||
public bool HasEighthReward { get; set; }
|
[SugarColumn(IsJson = true, IsNullable = true)]
|
||||||
|
public Dictionary<int, long>? WinRecords { get; set; }
|
||||||
/// <summary>
|
|
||||||
/// 第8次奖励金额(100-300w)
|
|
||||||
/// </summary>
|
|
||||||
public long? EighthRewardAmount { get; set; }
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 是否已获得第9次奖励
|
|
||||||
/// </summary>
|
|
||||||
public bool HasNinthReward { get; set; }
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 第9次奖励金额(100-500w)
|
|
||||||
/// </summary>
|
|
||||||
public long? NinthRewardAmount { get; set; }
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 是否已获得第10次奖励
|
|
||||||
/// </summary>
|
|
||||||
public bool HasTenthReward { get; set; }
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 第10次奖励金额(100-1000w)
|
|
||||||
/// </summary>
|
|
||||||
public long? TenthRewardAmount { get; set; }
|
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 备注信息
|
/// 备注信息
|
||||||
@@ -135,33 +110,17 @@ public class CardFlipTaskAggregateRoot : FullAuditedAggregateRoot<Guid>
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 记录第8次奖励
|
/// 记录中奖
|
||||||
/// </summary>
|
/// </summary>
|
||||||
|
/// <param name="flipCount">第几次翻牌(1-10)</param>
|
||||||
/// <param name="amount">奖励金额</param>
|
/// <param name="amount">奖励金额</param>
|
||||||
public void SetEighthReward(long amount)
|
public void SetWinReward(int flipCount, long amount)
|
||||||
{
|
{
|
||||||
HasEighthReward = true;
|
if (WinRecords == null)
|
||||||
EighthRewardAmount = amount;
|
{
|
||||||
}
|
WinRecords = new Dictionary<int, long>();
|
||||||
|
}
|
||||||
/// <summary>
|
WinRecords[flipCount] = amount;
|
||||||
/// 记录第9次奖励
|
|
||||||
/// </summary>
|
|
||||||
/// <param name="amount">奖励金额</param>
|
|
||||||
public void SetNinthReward(long amount)
|
|
||||||
{
|
|
||||||
HasNinthReward = true;
|
|
||||||
NinthRewardAmount = amount;
|
|
||||||
}
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 记录第10次奖励
|
|
||||||
/// </summary>
|
|
||||||
/// <param name="amount">奖励金额</param>
|
|
||||||
public void SetTenthReward(long amount)
|
|
||||||
{
|
|
||||||
HasTenthReward = true;
|
|
||||||
TenthRewardAmount = amount;
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -20,10 +20,11 @@ public class MessageAggregateRoot : FullAuditedAggregateRoot<Guid>
|
|||||||
}
|
}
|
||||||
|
|
||||||
public MessageAggregateRoot(Guid? userId, Guid? sessionId, string content, string role, string modelId,
|
public MessageAggregateRoot(Guid? userId, Guid? sessionId, string content, string role, string modelId,
|
||||||
ThorUsageResponse? tokenUsage)
|
ThorUsageResponse? tokenUsage, Guid? tokenId = null)
|
||||||
{
|
{
|
||||||
UserId = userId;
|
UserId = userId;
|
||||||
SessionId = sessionId;
|
SessionId = sessionId;
|
||||||
|
TokenId = tokenId ?? Guid.Empty;
|
||||||
//如果没有会话,不存储对话内容
|
//如果没有会话,不存储对话内容
|
||||||
Content = sessionId is null ? null : content;
|
Content = sessionId is null ? null : content;
|
||||||
Role = role;
|
Role = role;
|
||||||
@@ -59,6 +60,11 @@ public class MessageAggregateRoot : FullAuditedAggregateRoot<Guid>
|
|||||||
public Guid? UserId { get; set; }
|
public Guid? UserId { get; set; }
|
||||||
public Guid? SessionId { get; set; }
|
public Guid? SessionId { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Token密钥Id(通过API调用时记录,Web调用为Guid.Empty)
|
||||||
|
/// </summary>
|
||||||
|
public Guid TokenId { get; set; }
|
||||||
|
|
||||||
[SugarColumn(ColumnDataType = StaticConfig.CodeFirst_BigString)]
|
[SugarColumn(ColumnDataType = StaticConfig.CodeFirst_BigString)]
|
||||||
public string? Content { get; set; }
|
public string? Content { get; set; }
|
||||||
|
|
||||||
|
|||||||
@@ -19,8 +19,6 @@ public class InviteCodeAggregateRoot : FullAuditedAggregateRoot<Guid>
|
|||||||
{
|
{
|
||||||
UserId = userId;
|
UserId = userId;
|
||||||
Code = code;
|
Code = code;
|
||||||
IsUsed = false;
|
|
||||||
IsUserInvited = false;
|
|
||||||
UsedCount = 0;
|
UsedCount = 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -36,53 +34,14 @@ public class InviteCodeAggregateRoot : FullAuditedAggregateRoot<Guid>
|
|||||||
public string Code { get; set; } = string.Empty;
|
public string Code { get; set; } = string.Empty;
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 是否已被使用(一个邀请码只能被使用一次)
|
/// 被使用次数(统计用,一个邀请码可以被多人使用)
|
||||||
/// </summary>
|
|
||||||
public bool IsUsed { get; set; }
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 邀请码拥有者是否已被他人邀请(被邀请后不可再提供邀请码)
|
|
||||||
/// </summary>
|
|
||||||
public bool IsUserInvited { get; set; }
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 被使用次数(统计用)
|
|
||||||
/// </summary>
|
/// </summary>
|
||||||
public int UsedCount { get; set; }
|
public int UsedCount { get; set; }
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 使用时间
|
|
||||||
/// </summary>
|
|
||||||
public DateTime? UsedTime { get; set; }
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 使用人ID
|
|
||||||
/// </summary>
|
|
||||||
public Guid? UsedByUserId { get; set; }
|
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 备注信息
|
/// 备注信息
|
||||||
/// </summary>
|
/// </summary>
|
||||||
[SugarColumn(Length = 500, IsNullable = true)]
|
[SugarColumn(Length = 500, IsNullable = true)]
|
||||||
public string? Remark { get; set; }
|
public string? Remark { get; set; }
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 标记邀请码已被使用
|
|
||||||
/// </summary>
|
|
||||||
/// <param name="usedByUserId">使用者ID</param>
|
|
||||||
public void MarkAsUsed(Guid usedByUserId)
|
|
||||||
{
|
|
||||||
IsUsed = true;
|
|
||||||
UsedTime = DateTime.Now;
|
|
||||||
UsedByUserId = usedByUserId;
|
|
||||||
UsedCount++;
|
|
||||||
}
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 标记用户已被邀请
|
|
||||||
/// </summary>
|
|
||||||
public void MarkUserAsInvited()
|
|
||||||
{
|
|
||||||
IsUserInvited = true;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -60,4 +60,24 @@ public class AiModelEntity : Entity<Guid>, IOrderNum, ISoftDelete
|
|||||||
/// 模型Api类型,现支持同一个模型id,多种接口格式
|
/// 模型Api类型,现支持同一个模型id,多种接口格式
|
||||||
/// </summary>
|
/// </summary>
|
||||||
public ModelApiTypeEnum ModelApiType { get; set; }
|
public ModelApiTypeEnum ModelApiType { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型倍率
|
||||||
|
/// </summary>
|
||||||
|
public decimal Multiplier { get; set; } = 1;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型显示倍率
|
||||||
|
/// </summary>
|
||||||
|
public decimal MultiplierShow { get; set; } = 1;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 供应商分组名称(如:OpenAI、Anthropic、Google等)
|
||||||
|
/// </summary>
|
||||||
|
public string? ProviderName { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型图标URL
|
||||||
|
/// </summary>
|
||||||
|
public string? IconUrl { get; set; }
|
||||||
}
|
}
|
||||||
@@ -5,27 +5,84 @@ using Volo.Abp.Domain.Entities.Auditing;
|
|||||||
namespace Yi.Framework.AiHub.Domain.Entities.OpenApi;
|
namespace Yi.Framework.AiHub.Domain.Entities.OpenApi;
|
||||||
|
|
||||||
[SugarTable("Ai_Token")]
|
[SugarTable("Ai_Token")]
|
||||||
|
[SugarIndex($"index_{{table}}_{nameof(UserId)}", nameof(UserId), OrderByType.Asc)]
|
||||||
public class TokenAggregateRoot : FullAuditedAggregateRoot<Guid>
|
public class TokenAggregateRoot : FullAuditedAggregateRoot<Guid>
|
||||||
{
|
{
|
||||||
public TokenAggregateRoot()
|
public TokenAggregateRoot()
|
||||||
{
|
{
|
||||||
}
|
}
|
||||||
|
|
||||||
public TokenAggregateRoot(Guid userId)
|
public TokenAggregateRoot(Guid userId, string name)
|
||||||
{
|
{
|
||||||
this.UserId = userId;
|
UserId = userId;
|
||||||
this.Token = GenerateToken();
|
Name = name;
|
||||||
|
Token = GenerateToken();
|
||||||
|
IsDisabled = false;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Token密钥
|
||||||
|
/// </summary>
|
||||||
public string Token { get; set; }
|
public string Token { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 用户Id
|
||||||
|
/// </summary>
|
||||||
public Guid UserId { get; set; }
|
public Guid UserId { get; set; }
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 重置token
|
/// 名称
|
||||||
/// </summary>
|
/// </summary>
|
||||||
public void ResetToken()
|
[SugarColumn(Length = 100)]
|
||||||
|
public string Name { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 过期时间(空为永不过期)
|
||||||
|
/// </summary>
|
||||||
|
public DateTime? ExpireTime { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 尊享包额度限制(空为不限制)
|
||||||
|
/// </summary>
|
||||||
|
public long? PremiumQuotaLimit { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 是否禁用
|
||||||
|
/// </summary>
|
||||||
|
public bool IsDisabled { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 检查Token是否可用
|
||||||
|
/// </summary>
|
||||||
|
public bool IsAvailable()
|
||||||
{
|
{
|
||||||
this.Token = GenerateToken();
|
if (IsDisabled)
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (ExpireTime.HasValue && ExpireTime.Value < DateTime.Now)
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 禁用Token
|
||||||
|
/// </summary>
|
||||||
|
public void Disable()
|
||||||
|
{
|
||||||
|
IsDisabled = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 启用Token
|
||||||
|
/// </summary>
|
||||||
|
public void Enable()
|
||||||
|
{
|
||||||
|
IsDisabled = false;
|
||||||
}
|
}
|
||||||
|
|
||||||
private string GenerateToken(int length = 36)
|
private string GenerateToken(int length = 36)
|
||||||
|
|||||||
@@ -86,14 +86,14 @@ public class PremiumPackageAggregateRoot : FullAuditedAggregateRoot<Guid>
|
|||||||
/// 检查是否可用
|
/// 检查是否可用
|
||||||
/// </summary>
|
/// </summary>
|
||||||
/// <returns>是否可用</returns>
|
/// <returns>是否可用</returns>
|
||||||
public bool IsAvailable()
|
public bool IsAvailable(bool isVerifyRemainingToken=true)
|
||||||
{
|
{
|
||||||
if (!IsActive)
|
if (!IsActive)
|
||||||
{
|
{
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (RemainingTokens <= 0)
|
if (isVerifyRemainingToken&&RemainingTokens <= 0)
|
||||||
{
|
{
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -7,16 +7,22 @@ namespace Yi.Framework.AiHub.Domain.Entities;
|
|||||||
/// 用量统计
|
/// 用量统计
|
||||||
/// </summary>
|
/// </summary>
|
||||||
[SugarTable("Ai_UsageStatistics")]
|
[SugarTable("Ai_UsageStatistics")]
|
||||||
|
[SugarIndex($"index_{{table}}_{nameof(UserId)}_{nameof(ModelId)}_{nameof(TokenId)}",
|
||||||
|
nameof(UserId), OrderByType.Asc,
|
||||||
|
nameof(ModelId), OrderByType.Asc,
|
||||||
|
nameof(TokenId), OrderByType.Asc
|
||||||
|
)]
|
||||||
public class UsageStatisticsAggregateRoot : FullAuditedAggregateRoot<Guid>
|
public class UsageStatisticsAggregateRoot : FullAuditedAggregateRoot<Guid>
|
||||||
{
|
{
|
||||||
public UsageStatisticsAggregateRoot()
|
public UsageStatisticsAggregateRoot()
|
||||||
{
|
{
|
||||||
}
|
}
|
||||||
|
|
||||||
public UsageStatisticsAggregateRoot(Guid? userId, string modelId)
|
public UsageStatisticsAggregateRoot(Guid? userId, string modelId, Guid tokenId)
|
||||||
{
|
{
|
||||||
UserId = userId;
|
UserId = userId;
|
||||||
ModelId = modelId;
|
ModelId = modelId;
|
||||||
|
TokenId = tokenId;
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
@@ -29,6 +35,11 @@ public class UsageStatisticsAggregateRoot : FullAuditedAggregateRoot<Guid>
|
|||||||
/// </summary>
|
/// </summary>
|
||||||
public string ModelId { get; set; }
|
public string ModelId { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Token密钥Id(通过API调用时记录,Web调用为Guid.Empty)
|
||||||
|
/// </summary>
|
||||||
|
public Guid TokenId { get; set; }
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 对话次数
|
/// 对话次数
|
||||||
/// </summary>
|
/// </summary>
|
||||||
|
|||||||
@@ -7,6 +7,7 @@ using Microsoft.AspNetCore.Http;
|
|||||||
using Microsoft.Extensions.DependencyInjection;
|
using Microsoft.Extensions.DependencyInjection;
|
||||||
using Microsoft.Extensions.Logging;
|
using Microsoft.Extensions.Logging;
|
||||||
using Newtonsoft.Json;
|
using Newtonsoft.Json;
|
||||||
|
using Newtonsoft.Json.Linq;
|
||||||
using Newtonsoft.Json.Serialization;
|
using Newtonsoft.Json.Serialization;
|
||||||
using Volo.Abp.Domain.Services;
|
using Volo.Abp.Domain.Services;
|
||||||
using Yi.Framework.AiHub.Domain.AiGateWay;
|
using Yi.Framework.AiHub.Domain.AiGateWay;
|
||||||
@@ -18,7 +19,9 @@ using Yi.Framework.AiHub.Domain.Shared.Dtos.Anthropic;
|
|||||||
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi;
|
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi;
|
||||||
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi.Embeddings;
|
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi.Embeddings;
|
||||||
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi.Images;
|
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi.Images;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi.Responses;
|
||||||
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Extensions;
|
||||||
using Yi.Framework.Core.Extensions;
|
using Yi.Framework.Core.Extensions;
|
||||||
using Yi.Framework.SqlSugarCore.Abstractions;
|
using Yi.Framework.SqlSugarCore.Abstractions;
|
||||||
using JsonSerializer = System.Text.Json.JsonSerializer;
|
using JsonSerializer = System.Text.Json.JsonSerializer;
|
||||||
@@ -77,7 +80,8 @@ public class AiGateWayManager : DomainService
|
|||||||
ModelName = model.Name,
|
ModelName = model.Name,
|
||||||
Description = model.Description,
|
Description = model.Description,
|
||||||
AppExtraUrl = app.ExtraUrl,
|
AppExtraUrl = app.ExtraUrl,
|
||||||
ModelExtraInfo = model.ExtraInfo
|
ModelExtraInfo = model.ExtraInfo,
|
||||||
|
Multiplier = model.Multiplier
|
||||||
})
|
})
|
||||||
.FirstAsync();
|
.FirstAsync();
|
||||||
if (aiModelDescribe is null)
|
if (aiModelDescribe is null)
|
||||||
@@ -89,28 +93,6 @@ public class AiGateWayManager : DomainService
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 聊天完成-流式
|
|
||||||
/// </summary>
|
|
||||||
/// <param name="request"></param>
|
|
||||||
/// <param name="cancellationToken"></param>
|
|
||||||
/// <returns></returns>
|
|
||||||
public async IAsyncEnumerable<ThorChatCompletionsResponse> CompleteChatStreamAsync(
|
|
||||||
ThorChatCompletionsRequest request,
|
|
||||||
[EnumeratorCancellation] CancellationToken cancellationToken)
|
|
||||||
{
|
|
||||||
_specialCompatible.Compatible(request);
|
|
||||||
var modelDescribe = await GetModelAsync(ModelApiTypeEnum.OpenAi, request.Model);
|
|
||||||
var chatService =
|
|
||||||
LazyServiceProvider.GetRequiredKeyedService<IChatCompletionService>(modelDescribe.HandlerName);
|
|
||||||
|
|
||||||
await foreach (var result in chatService.CompleteChatStreamAsync(modelDescribe, request, cancellationToken))
|
|
||||||
{
|
|
||||||
yield return result;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 聊天完成-非流式
|
/// 聊天完成-非流式
|
||||||
/// </summary>
|
/// </summary>
|
||||||
@@ -118,12 +100,14 @@ public class AiGateWayManager : DomainService
|
|||||||
/// <param name="request"></param>
|
/// <param name="request"></param>
|
||||||
/// <param name="userId"></param>
|
/// <param name="userId"></param>
|
||||||
/// <param name="sessionId"></param>
|
/// <param name="sessionId"></param>
|
||||||
|
/// <param name="tokenId">Token Id(Web端传null或Guid.Empty)</param>
|
||||||
/// <param name="cancellationToken"></param>
|
/// <param name="cancellationToken"></param>
|
||||||
/// <returns></returns>
|
/// <returns></returns>
|
||||||
public async Task CompleteChatForStatisticsAsync(HttpContext httpContext,
|
public async Task CompleteChatForStatisticsAsync(HttpContext httpContext,
|
||||||
ThorChatCompletionsRequest request,
|
ThorChatCompletionsRequest request,
|
||||||
Guid? userId = null,
|
Guid? userId = null,
|
||||||
Guid? sessionId = null,
|
Guid? sessionId = null,
|
||||||
|
Guid? tokenId = null,
|
||||||
CancellationToken cancellationToken = default)
|
CancellationToken cancellationToken = default)
|
||||||
{
|
{
|
||||||
_specialCompatible.Compatible(request);
|
_specialCompatible.Compatible(request);
|
||||||
@@ -134,6 +118,7 @@ public class AiGateWayManager : DomainService
|
|||||||
var chatService =
|
var chatService =
|
||||||
LazyServiceProvider.GetRequiredKeyedService<IChatCompletionService>(modelDescribe.HandlerName);
|
LazyServiceProvider.GetRequiredKeyedService<IChatCompletionService>(modelDescribe.HandlerName);
|
||||||
var data = await chatService.CompleteChatAsync(modelDescribe, request, cancellationToken);
|
var data = await chatService.CompleteChatAsync(modelDescribe, request, cancellationToken);
|
||||||
|
data.SupplementalMultiplier(modelDescribe.Multiplier);
|
||||||
if (userId is not null)
|
if (userId is not null)
|
||||||
{
|
{
|
||||||
await _aiMessageManager.CreateUserMessageAsync(userId.Value, sessionId,
|
await _aiMessageManager.CreateUserMessageAsync(userId.Value, sessionId,
|
||||||
@@ -142,7 +127,7 @@ public class AiGateWayManager : DomainService
|
|||||||
Content = sessionId is null ? "不予存储" : request.Messages?.LastOrDefault().Content ?? string.Empty,
|
Content = sessionId is null ? "不予存储" : request.Messages?.LastOrDefault().Content ?? string.Empty,
|
||||||
ModelId = request.Model,
|
ModelId = request.Model,
|
||||||
TokenUsage = data.Usage,
|
TokenUsage = data.Usage,
|
||||||
});
|
}, tokenId);
|
||||||
|
|
||||||
await _aiMessageManager.CreateSystemMessageAsync(userId.Value, sessionId,
|
await _aiMessageManager.CreateSystemMessageAsync(userId.Value, sessionId,
|
||||||
new MessageInputDto
|
new MessageInputDto
|
||||||
@@ -151,14 +136,25 @@ public class AiGateWayManager : DomainService
|
|||||||
sessionId is null ? "不予存储" : data.Choices?.FirstOrDefault()?.Delta.Content ?? string.Empty,
|
sessionId is null ? "不予存储" : data.Choices?.FirstOrDefault()?.Delta.Content ?? string.Empty,
|
||||||
ModelId = request.Model,
|
ModelId = request.Model,
|
||||||
TokenUsage = data.Usage
|
TokenUsage = data.Usage
|
||||||
});
|
}, tokenId);
|
||||||
|
|
||||||
await _usageStatisticsManager.SetUsageAsync(userId.Value, request.Model, data.Usage);
|
await _usageStatisticsManager.SetUsageAsync(userId.Value, request.Model, data.Usage, tokenId);
|
||||||
|
|
||||||
|
// 扣减尊享token包用量
|
||||||
|
if (PremiumPackageConst.ModeIds.Contains(request.Model))
|
||||||
|
{
|
||||||
|
var totalTokens = data.Usage?.TotalTokens ?? 0;
|
||||||
|
if (totalTokens > 0)
|
||||||
|
{
|
||||||
|
await PremiumPackageManager.TryConsumeTokensAsync(userId.Value, totalTokens);
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
await response.WriteAsJsonAsync(data, cancellationToken);
|
await response.WriteAsJsonAsync(data, cancellationToken);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 聊天完成-缓存处理
|
/// 聊天完成-缓存处理
|
||||||
/// </summary>
|
/// </summary>
|
||||||
@@ -166,6 +162,7 @@ public class AiGateWayManager : DomainService
|
|||||||
/// <param name="request"></param>
|
/// <param name="request"></param>
|
||||||
/// <param name="userId"></param>
|
/// <param name="userId"></param>
|
||||||
/// <param name="sessionId"></param>
|
/// <param name="sessionId"></param>
|
||||||
|
/// <param name="tokenId">Token Id(Web端传null或Guid.Empty)</param>
|
||||||
/// <param name="cancellationToken"></param>
|
/// <param name="cancellationToken"></param>
|
||||||
/// <returns></returns>
|
/// <returns></returns>
|
||||||
public async Task CompleteChatStreamForStatisticsAsync(
|
public async Task CompleteChatStreamForStatisticsAsync(
|
||||||
@@ -173,6 +170,7 @@ public class AiGateWayManager : DomainService
|
|||||||
ThorChatCompletionsRequest request,
|
ThorChatCompletionsRequest request,
|
||||||
Guid? userId = null,
|
Guid? userId = null,
|
||||||
Guid? sessionId = null,
|
Guid? sessionId = null,
|
||||||
|
Guid? tokenId = null,
|
||||||
CancellationToken cancellationToken = default)
|
CancellationToken cancellationToken = default)
|
||||||
{
|
{
|
||||||
var response = httpContext.Response;
|
var response = httpContext.Response;
|
||||||
@@ -182,8 +180,12 @@ public class AiGateWayManager : DomainService
|
|||||||
response.Headers.TryAdd("Connection", "keep-alive");
|
response.Headers.TryAdd("Connection", "keep-alive");
|
||||||
|
|
||||||
|
|
||||||
var gateWay = LazyServiceProvider.GetRequiredService<AiGateWayManager>();
|
_specialCompatible.Compatible(request);
|
||||||
var completeChatResponse = gateWay.CompleteChatStreamAsync(request, cancellationToken);
|
var modelDescribe = await GetModelAsync(ModelApiTypeEnum.OpenAi, request.Model);
|
||||||
|
var chatService =
|
||||||
|
LazyServiceProvider.GetRequiredKeyedService<IChatCompletionService>(modelDescribe.HandlerName);
|
||||||
|
|
||||||
|
var completeChatResponse = chatService.CompleteChatStreamAsync(modelDescribe,request, cancellationToken);
|
||||||
var tokenUsage = new ThorUsageResponse();
|
var tokenUsage = new ThorUsageResponse();
|
||||||
|
|
||||||
//缓存队列算法
|
//缓存队列算法
|
||||||
@@ -225,7 +227,8 @@ public class AiGateWayManager : DomainService
|
|||||||
{
|
{
|
||||||
await foreach (var data in completeChatResponse)
|
await foreach (var data in completeChatResponse)
|
||||||
{
|
{
|
||||||
if (data.Usage is not null&&(data.Usage.CompletionTokens>0||data.Usage.OutputTokens>0))
|
data.SupplementalMultiplier(modelDescribe.Multiplier);
|
||||||
|
if (data.Usage is not null && (data.Usage.CompletionTokens > 0 || data.Usage.OutputTokens > 0))
|
||||||
{
|
{
|
||||||
tokenUsage = data.Usage;
|
tokenUsage = data.Usage;
|
||||||
}
|
}
|
||||||
@@ -239,7 +242,7 @@ public class AiGateWayManager : DomainService
|
|||||||
catch (Exception e)
|
catch (Exception e)
|
||||||
{
|
{
|
||||||
_logger.LogError(e, $"Ai对话异常");
|
_logger.LogError(e, $"Ai对话异常");
|
||||||
var errorContent = $"对话Ai异常,异常信息:\n当前Ai模型:{request.Model}\n异常信息:{e.Message}\n异常堆栈:{e}";
|
var errorContent = $"对话Ai异常,异常信息:\n当前Ai模型:{request.Model}\n异常信息:{e.Message}\n异常堆栈:{e}";
|
||||||
var model = new ThorChatCompletionsResponse()
|
var model = new ThorChatCompletionsResponse()
|
||||||
{
|
{
|
||||||
Choices = new List<ThorChatChoiceResponse>()
|
Choices = new List<ThorChatChoiceResponse>()
|
||||||
@@ -272,10 +275,10 @@ public class AiGateWayManager : DomainService
|
|||||||
await _aiMessageManager.CreateUserMessageAsync(userId, sessionId,
|
await _aiMessageManager.CreateUserMessageAsync(userId, sessionId,
|
||||||
new MessageInputDto
|
new MessageInputDto
|
||||||
{
|
{
|
||||||
Content = sessionId is null ? "不予存储" : request.Messages?.LastOrDefault()?.Content ?? string.Empty,
|
Content = sessionId is null ? "不予存储" : request.Messages?.LastOrDefault()?.MessagesStore ?? string.Empty,
|
||||||
ModelId = request.Model,
|
ModelId = request.Model,
|
||||||
TokenUsage = tokenUsage,
|
TokenUsage = tokenUsage,
|
||||||
});
|
}, tokenId);
|
||||||
|
|
||||||
await _aiMessageManager.CreateSystemMessageAsync(userId, sessionId,
|
await _aiMessageManager.CreateSystemMessageAsync(userId, sessionId,
|
||||||
new MessageInputDto
|
new MessageInputDto
|
||||||
@@ -283,9 +286,9 @@ public class AiGateWayManager : DomainService
|
|||||||
Content = sessionId is null ? "不予存储" : backupSystemContent.ToString(),
|
Content = sessionId is null ? "不予存储" : backupSystemContent.ToString(),
|
||||||
ModelId = request.Model,
|
ModelId = request.Model,
|
||||||
TokenUsage = tokenUsage
|
TokenUsage = tokenUsage
|
||||||
});
|
}, tokenId);
|
||||||
|
|
||||||
await _usageStatisticsManager.SetUsageAsync(userId, request.Model, tokenUsage);
|
await _usageStatisticsManager.SetUsageAsync(userId, request.Model, tokenUsage, tokenId);
|
||||||
|
|
||||||
// 扣减尊享token包用量
|
// 扣减尊享token包用量
|
||||||
if (userId is not null && PremiumPackageConst.ModeIds.Contains(request.Model))
|
if (userId is not null && PremiumPackageConst.ModeIds.Contains(request.Model))
|
||||||
@@ -306,10 +309,11 @@ public class AiGateWayManager : DomainService
|
|||||||
/// <param name="userId"></param>
|
/// <param name="userId"></param>
|
||||||
/// <param name="sessionId"></param>
|
/// <param name="sessionId"></param>
|
||||||
/// <param name="request"></param>
|
/// <param name="request"></param>
|
||||||
|
/// <param name="tokenId">Token Id(Web端传null或Guid.Empty)</param>
|
||||||
/// <exception cref="BusinessException"></exception>
|
/// <exception cref="BusinessException"></exception>
|
||||||
/// <exception cref="Exception"></exception>
|
/// <exception cref="Exception"></exception>
|
||||||
public async Task CreateImageForStatisticsAsync(HttpContext context, Guid? userId, Guid? sessionId,
|
public async Task CreateImageForStatisticsAsync(HttpContext context, Guid? userId, Guid? sessionId,
|
||||||
ImageCreateRequest request)
|
ImageCreateRequest request, Guid? tokenId = null)
|
||||||
{
|
{
|
||||||
try
|
try
|
||||||
{
|
{
|
||||||
@@ -337,7 +341,7 @@ public class AiGateWayManager : DomainService
|
|||||||
Content = sessionId is null ? "不予存储" : request.Prompt,
|
Content = sessionId is null ? "不予存储" : request.Prompt,
|
||||||
ModelId = model,
|
ModelId = model,
|
||||||
TokenUsage = response.Usage,
|
TokenUsage = response.Usage,
|
||||||
});
|
}, tokenId);
|
||||||
|
|
||||||
await _aiMessageManager.CreateSystemMessageAsync(userId, sessionId,
|
await _aiMessageManager.CreateSystemMessageAsync(userId, sessionId,
|
||||||
new MessageInputDto
|
new MessageInputDto
|
||||||
@@ -345,9 +349,9 @@ public class AiGateWayManager : DomainService
|
|||||||
Content = sessionId is null ? "不予存储" : response.Results?.FirstOrDefault()?.Url,
|
Content = sessionId is null ? "不予存储" : response.Results?.FirstOrDefault()?.Url,
|
||||||
ModelId = model,
|
ModelId = model,
|
||||||
TokenUsage = response.Usage
|
TokenUsage = response.Usage
|
||||||
});
|
}, tokenId);
|
||||||
|
|
||||||
await _usageStatisticsManager.SetUsageAsync(userId, model, response.Usage);
|
await _usageStatisticsManager.SetUsageAsync(userId, model, response.Usage, tokenId);
|
||||||
|
|
||||||
// 扣减尊享token包用量
|
// 扣减尊享token包用量
|
||||||
if (userId is not null && PremiumPackageConst.ModeIds.Contains(request.Model))
|
if (userId is not null && PremiumPackageConst.ModeIds.Contains(request.Model))
|
||||||
@@ -361,7 +365,7 @@ public class AiGateWayManager : DomainService
|
|||||||
}
|
}
|
||||||
catch (Exception e)
|
catch (Exception e)
|
||||||
{
|
{
|
||||||
var errorContent = $"图片生成Ai异常,异常信息:\n当前Ai模型:{request.Model}\n异常信息:{e.Message}\n异常堆栈:{e}";
|
var errorContent = $"图片生成Ai异常,异常信息:\n当前Ai模型:{request.Model}\n异常信息:{e.Message}\n异常堆栈:{e}";
|
||||||
throw new UserFriendlyException(errorContent);
|
throw new UserFriendlyException(errorContent);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -371,13 +375,14 @@ public class AiGateWayManager : DomainService
|
|||||||
/// 向量生成
|
/// 向量生成
|
||||||
/// </summary>
|
/// </summary>
|
||||||
/// <param name="context"></param>
|
/// <param name="context"></param>
|
||||||
|
/// <param name="userId"></param>
|
||||||
/// <param name="sessionId"></param>
|
/// <param name="sessionId"></param>
|
||||||
/// <param name="input"></param>
|
/// <param name="input"></param>
|
||||||
/// <param name="userId"></param>
|
/// <param name="tokenId">Token Id(Web端传null或Guid.Empty)</param>
|
||||||
/// <exception cref="Exception"></exception>
|
/// <exception cref="Exception"></exception>
|
||||||
/// <exception cref="BusinessException"></exception>
|
/// <exception cref="BusinessException"></exception>
|
||||||
public async Task EmbeddingForStatisticsAsync(HttpContext context, Guid? userId, Guid? sessionId,
|
public async Task EmbeddingForStatisticsAsync(HttpContext context, Guid? userId, Guid? sessionId,
|
||||||
ThorEmbeddingInput input)
|
ThorEmbeddingInput input, Guid? tokenId = null)
|
||||||
{
|
{
|
||||||
try
|
try
|
||||||
{
|
{
|
||||||
@@ -461,7 +466,7 @@ public class AiGateWayManager : DomainService
|
|||||||
// TokenUsage = usage
|
// TokenUsage = usage
|
||||||
// });
|
// });
|
||||||
|
|
||||||
await _usageStatisticsManager.SetUsageAsync(userId, input.Model, usage);
|
await _usageStatisticsManager.SetUsageAsync(userId, input.Model, usage, tokenId);
|
||||||
}
|
}
|
||||||
catch (ThorRateLimitException)
|
catch (ThorRateLimitException)
|
||||||
{
|
{
|
||||||
@@ -473,34 +478,12 @@ public class AiGateWayManager : DomainService
|
|||||||
}
|
}
|
||||||
catch (Exception e)
|
catch (Exception e)
|
||||||
{
|
{
|
||||||
var errorContent = $"嵌入Ai异常,异常信息:\n当前Ai模型:{input.Model}\n异常信息:{e.Message}\n异常堆栈:{e}";
|
var errorContent = $"嵌入Ai异常,异常信息:\n当前Ai模型:{input.Model}\n异常信息:{e.Message}\n异常堆栈:{e}";
|
||||||
throw new UserFriendlyException(errorContent);
|
throw new UserFriendlyException(errorContent);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// Anthropic聊天完成-流式
|
|
||||||
/// </summary>
|
|
||||||
/// <param name="request"></param>
|
|
||||||
/// <param name="cancellationToken"></param>
|
|
||||||
/// <returns></returns>
|
|
||||||
public async IAsyncEnumerable<(string, AnthropicStreamDto?)> AnthropicCompleteChatStreamAsync(
|
|
||||||
AnthropicInput request,
|
|
||||||
[EnumeratorCancellation] CancellationToken cancellationToken)
|
|
||||||
{
|
|
||||||
_specialCompatible.AnthropicCompatible(request);
|
|
||||||
var modelDescribe = await GetModelAsync(ModelApiTypeEnum.Claude, request.Model);
|
|
||||||
var chatService =
|
|
||||||
LazyServiceProvider.GetRequiredKeyedService<IAnthropicChatCompletionService>(modelDescribe.HandlerName);
|
|
||||||
|
|
||||||
await foreach (var result in chatService.StreamChatCompletionsAsync(modelDescribe, request, cancellationToken))
|
|
||||||
{
|
|
||||||
yield return result;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// Anthropic聊天完成-非流式
|
/// Anthropic聊天完成-非流式
|
||||||
/// </summary>
|
/// </summary>
|
||||||
@@ -508,12 +491,14 @@ public class AiGateWayManager : DomainService
|
|||||||
/// <param name="request"></param>
|
/// <param name="request"></param>
|
||||||
/// <param name="userId"></param>
|
/// <param name="userId"></param>
|
||||||
/// <param name="sessionId"></param>
|
/// <param name="sessionId"></param>
|
||||||
|
/// <param name="tokenId">Token Id(Web端传null或Guid.Empty)</param>
|
||||||
/// <param name="cancellationToken"></param>
|
/// <param name="cancellationToken"></param>
|
||||||
/// <returns></returns>
|
/// <returns></returns>
|
||||||
public async Task AnthropicCompleteChatForStatisticsAsync(HttpContext httpContext,
|
public async Task AnthropicCompleteChatForStatisticsAsync(HttpContext httpContext,
|
||||||
AnthropicInput request,
|
AnthropicInput request,
|
||||||
Guid? userId = null,
|
Guid? userId = null,
|
||||||
Guid? sessionId = null,
|
Guid? sessionId = null,
|
||||||
|
Guid? tokenId = null,
|
||||||
CancellationToken cancellationToken = default)
|
CancellationToken cancellationToken = default)
|
||||||
{
|
{
|
||||||
_specialCompatible.AnthropicCompatible(request);
|
_specialCompatible.AnthropicCompatible(request);
|
||||||
@@ -524,25 +509,28 @@ public class AiGateWayManager : DomainService
|
|||||||
var chatService =
|
var chatService =
|
||||||
LazyServiceProvider.GetRequiredKeyedService<IAnthropicChatCompletionService>(modelDescribe.HandlerName);
|
LazyServiceProvider.GetRequiredKeyedService<IAnthropicChatCompletionService>(modelDescribe.HandlerName);
|
||||||
var data = await chatService.ChatCompletionsAsync(modelDescribe, request, cancellationToken);
|
var data = await chatService.ChatCompletionsAsync(modelDescribe, request, cancellationToken);
|
||||||
|
|
||||||
|
data.SupplementalMultiplier(modelDescribe.Multiplier);
|
||||||
|
|
||||||
if (userId is not null)
|
if (userId is not null)
|
||||||
{
|
{
|
||||||
await _aiMessageManager.CreateUserMessageAsync(userId.Value, sessionId,
|
await _aiMessageManager.CreateUserMessageAsync(userId.Value, sessionId,
|
||||||
new MessageInputDto
|
new MessageInputDto
|
||||||
{
|
{
|
||||||
Content = sessionId is null ? "不予存储" : request.Messages?.FirstOrDefault()?.Content ?? string.Empty,
|
Content = "不予存储",
|
||||||
ModelId = request.Model,
|
ModelId = request.Model,
|
||||||
TokenUsage = data.TokenUsage,
|
TokenUsage = data.TokenUsage,
|
||||||
});
|
}, tokenId);
|
||||||
|
|
||||||
await _aiMessageManager.CreateSystemMessageAsync(userId.Value, sessionId,
|
await _aiMessageManager.CreateSystemMessageAsync(userId.Value, sessionId,
|
||||||
new MessageInputDto
|
new MessageInputDto
|
||||||
{
|
{
|
||||||
Content = sessionId is null ? "不予存储" : data.content?.FirstOrDefault()?.text,
|
Content = "不予存储",
|
||||||
ModelId = request.Model,
|
ModelId = request.Model,
|
||||||
TokenUsage = data.TokenUsage
|
TokenUsage = data.TokenUsage
|
||||||
});
|
}, tokenId);
|
||||||
|
|
||||||
await _usageStatisticsManager.SetUsageAsync(userId.Value, request.Model, data.TokenUsage);
|
await _usageStatisticsManager.SetUsageAsync(userId.Value, request.Model, data.TokenUsage, tokenId);
|
||||||
|
|
||||||
// 扣减尊享token包用量
|
// 扣减尊享token包用量
|
||||||
var totalTokens = data.TokenUsage.TotalTokens ?? 0;
|
var totalTokens = data.TokenUsage.TotalTokens ?? 0;
|
||||||
@@ -555,6 +543,7 @@ public class AiGateWayManager : DomainService
|
|||||||
await response.WriteAsJsonAsync(data, cancellationToken);
|
await response.WriteAsJsonAsync(data, cancellationToken);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// Anthropic聊天完成-缓存处理
|
/// Anthropic聊天完成-缓存处理
|
||||||
/// </summary>
|
/// </summary>
|
||||||
@@ -562,6 +551,7 @@ public class AiGateWayManager : DomainService
|
|||||||
/// <param name="request"></param>
|
/// <param name="request"></param>
|
||||||
/// <param name="userId"></param>
|
/// <param name="userId"></param>
|
||||||
/// <param name="sessionId"></param>
|
/// <param name="sessionId"></param>
|
||||||
|
/// <param name="tokenId">Token Id(Web端传null或Guid.Empty)</param>
|
||||||
/// <param name="cancellationToken"></param>
|
/// <param name="cancellationToken"></param>
|
||||||
/// <returns></returns>
|
/// <returns></returns>
|
||||||
public async Task AnthropicCompleteChatStreamForStatisticsAsync(
|
public async Task AnthropicCompleteChatStreamForStatisticsAsync(
|
||||||
@@ -569,6 +559,7 @@ public class AiGateWayManager : DomainService
|
|||||||
AnthropicInput request,
|
AnthropicInput request,
|
||||||
Guid? userId = null,
|
Guid? userId = null,
|
||||||
Guid? sessionId = null,
|
Guid? sessionId = null,
|
||||||
|
Guid? tokenId = null,
|
||||||
CancellationToken cancellationToken = default)
|
CancellationToken cancellationToken = default)
|
||||||
{
|
{
|
||||||
var response = httpContext.Response;
|
var response = httpContext.Response;
|
||||||
@@ -577,15 +568,19 @@ public class AiGateWayManager : DomainService
|
|||||||
response.Headers.TryAdd("Cache-Control", "no-cache");
|
response.Headers.TryAdd("Cache-Control", "no-cache");
|
||||||
response.Headers.TryAdd("Connection", "keep-alive");
|
response.Headers.TryAdd("Connection", "keep-alive");
|
||||||
|
|
||||||
|
_specialCompatible.AnthropicCompatible(request);
|
||||||
|
var modelDescribe = await GetModelAsync(ModelApiTypeEnum.Claude, request.Model);
|
||||||
|
var chatService =
|
||||||
|
LazyServiceProvider.GetRequiredKeyedService<IAnthropicChatCompletionService>(modelDescribe.HandlerName);
|
||||||
|
|
||||||
var gateWay = LazyServiceProvider.GetRequiredService<AiGateWayManager>();
|
var completeChatResponse = chatService.StreamChatCompletionsAsync(modelDescribe,request, cancellationToken);
|
||||||
var completeChatResponse = gateWay.AnthropicCompleteChatStreamAsync(request, cancellationToken);
|
|
||||||
ThorUsageResponse? tokenUsage = null;
|
ThorUsageResponse? tokenUsage = null;
|
||||||
StringBuilder backupSystemContent = new StringBuilder();
|
StringBuilder backupSystemContent = new StringBuilder();
|
||||||
try
|
try
|
||||||
{
|
{
|
||||||
await foreach (var responseResult in completeChatResponse)
|
await foreach (var responseResult in completeChatResponse)
|
||||||
{
|
{
|
||||||
|
responseResult.Item2.SupplementalMultiplier(modelDescribe.Multiplier);
|
||||||
//message_start是为了保底机制
|
//message_start是为了保底机制
|
||||||
if (responseResult.Item1.Contains("message_delta") || responseResult.Item1.Contains("message_start"))
|
if (responseResult.Item1.Contains("message_delta") || responseResult.Item1.Contains("message_start"))
|
||||||
{
|
{
|
||||||
@@ -600,27 +595,27 @@ public class AiGateWayManager : DomainService
|
|||||||
catch (Exception e)
|
catch (Exception e)
|
||||||
{
|
{
|
||||||
_logger.LogError(e, $"Ai对话异常");
|
_logger.LogError(e, $"Ai对话异常");
|
||||||
var errorContent = $"对话Ai异常,异常信息:\n当前Ai模型:{request.Model}\n异常信息:{e.Message}\n异常堆栈:{e}";
|
var errorContent = $"对话Ai异常,异常信息:\n当前Ai模型:{request.Model}\n异常信息:{e.Message}\n异常堆栈:{e}";
|
||||||
throw new UserFriendlyException(errorContent);
|
throw new UserFriendlyException(errorContent);
|
||||||
}
|
}
|
||||||
|
|
||||||
await _aiMessageManager.CreateUserMessageAsync(userId, sessionId,
|
await _aiMessageManager.CreateUserMessageAsync(userId, sessionId,
|
||||||
new MessageInputDto
|
new MessageInputDto
|
||||||
{
|
{
|
||||||
Content = sessionId is null ? "不予存储" : request.Messages?.LastOrDefault()?.Content ?? string.Empty,
|
Content = "不予存储",
|
||||||
ModelId = request.Model,
|
ModelId = request.Model,
|
||||||
TokenUsage = tokenUsage,
|
TokenUsage = tokenUsage,
|
||||||
});
|
}, tokenId);
|
||||||
|
|
||||||
await _aiMessageManager.CreateSystemMessageAsync(userId, sessionId,
|
await _aiMessageManager.CreateSystemMessageAsync(userId, sessionId,
|
||||||
new MessageInputDto
|
new MessageInputDto
|
||||||
{
|
{
|
||||||
Content = sessionId is null ? "不予存储" : backupSystemContent.ToString(),
|
Content = "不予存储",
|
||||||
ModelId = request.Model,
|
ModelId = request.Model,
|
||||||
TokenUsage = tokenUsage
|
TokenUsage = tokenUsage
|
||||||
});
|
}, tokenId);
|
||||||
|
|
||||||
await _usageStatisticsManager.SetUsageAsync(userId, request.Model, tokenUsage);
|
await _usageStatisticsManager.SetUsageAsync(userId, request.Model, tokenUsage, tokenId);
|
||||||
|
|
||||||
// 扣减尊享token包用量
|
// 扣减尊享token包用量
|
||||||
if (userId.HasValue && tokenUsage is not null)
|
if (userId.HasValue && tokenUsage is not null)
|
||||||
@@ -633,7 +628,167 @@ public class AiGateWayManager : DomainService
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#region Anthropic格式Http响应
|
|
||||||
|
/// <summary>
|
||||||
|
/// OpenAi 响应-非流式-缓存处理
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="httpContext"></param>
|
||||||
|
/// <param name="request"></param>
|
||||||
|
/// <param name="userId"></param>
|
||||||
|
/// <param name="sessionId"></param>
|
||||||
|
/// <param name="tokenId"></param>
|
||||||
|
/// <param name="cancellationToken"></param>
|
||||||
|
public async Task OpenAiResponsesAsyncForStatisticsAsync(HttpContext httpContext,
|
||||||
|
OpenAiResponsesInput request,
|
||||||
|
Guid? userId = null,
|
||||||
|
Guid? sessionId = null,
|
||||||
|
Guid? tokenId = null,
|
||||||
|
CancellationToken cancellationToken = default)
|
||||||
|
{
|
||||||
|
// _specialCompatible.AnthropicCompatible(request);
|
||||||
|
var response = httpContext.Response;
|
||||||
|
// 设置响应头,声明是 json
|
||||||
|
//response.ContentType = "application/json; charset=UTF-8";
|
||||||
|
var modelDescribe = await GetModelAsync(ModelApiTypeEnum.Response, request.Model);
|
||||||
|
|
||||||
|
var chatService =
|
||||||
|
LazyServiceProvider.GetRequiredKeyedService<IOpenAiResponseService>(modelDescribe.HandlerName);
|
||||||
|
var data = await chatService.ResponsesAsync(modelDescribe, request, cancellationToken);
|
||||||
|
|
||||||
|
data.SupplementalMultiplier(modelDescribe.Multiplier);
|
||||||
|
|
||||||
|
var tokenUsage= new ThorUsageResponse
|
||||||
|
{
|
||||||
|
InputTokens = data.Usage.InputTokens,
|
||||||
|
OutputTokens = data.Usage.OutputTokens,
|
||||||
|
TotalTokens = data.Usage.InputTokens + data.Usage.OutputTokens,
|
||||||
|
};
|
||||||
|
if (userId is not null)
|
||||||
|
{
|
||||||
|
await _aiMessageManager.CreateUserMessageAsync(userId.Value, sessionId,
|
||||||
|
new MessageInputDto
|
||||||
|
{
|
||||||
|
Content = "不予存储",
|
||||||
|
ModelId = request.Model,
|
||||||
|
TokenUsage = tokenUsage,
|
||||||
|
}, tokenId);
|
||||||
|
|
||||||
|
await _aiMessageManager.CreateSystemMessageAsync(userId.Value, sessionId,
|
||||||
|
new MessageInputDto
|
||||||
|
{
|
||||||
|
Content = "不予存储",
|
||||||
|
ModelId = request.Model,
|
||||||
|
TokenUsage = tokenUsage
|
||||||
|
}, tokenId);
|
||||||
|
|
||||||
|
await _usageStatisticsManager.SetUsageAsync(userId.Value, request.Model, tokenUsage, tokenId);
|
||||||
|
|
||||||
|
// 扣减尊享token包用量
|
||||||
|
var totalTokens = tokenUsage.TotalTokens ?? 0;
|
||||||
|
if (totalTokens > 0)
|
||||||
|
{
|
||||||
|
await PremiumPackageManager.TryConsumeTokensAsync(userId.Value, totalTokens);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
await response.WriteAsJsonAsync(data, cancellationToken);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// OpenAi响应-流式-缓存处理
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="httpContext"></param>
|
||||||
|
/// <param name="request"></param>
|
||||||
|
/// <param name="userId"></param>
|
||||||
|
/// <param name="sessionId"></param>
|
||||||
|
/// <param name="tokenId">Token Id(Web端传null或Guid.Empty)</param>
|
||||||
|
/// <param name="cancellationToken"></param>
|
||||||
|
/// <returns></returns>
|
||||||
|
public async Task OpenAiResponsesStreamForStatisticsAsync(
|
||||||
|
HttpContext httpContext,
|
||||||
|
OpenAiResponsesInput request,
|
||||||
|
Guid? userId = null,
|
||||||
|
Guid? sessionId = null,
|
||||||
|
Guid? tokenId = null,
|
||||||
|
CancellationToken cancellationToken = default)
|
||||||
|
{
|
||||||
|
var response = httpContext.Response;
|
||||||
|
// 设置响应头,声明是 SSE 流
|
||||||
|
response.ContentType = "text/event-stream;charset=utf-8;";
|
||||||
|
response.Headers.TryAdd("Cache-Control", "no-cache");
|
||||||
|
response.Headers.TryAdd("Connection", "keep-alive");
|
||||||
|
|
||||||
|
var modelDescribe = await GetModelAsync(ModelApiTypeEnum.Response, request.Model);
|
||||||
|
var chatService =
|
||||||
|
LazyServiceProvider.GetRequiredKeyedService<IOpenAiResponseService>(modelDescribe.HandlerName);
|
||||||
|
|
||||||
|
var completeChatResponse = chatService.ResponsesStreamAsync(modelDescribe,request, cancellationToken);
|
||||||
|
ThorUsageResponse? tokenUsage = null;
|
||||||
|
try
|
||||||
|
{
|
||||||
|
await foreach (var responseResult in completeChatResponse)
|
||||||
|
{
|
||||||
|
//message_start是为了保底机制
|
||||||
|
if (responseResult.Item1.Contains("response.completed"))
|
||||||
|
{
|
||||||
|
var obj = responseResult.Item2!.Value;
|
||||||
|
int inputTokens = obj.GetPath("response","usage","input_tokens").GetInt();
|
||||||
|
int outputTokens = obj.GetPath("response","usage","output_tokens").GetInt();
|
||||||
|
inputTokens=Convert.ToInt32(inputTokens * modelDescribe.Multiplier);
|
||||||
|
outputTokens=Convert.ToInt32(outputTokens * modelDescribe.Multiplier);
|
||||||
|
tokenUsage = new ThorUsageResponse
|
||||||
|
{
|
||||||
|
PromptTokens =inputTokens,
|
||||||
|
InputTokens = inputTokens,
|
||||||
|
OutputTokens = outputTokens,
|
||||||
|
CompletionTokens = outputTokens,
|
||||||
|
TotalTokens = inputTokens+outputTokens,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
await WriteAsEventStreamDataAsync(httpContext, responseResult.Item1, responseResult.Item2,
|
||||||
|
cancellationToken);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch (Exception e)
|
||||||
|
{
|
||||||
|
_logger.LogError(e, $"Ai响应异常");
|
||||||
|
var errorContent = $"响应Ai异常,异常信息:\n当前Ai模型:{request.Model}\n异常信息:{e.Message}\n异常堆栈:{e}";
|
||||||
|
throw new UserFriendlyException(errorContent);
|
||||||
|
}
|
||||||
|
|
||||||
|
await _aiMessageManager.CreateUserMessageAsync(userId, sessionId,
|
||||||
|
new MessageInputDto
|
||||||
|
{
|
||||||
|
Content = "不予存储" ,
|
||||||
|
ModelId = request.Model,
|
||||||
|
TokenUsage = tokenUsage,
|
||||||
|
}, tokenId);
|
||||||
|
|
||||||
|
await _aiMessageManager.CreateSystemMessageAsync(userId, sessionId,
|
||||||
|
new MessageInputDto
|
||||||
|
{
|
||||||
|
Content = "不予存储" ,
|
||||||
|
ModelId = request.Model,
|
||||||
|
TokenUsage = tokenUsage
|
||||||
|
}, tokenId);
|
||||||
|
|
||||||
|
await _usageStatisticsManager.SetUsageAsync(userId, request.Model, tokenUsage, tokenId);
|
||||||
|
|
||||||
|
// 扣减尊享token包用量
|
||||||
|
if (userId.HasValue && tokenUsage is not null)
|
||||||
|
{
|
||||||
|
var totalTokens = tokenUsage.TotalTokens ?? 0;
|
||||||
|
if (tokenUsage.TotalTokens > 0)
|
||||||
|
{
|
||||||
|
await PremiumPackageManager.TryConsumeTokensAsync(userId.Value, totalTokens);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
#region 流式传输格式Http响应
|
||||||
|
|
||||||
private static readonly byte[] EventPrefix = "event: "u8.ToArray();
|
private static readonly byte[] EventPrefix = "event: "u8.ToArray();
|
||||||
private static readonly byte[] DataPrefix = "data: "u8.ToArray();
|
private static readonly byte[] DataPrefix = "data: "u8.ToArray();
|
||||||
@@ -648,7 +803,6 @@ public class AiGateWayManager : DomainService
|
|||||||
string @event,
|
string @event,
|
||||||
T value,
|
T value,
|
||||||
CancellationToken cancellationToken = default)
|
CancellationToken cancellationToken = default)
|
||||||
where T : class
|
|
||||||
{
|
{
|
||||||
var response = context.Response;
|
var response = context.Response;
|
||||||
var bodyStream = response.Body;
|
var bodyStream = response.Body;
|
||||||
|
|||||||
@@ -19,28 +19,30 @@ public class AiMessageManager : DomainService
|
|||||||
/// <summary>
|
/// <summary>
|
||||||
/// 创建系统消息
|
/// 创建系统消息
|
||||||
/// </summary>
|
/// </summary>
|
||||||
/// <param name="sessionId"></param>
|
/// <param name="userId">用户Id</param>
|
||||||
/// <param name="userId"></param>
|
/// <param name="sessionId">会话Id</param>
|
||||||
/// <param name="input"></param>
|
/// <param name="input">消息输入</param>
|
||||||
|
/// <param name="tokenId">Token Id(Web端传Guid.Empty)</param>
|
||||||
/// <returns></returns>
|
/// <returns></returns>
|
||||||
public async Task CreateSystemMessageAsync(Guid? userId, Guid? sessionId, MessageInputDto input)
|
public async Task CreateSystemMessageAsync(Guid? userId, Guid? sessionId, MessageInputDto input, Guid? tokenId = null)
|
||||||
{
|
{
|
||||||
input.Role = "system";
|
input.Role = "system";
|
||||||
var message = new MessageAggregateRoot(userId, sessionId, input.Content, input.Role, input.ModelId,input.TokenUsage);
|
var message = new MessageAggregateRoot(userId, sessionId, input.Content, input.Role, input.ModelId, input.TokenUsage, tokenId);
|
||||||
await _repository.InsertAsync(message);
|
await _repository.InsertAsync(message);
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 创建系统消息
|
/// 创建用户消息
|
||||||
/// </summary>
|
/// </summary>
|
||||||
/// <param name="sessionId"></param>
|
/// <param name="userId">用户Id</param>
|
||||||
/// <param name="userId"></param>
|
/// <param name="sessionId">会话Id</param>
|
||||||
/// <param name="input"></param>
|
/// <param name="input">消息输入</param>
|
||||||
|
/// <param name="tokenId">Token Id(Web端传Guid.Empty)</param>
|
||||||
/// <returns></returns>
|
/// <returns></returns>
|
||||||
public async Task CreateUserMessageAsync(Guid? userId, Guid? sessionId, MessageInputDto input)
|
public async Task CreateUserMessageAsync(Guid? userId, Guid? sessionId, MessageInputDto input, Guid? tokenId = null)
|
||||||
{
|
{
|
||||||
input.Role = "user";
|
input.Role = "user";
|
||||||
var message = new MessageAggregateRoot(userId, sessionId, input.Content, input.Role, input.ModelId,input.TokenUsage);
|
var message = new MessageAggregateRoot(userId, sessionId, input.Content, input.Role, input.ModelId, input.TokenUsage, tokenId);
|
||||||
await _repository.InsertAsync(message);
|
await _repository.InsertAsync(message);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -24,6 +24,11 @@ public class CardFlipManager : DomainService
|
|||||||
private const int NINTH_FLIP = 9; // 第9次翻牌
|
private const int NINTH_FLIP = 9; // 第9次翻牌
|
||||||
private const int TENTH_FLIP = 10; // 第10次翻牌
|
private const int TENTH_FLIP = 10; // 第10次翻牌
|
||||||
|
|
||||||
|
// 前7次免费翻牌奖励配置
|
||||||
|
private const long FREE_MIN_REWARD = 10000; // 前7次最小奖励 1w
|
||||||
|
private const long FREE_MAX_REWARD = 30000; // 前7次最大奖励 3w
|
||||||
|
private const double FREE_WIN_RATE = 0.5; // 前7次中奖概率 50%
|
||||||
|
|
||||||
private const long EIGHTH_MIN_REWARD = 1000000; // 第8次最小奖励 100w
|
private const long EIGHTH_MIN_REWARD = 1000000; // 第8次最小奖励 100w
|
||||||
private const long EIGHTH_MAX_REWARD = 3000000; // 第8次最大奖励 300w
|
private const long EIGHTH_MAX_REWARD = 3000000; // 第8次最大奖励 300w
|
||||||
private const long NINTH_MIN_REWARD = 1000000; // 第9次最小奖励 100w
|
private const long NINTH_MIN_REWARD = 1000000; // 第9次最小奖励 100w
|
||||||
@@ -112,23 +117,23 @@ public class CardFlipManager : DomainService
|
|||||||
throw new UserFriendlyException(GetFlipTypeErrorMessage(flipType));
|
throw new UserFriendlyException(GetFlipTypeErrorMessage(flipType));
|
||||||
}
|
}
|
||||||
|
|
||||||
// 如果是邀请类型翻牌,必须验证用户本周填写的邀请码数量足够
|
// 如果是邀请类型翻牌,必须验证用户本周的邀请记录数量足够(包括填写别人的邀请码和别人填写我的邀请码)
|
||||||
if (flipType == FlipType.Invite)
|
if (flipType == FlipType.Invite)
|
||||||
{
|
{
|
||||||
// 查询本周已使用的邀请码数量
|
// 查询本周作为邀请人或被邀请人的记录数量(双方都会增加翻牌次数)
|
||||||
var weeklyInviteCodeUsedCount = await _invitationRecordRepository._DbQueryable
|
var weeklyInviteRecordCount = await _invitationRecordRepository._DbQueryable
|
||||||
.Where(x => x.InvitedUserId == userId)
|
.Where(x => x.InviterId == userId || x.InvitedUserId == userId)
|
||||||
.Where(x => x.InvitationTime >= weekStart)
|
.Where(x => x.InvitationTime >= weekStart)
|
||||||
.CountAsync();
|
.CountAsync();
|
||||||
|
|
||||||
// 本周填写的邀请码数量必须 >= 即将使用的邀请翻牌次数
|
// 本周邀请记录数量必须 >= 即将使用的邀请翻牌次数
|
||||||
// 例如: 要翻第8次(InviteFlipsUsed=0->1), 需要至少填写了1个邀请码
|
// 例如: 要翻第8次(InviteFlipsUsed=0->1), 需要至少有1条邀请记录(我邀请别人或别人邀请我)
|
||||||
// 要翻第9次(InviteFlipsUsed=1->2), 需要至少填写了2个邀请码
|
// 要翻第9次(InviteFlipsUsed=1->2), 需要至少有2条邀请记录
|
||||||
// 要翻第10次(InviteFlipsUsed=2->3), 需要至少填写了3个邀请码
|
// 要翻第10次(InviteFlipsUsed=2->3), 需要至少有3条邀请记录
|
||||||
var requiredInviteCodeCount = task.InviteFlipsUsed + 1;
|
var requiredInviteRecordCount = task.InviteFlipsUsed + 1;
|
||||||
if (weeklyInviteCodeUsedCount < requiredInviteCodeCount)
|
if (weeklyInviteRecordCount < requiredInviteRecordCount)
|
||||||
{
|
{
|
||||||
throw new UserFriendlyException($"需本周累积使用{requiredInviteCodeCount}个他人邀请码才能解锁第{task.TotalFlips + 1}次翻牌,您还差一个~");
|
throw new UserFriendlyException($"需本周累积{requiredInviteRecordCount}次邀请记录(填写别人的邀请码或别人填写你的邀请码)才能解锁第{task.TotalFlips + 1}次翻牌");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -152,18 +157,7 @@ public class CardFlipManager : DomainService
|
|||||||
// 如果中奖,记录奖励金额(用于后续查询显示)
|
// 如果中奖,记录奖励金额(用于后续查询显示)
|
||||||
if (result.IsWin)
|
if (result.IsWin)
|
||||||
{
|
{
|
||||||
if (flipCount == EIGHTH_FLIP)
|
task.SetWinReward(flipCount, result.RewardAmount);
|
||||||
{
|
|
||||||
task.SetEighthReward(result.RewardAmount);
|
|
||||||
}
|
|
||||||
else if (flipCount == NINTH_FLIP)
|
|
||||||
{
|
|
||||||
task.SetNinthReward(result.RewardAmount);
|
|
||||||
}
|
|
||||||
else if (flipCount == TENTH_FLIP)
|
|
||||||
{
|
|
||||||
task.SetTenthReward(result.RewardAmount);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
await _cardFlipTaskRepository.UpdateAsync(task);
|
await _cardFlipTaskRepository.UpdateAsync(task);
|
||||||
@@ -223,11 +217,24 @@ public class CardFlipManager : DomainService
|
|||||||
IsWin = false
|
IsWin = false
|
||||||
};
|
};
|
||||||
|
|
||||||
// 前7次固定失败
|
var random = new Random();
|
||||||
|
|
||||||
|
// 前7次: 50%概率中奖,奖励1w-3w
|
||||||
if (flipCount <= 7)
|
if (flipCount <= 7)
|
||||||
{
|
{
|
||||||
result.IsWin = false;
|
// 50%概率中奖
|
||||||
result.RewardDesc = "很遗憾,未中奖";
|
if (random.NextDouble() < FREE_WIN_RATE)
|
||||||
|
{
|
||||||
|
var rewardAmount = GenerateRandomReward(FREE_MIN_REWARD, FREE_MAX_REWARD);
|
||||||
|
result.IsWin = true;
|
||||||
|
result.RewardAmount = rewardAmount;
|
||||||
|
result.RewardDesc = $"恭喜获得尊享包 {rewardAmount / 10000m:0.##}w tokens!";
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
result.IsWin = false;
|
||||||
|
result.RewardDesc = "很遗憾,未中奖";
|
||||||
|
}
|
||||||
}
|
}
|
||||||
// 第8次中奖 (邀请码解锁)
|
// 第8次中奖 (邀请码解锁)
|
||||||
else if (flipCount == EIGHTH_FLIP)
|
else if (flipCount == EIGHTH_FLIP)
|
||||||
@@ -271,21 +278,24 @@ public class CardFlipManager : DomainService
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 生成随机奖励金额 (最小单位100w)
|
/// 生成随机奖励金额
|
||||||
/// </summary>
|
/// </summary>
|
||||||
private long GenerateRandomReward(long min, long max)
|
private long GenerateRandomReward(long min, long max)
|
||||||
{
|
{
|
||||||
var random = new Random();
|
var random = new Random();
|
||||||
const long unit = 1000000; // 100w的单位
|
|
||||||
|
|
||||||
// 将min和max转换为100w的倍数
|
// 根据最小值判断单位
|
||||||
|
// 如果min小于100000,则使用1w(10000)作为单位;否则使用100w(1000000)作为单位
|
||||||
|
long unit = min < 100000 ? 10000 : 1000000;
|
||||||
|
|
||||||
|
// 将min和max转换为单位的倍数
|
||||||
long minUnits = min / unit;
|
long minUnits = min / unit;
|
||||||
long maxUnits = max / unit;
|
long maxUnits = max / unit;
|
||||||
|
|
||||||
// 在倍数范围内随机
|
// 在倍数范围内随机
|
||||||
long randomUnits = random.Next((int)minUnits, (int)maxUnits + 1);
|
long randomUnits = random.Next((int)minUnits, (int)maxUnits + 1);
|
||||||
|
|
||||||
// 返回100w的倍数
|
// 返回单位的倍数
|
||||||
return randomUnits * unit;
|
return randomUnits * unit;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -63,14 +63,19 @@ public class InviteCodeManager : DomainService
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 统计用户本周邀请人数
|
/// 统计用户本周邀请人数(别人填写我的邀请码的次数/或者我填写别人邀请码)
|
||||||
/// </summary>
|
/// </summary>
|
||||||
public async Task<int> GetWeeklyInvitationCountAsync(Guid userId, DateTime weekStart)
|
public async Task<int> GetWeeklyInvitationCountAsync(Guid userId, DateTime weekStart)
|
||||||
{
|
{
|
||||||
return await _invitationRecordRepository._DbQueryable
|
var inviterCount= await _invitationRecordRepository._DbQueryable
|
||||||
|
.Where(x => x.InviterId == userId)
|
||||||
|
.Where(x => x.InvitationTime >= weekStart)
|
||||||
|
.CountAsync();
|
||||||
|
var invitedUserIdCount= await _invitationRecordRepository._DbQueryable
|
||||||
.Where(x => x.InvitedUserId == userId)
|
.Where(x => x.InvitedUserId == userId)
|
||||||
.Where(x => x.InvitationTime >= weekStart)
|
.Where(x => x.InvitationTime >= weekStart)
|
||||||
.CountAsync();
|
.CountAsync();
|
||||||
|
return inviterCount + invitedUserIdCount;
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
@@ -91,7 +96,7 @@ public class InviteCodeManager : DomainService
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 使用邀请码
|
/// 使用邀请码(双方都增加翻牌次数)
|
||||||
/// </summary>
|
/// </summary>
|
||||||
/// <param name="userId">使用者ID</param>
|
/// <param name="userId">使用者ID</param>
|
||||||
/// <param name="inviteCode">邀请码</param>
|
/// <param name="inviteCode">邀请码</param>
|
||||||
@@ -119,74 +124,39 @@ public class InviteCodeManager : DomainService
|
|||||||
throw new UserFriendlyException("不能使用自己的邀请码");
|
throw new UserFriendlyException("不能使用自己的邀请码");
|
||||||
}
|
}
|
||||||
|
|
||||||
// 验证邀请码是否已被使用
|
// 检查当前用户是否已经填写过别人的邀请码(一辈子只能填写一次)
|
||||||
if (inviteCodeEntity.IsUsed)
|
var hasUsedOthersCode = await IsFilledInviteCodeAsync(userId);
|
||||||
|
|
||||||
|
if (hasUsedOthersCode)
|
||||||
{
|
{
|
||||||
throw new UserFriendlyException("该邀请码已被使用");
|
throw new UserFriendlyException("您已经填写过别人的邀请码了,每个账号只能填写一次");
|
||||||
}
|
}
|
||||||
|
|
||||||
// 验证邀请码拥有者是否已被邀请
|
// 增加邀请码被使用次数
|
||||||
if (inviteCodeEntity.IsUserInvited)
|
inviteCodeEntity.UsedCount++;
|
||||||
{
|
|
||||||
throw new UserFriendlyException("该用户已被邀请,邀请码无效");
|
|
||||||
}
|
|
||||||
|
|
||||||
// 验证本周邀请码使用次数
|
|
||||||
var weekStart = CardFlipManager.GetWeekStartDate(DateTime.Now);
|
|
||||||
var weeklyUseCount = await _invitationRecordRepository._DbQueryable
|
|
||||||
.Where(x => x.InvitedUserId == userId)
|
|
||||||
.Where(x => x.InvitationTime >= weekStart)
|
|
||||||
.CountAsync();
|
|
||||||
|
|
||||||
if (weeklyUseCount >= CardFlipManager.MAX_INVITE_FLIPS)
|
|
||||||
{
|
|
||||||
throw new UserFriendlyException($"本周邀请码使用次数已达上限({CardFlipManager.MAX_INVITE_FLIPS}次),请下周再来");
|
|
||||||
}
|
|
||||||
|
|
||||||
// 检查当前用户的邀请码信息
|
|
||||||
var myInviteCode = await _inviteCodeRepository._DbQueryable
|
|
||||||
.Where(x => x.UserId == userId)
|
|
||||||
.FirstAsync();
|
|
||||||
|
|
||||||
// 标记邀请码为已使用
|
|
||||||
inviteCodeEntity.MarkAsUsed(userId);
|
|
||||||
await _inviteCodeRepository.UpdateAsync(inviteCodeEntity);
|
await _inviteCodeRepository.UpdateAsync(inviteCodeEntity);
|
||||||
|
|
||||||
// 标记当前用户已被邀请(仅第一次使用邀请码时标记)
|
// 创建邀请记录(双方都会因为这条记录增加一次翻牌机会)
|
||||||
if (myInviteCode == null)
|
|
||||||
{
|
|
||||||
myInviteCode = new InviteCodeAggregateRoot(userId, GenerateUniqueInviteCode());
|
|
||||||
myInviteCode.MarkUserAsInvited();
|
|
||||||
await _inviteCodeRepository.InsertAsync(myInviteCode);
|
|
||||||
}
|
|
||||||
else if (!myInviteCode.IsUserInvited)
|
|
||||||
{
|
|
||||||
myInviteCode.MarkUserAsInvited();
|
|
||||||
await _inviteCodeRepository.UpdateAsync(myInviteCode);
|
|
||||||
}
|
|
||||||
|
|
||||||
// 创建邀请记录
|
|
||||||
var invitationRecord = new InvitationRecordAggregateRoot(
|
var invitationRecord = new InvitationRecordAggregateRoot(
|
||||||
inviteCodeEntity.UserId,
|
inviteCodeEntity.UserId,
|
||||||
userId,
|
userId,
|
||||||
inviteCode);
|
inviteCode);
|
||||||
await _invitationRecordRepository.InsertAsync(invitationRecord);
|
await _invitationRecordRepository.InsertAsync(invitationRecord);
|
||||||
|
|
||||||
_logger.LogInformation($"用户 {userId} 使用邀请码 {inviteCode} 成功");
|
_logger.LogInformation($"用户 {userId} 使用邀请码 {inviteCode} 成功,邀请人 {inviteCodeEntity.UserId} 和被邀请人 {userId} 都增加一次翻牌机会");
|
||||||
|
|
||||||
return inviteCodeEntity.UserId;
|
return inviteCodeEntity.UserId;
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 检查用户是否已被邀请
|
/// 检查用户是否已填写过邀请码
|
||||||
/// </summary>
|
/// </summary>
|
||||||
public async Task<bool> IsUserInvitedAsync(Guid userId)
|
public async Task<bool> IsFilledInviteCodeAsync(Guid userId)
|
||||||
{
|
{
|
||||||
var inviteCode = await _inviteCodeRepository._DbQueryable
|
// 检查当前用户是否已经填写过别人的邀请码(一辈子只能填写一次)
|
||||||
.Where(x => x.UserId == userId)
|
return await _invitationRecordRepository._DbQueryable
|
||||||
.FirstAsync();
|
.Where(x => x.InvitedUserId == userId)
|
||||||
|
.AnyAsync();
|
||||||
return inviteCode?.IsUserInvited ?? false;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
|
|||||||
@@ -1,64 +1,134 @@
|
|||||||
using Volo.Abp.Domain.Services;
|
using SqlSugar;
|
||||||
using Volo.Abp.Users;
|
using Volo.Abp.Domain.Services;
|
||||||
|
using Yi.Framework.AiHub.Domain.Entities;
|
||||||
using Yi.Framework.AiHub.Domain.Entities.OpenApi;
|
using Yi.Framework.AiHub.Domain.Entities.OpenApi;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Consts;
|
||||||
using Yi.Framework.SqlSugarCore.Abstractions;
|
using Yi.Framework.SqlSugarCore.Abstractions;
|
||||||
|
|
||||||
namespace Yi.Framework.AiHub.Domain.Managers;
|
namespace Yi.Framework.AiHub.Domain.Managers;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Token验证结果
|
||||||
|
/// </summary>
|
||||||
|
public class TokenValidationResult
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 用户Id
|
||||||
|
/// </summary>
|
||||||
|
public Guid UserId { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Token Id
|
||||||
|
/// </summary>
|
||||||
|
public Guid TokenId { get; set; }
|
||||||
|
}
|
||||||
|
|
||||||
public class TokenManager : DomainService
|
public class TokenManager : DomainService
|
||||||
{
|
{
|
||||||
private readonly ISqlSugarRepository<TokenAggregateRoot> _tokenRepository;
|
private readonly ISqlSugarRepository<TokenAggregateRoot> _tokenRepository;
|
||||||
|
private readonly ISqlSugarRepository<UsageStatisticsAggregateRoot> _usageStatisticsRepository;
|
||||||
|
|
||||||
public TokenManager(ISqlSugarRepository<TokenAggregateRoot> tokenRepository)
|
public TokenManager(
|
||||||
|
ISqlSugarRepository<TokenAggregateRoot> tokenRepository,
|
||||||
|
ISqlSugarRepository<UsageStatisticsAggregateRoot> usageStatisticsRepository)
|
||||||
{
|
{
|
||||||
_tokenRepository = tokenRepository;
|
_tokenRepository = tokenRepository;
|
||||||
|
_usageStatisticsRepository = usageStatisticsRepository;
|
||||||
}
|
}
|
||||||
|
|
||||||
public async Task<string?> GetAsync(Guid userId)
|
/// <summary>
|
||||||
{
|
/// 验证Token并返回用户Id和TokenId
|
||||||
var entity = await _tokenRepository._DbQueryable.FirstAsync(x => x.UserId == userId);
|
/// </summary>
|
||||||
if (entity is not null)
|
/// <param name="token">Token密钥</param>
|
||||||
{
|
/// <param name="modelId">模型Id(用于判断是否是尊享模型需要检查额度)</param>
|
||||||
return entity.Token;
|
/// <returns>Token验证结果</returns>
|
||||||
}
|
public async Task<TokenValidationResult> ValidateTokenAsync(string? token, string? modelId = null)
|
||||||
else
|
|
||||||
{
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
public async Task CreateAsync(Guid userId)
|
|
||||||
{
|
|
||||||
var entity = await _tokenRepository._DbQueryable.FirstAsync(x => x.UserId == userId);
|
|
||||||
if (entity is not null)
|
|
||||||
{
|
|
||||||
entity.ResetToken();
|
|
||||||
await _tokenRepository.UpdateAsync(entity);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
var token = new TokenAggregateRoot(userId);
|
|
||||||
await _tokenRepository.InsertAsync(token);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
public async Task<Guid> GetUserIdAsync(string? token)
|
|
||||||
{
|
{
|
||||||
if (token is null)
|
if (token is null)
|
||||||
{
|
{
|
||||||
throw new UserFriendlyException("当前请求未包含token", "401");
|
throw new UserFriendlyException("当前请求未包含token", "401");
|
||||||
}
|
}
|
||||||
|
|
||||||
if (token.StartsWith("yi-"))
|
if (!token.StartsWith("yi-"))
|
||||||
{
|
{
|
||||||
var entity = await _tokenRepository._DbQueryable.Where(x => x.Token == token).FirstAsync();
|
throw new UserFriendlyException("当前请求token非法", "401");
|
||||||
if (entity is null)
|
|
||||||
{
|
|
||||||
throw new UserFriendlyException("当前请求token无效", "401");
|
|
||||||
}
|
|
||||||
|
|
||||||
return entity.UserId;
|
|
||||||
}
|
}
|
||||||
throw new UserFriendlyException("当前请求token非法", "401");
|
|
||||||
|
var entity = await _tokenRepository._DbQueryable
|
||||||
|
.Where(x => x.Token == token)
|
||||||
|
.FirstAsync();
|
||||||
|
|
||||||
|
if (entity is null)
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("当前请求token无效", "401");
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查Token是否被禁用
|
||||||
|
if (entity.IsDisabled)
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("当前Token已被禁用,请启用后再使用", "403");
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查Token是否过期
|
||||||
|
if (entity.ExpireTime.HasValue && entity.ExpireTime.Value < DateTime.Now)
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("当前Token已过期,请更新过期时间或创建新的Token", "403");
|
||||||
|
}
|
||||||
|
|
||||||
|
// 如果是尊享模型且Token设置了额度限制,检查是否超限
|
||||||
|
if (!string.IsNullOrEmpty(modelId) &&
|
||||||
|
PremiumPackageConst.ModeIds.Contains(modelId) &&
|
||||||
|
entity.PremiumQuotaLimit.HasValue)
|
||||||
|
{
|
||||||
|
var usedQuota = await GetTokenPremiumUsedQuotaAsync(entity.UserId, entity.Id);
|
||||||
|
if (usedQuota >= entity.PremiumQuotaLimit.Value)
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException($"当前Token的尊享包额度已用完(已使用:{usedQuota},限制:{entity.PremiumQuotaLimit.Value}),请调整额度限制或使用其他Token", "403");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return new TokenValidationResult
|
||||||
|
{
|
||||||
|
UserId = entity.UserId,
|
||||||
|
TokenId = entity.Id
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取Token的尊享包已使用额度
|
||||||
|
/// </summary>
|
||||||
|
private async Task<long> GetTokenPremiumUsedQuotaAsync(Guid userId, Guid tokenId)
|
||||||
|
{
|
||||||
|
var premiumModelIds = PremiumPackageConst.ModeIds;
|
||||||
|
|
||||||
|
var usedQuota = await _usageStatisticsRepository._DbQueryable
|
||||||
|
.Where(x => x.UserId == userId && x.TokenId == tokenId && premiumModelIds.Contains(x.ModelId))
|
||||||
|
.SumAsync(x => x.TotalTokenCount);
|
||||||
|
|
||||||
|
return usedQuota;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取用户的Token(兼容旧接口,返回第一个可用的Token)
|
||||||
|
/// </summary>
|
||||||
|
[Obsolete("请使用 ValidateTokenAsync 方法")]
|
||||||
|
public async Task<string?> GetAsync(Guid userId)
|
||||||
|
{
|
||||||
|
var entity = await _tokenRepository._DbQueryable
|
||||||
|
.Where(x => x.UserId == userId && !x.IsDisabled)
|
||||||
|
.OrderBy(x => x.CreationTime)
|
||||||
|
.FirstAsync();
|
||||||
|
|
||||||
|
return entity?.Token;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取用户Id(兼容旧接口)
|
||||||
|
/// </summary>
|
||||||
|
[Obsolete("请使用 ValidateTokenAsync 方法")]
|
||||||
|
public async Task<Guid> GetUserIdAsync(string? token)
|
||||||
|
{
|
||||||
|
var result = await ValidateTokenAsync(token);
|
||||||
|
return result.UserId;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -18,8 +18,10 @@ public class UsageStatisticsManager : DomainService
|
|||||||
private IDistributedLockProvider DistributedLock =>
|
private IDistributedLockProvider DistributedLock =>
|
||||||
LazyServiceProvider.LazyGetRequiredService<IDistributedLockProvider>();
|
LazyServiceProvider.LazyGetRequiredService<IDistributedLockProvider>();
|
||||||
|
|
||||||
public async Task SetUsageAsync(Guid? userId, string modelId, ThorUsageResponse? tokenUsage)
|
public async Task SetUsageAsync(Guid? userId, string modelId, ThorUsageResponse? tokenUsage, Guid? tokenId = null)
|
||||||
{
|
{
|
||||||
|
var actualTokenId = tokenId ?? Guid.Empty;
|
||||||
|
|
||||||
long inputTokenCount = tokenUsage?.PromptTokens
|
long inputTokenCount = tokenUsage?.PromptTokens
|
||||||
?? tokenUsage?.InputTokens
|
?? tokenUsage?.InputTokens
|
||||||
?? 0;
|
?? 0;
|
||||||
@@ -28,10 +30,10 @@ public class UsageStatisticsManager : DomainService
|
|||||||
?? tokenUsage?.OutputTokens
|
?? tokenUsage?.OutputTokens
|
||||||
?? 0;
|
?? 0;
|
||||||
|
|
||||||
await using (await DistributedLock.AcquireLockAsync($"UsageStatistics:{userId?.ToString()}"))
|
await using (await DistributedLock.AcquireLockAsync($"UsageStatistics:{userId?.ToString()}:{actualTokenId}:{modelId}"))
|
||||||
{
|
{
|
||||||
var entity = await _repository._DbQueryable.FirstAsync(x => x.UserId == userId && x.ModelId == modelId);
|
var entity = await _repository._DbQueryable.FirstAsync(x => x.UserId == userId && x.ModelId == modelId && x.TokenId == actualTokenId);
|
||||||
//存在数据,更细
|
//存在数据,更新
|
||||||
if (entity is not null)
|
if (entity is not null)
|
||||||
{
|
{
|
||||||
entity.AddOnceChat(inputTokenCount, outputTokenCount);
|
entity.AddOnceChat(inputTokenCount, outputTokenCount);
|
||||||
@@ -40,7 +42,7 @@ public class UsageStatisticsManager : DomainService
|
|||||||
//不存在插入
|
//不存在插入
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
var usage = new UsageStatisticsAggregateRoot(userId, modelId);
|
var usage = new UsageStatisticsAggregateRoot(userId, modelId, actualTokenId);
|
||||||
usage.AddOnceChat(inputTokenCount, outputTokenCount);
|
usage.AddOnceChat(inputTokenCount, outputTokenCount);
|
||||||
await _repository.InsertAsync(usage);
|
await _repository.InsertAsync(usage);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -44,17 +44,22 @@ namespace Yi.Framework.AiHub.Domain
|
|||||||
nameof(OpenAiChatCompletionsService));
|
nameof(OpenAiChatCompletionsService));
|
||||||
services.AddKeyedTransient<IChatCompletionService, ClaudiaChatCompletionsService>(
|
services.AddKeyedTransient<IChatCompletionService, ClaudiaChatCompletionsService>(
|
||||||
nameof(ClaudiaChatCompletionsService));
|
nameof(ClaudiaChatCompletionsService));
|
||||||
|
|
||||||
#endregion
|
#endregion
|
||||||
|
|
||||||
#region Anthropic ChatCompletion
|
#region Anthropic ChatCompletion
|
||||||
|
|
||||||
services.AddKeyedTransient<IAnthropicChatCompletionService, CustomOpenAIAnthropicChatCompletionsService>(
|
|
||||||
nameof(CustomOpenAIAnthropicChatCompletionsService));
|
|
||||||
services.AddKeyedTransient<IAnthropicChatCompletionService, AnthropicChatCompletionsService>(
|
services.AddKeyedTransient<IAnthropicChatCompletionService, AnthropicChatCompletionsService>(
|
||||||
nameof(AnthropicChatCompletionsService));
|
nameof(AnthropicChatCompletionsService));
|
||||||
|
|
||||||
#endregion
|
#endregion
|
||||||
|
|
||||||
|
#region OpenAi Response
|
||||||
|
|
||||||
|
services.AddKeyedTransient<IOpenAiResponseService, OpenAiResponseService>(
|
||||||
|
nameof(OpenAiResponseService));
|
||||||
|
|
||||||
|
#endregion
|
||||||
|
|
||||||
#region Image
|
#region Image
|
||||||
|
|
||||||
@@ -73,6 +78,34 @@ namespace Yi.Framework.AiHub.Domain
|
|||||||
//ai模型特殊性兼容处理
|
//ai模型特殊性兼容处理
|
||||||
Configure<SpecialCompatibleOptions>(options =>
|
Configure<SpecialCompatibleOptions>(options =>
|
||||||
{
|
{
|
||||||
|
options.Handles.Add(request =>
|
||||||
|
{
|
||||||
|
if (request.Model == "gpt-5.1-chat")
|
||||||
|
{
|
||||||
|
request.Temperature = null;
|
||||||
|
request.TopP = null;
|
||||||
|
request.MaxCompletionTokens = request.MaxTokens;
|
||||||
|
request.MaxTokens = null;
|
||||||
|
request.PresencePenalty = null;
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
options.Handles.Add(request =>
|
||||||
|
{
|
||||||
|
if (request.Model =="gpt-5-mini")
|
||||||
|
{
|
||||||
|
request.Temperature = null;
|
||||||
|
request.TopP = null;
|
||||||
|
}
|
||||||
|
});
|
||||||
|
options.Handles.Add(request =>
|
||||||
|
{
|
||||||
|
if (request.Model == "databricks-claude-sonnet-4")
|
||||||
|
{
|
||||||
|
request.PresencePenalty = null;
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
options.Handles.Add(request =>
|
options.Handles.Add(request =>
|
||||||
{
|
{
|
||||||
if (request.Model == "o1")
|
if (request.Model == "o1")
|
||||||
@@ -101,9 +134,9 @@ namespace Yi.Framework.AiHub.Domain
|
|||||||
});
|
});
|
||||||
options.Handles.Add(request =>
|
options.Handles.Add(request =>
|
||||||
{
|
{
|
||||||
if (request.MaxTokens >= 16384)
|
if (request.MaxTokens > 128000)
|
||||||
{
|
{
|
||||||
request.MaxTokens = 16384;
|
request.MaxTokens = 128000;
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
options.AnthropicHandles.add(request =>
|
options.AnthropicHandles.add(request =>
|
||||||
@@ -128,7 +161,7 @@ namespace Yi.Framework.AiHub.Domain
|
|||||||
{
|
{
|
||||||
builder.ConfigureHttpClient(client =>
|
builder.ConfigureHttpClient(client =>
|
||||||
{
|
{
|
||||||
client.DefaultRequestHeaders.Add("User-Agent","Apifox/1.0.0 (https://apifox.com)");
|
client.DefaultRequestHeaders.Add("User-Agent", "Apifox/1.0.0 (https://apifox.com)");
|
||||||
client.Timeout = TimeSpan.FromMinutes(10);
|
client.Timeout = TimeSpan.FromMinutes(10);
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
|
|||||||
@@ -3,7 +3,6 @@ using Mapster;
|
|||||||
using Microsoft.AspNetCore.Authorization;
|
using Microsoft.AspNetCore.Authorization;
|
||||||
using Microsoft.AspNetCore.Mvc;
|
using Microsoft.AspNetCore.Mvc;
|
||||||
using SqlSugar;
|
using SqlSugar;
|
||||||
using TencentCloud.Pds.V20210701.Models;
|
|
||||||
using Volo.Abp;
|
using Volo.Abp;
|
||||||
using Volo.Abp.Application.Dtos;
|
using Volo.Abp.Application.Dtos;
|
||||||
using Volo.Abp.EventBus.Local;
|
using Volo.Abp.EventBus.Local;
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
using TencentCloud.Tbm.V20180129.Models;
|
using Volo.Abp.DependencyInjection;
|
||||||
using Volo.Abp.DependencyInjection;
|
|
||||||
using Volo.Abp.Domain.Entities.Events;
|
using Volo.Abp.Domain.Entities.Events;
|
||||||
using Volo.Abp.EventBus;
|
using Volo.Abp.EventBus;
|
||||||
using Volo.Abp.EventBus.Local;
|
using Volo.Abp.EventBus.Local;
|
||||||
|
|||||||
@@ -56,8 +56,8 @@ namespace Yi.Framework.Rbac.Application.Services
|
|||||||
return new NotFoundResult();
|
return new NotFoundResult();
|
||||||
}
|
}
|
||||||
|
|
||||||
var steam = await File.ReadAllBytesAsync(path);
|
var stream = new FileStream(path, FileMode.Open, FileAccess.Read, FileShare.Read);
|
||||||
return new FileContentResult(steam, file.GetMimeMapping());
|
return new FileStreamResult(stream, file!.GetMimeMapping());
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
|
|||||||
@@ -3,7 +3,6 @@ using Microsoft.AspNetCore.Mvc;
|
|||||||
using Microsoft.Extensions.Configuration;
|
using Microsoft.Extensions.Configuration;
|
||||||
using Microsoft.Extensions.Options;
|
using Microsoft.Extensions.Options;
|
||||||
using Microsoft.VisualBasic;
|
using Microsoft.VisualBasic;
|
||||||
using TencentCloud.Mna.V20210119.Models;
|
|
||||||
using Volo.Abp.Application.Services;
|
using Volo.Abp.Application.Services;
|
||||||
using Volo.Abp.Caching;
|
using Volo.Abp.Caching;
|
||||||
using Volo.Abp.DependencyInjection;
|
using Volo.Abp.DependencyInjection;
|
||||||
|
|||||||
@@ -51,7 +51,7 @@ namespace Yi.Framework.Rbac.Application.Services.System
|
|||||||
.WhereIF(!string.IsNullOrEmpty(input.DeptName), u => u.DeptName.Contains(input.DeptName!))
|
.WhereIF(!string.IsNullOrEmpty(input.DeptName), u => u.DeptName.Contains(input.DeptName!))
|
||||||
.WhereIF(input.State is not null, u => u.State == input.State)
|
.WhereIF(input.State is not null, u => u.State == input.State)
|
||||||
.OrderBy(u => u.OrderNum, OrderByType.Asc)
|
.OrderBy(u => u.OrderNum, OrderByType.Asc)
|
||||||
.ToPageListAsync(input.SkipCount, input.MaxResultCount, total);
|
.ToListAsync();
|
||||||
return new PagedResultDto<DeptGetListOutputDto>
|
return new PagedResultDto<DeptGetListOutputDto>
|
||||||
{
|
{
|
||||||
Items = await MapToGetListOutputDtosAsync(entities),
|
Items = await MapToGetListOutputDtosAsync(entities),
|
||||||
|
|||||||
@@ -1,3 +1,4 @@
|
|||||||
|
using Microsoft.AspNetCore.Mvc;
|
||||||
using SqlSugar;
|
using SqlSugar;
|
||||||
using Volo.Abp.Application.Dtos;
|
using Volo.Abp.Application.Dtos;
|
||||||
using Yi.Framework.Ddd.Application;
|
using Yi.Framework.Ddd.Application;
|
||||||
@@ -54,5 +55,25 @@ namespace Yi.Framework.Rbac.Application.Services.System
|
|||||||
throw new UserFriendlyException(RoleConst.Exist);
|
throw new UserFriendlyException(RoleConst.Exist);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 更新状态
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="id"></param>
|
||||||
|
/// <param name="state"></param>
|
||||||
|
/// <returns></returns>
|
||||||
|
[Route("post/{id}/{state}")]
|
||||||
|
public async Task<PostGetOutputDto> UpdateStateAsync([FromRoute] Guid id, [FromRoute] bool state)
|
||||||
|
{
|
||||||
|
var entity = await _repository.GetByIdAsync(id);
|
||||||
|
if (entity is null)
|
||||||
|
{
|
||||||
|
throw new ApplicationException("岗位未存在");
|
||||||
|
}
|
||||||
|
|
||||||
|
entity.State = state;
|
||||||
|
await _repository.UpdateAsync(entity);
|
||||||
|
return await MapToGetOutputDtoAsync(entity);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -1,6 +1,5 @@
|
|||||||
using Microsoft.AspNetCore.Mvc;
|
using Microsoft.AspNetCore.Mvc;
|
||||||
using SqlSugar;
|
using SqlSugar;
|
||||||
using TencentCloud.Tcr.V20190924.Models;
|
|
||||||
using Volo.Abp;
|
using Volo.Abp;
|
||||||
using Volo.Abp.Application.Dtos;
|
using Volo.Abp.Application.Dtos;
|
||||||
using Volo.Abp.Caching;
|
using Volo.Abp.Caching;
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
namespace Yi.Framework.Rbac.Domain.Shared.Caches;
|
namespace Yi.Framework.Rbac.Domain.Shared.Caches;
|
||||||
|
|
||||||
public class FileCacheItem
|
public class FileCacheItem
|
||||||
{
|
{
|
||||||
|
|||||||
@@ -34,8 +34,7 @@ namespace Yi.Framework.Rbac.Domain.Entities
|
|||||||
var type = GetFileType();
|
var type = GetFileType();
|
||||||
|
|
||||||
var savePath = GetSaveFilePath();
|
var savePath = GetSaveFilePath();
|
||||||
var filePath = Path.Combine(savePath, this.FileName);
|
this.FilePath = savePath;
|
||||||
this.FilePath = filePath;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
|
|||||||
@@ -1,59 +1,59 @@
|
|||||||
using System;
|
// using System;
|
||||||
using System.Collections.Generic;
|
// using System.Collections.Generic;
|
||||||
using System.Linq;
|
// using System.Linq;
|
||||||
using System.Text;
|
// using System.Text;
|
||||||
using System.Threading.Tasks;
|
// using System.Threading.Tasks;
|
||||||
using TencentCloud.Common.Profile;
|
// using TencentCloud.Common.Profile;
|
||||||
using TencentCloud.Common;
|
// using TencentCloud.Common;
|
||||||
using TencentCloud.Sms.V20210111.Models;
|
// using TencentCloud.Sms.V20210111.Models;
|
||||||
using TencentCloud.Sms.V20210111;
|
// using TencentCloud.Sms.V20210111;
|
||||||
using Volo.Abp.Domain.Services;
|
// using Volo.Abp.Domain.Services;
|
||||||
using Microsoft.Extensions.Logging;
|
// using Microsoft.Extensions.Logging;
|
||||||
|
//
|
||||||
namespace Yi.Framework.Rbac.Domain.Managers
|
// namespace Yi.Framework.Rbac.Domain.Managers
|
||||||
{
|
// {
|
||||||
public class TencentCloudManager : DomainService
|
// public class TencentCloudManager : DomainService
|
||||||
{
|
// {
|
||||||
private ILogger<TencentCloudManager> _logger;
|
// private ILogger<TencentCloudManager> _logger;
|
||||||
public TencentCloudManager(ILogger<TencentCloudManager> logger)
|
// public TencentCloudManager(ILogger<TencentCloudManager> logger)
|
||||||
{
|
// {
|
||||||
_logger= logger;
|
// _logger= logger;
|
||||||
}
|
// }
|
||||||
|
//
|
||||||
public async Task SendSmsAsync()
|
// public async Task SendSmsAsync()
|
||||||
{
|
// {
|
||||||
|
//
|
||||||
try
|
// try
|
||||||
{
|
// {
|
||||||
// 实例化一个认证对象,入参需要传入腾讯云账户 SecretId 和 SecretKey,此处还需注意密钥对的保密
|
// // 实例化一个认证对象,入参需要传入腾讯云账户 SecretId 和 SecretKey,此处还需注意密钥对的保密
|
||||||
// 代码泄露可能会导致 SecretId 和 SecretKey 泄露,并威胁账号下所有资源的安全性。以下代码示例仅供参考,建议采用更安全的方式来使用密钥,请参见:https://cloud.tencent.com/document/product/1278/85305
|
// // 代码泄露可能会导致 SecretId 和 SecretKey 泄露,并威胁账号下所有资源的安全性。以下代码示例仅供参考,建议采用更安全的方式来使用密钥,请参见:https://cloud.tencent.com/document/product/1278/85305
|
||||||
// 密钥可前往官网控制台 https://console.cloud.tencent.com/cam/capi 进行获取
|
// // 密钥可前往官网控制台 https://console.cloud.tencent.com/cam/capi 进行获取
|
||||||
Credential cred = new Credential
|
// Credential cred = new Credential
|
||||||
{
|
// {
|
||||||
SecretId = "SecretId",
|
// SecretId = "SecretId",
|
||||||
SecretKey = "SecretKey"
|
// SecretKey = "SecretKey"
|
||||||
};
|
// };
|
||||||
// 实例化一个client选项,可选的,没有特殊需求可以跳过
|
// // 实例化一个client选项,可选的,没有特殊需求可以跳过
|
||||||
ClientProfile clientProfile = new ClientProfile();
|
// ClientProfile clientProfile = new ClientProfile();
|
||||||
// 实例化一个http选项,可选的,没有特殊需求可以跳过
|
// // 实例化一个http选项,可选的,没有特殊需求可以跳过
|
||||||
HttpProfile httpProfile = new HttpProfile();
|
// HttpProfile httpProfile = new HttpProfile();
|
||||||
httpProfile.Endpoint = ("sms.tencentcloudapi.com");
|
// httpProfile.Endpoint = ("sms.tencentcloudapi.com");
|
||||||
clientProfile.HttpProfile = httpProfile;
|
// clientProfile.HttpProfile = httpProfile;
|
||||||
|
//
|
||||||
// 实例化要请求产品的client对象,clientProfile是可选的
|
// // 实例化要请求产品的client对象,clientProfile是可选的
|
||||||
SmsClient client = new SmsClient(cred, "", clientProfile);
|
// SmsClient client = new SmsClient(cred, "", clientProfile);
|
||||||
// 实例化一个请求对象,每个接口都会对应一个request对象
|
// // 实例化一个请求对象,每个接口都会对应一个request对象
|
||||||
SendSmsRequest req = new SendSmsRequest();
|
// SendSmsRequest req = new SendSmsRequest();
|
||||||
|
//
|
||||||
// 返回的resp是一个SendSmsResponse的实例,与请求对象对应
|
// // 返回的resp是一个SendSmsResponse的实例,与请求对象对应
|
||||||
SendSmsResponse resp = await client.SendSms(req);
|
// SendSmsResponse resp = await client.SendSms(req);
|
||||||
// 输出json格式的字符串回包
|
// // 输出json格式的字符串回包
|
||||||
_logger.LogInformation("腾讯云Sms返回:"+AbstractModel.ToJsonString(resp));
|
// _logger.LogInformation("腾讯云Sms返回:"+AbstractModel.ToJsonString(resp));
|
||||||
}
|
// }
|
||||||
catch (Exception e)
|
// catch (Exception e)
|
||||||
{
|
// {
|
||||||
_logger.LogError(e,e.ToString());
|
// _logger.LogError(e,e.ToString());
|
||||||
}
|
// }
|
||||||
}
|
// }
|
||||||
}
|
// }
|
||||||
}
|
// }
|
||||||
|
|||||||
@@ -8,7 +8,7 @@
|
|||||||
|
|
||||||
<PackageReference Include="IPTools.China" Version="1.6.0" />
|
<PackageReference Include="IPTools.China" Version="1.6.0" />
|
||||||
|
|
||||||
<PackageReference Include="TencentCloudSDK" Version="3.0.966" />
|
<!-- <PackageReference Include="TencentCloudSDK" Version="3.0.966" />-->
|
||||||
|
|
||||||
<PackageReference Include="UAParser" Version="3.1.47" />
|
<PackageReference Include="UAParser" Version="3.1.47" />
|
||||||
|
|
||||||
|
|||||||
@@ -29,13 +29,18 @@ using Volo.Abp.Swashbuckle;
|
|||||||
using Yi.Abp.Application;
|
using Yi.Abp.Application;
|
||||||
using Yi.Abp.SqlsugarCore;
|
using Yi.Abp.SqlsugarCore;
|
||||||
using Yi.Framework.AiHub.Application;
|
using Yi.Framework.AiHub.Application;
|
||||||
|
using Yi.Framework.AiHub.Application.Services;
|
||||||
using Yi.Framework.AiHub.Domain.Entities;
|
using Yi.Framework.AiHub.Domain.Entities;
|
||||||
|
using Yi.Framework.AiHub.Domain.Entities.Chat;
|
||||||
|
using Yi.Framework.AiHub.Domain.Entities.Model;
|
||||||
|
using Yi.Framework.AiHub.Domain.Entities.OpenApi;
|
||||||
using Yi.Framework.AspNetCore;
|
using Yi.Framework.AspNetCore;
|
||||||
using Yi.Framework.AspNetCore.Authentication.OAuth;
|
using Yi.Framework.AspNetCore.Authentication.OAuth;
|
||||||
using Yi.Framework.AspNetCore.Authentication.OAuth.Gitee;
|
using Yi.Framework.AspNetCore.Authentication.OAuth.Gitee;
|
||||||
using Yi.Framework.AspNetCore.Authentication.OAuth.QQ;
|
using Yi.Framework.AspNetCore.Authentication.OAuth.QQ;
|
||||||
using Yi.Framework.AspNetCore.Microsoft.AspNetCore.Builder;
|
using Yi.Framework.AspNetCore.Microsoft.AspNetCore.Builder;
|
||||||
using Yi.Framework.AspNetCore.Microsoft.Extensions.DependencyInjection;
|
using Yi.Framework.AspNetCore.Microsoft.Extensions.DependencyInjection;
|
||||||
|
using Yi.Framework.AspNetCore.UnifyResult;
|
||||||
using Yi.Framework.BackgroundWorkers.Hangfire;
|
using Yi.Framework.BackgroundWorkers.Hangfire;
|
||||||
using Yi.Framework.Bbs.Application;
|
using Yi.Framework.Bbs.Application;
|
||||||
using Yi.Framework.Bbs.Application.Extensions;
|
using Yi.Framework.Bbs.Application.Extensions;
|
||||||
@@ -125,6 +130,7 @@ namespace Yi.Abp.Web
|
|||||||
});
|
});
|
||||||
|
|
||||||
//采用furion格式的规范化api,默认不开启,使用abp优雅的方式
|
//采用furion格式的规范化api,默认不开启,使用abp优雅的方式
|
||||||
|
//前置:需要将管道工作单元前加上app.Properties.Add("_AbpExceptionHandlingMiddleware_Added",false);
|
||||||
//你没看错。。。
|
//你没看错。。。
|
||||||
//service.AddFurionUnifyResultApi();
|
//service.AddFurionUnifyResultApi();
|
||||||
|
|
||||||
@@ -282,19 +288,19 @@ namespace Yi.Abp.Web
|
|||||||
};
|
};
|
||||||
options.Events = new JwtBearerEvents
|
options.Events = new JwtBearerEvents
|
||||||
{
|
{
|
||||||
OnMessageReceived = context =>
|
OnMessageReceived = messageContext =>
|
||||||
{
|
{
|
||||||
//优先Query中获取,再去cookies中获取
|
//优先Query中获取,再去cookies中获取
|
||||||
var accessToken = context.Request.Query["access_token"];
|
var accessToken = messageContext.Request.Query["access_token"];
|
||||||
if (!string.IsNullOrEmpty(accessToken))
|
if (!string.IsNullOrEmpty(accessToken))
|
||||||
{
|
{
|
||||||
context.Token = accessToken;
|
messageContext.Token = accessToken;
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
if (context.Request.Cookies.TryGetValue("Token", out var cookiesToken))
|
if (messageContext.Request.Cookies.TryGetValue("Token", out var cookiesToken))
|
||||||
{
|
{
|
||||||
context.Token = cookiesToken;
|
messageContext.Token = cookiesToken;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -315,19 +321,19 @@ namespace Yi.Abp.Web
|
|||||||
};
|
};
|
||||||
options.Events = new JwtBearerEvents
|
options.Events = new JwtBearerEvents
|
||||||
{
|
{
|
||||||
OnMessageReceived = context =>
|
OnMessageReceived = messageContext =>
|
||||||
{
|
{
|
||||||
var refresh_token = context.Request.Headers["refresh_token"];
|
var headerRefreshToken = messageContext.Request.Headers["refresh_token"];
|
||||||
if (!string.IsNullOrEmpty(refresh_token))
|
if (!string.IsNullOrEmpty(headerRefreshToken))
|
||||||
{
|
{
|
||||||
context.Token = refresh_token;
|
messageContext.Token = headerRefreshToken;
|
||||||
return Task.CompletedTask;
|
return Task.CompletedTask;
|
||||||
}
|
}
|
||||||
|
|
||||||
var refreshToken = context.Request.Query["refresh_token"];
|
var queryRefreshToken = messageContext.Request.Query["refresh_token"];
|
||||||
if (!string.IsNullOrEmpty(refreshToken))
|
if (!string.IsNullOrEmpty(queryRefreshToken))
|
||||||
{
|
{
|
||||||
context.Token = refreshToken;
|
messageContext.Token = queryRefreshToken;
|
||||||
}
|
}
|
||||||
|
|
||||||
return Task.CompletedTask;
|
return Task.CompletedTask;
|
||||||
@@ -352,9 +358,9 @@ namespace Yi.Abp.Web
|
|||||||
var app = context.GetApplicationBuilder();
|
var app = context.GetApplicationBuilder();
|
||||||
app.UseRouting();
|
app.UseRouting();
|
||||||
|
|
||||||
// app.ApplicationServices.GetRequiredService<ISqlSugarDbContext>().SqlSugarClient.CodeFirst.InitTables<CardFlipTaskAggregateRoot>();
|
// app.ApplicationServices.GetRequiredService<ISqlSugarDbContext>().SqlSugarClient.CodeFirst.InitTables<AiModelEntity>();
|
||||||
// app.ApplicationServices.GetRequiredService<ISqlSugarDbContext>().SqlSugarClient.CodeFirst.InitTables<InvitationRecordAggregateRoot>();
|
// app.ApplicationServices.GetRequiredService<ISqlSugarDbContext>().SqlSugarClient.CodeFirst.InitTables<TokenAggregateRoot>();
|
||||||
// app.ApplicationServices.GetRequiredService<ISqlSugarDbContext>().SqlSugarClient.CodeFirst.InitTables<InviteCodeAggregateRoot>();
|
// app.ApplicationServices.GetRequiredService<ISqlSugarDbContext>().SqlSugarClient.CodeFirst.InitTables<UsageStatisticsAggregateRoot>();
|
||||||
|
|
||||||
//跨域
|
//跨域
|
||||||
app.UseCors(DefaultCorsPolicyName);
|
app.UseCors(DefaultCorsPolicyName);
|
||||||
@@ -400,8 +406,7 @@ namespace Yi.Abp.Web
|
|||||||
app.UseDefaultFiles();
|
app.UseDefaultFiles();
|
||||||
app.UseDirectoryBrowser("/api/app/wwwroot");
|
app.UseDirectoryBrowser("/api/app/wwwroot");
|
||||||
|
|
||||||
|
//app.Properties.Add("_AbpExceptionHandlingMiddleware_Added",false);
|
||||||
// app.Properties.Add("_AbpExceptionHandlingMiddleware_Added",false);
|
|
||||||
//工作单元
|
//工作单元
|
||||||
app.UseUnitOfWork();
|
app.UseUnitOfWork();
|
||||||
|
|
||||||
|
|||||||
@@ -1,7 +1,5 @@
|
|||||||
using Microsoft.Extensions.DependencyInjection;
|
using Microsoft.Extensions.DependencyInjection;
|
||||||
using Shouldly;
|
using Shouldly;
|
||||||
using TencentCloud.Ame.V20190916.Models;
|
|
||||||
using TencentCloud.Tiw.V20190919.Models;
|
|
||||||
using Volo.Abp.Domain.Repositories;
|
using Volo.Abp.Domain.Repositories;
|
||||||
using Xunit;
|
using Xunit;
|
||||||
using Yi.Framework.Rbac.Application.Contracts.Dtos.User;
|
using Yi.Framework.Rbac.Application.Contracts.Dtos.User;
|
||||||
|
|||||||
@@ -1,7 +0,0 @@
|
|||||||
{
|
|
||||||
"env": {
|
|
||||||
"ANTHROPIC_AUTH_TOKEN": "sk-lVcGFMOQfXYtZWp1rD4SaBNDNpM270UX2wDqWh",
|
|
||||||
"ANTHROPIC_BASE_URL": "https://api.token-ai.cn",
|
|
||||||
"ANTHROPIC_MODEL": "gpt-4o-mini"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,7 +1,8 @@
|
|||||||
{
|
{
|
||||||
"permissions": {
|
"permissions": {
|
||||||
"allow": [
|
"allow": [
|
||||||
"WebFetch(domain:www.donet5.com)"
|
"Bash(npx vue-tsc --noEmit)",
|
||||||
|
"Bash(timeout 60 npx vue-tsc:*)"
|
||||||
],
|
],
|
||||||
"deny": [],
|
"deny": [],
|
||||||
"ask": []
|
"ask": []
|
||||||
@@ -15,6 +15,9 @@ VITE_WEB_BASE_API = '/dev-api'
|
|||||||
VITE_API_URL = http://localhost:19001/api/app
|
VITE_API_URL = http://localhost:19001/api/app
|
||||||
#VITE_API_URL=http://data.ccnetcore.com:19001/api/app
|
#VITE_API_URL=http://data.ccnetcore.com:19001/api/app
|
||||||
|
|
||||||
|
# 文件上传接口域名
|
||||||
|
VITE_FILE_UPLOAD_API = https://ai.ccnetcore.com
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
# SSO单点登录url
|
# SSO单点登录url
|
||||||
|
|||||||
@@ -13,6 +13,9 @@ VITE_WEB_BASE_API = '/prod-api'
|
|||||||
# 本地接口
|
# 本地接口
|
||||||
VITE_API_URL = http://data.ccnetcore.com:19001/api/app
|
VITE_API_URL = http://data.ccnetcore.com:19001/api/app
|
||||||
|
|
||||||
|
# 文件上传接口域名
|
||||||
|
VITE_FILE_UPLOAD_API = https://ai.ccnetcore.com
|
||||||
|
|
||||||
# 是否在打包时开启压缩,支持 gzip 和 brotli
|
# 是否在打包时开启压缩,支持 gzip 和 brotli
|
||||||
VITE_BUILD_COMPRESS = gzip
|
VITE_BUILD_COMPRESS = gzip
|
||||||
|
|
||||||
|
|||||||
@@ -1,78 +0,0 @@
|
|||||||
{
|
|
||||||
"globals": {
|
|
||||||
"Component": true,
|
|
||||||
"ComponentPublicInstance": true,
|
|
||||||
"ComputedRef": true,
|
|
||||||
"DirectiveBinding": true,
|
|
||||||
"EffectScope": true,
|
|
||||||
"ElMessage": true,
|
|
||||||
"ElMessageBox": true,
|
|
||||||
"ExtractDefaultPropTypes": true,
|
|
||||||
"ExtractPropTypes": true,
|
|
||||||
"ExtractPublicPropTypes": true,
|
|
||||||
"InjectionKey": true,
|
|
||||||
"MaybeRef": true,
|
|
||||||
"MaybeRefOrGetter": true,
|
|
||||||
"PropType": true,
|
|
||||||
"Ref": true,
|
|
||||||
"Slot": true,
|
|
||||||
"Slots": true,
|
|
||||||
"VNode": true,
|
|
||||||
"WritableComputedRef": true,
|
|
||||||
"computed": true,
|
|
||||||
"createApp": true,
|
|
||||||
"customRef": true,
|
|
||||||
"defineAsyncComponent": true,
|
|
||||||
"defineComponent": true,
|
|
||||||
"effectScope": true,
|
|
||||||
"getCurrentInstance": true,
|
|
||||||
"getCurrentScope": true,
|
|
||||||
"h": true,
|
|
||||||
"inject": true,
|
|
||||||
"isProxy": true,
|
|
||||||
"isReactive": true,
|
|
||||||
"isReadonly": true,
|
|
||||||
"isRef": true,
|
|
||||||
"markRaw": true,
|
|
||||||
"nextTick": true,
|
|
||||||
"onActivated": true,
|
|
||||||
"onBeforeMount": true,
|
|
||||||
"onBeforeUnmount": true,
|
|
||||||
"onBeforeUpdate": true,
|
|
||||||
"onDeactivated": true,
|
|
||||||
"onErrorCaptured": true,
|
|
||||||
"onMounted": true,
|
|
||||||
"onRenderTracked": true,
|
|
||||||
"onRenderTriggered": true,
|
|
||||||
"onScopeDispose": true,
|
|
||||||
"onServerPrefetch": true,
|
|
||||||
"onUnmounted": true,
|
|
||||||
"onUpdated": true,
|
|
||||||
"onWatcherCleanup": true,
|
|
||||||
"provide": true,
|
|
||||||
"reactive": true,
|
|
||||||
"readonly": true,
|
|
||||||
"ref": true,
|
|
||||||
"resolveComponent": true,
|
|
||||||
"shallowReactive": true,
|
|
||||||
"shallowReadonly": true,
|
|
||||||
"shallowRef": true,
|
|
||||||
"toRaw": true,
|
|
||||||
"toRef": true,
|
|
||||||
"toRefs": true,
|
|
||||||
"toValue": true,
|
|
||||||
"triggerRef": true,
|
|
||||||
"unref": true,
|
|
||||||
"useAttrs": true,
|
|
||||||
"useCssModule": true,
|
|
||||||
"useCssVars": true,
|
|
||||||
"useId": true,
|
|
||||||
"useModel": true,
|
|
||||||
"useSlots": true,
|
|
||||||
"useTemplateRef": true,
|
|
||||||
"watch": true,
|
|
||||||
"watchEffect": true,
|
|
||||||
"watchPostEffect": true,
|
|
||||||
"watchSyncEffect": true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
7
Yi.Ai.Vue3/.gitignore
vendored
7
Yi.Ai.Vue3/.gitignore
vendored
@@ -23,3 +23,10 @@ dist-ssr
|
|||||||
*.njsproj
|
*.njsproj
|
||||||
*.sln
|
*.sln
|
||||||
*.sw?
|
*.sw?
|
||||||
|
|
||||||
|
/.eslintrc-auto-import.json
|
||||||
|
/types/auto-imports.d.ts
|
||||||
|
/types/components.d.ts
|
||||||
|
/types/import_meta.d.ts
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -112,7 +112,7 @@
|
|||||||
<body>
|
<body>
|
||||||
<!-- 加载动画容器 -->
|
<!-- 加载动画容器 -->
|
||||||
<div id="yixinai-loader" class="loader-container">
|
<div id="yixinai-loader" class="loader-container">
|
||||||
<div class="loader-title">意心Ai 2.2</div>
|
<div class="loader-title">意心Ai 2.8</div>
|
||||||
<div class="loader-subtitle">海外地址,仅首次访问预计加载约10秒</div>
|
<div class="loader-subtitle">海外地址,仅首次访问预计加载约10秒</div>
|
||||||
<div class="loader-logo">
|
<div class="loader-logo">
|
||||||
<div class="pulse-box"></div>
|
<div class="pulse-box"></div>
|
||||||
|
|||||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user