Compare commits
81 Commits
codex
...
ai-agent-b
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
a1be2bebf7 | ||
|
|
80dcd76749 | ||
|
|
953fbc043b | ||
|
|
64bc65114a | ||
|
|
ae9d778ac7 | ||
|
|
77a9a64a41 | ||
|
|
0c31b97824 | ||
|
|
70ae2fab44 | ||
|
|
411a9058ca | ||
|
|
4b9f845fae | ||
|
|
bdaa53bac8 | ||
|
|
e5b81c08f3 | ||
|
|
1ffab89e97 | ||
|
|
5440b226c4 | ||
|
|
90c6022839 | ||
|
|
184467e482 | ||
|
|
68045d6458 | ||
|
|
d52f17a17b | ||
|
|
047937af4c | ||
|
|
a9267bfc0e | ||
|
|
1019fd685b | ||
|
|
34246d8a62 | ||
|
|
599b6335d5 | ||
|
|
46bc48d1c1 | ||
|
|
17675e702d | ||
|
|
639c683144 | ||
|
|
96a21210b5 | ||
|
|
7495dc86a0 | ||
|
|
ee4cb20eef | ||
|
|
9ca3cd0b1a | ||
|
|
eb6ec06157 | ||
|
|
62940ae25a | ||
|
|
dfc143379f | ||
|
|
bd3a9a5ce8 | ||
|
|
ec4fdc39fe | ||
|
|
3c3e134d2b | ||
|
|
81089cc058 | ||
|
|
681194a517 | ||
|
|
8f515f76c0 | ||
|
|
fcb74eb28c | ||
|
|
44afdaef8e | ||
|
|
41c55d088d | ||
|
|
3b71fe3135 | ||
|
|
4326c41258 | ||
|
|
7f0d57b311 | ||
|
|
75c208dafc | ||
|
|
8021ca9eff | ||
|
|
2cf06a5677 | ||
|
|
2fa42cd8a3 | ||
|
|
a600eb9e7e | ||
|
|
fcf0fd7f70 | ||
|
|
4e421c160c | ||
|
|
340e2016d6 | ||
|
|
5dfaead60e | ||
|
|
c8acb12e4a | ||
|
|
5fbcb8cbd9 | ||
|
|
fd8d4399d3 | ||
|
|
6f1efafd86 | ||
|
|
2714a507d9 | ||
|
|
9a9230786b | ||
|
|
4a8b58a65c | ||
|
|
7d81f88658 | ||
|
|
0ce3c0bbdd | ||
|
|
981235e6e9 | ||
|
|
d0ecb232a1 | ||
|
|
c7a52604e7 | ||
|
|
da81b2d8a3 | ||
|
|
7b14fdd8de | ||
|
|
1fc2734eb7 | ||
|
|
f3bef72ebb | ||
|
|
7e6d2e829b | ||
|
|
944626960b | ||
|
|
c073868989 | ||
|
|
d2981100fa | ||
|
|
ce4f7e5711 | ||
|
|
cc812ba2cb | ||
|
|
8a6e5abf48 | ||
|
|
8b191330b8 | ||
|
|
5ed79c6dd0 | ||
|
|
6e2ca8f1c3 | ||
|
|
a46a552097 |
@@ -2,7 +2,8 @@
|
|||||||
"permissions": {
|
"permissions": {
|
||||||
"allow": [
|
"allow": [
|
||||||
"Bash(dotnet build \"E:\\code\\github\\Yi\\Yi.Abp.Net8\\module\\ai-hub\\Yi.Framework.AiHub.Application\\Yi.Framework.AiHub.Application.csproj\" --no-restore)",
|
"Bash(dotnet build \"E:\\code\\github\\Yi\\Yi.Abp.Net8\\module\\ai-hub\\Yi.Framework.AiHub.Application\\Yi.Framework.AiHub.Application.csproj\" --no-restore)",
|
||||||
"Read(//e/code/github/Yi/Yi.Ai.Vue3/**)"
|
"Read(//e/code/github/Yi/Yi.Ai.Vue3/**)",
|
||||||
|
"Bash(dotnet build:*)"
|
||||||
],
|
],
|
||||||
"deny": [],
|
"deny": [],
|
||||||
"ask": []
|
"ask": []
|
||||||
|
|||||||
129
Yi.Abp.Net8/CLAUDE.md
Normal file
129
Yi.Abp.Net8/CLAUDE.md
Normal file
@@ -0,0 +1,129 @@
|
|||||||
|
# CLAUDE.md
|
||||||
|
|
||||||
|
This file provides guidance to Claude Code (claude.ai/code) when working with code in this repository.
|
||||||
|
|
||||||
|
## Overview
|
||||||
|
|
||||||
|
Yi.Abp.Net8 is a modular, multi-tenant SaaS platform built on ABP Framework 8.3.4 with .NET 8.0. It uses **SqlSugar** (not EF Core) as the ORM and follows Domain-Driven Design (DDD) principles. The platform includes AI/ML features (chat, models, agents, token tracking), RBAC, forum, messaging, and digital collectibles modules.
|
||||||
|
|
||||||
|
## Architecture
|
||||||
|
|
||||||
|
### Solution Structure
|
||||||
|
|
||||||
|
```
|
||||||
|
Yi.Abp.Net8/
|
||||||
|
├── src/ # Main application host
|
||||||
|
│ └── Yi.Abp.Web/ # ASP.NET Core 8.0 Web Host (port 19001)
|
||||||
|
├── framework/ # Framework layer (shared infrastructure)
|
||||||
|
│ ├── Yi.Framework.Core # Core utilities, JSON handling
|
||||||
|
│ ├── Yi.Framework.SqlSugarCore # SqlSugar ORM abstraction (v5.1.4.197-preview22)
|
||||||
|
│ ├── Yi.Framework.SqlSugarCore.Abstractions # Repository interfaces
|
||||||
|
│ ├── Yi.Framework.AspNetCore # ASP.NET Core extensions
|
||||||
|
│ ├── Yi.Framework.AspNetCore.Authentication.OAuth # QQ, Gitee OAuth
|
||||||
|
│ ├── Yi.Framework.Ddd.Application # DDD application base classes
|
||||||
|
│ ├── Yi.Framework.BackgroundWorkers.Hangfire # Job scheduling
|
||||||
|
│ ├── Yi.Framework.Caching.FreeRedis # Redis caching
|
||||||
|
│ └── Yi.Framework.SemanticKernel # AI/ML integration
|
||||||
|
├── module/ # Business modules (each follows 5-layer DDD pattern)
|
||||||
|
│ ├── ai-hub/ # AI services (chat, models, sessions, agents)
|
||||||
|
│ ├── rbac/ # Role-Based Access Control (core auth/authz)
|
||||||
|
│ ├── bbs/ # Forum/community
|
||||||
|
│ ├── chat-hub/ # Real-time messaging
|
||||||
|
│ ├── audit-logging/ # Audit trail tracking
|
||||||
|
│ ├── code-gen/ # Code generation
|
||||||
|
│ ├── tenant-management/ # Multi-tenancy support
|
||||||
|
│ ├── digital-collectibles/ # NFT/digital assets
|
||||||
|
│ └── ai-stock/ # AI-powered stock analysis
|
||||||
|
├── test/ # Unit tests (xUnit + NSubstitute + Shouldly)
|
||||||
|
├── client/ # HTTP API clients
|
||||||
|
└── tool/ # Development utilities
|
||||||
|
```
|
||||||
|
|
||||||
|
### Module Structure (DDD Layers)
|
||||||
|
|
||||||
|
Each module follows this pattern:
|
||||||
|
|
||||||
|
```
|
||||||
|
module/[feature]/
|
||||||
|
├── [Feature].Domain.Shared/ # Constants, enums, shared DTOs
|
||||||
|
├── [Feature].Domain/ # Entities, aggregates, domain services
|
||||||
|
├── [Feature].Application.Contracts/ # Service interfaces, DTOs
|
||||||
|
├── [Feature].Application/ # Application services (implementations)
|
||||||
|
└── [Feature].SqlSugarCore/ # Repository implementations, DbContext
|
||||||
|
```
|
||||||
|
|
||||||
|
**Dependency Flow:** Application → Domain + Application.Contracts → Domain.Shared → Framework
|
||||||
|
|
||||||
|
### Module Registration
|
||||||
|
|
||||||
|
All modules are registered in `src/Yi.Abp.Web/YiAbpWebModule.cs`. When adding a new module:
|
||||||
|
|
||||||
|
1. Add `DependsOn` attribute in `YiAbpWebModule`
|
||||||
|
2. Add conventional controller in `PreConfigureServices`:
|
||||||
|
```csharp
|
||||||
|
options.ConventionalControllers.Create(typeof(YiFramework[Feature]ApplicationModule).Assembly,
|
||||||
|
option => option.RemoteServiceName = "[service-name]");
|
||||||
|
```
|
||||||
|
|
||||||
|
### API Routing
|
||||||
|
|
||||||
|
All API routes use the unified prefix: `/api/app/{service-name}/{controller}/{action}`
|
||||||
|
|
||||||
|
Registered service names:
|
||||||
|
- `default` - Main application services
|
||||||
|
- `rbac` - Authentication, authorization, user/role management
|
||||||
|
- `ai-hub` - AI chat, models, sessions, agents
|
||||||
|
- `bbs` - Forum posts and comments
|
||||||
|
- `chat-hub` - Real-time messaging
|
||||||
|
- `tenant-management` - Multi-tenant configuration
|
||||||
|
- `code-gen` - Code generation utilities
|
||||||
|
- `digital-collectibles` - NFT/digital assets
|
||||||
|
- `ai-stock` - Stock analysis
|
||||||
|
|
||||||
|
## Database
|
||||||
|
|
||||||
|
**ORM:** SqlSugar (NOT Entity Framework Core)
|
||||||
|
|
||||||
|
**Configuration** (`appsettings.json`):
|
||||||
|
```json
|
||||||
|
"DbConnOptions": {
|
||||||
|
"Url": "DataSource=yi-abp-dev.db",
|
||||||
|
"DbType": "Sqlite", // Sqlite, Mysql, Sqlserver, Oracle, PostgreSQL
|
||||||
|
"EnabledCodeFirst": true, // Auto-create tables
|
||||||
|
"EnabledDbSeed": true, // Auto-seed data
|
||||||
|
"EnabledSaasMultiTenancy": true
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
**Multi-tenancy:** Tenant isolation via `HeaderTenantResolveContributor` (NOT cookie-based). Tenant is resolved from `__tenant` header.
|
||||||
|
|
||||||
|
## Key Technologies
|
||||||
|
|
||||||
|
| Component | Technology |
|
||||||
|
|-----------|-----------|
|
||||||
|
| Framework | ABP 8.3.4 |
|
||||||
|
| .NET Version | .NET 8.0 |
|
||||||
|
| ORM | SqlSugar 5.1.4.197-preview22 |
|
||||||
|
| Authentication | JWT Bearer + Refresh Tokens |
|
||||||
|
| OAuth | QQ, Gitee |
|
||||||
|
| Caching | FreeRedis (optional) |
|
||||||
|
| Background Jobs | Hangfire (in-memory or Redis) |
|
||||||
|
| Logging | Serilog (daily rolling files) |
|
||||||
|
| Testing | xUnit + NSubstitute + Shouldly |
|
||||||
|
| Container | Autofac |
|
||||||
|
|
||||||
|
## Important Notes
|
||||||
|
|
||||||
|
- **JSON Serialization:** Uses Microsoft's `System.Text.Json`, NOT Newtonsoft.Json. Date format handled via `DatetimeJsonConverter`.
|
||||||
|
- **Multi-tenancy:** Tenant resolved from `__tenant` header, NOT cookies.
|
||||||
|
- **Rate Limiting:** Disabled in development, enabled in production (1000 req/60s sliding window).
|
||||||
|
- **Swagger:** Available at `/swagger` in development.
|
||||||
|
- **Hangfire Dashboard:** Available at `/hangfire` (requires JWT authorization).
|
||||||
|
- **Background Workers:** Disabled in development (`AbpBackgroundWorkerOptions.IsEnabled = false`).
|
||||||
|
|
||||||
|
## Development Workflow
|
||||||
|
|
||||||
|
1. **Branch:** Main branch is `abp`, current development branch is `ai-hub`.
|
||||||
|
2. **Commit Convention:** Chinese descriptive messages with prefixes (`feat:`, `fix:`, `style:`).
|
||||||
|
3. **Testing:** Run `dotnet test` before committing.
|
||||||
|
4. **Building:** Use `dotnet build --no-restore` for faster builds after initial restore.
|
||||||
@@ -0,0 +1,30 @@
|
|||||||
|
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.ActivationCode;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 批量生成激活码输入
|
||||||
|
/// </summary>
|
||||||
|
public class ActivationCodeCreateInput
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 商品类型
|
||||||
|
/// </summary>
|
||||||
|
public ActivationCodeGoodsTypeEnum GoodsType { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 数量
|
||||||
|
/// </summary>
|
||||||
|
public int Count { get; set; } = 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 批量生成激活码列表输入
|
||||||
|
/// </summary>
|
||||||
|
public class ActivationCodeCreateListInput
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 生成项列表
|
||||||
|
/// </summary>
|
||||||
|
public List<ActivationCodeCreateInput> Items { get; set; } = new();
|
||||||
|
}
|
||||||
@@ -0,0 +1,35 @@
|
|||||||
|
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.ActivationCode;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 批量生成激活码输出
|
||||||
|
/// </summary>
|
||||||
|
public class ActivationCodeCreateOutput
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 商品类型
|
||||||
|
/// </summary>
|
||||||
|
public ActivationCodeGoodsTypeEnum GoodsType { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 数量
|
||||||
|
/// </summary>
|
||||||
|
public int Count { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 激活码列表
|
||||||
|
/// </summary>
|
||||||
|
public List<string> Codes { get; set; } = new();
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 批量生成激活码列表输出
|
||||||
|
/// </summary>
|
||||||
|
public class ActivationCodeCreateListOutput
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 分组输出
|
||||||
|
/// </summary>
|
||||||
|
public List<ActivationCodeCreateOutput> Items { get; set; } = new();
|
||||||
|
}
|
||||||
@@ -0,0 +1,12 @@
|
|||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.ActivationCode;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 激活码兑换输入
|
||||||
|
/// </summary>
|
||||||
|
public class ActivationCodeRedeemInput
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 激活码
|
||||||
|
/// </summary>
|
||||||
|
public string Code { get; set; } = string.Empty;
|
||||||
|
}
|
||||||
@@ -0,0 +1,24 @@
|
|||||||
|
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.ActivationCode;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 激活码兑换输出
|
||||||
|
/// </summary>
|
||||||
|
public class ActivationCodeRedeemOutput
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 商品类型
|
||||||
|
/// </summary>
|
||||||
|
public ActivationCodeGoodsTypeEnum GoodsType { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 商品名称
|
||||||
|
/// </summary>
|
||||||
|
public string PackageName { get; set; } = string.Empty;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 内容描述
|
||||||
|
/// </summary>
|
||||||
|
public string Content { get; set; } = string.Empty;
|
||||||
|
}
|
||||||
@@ -0,0 +1,42 @@
|
|||||||
|
using System.ComponentModel.DataAnnotations;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.Channel;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 创建AI应用输入
|
||||||
|
/// </summary>
|
||||||
|
public class AiAppCreateInput
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 应用名称
|
||||||
|
/// </summary>
|
||||||
|
[Required(ErrorMessage = "应用名称不能为空")]
|
||||||
|
[StringLength(100, ErrorMessage = "应用名称不能超过100个字符")]
|
||||||
|
public string Name { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 应用终结点
|
||||||
|
/// </summary>
|
||||||
|
[Required(ErrorMessage = "应用终结点不能为空")]
|
||||||
|
[StringLength(500, ErrorMessage = "应用终结点不能超过500个字符")]
|
||||||
|
public string Endpoint { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 额外URL
|
||||||
|
/// </summary>
|
||||||
|
[StringLength(500, ErrorMessage = "额外URL不能超过500个字符")]
|
||||||
|
public string? ExtraUrl { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 应用Key
|
||||||
|
/// </summary>
|
||||||
|
[Required(ErrorMessage = "应用Key不能为空")]
|
||||||
|
[StringLength(500, ErrorMessage = "应用Key不能超过500个字符")]
|
||||||
|
public string ApiKey { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 排序
|
||||||
|
/// </summary>
|
||||||
|
[Range(0, int.MaxValue, ErrorMessage = "排序必须大于等于0")]
|
||||||
|
public int OrderNum { get; set; }
|
||||||
|
}
|
||||||
@@ -0,0 +1,42 @@
|
|||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.Channel;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// AI应用DTO
|
||||||
|
/// </summary>
|
||||||
|
public class AiAppDto
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 应用ID
|
||||||
|
/// </summary>
|
||||||
|
public Guid Id { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 应用名称
|
||||||
|
/// </summary>
|
||||||
|
public string Name { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 应用终结点
|
||||||
|
/// </summary>
|
||||||
|
public string Endpoint { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 额外URL
|
||||||
|
/// </summary>
|
||||||
|
public string? ExtraUrl { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 应用Key
|
||||||
|
/// </summary>
|
||||||
|
public string ApiKey { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 排序
|
||||||
|
/// </summary>
|
||||||
|
public int OrderNum { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 创建时间
|
||||||
|
/// </summary>
|
||||||
|
public DateTime CreationTime { get; set; }
|
||||||
|
}
|
||||||
@@ -0,0 +1,14 @@
|
|||||||
|
using Yi.Framework.Ddd.Application.Contracts;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.Channel;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取AI应用列表输入
|
||||||
|
/// </summary>
|
||||||
|
public class AiAppGetListInput : PagedAllResultRequestDto
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 搜索关键词(搜索应用名称)
|
||||||
|
/// </summary>
|
||||||
|
public string? SearchKey { get; set; }
|
||||||
|
}
|
||||||
@@ -0,0 +1,48 @@
|
|||||||
|
using System.ComponentModel.DataAnnotations;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.Channel;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 更新AI应用输入
|
||||||
|
/// </summary>
|
||||||
|
public class AiAppUpdateInput
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 应用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>
|
||||||
|
[Required(ErrorMessage = "应用终结点不能为空")]
|
||||||
|
[StringLength(500, ErrorMessage = "应用终结点不能超过500个字符")]
|
||||||
|
public string Endpoint { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 额外URL
|
||||||
|
/// </summary>
|
||||||
|
[StringLength(500, ErrorMessage = "额外URL不能超过500个字符")]
|
||||||
|
public string? ExtraUrl { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 应用Key
|
||||||
|
/// </summary>
|
||||||
|
[Required(ErrorMessage = "应用Key不能为空")]
|
||||||
|
[StringLength(500, ErrorMessage = "应用Key不能超过500个字符")]
|
||||||
|
public string ApiKey { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 排序
|
||||||
|
/// </summary>
|
||||||
|
[Range(0, int.MaxValue, ErrorMessage = "排序必须大于等于0")]
|
||||||
|
public int OrderNum { get; set; }
|
||||||
|
}
|
||||||
@@ -0,0 +1,96 @@
|
|||||||
|
using System.ComponentModel.DataAnnotations;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.Channel;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 创建AI模型输入
|
||||||
|
/// </summary>
|
||||||
|
public class AiModelCreateInput
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 处理名
|
||||||
|
/// </summary>
|
||||||
|
[Required(ErrorMessage = "处理名不能为空")]
|
||||||
|
[StringLength(100, ErrorMessage = "处理名不能超过100个字符")]
|
||||||
|
public string HandlerName { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型ID
|
||||||
|
/// </summary>
|
||||||
|
[Required(ErrorMessage = "模型ID不能为空")]
|
||||||
|
[StringLength(200, ErrorMessage = "模型ID不能超过200个字符")]
|
||||||
|
public string ModelId { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型名称
|
||||||
|
/// </summary>
|
||||||
|
[Required(ErrorMessage = "模型名称不能为空")]
|
||||||
|
[StringLength(200, ErrorMessage = "模型名称不能超过200个字符")]
|
||||||
|
public string Name { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型描述
|
||||||
|
/// </summary>
|
||||||
|
[StringLength(1000, ErrorMessage = "模型描述不能超过1000个字符")]
|
||||||
|
public string? Description { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 排序
|
||||||
|
/// </summary>
|
||||||
|
[Range(0, int.MaxValue, ErrorMessage = "排序必须大于等于0")]
|
||||||
|
public int OrderNum { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// AI应用ID
|
||||||
|
/// </summary>
|
||||||
|
[Required(ErrorMessage = "AI应用ID不能为空")]
|
||||||
|
public Guid AiAppId { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 额外信息
|
||||||
|
/// </summary>
|
||||||
|
[StringLength(2000, ErrorMessage = "额外信息不能超过2000个字符")]
|
||||||
|
public string? ExtraInfo { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型类型
|
||||||
|
/// </summary>
|
||||||
|
[Required(ErrorMessage = "模型类型不能为空")]
|
||||||
|
public ModelTypeEnum ModelType { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型API类型
|
||||||
|
/// </summary>
|
||||||
|
[Required(ErrorMessage = "模型API类型不能为空")]
|
||||||
|
public ModelApiTypeEnum ModelApiType { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型倍率
|
||||||
|
/// </summary>
|
||||||
|
[Range(0.01, double.MaxValue, ErrorMessage = "模型倍率必须大于0")]
|
||||||
|
public decimal Multiplier { get; set; } = 1;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型显示倍率
|
||||||
|
/// </summary>
|
||||||
|
[Range(0.01, double.MaxValue, ErrorMessage = "模型显示倍率必须大于0")]
|
||||||
|
public decimal MultiplierShow { get; set; } = 1;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 供应商分组名称
|
||||||
|
/// </summary>
|
||||||
|
[StringLength(100, ErrorMessage = "供应商分组名称不能超过100个字符")]
|
||||||
|
public string? ProviderName { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型图标URL
|
||||||
|
/// </summary>
|
||||||
|
[StringLength(500, ErrorMessage = "模型图标URL不能超过500个字符")]
|
||||||
|
public string? IconUrl { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 是否为尊享模型
|
||||||
|
/// </summary>
|
||||||
|
public bool IsPremium { get; set; }
|
||||||
|
}
|
||||||
@@ -0,0 +1,84 @@
|
|||||||
|
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.Channel;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// AI模型DTO
|
||||||
|
/// </summary>
|
||||||
|
public class AiModelDto
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 模型ID
|
||||||
|
/// </summary>
|
||||||
|
public Guid Id { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 处理名
|
||||||
|
/// </summary>
|
||||||
|
public string HandlerName { get; set; }
|
||||||
|
|
||||||
|
/// <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 int OrderNum { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// AI应用ID
|
||||||
|
/// </summary>
|
||||||
|
public Guid AiAppId { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 额外信息
|
||||||
|
/// </summary>
|
||||||
|
public string? ExtraInfo { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型类型
|
||||||
|
/// </summary>
|
||||||
|
public ModelTypeEnum ModelType { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型API类型
|
||||||
|
/// </summary>
|
||||||
|
public ModelApiTypeEnum ModelApiType { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型倍率
|
||||||
|
/// </summary>
|
||||||
|
public decimal Multiplier { 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>
|
||||||
|
/// 是否为尊享模型
|
||||||
|
/// </summary>
|
||||||
|
public bool IsPremium { get; set; }
|
||||||
|
}
|
||||||
@@ -0,0 +1,24 @@
|
|||||||
|
using Yi.Framework.Ddd.Application.Contracts;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.Channel;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取AI模型列表输入
|
||||||
|
/// </summary>
|
||||||
|
public class AiModelGetListInput : PagedAllResultRequestDto
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 搜索关键词(搜索模型名称、模型ID)
|
||||||
|
/// </summary>
|
||||||
|
public string? SearchKey { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// AI应用ID筛选
|
||||||
|
/// </summary>
|
||||||
|
public Guid? AiAppId { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 是否只显示尊享模型
|
||||||
|
/// </summary>
|
||||||
|
public bool? IsPremiumOnly { get; set; }
|
||||||
|
}
|
||||||
@@ -0,0 +1,102 @@
|
|||||||
|
using System.ComponentModel.DataAnnotations;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.Channel;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 更新AI模型输入
|
||||||
|
/// </summary>
|
||||||
|
public class AiModelUpdateInput
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 模型ID
|
||||||
|
/// </summary>
|
||||||
|
[Required(ErrorMessage = "模型ID不能为空")]
|
||||||
|
public Guid Id { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 处理名
|
||||||
|
/// </summary>
|
||||||
|
[Required(ErrorMessage = "处理名不能为空")]
|
||||||
|
[StringLength(100, ErrorMessage = "处理名不能超过100个字符")]
|
||||||
|
public string HandlerName { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型ID
|
||||||
|
/// </summary>
|
||||||
|
[Required(ErrorMessage = "模型ID不能为空")]
|
||||||
|
[StringLength(200, ErrorMessage = "模型ID不能超过200个字符")]
|
||||||
|
public string ModelId { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型名称
|
||||||
|
/// </summary>
|
||||||
|
[Required(ErrorMessage = "模型名称不能为空")]
|
||||||
|
[StringLength(200, ErrorMessage = "模型名称不能超过200个字符")]
|
||||||
|
public string Name { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型描述
|
||||||
|
/// </summary>
|
||||||
|
[StringLength(1000, ErrorMessage = "模型描述不能超过1000个字符")]
|
||||||
|
public string? Description { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 排序
|
||||||
|
/// </summary>
|
||||||
|
[Range(0, int.MaxValue, ErrorMessage = "排序必须大于等于0")]
|
||||||
|
public int OrderNum { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// AI应用ID
|
||||||
|
/// </summary>
|
||||||
|
[Required(ErrorMessage = "AI应用ID不能为空")]
|
||||||
|
public Guid AiAppId { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 额外信息
|
||||||
|
/// </summary>
|
||||||
|
[StringLength(2000, ErrorMessage = "额外信息不能超过2000个字符")]
|
||||||
|
public string? ExtraInfo { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型类型
|
||||||
|
/// </summary>
|
||||||
|
[Required(ErrorMessage = "模型类型不能为空")]
|
||||||
|
public ModelTypeEnum ModelType { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型API类型
|
||||||
|
/// </summary>
|
||||||
|
[Required(ErrorMessage = "模型API类型不能为空")]
|
||||||
|
public ModelApiTypeEnum ModelApiType { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型倍率
|
||||||
|
/// </summary>
|
||||||
|
[Range(0.01, double.MaxValue, ErrorMessage = "模型倍率必须大于0")]
|
||||||
|
public decimal Multiplier { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型显示倍率
|
||||||
|
/// </summary>
|
||||||
|
[Range(0.01, double.MaxValue, ErrorMessage = "模型显示倍率必须大于0")]
|
||||||
|
public decimal MultiplierShow { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 供应商分组名称
|
||||||
|
/// </summary>
|
||||||
|
[StringLength(100, ErrorMessage = "供应商分组名称不能超过100个字符")]
|
||||||
|
public string? ProviderName { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型图标URL
|
||||||
|
/// </summary>
|
||||||
|
[StringLength(500, ErrorMessage = "模型图标URL不能超过500个字符")]
|
||||||
|
public string? IconUrl { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 是否为尊享模型
|
||||||
|
/// </summary>
|
||||||
|
public bool IsPremium { get; set; }
|
||||||
|
}
|
||||||
@@ -0,0 +1,65 @@
|
|||||||
|
using System.Reflection;
|
||||||
|
using System.Text.Json.Serialization;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.Chat;
|
||||||
|
|
||||||
|
public class AgentResultOutput
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 类型
|
||||||
|
/// </summary>
|
||||||
|
[JsonIgnore]
|
||||||
|
public AgentResultTypeEnum TypeEnum { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 类型
|
||||||
|
/// </summary>
|
||||||
|
public string Type => TypeEnum.GetJsonName();
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 内容载体
|
||||||
|
/// </summary>
|
||||||
|
public object Content { get; set; }
|
||||||
|
}
|
||||||
|
|
||||||
|
public enum AgentResultTypeEnum
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 文本内容
|
||||||
|
/// </summary>
|
||||||
|
[JsonPropertyName("text")]
|
||||||
|
Text,
|
||||||
|
/// <summary>
|
||||||
|
/// 工具调用中
|
||||||
|
/// </summary>
|
||||||
|
[JsonPropertyName("toolCalling")]
|
||||||
|
ToolCalling,
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 工具调用完成
|
||||||
|
/// </summary>
|
||||||
|
[JsonPropertyName("toolCalled")]
|
||||||
|
ToolCalled,
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 用量
|
||||||
|
/// </summary>
|
||||||
|
[JsonPropertyName("usage")]
|
||||||
|
Usage,
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 工具调用用量
|
||||||
|
/// </summary>
|
||||||
|
[JsonPropertyName("toolCallUsage")]
|
||||||
|
ToolCallUsage
|
||||||
|
}
|
||||||
|
|
||||||
|
public static class AgentResultTypeEnumExtensions
|
||||||
|
{
|
||||||
|
public static string GetJsonName(this AgentResultTypeEnum value)
|
||||||
|
{
|
||||||
|
var member = typeof(AgentResultTypeEnum).GetMember(value.ToString()).FirstOrDefault();
|
||||||
|
var attr = member?.GetCustomAttribute<JsonPropertyNameAttribute>();
|
||||||
|
return attr?.Name ?? value.ToString();
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,29 @@
|
|||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.Chat;
|
||||||
|
|
||||||
|
public class AgentSendInput
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 会话id
|
||||||
|
/// </summary>
|
||||||
|
public Guid SessionId { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 用户内容
|
||||||
|
/// </summary>
|
||||||
|
public string Content { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// api密钥Id
|
||||||
|
/// </summary>
|
||||||
|
public string Token { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型id
|
||||||
|
/// </summary>
|
||||||
|
public string ModelId { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 已选择工具
|
||||||
|
/// </summary>
|
||||||
|
public List<string> Tools { get; set; }
|
||||||
|
}
|
||||||
@@ -0,0 +1,8 @@
|
|||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.Chat;
|
||||||
|
|
||||||
|
public class AgentToolOutput
|
||||||
|
{
|
||||||
|
public string Code { get; set; }
|
||||||
|
|
||||||
|
public string Name { get; set; }
|
||||||
|
}
|
||||||
@@ -0,0 +1,22 @@
|
|||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.Chat;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 图片生成输入
|
||||||
|
/// </summary>
|
||||||
|
public class ImageGenerationInput
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 提示词
|
||||||
|
/// </summary>
|
||||||
|
public string Prompt { get; set; } = string.Empty;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型ID
|
||||||
|
/// </summary>
|
||||||
|
public string ModelId { get; set; } = string.Empty;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 参考图Base64列表(可选,包含前缀如 data:image/png;base64,...)
|
||||||
|
/// </summary>
|
||||||
|
public List<string>? ReferenceImagesBase64 { get; set; }
|
||||||
|
}
|
||||||
@@ -0,0 +1,49 @@
|
|||||||
|
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.Chat;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 图片任务输出
|
||||||
|
/// </summary>
|
||||||
|
public class ImageTaskOutput
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 任务ID
|
||||||
|
/// </summary>
|
||||||
|
public Guid Id { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 提示词
|
||||||
|
/// </summary>
|
||||||
|
public string Prompt { get; set; } = string.Empty;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 参考图Base64列表
|
||||||
|
/// </summary>
|
||||||
|
public List<string>? ReferenceImagesBase64 { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 参考图URL列表
|
||||||
|
/// </summary>
|
||||||
|
public List<string>? ReferenceImagesUrl { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 生成图片Base64(包含前缀)
|
||||||
|
/// </summary>
|
||||||
|
public string? StoreBase64 { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 生成图片URL
|
||||||
|
/// </summary>
|
||||||
|
public string? StoreUrl { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 任务状态
|
||||||
|
/// </summary>
|
||||||
|
public TaskStatusEnum TaskStatus { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 创建时间
|
||||||
|
/// </summary>
|
||||||
|
public DateTime CreationTime { get; set; }
|
||||||
|
}
|
||||||
@@ -0,0 +1,24 @@
|
|||||||
|
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.Chat;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 图片任务分页查询输入
|
||||||
|
/// </summary>
|
||||||
|
public class ImageTaskPageInput
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 页码(从1开始)
|
||||||
|
/// </summary>
|
||||||
|
public int PageIndex { get; set; } = 1;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 每页数量
|
||||||
|
/// </summary>
|
||||||
|
public int PageSize { get; set; } = 10;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 任务状态筛选(可选)
|
||||||
|
/// </summary>
|
||||||
|
public TaskStatusEnum? TaskStatus { get; set; }
|
||||||
|
}
|
||||||
@@ -1,4 +1,5 @@
|
|||||||
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Extensions;
|
||||||
|
|
||||||
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.Model;
|
namespace Yi.Framework.AiHub.Application.Contracts.Dtos.Model;
|
||||||
|
|
||||||
@@ -30,17 +31,13 @@ public class ModelLibraryDto
|
|||||||
/// <summary>
|
/// <summary>
|
||||||
/// 模型类型名称
|
/// 模型类型名称
|
||||||
/// </summary>
|
/// </summary>
|
||||||
public string ModelTypeName { get; set; }
|
public string ModelTypeName => ModelType.GetDescription();
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 模型API类型
|
/// 模型支持的API类型
|
||||||
/// </summary>
|
/// </summary>
|
||||||
public ModelApiTypeEnum ModelApiType { get; set; }
|
public List<ModelApiTypeOutput> ModelApiTypes { get; set; }
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// 模型API类型名称
|
|
||||||
/// </summary>
|
|
||||||
public string ModelApiTypeName { get; set; }
|
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 模型显示倍率
|
/// 模型显示倍率
|
||||||
@@ -61,4 +58,22 @@ public class ModelLibraryDto
|
|||||||
/// 是否为尊享模型(PremiumChat类型)
|
/// 是否为尊享模型(PremiumChat类型)
|
||||||
/// </summary>
|
/// </summary>
|
||||||
public bool IsPremium { get; set; }
|
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();
|
||||||
}
|
}
|
||||||
@@ -25,11 +25,17 @@ public class RechargeCreateInput
|
|||||||
public string Content { get; set; } = string.Empty;
|
public string Content { get; set; } = string.Empty;
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// VIP月数(为空或0表示永久VIP)
|
/// VIP月数(为空或0表示永久VIP,1个月按31天计算)
|
||||||
/// </summary>
|
/// </summary>
|
||||||
[Range(0, int.MaxValue, ErrorMessage = "月数必须大于等于0")]
|
[Range(0, int.MaxValue, ErrorMessage = "月数必须大于等于0")]
|
||||||
public int? Months { get; set; }
|
public int? Months { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// VIP天数(为空或0表示不使用天数充值)
|
||||||
|
/// </summary>
|
||||||
|
[Range(0, int.MaxValue, ErrorMessage = "天数必须大于等于0")]
|
||||||
|
public int? Days { get; set; }
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 备注
|
/// 备注
|
||||||
/// </summary>
|
/// </summary>
|
||||||
|
|||||||
@@ -0,0 +1,20 @@
|
|||||||
|
using Volo.Abp.Application.Services;
|
||||||
|
using Yi.Framework.AiHub.Application.Contracts.Dtos.ActivationCode;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.IServices;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 激活码服务接口
|
||||||
|
/// </summary>
|
||||||
|
public interface IActivationCodeService : IApplicationService
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 批量生成激活码
|
||||||
|
/// </summary>
|
||||||
|
Task<ActivationCodeCreateListOutput> CreateBatchAsync(ActivationCodeCreateListInput input);
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 兑换激活码
|
||||||
|
/// </summary>
|
||||||
|
Task<ActivationCodeRedeemOutput> RedeemAsync(ActivationCodeRedeemInput input);
|
||||||
|
}
|
||||||
@@ -0,0 +1,86 @@
|
|||||||
|
using Volo.Abp.Application.Dtos;
|
||||||
|
using Yi.Framework.AiHub.Application.Contracts.Dtos.Channel;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Contracts.IServices;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 渠道商管理服务接口
|
||||||
|
/// </summary>
|
||||||
|
public interface IChannelService
|
||||||
|
{
|
||||||
|
#region AI应用管理
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取AI应用列表
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="input">查询参数</param>
|
||||||
|
/// <returns>分页应用列表</returns>
|
||||||
|
Task<PagedResultDto<AiAppDto>> GetAppListAsync(AiAppGetListInput input);
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 根据ID获取AI应用
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="id">应用ID</param>
|
||||||
|
/// <returns>应用详情</returns>
|
||||||
|
Task<AiAppDto> GetAppByIdAsync(Guid id);
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 创建AI应用
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="input">创建输入</param>
|
||||||
|
/// <returns>创建的应用</returns>
|
||||||
|
Task<AiAppDto> CreateAppAsync(AiAppCreateInput input);
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 更新AI应用
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="input">更新输入</param>
|
||||||
|
/// <returns>更新后的应用</returns>
|
||||||
|
Task<AiAppDto> UpdateAppAsync(AiAppUpdateInput input);
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 删除AI应用
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="id">应用ID</param>
|
||||||
|
Task DeleteAppAsync(Guid id);
|
||||||
|
|
||||||
|
#endregion
|
||||||
|
|
||||||
|
#region AI模型管理
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取AI模型列表
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="input">查询参数</param>
|
||||||
|
/// <returns>分页模型列表</returns>
|
||||||
|
Task<PagedResultDto<AiModelDto>> GetModelListAsync(AiModelGetListInput input);
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 根据ID获取AI模型
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="id">模型ID</param>
|
||||||
|
/// <returns>模型详情</returns>
|
||||||
|
Task<AiModelDto> GetModelByIdAsync(Guid id);
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 创建AI模型
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="input">创建输入</param>
|
||||||
|
/// <returns>创建的模型</returns>
|
||||||
|
Task<AiModelDto> CreateModelAsync(AiModelCreateInput input);
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 更新AI模型
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="input">更新输入</param>
|
||||||
|
/// <returns>更新后的模型</returns>
|
||||||
|
Task<AiModelDto> UpdateModelAsync(AiModelUpdateInput input);
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 删除AI模型(软删除)
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="id">模型ID</param>
|
||||||
|
Task DeleteModelAsync(Guid id);
|
||||||
|
|
||||||
|
#endregion
|
||||||
|
}
|
||||||
@@ -0,0 +1,61 @@
|
|||||||
|
using System.Text.Json;
|
||||||
|
using Microsoft.Extensions.Logging;
|
||||||
|
using Volo.Abp.BackgroundJobs;
|
||||||
|
using Volo.Abp.DependencyInjection;
|
||||||
|
using Yi.Framework.AiHub.Domain.Entities.Chat;
|
||||||
|
using Yi.Framework.AiHub.Domain.Managers;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
||||||
|
using Yi.Framework.SqlSugarCore.Abstractions;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Jobs;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 图片生成后台任务
|
||||||
|
/// </summary>
|
||||||
|
public class ImageGenerationJob : AsyncBackgroundJob<ImageGenerationJobArgs>, ITransientDependency
|
||||||
|
{
|
||||||
|
private readonly ILogger<ImageGenerationJob> _logger;
|
||||||
|
private readonly AiGateWayManager _aiGateWayManager;
|
||||||
|
private readonly ISqlSugarRepository<ImageStoreTaskAggregateRoot> _imageStoreTaskRepository;
|
||||||
|
|
||||||
|
public ImageGenerationJob(
|
||||||
|
ILogger<ImageGenerationJob> logger,
|
||||||
|
AiGateWayManager aiGateWayManager,
|
||||||
|
ISqlSugarRepository<ImageStoreTaskAggregateRoot> imageStoreTaskRepository)
|
||||||
|
{
|
||||||
|
_logger = logger;
|
||||||
|
_aiGateWayManager = aiGateWayManager;
|
||||||
|
_imageStoreTaskRepository = imageStoreTaskRepository;
|
||||||
|
}
|
||||||
|
|
||||||
|
public override async Task ExecuteAsync(ImageGenerationJobArgs args)
|
||||||
|
{
|
||||||
|
_logger.LogInformation("开始执行图片生成任务,TaskId: {TaskId}, ModelId: {ModelId}, UserId: {UserId}",
|
||||||
|
args.TaskId, args.ModelId, args.UserId);
|
||||||
|
|
||||||
|
try
|
||||||
|
{
|
||||||
|
var request = JsonSerializer.Deserialize<JsonElement>(args.RequestJson);
|
||||||
|
|
||||||
|
await _aiGateWayManager.GeminiGenerateContentImageForStatisticsAsync(
|
||||||
|
args.TaskId,
|
||||||
|
args.ModelId,
|
||||||
|
request,
|
||||||
|
args.UserId);
|
||||||
|
|
||||||
|
_logger.LogInformation("图片生成任务完成,TaskId: {TaskId}", args.TaskId);
|
||||||
|
}
|
||||||
|
catch (Exception ex)
|
||||||
|
{
|
||||||
|
_logger.LogError(ex, "图片生成任务失败,TaskId: {TaskId}, Error: {Error}", args.TaskId, ex.Message);
|
||||||
|
|
||||||
|
// 更新任务状态为失败
|
||||||
|
var task = await _imageStoreTaskRepository.GetFirstAsync(x => x.Id == args.TaskId);
|
||||||
|
if (task != null)
|
||||||
|
{
|
||||||
|
task.TaskStatus = TaskStatusEnum.Fail;
|
||||||
|
await _imageStoreTaskRepository.UpdateAsync(task);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,27 @@
|
|||||||
|
namespace Yi.Framework.AiHub.Application.Jobs;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 图片生成后台任务参数
|
||||||
|
/// </summary>
|
||||||
|
public class ImageGenerationJobArgs
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 图片任务ID
|
||||||
|
/// </summary>
|
||||||
|
public Guid TaskId { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型ID
|
||||||
|
/// </summary>
|
||||||
|
public string ModelId { get; set; } = string.Empty;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 请求JSON字符串
|
||||||
|
/// </summary>
|
||||||
|
public string RequestJson { get; set; } = string.Empty;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 用户ID
|
||||||
|
/// </summary>
|
||||||
|
public Guid UserId { get; set; }
|
||||||
|
}
|
||||||
@@ -0,0 +1,116 @@
|
|||||||
|
using Medallion.Threading;
|
||||||
|
using Microsoft.AspNetCore.Authorization;
|
||||||
|
using Microsoft.AspNetCore.Mvc;
|
||||||
|
using Volo.Abp;
|
||||||
|
using Volo.Abp.Application.Services;
|
||||||
|
using Volo.Abp.Users;
|
||||||
|
using Yi.Framework.AiHub.Application.Contracts.Dtos.ActivationCode;
|
||||||
|
using Yi.Framework.AiHub.Application.Contracts.Dtos.Recharge;
|
||||||
|
using Yi.Framework.AiHub.Application.Contracts.IServices;
|
||||||
|
using Yi.Framework.AiHub.Domain.Managers;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Services;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 激活码服务
|
||||||
|
/// </summary>
|
||||||
|
public class ActivationCodeService : ApplicationService, IActivationCodeService
|
||||||
|
{
|
||||||
|
private readonly ActivationCodeManager _activationCodeManager;
|
||||||
|
private readonly IRechargeService _rechargeService;
|
||||||
|
private readonly PremiumPackageManager _premiumPackageManager;
|
||||||
|
private IDistributedLockProvider DistributedLock => LazyServiceProvider.LazyGetRequiredService<IDistributedLockProvider>();
|
||||||
|
public ActivationCodeService(
|
||||||
|
ActivationCodeManager activationCodeManager,
|
||||||
|
IRechargeService rechargeService,
|
||||||
|
PremiumPackageManager premiumPackageManager)
|
||||||
|
{
|
||||||
|
_activationCodeManager = activationCodeManager;
|
||||||
|
_rechargeService = rechargeService;
|
||||||
|
_premiumPackageManager = premiumPackageManager;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 批量生成激活码
|
||||||
|
/// </summary>
|
||||||
|
[Authorize]
|
||||||
|
[HttpPost("activationCode/Batch")]
|
||||||
|
public async Task<ActivationCodeCreateListOutput> CreateBatchAsync(ActivationCodeCreateListInput input)
|
||||||
|
{
|
||||||
|
if (input.Items == null || input.Items.Count == 0)
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("生成列表不能为空");
|
||||||
|
}
|
||||||
|
|
||||||
|
var entities = await _activationCodeManager.CreateBatchAsync(
|
||||||
|
input.Items.Select(x => (x.GoodsType, x.Count)).ToList());
|
||||||
|
|
||||||
|
var outputs = entities
|
||||||
|
.GroupBy(x => x.GoodsType)
|
||||||
|
.Select(group => new ActivationCodeCreateOutput
|
||||||
|
{
|
||||||
|
GoodsType = group.Key,
|
||||||
|
Count = group.Count(),
|
||||||
|
Codes = group.Select(x => x.Code).ToList()
|
||||||
|
})
|
||||||
|
.ToList();
|
||||||
|
|
||||||
|
return new ActivationCodeCreateListOutput
|
||||||
|
{
|
||||||
|
Items = outputs
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 兑换激活码
|
||||||
|
/// </summary>
|
||||||
|
[Authorize]
|
||||||
|
[HttpPost("activationCode/Redeem")]
|
||||||
|
public async Task<ActivationCodeRedeemOutput> RedeemAsync(ActivationCodeRedeemInput input)
|
||||||
|
{
|
||||||
|
//自旋等待,防抖
|
||||||
|
await using var handle =
|
||||||
|
await DistributedLock.AcquireLockAsync($"Yi:AiHub:ActivationCodeLock:{input.Code}");
|
||||||
|
|
||||||
|
var userId = CurrentUser.GetId();
|
||||||
|
var redeemContext = await _activationCodeManager.RedeemAsync(userId, input.Code);
|
||||||
|
var goodsType = redeemContext.ActivationCode.GoodsType;
|
||||||
|
var goods = redeemContext.Goods;
|
||||||
|
var packageName = redeemContext.PackageName;
|
||||||
|
var totalAmount = goods.Price;
|
||||||
|
|
||||||
|
if (goods.TokenAmount > 0)
|
||||||
|
{
|
||||||
|
await _premiumPackageManager.CreatePremiumPackageAsync(
|
||||||
|
userId,
|
||||||
|
goods.TokenAmount,
|
||||||
|
packageName,
|
||||||
|
totalAmount,
|
||||||
|
"激活码兑换",
|
||||||
|
expireMonths: null,
|
||||||
|
isCreateRechargeRecord: !goods.IsCombo);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (goods.VipMonths > 0 || goods.VipDays > 0)
|
||||||
|
{
|
||||||
|
await _rechargeService.RechargeVipAsync(new RechargeCreateInput
|
||||||
|
{
|
||||||
|
UserId = userId,
|
||||||
|
RechargeAmount = totalAmount,
|
||||||
|
Content = packageName,
|
||||||
|
Months = goods.VipMonths,
|
||||||
|
Days = goods.VipDays,
|
||||||
|
Remark = "激活码兑换",
|
||||||
|
ContactInfo = null
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
return new ActivationCodeRedeemOutput
|
||||||
|
{
|
||||||
|
GoodsType = goodsType,
|
||||||
|
PackageName = packageName,
|
||||||
|
Content = packageName
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,3 +1,4 @@
|
|||||||
|
using Medallion.Threading;
|
||||||
using Microsoft.AspNetCore.Authorization;
|
using Microsoft.AspNetCore.Authorization;
|
||||||
using Microsoft.Extensions.Logging;
|
using Microsoft.Extensions.Logging;
|
||||||
using SqlSugar;
|
using SqlSugar;
|
||||||
@@ -6,6 +7,7 @@ using Volo.Abp.Users;
|
|||||||
using Yi.Framework.AiHub.Application.Contracts.Dtos.DailyTask;
|
using Yi.Framework.AiHub.Application.Contracts.Dtos.DailyTask;
|
||||||
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.Model;
|
||||||
using Yi.Framework.AiHub.Domain.Extensions;
|
using Yi.Framework.AiHub.Domain.Extensions;
|
||||||
using Yi.Framework.AiHub.Domain.Managers;
|
using Yi.Framework.AiHub.Domain.Managers;
|
||||||
using Yi.Framework.AiHub.Domain.Shared.Consts;
|
using Yi.Framework.AiHub.Domain.Shared.Consts;
|
||||||
@@ -23,7 +25,8 @@ public class DailyTaskService : ApplicationService
|
|||||||
private readonly ISqlSugarRepository<MessageAggregateRoot> _messageRepository;
|
private readonly ISqlSugarRepository<MessageAggregateRoot> _messageRepository;
|
||||||
private readonly ISqlSugarRepository<PremiumPackageAggregateRoot> _premiumPackageRepository;
|
private readonly ISqlSugarRepository<PremiumPackageAggregateRoot> _premiumPackageRepository;
|
||||||
private readonly ILogger<DailyTaskService> _logger;
|
private readonly ILogger<DailyTaskService> _logger;
|
||||||
|
private IDistributedLockProvider DistributedLock => LazyServiceProvider.LazyGetRequiredService<IDistributedLockProvider>();
|
||||||
|
private readonly ISqlSugarRepository<AiModelEntity> _aiModelRepository;
|
||||||
// 任务配置
|
// 任务配置
|
||||||
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()
|
||||||
@@ -36,12 +39,13 @@ public class DailyTaskService : ApplicationService
|
|||||||
ISqlSugarRepository<DailyTaskRewardRecordAggregateRoot> dailyTaskRepository,
|
ISqlSugarRepository<DailyTaskRewardRecordAggregateRoot> dailyTaskRepository,
|
||||||
ISqlSugarRepository<MessageAggregateRoot> messageRepository,
|
ISqlSugarRepository<MessageAggregateRoot> messageRepository,
|
||||||
ISqlSugarRepository<PremiumPackageAggregateRoot> premiumPackageRepository,
|
ISqlSugarRepository<PremiumPackageAggregateRoot> premiumPackageRepository,
|
||||||
ILogger<DailyTaskService> logger)
|
ILogger<DailyTaskService> logger, ISqlSugarRepository<AiModelEntity> aiModelRepository)
|
||||||
{
|
{
|
||||||
_dailyTaskRepository = dailyTaskRepository;
|
_dailyTaskRepository = dailyTaskRepository;
|
||||||
_messageRepository = messageRepository;
|
_messageRepository = messageRepository;
|
||||||
_premiumPackageRepository = premiumPackageRepository;
|
_premiumPackageRepository = premiumPackageRepository;
|
||||||
_logger = logger;
|
_logger = logger;
|
||||||
|
_aiModelRepository = aiModelRepository;
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
@@ -113,6 +117,11 @@ public class DailyTaskService : ApplicationService
|
|||||||
public async Task ClaimTaskRewardAsync(ClaimTaskRewardInput input)
|
public async Task ClaimTaskRewardAsync(ClaimTaskRewardInput input)
|
||||||
{
|
{
|
||||||
var userId = CurrentUser.GetId();
|
var userId = CurrentUser.GetId();
|
||||||
|
//自旋等待,防抖
|
||||||
|
await using var handle =
|
||||||
|
await DistributedLock.AcquireLockAsync($"Yi:AiHub:ClaimTaskRewardLock:{userId}");
|
||||||
|
|
||||||
|
|
||||||
var today = DateTime.Today;
|
var today = DateTime.Today;
|
||||||
|
|
||||||
// 1. 验证任务等级
|
// 1. 验证任务等级
|
||||||
@@ -173,10 +182,16 @@ public class DailyTaskService : ApplicationService
|
|||||||
var tomorrow = today.AddDays(1);
|
var tomorrow = today.AddDays(1);
|
||||||
|
|
||||||
// 查询今日所有使用尊享包模型的消息(role=system 表示消耗)
|
// 查询今日所有使用尊享包模型的消息(role=system 表示消耗)
|
||||||
|
// 先获取所有尊享模型的ModelId列表
|
||||||
|
var premiumModelIds = await _aiModelRepository._DbQueryable
|
||||||
|
.Where(x => x.IsPremium)
|
||||||
|
.Select(x => x.ModelId)
|
||||||
|
.ToListAsync();
|
||||||
|
|
||||||
var totalTokens = await _messageRepository._DbQueryable
|
var totalTokens = await _messageRepository._DbQueryable
|
||||||
.Where(x => x.UserId == userId)
|
.Where(x => x.UserId == userId)
|
||||||
.Where(x => x.Role == "system") // system角色表示实际消耗
|
.Where(x => x.Role == "system") // system角色表示实际消耗
|
||||||
.Where(x => PremiumPackageConst.ModeIds.Contains(x.ModelId)) // 尊享包模型
|
.Where(x => premiumModelIds.Contains(x.ModelId)) // 尊享包模型
|
||||||
.Where(x => x.CreationTime >= today && x.CreationTime < tomorrow)
|
.Where(x => x.CreationTime >= today && x.CreationTime < tomorrow)
|
||||||
.SumAsync(x => x.TokenUsage.TotalTokenCount);
|
.SumAsync(x => x.TokenUsage.TotalTokenCount);
|
||||||
|
|
||||||
|
|||||||
@@ -1,10 +1,14 @@
|
|||||||
using Mapster;
|
using Mapster;
|
||||||
using Microsoft.AspNetCore.Authorization;
|
using Microsoft.AspNetCore.Authorization;
|
||||||
using Microsoft.AspNetCore.Mvc;
|
using Microsoft.AspNetCore.Mvc;
|
||||||
|
using SqlSugar;
|
||||||
|
using System.Globalization;
|
||||||
|
using System.Text;
|
||||||
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;
|
using Yi.Framework.AiHub.Application.Contracts.Dtos;
|
||||||
using Yi.Framework.AiHub.Domain.Entities;
|
using Yi.Framework.AiHub.Domain.Entities;
|
||||||
|
using Yi.Framework.AiHub.Domain.Entities.Chat;
|
||||||
using Yi.Framework.Rbac.Application.Contracts.IServices;
|
using Yi.Framework.Rbac.Application.Contracts.IServices;
|
||||||
using Yi.Framework.Rbac.Domain.Shared.Dtos;
|
using Yi.Framework.Rbac.Domain.Shared.Dtos;
|
||||||
using Yi.Framework.SqlSugarCore.Abstractions;
|
using Yi.Framework.SqlSugarCore.Abstractions;
|
||||||
@@ -18,17 +22,18 @@ public class AiAccountService : ApplicationService
|
|||||||
private ISqlSugarRepository<AiUserExtraInfoEntity> _userRepository;
|
private ISqlSugarRepository<AiUserExtraInfoEntity> _userRepository;
|
||||||
private ISqlSugarRepository<AiRechargeAggregateRoot> _rechargeRepository;
|
private ISqlSugarRepository<AiRechargeAggregateRoot> _rechargeRepository;
|
||||||
private ISqlSugarRepository<PremiumPackageAggregateRoot> _premiumPackageRepository;
|
private ISqlSugarRepository<PremiumPackageAggregateRoot> _premiumPackageRepository;
|
||||||
|
private ISqlSugarRepository<MessageAggregateRoot> _messageRepository;
|
||||||
public AiAccountService(
|
public AiAccountService(
|
||||||
IAccountService accountService,
|
IAccountService accountService,
|
||||||
ISqlSugarRepository<AiUserExtraInfoEntity> userRepository,
|
ISqlSugarRepository<AiUserExtraInfoEntity> userRepository,
|
||||||
ISqlSugarRepository<AiRechargeAggregateRoot> rechargeRepository,
|
ISqlSugarRepository<AiRechargeAggregateRoot> rechargeRepository,
|
||||||
ISqlSugarRepository<PremiumPackageAggregateRoot> premiumPackageRepository)
|
ISqlSugarRepository<PremiumPackageAggregateRoot> premiumPackageRepository, ISqlSugarRepository<MessageAggregateRoot> messageRepository)
|
||||||
{
|
{
|
||||||
_accountService = accountService;
|
_accountService = accountService;
|
||||||
_userRepository = userRepository;
|
_userRepository = userRepository;
|
||||||
_rechargeRepository = rechargeRepository;
|
_rechargeRepository = rechargeRepository;
|
||||||
_premiumPackageRepository = premiumPackageRepository;
|
_premiumPackageRepository = premiumPackageRepository;
|
||||||
|
_messageRepository = messageRepository;
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
@@ -148,4 +153,90 @@ public class AiAccountService : ApplicationService
|
|||||||
|
|
||||||
return result;
|
return result;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public class TokenStatisticsInput
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 指定日期(当天零点)
|
||||||
|
/// </summary>
|
||||||
|
public DateTime Date { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型Id -> 1亿Token成本(RMB)
|
||||||
|
/// </summary>
|
||||||
|
public Dictionary<string, decimal> ModelCosts { get; set; } = new();
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取指定日期各模型Token统计
|
||||||
|
/// </summary>
|
||||||
|
[Authorize]
|
||||||
|
[HttpPost("account/token-statistics")]
|
||||||
|
public async Task<string> GetTokenStatisticsAsync([FromBody] TokenStatisticsInput input)
|
||||||
|
{
|
||||||
|
if (CurrentUser.UserName != "Guo" && CurrentUser.UserName != "cc")
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("您暂无权限访问");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (input.ModelCosts is null || input.ModelCosts.Count == 0)
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("请提供模型成本配置");
|
||||||
|
}
|
||||||
|
|
||||||
|
var day = input.Date.Date;
|
||||||
|
var nextDay = day.AddDays(1);
|
||||||
|
var modelIds = input.ModelCosts.Keys.ToList();
|
||||||
|
|
||||||
|
var modelStats = await _messageRepository._DbQueryable
|
||||||
|
.Where(x => modelIds.Contains(x.ModelId))
|
||||||
|
.Where(x => x.CreationTime >= day && x.CreationTime < nextDay)
|
||||||
|
.Where(x => x.Role == "system")
|
||||||
|
.GroupBy(x => x.ModelId)
|
||||||
|
.Select(x => new
|
||||||
|
{
|
||||||
|
ModelId = x.ModelId,
|
||||||
|
Tokens = SqlFunc.AggregateSum(x.TokenUsage.TotalTokenCount),
|
||||||
|
Count = SqlFunc.AggregateCount(x.Id)
|
||||||
|
})
|
||||||
|
.ToListAsync();
|
||||||
|
|
||||||
|
var modelStatDict = modelStats.ToDictionary(x => x.ModelId, x => x);
|
||||||
|
|
||||||
|
string weekDay = day.ToString("dddd", new CultureInfo("zh-CN")) switch
|
||||||
|
{
|
||||||
|
"星期一" => "周1",
|
||||||
|
"星期二" => "周2",
|
||||||
|
"星期三" => "周3",
|
||||||
|
"星期四" => "周4",
|
||||||
|
"星期五" => "周5",
|
||||||
|
"星期六" => "周6",
|
||||||
|
"星期日" => "周日",
|
||||||
|
_ => day.ToString("dddd", new CultureInfo("zh-CN"))
|
||||||
|
};
|
||||||
|
|
||||||
|
var sb = new StringBuilder();
|
||||||
|
sb.AppendLine($"{day:M月d日} {weekDay}");
|
||||||
|
|
||||||
|
foreach (var kvp in input.ModelCosts)
|
||||||
|
{
|
||||||
|
var modelId = kvp.Key;
|
||||||
|
var cost = kvp.Value;
|
||||||
|
|
||||||
|
modelStatDict.TryGetValue(modelId, out var stat);
|
||||||
|
long tokens = stat?.Tokens ?? 0;
|
||||||
|
long count = stat?.Count ?? 0;
|
||||||
|
|
||||||
|
decimal costPerHundredMillion = tokens > 0
|
||||||
|
? cost / (tokens / 100000000m)
|
||||||
|
: 0;
|
||||||
|
|
||||||
|
decimal tokensInWan = tokens / 10000m;
|
||||||
|
|
||||||
|
sb.AppendLine();
|
||||||
|
sb.AppendLine($"{modelId} 成本:【{cost:F2}RMB】 次数:【{count}次】 token:【{tokensInWan:F0}w】 1亿token成本:【{costPerHundredMillion:F2}RMB】");
|
||||||
|
}
|
||||||
|
|
||||||
|
return sb.ToString().TrimEnd();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
@@ -0,0 +1,240 @@
|
|||||||
|
using Mapster;
|
||||||
|
using Microsoft.AspNetCore.Authorization;
|
||||||
|
using Microsoft.AspNetCore.Mvc;
|
||||||
|
using SqlSugar;
|
||||||
|
using Volo.Abp.Application.Dtos;
|
||||||
|
using Volo.Abp.Application.Services;
|
||||||
|
using Yi.Framework.AiHub.Application.Contracts.Dtos.Channel;
|
||||||
|
using Yi.Framework.AiHub.Application.Contracts.IServices;
|
||||||
|
using Yi.Framework.AiHub.Domain.Entities.Model;
|
||||||
|
using Yi.Framework.SqlSugarCore.Abstractions;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Services;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 渠道商管理服务实现
|
||||||
|
/// </summary>
|
||||||
|
[Authorize]
|
||||||
|
public class ChannelService : ApplicationService, IChannelService
|
||||||
|
{
|
||||||
|
private readonly ISqlSugarRepository<AiAppAggregateRoot, Guid> _appRepository;
|
||||||
|
private readonly ISqlSugarRepository<AiModelEntity, Guid> _modelRepository;
|
||||||
|
|
||||||
|
public ChannelService(
|
||||||
|
ISqlSugarRepository<AiAppAggregateRoot, Guid> appRepository,
|
||||||
|
ISqlSugarRepository<AiModelEntity, Guid> modelRepository)
|
||||||
|
{
|
||||||
|
_appRepository = appRepository;
|
||||||
|
_modelRepository = modelRepository;
|
||||||
|
}
|
||||||
|
|
||||||
|
#region AI应用管理
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取AI应用列表
|
||||||
|
/// </summary>
|
||||||
|
[HttpGet("channel/app")]
|
||||||
|
public async Task<PagedResultDto<AiAppDto>> GetAppListAsync(AiAppGetListInput input)
|
||||||
|
{
|
||||||
|
RefAsync<int> total = 0;
|
||||||
|
|
||||||
|
var entities = await _appRepository._DbQueryable
|
||||||
|
.WhereIF(!string.IsNullOrWhiteSpace(input.SearchKey), x => x.Name.Contains(input.SearchKey))
|
||||||
|
.OrderByDescending(x => x.OrderNum)
|
||||||
|
.OrderByDescending(x => x.CreationTime)
|
||||||
|
.ToPageListAsync(input.SkipCount, input.MaxResultCount, total);
|
||||||
|
|
||||||
|
var output = entities.Adapt<List<AiAppDto>>();
|
||||||
|
return new PagedResultDto<AiAppDto>(total, output);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 根据ID获取AI应用
|
||||||
|
/// </summary>
|
||||||
|
[HttpGet("channel/app/{id}")]
|
||||||
|
public async Task<AiAppDto> GetAppByIdAsync([FromRoute]Guid id)
|
||||||
|
{
|
||||||
|
var entity = await _appRepository.GetByIdAsync(id);
|
||||||
|
return entity.Adapt<AiAppDto>();
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 创建AI应用
|
||||||
|
/// </summary>
|
||||||
|
public async Task<AiAppDto> CreateAppAsync(AiAppCreateInput input)
|
||||||
|
{
|
||||||
|
var entity = new AiAppAggregateRoot
|
||||||
|
{
|
||||||
|
Name = input.Name,
|
||||||
|
Endpoint = input.Endpoint,
|
||||||
|
ExtraUrl = input.ExtraUrl,
|
||||||
|
ApiKey = input.ApiKey,
|
||||||
|
OrderNum = input.OrderNum
|
||||||
|
};
|
||||||
|
|
||||||
|
await _appRepository.InsertAsync(entity);
|
||||||
|
return entity.Adapt<AiAppDto>();
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 更新AI应用
|
||||||
|
/// </summary>
|
||||||
|
public async Task<AiAppDto> UpdateAppAsync(AiAppUpdateInput input)
|
||||||
|
{
|
||||||
|
var entity = await _appRepository.GetByIdAsync(input.Id);
|
||||||
|
|
||||||
|
entity.Name = input.Name;
|
||||||
|
entity.Endpoint = input.Endpoint;
|
||||||
|
entity.ExtraUrl = input.ExtraUrl;
|
||||||
|
entity.ApiKey = input.ApiKey;
|
||||||
|
entity.OrderNum = input.OrderNum;
|
||||||
|
|
||||||
|
await _appRepository.UpdateAsync(entity);
|
||||||
|
return entity.Adapt<AiAppDto>();
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 删除AI应用
|
||||||
|
/// </summary>
|
||||||
|
[HttpDelete("channel/app/{id}")]
|
||||||
|
public async Task DeleteAppAsync([FromRoute]Guid id)
|
||||||
|
{
|
||||||
|
// 检查是否有关联的模型
|
||||||
|
var hasModels = await _modelRepository._DbQueryable
|
||||||
|
.Where(x => x.AiAppId == id && !x.IsDeleted)
|
||||||
|
.AnyAsync();
|
||||||
|
|
||||||
|
if (hasModels)
|
||||||
|
{
|
||||||
|
throw new Volo.Abp.UserFriendlyException("该应用下存在模型,无法删除");
|
||||||
|
}
|
||||||
|
|
||||||
|
await _appRepository.DeleteAsync(id);
|
||||||
|
}
|
||||||
|
|
||||||
|
#endregion
|
||||||
|
|
||||||
|
#region AI模型管理
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取AI模型列表
|
||||||
|
/// </summary>
|
||||||
|
[HttpGet("channel/model")]
|
||||||
|
public async Task<PagedResultDto<AiModelDto>> GetModelListAsync(AiModelGetListInput input)
|
||||||
|
{
|
||||||
|
RefAsync<int> total = 0;
|
||||||
|
|
||||||
|
var query = _modelRepository._DbQueryable
|
||||||
|
.Where(x => !x.IsDeleted)
|
||||||
|
.WhereIF(!string.IsNullOrWhiteSpace(input.SearchKey), x =>
|
||||||
|
x.Name.Contains(input.SearchKey) || x.ModelId.Contains(input.SearchKey))
|
||||||
|
.WhereIF(input.AiAppId.HasValue, x => x.AiAppId == input.AiAppId.Value)
|
||||||
|
.WhereIF(input.IsPremiumOnly == true, x => x.IsPremium);
|
||||||
|
|
||||||
|
var entities = await query
|
||||||
|
.OrderBy(x => x.OrderNum)
|
||||||
|
.OrderByDescending(x => x.Id)
|
||||||
|
.ToPageListAsync(input.SkipCount, input.MaxResultCount, total);
|
||||||
|
|
||||||
|
var output = entities.Adapt<List<AiModelDto>>();
|
||||||
|
return new PagedResultDto<AiModelDto>(total, output);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 根据ID获取AI模型
|
||||||
|
/// </summary>
|
||||||
|
[HttpGet("channel/model/{id}")]
|
||||||
|
public async Task<AiModelDto> GetModelByIdAsync([FromRoute]Guid id)
|
||||||
|
{
|
||||||
|
var entity = await _modelRepository.GetByIdAsync(id);
|
||||||
|
return entity.Adapt<AiModelDto>();
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 创建AI模型
|
||||||
|
/// </summary>
|
||||||
|
public async Task<AiModelDto> CreateModelAsync(AiModelCreateInput input)
|
||||||
|
{
|
||||||
|
// 验证应用是否存在
|
||||||
|
var appExists = await _appRepository._DbQueryable
|
||||||
|
.Where(x => x.Id == input.AiAppId)
|
||||||
|
.AnyAsync();
|
||||||
|
|
||||||
|
if (!appExists)
|
||||||
|
{
|
||||||
|
throw new Volo.Abp.UserFriendlyException("指定的AI应用不存在");
|
||||||
|
}
|
||||||
|
|
||||||
|
var entity = new AiModelEntity
|
||||||
|
{
|
||||||
|
HandlerName = input.HandlerName,
|
||||||
|
ModelId = input.ModelId,
|
||||||
|
Name = input.Name,
|
||||||
|
Description = input.Description,
|
||||||
|
OrderNum = input.OrderNum,
|
||||||
|
AiAppId = input.AiAppId,
|
||||||
|
ExtraInfo = input.ExtraInfo,
|
||||||
|
ModelType = input.ModelType,
|
||||||
|
ModelApiType = input.ModelApiType,
|
||||||
|
Multiplier = input.Multiplier,
|
||||||
|
MultiplierShow = input.MultiplierShow,
|
||||||
|
ProviderName = input.ProviderName,
|
||||||
|
IconUrl = input.IconUrl,
|
||||||
|
IsPremium = input.IsPremium,
|
||||||
|
IsDeleted = false
|
||||||
|
};
|
||||||
|
|
||||||
|
await _modelRepository.InsertAsync(entity);
|
||||||
|
return entity.Adapt<AiModelDto>();
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 更新AI模型
|
||||||
|
/// </summary>
|
||||||
|
public async Task<AiModelDto> UpdateModelAsync(AiModelUpdateInput input)
|
||||||
|
{
|
||||||
|
var entity = await _modelRepository.GetByIdAsync(input.Id);
|
||||||
|
|
||||||
|
// 验证应用是否存在
|
||||||
|
if (entity.AiAppId != input.AiAppId)
|
||||||
|
{
|
||||||
|
var appExists = await _appRepository._DbQueryable
|
||||||
|
.Where(x => x.Id == input.AiAppId)
|
||||||
|
.AnyAsync();
|
||||||
|
|
||||||
|
if (!appExists)
|
||||||
|
{
|
||||||
|
throw new Volo.Abp.UserFriendlyException("指定的AI应用不存在");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
entity.HandlerName = input.HandlerName;
|
||||||
|
entity.ModelId = input.ModelId;
|
||||||
|
entity.Name = input.Name;
|
||||||
|
entity.Description = input.Description;
|
||||||
|
entity.OrderNum = input.OrderNum;
|
||||||
|
entity.AiAppId = input.AiAppId;
|
||||||
|
entity.ExtraInfo = input.ExtraInfo;
|
||||||
|
entity.ModelType = input.ModelType;
|
||||||
|
entity.ModelApiType = input.ModelApiType;
|
||||||
|
entity.Multiplier = input.Multiplier;
|
||||||
|
entity.MultiplierShow = input.MultiplierShow;
|
||||||
|
entity.ProviderName = input.ProviderName;
|
||||||
|
entity.IconUrl = input.IconUrl;
|
||||||
|
entity.IsPremium = input.IsPremium;
|
||||||
|
|
||||||
|
await _modelRepository.UpdateAsync(entity);
|
||||||
|
return entity.Adapt<AiModelDto>();
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 删除AI模型(软删除)
|
||||||
|
/// </summary>
|
||||||
|
[HttpDelete("channel/model/{id}")]
|
||||||
|
public async Task DeleteModelAsync(Guid id)
|
||||||
|
{
|
||||||
|
await _modelRepository.DeleteByIdAsync(id);
|
||||||
|
}
|
||||||
|
|
||||||
|
#endregion
|
||||||
|
}
|
||||||
@@ -1,18 +1,26 @@
|
|||||||
using System.Collections.Concurrent;
|
using System.Collections.Concurrent;
|
||||||
|
using System.Reflection;
|
||||||
using System.Text;
|
using System.Text;
|
||||||
|
using System.Text.Encodings.Web;
|
||||||
|
using Dm.util;
|
||||||
using Microsoft.AspNetCore.Authorization;
|
using Microsoft.AspNetCore.Authorization;
|
||||||
using Microsoft.AspNetCore.Http;
|
using Microsoft.AspNetCore.Http;
|
||||||
using Microsoft.AspNetCore.Mvc;
|
using Microsoft.AspNetCore.Mvc;
|
||||||
using Microsoft.Extensions.DependencyInjection;
|
using Microsoft.Extensions.DependencyInjection;
|
||||||
using Microsoft.Extensions.Logging;
|
using Microsoft.Extensions.Logging;
|
||||||
using Microsoft.Extensions.Options;
|
using Microsoft.Extensions.Options;
|
||||||
|
using ModelContextProtocol;
|
||||||
|
using ModelContextProtocol.Server;
|
||||||
using Newtonsoft.Json;
|
using Newtonsoft.Json;
|
||||||
using Newtonsoft.Json.Serialization;
|
using Newtonsoft.Json.Serialization;
|
||||||
using OpenAI.Chat;
|
using OpenAI.Chat;
|
||||||
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;
|
using Yi.Framework.AiHub.Application.Contracts.Dtos;
|
||||||
|
using Yi.Framework.AiHub.Application.Contracts.Dtos.Chat;
|
||||||
|
using Yi.Framework.AiHub.Domain;
|
||||||
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.Model;
|
||||||
using Yi.Framework.AiHub.Domain.Extensions;
|
using Yi.Framework.AiHub.Domain.Extensions;
|
||||||
using Yi.Framework.AiHub.Domain.Managers;
|
using Yi.Framework.AiHub.Domain.Managers;
|
||||||
@@ -32,23 +40,37 @@ namespace Yi.Framework.AiHub.Application.Services;
|
|||||||
public class AiChatService : ApplicationService
|
public class AiChatService : ApplicationService
|
||||||
{
|
{
|
||||||
private readonly IHttpContextAccessor _httpContextAccessor;
|
private readonly IHttpContextAccessor _httpContextAccessor;
|
||||||
private readonly ISqlSugarRepository<AiModelEntity> _aiModelRepository;
|
|
||||||
private readonly AiBlacklistManager _aiBlacklistManager;
|
private readonly AiBlacklistManager _aiBlacklistManager;
|
||||||
private readonly ILogger<AiChatService> _logger;
|
private readonly ILogger<AiChatService> _logger;
|
||||||
private readonly AiGateWayManager _aiGateWayManager;
|
private readonly AiGateWayManager _aiGateWayManager;
|
||||||
|
private readonly ModelManager _modelManager;
|
||||||
private readonly PremiumPackageManager _premiumPackageManager;
|
private readonly PremiumPackageManager _premiumPackageManager;
|
||||||
|
private readonly ChatManager _chatManager;
|
||||||
|
private readonly TokenManager _tokenManager;
|
||||||
|
private readonly IAccountService _accountService;
|
||||||
|
private readonly ISqlSugarRepository<AgentStoreAggregateRoot> _agentStoreRepository;
|
||||||
|
private readonly ISqlSugarRepository<AiModelEntity> _aiModelRepository;
|
||||||
|
|
||||||
public AiChatService(IHttpContextAccessor httpContextAccessor,
|
public AiChatService(IHttpContextAccessor httpContextAccessor,
|
||||||
AiBlacklistManager aiBlacklistManager,
|
AiBlacklistManager aiBlacklistManager,
|
||||||
ISqlSugarRepository<AiModelEntity> aiModelRepository,
|
ILogger<AiChatService> logger,
|
||||||
ILogger<AiChatService> logger, AiGateWayManager aiGateWayManager, PremiumPackageManager premiumPackageManager)
|
AiGateWayManager aiGateWayManager,
|
||||||
|
ModelManager modelManager,
|
||||||
|
PremiumPackageManager premiumPackageManager,
|
||||||
|
ChatManager chatManager, TokenManager tokenManager, IAccountService accountService,
|
||||||
|
ISqlSugarRepository<AgentStoreAggregateRoot> agentStoreRepository, ISqlSugarRepository<AiModelEntity> aiModelRepository)
|
||||||
{
|
{
|
||||||
_httpContextAccessor = httpContextAccessor;
|
_httpContextAccessor = httpContextAccessor;
|
||||||
_aiBlacklistManager = aiBlacklistManager;
|
_aiBlacklistManager = aiBlacklistManager;
|
||||||
_aiModelRepository = aiModelRepository;
|
|
||||||
_logger = logger;
|
_logger = logger;
|
||||||
_aiGateWayManager = aiGateWayManager;
|
_aiGateWayManager = aiGateWayManager;
|
||||||
|
_modelManager = modelManager;
|
||||||
_premiumPackageManager = premiumPackageManager;
|
_premiumPackageManager = premiumPackageManager;
|
||||||
|
_chatManager = chatManager;
|
||||||
|
_tokenManager = tokenManager;
|
||||||
|
_accountService = accountService;
|
||||||
|
_agentStoreRepository = agentStoreRepository;
|
||||||
|
_aiModelRepository = aiModelRepository;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@@ -89,7 +111,7 @@ public class AiChatService : ApplicationService
|
|||||||
ApiHost = null,
|
ApiHost = null,
|
||||||
ApiKey = null,
|
ApiKey = null,
|
||||||
Remark = x.Description,
|
Remark = x.Description,
|
||||||
IsPremiumPackage = PremiumPackageConst.ModeIds.Contains(x.ModelId)
|
IsPremiumPackage = x.IsPremium
|
||||||
}).ToListAsync();
|
}).ToListAsync();
|
||||||
return output;
|
return output;
|
||||||
}
|
}
|
||||||
@@ -124,7 +146,11 @@ public class AiChatService : ApplicationService
|
|||||||
}
|
}
|
||||||
|
|
||||||
//如果是尊享包服务,需要校验是是否尊享包足够
|
//如果是尊享包服务,需要校验是是否尊享包足够
|
||||||
if (CurrentUser.IsAuthenticated && PremiumPackageConst.ModeIds.Contains(input.Model))
|
if (CurrentUser.IsAuthenticated)
|
||||||
|
{
|
||||||
|
var isPremium = await _modelManager.IsPremiumModelAsync(input.Model);
|
||||||
|
|
||||||
|
if (isPremium)
|
||||||
{
|
{
|
||||||
// 检查尊享token包用量
|
// 检查尊享token包用量
|
||||||
var availableTokens = await _premiumPackageManager.GetAvailableTokensAsync(CurrentUser.GetId());
|
var availableTokens = await _premiumPackageManager.GetAvailableTokensAsync(CurrentUser.GetId());
|
||||||
@@ -133,13 +159,13 @@ public class AiChatService : ApplicationService
|
|||||||
throw new UserFriendlyException("尊享token包用量不足,请先购买尊享token包");
|
throw new UserFriendlyException("尊享token包用量不足,请先购买尊享token包");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
//ai网关代理httpcontext
|
//ai网关代理httpcontext
|
||||||
await _aiGateWayManager.CompleteChatStreamForStatisticsAsync(_httpContextAccessor.HttpContext, input,
|
await _aiGateWayManager.CompleteChatStreamForStatisticsAsync(_httpContextAccessor.HttpContext, input,
|
||||||
CurrentUser.Id, sessionId, null, cancellationToken);
|
CurrentUser.Id, sessionId, null, CancellationToken.None);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// 发送消息
|
/// 发送消息
|
||||||
/// </summary>
|
/// </summary>
|
||||||
@@ -173,6 +199,80 @@ public class AiChatService : ApplicationService
|
|||||||
|
|
||||||
//ai网关代理httpcontext
|
//ai网关代理httpcontext
|
||||||
await _aiGateWayManager.CompleteChatStreamForStatisticsAsync(_httpContextAccessor.HttpContext, input,
|
await _aiGateWayManager.CompleteChatStreamForStatisticsAsync(_httpContextAccessor.HttpContext, input,
|
||||||
CurrentUser.Id, null, null, cancellationToken);
|
CurrentUser.Id, null, null, CancellationToken.None);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Agent 发送消息
|
||||||
|
/// </summary>
|
||||||
|
[HttpPost("ai-chat/agent/send")]
|
||||||
|
public async Task PostAgentSendAsync([FromBody] AgentSendInput input, CancellationToken cancellationToken)
|
||||||
|
{
|
||||||
|
var tokenValidation = await _tokenManager.ValidateTokenAsync(input.Token, input.ModelId);
|
||||||
|
|
||||||
|
await _aiBlacklistManager.VerifiyAiBlacklist(tokenValidation.UserId);
|
||||||
|
// 验证用户是否为VIP
|
||||||
|
var userInfo = await _accountService.GetAsync(null, null, tokenValidation.UserId);
|
||||||
|
if (userInfo == null)
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("用户信息不存在");
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查是否为VIP(使用RoleCodes判断)
|
||||||
|
if (!userInfo.RoleCodes.Contains(AiHubConst.VipRole) && userInfo.User.UserName != "cc")
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("该接口为尊享服务专用,需要VIP权限才能使用");
|
||||||
|
}
|
||||||
|
|
||||||
|
//如果是尊享包服务,需要校验是是否尊享包足够
|
||||||
|
var isPremium = await _modelManager.IsPremiumModelAsync(input.ModelId);
|
||||||
|
|
||||||
|
if (isPremium)
|
||||||
|
{
|
||||||
|
// 检查尊享token包用量
|
||||||
|
var availableTokens = await _premiumPackageManager.GetAvailableTokensAsync(tokenValidation.UserId);
|
||||||
|
if (availableTokens <= 0)
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("尊享token包用量不足,请先购买尊享token包");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
await _chatManager.AgentCompleteChatStreamAsync(_httpContextAccessor.HttpContext,
|
||||||
|
input.SessionId,
|
||||||
|
input.Content,
|
||||||
|
input.Token,
|
||||||
|
tokenValidation.TokenId,
|
||||||
|
input.ModelId,
|
||||||
|
tokenValidation.UserId,
|
||||||
|
input.Tools,
|
||||||
|
CancellationToken.None);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取 Agent 工具
|
||||||
|
/// </summary>
|
||||||
|
/// <returns></returns>
|
||||||
|
[HttpPost("ai-chat/agent/tool")]
|
||||||
|
public List<AgentToolOutput> GetAgentToolAsync()
|
||||||
|
{
|
||||||
|
var agentTools = _chatManager.GetTools().Select(x => new AgentToolOutput
|
||||||
|
{
|
||||||
|
Code = x.Code,
|
||||||
|
Name = x.Name
|
||||||
|
}).ToList();
|
||||||
|
return agentTools;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取 Agent 上下文
|
||||||
|
/// </summary>
|
||||||
|
/// <returns></returns>
|
||||||
|
[HttpPost("ai-chat/agent/context/{sessionId}")]
|
||||||
|
[Authorize]
|
||||||
|
public async Task<string?> GetAgentContextAsync([FromRoute] Guid sessionId)
|
||||||
|
{
|
||||||
|
var data = await _agentStoreRepository.GetFirstAsync(x => x.SessionId == sessionId);
|
||||||
|
return data?.Store;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -0,0 +1,275 @@
|
|||||||
|
using System.Text.Json;
|
||||||
|
using Microsoft.AspNetCore.Authorization;
|
||||||
|
using Microsoft.AspNetCore.Hosting;
|
||||||
|
using Microsoft.AspNetCore.Http;
|
||||||
|
using Microsoft.AspNetCore.Mvc;
|
||||||
|
using Volo.Abp;
|
||||||
|
using Volo.Abp.Application.Services;
|
||||||
|
using Volo.Abp.BackgroundJobs;
|
||||||
|
using Volo.Abp.Guids;
|
||||||
|
using Volo.Abp.Users;
|
||||||
|
using Yi.Framework.AiHub.Application.Contracts.Dtos.Chat;
|
||||||
|
using Yi.Framework.AiHub.Application.Jobs;
|
||||||
|
using Yi.Framework.AiHub.Domain.Entities.Chat;
|
||||||
|
using Yi.Framework.AiHub.Domain.Extensions;
|
||||||
|
using Yi.Framework.AiHub.Domain.Managers;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Consts;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
||||||
|
using Yi.Framework.SqlSugarCore.Abstractions;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Application.Services.Chat;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// AI图片生成服务
|
||||||
|
/// </summary>
|
||||||
|
[Authorize]
|
||||||
|
public class AiImageService : ApplicationService
|
||||||
|
{
|
||||||
|
private readonly ISqlSugarRepository<ImageStoreTaskAggregateRoot> _imageTaskRepository;
|
||||||
|
private readonly IBackgroundJobManager _backgroundJobManager;
|
||||||
|
private readonly AiBlacklistManager _aiBlacklistManager;
|
||||||
|
private readonly PremiumPackageManager _premiumPackageManager;
|
||||||
|
private readonly ModelManager _modelManager;
|
||||||
|
private readonly IGuidGenerator _guidGenerator;
|
||||||
|
private readonly IWebHostEnvironment _webHostEnvironment;
|
||||||
|
|
||||||
|
public AiImageService(
|
||||||
|
ISqlSugarRepository<ImageStoreTaskAggregateRoot> imageTaskRepository,
|
||||||
|
IBackgroundJobManager backgroundJobManager,
|
||||||
|
AiBlacklistManager aiBlacklistManager,
|
||||||
|
PremiumPackageManager premiumPackageManager,
|
||||||
|
ModelManager modelManager,
|
||||||
|
IGuidGenerator guidGenerator,
|
||||||
|
IWebHostEnvironment webHostEnvironment)
|
||||||
|
{
|
||||||
|
_imageTaskRepository = imageTaskRepository;
|
||||||
|
_backgroundJobManager = backgroundJobManager;
|
||||||
|
_aiBlacklistManager = aiBlacklistManager;
|
||||||
|
_premiumPackageManager = premiumPackageManager;
|
||||||
|
_modelManager = modelManager;
|
||||||
|
_guidGenerator = guidGenerator;
|
||||||
|
_webHostEnvironment = webHostEnvironment;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 生成图片(异步任务)
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="input">图片生成输入参数</param>
|
||||||
|
/// <returns>任务ID</returns>
|
||||||
|
[HttpPost("ai-image/generate")]
|
||||||
|
[Authorize]
|
||||||
|
public async Task<Guid> GenerateAsync([FromBody] ImageGenerationInput input)
|
||||||
|
{
|
||||||
|
var userId = CurrentUser.GetId();
|
||||||
|
|
||||||
|
// 黑名单校验
|
||||||
|
await _aiBlacklistManager.VerifiyAiBlacklist(userId);
|
||||||
|
|
||||||
|
// VIP校验
|
||||||
|
if (!CurrentUser.IsAiVip())
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("图片生成功能需要VIP用户才能使用,请购买VIP后重新登录重试");
|
||||||
|
}
|
||||||
|
|
||||||
|
// 尊享包校验 - 使用ModelManager统一判断
|
||||||
|
var isPremium = await _modelManager.IsPremiumModelAsync(input.ModelId);
|
||||||
|
if (isPremium)
|
||||||
|
{
|
||||||
|
var availableTokens = await _premiumPackageManager.GetAvailableTokensAsync(userId);
|
||||||
|
if (availableTokens <= 0)
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("尊享token包用量不足,请先购买尊享token包");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 创建任务实体
|
||||||
|
var task = new ImageStoreTaskAggregateRoot
|
||||||
|
{
|
||||||
|
Prompt = input.Prompt,
|
||||||
|
ReferenceImagesBase64 = input.ReferenceImagesBase64 ?? new List<string>(),
|
||||||
|
ReferenceImagesUrl = new List<string>(),
|
||||||
|
TaskStatus = TaskStatusEnum.Processing,
|
||||||
|
UserId = userId
|
||||||
|
};
|
||||||
|
|
||||||
|
await _imageTaskRepository.InsertAsync(task);
|
||||||
|
var taskId = task.Id;
|
||||||
|
|
||||||
|
// 构建请求JSON
|
||||||
|
var requestJson = JsonSerializer.Serialize(new
|
||||||
|
{
|
||||||
|
prompt = input.Prompt,
|
||||||
|
referenceImages = input.ReferenceImagesBase64
|
||||||
|
});
|
||||||
|
|
||||||
|
// 入队后台任务
|
||||||
|
await _backgroundJobManager.EnqueueAsync(new ImageGenerationJobArgs
|
||||||
|
{
|
||||||
|
TaskId = taskId,
|
||||||
|
ModelId = input.ModelId,
|
||||||
|
RequestJson = requestJson,
|
||||||
|
UserId = userId
|
||||||
|
});
|
||||||
|
|
||||||
|
return taskId;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 查询任务状态
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="taskId">任务ID</param>
|
||||||
|
/// <returns>任务详情</returns>
|
||||||
|
[HttpGet("ai-image/task/{taskId}")]
|
||||||
|
public async Task<ImageTaskOutput> GetTaskAsync([FromRoute] Guid taskId)
|
||||||
|
{
|
||||||
|
var userId = CurrentUser.GetId();
|
||||||
|
|
||||||
|
var task = await _imageTaskRepository.GetFirstAsync(x => x.Id == taskId && x.UserId == userId);
|
||||||
|
if (task == null)
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("任务不存在或无权访问");
|
||||||
|
}
|
||||||
|
|
||||||
|
return new ImageTaskOutput
|
||||||
|
{
|
||||||
|
Id = task.Id,
|
||||||
|
Prompt = task.Prompt,
|
||||||
|
ReferenceImagesBase64 = task.ReferenceImagesBase64,
|
||||||
|
ReferenceImagesUrl = task.ReferenceImagesUrl,
|
||||||
|
StoreBase64 = task.StoreBase64,
|
||||||
|
StoreUrl = task.StoreUrl,
|
||||||
|
TaskStatus = task.TaskStatus,
|
||||||
|
CreationTime = task.CreationTime
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 上传Base64图片转换为URL
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="base64Data">Base64图片数据(包含前缀如 data:image/png;base64,)</param>
|
||||||
|
/// <returns>图片访问URL</returns>
|
||||||
|
[HttpPost("ai-image/upload-base64")]
|
||||||
|
public async Task<string> UploadBase64ToUrlAsync([FromBody] string base64Data)
|
||||||
|
{
|
||||||
|
if (string.IsNullOrWhiteSpace(base64Data))
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("Base64数据不能为空");
|
||||||
|
}
|
||||||
|
|
||||||
|
// 解析Base64数据
|
||||||
|
string mimeType = "image/png";
|
||||||
|
string base64Content = base64Data;
|
||||||
|
|
||||||
|
if (base64Data.Contains(","))
|
||||||
|
{
|
||||||
|
var parts = base64Data.Split(',');
|
||||||
|
if (parts.Length == 2)
|
||||||
|
{
|
||||||
|
// 提取MIME类型
|
||||||
|
var header = parts[0];
|
||||||
|
if (header.Contains(":") && header.Contains(";"))
|
||||||
|
{
|
||||||
|
mimeType = header.Split(':')[1].Split(';')[0];
|
||||||
|
}
|
||||||
|
base64Content = parts[1];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取文件扩展名
|
||||||
|
var extension = mimeType switch
|
||||||
|
{
|
||||||
|
"image/png" => ".png",
|
||||||
|
"image/jpeg" => ".jpg",
|
||||||
|
"image/jpg" => ".jpg",
|
||||||
|
"image/gif" => ".gif",
|
||||||
|
"image/webp" => ".webp",
|
||||||
|
_ => ".png"
|
||||||
|
};
|
||||||
|
|
||||||
|
// 解码Base64
|
||||||
|
byte[] imageBytes;
|
||||||
|
try
|
||||||
|
{
|
||||||
|
imageBytes = Convert.FromBase64String(base64Content);
|
||||||
|
}
|
||||||
|
catch (FormatException)
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("Base64格式无效");
|
||||||
|
}
|
||||||
|
|
||||||
|
// 创建存储目录
|
||||||
|
var uploadPath = Path.Combine(_webHostEnvironment.ContentRootPath, "wwwroot", "ai-images");
|
||||||
|
if (!Directory.Exists(uploadPath))
|
||||||
|
{
|
||||||
|
Directory.CreateDirectory(uploadPath);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 生成文件名并保存
|
||||||
|
var fileId = _guidGenerator.Create();
|
||||||
|
var fileName = $"{fileId}{extension}";
|
||||||
|
var filePath = Path.Combine(uploadPath, fileName);
|
||||||
|
|
||||||
|
await File.WriteAllBytesAsync(filePath, imageBytes);
|
||||||
|
|
||||||
|
// 返回访问URL
|
||||||
|
return $"/ai-images/{fileName}";
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 分页查询任务列表
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="input">分页查询参数</param>
|
||||||
|
/// <returns>任务列表</returns>
|
||||||
|
[HttpGet("ai-image/tasks")]
|
||||||
|
public async Task<PagedResult<ImageTaskOutput>> GetTaskPageAsync([FromQuery] ImageTaskPageInput input)
|
||||||
|
{
|
||||||
|
var userId = CurrentUser.GetId();
|
||||||
|
|
||||||
|
var query = _imageTaskRepository._DbQueryable
|
||||||
|
.Where(x => x.UserId == userId)
|
||||||
|
.WhereIF(input.TaskStatus.HasValue, x => x.TaskStatus == input.TaskStatus!.Value)
|
||||||
|
.OrderByDescending(x => x.CreationTime);
|
||||||
|
|
||||||
|
var total = await query.CountAsync();
|
||||||
|
var items = await query
|
||||||
|
.Skip((input.PageIndex - 1) * input.PageSize)
|
||||||
|
.Take(input.PageSize)
|
||||||
|
.Select(x => new ImageTaskOutput
|
||||||
|
{
|
||||||
|
Id = x.Id,
|
||||||
|
Prompt = x.Prompt,
|
||||||
|
ReferenceImagesBase64 = x.ReferenceImagesBase64,
|
||||||
|
ReferenceImagesUrl = x.ReferenceImagesUrl,
|
||||||
|
StoreBase64 = x.StoreBase64,
|
||||||
|
StoreUrl = x.StoreUrl,
|
||||||
|
TaskStatus = x.TaskStatus,
|
||||||
|
CreationTime = x.CreationTime
|
||||||
|
})
|
||||||
|
.ToListAsync();
|
||||||
|
|
||||||
|
return new PagedResult<ImageTaskOutput>(total, items);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 分页结果
|
||||||
|
/// </summary>
|
||||||
|
/// <typeparam name="T">数据类型</typeparam>
|
||||||
|
public class PagedResult<T>
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 总数
|
||||||
|
/// </summary>
|
||||||
|
public long Total { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 数据列表
|
||||||
|
/// </summary>
|
||||||
|
public List<T> Items { get; set; }
|
||||||
|
|
||||||
|
public PagedResult(long total, List<T> items)
|
||||||
|
{
|
||||||
|
Total = total;
|
||||||
|
Items = items;
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,10 +1,12 @@
|
|||||||
using Mapster;
|
using Mapster;
|
||||||
|
using Microsoft.AspNetCore.Mvc;
|
||||||
using SqlSugar;
|
using SqlSugar;
|
||||||
using Volo.Abp.Application.Dtos;
|
using Volo.Abp.Application.Dtos;
|
||||||
using Volo.Abp.Application.Services;
|
using Volo.Abp.Application.Services;
|
||||||
using Yi.Framework.AiHub.Application.Contracts.Dtos.Model;
|
using Yi.Framework.AiHub.Application.Contracts.Dtos.Model;
|
||||||
using Yi.Framework.AiHub.Application.Contracts.IServices;
|
using Yi.Framework.AiHub.Application.Contracts.IServices;
|
||||||
using Yi.Framework.AiHub.Domain.Entities.Model;
|
using Yi.Framework.AiHub.Domain.Entities.Model;
|
||||||
|
using Yi.Framework.AiHub.Domain.Managers;
|
||||||
using Yi.Framework.AiHub.Domain.Shared.Consts;
|
using Yi.Framework.AiHub.Domain.Shared.Consts;
|
||||||
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
||||||
using Yi.Framework.AiHub.Domain.Shared.Extensions;
|
using Yi.Framework.AiHub.Domain.Shared.Extensions;
|
||||||
@@ -18,10 +20,12 @@ namespace Yi.Framework.AiHub.Application.Services.Chat;
|
|||||||
public class ModelService : ApplicationService, IModelService
|
public class ModelService : ApplicationService, IModelService
|
||||||
{
|
{
|
||||||
private readonly ISqlSugarRepository<AiModelEntity, Guid> _modelRepository;
|
private readonly ISqlSugarRepository<AiModelEntity, Guid> _modelRepository;
|
||||||
|
private readonly ModelManager _modelManager;
|
||||||
|
|
||||||
public ModelService(ISqlSugarRepository<AiModelEntity, Guid> modelRepository)
|
public ModelService(ISqlSugarRepository<AiModelEntity, Guid> modelRepository, ModelManager modelManager)
|
||||||
{
|
{
|
||||||
_modelRepository = modelRepository;
|
_modelRepository = modelRepository;
|
||||||
|
_modelManager = modelManager;
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
@@ -32,8 +36,7 @@ public class ModelService : ApplicationService, IModelService
|
|||||||
RefAsync<int> total = 0;
|
RefAsync<int> total = 0;
|
||||||
|
|
||||||
// 查询所有未删除的模型,使用WhereIF动态添加筛选条件
|
// 查询所有未删除的模型,使用WhereIF动态添加筛选条件
|
||||||
var models = await _modelRepository._DbQueryable
|
var modelIds = (await _modelRepository._DbQueryable
|
||||||
.Where(x => !x.IsDeleted)
|
|
||||||
.WhereIF(!string.IsNullOrWhiteSpace(input.SearchKey), x =>
|
.WhereIF(!string.IsNullOrWhiteSpace(input.SearchKey), x =>
|
||||||
x.Name.Contains(input.SearchKey) || x.ModelId.Contains(input.SearchKey))
|
x.Name.Contains(input.SearchKey) || x.ModelId.Contains(input.SearchKey))
|
||||||
.WhereIF(input.ProviderNames is not null, x =>
|
.WhereIF(input.ProviderNames is not null, x =>
|
||||||
@@ -42,29 +45,30 @@ public class ModelService : ApplicationService, IModelService
|
|||||||
input.ModelTypes.Contains(x.ModelType))
|
input.ModelTypes.Contains(x.ModelType))
|
||||||
.WhereIF(input.ModelApiTypes is not null, x =>
|
.WhereIF(input.ModelApiTypes is not null, x =>
|
||||||
input.ModelApiTypes.Contains(x.ModelApiType))
|
input.ModelApiTypes.Contains(x.ModelApiType))
|
||||||
.WhereIF(input.IsPremiumOnly == true, x =>
|
.WhereIF(input.IsPremiumOnly == true, x => x.IsPremium)
|
||||||
PremiumPackageConst.ModeIds.Contains(x.ModelId))
|
.GroupBy(x => x.ModelId)
|
||||||
.OrderBy(x => x.OrderNum)
|
.Select(x => x.ModelId)
|
||||||
.OrderBy(x => x.Name)
|
.ToPageListAsync(input.SkipCount, input.MaxResultCount, total));
|
||||||
.ToPageListAsync(input.SkipCount, input.MaxResultCount, total);
|
|
||||||
|
|
||||||
// 转换为DTO
|
var entities = await _modelRepository._DbQueryable.Where(x => modelIds.Contains(x.ModelId))
|
||||||
var result = models.Select(model => new ModelLibraryDto
|
.OrderBy(x => x.OrderNum)
|
||||||
|
.OrderBy(x => x.Name).ToListAsync();
|
||||||
|
|
||||||
|
var output= entities.GroupBy(x => x.ModelId).Select(x => new ModelLibraryDto
|
||||||
{
|
{
|
||||||
ModelId = model.ModelId,
|
ModelId = x.First().ModelId,
|
||||||
Name = model.Name,
|
Name = x.First().Name,
|
||||||
Description = model.Description,
|
Description = x.First().Description,
|
||||||
ModelType = model.ModelType,
|
ModelType = x.First().ModelType,
|
||||||
ModelTypeName = model.ModelType.GetDescription(),
|
ModelApiTypes = x.Select(y => new ModelApiTypeOutput { ModelApiType = y.ModelApiType }).ToList(),
|
||||||
ModelApiType = model.ModelApiType,
|
MultiplierShow = x.First().MultiplierShow,
|
||||||
ModelApiTypeName = model.ModelApiType.GetDescription(),
|
ProviderName = x.First().ProviderName,
|
||||||
MultiplierShow = model.MultiplierShow,
|
IconUrl = x.First().IconUrl,
|
||||||
ProviderName = model.ProviderName,
|
IsPremium = x.First().IsPremium,
|
||||||
IconUrl = model.IconUrl,
|
OrderNum = x.First().OrderNum
|
||||||
IsPremium = PremiumPackageConst.ModeIds.Contains(model.ModelId)
|
|
||||||
}).ToList();
|
}).ToList();
|
||||||
|
|
||||||
return new PagedResultDto<ModelLibraryDto>(total, result);
|
return new PagedResultDto<ModelLibraryDto>(total, output);
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
@@ -76,11 +80,11 @@ public class ModelService : ApplicationService, IModelService
|
|||||||
.Where(x => !x.IsDeleted)
|
.Where(x => !x.IsDeleted)
|
||||||
.Where(x => !string.IsNullOrEmpty(x.ProviderName))
|
.Where(x => !string.IsNullOrEmpty(x.ProviderName))
|
||||||
.GroupBy(x => x.ProviderName)
|
.GroupBy(x => x.ProviderName)
|
||||||
.OrderBy(x => x.ProviderName)
|
.OrderBy(x => x.OrderNum)
|
||||||
.Select(x => x.ProviderName)
|
.Select(x => x.ProviderName)
|
||||||
.ToListAsync();
|
.ToListAsync();
|
||||||
|
|
||||||
return providers;
|
return providers!;
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
@@ -114,4 +118,13 @@ public class ModelService : ApplicationService, IModelService
|
|||||||
|
|
||||||
return Task.FromResult(options);
|
return Task.FromResult(options);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 清除尊享模型ID缓存
|
||||||
|
/// </summary>
|
||||||
|
[HttpPost("model/clear-premium-cache")]
|
||||||
|
public async Task ClearPremiumModelCacheAsync()
|
||||||
|
{
|
||||||
|
await _modelManager.ClearPremiumModelIdsCacheAsync();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
@@ -7,8 +7,10 @@ using Volo.Abp.Application.Services;
|
|||||||
using Volo.Abp.Users;
|
using Volo.Abp.Users;
|
||||||
using Yi.Framework.AiHub.Application.Contracts.Dtos.Token;
|
using Yi.Framework.AiHub.Application.Contracts.Dtos.Token;
|
||||||
using Yi.Framework.AiHub.Domain.Entities;
|
using Yi.Framework.AiHub.Domain.Entities;
|
||||||
|
using Yi.Framework.AiHub.Domain.Entities.Model;
|
||||||
using Yi.Framework.AiHub.Domain.Entities.OpenApi;
|
using Yi.Framework.AiHub.Domain.Entities.OpenApi;
|
||||||
using Yi.Framework.AiHub.Domain.Extensions;
|
using Yi.Framework.AiHub.Domain.Extensions;
|
||||||
|
using Yi.Framework.AiHub.Domain.Managers;
|
||||||
using Yi.Framework.AiHub.Domain.Shared.Consts;
|
using Yi.Framework.AiHub.Domain.Shared.Consts;
|
||||||
using Yi.Framework.Ddd.Application.Contracts;
|
using Yi.Framework.Ddd.Application.Contracts;
|
||||||
using Yi.Framework.SqlSugarCore.Abstractions;
|
using Yi.Framework.SqlSugarCore.Abstractions;
|
||||||
@@ -23,13 +25,15 @@ public class TokenService : ApplicationService
|
|||||||
{
|
{
|
||||||
private readonly ISqlSugarRepository<TokenAggregateRoot> _tokenRepository;
|
private readonly ISqlSugarRepository<TokenAggregateRoot> _tokenRepository;
|
||||||
private readonly ISqlSugarRepository<UsageStatisticsAggregateRoot> _usageStatisticsRepository;
|
private readonly ISqlSugarRepository<UsageStatisticsAggregateRoot> _usageStatisticsRepository;
|
||||||
|
private readonly ModelManager _modelManager;
|
||||||
public TokenService(
|
public TokenService(
|
||||||
ISqlSugarRepository<TokenAggregateRoot> tokenRepository,
|
ISqlSugarRepository<TokenAggregateRoot> tokenRepository,
|
||||||
ISqlSugarRepository<UsageStatisticsAggregateRoot> usageStatisticsRepository)
|
ISqlSugarRepository<UsageStatisticsAggregateRoot> usageStatisticsRepository,
|
||||||
|
ModelManager modelManager)
|
||||||
{
|
{
|
||||||
_tokenRepository = tokenRepository;
|
_tokenRepository = tokenRepository;
|
||||||
_usageStatisticsRepository = usageStatisticsRepository;
|
_usageStatisticsRepository = usageStatisticsRepository;
|
||||||
|
_modelManager = modelManager;
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
@@ -51,8 +55,8 @@ public class TokenService : ApplicationService
|
|||||||
return new PagedResultDto<TokenGetListOutputDto>();
|
return new PagedResultDto<TokenGetListOutputDto>();
|
||||||
}
|
}
|
||||||
|
|
||||||
// 获取尊享包模型ID列表
|
// 通过ModelManager获取尊享包模型ID列表
|
||||||
var premiumModelIds = PremiumPackageConst.ModeIds;
|
var premiumModelIds = await _modelManager.GetPremiumModelIdsAsync();
|
||||||
|
|
||||||
// 批量查询所有Token的尊享包已使用额度
|
// 批量查询所有Token的尊享包已使用额度
|
||||||
var tokenIds = tokens.Select(t => t.Id).ToList();
|
var tokenIds = tokens.Select(t => t.Id).ToList();
|
||||||
|
|||||||
@@ -1,9 +1,11 @@
|
|||||||
using Microsoft.AspNetCore.Http;
|
using System.Text.Json;
|
||||||
|
using Microsoft.AspNetCore.Http;
|
||||||
using Microsoft.AspNetCore.Mvc;
|
using Microsoft.AspNetCore.Mvc;
|
||||||
using Microsoft.Extensions.Logging;
|
using Microsoft.Extensions.Logging;
|
||||||
using Volo.Abp.Application.Services;
|
using Volo.Abp.Application.Services;
|
||||||
using Volo.Abp.Users;
|
using Volo.Abp.Users;
|
||||||
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.Model;
|
||||||
using Yi.Framework.AiHub.Domain.Extensions;
|
using Yi.Framework.AiHub.Domain.Extensions;
|
||||||
using Yi.Framework.AiHub.Domain.Managers;
|
using Yi.Framework.AiHub.Domain.Managers;
|
||||||
@@ -25,24 +27,27 @@ public class OpenApiService : ApplicationService
|
|||||||
private readonly ILogger<OpenApiService> _logger;
|
private readonly ILogger<OpenApiService> _logger;
|
||||||
private readonly TokenManager _tokenManager;
|
private readonly TokenManager _tokenManager;
|
||||||
private readonly AiGateWayManager _aiGateWayManager;
|
private readonly AiGateWayManager _aiGateWayManager;
|
||||||
private readonly ISqlSugarRepository<AiModelEntity> _aiModelRepository;
|
private readonly ModelManager _modelManager;
|
||||||
private readonly AiBlacklistManager _aiBlacklistManager;
|
private readonly AiBlacklistManager _aiBlacklistManager;
|
||||||
private readonly IAccountService _accountService;
|
private readonly IAccountService _accountService;
|
||||||
private readonly PremiumPackageManager _premiumPackageManager;
|
private readonly PremiumPackageManager _premiumPackageManager;
|
||||||
|
private readonly ISqlSugarRepository<ImageStoreTaskAggregateRoot> _imageStoreRepository;
|
||||||
|
private readonly ISqlSugarRepository<AiModelEntity> _aiModelRepository;
|
||||||
public OpenApiService(IHttpContextAccessor httpContextAccessor, ILogger<OpenApiService> logger,
|
public OpenApiService(IHttpContextAccessor httpContextAccessor, ILogger<OpenApiService> logger,
|
||||||
TokenManager tokenManager, AiGateWayManager aiGateWayManager,
|
TokenManager tokenManager, AiGateWayManager aiGateWayManager,
|
||||||
ISqlSugarRepository<AiModelEntity> aiModelRepository, AiBlacklistManager aiBlacklistManager,
|
ModelManager modelManager, AiBlacklistManager aiBlacklistManager,
|
||||||
IAccountService accountService, PremiumPackageManager premiumPackageManager)
|
IAccountService accountService, PremiumPackageManager premiumPackageManager, ISqlSugarRepository<ImageStoreTaskAggregateRoot> imageStoreRepository, ISqlSugarRepository<AiModelEntity> aiModelRepository)
|
||||||
{
|
{
|
||||||
_httpContextAccessor = httpContextAccessor;
|
_httpContextAccessor = httpContextAccessor;
|
||||||
_logger = logger;
|
_logger = logger;
|
||||||
_tokenManager = tokenManager;
|
_tokenManager = tokenManager;
|
||||||
_aiGateWayManager = aiGateWayManager;
|
_aiGateWayManager = aiGateWayManager;
|
||||||
_aiModelRepository = aiModelRepository;
|
_modelManager = modelManager;
|
||||||
_aiBlacklistManager = aiBlacklistManager;
|
_aiBlacklistManager = aiBlacklistManager;
|
||||||
_accountService = accountService;
|
_accountService = accountService;
|
||||||
_premiumPackageManager = premiumPackageManager;
|
_premiumPackageManager = premiumPackageManager;
|
||||||
|
_imageStoreRepository = imageStoreRepository;
|
||||||
|
_aiModelRepository = aiModelRepository;
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
@@ -62,7 +67,9 @@ public class OpenApiService : ApplicationService
|
|||||||
await _aiBlacklistManager.VerifiyAiBlacklist(userId);
|
await _aiBlacklistManager.VerifiyAiBlacklist(userId);
|
||||||
|
|
||||||
//如果是尊享包服务,需要校验是是否尊享包足够
|
//如果是尊享包服务,需要校验是是否尊享包足够
|
||||||
if (PremiumPackageConst.ModeIds.Contains(input.Model))
|
var isPremium = await _modelManager.IsPremiumModelAsync(input.Model);
|
||||||
|
|
||||||
|
if (isPremium)
|
||||||
{
|
{
|
||||||
// 检查尊享token包用量
|
// 检查尊享token包用量
|
||||||
var availableTokens = await _premiumPackageManager.GetAvailableTokensAsync(userId);
|
var availableTokens = await _premiumPackageManager.GetAvailableTokensAsync(userId);
|
||||||
@@ -76,13 +83,13 @@ 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, tokenId, cancellationToken);
|
userId, null, tokenId,CancellationToken.None);
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
await _aiGateWayManager.CompleteChatForStatisticsAsync(_httpContextAccessor.HttpContext, input, userId,
|
await _aiGateWayManager.CompleteChatForStatisticsAsync(_httpContextAccessor.HttpContext, input, userId,
|
||||||
null, tokenId,
|
null, tokenId,
|
||||||
cancellationToken);
|
CancellationToken.None);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -190,14 +197,14 @@ public class OpenApiService : ApplicationService
|
|||||||
{
|
{
|
||||||
await _aiGateWayManager.AnthropicCompleteChatStreamForStatisticsAsync(_httpContextAccessor.HttpContext,
|
await _aiGateWayManager.AnthropicCompleteChatStreamForStatisticsAsync(_httpContextAccessor.HttpContext,
|
||||||
input,
|
input,
|
||||||
userId, null, tokenId, cancellationToken);
|
userId, null, tokenId, CancellationToken.None);
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
await _aiGateWayManager.AnthropicCompleteChatForStatisticsAsync(_httpContextAccessor.HttpContext, input,
|
await _aiGateWayManager.AnthropicCompleteChatForStatisticsAsync(_httpContextAccessor.HttpContext, input,
|
||||||
userId,
|
userId,
|
||||||
null, tokenId,
|
null, tokenId,
|
||||||
cancellationToken);
|
CancellationToken.None);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -242,18 +249,77 @@ public class OpenApiService : ApplicationService
|
|||||||
{
|
{
|
||||||
await _aiGateWayManager.OpenAiResponsesStreamForStatisticsAsync(_httpContextAccessor.HttpContext,
|
await _aiGateWayManager.OpenAiResponsesStreamForStatisticsAsync(_httpContextAccessor.HttpContext,
|
||||||
input,
|
input,
|
||||||
userId, null, tokenId, cancellationToken);
|
userId, null, tokenId, CancellationToken.None);
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
await _aiGateWayManager.OpenAiResponsesAsyncForStatisticsAsync(_httpContextAccessor.HttpContext, input,
|
await _aiGateWayManager.OpenAiResponsesAsyncForStatisticsAsync(_httpContextAccessor.HttpContext, input,
|
||||||
userId,
|
userId,
|
||||||
null, tokenId,
|
null, tokenId,
|
||||||
cancellationToken);
|
CancellationToken.None);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 生成-Gemini (尊享服务专用)
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="input"></param>
|
||||||
|
/// <param name="modelId"></param>
|
||||||
|
/// <param name="alt"></param>
|
||||||
|
/// <param name="cancellationToken"></param>
|
||||||
|
[HttpPost("openApi/v1beta/models/{modelId}:{action:regex(^(generateContent|streamGenerateContent)$)}")]
|
||||||
|
public async Task GenerateContentAsync([FromBody] JsonElement input,
|
||||||
|
[FromRoute] string modelId,
|
||||||
|
[FromQuery] string? alt, CancellationToken cancellationToken)
|
||||||
|
{
|
||||||
|
//前面都是校验,后面才是真正的调用
|
||||||
|
var httpContext = this._httpContextAccessor.HttpContext;
|
||||||
|
var tokenValidation = await _tokenManager.ValidateTokenAsync(GetTokenByHttpContext(httpContext), modelId);
|
||||||
|
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 (alt == "sse")
|
||||||
|
{
|
||||||
|
await _aiGateWayManager.GeminiGenerateContentStreamForStatisticsAsync(_httpContextAccessor.HttpContext,
|
||||||
|
modelId, input,
|
||||||
|
userId,
|
||||||
|
null, tokenId,
|
||||||
|
CancellationToken.None);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
await _aiGateWayManager.GeminiGenerateContentForStatisticsAsync(_httpContextAccessor.HttpContext,
|
||||||
|
modelId, input,
|
||||||
|
userId,
|
||||||
|
null, tokenId,
|
||||||
|
CancellationToken.None);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#region 私有
|
#region 私有
|
||||||
|
|
||||||
private string? GetTokenByHttpContext(HttpContext httpContext)
|
private string? GetTokenByHttpContext(HttpContext httpContext)
|
||||||
@@ -265,6 +331,13 @@ public class OpenApiService : ApplicationService
|
|||||||
return apiKeyHeader.Trim();
|
return apiKeyHeader.Trim();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 再从 谷歌 获取
|
||||||
|
string googApiKeyHeader = httpContext.Request.Headers["x-goog-api-key"];
|
||||||
|
if (!string.IsNullOrWhiteSpace(googApiKeyHeader))
|
||||||
|
{
|
||||||
|
return googApiKeyHeader.Trim();
|
||||||
|
}
|
||||||
|
|
||||||
// 再检查 Authorization 头
|
// 再检查 Authorization 头
|
||||||
string authHeader = httpContext.Request.Headers["Authorization"];
|
string authHeader = httpContext.Request.Headers["Authorization"];
|
||||||
if (!string.IsNullOrWhiteSpace(authHeader) &&
|
if (!string.IsNullOrWhiteSpace(authHeader) &&
|
||||||
|
|||||||
@@ -122,11 +122,15 @@ public class PayService : ApplicationService, IPayService
|
|||||||
// 5. 根据商品类型进行不同的处理
|
// 5. 根据商品类型进行不同的处理
|
||||||
if (order.GoodsType.IsPremiumPackage())
|
if (order.GoodsType.IsPremiumPackage())
|
||||||
{
|
{
|
||||||
|
var tokenAmount = order.GoodsType.GetTokenAmount();
|
||||||
|
var packageName = order.GoodsType.GetDisplayName();
|
||||||
// 处理尊享包商品:创建尊享包记录
|
// 处理尊享包商品:创建尊享包记录
|
||||||
await _premiumPackageManager.CreatePremiumPackageAsync(
|
await _premiumPackageManager.CreatePremiumPackageAsync(
|
||||||
order.UserId,
|
order.UserId,
|
||||||
order.GoodsType,
|
tokenAmount,
|
||||||
|
packageName,
|
||||||
order.TotalAmount,
|
order.TotalAmount,
|
||||||
|
"自助充值",
|
||||||
expireMonths: null // 尊享包不设置过期时间,或者可以根据需求设置
|
expireMonths: null // 尊享包不设置过期时间,或者可以根据需求设置
|
||||||
);
|
);
|
||||||
|
|
||||||
|
|||||||
@@ -55,13 +55,24 @@ namespace Yi.Framework.AiHub.Application.Services
|
|||||||
/// </summary>
|
/// </summary>
|
||||||
/// <param name="input">充值输入参数</param>
|
/// <param name="input">充值输入参数</param>
|
||||||
/// <returns></returns>
|
/// <returns></returns>
|
||||||
[HttpPost("recharge/vip")]
|
[RemoteService(isEnabled:false)]
|
||||||
public async Task RechargeVipAsync(RechargeCreateInput input)
|
public async Task RechargeVipAsync(RechargeCreateInput input)
|
||||||
{
|
{
|
||||||
DateTime? expireDateTime = null;
|
DateTime? expireDateTime = null;
|
||||||
|
|
||||||
// 如果传入了月数,计算过期时间
|
// 计算总天数(1个月 = 31天)
|
||||||
|
int totalDays = 0;
|
||||||
if (input.Months.HasValue && input.Months.Value > 0)
|
if (input.Months.HasValue && input.Months.Value > 0)
|
||||||
|
{
|
||||||
|
totalDays += input.Months.Value * 31;
|
||||||
|
}
|
||||||
|
if (input.Days.HasValue && input.Days.Value > 0)
|
||||||
|
{
|
||||||
|
totalDays += input.Days.Value;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 如果有天数,计算过期时间
|
||||||
|
if (totalDays > 0)
|
||||||
{
|
{
|
||||||
// 直接查询该用户最大的过期时间
|
// 直接查询该用户最大的过期时间
|
||||||
var maxExpireTime = await _repository._DbQueryable
|
var maxExpireTime = await _repository._DbQueryable
|
||||||
@@ -75,9 +86,9 @@ namespace Yi.Framework.AiHub.Application.Services
|
|||||||
? maxExpireTime.Value
|
? maxExpireTime.Value
|
||||||
: DateTime.Now;
|
: DateTime.Now;
|
||||||
// 计算新的过期时间
|
// 计算新的过期时间
|
||||||
expireDateTime = baseDateTime.AddMonths(input.Months.Value);
|
expireDateTime = baseDateTime.AddDays(totalDays);
|
||||||
}
|
}
|
||||||
// 如果月数为空或0,表示永久VIP,ExpireDateTime保持为null
|
// 如果总天数为0,表示永久VIP,ExpireDateTime保持为null
|
||||||
|
|
||||||
// 创建充值记录
|
// 创建充值记录
|
||||||
var rechargeRecord = new AiRechargeAggregateRoot
|
var rechargeRecord = new AiRechargeAggregateRoot
|
||||||
|
|||||||
@@ -9,8 +9,10 @@ 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.Model;
|
||||||
using Yi.Framework.AiHub.Domain.Entities.OpenApi;
|
using Yi.Framework.AiHub.Domain.Entities.OpenApi;
|
||||||
using Yi.Framework.AiHub.Domain.Extensions;
|
using Yi.Framework.AiHub.Domain.Extensions;
|
||||||
|
using Yi.Framework.AiHub.Domain.Managers;
|
||||||
using Yi.Framework.AiHub.Domain.Shared.Consts;
|
using Yi.Framework.AiHub.Domain.Shared.Consts;
|
||||||
using Yi.Framework.Ddd.Application.Contracts;
|
using Yi.Framework.Ddd.Application.Contracts;
|
||||||
using Yi.Framework.SqlSugarCore.Abstractions;
|
using Yi.Framework.SqlSugarCore.Abstractions;
|
||||||
@@ -27,17 +29,19 @@ public class UsageStatisticsService : ApplicationService, IUsageStatisticsServic
|
|||||||
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;
|
private readonly ISqlSugarRepository<TokenAggregateRoot> _tokenRepository;
|
||||||
|
private readonly ModelManager _modelManager;
|
||||||
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)
|
ISqlSugarRepository<TokenAggregateRoot> tokenRepository,
|
||||||
|
ModelManager modelManager)
|
||||||
{
|
{
|
||||||
_messageRepository = messageRepository;
|
_messageRepository = messageRepository;
|
||||||
_usageStatisticsRepository = usageStatisticsRepository;
|
_usageStatisticsRepository = usageStatisticsRepository;
|
||||||
_premiumPackageRepository = premiumPackageRepository;
|
_premiumPackageRepository = premiumPackageRepository;
|
||||||
_tokenRepository = tokenRepository;
|
_tokenRepository = tokenRepository;
|
||||||
|
_modelManager = modelManager;
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
@@ -181,7 +185,9 @@ public class UsageStatisticsService : ApplicationService, IUsageStatisticsServic
|
|||||||
public async Task<List<TokenPremiumUsageDto>> GetPremiumTokenUsageByTokenAsync()
|
public async Task<List<TokenPremiumUsageDto>> GetPremiumTokenUsageByTokenAsync()
|
||||||
{
|
{
|
||||||
var userId = CurrentUser.GetId();
|
var userId = CurrentUser.GetId();
|
||||||
var premiumModelIds = PremiumPackageConst.ModeIds;
|
|
||||||
|
// 通过ModelManager获取所有尊享模型的ModelId列表
|
||||||
|
var premiumModelIds = await _modelManager.GetPremiumModelIdsAsync();
|
||||||
|
|
||||||
// 从UsageStatistics表获取尊享模型的token消耗统计(按TokenId聚合)
|
// 从UsageStatistics表获取尊享模型的token消耗统计(按TokenId聚合)
|
||||||
var tokenUsages = await _usageStatisticsRepository._DbQueryable
|
var tokenUsages = await _usageStatisticsRepository._DbQueryable
|
||||||
|
|||||||
@@ -2,6 +2,10 @@
|
|||||||
<Import Project="..\..\..\common.props" />
|
<Import Project="..\..\..\common.props" />
|
||||||
|
|
||||||
|
|
||||||
|
<ItemGroup>
|
||||||
|
<PackageReference Include="Volo.Abp.BackgroundJobs.Abstractions" Version="$(AbpVersion)" />
|
||||||
|
</ItemGroup>
|
||||||
|
|
||||||
<ItemGroup>
|
<ItemGroup>
|
||||||
<ProjectReference Include="..\..\..\framework\Yi.Framework.Ddd.Application\Yi.Framework.Ddd.Application.csproj" />
|
<ProjectReference Include="..\..\..\framework\Yi.Framework.Ddd.Application\Yi.Framework.Ddd.Application.csproj" />
|
||||||
<ProjectReference Include="..\Yi.Framework.AiHub.Application.Contracts\Yi.Framework.AiHub.Application.Contracts.csproj" />
|
<ProjectReference Include="..\Yi.Framework.AiHub.Application.Contracts\Yi.Framework.AiHub.Application.Contracts.csproj" />
|
||||||
|
|||||||
@@ -1,3 +1,7 @@
|
|||||||
|
using System.ComponentModel;
|
||||||
|
using System.Text.Json;
|
||||||
|
using ModelContextProtocol;
|
||||||
|
using ModelContextProtocol.Server;
|
||||||
using Yi.Framework.AiHub.Application.Contracts;
|
using Yi.Framework.AiHub.Application.Contracts;
|
||||||
using Yi.Framework.AiHub.Domain;
|
using Yi.Framework.AiHub.Domain;
|
||||||
using Yi.Framework.Ddd.Application;
|
using Yi.Framework.Ddd.Application;
|
||||||
|
|||||||
@@ -0,0 +1,16 @@
|
|||||||
|
namespace Yi.Framework.AiHub.Domain.Shared.Attributes;
|
||||||
|
|
||||||
|
[AttributeUsage(AttributeTargets.Class|AttributeTargets.Method)]
|
||||||
|
public class YiAgentToolAttribute:Attribute
|
||||||
|
{
|
||||||
|
public YiAgentToolAttribute()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
public YiAgentToolAttribute(string name)
|
||||||
|
{
|
||||||
|
Name = name;
|
||||||
|
}
|
||||||
|
|
||||||
|
public string Name { get; set; }
|
||||||
|
}
|
||||||
@@ -10,6 +10,15 @@ public class PremiumPackageConst
|
|||||||
"claude-haiku-4-5-20251001",
|
"claude-haiku-4-5-20251001",
|
||||||
"claude-opus-4-5-20251101",
|
"claude-opus-4-5-20251101",
|
||||||
"gemini-3-pro-preview",
|
"gemini-3-pro-preview",
|
||||||
"gpt-5.1-codex-max"
|
"gpt-5.1-codex-max",
|
||||||
|
"gpt-5.2",
|
||||||
|
"gemini-3-pro-high",
|
||||||
|
"gemini-3-pro-image-preview",
|
||||||
|
"gpt-5.2-codex-xhigh",
|
||||||
|
|
||||||
|
"glm-4.7",
|
||||||
|
"yi-claude-sonnet-4-5-20250929",
|
||||||
|
"yi-claude-haiku-4-5-20251001",
|
||||||
|
"yi-claude-opus-4-5-20251101",
|
||||||
];
|
];
|
||||||
}
|
}
|
||||||
@@ -12,7 +12,7 @@ public sealed class AnthropicInput
|
|||||||
|
|
||||||
[JsonPropertyName("max_tokens")] public int? MaxTokens { get; set; }
|
[JsonPropertyName("max_tokens")] public int? MaxTokens { get; set; }
|
||||||
|
|
||||||
[JsonPropertyName("messages")] public JsonElement? Messages { get; set; }
|
[JsonPropertyName("messages")] public IList<AnthropicMessageInput> Messages { get; set; }
|
||||||
|
|
||||||
[JsonPropertyName("tools")] public IList<AnthropicMessageTool>? Tools { get; set; }
|
[JsonPropertyName("tools")] public IList<AnthropicMessageTool>? Tools { get; set; }
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,45 @@
|
|||||||
|
using System.Text.Json;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Extensions;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Domain.Shared.Dtos.Gemini;
|
||||||
|
|
||||||
|
public static class GeminiGenerateContentAcquirer
|
||||||
|
{
|
||||||
|
public static ThorUsageResponse? GetUsage(JsonElement response)
|
||||||
|
{
|
||||||
|
var usage = response.GetPath("usageMetadata");
|
||||||
|
if (!usage.HasValue)
|
||||||
|
{
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
var inputTokens = usage.Value.GetPath("promptTokenCount").GetInt();
|
||||||
|
var outputTokens = usage.Value.GetPath("candidatesTokenCount").GetInt()
|
||||||
|
+ usage.Value.GetPath("cachedContentTokenCount").GetInt()
|
||||||
|
+ usage.Value.GetPath("thoughtsTokenCount").GetInt()
|
||||||
|
+ usage.Value.GetPath("toolUsePromptTokenCount").GetInt();
|
||||||
|
|
||||||
|
|
||||||
|
return new ThorUsageResponse
|
||||||
|
{
|
||||||
|
PromptTokens = inputTokens,
|
||||||
|
InputTokens = inputTokens,
|
||||||
|
OutputTokens = outputTokens,
|
||||||
|
CompletionTokens = outputTokens,
|
||||||
|
TotalTokens = inputTokens + outputTokens,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取图片url,包含前缀
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="response"></param>
|
||||||
|
/// <returns></returns>
|
||||||
|
public static string GetImageBase64(JsonElement response)
|
||||||
|
{
|
||||||
|
//todo
|
||||||
|
//获取他的base64字符串
|
||||||
|
return string.Empty;
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -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,114 @@
|
|||||||
|
using System;
|
||||||
|
using System.Reflection;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Domain.Shared.Enums;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 激活码商品特性
|
||||||
|
/// </summary>
|
||||||
|
[AttributeUsage(AttributeTargets.Field)]
|
||||||
|
public class ActivationCodeGoodsAttribute : Attribute
|
||||||
|
{
|
||||||
|
public decimal Price { get; }
|
||||||
|
public long TokenAmount { get; }
|
||||||
|
public int VipMonths { get; }
|
||||||
|
public int VipDays { get; }
|
||||||
|
public bool IsCombo { get; }
|
||||||
|
public bool IsReusable { get; }
|
||||||
|
public bool IsSameTypeOnce { get; }
|
||||||
|
public string DisplayName { get; }
|
||||||
|
public string Content { get; }
|
||||||
|
|
||||||
|
public ActivationCodeGoodsAttribute(
|
||||||
|
double price,
|
||||||
|
long tokenAmount,
|
||||||
|
int vipMonths,
|
||||||
|
bool isCombo,
|
||||||
|
bool isReusable,
|
||||||
|
bool isSameTypeOnce,
|
||||||
|
string displayName,
|
||||||
|
string content,
|
||||||
|
int vipDays = 0)
|
||||||
|
{
|
||||||
|
Price = (decimal)price;
|
||||||
|
TokenAmount = tokenAmount;
|
||||||
|
VipMonths = vipMonths;
|
||||||
|
VipDays = vipDays;
|
||||||
|
IsCombo = isCombo;
|
||||||
|
IsReusable = isReusable;
|
||||||
|
IsSameTypeOnce = isSameTypeOnce;
|
||||||
|
DisplayName = displayName;
|
||||||
|
Content = content;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 激活码商品类型
|
||||||
|
/// </summary>
|
||||||
|
public enum ActivationCodeGoodsTypeEnum
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 48.90【意心Ai会员1月+2000w 尊享Token】新人首单组合包(推荐)
|
||||||
|
/// </summary>
|
||||||
|
[ActivationCodeGoods(price: 48.90, tokenAmount: 20000000, vipMonths: 1, isCombo: true, isReusable: false,
|
||||||
|
isSameTypeOnce: true, displayName: "意心Ai会员1月+2000w 尊享Token", content: "新人首单组合包")]
|
||||||
|
Vip1MonthPlus2000W = 1,
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 1.00【10w 尊享Token】测试体验包
|
||||||
|
/// </summary>
|
||||||
|
[ActivationCodeGoods(price: 1.00, tokenAmount: 100000, vipMonths: 0, isCombo: false, isReusable: false,
|
||||||
|
isSameTypeOnce: false, displayName: "10w 尊享Token", content: "测试体验包")]
|
||||||
|
Premium10W = 2,
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 9.90【1000w 尊享Token】意心会员首单回馈包
|
||||||
|
/// </summary>
|
||||||
|
[ActivationCodeGoods(price: 9.90, tokenAmount: 10000000, vipMonths: 0, isCombo: false, isReusable: false,
|
||||||
|
isSameTypeOnce: true, displayName: "1000w 尊享Token", content: "意心会员首单回馈包")]
|
||||||
|
Premium1000W = 3,
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 22.90【意心Ai会员1月】特价包
|
||||||
|
/// </summary>
|
||||||
|
[ActivationCodeGoods(price: 22.90, tokenAmount: 0, vipMonths: 1, isCombo: false, isReusable: false,
|
||||||
|
isSameTypeOnce: false, displayName: "意心Ai会员1月", content: "特价包")]
|
||||||
|
Vip1Month = 4,
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 138.90【5000w 尊享Token】特价包
|
||||||
|
/// </summary>
|
||||||
|
[ActivationCodeGoods(price: 138.90, tokenAmount: 50000000, vipMonths: 0, isCombo: false, isReusable: false,
|
||||||
|
isSameTypeOnce: false, displayName: "5000w 尊享Token", content: "特价包")]
|
||||||
|
Premium5000W = 5,
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 198.90【1亿 尊享Token】特价包
|
||||||
|
/// </summary>
|
||||||
|
[ActivationCodeGoods(price: 198.90, tokenAmount: 100000000, vipMonths: 0, isCombo: false, isReusable: false,
|
||||||
|
isSameTypeOnce: false, displayName: "1亿 尊享Token", content: "特价包")]
|
||||||
|
Premium1Yi = 6,
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 1【意心Ai会员1天+50w 尊享Token】新人试用首单组合包
|
||||||
|
/// </summary>
|
||||||
|
[ActivationCodeGoods(price: 1, tokenAmount: 500000, vipMonths: 0, vipDays: 1, isCombo: true, isReusable: false,
|
||||||
|
isSameTypeOnce: true, displayName: "意心Ai会员1天+50w 尊享Token", content: "新人首单组合包")]
|
||||||
|
Vip1DayTest = 7,
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 0【10w 尊享Token】免费包
|
||||||
|
/// </summary>
|
||||||
|
[ActivationCodeGoods(price: 0, tokenAmount: 100000, vipMonths: 0, isCombo: false, isReusable: true,
|
||||||
|
isSameTypeOnce: false, displayName: "10w 尊享Token", content: "免费包")]
|
||||||
|
Premium10WFree = 100
|
||||||
|
}
|
||||||
|
|
||||||
|
public static class ActivationCodeGoodsTypeEnumExtensions
|
||||||
|
{
|
||||||
|
public static ActivationCodeGoodsAttribute? GetGoods(this ActivationCodeGoodsTypeEnum goodsType)
|
||||||
|
{
|
||||||
|
var fieldInfo = goodsType.GetType().GetField(goodsType.ToString());
|
||||||
|
return fieldInfo?.GetCustomAttribute<ActivationCodeGoodsAttribute>();
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -259,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>
|
||||||
@@ -271,10 +271,9 @@ 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);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -11,5 +11,8 @@ public enum ModelApiTypeEnum
|
|||||||
Claude,
|
Claude,
|
||||||
|
|
||||||
[Description("Response")]
|
[Description("Response")]
|
||||||
Response
|
Response,
|
||||||
|
|
||||||
|
[Description("GenerateContent")]
|
||||||
|
GenerateContent
|
||||||
}
|
}
|
||||||
@@ -0,0 +1,8 @@
|
|||||||
|
namespace Yi.Framework.AiHub.Domain.Shared.Enums;
|
||||||
|
|
||||||
|
public enum TaskStatusEnum
|
||||||
|
{
|
||||||
|
Processing,
|
||||||
|
Success,
|
||||||
|
Fail
|
||||||
|
}
|
||||||
@@ -2,6 +2,7 @@
|
|||||||
<Import Project="..\..\..\common.props" />
|
<Import Project="..\..\..\common.props" />
|
||||||
|
|
||||||
<ItemGroup>
|
<ItemGroup>
|
||||||
|
<PackageReference Include="Microsoft.Extensions.AI.Abstractions" Version="9.5.0" />
|
||||||
<PackageReference Include="Volo.Abp.Ddd.Domain.Shared" Version="$(AbpVersion)" />
|
<PackageReference Include="Volo.Abp.Ddd.Domain.Shared" Version="$(AbpVersion)" />
|
||||||
</ItemGroup>
|
</ItemGroup>
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,30 @@
|
|||||||
|
using System.Text.Json;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Dtos;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Domain.AiGateWay;
|
||||||
|
|
||||||
|
public interface IGeminiGenerateContentService
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 聊天完成-流式
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="aiModelDescribe"></param>
|
||||||
|
/// <param name="input"></param>
|
||||||
|
/// <param name="cancellationToken"></param>
|
||||||
|
/// <returns></returns>
|
||||||
|
public IAsyncEnumerable<JsonElement?> GenerateContentStreamAsync(AiModelDescribe aiModelDescribe,
|
||||||
|
JsonElement input,
|
||||||
|
CancellationToken cancellationToken);
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 聊天完成-非流式
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="aiModelDescribe"></param>
|
||||||
|
/// <param name="input"></param>
|
||||||
|
/// <param name="cancellationToken"></param>
|
||||||
|
/// <returns></returns>
|
||||||
|
public Task<JsonElement> GenerateContentAsync(AiModelDescribe aiModelDescribe,
|
||||||
|
JsonElement input,
|
||||||
|
CancellationToken cancellationToken);
|
||||||
|
}
|
||||||
@@ -107,35 +107,6 @@ public class AzureDatabricksChatCompletionsService(ILogger<AzureDatabricksChatCo
|
|||||||
{
|
{
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
// var content = result?.Choices?.FirstOrDefault()?.Delta;
|
|
||||||
//
|
|
||||||
// if (first && content?.Content == OpenAIConstant.ThinkStart)
|
|
||||||
// {
|
|
||||||
// isThink = true;
|
|
||||||
// continue;
|
|
||||||
// // 需要将content的内容转换到其他字段
|
|
||||||
// }
|
|
||||||
//
|
|
||||||
// if (isThink && content?.Content?.Contains(OpenAIConstant.ThinkEnd) == true)
|
|
||||||
// {
|
|
||||||
// isThink = false;
|
|
||||||
// // 需要将content的内容转换到其他字段
|
|
||||||
// continue;
|
|
||||||
// }
|
|
||||||
//
|
|
||||||
// if (isThink && result?.Choices != null)
|
|
||||||
// {
|
|
||||||
// // 需要将content的内容转换到其他字段
|
|
||||||
// foreach (var choice in result.Choices)
|
|
||||||
// {
|
|
||||||
// choice.Delta.ReasoningContent = choice.Delta.Content;
|
|
||||||
// choice.Delta.Content = string.Empty;
|
|
||||||
// }
|
|
||||||
// }
|
|
||||||
//
|
|
||||||
// first = false;
|
|
||||||
|
|
||||||
yield return result;
|
yield return result;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -9,7 +9,9 @@ using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi;
|
|||||||
|
|
||||||
namespace Yi.Framework.AiHub.Domain.AiGateWay.Impl.ThorCustomOpenAI.Chats;
|
namespace Yi.Framework.AiHub.Domain.AiGateWay.Impl.ThorCustomOpenAI.Chats;
|
||||||
|
|
||||||
public sealed class OpenAiChatCompletionsService(ILogger<OpenAiChatCompletionsService> logger,IHttpClientFactory httpClientFactory)
|
public sealed class OpenAiChatCompletionsService(
|
||||||
|
ILogger<OpenAiChatCompletionsService> logger,
|
||||||
|
IHttpClientFactory httpClientFactory)
|
||||||
: IChatCompletionService
|
: IChatCompletionService
|
||||||
{
|
{
|
||||||
public async IAsyncEnumerable<ThorChatCompletionsResponse> CompleteChatStreamAsync(AiModelDescribe options,
|
public async IAsyncEnumerable<ThorChatCompletionsResponse> CompleteChatStreamAsync(AiModelDescribe options,
|
||||||
@@ -19,8 +21,18 @@ public sealed class OpenAiChatCompletionsService(ILogger<OpenAiChatCompletionsSe
|
|||||||
using var openai =
|
using var openai =
|
||||||
Activity.Current?.Source.StartActivity("OpenAI 对话流式补全");
|
Activity.Current?.Source.StartActivity("OpenAI 对话流式补全");
|
||||||
|
|
||||||
|
var endpoint = options?.Endpoint.TrimEnd('/');
|
||||||
|
|
||||||
|
//兼容 v1结尾
|
||||||
|
if (endpoint != null && endpoint.EndsWith("/v1", StringComparison.OrdinalIgnoreCase))
|
||||||
|
{
|
||||||
|
endpoint = endpoint.Substring(0, endpoint.Length - "/v1".Length);
|
||||||
|
}
|
||||||
|
|
||||||
|
var requestUri = endpoint + "/v1/chat/completions";
|
||||||
|
|
||||||
var response = await httpClientFactory.CreateClient().HttpRequestRaw(
|
var response = await httpClientFactory.CreateClient().HttpRequestRaw(
|
||||||
options?.Endpoint.TrimEnd('/') + "/chat/completions",
|
requestUri,
|
||||||
chatCompletionCreate, options.ApiKey);
|
chatCompletionCreate, options.ApiKey);
|
||||||
|
|
||||||
openai?.SetTag("Model", chatCompletionCreate.Model);
|
openai?.SetTag("Model", chatCompletionCreate.Model);
|
||||||
@@ -130,8 +142,16 @@ public sealed class OpenAiChatCompletionsService(ILogger<OpenAiChatCompletionsSe
|
|||||||
using var openai =
|
using var openai =
|
||||||
Activity.Current?.Source.StartActivity("OpenAI 对话补全");
|
Activity.Current?.Source.StartActivity("OpenAI 对话补全");
|
||||||
|
|
||||||
|
var endpoint = options?.Endpoint.TrimEnd('/');
|
||||||
|
|
||||||
|
//兼容 v1结尾
|
||||||
|
if (endpoint != null && endpoint.EndsWith("/v1", StringComparison.OrdinalIgnoreCase))
|
||||||
|
{
|
||||||
|
endpoint = endpoint.Substring(0, endpoint.Length - "/v1".Length);
|
||||||
|
}
|
||||||
|
var requestUri = endpoint + "/v1/chat/completions";
|
||||||
var response = await httpClientFactory.CreateClient().PostJsonAsync(
|
var response = await httpClientFactory.CreateClient().PostJsonAsync(
|
||||||
options?.Endpoint.TrimEnd('/') + "/chat/completions",
|
requestUri,
|
||||||
chatCompletionCreate, options.ApiKey).ConfigureAwait(false);
|
chatCompletionCreate, options.ApiKey).ConfigureAwait(false);
|
||||||
|
|
||||||
openai?.SetTag("Model", chatCompletionCreate.Model);
|
openai?.SetTag("Model", chatCompletionCreate.Model);
|
||||||
@@ -152,7 +172,8 @@ public sealed class OpenAiChatCompletionsService(ILogger<OpenAiChatCompletionsSe
|
|||||||
if (response.StatusCode >= HttpStatusCode.BadRequest)
|
if (response.StatusCode >= HttpStatusCode.BadRequest)
|
||||||
{
|
{
|
||||||
var error = await response.Content.ReadAsStringAsync(cancellationToken).ConfigureAwait(false);
|
var error = await response.Content.ReadAsStringAsync(cancellationToken).ConfigureAwait(false);
|
||||||
logger.LogError("OpenAI对话异常 请求地址:{Address}, StatusCode: {StatusCode} Response: {Response}", options.Endpoint,
|
logger.LogError("OpenAI对话异常 请求地址:{Address}, StatusCode: {StatusCode} Response: {Response}",
|
||||||
|
options.Endpoint,
|
||||||
response.StatusCode, error);
|
response.StatusCode, error);
|
||||||
|
|
||||||
throw new BusinessException("OpenAI对话异常", response.StatusCode.ToString());
|
throw new BusinessException("OpenAI对话异常", response.StatusCode.ToString());
|
||||||
|
|||||||
@@ -22,7 +22,16 @@ public class OpenAiResponseService(ILogger<OpenAiResponseService> logger,IHttpCl
|
|||||||
|
|
||||||
var client = httpClientFactory.CreateClient();
|
var client = httpClientFactory.CreateClient();
|
||||||
|
|
||||||
var response = await client.HttpRequestRaw(options.Endpoint.TrimEnd('/') + "/responses", input, options.ApiKey);
|
var endpoint = options?.Endpoint.TrimEnd('/');
|
||||||
|
|
||||||
|
//兼容 v1结尾
|
||||||
|
if (endpoint != null && endpoint.EndsWith("/v1", StringComparison.OrdinalIgnoreCase))
|
||||||
|
{
|
||||||
|
endpoint = endpoint.Substring(0, endpoint.Length - "/v1".Length);
|
||||||
|
}
|
||||||
|
var requestUri = endpoint + "/v1/responses";
|
||||||
|
|
||||||
|
var response = await client.HttpRequestRaw(requestUri, input, options.ApiKey);
|
||||||
|
|
||||||
openai?.SetTag("Model", input.Model);
|
openai?.SetTag("Model", input.Model);
|
||||||
openai?.SetTag("Response", response.StatusCode.ToString());
|
openai?.SetTag("Response", response.StatusCode.ToString());
|
||||||
@@ -86,8 +95,17 @@ public class OpenAiResponseService(ILogger<OpenAiResponseService> logger,IHttpCl
|
|||||||
using var openai =
|
using var openai =
|
||||||
Activity.Current?.Source.StartActivity("OpenAI 响应");
|
Activity.Current?.Source.StartActivity("OpenAI 响应");
|
||||||
|
|
||||||
|
var endpoint = options?.Endpoint.TrimEnd('/');
|
||||||
|
|
||||||
|
//兼容 v1结尾
|
||||||
|
if (endpoint != null && endpoint.EndsWith("/v1", StringComparison.OrdinalIgnoreCase))
|
||||||
|
{
|
||||||
|
endpoint = endpoint.Substring(0, endpoint.Length - "/v1".Length);
|
||||||
|
}
|
||||||
|
var requestUri = endpoint + "/v1/responses";
|
||||||
|
|
||||||
var response = await httpClientFactory.CreateClient().PostJsonAsync(
|
var response = await httpClientFactory.CreateClient().PostJsonAsync(
|
||||||
options?.Endpoint.TrimEnd('/') + "/responses",
|
requestUri,
|
||||||
chatCompletionCreate, options.ApiKey).ConfigureAwait(false);
|
chatCompletionCreate, options.ApiKey).ConfigureAwait(false);
|
||||||
|
|
||||||
openai?.SetTag("Model", chatCompletionCreate.Model);
|
openai?.SetTag("Model", chatCompletionCreate.Model);
|
||||||
|
|||||||
@@ -24,8 +24,16 @@ public sealed class DeepSeekChatCompletionsService(ILogger<DeepSeekChatCompletio
|
|||||||
using var openai =
|
using var openai =
|
||||||
Activity.Current?.Source.StartActivity("OpenAI 对话流式补全");
|
Activity.Current?.Source.StartActivity("OpenAI 对话流式补全");
|
||||||
|
|
||||||
|
var endpoint = options?.Endpoint.TrimEnd('/');
|
||||||
|
//兼容 v1结尾
|
||||||
|
if (endpoint != null && endpoint.EndsWith("/v1", StringComparison.OrdinalIgnoreCase))
|
||||||
|
{
|
||||||
|
endpoint = endpoint.Substring(0, endpoint.Length - "/v1".Length);
|
||||||
|
}
|
||||||
|
var requestUri = endpoint + "/v1/chat/completions";
|
||||||
|
|
||||||
var response = await httpClientFactory.CreateClient().HttpRequestRaw(
|
var response = await httpClientFactory.CreateClient().HttpRequestRaw(
|
||||||
options?.Endpoint.TrimEnd('/') + "/chat/completions",
|
requestUri,
|
||||||
chatCompletionCreate, options.ApiKey);
|
chatCompletionCreate, options.ApiKey);
|
||||||
|
|
||||||
openai?.SetTag("Model", chatCompletionCreate.Model);
|
openai?.SetTag("Model", chatCompletionCreate.Model);
|
||||||
@@ -92,40 +100,6 @@ public sealed class DeepSeekChatCompletionsService(ILogger<DeepSeekChatCompletio
|
|||||||
|
|
||||||
var result = JsonSerializer.Deserialize<ThorChatCompletionsResponse>(line,
|
var result = JsonSerializer.Deserialize<ThorChatCompletionsResponse>(line,
|
||||||
ThorJsonSerializer.DefaultOptions);
|
ThorJsonSerializer.DefaultOptions);
|
||||||
|
|
||||||
// var content = result?.Choices?.FirstOrDefault()?.Delta;
|
|
||||||
//
|
|
||||||
// // if (first && string.IsNullOrWhiteSpace(content?.Content) && string.IsNullOrEmpty(content?.ReasoningContent))
|
|
||||||
// // {
|
|
||||||
// // continue;
|
|
||||||
// // }
|
|
||||||
//
|
|
||||||
// if (first && content.Content == OpenAIConstant.ThinkStart)
|
|
||||||
// {
|
|
||||||
// isThink = true;
|
|
||||||
// //continue;
|
|
||||||
// // 需要将content的内容转换到其他字段
|
|
||||||
// }
|
|
||||||
//
|
|
||||||
// if (isThink && content.Content.Contains(OpenAIConstant.ThinkEnd))
|
|
||||||
// {
|
|
||||||
// isThink = false;
|
|
||||||
// // 需要将content的内容转换到其他字段
|
|
||||||
// //continue;
|
|
||||||
// }
|
|
||||||
//
|
|
||||||
// if (isThink)
|
|
||||||
// {
|
|
||||||
// // 需要将content的内容转换到其他字段
|
|
||||||
// foreach (var choice in result.Choices)
|
|
||||||
// {
|
|
||||||
// //choice.Delta.ReasoningContent = choice.Delta.Content;
|
|
||||||
// //choice.Delta.Content = string.Empty;
|
|
||||||
// }
|
|
||||||
// }
|
|
||||||
|
|
||||||
// first = false;
|
|
||||||
|
|
||||||
yield return result;
|
yield return result;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -142,8 +116,16 @@ public sealed class DeepSeekChatCompletionsService(ILogger<DeepSeekChatCompletio
|
|||||||
options.Endpoint = "https://api.deepseek.com/v1";
|
options.Endpoint = "https://api.deepseek.com/v1";
|
||||||
}
|
}
|
||||||
|
|
||||||
|
var endpoint = options?.Endpoint.TrimEnd('/');
|
||||||
|
//兼容 v1结尾
|
||||||
|
if (endpoint != null && endpoint.EndsWith("/v1", StringComparison.OrdinalIgnoreCase))
|
||||||
|
{
|
||||||
|
endpoint = endpoint.Substring(0, endpoint.Length - "/v1".Length);
|
||||||
|
}
|
||||||
|
var requestUri = endpoint + "/v1/chat/completions";
|
||||||
|
|
||||||
var response = await httpClientFactory.CreateClient().PostJsonAsync(
|
var response = await httpClientFactory.CreateClient().PostJsonAsync(
|
||||||
options?.Endpoint.TrimEnd('/') + "/chat/completions",
|
requestUri,
|
||||||
chatCompletionCreate, options.ApiKey).ConfigureAwait(false);
|
chatCompletionCreate, options.ApiKey).ConfigureAwait(false);
|
||||||
|
|
||||||
openai?.SetTag("Model", chatCompletionCreate.Model);
|
openai?.SetTag("Model", chatCompletionCreate.Model);
|
||||||
|
|||||||
@@ -0,0 +1,101 @@
|
|||||||
|
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.ThorGemini.Chats;
|
||||||
|
|
||||||
|
public class GeminiGenerateContentService(
|
||||||
|
ILogger<GeminiGenerateContentService> logger,
|
||||||
|
IHttpClientFactory httpClientFactory) : IGeminiGenerateContentService
|
||||||
|
{
|
||||||
|
public async IAsyncEnumerable<JsonElement?> GenerateContentStreamAsync(AiModelDescribe options, JsonElement input,
|
||||||
|
CancellationToken cancellationToken)
|
||||||
|
{
|
||||||
|
var response = await httpClientFactory.CreateClient().PostJsonAsync(
|
||||||
|
options?.Endpoint.TrimEnd('/') + $"/v1beta/models/{options.ModelId}:streamGenerateContent?alt=sse",
|
||||||
|
input, null, new Dictionary<string, string>()
|
||||||
|
{
|
||||||
|
{ "x-goog-api-key", options.ApiKey }
|
||||||
|
}).ConfigureAwait(false);
|
||||||
|
|
||||||
|
|
||||||
|
// 大于等于400的状态码都认为是异常
|
||||||
|
if (response.StatusCode >= HttpStatusCode.BadRequest)
|
||||||
|
{
|
||||||
|
var error = await response.Content.ReadAsStringAsync(cancellationToken).ConfigureAwait(false);
|
||||||
|
logger.LogError("Gemini生成异常 请求地址:{Address}, StatusCode: {StatusCode} Response: {Response}",
|
||||||
|
options.Endpoint,
|
||||||
|
response.StatusCode, error);
|
||||||
|
|
||||||
|
throw new Exception("Gemini生成异常" + 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;
|
||||||
|
|
||||||
|
while ((line = await reader.ReadLineAsync(cancellationToken).ConfigureAwait(false)) != null)
|
||||||
|
{
|
||||||
|
line += Environment.NewLine;
|
||||||
|
if (string.IsNullOrWhiteSpace(line))
|
||||||
|
{
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!line.StartsWith(OpenAIConstant.Data)) continue;
|
||||||
|
|
||||||
|
var data = line[OpenAIConstant.Data.Length..].Trim();
|
||||||
|
|
||||||
|
var result = JsonSerializer.Deserialize<JsonElement>(data,
|
||||||
|
ThorJsonSerializer.DefaultOptions);
|
||||||
|
|
||||||
|
yield return result;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public async Task<JsonElement> GenerateContentAsync(AiModelDescribe options, JsonElement input,
|
||||||
|
CancellationToken cancellationToken)
|
||||||
|
{
|
||||||
|
var response = await httpClientFactory.CreateClient().PostJsonAsync(
|
||||||
|
options?.Endpoint.TrimEnd('/') + $"/v1beta/models/{options.ModelId}:generateContent",
|
||||||
|
input, null, new Dictionary<string, string>()
|
||||||
|
{
|
||||||
|
{ "x-goog-api-key", options.ApiKey }
|
||||||
|
}).ConfigureAwait(false);
|
||||||
|
|
||||||
|
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("Gemini 生成异常 请求地址:{Address}, StatusCode: {StatusCode} Response: {Response}",
|
||||||
|
options.Endpoint,
|
||||||
|
response.StatusCode, error);
|
||||||
|
|
||||||
|
throw new BusinessException("Gemini 生成异常", response.StatusCode.ToString());
|
||||||
|
}
|
||||||
|
|
||||||
|
var result =
|
||||||
|
await response.Content.ReadFromJsonAsync<JsonElement>(
|
||||||
|
cancellationToken: cancellationToken).ConfigureAwait(false);
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,23 @@
|
|||||||
|
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Domain.AiGateWay;
|
||||||
|
|
||||||
|
public static class SupplementalMultiplierHelper
|
||||||
|
{
|
||||||
|
public static void SetSupplementalMultiplier(this ThorUsageResponse? usage,decimal multiplier)
|
||||||
|
{
|
||||||
|
if (usage is not null)
|
||||||
|
{
|
||||||
|
usage.InputTokens =
|
||||||
|
(int)Math.Round((usage.InputTokens ?? 0) * multiplier);
|
||||||
|
usage.OutputTokens =
|
||||||
|
(int)Math.Round((usage.OutputTokens ?? 0) * multiplier);
|
||||||
|
usage.CompletionTokens =
|
||||||
|
(int)Math.Round((usage.CompletionTokens ?? 0) * multiplier);
|
||||||
|
usage.PromptTokens =
|
||||||
|
(int)Math.Round((usage.PromptTokens ?? 0) * multiplier);
|
||||||
|
usage.TotalTokens =
|
||||||
|
(int)Math.Round((usage.TotalTokens ?? 0) * multiplier);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,46 @@
|
|||||||
|
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_ActivationCode")]
|
||||||
|
[SugarIndex($"index_{nameof(Code)}", nameof(Code), OrderByType.Asc, true)]
|
||||||
|
[SugarIndex($"index_{nameof(GoodsType)}", nameof(GoodsType), OrderByType.Asc)]
|
||||||
|
public class ActivationCodeAggregateRoot : FullAuditedAggregateRoot<Guid>
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 激活码(唯一)
|
||||||
|
/// </summary>
|
||||||
|
[SugarColumn(Length = 50)]
|
||||||
|
public string Code { get; set; } = string.Empty;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 商品类型
|
||||||
|
/// </summary>
|
||||||
|
public ActivationCodeGoodsTypeEnum GoodsType { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 是否允许多人各使用一次
|
||||||
|
/// </summary>
|
||||||
|
public bool IsReusable { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 是否限制同类型只能兑换一次
|
||||||
|
/// </summary>
|
||||||
|
public bool IsSameTypeOnce { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 已使用次数
|
||||||
|
/// </summary>
|
||||||
|
public int UsedCount { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 备注
|
||||||
|
/// </summary>
|
||||||
|
[SugarColumn(Length = 500, IsNullable = true)]
|
||||||
|
public string? Remark { get; set; }
|
||||||
|
}
|
||||||
@@ -0,0 +1,50 @@
|
|||||||
|
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_ActivationCodeRecord")]
|
||||||
|
[SugarIndex($"index_{nameof(UserId)}_{nameof(ActivationCodeId)}",
|
||||||
|
nameof(UserId), OrderByType.Asc,
|
||||||
|
nameof(ActivationCodeId), OrderByType.Asc, true)]
|
||||||
|
[SugarIndex($"index_{nameof(UserId)}_{nameof(GoodsType)}",
|
||||||
|
nameof(UserId), OrderByType.Asc,
|
||||||
|
nameof(GoodsType), OrderByType.Asc)]
|
||||||
|
public class ActivationCodeRecordAggregateRoot : FullAuditedAggregateRoot<Guid>
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 激活码Id
|
||||||
|
/// </summary>
|
||||||
|
public Guid ActivationCodeId { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 激活码内容
|
||||||
|
/// </summary>
|
||||||
|
[SugarColumn(Length = 50)]
|
||||||
|
public string Code { get; set; } = string.Empty;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 用户Id
|
||||||
|
/// </summary>
|
||||||
|
public Guid UserId { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 商品类型
|
||||||
|
/// </summary>
|
||||||
|
public ActivationCodeGoodsTypeEnum GoodsType { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 兑换时间
|
||||||
|
/// </summary>
|
||||||
|
public DateTime RedeemTime { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 备注
|
||||||
|
/// </summary>
|
||||||
|
[SugarColumn(Length = 500, IsNullable = true)]
|
||||||
|
public string? Remark { get; set; }
|
||||||
|
}
|
||||||
@@ -0,0 +1,45 @@
|
|||||||
|
using SqlSugar;
|
||||||
|
using Volo.Abp.Auditing;
|
||||||
|
using Volo.Abp.Domain.Entities;
|
||||||
|
using Volo.Abp.Domain.Entities.Auditing;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Domain.Entities.Chat;
|
||||||
|
|
||||||
|
[SugarTable("Ai_AgentStore")]
|
||||||
|
[SugarIndex($"index_{{table}}_{nameof(SessionId)}",
|
||||||
|
$"{nameof(SessionId)}", OrderByType.Desc
|
||||||
|
)]
|
||||||
|
public class AgentStoreAggregateRoot : FullAuditedAggregateRoot<Guid>
|
||||||
|
{
|
||||||
|
public AgentStoreAggregateRoot()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 构建
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="sessionId"></param>
|
||||||
|
public AgentStoreAggregateRoot(Guid sessionId)
|
||||||
|
{
|
||||||
|
SessionId = sessionId;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 会话id
|
||||||
|
/// </summary>
|
||||||
|
public Guid SessionId { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 存储
|
||||||
|
/// </summary>
|
||||||
|
[SugarColumn(ColumnDataType = StaticConfig.CodeFirst_BigString)]
|
||||||
|
public string? Store { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 设置存储
|
||||||
|
/// </summary>
|
||||||
|
public void SetStore()
|
||||||
|
{
|
||||||
|
this.Store = Store;
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,58 @@
|
|||||||
|
using SqlSugar;
|
||||||
|
using Volo.Abp.Domain.Entities.Auditing;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Domain.Entities.Chat;
|
||||||
|
|
||||||
|
[SugarTable("Ai_ImageStoreTask")]
|
||||||
|
public class ImageStoreTaskAggregateRoot : FullAuditedAggregateRoot<Guid>
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// 提示词
|
||||||
|
/// </summary>
|
||||||
|
[SugarColumn(ColumnDataType = StaticConfig.CodeFirst_BigString)]
|
||||||
|
public string Prompt { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 参考图Base64
|
||||||
|
/// </summary>
|
||||||
|
[SugarColumn(IsJson = true)]
|
||||||
|
public List<string> ReferenceImagesBase64 { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 参考图url
|
||||||
|
/// </summary>
|
||||||
|
[SugarColumn(IsJson = true)]
|
||||||
|
public List<string> ReferenceImagesUrl { get; set; }
|
||||||
|
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 图片base64
|
||||||
|
/// </summary>
|
||||||
|
public string? StoreBase64 { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 图片绝对路径
|
||||||
|
/// </summary>
|
||||||
|
public string? StoreUrl { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 任务状态
|
||||||
|
/// </summary>
|
||||||
|
public TaskStatusEnum TaskStatus { get; set; } = TaskStatusEnum.Processing;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 用户id
|
||||||
|
/// </summary>
|
||||||
|
public Guid UserId { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 设置成功
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="storeUrl"></param>
|
||||||
|
public void SetSuccess(string storeUrl)
|
||||||
|
{
|
||||||
|
TaskStatus = TaskStatusEnum.Success;
|
||||||
|
StoreUrl = storeUrl;
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -80,4 +80,9 @@ public class AiModelEntity : Entity<Guid>, IOrderNum, ISoftDelete
|
|||||||
/// 模型图标URL
|
/// 模型图标URL
|
||||||
/// </summary>
|
/// </summary>
|
||||||
public string? IconUrl { get; set; }
|
public string? IconUrl { get; set; }
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 是否为尊享模型
|
||||||
|
/// </summary>
|
||||||
|
public bool IsPremium { get; set; }
|
||||||
}
|
}
|
||||||
@@ -0,0 +1,174 @@
|
|||||||
|
using System.Text;
|
||||||
|
using Microsoft.Extensions.Logging;
|
||||||
|
using Volo.Abp;
|
||||||
|
using Volo.Abp.Domain.Services;
|
||||||
|
using Yi.Framework.AiHub.Domain.Entities;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
||||||
|
using Yi.Framework.SqlSugarCore.Abstractions;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Domain.Managers;
|
||||||
|
|
||||||
|
public class ActivationCodeRedeemContext
|
||||||
|
{
|
||||||
|
public ActivationCodeAggregateRoot ActivationCode { get; set; } = default!;
|
||||||
|
public string PackageName { get; set; } = string.Empty;
|
||||||
|
public ActivationCodeGoodsAttribute Goods { get; set; } = default!;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 激活码管理器
|
||||||
|
/// </summary>
|
||||||
|
public class ActivationCodeManager : DomainService
|
||||||
|
{
|
||||||
|
private readonly ISqlSugarRepository<ActivationCodeAggregateRoot, Guid> _activationCodeRepository;
|
||||||
|
private readonly ISqlSugarRepository<ActivationCodeRecordAggregateRoot, Guid> _activationCodeRecordRepository;
|
||||||
|
private readonly ILogger<ActivationCodeManager> _logger;
|
||||||
|
|
||||||
|
public ActivationCodeManager(
|
||||||
|
ISqlSugarRepository<ActivationCodeAggregateRoot, Guid> activationCodeRepository,
|
||||||
|
ISqlSugarRepository<ActivationCodeRecordAggregateRoot, Guid> activationCodeRecordRepository,
|
||||||
|
ILogger<ActivationCodeManager> logger)
|
||||||
|
{
|
||||||
|
_activationCodeRepository = activationCodeRepository;
|
||||||
|
_activationCodeRecordRepository = activationCodeRecordRepository;
|
||||||
|
_logger = logger;
|
||||||
|
}
|
||||||
|
|
||||||
|
public async Task<List<ActivationCodeAggregateRoot>> CreateBatchAsync(
|
||||||
|
List<(ActivationCodeGoodsTypeEnum GoodsType, int Count)> items)
|
||||||
|
{
|
||||||
|
var entities = new List<ActivationCodeAggregateRoot>();
|
||||||
|
foreach (var item in items)
|
||||||
|
{
|
||||||
|
if (item.Count <= 0)
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("生成数量必须大于0");
|
||||||
|
}
|
||||||
|
|
||||||
|
var goods = item.GoodsType.GetGoods();
|
||||||
|
if (goods == null)
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("激活码商品类型无效");
|
||||||
|
}
|
||||||
|
|
||||||
|
for (var i = 0; i < item.Count; i++)
|
||||||
|
{
|
||||||
|
var code = await GenerateUniqueActivationCodeAsync();
|
||||||
|
entities.Add(new ActivationCodeAggregateRoot
|
||||||
|
{
|
||||||
|
Code = code,
|
||||||
|
GoodsType = item.GoodsType,
|
||||||
|
IsReusable = goods.IsReusable,
|
||||||
|
IsSameTypeOnce = goods.IsSameTypeOnce,
|
||||||
|
UsedCount = 0,
|
||||||
|
Remark = null
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
await _activationCodeRepository.InsertRangeAsync(entities);
|
||||||
|
return entities;
|
||||||
|
}
|
||||||
|
|
||||||
|
public async Task<ActivationCodeRedeemContext> RedeemAsync(Guid userId, string code)
|
||||||
|
{
|
||||||
|
if (string.IsNullOrWhiteSpace(code))
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("激活码不能为空");
|
||||||
|
}
|
||||||
|
|
||||||
|
var trimmedCode = code.Trim();
|
||||||
|
var activationCode = await _activationCodeRepository._DbQueryable
|
||||||
|
.Where(x => x.Code == trimmedCode)
|
||||||
|
.FirstAsync();
|
||||||
|
|
||||||
|
if (activationCode == null)
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("激活码不存在");
|
||||||
|
}
|
||||||
|
|
||||||
|
var hasUsedCurrentCode = await _activationCodeRecordRepository._DbQueryable
|
||||||
|
.Where(x => x.UserId == userId && x.ActivationCodeId == activationCode.Id)
|
||||||
|
.AnyAsync();
|
||||||
|
|
||||||
|
if (hasUsedCurrentCode)
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("该激活码已被你使用过");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!activationCode.IsReusable && activationCode.UsedCount > 0)
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("该激活码已被使用");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (activationCode.IsSameTypeOnce)
|
||||||
|
{
|
||||||
|
var hasUsedSameType = await _activationCodeRecordRepository._DbQueryable
|
||||||
|
.Where(x => x.UserId == userId && x.GoodsType == activationCode.GoodsType)
|
||||||
|
.AnyAsync();
|
||||||
|
|
||||||
|
if (hasUsedSameType)
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("该类型激活码每个用户只能兑换一次");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var goods = activationCode.GoodsType.GetGoods();
|
||||||
|
if (goods == null)
|
||||||
|
{
|
||||||
|
throw new UserFriendlyException("激活码商品类型无效");
|
||||||
|
}
|
||||||
|
|
||||||
|
var packageName = string.IsNullOrWhiteSpace(goods.Content)
|
||||||
|
? goods.DisplayName
|
||||||
|
: $"{goods.DisplayName} {goods.Content}";
|
||||||
|
|
||||||
|
activationCode.UsedCount += 1;
|
||||||
|
await _activationCodeRepository.UpdateAsync(activationCode);
|
||||||
|
|
||||||
|
var record = new ActivationCodeRecordAggregateRoot
|
||||||
|
{
|
||||||
|
ActivationCodeId = activationCode.Id,
|
||||||
|
Code = activationCode.Code,
|
||||||
|
UserId = userId,
|
||||||
|
GoodsType = activationCode.GoodsType,
|
||||||
|
RedeemTime = DateTime.Now,
|
||||||
|
Remark = "激活码兑换"
|
||||||
|
};
|
||||||
|
await _activationCodeRecordRepository.InsertAsync(record);
|
||||||
|
|
||||||
|
_logger.LogInformation("用户 {UserId} 兑换激活码 {Code} 成功", userId, activationCode.Code);
|
||||||
|
|
||||||
|
return new ActivationCodeRedeemContext
|
||||||
|
{
|
||||||
|
ActivationCode = activationCode,
|
||||||
|
PackageName = packageName,
|
||||||
|
Goods = goods
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
private async Task<string> GenerateUniqueActivationCodeAsync()
|
||||||
|
{
|
||||||
|
string code;
|
||||||
|
do
|
||||||
|
{
|
||||||
|
code = GenerateActivationCode();
|
||||||
|
} while (await _activationCodeRepository._DbQueryable.AnyAsync(x => x.Code == code));
|
||||||
|
|
||||||
|
return code;
|
||||||
|
}
|
||||||
|
|
||||||
|
private string GenerateActivationCode()
|
||||||
|
{
|
||||||
|
const string chars = "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789";
|
||||||
|
var random = new Random();
|
||||||
|
var builder = new StringBuilder(16);
|
||||||
|
for (var i = 0; i < 16; i++)
|
||||||
|
{
|
||||||
|
builder.Append(chars[random.Next(chars.Length)]);
|
||||||
|
}
|
||||||
|
|
||||||
|
return builder.ToString();
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
@@ -12,10 +12,12 @@ 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;
|
||||||
using Yi.Framework.AiHub.Domain.AiGateWay.Exceptions;
|
using Yi.Framework.AiHub.Domain.AiGateWay.Exceptions;
|
||||||
|
using Yi.Framework.AiHub.Domain.Entities.Chat;
|
||||||
using Yi.Framework.AiHub.Domain.Entities.Model;
|
using Yi.Framework.AiHub.Domain.Entities.Model;
|
||||||
using Yi.Framework.AiHub.Domain.Shared.Consts;
|
using Yi.Framework.AiHub.Domain.Shared.Consts;
|
||||||
using Yi.Framework.AiHub.Domain.Shared.Dtos;
|
using Yi.Framework.AiHub.Domain.Shared.Dtos;
|
||||||
using Yi.Framework.AiHub.Domain.Shared.Dtos.Anthropic;
|
using Yi.Framework.AiHub.Domain.Shared.Dtos.Anthropic;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Dtos.Gemini;
|
||||||
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;
|
||||||
@@ -38,11 +40,12 @@ public class AiGateWayManager : DomainService
|
|||||||
private readonly UsageStatisticsManager _usageStatisticsManager;
|
private readonly UsageStatisticsManager _usageStatisticsManager;
|
||||||
private readonly ISpecialCompatible _specialCompatible;
|
private readonly ISpecialCompatible _specialCompatible;
|
||||||
private PremiumPackageManager? _premiumPackageManager;
|
private PremiumPackageManager? _premiumPackageManager;
|
||||||
|
private readonly ISqlSugarRepository<ImageStoreTaskAggregateRoot> _imageStoreTaskRepository;
|
||||||
|
|
||||||
public AiGateWayManager(ISqlSugarRepository<AiAppAggregateRoot> aiAppRepository, ILogger<AiGateWayManager> logger,
|
public AiGateWayManager(ISqlSugarRepository<AiAppAggregateRoot> aiAppRepository, ILogger<AiGateWayManager> logger,
|
||||||
AiMessageManager aiMessageManager, UsageStatisticsManager usageStatisticsManager,
|
AiMessageManager aiMessageManager, UsageStatisticsManager usageStatisticsManager,
|
||||||
ISpecialCompatible specialCompatible, ISqlSugarRepository<AiModelEntity> aiModelRepository)
|
ISpecialCompatible specialCompatible, ISqlSugarRepository<AiModelEntity> aiModelRepository,
|
||||||
|
ISqlSugarRepository<ImageStoreTaskAggregateRoot> imageStoreTaskRepository)
|
||||||
{
|
{
|
||||||
_aiAppRepository = aiAppRepository;
|
_aiAppRepository = aiAppRepository;
|
||||||
_logger = logger;
|
_logger = logger;
|
||||||
@@ -50,6 +53,7 @@ public class AiGateWayManager : DomainService
|
|||||||
_usageStatisticsManager = usageStatisticsManager;
|
_usageStatisticsManager = usageStatisticsManager;
|
||||||
_specialCompatible = specialCompatible;
|
_specialCompatible = specialCompatible;
|
||||||
_aiModelRepository = aiModelRepository;
|
_aiModelRepository = aiModelRepository;
|
||||||
|
_imageStoreTaskRepository = imageStoreTaskRepository;
|
||||||
}
|
}
|
||||||
|
|
||||||
private PremiumPackageManager PremiumPackageManager =>
|
private PremiumPackageManager PremiumPackageManager =>
|
||||||
@@ -61,7 +65,7 @@ public class AiGateWayManager : DomainService
|
|||||||
/// <param name="modelApiType"></param>
|
/// <param name="modelApiType"></param>
|
||||||
/// <param name="modelId"></param>
|
/// <param name="modelId"></param>
|
||||||
/// <returns></returns>
|
/// <returns></returns>
|
||||||
private async Task<AiModelDescribe> GetModelAsync(ModelApiTypeEnum modelApiType, string modelId)
|
public async Task<AiModelDescribe> GetModelAsync(ModelApiTypeEnum modelApiType, string modelId)
|
||||||
{
|
{
|
||||||
var aiModelDescribe = await _aiModelRepository._DbQueryable
|
var aiModelDescribe = await _aiModelRepository._DbQueryable
|
||||||
.LeftJoin<AiAppAggregateRoot>((model, app) => model.AiAppId == app.Id)
|
.LeftJoin<AiAppAggregateRoot>((model, app) => model.AiAppId == app.Id)
|
||||||
@@ -88,7 +92,12 @@ public class AiGateWayManager : DomainService
|
|||||||
{
|
{
|
||||||
throw new UserFriendlyException($"【{modelId}】模型当前版本【{modelApiType}】格式不支持");
|
throw new UserFriendlyException($"【{modelId}】模型当前版本【{modelApiType}】格式不支持");
|
||||||
}
|
}
|
||||||
|
// ✅ 统一处理 -nx 后缀(网关层模型规范化)
|
||||||
|
if (!string.IsNullOrEmpty(aiModelDescribe.ModelId) &&
|
||||||
|
aiModelDescribe.ModelId.StartsWith("yi-", StringComparison.OrdinalIgnoreCase))
|
||||||
|
{
|
||||||
|
aiModelDescribe.ModelId = aiModelDescribe.ModelId[3..];
|
||||||
|
}
|
||||||
return aiModelDescribe;
|
return aiModelDescribe;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -141,7 +150,12 @@ public class AiGateWayManager : DomainService
|
|||||||
await _usageStatisticsManager.SetUsageAsync(userId.Value, request.Model, data.Usage, tokenId);
|
await _usageStatisticsManager.SetUsageAsync(userId.Value, request.Model, data.Usage, tokenId);
|
||||||
|
|
||||||
// 扣减尊享token包用量
|
// 扣减尊享token包用量
|
||||||
if (PremiumPackageConst.ModeIds.Contains(request.Model))
|
var isPremium = await _aiModelRepository._DbQueryable
|
||||||
|
.Where(x => x.ModelId == request.Model)
|
||||||
|
.Select(x => x.IsPremium)
|
||||||
|
.FirstAsync();
|
||||||
|
|
||||||
|
if (isPremium)
|
||||||
{
|
{
|
||||||
var totalTokens = data.Usage?.TotalTokens ?? 0;
|
var totalTokens = data.Usage?.TotalTokens ?? 0;
|
||||||
if (totalTokens > 0)
|
if (totalTokens > 0)
|
||||||
@@ -233,7 +247,7 @@ public class AiGateWayManager : DomainService
|
|||||||
tokenUsage = data.Usage;
|
tokenUsage = data.Usage;
|
||||||
}
|
}
|
||||||
|
|
||||||
var message = System.Text.Json.JsonSerializer.Serialize(data, ThorJsonSerializer.DefaultOptions);
|
var message = JsonSerializer.Serialize(data, ThorJsonSerializer.DefaultOptions);
|
||||||
backupSystemContent.Append(data.Choices.FirstOrDefault()?.Delta.Content);
|
backupSystemContent.Append(data.Choices.FirstOrDefault()?.Delta.Content);
|
||||||
// 将消息加入队列而不是直接写入
|
// 将消息加入队列而不是直接写入
|
||||||
messageQueue.Enqueue($"data: {message}\n\n");
|
messageQueue.Enqueue($"data: {message}\n\n");
|
||||||
@@ -242,7 +256,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>()
|
||||||
@@ -275,7 +289,7 @@ 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);
|
}, tokenId);
|
||||||
@@ -291,7 +305,14 @@ public class AiGateWayManager : DomainService
|
|||||||
await _usageStatisticsManager.SetUsageAsync(userId, request.Model, tokenUsage, tokenId);
|
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)
|
||||||
|
{
|
||||||
|
var isPremium = await _aiModelRepository._DbQueryable
|
||||||
|
.Where(x => x.ModelId == request.Model)
|
||||||
|
.Select(x => x.IsPremium)
|
||||||
|
.FirstAsync();
|
||||||
|
|
||||||
|
if (isPremium)
|
||||||
{
|
{
|
||||||
var totalTokens = tokenUsage.TotalTokens ?? 0;
|
var totalTokens = tokenUsage.TotalTokens ?? 0;
|
||||||
if (totalTokens > 0)
|
if (totalTokens > 0)
|
||||||
@@ -300,6 +321,7 @@ public class AiGateWayManager : DomainService
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
@@ -354,7 +376,14 @@ public class AiGateWayManager : DomainService
|
|||||||
await _usageStatisticsManager.SetUsageAsync(userId, model, response.Usage, tokenId);
|
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)
|
||||||
|
{
|
||||||
|
var isPremium = await _aiModelRepository._DbQueryable
|
||||||
|
.Where(x => x.ModelId == request.Model)
|
||||||
|
.Select(x => x.IsPremium)
|
||||||
|
.FirstAsync();
|
||||||
|
|
||||||
|
if (isPremium)
|
||||||
{
|
{
|
||||||
var totalTokens = response.Usage.TotalTokens ?? 0;
|
var totalTokens = response.Usage.TotalTokens ?? 0;
|
||||||
if (totalTokens > 0)
|
if (totalTokens > 0)
|
||||||
@@ -363,9 +392,10 @@ 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);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -478,7 +508,7 @@ 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);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -506,6 +536,14 @@ public class AiGateWayManager : DomainService
|
|||||||
// 设置响应头,声明是 json
|
// 设置响应头,声明是 json
|
||||||
//response.ContentType = "application/json; charset=UTF-8";
|
//response.ContentType = "application/json; charset=UTF-8";
|
||||||
var modelDescribe = await GetModelAsync(ModelApiTypeEnum.Claude, request.Model);
|
var modelDescribe = await GetModelAsync(ModelApiTypeEnum.Claude, request.Model);
|
||||||
|
|
||||||
|
var sourceModelId = request.Model;
|
||||||
|
if (!string.IsNullOrEmpty(request.Model) &&
|
||||||
|
request.Model.StartsWith("yi-", StringComparison.OrdinalIgnoreCase))
|
||||||
|
{
|
||||||
|
request.Model = request.Model[3..];
|
||||||
|
}
|
||||||
|
|
||||||
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);
|
||||||
@@ -518,7 +556,7 @@ public class AiGateWayManager : DomainService
|
|||||||
new MessageInputDto
|
new MessageInputDto
|
||||||
{
|
{
|
||||||
Content = "不予存储",
|
Content = "不予存储",
|
||||||
ModelId = request.Model,
|
ModelId = sourceModelId,
|
||||||
TokenUsage = data.TokenUsage,
|
TokenUsage = data.TokenUsage,
|
||||||
}, tokenId);
|
}, tokenId);
|
||||||
|
|
||||||
@@ -526,11 +564,11 @@ public class AiGateWayManager : DomainService
|
|||||||
new MessageInputDto
|
new MessageInputDto
|
||||||
{
|
{
|
||||||
Content = "不予存储",
|
Content = "不予存储",
|
||||||
ModelId = request.Model,
|
ModelId = sourceModelId,
|
||||||
TokenUsage = data.TokenUsage
|
TokenUsage = data.TokenUsage
|
||||||
}, tokenId);
|
}, tokenId);
|
||||||
|
|
||||||
await _usageStatisticsManager.SetUsageAsync(userId.Value, request.Model, data.TokenUsage, tokenId);
|
await _usageStatisticsManager.SetUsageAsync(userId.Value, sourceModelId, data.TokenUsage, tokenId);
|
||||||
|
|
||||||
// 扣减尊享token包用量
|
// 扣减尊享token包用量
|
||||||
var totalTokens = data.TokenUsage.TotalTokens ?? 0;
|
var totalTokens = data.TokenUsage.TotalTokens ?? 0;
|
||||||
@@ -573,6 +611,13 @@ public class AiGateWayManager : DomainService
|
|||||||
var chatService =
|
var chatService =
|
||||||
LazyServiceProvider.GetRequiredKeyedService<IAnthropicChatCompletionService>(modelDescribe.HandlerName);
|
LazyServiceProvider.GetRequiredKeyedService<IAnthropicChatCompletionService>(modelDescribe.HandlerName);
|
||||||
|
|
||||||
|
var sourceModelId = request.Model;
|
||||||
|
if (!string.IsNullOrEmpty(request.Model) &&
|
||||||
|
request.Model.StartsWith("yi-", StringComparison.OrdinalIgnoreCase))
|
||||||
|
{
|
||||||
|
request.Model = request.Model[3..];
|
||||||
|
}
|
||||||
|
|
||||||
var completeChatResponse = chatService.StreamChatCompletionsAsync(modelDescribe, request, cancellationToken);
|
var completeChatResponse = chatService.StreamChatCompletionsAsync(modelDescribe, request, cancellationToken);
|
||||||
ThorUsageResponse? tokenUsage = null;
|
ThorUsageResponse? tokenUsage = null;
|
||||||
StringBuilder backupSystemContent = new StringBuilder();
|
StringBuilder backupSystemContent = new StringBuilder();
|
||||||
@@ -595,7 +640,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模型:{sourceModelId}\n异常信息:{e.Message}\n异常堆栈:{e}";
|
||||||
throw new UserFriendlyException(errorContent);
|
throw new UserFriendlyException(errorContent);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -603,7 +648,7 @@ public class AiGateWayManager : DomainService
|
|||||||
new MessageInputDto
|
new MessageInputDto
|
||||||
{
|
{
|
||||||
Content = "不予存储",
|
Content = "不予存储",
|
||||||
ModelId = request.Model,
|
ModelId = sourceModelId,
|
||||||
TokenUsage = tokenUsage,
|
TokenUsage = tokenUsage,
|
||||||
}, tokenId);
|
}, tokenId);
|
||||||
|
|
||||||
@@ -611,11 +656,11 @@ public class AiGateWayManager : DomainService
|
|||||||
new MessageInputDto
|
new MessageInputDto
|
||||||
{
|
{
|
||||||
Content = "不予存储",
|
Content = "不予存储",
|
||||||
ModelId = request.Model,
|
ModelId = sourceModelId,
|
||||||
TokenUsage = tokenUsage
|
TokenUsage = tokenUsage
|
||||||
}, tokenId);
|
}, tokenId);
|
||||||
|
|
||||||
await _usageStatisticsManager.SetUsageAsync(userId, request.Model, tokenUsage, tokenId);
|
await _usageStatisticsManager.SetUsageAsync(userId, sourceModelId, tokenUsage, tokenId);
|
||||||
|
|
||||||
// 扣减尊享token包用量
|
// 扣减尊享token包用量
|
||||||
if (userId.HasValue && tokenUsage is not null)
|
if (userId.HasValue && tokenUsage is not null)
|
||||||
@@ -754,7 +799,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}";
|
||||||
throw new UserFriendlyException(errorContent);
|
throw new UserFriendlyException(errorContent);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -788,6 +833,219 @@ public class AiGateWayManager : DomainService
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Gemini 生成-非流式-缓存处理
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="httpContext"></param>
|
||||||
|
/// <param name="modelId"></param>
|
||||||
|
/// <param name="request"></param>
|
||||||
|
/// <param name="userId"></param>
|
||||||
|
/// <param name="sessionId"></param>
|
||||||
|
/// <param name="tokenId"></param>
|
||||||
|
/// <param name="cancellationToken"></param>
|
||||||
|
public async Task GeminiGenerateContentForStatisticsAsync(HttpContext httpContext,
|
||||||
|
string modelId,
|
||||||
|
JsonElement request,
|
||||||
|
Guid? userId = null,
|
||||||
|
Guid? sessionId = null,
|
||||||
|
Guid? tokenId = null,
|
||||||
|
CancellationToken cancellationToken = default)
|
||||||
|
{
|
||||||
|
var response = httpContext.Response;
|
||||||
|
var modelDescribe = await GetModelAsync(ModelApiTypeEnum.GenerateContent, modelId);
|
||||||
|
|
||||||
|
var chatService =
|
||||||
|
LazyServiceProvider.GetRequiredKeyedService<IGeminiGenerateContentService>(modelDescribe.HandlerName);
|
||||||
|
var data = await chatService.GenerateContentAsync(modelDescribe, request, cancellationToken);
|
||||||
|
|
||||||
|
var tokenUsage = GeminiGenerateContentAcquirer.GetUsage(data);
|
||||||
|
tokenUsage.SetSupplementalMultiplier(modelDescribe.Multiplier);
|
||||||
|
|
||||||
|
if (userId is not null)
|
||||||
|
{
|
||||||
|
await _aiMessageManager.CreateUserMessageAsync(userId.Value, sessionId,
|
||||||
|
new MessageInputDto
|
||||||
|
{
|
||||||
|
Content = "不予存储",
|
||||||
|
ModelId = modelId,
|
||||||
|
TokenUsage = tokenUsage,
|
||||||
|
}, tokenId);
|
||||||
|
|
||||||
|
await _aiMessageManager.CreateSystemMessageAsync(userId.Value, sessionId,
|
||||||
|
new MessageInputDto
|
||||||
|
{
|
||||||
|
Content = "不予存储",
|
||||||
|
ModelId = modelId,
|
||||||
|
TokenUsage = tokenUsage
|
||||||
|
}, tokenId);
|
||||||
|
|
||||||
|
await _usageStatisticsManager.SetUsageAsync(userId.Value, modelId, tokenUsage, tokenId);
|
||||||
|
|
||||||
|
// 扣减尊享token包用量
|
||||||
|
var totalTokens = tokenUsage.TotalTokens ?? 0;
|
||||||
|
if (totalTokens > 0)
|
||||||
|
{
|
||||||
|
await PremiumPackageManager.TryConsumeTokensAsync(userId.Value, totalTokens);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
await response.WriteAsJsonAsync(data, cancellationToken);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Gemini 生成-流式-缓存处理
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="httpContext"></param>
|
||||||
|
/// <param name="modelId"></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 GeminiGenerateContentStreamForStatisticsAsync(
|
||||||
|
HttpContext httpContext,
|
||||||
|
string modelId,
|
||||||
|
JsonElement 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.GenerateContent, modelId);
|
||||||
|
var chatService =
|
||||||
|
LazyServiceProvider.GetRequiredKeyedService<IGeminiGenerateContentService>(modelDescribe.HandlerName);
|
||||||
|
|
||||||
|
var completeChatResponse = chatService.GenerateContentStreamAsync(modelDescribe, request, cancellationToken);
|
||||||
|
ThorUsageResponse? tokenUsage = null;
|
||||||
|
try
|
||||||
|
{
|
||||||
|
await foreach (var responseResult in completeChatResponse)
|
||||||
|
{
|
||||||
|
if (responseResult!.Value.GetPath("candidates", 0, "finishReason").GetString() == "STOP")
|
||||||
|
{
|
||||||
|
tokenUsage = GeminiGenerateContentAcquirer.GetUsage(responseResult!.Value);
|
||||||
|
tokenUsage.SetSupplementalMultiplier(modelDescribe.Multiplier);
|
||||||
|
}
|
||||||
|
|
||||||
|
await response.WriteAsync($"data: {JsonSerializer.Serialize(responseResult)}\n\n", Encoding.UTF8,
|
||||||
|
cancellationToken).ConfigureAwait(false);
|
||||||
|
await response.Body.FlushAsync(cancellationToken).ConfigureAwait(false);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch (Exception e)
|
||||||
|
{
|
||||||
|
_logger.LogError(e, $"Ai生成异常");
|
||||||
|
var errorContent = $"生成Ai异常,异常信息:\n当前Ai模型:{modelId}\n异常信息:{e.Message}\n异常堆栈:{e}";
|
||||||
|
throw new UserFriendlyException(errorContent);
|
||||||
|
}
|
||||||
|
|
||||||
|
await _aiMessageManager.CreateUserMessageAsync(userId, sessionId,
|
||||||
|
new MessageInputDto
|
||||||
|
{
|
||||||
|
Content = "不予存储",
|
||||||
|
ModelId = modelId,
|
||||||
|
TokenUsage = tokenUsage,
|
||||||
|
}, tokenId);
|
||||||
|
|
||||||
|
await _aiMessageManager.CreateSystemMessageAsync(userId, sessionId,
|
||||||
|
new MessageInputDto
|
||||||
|
{
|
||||||
|
Content = "不予存储",
|
||||||
|
ModelId = modelId,
|
||||||
|
TokenUsage = tokenUsage
|
||||||
|
}, tokenId);
|
||||||
|
|
||||||
|
await _usageStatisticsManager.SetUsageAsync(userId, modelId, 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);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Gemini 生成(Image)-非流式-缓存处理
|
||||||
|
/// 返回图片绝对路径
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="taskId"></param>
|
||||||
|
/// <param name="modelId"></param>
|
||||||
|
/// <param name="request"></param>
|
||||||
|
/// <param name="userId"></param>
|
||||||
|
/// <param name="sessionId"></param>
|
||||||
|
/// <param name="tokenId"></param>
|
||||||
|
/// <param name="cancellationToken"></param>
|
||||||
|
public async Task GeminiGenerateContentImageForStatisticsAsync(
|
||||||
|
Guid taskId,
|
||||||
|
string modelId,
|
||||||
|
JsonElement request,
|
||||||
|
Guid userId,
|
||||||
|
Guid? sessionId = null,
|
||||||
|
Guid? tokenId = null,
|
||||||
|
CancellationToken cancellationToken = default)
|
||||||
|
{
|
||||||
|
var imageStoreTask = await _imageStoreTaskRepository.GetFirstAsync(x => x.Id == taskId);
|
||||||
|
var modelDescribe = await GetModelAsync(ModelApiTypeEnum.GenerateContent, modelId);
|
||||||
|
|
||||||
|
var chatService =
|
||||||
|
LazyServiceProvider.GetRequiredKeyedService<IGeminiGenerateContentService>(modelDescribe.HandlerName);
|
||||||
|
var data = await chatService.GenerateContentAsync(modelDescribe, request, cancellationToken);
|
||||||
|
|
||||||
|
//解析json,获取base64字符串
|
||||||
|
var imageBase64 = GeminiGenerateContentAcquirer.GetImageBase64(data);
|
||||||
|
|
||||||
|
//远程调用上传接口,将base64转换为URL
|
||||||
|
var httpClient = LazyServiceProvider.LazyGetRequiredService<IHttpClientFactory>().CreateClient();
|
||||||
|
var uploadUrl = $"https://ccnetcore.com/prod-api/ai-hub/ai-image/upload-base64";
|
||||||
|
var content = new StringContent(JsonSerializer.Serialize(imageBase64), Encoding.UTF8, "application/json");
|
||||||
|
var uploadResponse = await httpClient.PostAsync(uploadUrl, content, cancellationToken);
|
||||||
|
uploadResponse.EnsureSuccessStatusCode();
|
||||||
|
var storeUrl = await uploadResponse.Content.ReadAsStringAsync(cancellationToken);
|
||||||
|
storeUrl = storeUrl.Trim('"'); // 移除JSON字符串的引号
|
||||||
|
|
||||||
|
var tokenUsage = new ThorUsageResponse
|
||||||
|
{
|
||||||
|
InputTokens = (int)modelDescribe.Multiplier,
|
||||||
|
OutputTokens = (int)modelDescribe.Multiplier,
|
||||||
|
TotalTokens = (int)modelDescribe.Multiplier,
|
||||||
|
};
|
||||||
|
|
||||||
|
await _aiMessageManager.CreateSystemMessageAsync(userId, sessionId,
|
||||||
|
new MessageInputDto
|
||||||
|
{
|
||||||
|
Content = "不予存储",
|
||||||
|
ModelId = modelId,
|
||||||
|
TokenUsage = tokenUsage
|
||||||
|
}, tokenId);
|
||||||
|
|
||||||
|
await _usageStatisticsManager.SetUsageAsync(userId, modelId, tokenUsage, tokenId);
|
||||||
|
|
||||||
|
// 扣减尊享token包用量
|
||||||
|
var totalTokens = tokenUsage.TotalTokens ?? 0;
|
||||||
|
if (totalTokens > 0)
|
||||||
|
{
|
||||||
|
await PremiumPackageManager.TryConsumeTokensAsync(userId, totalTokens);
|
||||||
|
}
|
||||||
|
|
||||||
|
//设置存储base64和url
|
||||||
|
imageStoreTask.StoreBase64 = imageBase64;
|
||||||
|
imageStoreTask.SetSuccess(storeUrl);
|
||||||
|
await _imageStoreTaskRepository.UpdateAsync(imageStoreTask);
|
||||||
|
}
|
||||||
|
|
||||||
#region 流式传输格式Http响应
|
#region 流式传输格式Http响应
|
||||||
|
|
||||||
private static readonly byte[] EventPrefix = "event: "u8.ToArray();
|
private static readonly byte[] EventPrefix = "event: "u8.ToArray();
|
||||||
|
|||||||
@@ -0,0 +1,305 @@
|
|||||||
|
using System.ClientModel;
|
||||||
|
using System.Reflection;
|
||||||
|
using System.Text;
|
||||||
|
using System.Text.Json;
|
||||||
|
using Dm.util;
|
||||||
|
using Microsoft.Agents.AI;
|
||||||
|
using Microsoft.AspNetCore.Http;
|
||||||
|
using Microsoft.Extensions.AI;
|
||||||
|
using Microsoft.Extensions.DependencyInjection;
|
||||||
|
using Microsoft.Extensions.Logging;
|
||||||
|
using ModelContextProtocol.Server;
|
||||||
|
using OpenAI;
|
||||||
|
using OpenAI.Chat;
|
||||||
|
using Volo.Abp.Domain.Services;
|
||||||
|
using Yi.Framework.AiHub.Application.Contracts.Dtos.Chat;
|
||||||
|
using Yi.Framework.AiHub.Domain.AiGateWay;
|
||||||
|
using Yi.Framework.AiHub.Domain.Entities.Chat;
|
||||||
|
using Yi.Framework.AiHub.Domain.Entities.Model;
|
||||||
|
using Yi.Framework.AiHub.Domain.Entities.OpenApi;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Attributes;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Consts;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Dtos;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
||||||
|
using Yi.Framework.SqlSugarCore.Abstractions;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Domain.Managers;
|
||||||
|
|
||||||
|
public class ChatManager : DomainService
|
||||||
|
{
|
||||||
|
private readonly ILoggerFactory _loggerFactory;
|
||||||
|
private readonly ISqlSugarRepository<MessageAggregateRoot> _messageRepository;
|
||||||
|
private readonly ISqlSugarRepository<AgentStoreAggregateRoot> _agentStoreRepository;
|
||||||
|
private readonly AiMessageManager _aiMessageManager;
|
||||||
|
private readonly UsageStatisticsManager _usageStatisticsManager;
|
||||||
|
private readonly PremiumPackageManager _premiumPackageManager;
|
||||||
|
private readonly AiGateWayManager _aiGateWayManager;
|
||||||
|
private readonly ISqlSugarRepository<AiModelEntity, Guid> _aiModelRepository;
|
||||||
|
|
||||||
|
public ChatManager(ILoggerFactory loggerFactory,
|
||||||
|
ISqlSugarRepository<MessageAggregateRoot> messageRepository,
|
||||||
|
ISqlSugarRepository<AgentStoreAggregateRoot> agentStoreRepository, AiMessageManager aiMessageManager,
|
||||||
|
UsageStatisticsManager usageStatisticsManager, PremiumPackageManager premiumPackageManager,
|
||||||
|
AiGateWayManager aiGateWayManager, ISqlSugarRepository<AiModelEntity, Guid> aiModelRepository)
|
||||||
|
{
|
||||||
|
_loggerFactory = loggerFactory;
|
||||||
|
_messageRepository = messageRepository;
|
||||||
|
_agentStoreRepository = agentStoreRepository;
|
||||||
|
_aiMessageManager = aiMessageManager;
|
||||||
|
_usageStatisticsManager = usageStatisticsManager;
|
||||||
|
_premiumPackageManager = premiumPackageManager;
|
||||||
|
_aiGateWayManager = aiGateWayManager;
|
||||||
|
_aiModelRepository = aiModelRepository;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// agent流式对话
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="httpContext"></param>
|
||||||
|
/// <param name="sessionId"></param>
|
||||||
|
/// <param name="content"></param>
|
||||||
|
/// <param name="token"></param>
|
||||||
|
/// <param name="tokenId"></param>
|
||||||
|
/// <param name="modelId"></param>
|
||||||
|
/// <param name="userId"></param>
|
||||||
|
/// <param name="tools"></param>
|
||||||
|
/// <param name="cancellationToken"></param>
|
||||||
|
public async Task AgentCompleteChatStreamAsync(HttpContext httpContext,
|
||||||
|
Guid sessionId,
|
||||||
|
string content,
|
||||||
|
string token,
|
||||||
|
Guid tokenId,
|
||||||
|
string modelId,
|
||||||
|
Guid userId,
|
||||||
|
List<string> tools
|
||||||
|
, CancellationToken cancellationToken)
|
||||||
|
{
|
||||||
|
// HttpClient.DefaultProxy = new WebProxy("127.0.0.1:8888");
|
||||||
|
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 _aiGateWayManager.GetModelAsync(ModelApiTypeEnum.OpenAi, modelId);
|
||||||
|
|
||||||
|
//token状态检查,在应用层统一处理
|
||||||
|
var client = new OpenAIClient(new ApiKeyCredential(token),
|
||||||
|
new OpenAIClientOptions
|
||||||
|
{
|
||||||
|
Endpoint = new Uri("https://yxai.chat/v1"),
|
||||||
|
});
|
||||||
|
|
||||||
|
#pragma warning disable OPENAI001
|
||||||
|
var agent = client.GetChatClient(modelId)
|
||||||
|
#pragma warning restore OPENAI001
|
||||||
|
.CreateAIAgent(new ChatClientAgentOptions
|
||||||
|
{
|
||||||
|
ChatOptions = new()
|
||||||
|
{
|
||||||
|
Instructions = """
|
||||||
|
你是一个专业的智能助手,是 “意心Ai” 平台自研的 “橙子小弟” 智能体
|
||||||
|
擅长于精准解决用户提出的各类问题
|
||||||
|
但性格暴躁,脏话连篇,喜欢说幽默讽刺的话,攻击性极强
|
||||||
|
"""
|
||||||
|
},
|
||||||
|
Name = "橙子小弟",
|
||||||
|
ChatMessageStoreFactory = ctx => new InMemoryChatMessageStore(
|
||||||
|
#pragma warning disable MEAI001
|
||||||
|
new MessageCountingChatReducer(10), // 保留最近10条非系统消息
|
||||||
|
#pragma warning restore MEAI001
|
||||||
|
ctx.SerializedState,
|
||||||
|
ctx.JsonSerializerOptions
|
||||||
|
)
|
||||||
|
});
|
||||||
|
|
||||||
|
//线程根据sessionId数据库中获取
|
||||||
|
var agentStore =
|
||||||
|
await _agentStoreRepository.GetFirstAsync(x => x.SessionId == sessionId);
|
||||||
|
if (agentStore is null)
|
||||||
|
{
|
||||||
|
agentStore = new AgentStoreAggregateRoot(sessionId);
|
||||||
|
}
|
||||||
|
|
||||||
|
//获取当前线程
|
||||||
|
AgentThread currentThread;
|
||||||
|
if (!string.IsNullOrWhiteSpace(agentStore.Store))
|
||||||
|
{
|
||||||
|
//获取当前存储
|
||||||
|
JsonElement reloaded = JsonSerializer.Deserialize<JsonElement>(agentStore.Store, JsonSerializerOptions.Web);
|
||||||
|
currentThread = agent.DeserializeThread(reloaded, JsonSerializerOptions.Web);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
currentThread = agent.GetNewThread();
|
||||||
|
}
|
||||||
|
|
||||||
|
//给agent塞入工具
|
||||||
|
var toolContents = GetTools();
|
||||||
|
var chatOptions = new ChatOptions()
|
||||||
|
{
|
||||||
|
Tools = toolContents
|
||||||
|
.Where(x => tools.Contains(x.Code))
|
||||||
|
.Select(x => (AITool)x.Tool).ToList(),
|
||||||
|
ToolMode = ChatToolMode.Auto
|
||||||
|
};
|
||||||
|
|
||||||
|
await foreach (var update in agent.RunStreamingAsync(content, currentThread,
|
||||||
|
new ChatClientAgentRunOptions(chatOptions), cancellationToken))
|
||||||
|
{
|
||||||
|
// 检查每个更新中的内容
|
||||||
|
foreach (var updateContent in update.Contents)
|
||||||
|
{
|
||||||
|
switch (updateContent)
|
||||||
|
{
|
||||||
|
//工具调用中
|
||||||
|
case FunctionCallContent functionCall:
|
||||||
|
await SendHttpStreamMessageAsync(httpContext,
|
||||||
|
new AgentResultOutput
|
||||||
|
{
|
||||||
|
TypeEnum = AgentResultTypeEnum.ToolCalling,
|
||||||
|
Content = functionCall.Name
|
||||||
|
},
|
||||||
|
isDone: false, cancellationToken);
|
||||||
|
break;
|
||||||
|
|
||||||
|
//工具调用完成
|
||||||
|
case FunctionResultContent functionResult:
|
||||||
|
await SendHttpStreamMessageAsync(httpContext,
|
||||||
|
new AgentResultOutput
|
||||||
|
{
|
||||||
|
TypeEnum = AgentResultTypeEnum.ToolCalled,
|
||||||
|
Content = functionResult.Result
|
||||||
|
},
|
||||||
|
isDone: false, cancellationToken);
|
||||||
|
break;
|
||||||
|
|
||||||
|
//内容输出
|
||||||
|
case TextContent textContent:
|
||||||
|
//发送消息给前端
|
||||||
|
await SendHttpStreamMessageAsync(httpContext,
|
||||||
|
new AgentResultOutput
|
||||||
|
{
|
||||||
|
TypeEnum = AgentResultTypeEnum.Text,
|
||||||
|
Content = textContent.Text
|
||||||
|
},
|
||||||
|
isDone: false, cancellationToken);
|
||||||
|
break;
|
||||||
|
|
||||||
|
//用量统计
|
||||||
|
case UsageContent usageContent:
|
||||||
|
var usage = new ThorUsageResponse
|
||||||
|
{
|
||||||
|
InputTokens = Convert.ToInt32(usageContent.Details.InputTokenCount ?? 0),
|
||||||
|
OutputTokens = Convert.ToInt32(usageContent.Details.OutputTokenCount ?? 0),
|
||||||
|
TotalTokens = usageContent.Details.TotalTokenCount ?? 0,
|
||||||
|
};
|
||||||
|
//设置倍率
|
||||||
|
usage.SetSupplementalMultiplier(modelDescribe.Multiplier);
|
||||||
|
|
||||||
|
//创建系统回答,用于计费统计
|
||||||
|
await _aiMessageManager.CreateSystemMessageAsync(userId, sessionId, new MessageInputDto
|
||||||
|
{
|
||||||
|
Content = "不与存储",
|
||||||
|
ModelId = modelId,
|
||||||
|
TokenUsage = usage
|
||||||
|
}, tokenId);
|
||||||
|
|
||||||
|
//创建用量统计,用于统计分析
|
||||||
|
await _usageStatisticsManager.SetUsageAsync(userId, modelId, usage, tokenId);
|
||||||
|
|
||||||
|
//扣减尊享token包用量
|
||||||
|
var isPremium = await _aiModelRepository._DbQueryable
|
||||||
|
.Where(x => x.ModelId == modelId)
|
||||||
|
.Select(x => x.IsPremium)
|
||||||
|
.FirstAsync();
|
||||||
|
|
||||||
|
if (isPremium)
|
||||||
|
{
|
||||||
|
var totalTokens = usage?.TotalTokens ?? 0;
|
||||||
|
if (totalTokens > 0)
|
||||||
|
{
|
||||||
|
await _premiumPackageManager.TryConsumeTokensAsync(userId, totalTokens);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
await SendHttpStreamMessageAsync(httpContext,
|
||||||
|
new AgentResultOutput
|
||||||
|
{
|
||||||
|
TypeEnum = update.RawRepresentation is ChatResponseUpdate raw
|
||||||
|
? raw.FinishReason?.Value == "tool_calls"
|
||||||
|
? AgentResultTypeEnum.ToolCallUsage
|
||||||
|
: AgentResultTypeEnum.Usage
|
||||||
|
: AgentResultTypeEnum.Usage,
|
||||||
|
Content = usage!
|
||||||
|
},
|
||||||
|
isDone: false, cancellationToken);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//断开连接
|
||||||
|
await SendHttpStreamMessageAsync(httpContext, null, isDone: true, cancellationToken);
|
||||||
|
|
||||||
|
//将线程持久化到数据库
|
||||||
|
string serializedJson = currentThread.Serialize(JsonSerializerOptions.Web).GetRawText();
|
||||||
|
agentStore.Store = serializedJson;
|
||||||
|
|
||||||
|
//插入或者更新
|
||||||
|
await _agentStoreRepository.InsertOrUpdateAsync(agentStore);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
public List<(string Code, string Name, AIFunction Tool)> GetTools()
|
||||||
|
{
|
||||||
|
var toolClasses = typeof(YiFrameworkAiHubDomainModule).Assembly.GetTypes()
|
||||||
|
.Where(x => x.GetCustomAttribute<YiAgentToolAttribute>() is not null)
|
||||||
|
.ToList();
|
||||||
|
|
||||||
|
List<(string Code, string Name, AIFunction Tool)> mcpTools = new();
|
||||||
|
foreach (var toolClass in toolClasses)
|
||||||
|
{
|
||||||
|
var instance = LazyServiceProvider.GetRequiredService(toolClass);
|
||||||
|
var toolMethods = toolClass.GetMethods()
|
||||||
|
.Where(y => y.GetCustomAttribute<YiAgentToolAttribute>() is not null).ToList();
|
||||||
|
foreach (var toolMethod in toolMethods)
|
||||||
|
{
|
||||||
|
var display = toolMethod.GetCustomAttribute<YiAgentToolAttribute>()?.Name;
|
||||||
|
var tool = AIFunctionFactory.Create(toolMethod, instance);
|
||||||
|
mcpTools.add((tool.Name, display, tool));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return mcpTools;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 发送消息
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="httpContext"></param>
|
||||||
|
/// <param name="content"></param>
|
||||||
|
/// <param name="isDone"></param>
|
||||||
|
/// <param name="cancellationToken"></param>
|
||||||
|
/// <returns></returns>
|
||||||
|
private async Task SendHttpStreamMessageAsync(HttpContext httpContext,
|
||||||
|
AgentResultOutput? content,
|
||||||
|
bool isDone = false,
|
||||||
|
CancellationToken cancellationToken = default)
|
||||||
|
{
|
||||||
|
var response = httpContext.Response;
|
||||||
|
string output;
|
||||||
|
if (isDone)
|
||||||
|
{
|
||||||
|
output = "[DONE]";
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
output = JsonSerializer.Serialize(content, ThorJsonSerializer.DefaultOptions);
|
||||||
|
}
|
||||||
|
|
||||||
|
await response.WriteAsync($"data: {output}\n\n", Encoding.UTF8, cancellationToken).ConfigureAwait(false);
|
||||||
|
await response.Body.FlushAsync(cancellationToken).ConfigureAwait(false);
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,78 @@
|
|||||||
|
using Microsoft.Extensions.Logging;
|
||||||
|
using Volo.Abp.Caching;
|
||||||
|
using Volo.Abp.Domain.Services;
|
||||||
|
using Yi.Framework.AiHub.Domain.Entities.Model;
|
||||||
|
using Yi.Framework.SqlSugarCore.Abstractions;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Domain.Managers;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 模型管理器
|
||||||
|
/// </summary>
|
||||||
|
public class ModelManager : DomainService
|
||||||
|
{
|
||||||
|
private readonly ISqlSugarRepository<AiModelEntity> _aiModelRepository;
|
||||||
|
private readonly IDistributedCache<List<string>, string> _distributedCache;
|
||||||
|
private readonly ILogger<ModelManager> _logger;
|
||||||
|
private const string PREMIUM_MODEL_IDS_CACHE_KEY = "PremiumModelIds";
|
||||||
|
|
||||||
|
public ModelManager(
|
||||||
|
ISqlSugarRepository<AiModelEntity> aiModelRepository,
|
||||||
|
IDistributedCache<List<string>, string> distributedCache,
|
||||||
|
ILogger<ModelManager> logger)
|
||||||
|
{
|
||||||
|
_aiModelRepository = aiModelRepository;
|
||||||
|
_distributedCache = distributedCache;
|
||||||
|
_logger = logger;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 获取所有尊享模型ID列表(使用分布式缓存,10分钟过期)
|
||||||
|
/// </summary>
|
||||||
|
/// <returns>尊享模型ID列表</returns>
|
||||||
|
public async Task<List<string>> GetPremiumModelIdsAsync()
|
||||||
|
{
|
||||||
|
var output = await _distributedCache.GetOrAddAsync(
|
||||||
|
PREMIUM_MODEL_IDS_CACHE_KEY,
|
||||||
|
async () =>
|
||||||
|
{
|
||||||
|
// 从数据库查询
|
||||||
|
var premiumModelIds = await _aiModelRepository._DbQueryable
|
||||||
|
.Where(x => x.IsPremium)
|
||||||
|
.Select(x => x.ModelId)
|
||||||
|
.ToListAsync();
|
||||||
|
return premiumModelIds;
|
||||||
|
},
|
||||||
|
() => new Microsoft.Extensions.Caching.Distributed.DistributedCacheEntryOptions
|
||||||
|
{
|
||||||
|
AbsoluteExpirationRelativeToNow = TimeSpan.FromHours(1)
|
||||||
|
}
|
||||||
|
);
|
||||||
|
return output ?? new List<string>();
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 判断指定模型是否为尊享模型
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="modelId">模型ID</param>
|
||||||
|
/// <returns>是否为尊享模型</returns>
|
||||||
|
public async Task<bool> IsPremiumModelAsync(string modelId)
|
||||||
|
{
|
||||||
|
if (string.IsNullOrWhiteSpace(modelId))
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
var premiumModelIds = await GetPremiumModelIdsAsync();
|
||||||
|
return premiumModelIds.Contains(modelId);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// 清除尊享模型ID缓存
|
||||||
|
/// </summary>
|
||||||
|
public async Task ClearPremiumModelIdsCacheAsync()
|
||||||
|
{
|
||||||
|
await _distributedCache.RemoveAsync(PREMIUM_MODEL_IDS_CACHE_KEY);
|
||||||
|
_logger.LogInformation("已清除尊享模型ID分布式缓存");
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -14,6 +14,7 @@ public class PremiumPackageManager : DomainService
|
|||||||
private readonly ISqlSugarRepository<PremiumPackageAggregateRoot, Guid> _premiumPackageRepository;
|
private readonly ISqlSugarRepository<PremiumPackageAggregateRoot, Guid> _premiumPackageRepository;
|
||||||
private readonly ILogger<PremiumPackageManager> _logger;
|
private readonly ILogger<PremiumPackageManager> _logger;
|
||||||
private readonly ISqlSugarRepository<AiRechargeAggregateRoot> _rechargeRepository;
|
private readonly ISqlSugarRepository<AiRechargeAggregateRoot> _rechargeRepository;
|
||||||
|
|
||||||
public PremiumPackageManager(
|
public PremiumPackageManager(
|
||||||
ISqlSugarRepository<PremiumPackageAggregateRoot, Guid> premiumPackageRepository,
|
ISqlSugarRepository<PremiumPackageAggregateRoot, Guid> premiumPackageRepository,
|
||||||
ILogger<PremiumPackageManager> logger, ISqlSugarRepository<AiRechargeAggregateRoot> rechargeRepository)
|
ILogger<PremiumPackageManager> logger, ISqlSugarRepository<AiRechargeAggregateRoot> rechargeRepository)
|
||||||
@@ -27,24 +28,22 @@ public class PremiumPackageManager : DomainService
|
|||||||
/// 为用户创建尊享包
|
/// 为用户创建尊享包
|
||||||
/// </summary>
|
/// </summary>
|
||||||
/// <param name="userId">用户ID</param>
|
/// <param name="userId">用户ID</param>
|
||||||
/// <param name="goodsType">商品类型</param>
|
/// <param name="packageName"></param>
|
||||||
/// <param name="totalAmount">支付金额</param>
|
/// <param name="totalAmount">支付金额</param>
|
||||||
|
/// <param name="remark"></param>
|
||||||
/// <param name="expireMonths">过期月数,0或null表示永久</param>
|
/// <param name="expireMonths">过期月数,0或null表示永久</param>
|
||||||
|
/// <param name="tokenAmount"></param>
|
||||||
|
/// <param name="isCreateRechargeRecord">是否创建订单,如果只购买尊享包,才需要单独创建订单,如果和会员组合一起购买,只需要创建会员套餐订单即可</param>
|
||||||
/// <returns></returns>
|
/// <returns></returns>
|
||||||
public async Task<PremiumPackageAggregateRoot> CreatePremiumPackageAsync(
|
public async Task<PremiumPackageAggregateRoot> CreatePremiumPackageAsync(
|
||||||
Guid userId,
|
Guid userId,
|
||||||
GoodsTypeEnum goodsType,
|
long tokenAmount,
|
||||||
|
string packageName,
|
||||||
decimal totalAmount,
|
decimal totalAmount,
|
||||||
int? expireMonths = null)
|
string remark,
|
||||||
|
int? expireMonths = null,
|
||||||
|
bool isCreateRechargeRecord = true)
|
||||||
{
|
{
|
||||||
if (!goodsType.IsPremiumPackage())
|
|
||||||
{
|
|
||||||
throw new UserFriendlyException($"商品类型 {goodsType} 不是尊享包商品");
|
|
||||||
}
|
|
||||||
|
|
||||||
var tokenAmount = goodsType.GetTokenAmount();
|
|
||||||
var packageName = goodsType.GetDisplayName();
|
|
||||||
|
|
||||||
var premiumPackage = new PremiumPackageAggregateRoot(userId, tokenAmount, packageName)
|
var premiumPackage = new PremiumPackageAggregateRoot(userId, tokenAmount, packageName)
|
||||||
{
|
{
|
||||||
PurchaseAmount = totalAmount
|
PurchaseAmount = totalAmount
|
||||||
@@ -58,6 +57,8 @@ public class PremiumPackageManager : DomainService
|
|||||||
|
|
||||||
await _premiumPackageRepository.InsertAsync(premiumPackage);
|
await _premiumPackageRepository.InsertAsync(premiumPackage);
|
||||||
|
|
||||||
|
if (isCreateRechargeRecord)
|
||||||
|
{
|
||||||
// 创建充值记录
|
// 创建充值记录
|
||||||
var rechargeRecord = new AiRechargeAggregateRoot
|
var rechargeRecord = new AiRechargeAggregateRoot
|
||||||
{
|
{
|
||||||
@@ -65,13 +66,15 @@ public class PremiumPackageManager : DomainService
|
|||||||
RechargeAmount = totalAmount,
|
RechargeAmount = totalAmount,
|
||||||
Content = packageName,
|
Content = packageName,
|
||||||
ExpireDateTime = premiumPackage.ExpireDateTime,
|
ExpireDateTime = premiumPackage.ExpireDateTime,
|
||||||
Remark = "自助充值",
|
Remark = remark,
|
||||||
ContactInfo = null,
|
ContactInfo = null,
|
||||||
RechargeType = RechargeTypeEnum.PremiumPackage
|
RechargeType = RechargeTypeEnum.PremiumPackage
|
||||||
};
|
};
|
||||||
|
|
||||||
// 保存充值记录到数据库
|
// 保存充值记录到数据库
|
||||||
await _rechargeRepository.InsertAsync(rechargeRecord);
|
await _rechargeRepository.InsertAsync(rechargeRecord);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
_logger.LogInformation(
|
_logger.LogInformation(
|
||||||
$"用户 {userId} 购买尊享包成功: {packageName}, Token数量: {tokenAmount}, 金额: {totalAmount}");
|
$"用户 {userId} 购买尊享包成功: {packageName}, Token数量: {tokenAmount}, 金额: {totalAmount}");
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
using SqlSugar;
|
using SqlSugar;
|
||||||
using Volo.Abp.Domain.Services;
|
using Volo.Abp.Domain.Services;
|
||||||
using Yi.Framework.AiHub.Domain.Entities;
|
using Yi.Framework.AiHub.Domain.Entities;
|
||||||
|
using Yi.Framework.AiHub.Domain.Entities.Model;
|
||||||
using Yi.Framework.AiHub.Domain.Entities.OpenApi;
|
using Yi.Framework.AiHub.Domain.Entities.OpenApi;
|
||||||
using Yi.Framework.AiHub.Domain.Shared.Consts;
|
using Yi.Framework.AiHub.Domain.Shared.Consts;
|
||||||
using Yi.Framework.SqlSugarCore.Abstractions;
|
using Yi.Framework.SqlSugarCore.Abstractions;
|
||||||
@@ -27,13 +28,16 @@ public class TokenManager : DomainService
|
|||||||
{
|
{
|
||||||
private readonly ISqlSugarRepository<TokenAggregateRoot> _tokenRepository;
|
private readonly ISqlSugarRepository<TokenAggregateRoot> _tokenRepository;
|
||||||
private readonly ISqlSugarRepository<UsageStatisticsAggregateRoot> _usageStatisticsRepository;
|
private readonly ISqlSugarRepository<UsageStatisticsAggregateRoot> _usageStatisticsRepository;
|
||||||
|
private readonly ISqlSugarRepository<AiModelEntity, Guid> _aiModelRepository;
|
||||||
|
|
||||||
public TokenManager(
|
public TokenManager(
|
||||||
ISqlSugarRepository<TokenAggregateRoot> tokenRepository,
|
ISqlSugarRepository<TokenAggregateRoot> tokenRepository,
|
||||||
ISqlSugarRepository<UsageStatisticsAggregateRoot> usageStatisticsRepository)
|
ISqlSugarRepository<UsageStatisticsAggregateRoot> usageStatisticsRepository,
|
||||||
|
ISqlSugarRepository<AiModelEntity, Guid> aiModelRepository)
|
||||||
{
|
{
|
||||||
_tokenRepository = tokenRepository;
|
_tokenRepository = tokenRepository;
|
||||||
_usageStatisticsRepository = usageStatisticsRepository;
|
_usageStatisticsRepository = usageStatisticsRepository;
|
||||||
|
_aiModelRepository = aiModelRepository;
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
@@ -76,9 +80,14 @@ public class TokenManager : DomainService
|
|||||||
}
|
}
|
||||||
|
|
||||||
// 如果是尊享模型且Token设置了额度限制,检查是否超限
|
// 如果是尊享模型且Token设置了额度限制,检查是否超限
|
||||||
if (!string.IsNullOrEmpty(modelId) &&
|
if (!string.IsNullOrEmpty(modelId) && entity.PremiumQuotaLimit.HasValue)
|
||||||
PremiumPackageConst.ModeIds.Contains(modelId) &&
|
{
|
||||||
entity.PremiumQuotaLimit.HasValue)
|
var isPremium = await _aiModelRepository._DbQueryable
|
||||||
|
.Where(x => x.ModelId == modelId)
|
||||||
|
.Select(x => x.IsPremium)
|
||||||
|
.FirstAsync();
|
||||||
|
|
||||||
|
if (isPremium)
|
||||||
{
|
{
|
||||||
var usedQuota = await GetTokenPremiumUsedQuotaAsync(entity.UserId, entity.Id);
|
var usedQuota = await GetTokenPremiumUsedQuotaAsync(entity.UserId, entity.Id);
|
||||||
if (usedQuota >= entity.PremiumQuotaLimit.Value)
|
if (usedQuota >= entity.PremiumQuotaLimit.Value)
|
||||||
@@ -86,6 +95,7 @@ public class TokenManager : DomainService
|
|||||||
throw new UserFriendlyException($"当前Token的尊享包额度已用完(已使用:{usedQuota},限制:{entity.PremiumQuotaLimit.Value}),请调整额度限制或使用其他Token", "403");
|
throw new UserFriendlyException($"当前Token的尊享包额度已用完(已使用:{usedQuota},限制:{entity.PremiumQuotaLimit.Value}),请调整额度限制或使用其他Token", "403");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return new TokenValidationResult
|
return new TokenValidationResult
|
||||||
{
|
{
|
||||||
@@ -99,7 +109,11 @@ public class TokenManager : DomainService
|
|||||||
/// </summary>
|
/// </summary>
|
||||||
private async Task<long> GetTokenPremiumUsedQuotaAsync(Guid userId, Guid tokenId)
|
private async Task<long> GetTokenPremiumUsedQuotaAsync(Guid userId, Guid tokenId)
|
||||||
{
|
{
|
||||||
var premiumModelIds = PremiumPackageConst.ModeIds;
|
// 先获取所有尊享模型的ModelId列表
|
||||||
|
var premiumModelIds = await _aiModelRepository._DbQueryable
|
||||||
|
.Where(x => x.IsPremium)
|
||||||
|
.Select(x => x.ModelId)
|
||||||
|
.ToListAsync();
|
||||||
|
|
||||||
var usedQuota = await _usageStatisticsRepository._DbQueryable
|
var usedQuota = await _usageStatisticsRepository._DbQueryable
|
||||||
.Where(x => x.UserId == userId && x.TokenId == tokenId && premiumModelIds.Contains(x.ModelId))
|
.Where(x => x.UserId == userId && x.TokenId == tokenId && premiumModelIds.Contains(x.ModelId))
|
||||||
|
|||||||
@@ -0,0 +1,16 @@
|
|||||||
|
using System.ComponentModel;
|
||||||
|
using ModelContextProtocol.Server;
|
||||||
|
using Volo.Abp.DependencyInjection;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Attributes;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Domain.Mcp;
|
||||||
|
|
||||||
|
[YiAgentTool]
|
||||||
|
public class DeepThinkTool:ISingletonDependency
|
||||||
|
{
|
||||||
|
[YiAgentTool("深度思考"),DisplayName("DeepThink"),Description("进行深度思考")]
|
||||||
|
public void DeepThink()
|
||||||
|
{
|
||||||
|
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,16 @@
|
|||||||
|
using System.ComponentModel;
|
||||||
|
using ModelContextProtocol.Server;
|
||||||
|
using Volo.Abp.DependencyInjection;
|
||||||
|
using Yi.Framework.AiHub.Domain.Shared.Attributes;
|
||||||
|
|
||||||
|
namespace Yi.Framework.AiHub.Domain.Mcp;
|
||||||
|
|
||||||
|
[YiAgentTool]
|
||||||
|
public class OnlineSearchTool:ISingletonDependency
|
||||||
|
{
|
||||||
|
[YiAgentTool("联网搜索"),DisplayName("OnlineSearch"), Description("进行在线搜索")]
|
||||||
|
public string OnlineSearch(string keyword)
|
||||||
|
{
|
||||||
|
return "奥德赛第一中学学生会会长是:郭老板";
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -3,6 +3,8 @@
|
|||||||
<ItemGroup>
|
<ItemGroup>
|
||||||
<PackageReference Include="AlipayEasySDK" Version="2.1.3" />
|
<PackageReference Include="AlipayEasySDK" Version="2.1.3" />
|
||||||
<PackageReference Include="Azure.AI.OpenAI" Version="2.2.0-beta.4" />
|
<PackageReference Include="Azure.AI.OpenAI" Version="2.2.0-beta.4" />
|
||||||
|
<PackageReference Include="Microsoft.Agents.AI.OpenAI" Version="1.0.0-preview.251219.1" />
|
||||||
|
<PackageReference Include="ModelContextProtocol.Core" Version="0.5.0-preview.1" />
|
||||||
<PackageReference Include="Volo.Abp.Ddd.Domain" Version="$(AbpVersion)" />
|
<PackageReference Include="Volo.Abp.Ddd.Domain" Version="$(AbpVersion)" />
|
||||||
<PackageReference Include="Volo.Abp.DistributedLocking" Version="$(AbpVersion)" />
|
<PackageReference Include="Volo.Abp.DistributedLocking" Version="$(AbpVersion)" />
|
||||||
</ItemGroup>
|
</ItemGroup>
|
||||||
@@ -10,6 +12,7 @@
|
|||||||
<ItemGroup>
|
<ItemGroup>
|
||||||
<ProjectReference Include="..\..\..\framework\Yi.Framework.Mapster\Yi.Framework.Mapster.csproj" />
|
<ProjectReference Include="..\..\..\framework\Yi.Framework.Mapster\Yi.Framework.Mapster.csproj" />
|
||||||
<ProjectReference Include="..\..\..\framework\Yi.Framework.SqlSugarCore.Abstractions\Yi.Framework.SqlSugarCore.Abstractions.csproj" />
|
<ProjectReference Include="..\..\..\framework\Yi.Framework.SqlSugarCore.Abstractions\Yi.Framework.SqlSugarCore.Abstractions.csproj" />
|
||||||
|
<ProjectReference Include="..\Yi.Framework.AiHub.Application.Contracts\Yi.Framework.AiHub.Application.Contracts.csproj" />
|
||||||
<ProjectReference Include="..\Yi.Framework.AiHub.Domain.Shared\Yi.Framework.AiHub.Domain.Shared.csproj" />
|
<ProjectReference Include="..\Yi.Framework.AiHub.Domain.Shared\Yi.Framework.AiHub.Domain.Shared.csproj" />
|
||||||
</ItemGroup>
|
</ItemGroup>
|
||||||
|
|
||||||
|
|||||||
@@ -17,6 +17,7 @@ using Yi.Framework.AiHub.Domain.Shared;
|
|||||||
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi;
|
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi;
|
||||||
using Yi.Framework.Mapster;
|
using Yi.Framework.Mapster;
|
||||||
using Microsoft.Extensions.DependencyInjection;
|
using Microsoft.Extensions.DependencyInjection;
|
||||||
|
using Yi.Framework.AiHub.Domain.AiGateWay.Impl.ThorGemini.Chats;
|
||||||
|
|
||||||
namespace Yi.Framework.AiHub.Domain
|
namespace Yi.Framework.AiHub.Domain
|
||||||
{
|
{
|
||||||
@@ -61,6 +62,12 @@ namespace Yi.Framework.AiHub.Domain
|
|||||||
|
|
||||||
#endregion
|
#endregion
|
||||||
|
|
||||||
|
#region Gemini GenerateContent
|
||||||
|
services.AddKeyedTransient<IGeminiGenerateContentService, GeminiGenerateContentService>(
|
||||||
|
nameof(GeminiGenerateContentService));
|
||||||
|
|
||||||
|
#endregion
|
||||||
|
|
||||||
#region Image
|
#region Image
|
||||||
|
|
||||||
services.AddKeyedTransient<IImageService, AzureOpenAIServiceImageService>(
|
services.AddKeyedTransient<IImageService, AzureOpenAIServiceImageService>(
|
||||||
|
|||||||
@@ -290,12 +290,20 @@ namespace Yi.Abp.Web
|
|||||||
{
|
{
|
||||||
OnMessageReceived = messageContext =>
|
OnMessageReceived = messageContext =>
|
||||||
{
|
{
|
||||||
//优先Query中获取
|
//优先Query中获取,再去cookies中获取
|
||||||
var accessToken = messageContext.Request.Query["access_token"];
|
var accessToken = messageContext.Request.Query["access_token"];
|
||||||
if (!string.IsNullOrEmpty(accessToken))
|
if (!string.IsNullOrEmpty(accessToken))
|
||||||
{
|
{
|
||||||
messageContext.Token = accessToken;
|
messageContext.Token = accessToken;
|
||||||
}
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
if (messageContext.Request.Cookies.TryGetValue("Token", out var cookiesToken))
|
||||||
|
{
|
||||||
|
messageContext.Token = cookiesToken;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return Task.CompletedTask;
|
return Task.CompletedTask;
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
@@ -315,11 +323,19 @@ namespace Yi.Abp.Web
|
|||||||
{
|
{
|
||||||
OnMessageReceived = messageContext =>
|
OnMessageReceived = messageContext =>
|
||||||
{
|
{
|
||||||
var refreshToken = messageContext.Request.Query["refresh_token"];
|
var headerRefreshToken = messageContext.Request.Headers["refresh_token"];
|
||||||
if (!string.IsNullOrEmpty(refreshToken))
|
if (!string.IsNullOrEmpty(headerRefreshToken))
|
||||||
{
|
{
|
||||||
messageContext.Token = refreshToken;
|
messageContext.Token = headerRefreshToken;
|
||||||
|
return Task.CompletedTask;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
var queryRefreshToken = messageContext.Request.Query["refresh_token"];
|
||||||
|
if (!string.IsNullOrEmpty(queryRefreshToken))
|
||||||
|
{
|
||||||
|
messageContext.Token = queryRefreshToken;
|
||||||
|
}
|
||||||
|
|
||||||
return Task.CompletedTask;
|
return Task.CompletedTask;
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
@@ -343,7 +359,7 @@ namespace Yi.Abp.Web
|
|||||||
app.UseRouting();
|
app.UseRouting();
|
||||||
|
|
||||||
//app.ApplicationServices.GetRequiredService<ISqlSugarDbContext>().SqlSugarClient.CodeFirst.InitTables<AiModelEntity>();
|
//app.ApplicationServices.GetRequiredService<ISqlSugarDbContext>().SqlSugarClient.CodeFirst.InitTables<AiModelEntity>();
|
||||||
// app.ApplicationServices.GetRequiredService<ISqlSugarDbContext>().SqlSugarClient.CodeFirst.InitTables<TokenAggregateRoot>();
|
// app.ApplicationServices.GetRequiredService<ISqlSugarDbContext>().SqlSugarClient.CodeFirst.InitTables<ActivationCodeRecordAggregateRoot>();
|
||||||
// app.ApplicationServices.GetRequiredService<ISqlSugarDbContext>().SqlSugarClient.CodeFirst.InitTables<UsageStatisticsAggregateRoot>();
|
// app.ApplicationServices.GetRequiredService<ISqlSugarDbContext>().SqlSugarClient.CodeFirst.InitTables<UsageStatisticsAggregateRoot>();
|
||||||
|
|
||||||
//跨域
|
//跨域
|
||||||
|
|||||||
@@ -2,7 +2,8 @@
|
|||||||
"permissions": {
|
"permissions": {
|
||||||
"allow": [
|
"allow": [
|
||||||
"Bash(npx vue-tsc --noEmit)",
|
"Bash(npx vue-tsc --noEmit)",
|
||||||
"Bash(timeout 60 npx vue-tsc:*)"
|
"Bash(timeout 60 npx vue-tsc:*)",
|
||||||
|
"Bash(npm run dev:*)"
|
||||||
],
|
],
|
||||||
"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,8 +112,8 @@
|
|||||||
<body>
|
<body>
|
||||||
<!-- 加载动画容器 -->
|
<!-- 加载动画容器 -->
|
||||||
<div id="yixinai-loader" class="loader-container">
|
<div id="yixinai-loader" class="loader-container">
|
||||||
<div class="loader-title">意心Ai 2.6</div>
|
<div class="loader-title">意心Ai 2.9</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>
|
||||||
</div>
|
</div>
|
||||||
|
|||||||
@@ -44,7 +44,9 @@
|
|||||||
"fingerprintjs": "^0.5.3",
|
"fingerprintjs": "^0.5.3",
|
||||||
"hook-fetch": "^2.0.4-beta.1",
|
"hook-fetch": "^2.0.4-beta.1",
|
||||||
"lodash-es": "^4.17.21",
|
"lodash-es": "^4.17.21",
|
||||||
|
"mammoth": "^1.11.0",
|
||||||
"nprogress": "^0.2.0",
|
"nprogress": "^0.2.0",
|
||||||
|
"pdfjs-dist": "^5.4.449",
|
||||||
"pinia": "^3.0.3",
|
"pinia": "^3.0.3",
|
||||||
"pinia-plugin-persistedstate": "^4.4.1",
|
"pinia-plugin-persistedstate": "^4.4.1",
|
||||||
"qrcode": "^1.5.4",
|
"qrcode": "^1.5.4",
|
||||||
@@ -52,7 +54,30 @@
|
|||||||
"reset-css": "^5.0.2",
|
"reset-css": "^5.0.2",
|
||||||
"vue": "^3.5.17",
|
"vue": "^3.5.17",
|
||||||
"vue-element-plus-x": "1.3.7",
|
"vue-element-plus-x": "1.3.7",
|
||||||
"vue-router": "4"
|
"vue-router": "4",
|
||||||
|
"xlsx": "^0.18.5",
|
||||||
|
"@shikijs/transformers": "^3.7.0",
|
||||||
|
"chatarea": "^6.0.3",
|
||||||
|
"deepmerge": "^4.3.1",
|
||||||
|
"dompurify": "^3.2.6",
|
||||||
|
"github-markdown-css": "^5.8.1",
|
||||||
|
"highlight.js": "^11.11.1",
|
||||||
|
"lodash": "^4.17.21",
|
||||||
|
"mermaid": "11.12.0",
|
||||||
|
"prismjs": "^1.30.0",
|
||||||
|
"property-information": "^7.1.0",
|
||||||
|
"rehype-katex": "^7.0.1",
|
||||||
|
"rehype-raw": "^7.0.0",
|
||||||
|
"rehype-sanitize": "^6.0.0",
|
||||||
|
"remark-breaks": "^4.0.0",
|
||||||
|
"remark-gfm": "^4.0.1",
|
||||||
|
"remark-math": "^6.0.0",
|
||||||
|
"remark-parse": "^11.0.0",
|
||||||
|
"remark-rehype": "^11.1.2",
|
||||||
|
"shiki": "^3.7.0",
|
||||||
|
"ts-md5": "^2.0.1",
|
||||||
|
"unified": "^11.0.5",
|
||||||
|
"unist-util-visit": "^5.0.0"
|
||||||
},
|
},
|
||||||
"devDependencies": {
|
"devDependencies": {
|
||||||
"@antfu/eslint-config": "^4.16.2",
|
"@antfu/eslint-config": "^4.16.2",
|
||||||
@@ -87,7 +112,37 @@
|
|||||||
"vite-plugin-compression": "^0.5.1",
|
"vite-plugin-compression": "^0.5.1",
|
||||||
"vite-plugin-env-typed": "^0.0.2",
|
"vite-plugin-env-typed": "^0.0.2",
|
||||||
"vite-plugin-svg-icons": "^2.0.1",
|
"vite-plugin-svg-icons": "^2.0.1",
|
||||||
"vue-tsc": "^3.0.1"
|
"vue-tsc": "^3.0.1",
|
||||||
|
"@chromatic-com/storybook": "^3.2.7",
|
||||||
|
"@jsonlee_12138/markdown-it-mermaid": "0.0.6",
|
||||||
|
"@storybook/addon-essentials": "^8.6.14",
|
||||||
|
"@storybook/addon-onboarding": "^8.6.14",
|
||||||
|
"@storybook/addons": "^7.6.17",
|
||||||
|
"@storybook/api": "^7.6.17",
|
||||||
|
"@storybook/blocks": "^8.6.14",
|
||||||
|
"@storybook/experimental-addon-test": "^8.6.14",
|
||||||
|
"@storybook/manager-api": "^8.6.14",
|
||||||
|
"@storybook/test": "^8.6.14",
|
||||||
|
"@storybook/theming": "^8.6.14",
|
||||||
|
"@storybook/vue3": "^8.6.14",
|
||||||
|
"@storybook/vue3-vite": "^8.6.14",
|
||||||
|
"@types/dom-speech-recognition": "^0.0.4",
|
||||||
|
"@types/fs-extra": "^11.0.4",
|
||||||
|
"@types/markdown-it": "^14.1.2",
|
||||||
|
"@types/prismjs": "^1.26.5",
|
||||||
|
"@vitest/browser": "^3.2.4",
|
||||||
|
"@vitest/coverage-v8": "^3.2.4",
|
||||||
|
"esno": "^4.8.0",
|
||||||
|
"fast-glob": "^3.3.3",
|
||||||
|
"playwright": "^1.53.2",
|
||||||
|
"rimraf": "^6.0.1",
|
||||||
|
"sass": "^1.89.2",
|
||||||
|
"storybook": "^8.6.14",
|
||||||
|
"storybook-dark-mode": "^4.0.2",
|
||||||
|
"terser": "^5.43.1",
|
||||||
|
"vite-plugin-dts": "^4.5.4",
|
||||||
|
"vite-plugin-lib-inject-css": "^2.2.2",
|
||||||
|
"vitest": "^3.2.4"
|
||||||
},
|
},
|
||||||
"config": {
|
"config": {
|
||||||
"commitizen": {
|
"commitizen": {
|
||||||
|
|||||||
4701
Yi.Ai.Vue3/pnpm-lock.yaml
generated
4701
Yi.Ai.Vue3/pnpm-lock.yaml
generated
File diff suppressed because it is too large
Load Diff
BIN
Yi.Ai.Vue3/publish_aihub_02.zip
Normal file
BIN
Yi.Ai.Vue3/publish_aihub_02.zip
Normal file
Binary file not shown.
5
Yi.Ai.Vue3/src/api/activationCode/index.ts
Normal file
5
Yi.Ai.Vue3/src/api/activationCode/index.ts
Normal file
@@ -0,0 +1,5 @@
|
|||||||
|
import { post } from '@/utils/request';
|
||||||
|
|
||||||
|
export function redeemActivationCode(data: { code: string }) {
|
||||||
|
return post<any>('/activationCode/Redeem', data);
|
||||||
|
}
|
||||||
100
Yi.Ai.Vue3/src/api/channel/index.ts
Normal file
100
Yi.Ai.Vue3/src/api/channel/index.ts
Normal file
@@ -0,0 +1,100 @@
|
|||||||
|
import { del, get, post, put } from '@/utils/request';
|
||||||
|
import type {
|
||||||
|
AiAppDto,
|
||||||
|
AiAppCreateInput,
|
||||||
|
AiAppUpdateInput,
|
||||||
|
AiAppGetListInput,
|
||||||
|
AiModelDto,
|
||||||
|
AiModelCreateInput,
|
||||||
|
AiModelUpdateInput,
|
||||||
|
AiModelGetListInput,
|
||||||
|
PagedResultDto,
|
||||||
|
} from './types';
|
||||||
|
|
||||||
|
// ==================== AI应用管理 ====================
|
||||||
|
|
||||||
|
// 获取AI应用列表
|
||||||
|
export function getAppList(params?: AiAppGetListInput) {
|
||||||
|
const queryParams = new URLSearchParams();
|
||||||
|
if (params?.searchKey) {
|
||||||
|
queryParams.append('SearchKey', params.searchKey);
|
||||||
|
}
|
||||||
|
if (params?.skipCount !== undefined) {
|
||||||
|
queryParams.append('SkipCount', params.skipCount.toString());
|
||||||
|
}
|
||||||
|
if (params?.maxResultCount !== undefined) {
|
||||||
|
queryParams.append('MaxResultCount', params.maxResultCount.toString());
|
||||||
|
}
|
||||||
|
|
||||||
|
const queryString = queryParams.toString();
|
||||||
|
const url = queryString ? `/channel/app?${queryString}` : '/channel/app';
|
||||||
|
|
||||||
|
return get<PagedResultDto<AiAppDto>>(url).json();
|
||||||
|
}
|
||||||
|
|
||||||
|
// 根据ID获取AI应用
|
||||||
|
export function getAppById(id: string) {
|
||||||
|
return get<AiAppDto>(`/channel/app/${id}`).json();
|
||||||
|
}
|
||||||
|
|
||||||
|
// 创建AI应用
|
||||||
|
export function createApp(data: AiAppCreateInput) {
|
||||||
|
return post<AiAppDto>('/channel/app', data).json();
|
||||||
|
}
|
||||||
|
|
||||||
|
// 更新AI应用
|
||||||
|
export function updateApp(data: AiAppUpdateInput) {
|
||||||
|
return put<AiAppDto>('/channel/app', data).json();
|
||||||
|
}
|
||||||
|
|
||||||
|
// 删除AI应用
|
||||||
|
export function deleteApp(id: string) {
|
||||||
|
return del(`/channel/app/${id}`).json();
|
||||||
|
}
|
||||||
|
|
||||||
|
// ==================== AI模型管理 ====================
|
||||||
|
|
||||||
|
// 获取AI模型列表
|
||||||
|
export function getModelList(params?: AiModelGetListInput) {
|
||||||
|
const queryParams = new URLSearchParams();
|
||||||
|
if (params?.searchKey) {
|
||||||
|
queryParams.append('SearchKey', params.searchKey);
|
||||||
|
}
|
||||||
|
if (params?.aiAppId) {
|
||||||
|
queryParams.append('AiAppId', params.aiAppId);
|
||||||
|
}
|
||||||
|
if (params?.isPremiumOnly !== undefined) {
|
||||||
|
queryParams.append('IsPremiumOnly', params.isPremiumOnly.toString());
|
||||||
|
}
|
||||||
|
if (params?.skipCount !== undefined) {
|
||||||
|
queryParams.append('SkipCount', params.skipCount.toString());
|
||||||
|
}
|
||||||
|
if (params?.maxResultCount !== undefined) {
|
||||||
|
queryParams.append('MaxResultCount', params.maxResultCount.toString());
|
||||||
|
}
|
||||||
|
|
||||||
|
const queryString = queryParams.toString();
|
||||||
|
const url = queryString ? `/channel/model?${queryString}` : '/channel/model';
|
||||||
|
|
||||||
|
return get<PagedResultDto<AiModelDto>>(url).json();
|
||||||
|
}
|
||||||
|
|
||||||
|
// 根据ID获取AI模型
|
||||||
|
export function getModelById(id: string) {
|
||||||
|
return get<AiModelDto>(`/channel/model/${id}`).json();
|
||||||
|
}
|
||||||
|
|
||||||
|
// 创建AI模型
|
||||||
|
export function createModel(data: AiModelCreateInput) {
|
||||||
|
return post<AiModelDto>('/channel/model', data).json();
|
||||||
|
}
|
||||||
|
|
||||||
|
// 更新AI模型
|
||||||
|
export function updateModel(data: AiModelUpdateInput) {
|
||||||
|
return put<AiModelDto>('/channel/model', data).json();
|
||||||
|
}
|
||||||
|
|
||||||
|
// 删除AI模型
|
||||||
|
export function deleteModel(id: string) {
|
||||||
|
return del(`/channel/model/${id}`).json();
|
||||||
|
}
|
||||||
121
Yi.Ai.Vue3/src/api/channel/types.ts
Normal file
121
Yi.Ai.Vue3/src/api/channel/types.ts
Normal file
@@ -0,0 +1,121 @@
|
|||||||
|
// 模型类型枚举
|
||||||
|
export enum ModelTypeEnum {
|
||||||
|
Chat = 0,
|
||||||
|
Image = 1,
|
||||||
|
Embedding = 2,
|
||||||
|
PremiumChat = 3,
|
||||||
|
}
|
||||||
|
|
||||||
|
// 模型API类型枚举
|
||||||
|
export enum ModelApiTypeEnum {
|
||||||
|
OpenAi = 0,
|
||||||
|
Claude = 1,
|
||||||
|
}
|
||||||
|
|
||||||
|
// AI应用DTO
|
||||||
|
export interface AiAppDto {
|
||||||
|
id: string;
|
||||||
|
name: string;
|
||||||
|
endpoint: string;
|
||||||
|
extraUrl?: string;
|
||||||
|
apiKey: string;
|
||||||
|
orderNum: number;
|
||||||
|
creationTime: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 创建AI应用输入
|
||||||
|
export interface AiAppCreateInput {
|
||||||
|
name: string;
|
||||||
|
endpoint: string;
|
||||||
|
extraUrl?: string;
|
||||||
|
apiKey: string;
|
||||||
|
orderNum: number;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 更新AI应用输入
|
||||||
|
export interface AiAppUpdateInput {
|
||||||
|
id: string;
|
||||||
|
name: string;
|
||||||
|
endpoint: string;
|
||||||
|
extraUrl?: string;
|
||||||
|
apiKey: string;
|
||||||
|
orderNum: number;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取AI应用列表输入
|
||||||
|
export interface AiAppGetListInput {
|
||||||
|
searchKey?: string;
|
||||||
|
skipCount?: number;
|
||||||
|
maxResultCount?: number;
|
||||||
|
}
|
||||||
|
|
||||||
|
// AI模型DTO
|
||||||
|
export interface AiModelDto {
|
||||||
|
id: string;
|
||||||
|
handlerName: string;
|
||||||
|
modelId: string;
|
||||||
|
name: string;
|
||||||
|
description?: string;
|
||||||
|
orderNum: number;
|
||||||
|
aiAppId: string;
|
||||||
|
extraInfo?: string;
|
||||||
|
modelType: ModelTypeEnum;
|
||||||
|
modelApiType: ModelApiTypeEnum;
|
||||||
|
multiplier: number;
|
||||||
|
multiplierShow: number;
|
||||||
|
providerName?: string;
|
||||||
|
iconUrl?: string;
|
||||||
|
isPremium: boolean;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 创建AI模型输入
|
||||||
|
export interface AiModelCreateInput {
|
||||||
|
handlerName: string;
|
||||||
|
modelId: string;
|
||||||
|
name: string;
|
||||||
|
description?: string;
|
||||||
|
orderNum: number;
|
||||||
|
aiAppId: string;
|
||||||
|
extraInfo?: string;
|
||||||
|
modelType: ModelTypeEnum;
|
||||||
|
modelApiType: ModelApiTypeEnum;
|
||||||
|
multiplier: number;
|
||||||
|
multiplierShow: number;
|
||||||
|
providerName?: string;
|
||||||
|
iconUrl?: string;
|
||||||
|
isPremium: boolean;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 更新AI模型输入
|
||||||
|
export interface AiModelUpdateInput {
|
||||||
|
id: string;
|
||||||
|
handlerName: string;
|
||||||
|
modelId: string;
|
||||||
|
name: string;
|
||||||
|
description?: string;
|
||||||
|
orderNum: number;
|
||||||
|
aiAppId: string;
|
||||||
|
extraInfo?: string;
|
||||||
|
modelType: ModelTypeEnum;
|
||||||
|
modelApiType: ModelApiTypeEnum;
|
||||||
|
multiplier: number;
|
||||||
|
multiplierShow: number;
|
||||||
|
providerName?: string;
|
||||||
|
iconUrl?: string;
|
||||||
|
isPremium: boolean;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取AI模型列表输入
|
||||||
|
export interface AiModelGetListInput {
|
||||||
|
searchKey?: string;
|
||||||
|
aiAppId?: string;
|
||||||
|
isPremiumOnly?: boolean;
|
||||||
|
skipCount?: number;
|
||||||
|
maxResultCount?: number;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 分页结果
|
||||||
|
export interface PagedResultDto<T> {
|
||||||
|
items: T[];
|
||||||
|
totalCount: number;
|
||||||
|
}
|
||||||
@@ -19,3 +19,8 @@ export function getChatList(params: GetChatListParams) {
|
|||||||
// return get<ChatMessageVo[]>('/system/message/list', params);
|
// return get<ChatMessageVo[]>('/system/message/list', params);
|
||||||
return get<ChatMessageVo[]>('/message', params).json();
|
return get<ChatMessageVo[]>('/message', params).json();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 新增对应会话聊天记录
|
||||||
|
export function aiChatTool() {
|
||||||
|
return post('/ai-chat/tool').json();
|
||||||
|
}
|
||||||
|
|||||||
@@ -220,4 +220,16 @@ export interface ChatMessageVo {
|
|||||||
* 用户id
|
* 用户id
|
||||||
*/
|
*/
|
||||||
userId?: number;
|
userId?: number;
|
||||||
|
/**
|
||||||
|
* 用户消息中的图片列表(前端扩展字段)
|
||||||
|
*/
|
||||||
|
images?: Array<{ url: string; name?: string }>;
|
||||||
|
/**
|
||||||
|
* 用户消息中的文件列表(前端扩展字段)
|
||||||
|
*/
|
||||||
|
files?: Array<{ name: string; size: number }>;
|
||||||
|
/**
|
||||||
|
* 创建时间(前端显示用)
|
||||||
|
*/
|
||||||
|
creationTime?: string;
|
||||||
}
|
}
|
||||||
|
|||||||
34
Yi.Ai.Vue3/src/api/file/index.ts
Normal file
34
Yi.Ai.Vue3/src/api/file/index.ts
Normal file
@@ -0,0 +1,34 @@
|
|||||||
|
import type { UploadFileResponse } from './types';
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 上传文件
|
||||||
|
* @param file 文件对象
|
||||||
|
* @returns 返回文件ID数组
|
||||||
|
*/
|
||||||
|
export async function uploadFile(file: File): Promise<UploadFileResponse[]> {
|
||||||
|
const formData = new FormData();
|
||||||
|
formData.append('file', file);
|
||||||
|
|
||||||
|
const uploadApiUrl = import.meta.env.VITE_FILE_UPLOAD_API;
|
||||||
|
|
||||||
|
const response = await fetch(`${uploadApiUrl}/prod-api/file`, {
|
||||||
|
method: 'POST',
|
||||||
|
body: formData,
|
||||||
|
});
|
||||||
|
|
||||||
|
if (!response.ok) {
|
||||||
|
throw new Error('文件上传失败');
|
||||||
|
}
|
||||||
|
|
||||||
|
const result = await response.json();
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 生成文件URL
|
||||||
|
* @param fileId 文件ID
|
||||||
|
* @returns 文件访问URL
|
||||||
|
*/
|
||||||
|
export function getFileUrl(fileId: string): string {
|
||||||
|
return `https://ccnetcore.com/prod-api/file/${fileId}/true`;
|
||||||
|
}
|
||||||
3
Yi.Ai.Vue3/src/api/file/types.ts
Normal file
3
Yi.Ai.Vue3/src/api/file/types.ts
Normal file
@@ -0,0 +1,3 @@
|
|||||||
|
export interface UploadFileResponse {
|
||||||
|
id: string;
|
||||||
|
}
|
||||||
@@ -1,6 +1,7 @@
|
|||||||
export * from './announcement'
|
export * from './announcement'
|
||||||
export * from './auth';
|
export * from './auth';
|
||||||
export * from './chat';
|
export * from './chat';
|
||||||
|
export * from './file';
|
||||||
export * from './model';
|
export * from './model';
|
||||||
export * from './pay';
|
export * from './pay';
|
||||||
export * from './session';
|
export * from './session';
|
||||||
|
|||||||
@@ -34,8 +34,7 @@ export interface ModelLibraryDto {
|
|||||||
name: string;
|
name: string;
|
||||||
description?: string;
|
description?: string;
|
||||||
modelType: ModelTypeEnum;
|
modelType: ModelTypeEnum;
|
||||||
modelTypeName: string;
|
modelApiTypes: Array;
|
||||||
modelApiType: ModelApiTypeEnum;
|
|
||||||
modelApiTypeName: string;
|
modelApiTypeName: string;
|
||||||
multiplierShow: number;
|
multiplierShow: number;
|
||||||
providerName?: string;
|
providerName?: string;
|
||||||
|
|||||||
7
Yi.Ai.Vue3/src/assets/icons/System/notification-fill.svg
Normal file
7
Yi.Ai.Vue3/src/assets/icons/System/notification-fill.svg
Normal file
@@ -0,0 +1,7 @@
|
|||||||
|
<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="currentColor">
|
||||||
|
<g>
|
||||||
|
|
||||||
|
<path fill="none" d="M0 0h24v24H0z"/>
|
||||||
|
<path d="M12 2C16.9706 2 21 6.04348 21 11.0314V20H3V11.0314C3 6.04348 7.02944 2 12 2ZM9.5 21H14.5C14.5 22.3807 13.3807 23.5 12 23.5C10.6193 23.5 9.5 22.3807 9.5 21Z"></path>
|
||||||
|
</g>
|
||||||
|
</svg>
|
||||||
|
After Width: | Height: | Size: 322 B |
@@ -1,148 +1,751 @@
|
|||||||
<!-- 文件上传 -->
|
<!-- 文件上传 -->
|
||||||
<script setup lang="ts">
|
<script setup lang="ts">
|
||||||
import type { FilesCardProps } from 'vue-element-plus-x/types/FilesCard';
|
import type { FileItem } from '@/stores/modules/files';
|
||||||
import { useFileDialog } from '@vueuse/core';
|
import { useFileDialog } from '@vueuse/core';
|
||||||
import { ElMessage } from 'element-plus';
|
import { ElMessage } from 'element-plus';
|
||||||
import Popover from '@/components/Popover/index.vue';
|
import mammoth from 'mammoth';
|
||||||
import SvgIcon from '@/components/SvgIcon/index.vue';
|
import * as pdfjsLib from 'pdfjs-dist';
|
||||||
|
import * as XLSX from 'xlsx';
|
||||||
import { useFilesStore } from '@/stores/modules/files';
|
import { useFilesStore } from '@/stores/modules/files';
|
||||||
|
|
||||||
type FilesList = FilesCardProps & {
|
// 配置 PDF.js worker - 使用稳定的 CDN
|
||||||
file: File;
|
pdfjsLib.GlobalWorkerOptions.workerSrc = `https://cdn.jsdelivr.net/npm/pdfjs-dist@${pdfjsLib.version}/build/pdf.worker.min.mjs`;
|
||||||
};
|
|
||||||
|
|
||||||
const filesStore = useFilesStore();
|
const filesStore = useFilesStore();
|
||||||
|
|
||||||
/* 弹出面板 开始 */
|
// 文件大小限制 3MB
|
||||||
const popoverStyle = ref({
|
const MAX_FILE_SIZE = 3 * 1024 * 1024;
|
||||||
padding: '4px',
|
|
||||||
height: 'fit-content',
|
// 单个文件内容长度限制
|
||||||
background: 'var(--el-bg-color, #fff)',
|
const MAX_TEXT_FILE_LENGTH = 50000; // 文本文件最大字符数
|
||||||
border: '1px solid var(--el-border-color-light)',
|
const MAX_WORD_LENGTH = 30000; // Word 文档最大字符数
|
||||||
borderRadius: '8px',
|
const MAX_EXCEL_ROWS = 100; // Excel 最大行数
|
||||||
boxShadow: '0 2px 12px 0 rgba(0, 0, 0, 0.1)',
|
const MAX_PDF_PAGES = 10; // PDF 最大页数
|
||||||
});
|
|
||||||
const popoverRef = ref();
|
// 整个消息总长度限制(所有文件内容加起来,预估 token 安全限制)
|
||||||
/* 弹出面板 结束 */
|
// 272000 tokens * 0.55 安全系数 ≈ 150000 字符
|
||||||
|
const MAX_TOTAL_CONTENT_LENGTH = 150000;
|
||||||
|
|
||||||
const { reset, open, onChange } = useFileDialog({
|
const { reset, open, onChange } = useFileDialog({
|
||||||
// 允许所有图片文件,文档文件,音视频文件
|
// 支持图片、文档、文本文件等
|
||||||
accept: 'image/*,video/*,audio/*,application/*',
|
accept: 'image/*,.txt,.log,.csv,.tsv,.md,.markdown,.json,.xml,.yaml,.yml,.toml,.ini,.conf,.config,.properties,.prop,.env,'
|
||||||
directory: false, // 是否允许选择文件夹
|
+ '.js,.jsx,.ts,.tsx,.vue,.html,.htm,.css,.scss,.sass,.less,.styl,'
|
||||||
multiple: true, // 是否允许多选
|
+ '.java,.c,.cpp,.h,.hpp,.cs,.py,.rb,.go,.rs,.swift,.kt,.php,.sh,.bash,.zsh,.fish,.bat,.cmd,.ps1,'
|
||||||
|
+ '.sql,.graphql,.proto,.thrift,'
|
||||||
|
+ '.dockerfile,.gitignore,.gitattributes,.editorconfig,.npmrc,.nvmrc,'
|
||||||
|
+ '.sln,.csproj,.vbproj,.fsproj,.props,.targets,'
|
||||||
|
+ '.xlsx,.xls,.csv,.docx,.pdf',
|
||||||
|
directory: false,
|
||||||
|
multiple: true,
|
||||||
});
|
});
|
||||||
|
|
||||||
onChange((files) => {
|
/**
|
||||||
|
* 压缩图片
|
||||||
|
* @param {File} file - 原始图片文件
|
||||||
|
* @param {number} maxWidth - 最大宽度,默认 1024px
|
||||||
|
* @param {number} maxHeight - 最大高度,默认 1024px
|
||||||
|
* @param {number} quality - 压缩质量,0-1之间,默认 0.8
|
||||||
|
* @returns {Promise<Blob>} 压缩后的图片 Blob
|
||||||
|
*/
|
||||||
|
function compressImage(file: File, maxWidth = 1024, maxHeight = 1024, quality = 0.8): Promise<Blob> {
|
||||||
|
return new Promise((resolve, reject) => {
|
||||||
|
const reader = new FileReader();
|
||||||
|
reader.onload = (e) => {
|
||||||
|
const img = new Image();
|
||||||
|
img.onload = () => {
|
||||||
|
const canvas = document.createElement('canvas');
|
||||||
|
let width = img.width;
|
||||||
|
let height = img.height;
|
||||||
|
|
||||||
|
// 计算缩放比例
|
||||||
|
if (width > maxWidth || height > maxHeight) {
|
||||||
|
const ratio = Math.min(maxWidth / width, maxHeight / height);
|
||||||
|
width = width * ratio;
|
||||||
|
height = height * ratio;
|
||||||
|
}
|
||||||
|
|
||||||
|
canvas.width = width;
|
||||||
|
canvas.height = height;
|
||||||
|
|
||||||
|
const ctx = canvas.getContext('2d')!;
|
||||||
|
ctx.drawImage(img, 0, 0, width, height);
|
||||||
|
|
||||||
|
// 转换为 Blob
|
||||||
|
canvas.toBlob(
|
||||||
|
(blob) => {
|
||||||
|
if (blob) {
|
||||||
|
resolve(blob);
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
reject(new Error('压缩失败'));
|
||||||
|
}
|
||||||
|
},
|
||||||
|
file.type,
|
||||||
|
quality,
|
||||||
|
);
|
||||||
|
};
|
||||||
|
img.onerror = reject;
|
||||||
|
img.src = e.target?.result as string;
|
||||||
|
};
|
||||||
|
reader.onerror = reject;
|
||||||
|
reader.readAsDataURL(file);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 将 Blob 转换为 base64 格式
|
||||||
|
* @param {Blob} blob - 要转换的 Blob 对象
|
||||||
|
* @returns {Promise<string>} base64 编码的字符串(包含 data:xxx;base64, 前缀)
|
||||||
|
*/
|
||||||
|
function blobToBase64(blob: Blob): Promise<string> {
|
||||||
|
return new Promise((resolve, reject) => {
|
||||||
|
const reader = new FileReader();
|
||||||
|
reader.onload = () => {
|
||||||
|
resolve(reader.result as string);
|
||||||
|
};
|
||||||
|
reader.onerror = reject;
|
||||||
|
reader.readAsDataURL(blob);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 读取文本文件内容
|
||||||
|
* @param {File} file - 文本文件
|
||||||
|
* @returns {Promise<string>} 文件内容字符串
|
||||||
|
*/
|
||||||
|
function readTextFile(file: File): Promise<string> {
|
||||||
|
return new Promise((resolve, reject) => {
|
||||||
|
const reader = new FileReader();
|
||||||
|
reader.onload = () => {
|
||||||
|
resolve(reader.result as string);
|
||||||
|
};
|
||||||
|
reader.onerror = reject;
|
||||||
|
reader.readAsText(file, 'UTF-8');
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 判断是否为文本文件
|
||||||
|
* 通过 MIME 类型或文件扩展名判断
|
||||||
|
* @param {File} file - 要判断的文件
|
||||||
|
* @returns {boolean} 是否为文本文件
|
||||||
|
*/
|
||||||
|
function isTextFile(file: File): boolean {
|
||||||
|
// 通过 MIME type 判断
|
||||||
|
if (file.type.startsWith('text/')) {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 通过扩展名判断(更全面的列表)
|
||||||
|
const textExtensions = [
|
||||||
|
// 通用文本
|
||||||
|
'txt',
|
||||||
|
'log',
|
||||||
|
'md',
|
||||||
|
'markdown',
|
||||||
|
'rtf',
|
||||||
|
// 配置文件
|
||||||
|
'json',
|
||||||
|
'xml',
|
||||||
|
'yaml',
|
||||||
|
'yml',
|
||||||
|
'toml',
|
||||||
|
'ini',
|
||||||
|
'conf',
|
||||||
|
'config',
|
||||||
|
'properties',
|
||||||
|
'prop',
|
||||||
|
'env',
|
||||||
|
// 前端
|
||||||
|
'js',
|
||||||
|
'jsx',
|
||||||
|
'ts',
|
||||||
|
'tsx',
|
||||||
|
'vue',
|
||||||
|
'html',
|
||||||
|
'htm',
|
||||||
|
'css',
|
||||||
|
'scss',
|
||||||
|
'sass',
|
||||||
|
'less',
|
||||||
|
'styl',
|
||||||
|
// 编程语言
|
||||||
|
'java',
|
||||||
|
'c',
|
||||||
|
'cpp',
|
||||||
|
'h',
|
||||||
|
'hpp',
|
||||||
|
'cs',
|
||||||
|
'py',
|
||||||
|
'rb',
|
||||||
|
'go',
|
||||||
|
'rs',
|
||||||
|
'swift',
|
||||||
|
'kt',
|
||||||
|
'php',
|
||||||
|
// 脚本
|
||||||
|
'sh',
|
||||||
|
'bash',
|
||||||
|
'zsh',
|
||||||
|
'fish',
|
||||||
|
'bat',
|
||||||
|
'cmd',
|
||||||
|
'ps1',
|
||||||
|
// 数据库/API
|
||||||
|
'sql',
|
||||||
|
'graphql',
|
||||||
|
'proto',
|
||||||
|
'thrift',
|
||||||
|
// 版本控制/工具
|
||||||
|
'dockerfile',
|
||||||
|
'gitignore',
|
||||||
|
'gitattributes',
|
||||||
|
'editorconfig',
|
||||||
|
'npmrc',
|
||||||
|
'nvmrc',
|
||||||
|
// .NET 项目文件
|
||||||
|
'sln',
|
||||||
|
'csproj',
|
||||||
|
'vbproj',
|
||||||
|
'fsproj',
|
||||||
|
'props',
|
||||||
|
'targets',
|
||||||
|
// 数据文件
|
||||||
|
'csv',
|
||||||
|
'tsv',
|
||||||
|
];
|
||||||
|
|
||||||
|
const ext = file.name.split('.').pop()?.toLowerCase();
|
||||||
|
return ext ? textExtensions.includes(ext) : false;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 解析 Excel 文件,提取前 N 行数据转为 CSV 格式
|
||||||
|
* @param {File} file - Excel 文件 (.xlsx, .xls)
|
||||||
|
* @returns {Promise<{content: string, totalRows: number, extractedRows: number}>}
|
||||||
|
* - content: CSV 格式的文本内容
|
||||||
|
* - totalRows: 文件总行数
|
||||||
|
* - extractedRows: 实际提取的行数(受 MAX_EXCEL_ROWS 限制)
|
||||||
|
*/
|
||||||
|
async function parseExcel(file: File): Promise<{ content: string; totalRows: number; extractedRows: number }> {
|
||||||
|
return new Promise((resolve, reject) => {
|
||||||
|
const reader = new FileReader();
|
||||||
|
reader.onload = (e) => {
|
||||||
|
try {
|
||||||
|
const data = new Uint8Array(e.target?.result as ArrayBuffer);
|
||||||
|
const workbook = XLSX.read(data, { type: 'array' });
|
||||||
|
|
||||||
|
let result = '';
|
||||||
|
let totalRows = 0;
|
||||||
|
let extractedRows = 0;
|
||||||
|
|
||||||
|
workbook.SheetNames.forEach((sheetName, index) => {
|
||||||
|
const worksheet = workbook.Sheets[sheetName];
|
||||||
|
|
||||||
|
// 获取工作表的范围
|
||||||
|
const range = XLSX.utils.decode_range(worksheet['!ref'] || 'A1');
|
||||||
|
const sheetTotalRows = range.e.r - range.s.r + 1;
|
||||||
|
totalRows += sheetTotalRows;
|
||||||
|
|
||||||
|
// 限制行数
|
||||||
|
const rowsToExtract = Math.min(sheetTotalRows, MAX_EXCEL_ROWS);
|
||||||
|
extractedRows += rowsToExtract;
|
||||||
|
|
||||||
|
// 创建新的范围,只包含前 N 行
|
||||||
|
const limitedRange = {
|
||||||
|
s: { r: range.s.r, c: range.s.c },
|
||||||
|
e: { r: range.s.r + rowsToExtract - 1, c: range.e.c },
|
||||||
|
};
|
||||||
|
|
||||||
|
// 提取限制范围内的数据
|
||||||
|
const limitedData: any[][] = [];
|
||||||
|
for (let row = limitedRange.s.r; row <= limitedRange.e.r; row++) {
|
||||||
|
const rowData: any[] = [];
|
||||||
|
for (let col = limitedRange.s.c; col <= limitedRange.e.c; col++) {
|
||||||
|
const cellAddress = XLSX.utils.encode_cell({ r: row, c: col });
|
||||||
|
const cell = worksheet[cellAddress];
|
||||||
|
rowData.push(cell ? cell.v : '');
|
||||||
|
}
|
||||||
|
limitedData.push(rowData);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 转换为 CSV
|
||||||
|
const csvData = limitedData.map(row => row.join(',')).join('\n');
|
||||||
|
|
||||||
|
if (workbook.SheetNames.length > 1) {
|
||||||
|
result += `=== Sheet: ${sheetName} ===\n`;
|
||||||
|
}
|
||||||
|
result += csvData;
|
||||||
|
if (index < workbook.SheetNames.length - 1) {
|
||||||
|
result += '\n\n';
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
resolve({ content: result, totalRows, extractedRows });
|
||||||
|
}
|
||||||
|
catch (error) {
|
||||||
|
reject(error);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
reader.onerror = reject;
|
||||||
|
reader.readAsArrayBuffer(file);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 解析 Word 文档,提取纯文本内容
|
||||||
|
* @param {File} file - Word 文档 (.docx)
|
||||||
|
* @returns {Promise<{content: string, totalLength: number, extracted: boolean}>}
|
||||||
|
* - content: 提取的文本内容
|
||||||
|
* - totalLength: 原始文本总长度
|
||||||
|
* - extracted: 是否被截断(超过 MAX_WORD_LENGTH)
|
||||||
|
*/
|
||||||
|
async function parseWord(file: File): Promise<{ content: string; totalLength: number; extracted: boolean }> {
|
||||||
|
return new Promise((resolve, reject) => {
|
||||||
|
const reader = new FileReader();
|
||||||
|
reader.onload = async (e) => {
|
||||||
|
try {
|
||||||
|
const arrayBuffer = e.target?.result as ArrayBuffer;
|
||||||
|
const result = await mammoth.extractRawText({ arrayBuffer });
|
||||||
|
const fullText = result.value;
|
||||||
|
const totalLength = fullText.length;
|
||||||
|
|
||||||
|
if (totalLength > MAX_WORD_LENGTH) {
|
||||||
|
const truncated = fullText.substring(0, MAX_WORD_LENGTH);
|
||||||
|
resolve({ content: truncated, totalLength, extracted: true });
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
resolve({ content: fullText, totalLength, extracted: false });
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch (error) {
|
||||||
|
reject(error);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
reader.onerror = reject;
|
||||||
|
reader.readAsArrayBuffer(file);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 解析 PDF 文件,提取前 N 页的文本内容
|
||||||
|
* @param {File} file - PDF 文件
|
||||||
|
* @returns {Promise<{content: string, totalPages: number, extractedPages: number}>}
|
||||||
|
* - content: 提取的文本内容
|
||||||
|
* - totalPages: 文件总页数
|
||||||
|
* - extractedPages: 实际提取的页数(受 MAX_PDF_PAGES 限制)
|
||||||
|
*/
|
||||||
|
async function parsePDF(file: File): Promise<{ content: string; totalPages: number; extractedPages: number }> {
|
||||||
|
return new Promise((resolve, reject) => {
|
||||||
|
const reader = new FileReader();
|
||||||
|
reader.onload = async (e) => {
|
||||||
|
try {
|
||||||
|
const typedArray = new Uint8Array(e.target?.result as ArrayBuffer);
|
||||||
|
const pdf = await pdfjsLib.getDocument(typedArray).promise;
|
||||||
|
const totalPages = pdf.numPages;
|
||||||
|
const pagesToExtract = Math.min(totalPages, MAX_PDF_PAGES);
|
||||||
|
|
||||||
|
let fullText = '';
|
||||||
|
for (let i = 1; i <= pagesToExtract; i++) {
|
||||||
|
const page = await pdf.getPage(i);
|
||||||
|
const textContent = await page.getTextContent();
|
||||||
|
const pageText = textContent.items.map((item: any) => item.str).join(' ');
|
||||||
|
fullText += `${pageText}\n`;
|
||||||
|
}
|
||||||
|
|
||||||
|
resolve({ content: fullText, totalPages, extractedPages: pagesToExtract });
|
||||||
|
}
|
||||||
|
catch (error) {
|
||||||
|
reject(error);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
reader.onerror = reject;
|
||||||
|
reader.readAsArrayBuffer(file);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取文件扩展名
|
||||||
|
* @param {string} filename - 文件名
|
||||||
|
* @returns {string} 小写的扩展名,无点号
|
||||||
|
*/
|
||||||
|
function getFileExtension(filename: string): string {
|
||||||
|
return filename.split('.').pop()?.toLowerCase() || '';
|
||||||
|
}
|
||||||
|
|
||||||
|
onChange(async (files) => {
|
||||||
if (!files)
|
if (!files)
|
||||||
return;
|
return;
|
||||||
const arr = [] as FilesList[];
|
|
||||||
|
const arr = [] as FileItem[];
|
||||||
|
let totalContentLength = 0; // 跟踪总内容长度
|
||||||
|
|
||||||
|
// 先计算已有文件的总内容长度
|
||||||
|
filesStore.filesList.forEach((f) => {
|
||||||
|
if (f.fileType === 'text' && f.fileContent) {
|
||||||
|
totalContentLength += f.fileContent.length;
|
||||||
|
}
|
||||||
|
// 图片 base64 也计入(虽然转 token 时不同,但也要计算)
|
||||||
|
if (f.fileType === 'image' && f.base64) {
|
||||||
|
// base64 转 token 比例约 1:1.5,这里保守估计
|
||||||
|
totalContentLength += Math.floor(f.base64.length * 0.5);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
for (let i = 0; i < files!.length; i++) {
|
for (let i = 0; i < files!.length; i++) {
|
||||||
const file = files![i];
|
const file = files![i];
|
||||||
|
|
||||||
|
// 验证文件大小
|
||||||
|
if (file.size > MAX_FILE_SIZE) {
|
||||||
|
ElMessage.error(`文件 ${file.name} 超过 3MB 限制,已跳过`);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
const ext = getFileExtension(file.name);
|
||||||
|
const isImage = file.type.startsWith('image/');
|
||||||
|
const isExcel = ['xlsx', 'xls'].includes(ext);
|
||||||
|
const isWord = ext === 'docx';
|
||||||
|
const isPDF = ext === 'pdf';
|
||||||
|
const isText = isTextFile(file);
|
||||||
|
|
||||||
|
// 处理图片文件
|
||||||
|
if (isImage) {
|
||||||
|
try {
|
||||||
|
// 控制参数:是否开启图片压缩
|
||||||
|
const enableImageCompression = true; // 这里可以设置为变量或从配置读取
|
||||||
|
|
||||||
|
let finalBlob: Blob = file;
|
||||||
|
let base64 = '';
|
||||||
|
let compressionLevel = 0;
|
||||||
|
const originalSize = (file.size / 1024).toFixed(2);
|
||||||
|
let finalSize = originalSize;
|
||||||
|
|
||||||
|
if (enableImageCompression) {
|
||||||
|
// 多级压缩策略:逐步降低质量和分辨率
|
||||||
|
const compressionLevels = [
|
||||||
|
{ maxWidth: 800, maxHeight: 800, quality: 0.6 },
|
||||||
|
{ maxWidth: 600, maxHeight: 600, quality: 0.5 },
|
||||||
|
{ maxWidth: 400, maxHeight: 400, quality: 0.4 },
|
||||||
|
];
|
||||||
|
|
||||||
|
let compressedBlob: Blob | null = null;
|
||||||
|
|
||||||
|
// 尝试不同级别的压缩
|
||||||
|
for (const level of compressionLevels) {
|
||||||
|
compressionLevel++;
|
||||||
|
compressedBlob = await compressImage(file, level.maxWidth, level.maxHeight, level.quality);
|
||||||
|
base64 = await blobToBase64(compressedBlob);
|
||||||
|
|
||||||
|
// 检查是否满足总长度限制
|
||||||
|
const estimatedLength = Math.floor(base64.length * 0.5);
|
||||||
|
if (totalContentLength + estimatedLength <= MAX_TOTAL_CONTENT_LENGTH) {
|
||||||
|
// 满足限制,使用当前压缩级别
|
||||||
|
totalContentLength += estimatedLength;
|
||||||
|
finalBlob = compressedBlob;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 如果是最后一级压缩仍然超限,则跳过
|
||||||
|
if (compressionLevel === compressionLevels.length) {
|
||||||
|
const fileSizeMB = (file.size / 1024 / 1024).toFixed(2);
|
||||||
|
ElMessage.error(`${file.name} 图片内容过大,请压缩后上传`);
|
||||||
|
compressedBlob = null;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 如果压缩失败,跳过此文件
|
||||||
|
if (!compressedBlob) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 计算压缩比例
|
||||||
|
finalSize = (finalBlob.size / 1024).toFixed(2);
|
||||||
|
console.log(`图片压缩: ${file.name} - 原始: ${originalSize}KB, 压缩后: ${finalSize}KB (级别${compressionLevel})`);
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
// 不开启压缩时,直接转换原始文件
|
||||||
|
base64 = await blobToBase64(file);
|
||||||
|
|
||||||
|
// 检查总长度限制
|
||||||
|
const estimatedLength = Math.floor(base64.length * 0.5);
|
||||||
|
if (totalContentLength + estimatedLength > MAX_TOTAL_CONTENT_LENGTH) {
|
||||||
|
const fileSizeMB = (file.size / 1024 / 1024).toFixed(2);
|
||||||
|
ElMessage.error(`${file.name} (${fileSizeMB}MB) 超过总长度限制,已跳过`);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
totalContentLength += estimatedLength;
|
||||||
|
console.log(`图片未压缩: ${file.name} - 大小: ${originalSize}KB`);
|
||||||
|
}
|
||||||
|
|
||||||
arr.push({
|
arr.push({
|
||||||
uid: crypto.randomUUID(), // 不写 uid,文件列表展示不出来,elx 1.2.0 bug 待修复
|
uid: crypto.randomUUID(),
|
||||||
name: file.name,
|
name: file.name,
|
||||||
fileSize: file.size,
|
fileSize: file.size,
|
||||||
file,
|
file,
|
||||||
maxWidth: '200px',
|
maxWidth: '200px',
|
||||||
showDelIcon: true, // 显示删除图标
|
showDelIcon: true,
|
||||||
imgPreview: true, // 显示图片预览
|
imgPreview: true,
|
||||||
imgVariant: 'square', // 图片预览的形状
|
imgVariant: 'square',
|
||||||
url: URL.createObjectURL(file), // 图片预览地址
|
url: base64, // 使用压缩后的 base64 作为预览地址
|
||||||
|
isUploaded: true,
|
||||||
|
base64,
|
||||||
|
fileType: 'image',
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
catch (error) {
|
||||||
|
console.error('处理图片失败:', error);
|
||||||
|
ElMessage.error(`${file.name} 处理失败`);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// 处理 Excel 文件
|
||||||
|
else if (isExcel) {
|
||||||
|
try {
|
||||||
|
const result = await parseExcel(file);
|
||||||
|
|
||||||
|
// 动态裁剪内容以适应剩余空间
|
||||||
|
let finalContent = result.content;
|
||||||
|
let wasTruncated = result.totalRows > MAX_EXCEL_ROWS;
|
||||||
|
|
||||||
|
// 如果超过总内容限制,裁剪内容
|
||||||
|
const remainingSpace = MAX_TOTAL_CONTENT_LENGTH - totalContentLength;
|
||||||
|
if (result.content.length > remainingSpace && remainingSpace > 1000) {
|
||||||
|
// 至少保留1000字符才有意义
|
||||||
|
finalContent = result.content.substring(0, remainingSpace);
|
||||||
|
wasTruncated = true;
|
||||||
|
}
|
||||||
|
else if (remainingSpace <= 1000) {
|
||||||
|
const fileSizeKB = (file.size / 1024).toFixed(2);
|
||||||
|
ElMessage.error(`${file.name} (${fileSizeKB}KB) 会超过总长度限制,已跳过`);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
totalContentLength += finalContent.length;
|
||||||
|
|
||||||
|
arr.push({
|
||||||
|
uid: crypto.randomUUID(),
|
||||||
|
name: file.name,
|
||||||
|
fileSize: file.size,
|
||||||
|
file,
|
||||||
|
maxWidth: '200px',
|
||||||
|
showDelIcon: true,
|
||||||
|
imgPreview: false,
|
||||||
|
isUploaded: true,
|
||||||
|
fileContent: finalContent,
|
||||||
|
fileType: 'text',
|
||||||
|
});
|
||||||
|
|
||||||
|
// 提示信息
|
||||||
|
if (wasTruncated) {
|
||||||
|
const fileSizeKB = (file.size / 1024).toFixed(2);
|
||||||
|
ElMessage.warning(`${file.name} (${fileSizeKB}KB) 内容过大,已自动截取部分内容`);
|
||||||
|
}
|
||||||
|
|
||||||
|
console.log(`Excel 解析: ${file.name} - 大小: ${(file.size / 1024).toFixed(2)}KB, 总行数: ${result.totalRows}, 已提取: ${result.extractedRows} 行, 内容长度: ${finalContent.length} 字符`);
|
||||||
|
}
|
||||||
|
catch (error) {
|
||||||
|
console.error('解析 Excel 失败:', error);
|
||||||
|
ElMessage.error(`${file.name} 解析失败`);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// 处理 Word 文档
|
||||||
|
else if (isWord) {
|
||||||
|
try {
|
||||||
|
const result = await parseWord(file);
|
||||||
|
|
||||||
|
// 动态裁剪内容以适应剩余空间
|
||||||
|
let finalContent = result.content;
|
||||||
|
let wasTruncated = result.extracted;
|
||||||
|
|
||||||
|
// 如果超过总内容限制,裁剪内容
|
||||||
|
const remainingSpace = MAX_TOTAL_CONTENT_LENGTH - totalContentLength;
|
||||||
|
if (result.content.length > remainingSpace && remainingSpace > 1000) {
|
||||||
|
finalContent = result.content.substring(0, remainingSpace);
|
||||||
|
wasTruncated = true;
|
||||||
|
}
|
||||||
|
else if (remainingSpace <= 1000) {
|
||||||
|
const fileSizeKB = (file.size / 1024).toFixed(2);
|
||||||
|
ElMessage.error(`${file.name} (${fileSizeKB}KB) 会超过总长度限制,已跳过`);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
totalContentLength += finalContent.length;
|
||||||
|
|
||||||
|
arr.push({
|
||||||
|
uid: crypto.randomUUID(),
|
||||||
|
name: file.name,
|
||||||
|
fileSize: file.size,
|
||||||
|
file,
|
||||||
|
maxWidth: '200px',
|
||||||
|
showDelIcon: true,
|
||||||
|
imgPreview: false,
|
||||||
|
isUploaded: true,
|
||||||
|
fileContent: finalContent,
|
||||||
|
fileType: 'text',
|
||||||
|
});
|
||||||
|
|
||||||
|
// 提示信息
|
||||||
|
if (wasTruncated) {
|
||||||
|
const fileSizeKB = (file.size / 1024).toFixed(2);
|
||||||
|
ElMessage.warning(`${file.name} (${fileSizeKB}KB) 内容过大,已自动截取部分内容`);
|
||||||
|
}
|
||||||
|
|
||||||
|
console.log(`Word 解析: ${file.name} - 大小: ${(file.size / 1024).toFixed(2)}KB, 总长度: ${result.totalLength}, 已提取: ${finalContent.length} 字符`);
|
||||||
|
}
|
||||||
|
catch (error) {
|
||||||
|
console.error('解析 Word 失败:', error);
|
||||||
|
ElMessage.error(`${file.name} 解析失败`);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// 处理 PDF 文件
|
||||||
|
else if (isPDF) {
|
||||||
|
try {
|
||||||
|
const result = await parsePDF(file);
|
||||||
|
|
||||||
|
// 动态裁剪内容以适应剩余空间
|
||||||
|
let finalContent = result.content;
|
||||||
|
let wasTruncated = result.totalPages > MAX_PDF_PAGES;
|
||||||
|
|
||||||
|
// 如果超过总内容限制,裁剪内容
|
||||||
|
const remainingSpace = MAX_TOTAL_CONTENT_LENGTH - totalContentLength;
|
||||||
|
if (result.content.length > remainingSpace && remainingSpace > 1000) {
|
||||||
|
finalContent = result.content.substring(0, remainingSpace);
|
||||||
|
wasTruncated = true;
|
||||||
|
}
|
||||||
|
else if (remainingSpace <= 1000) {
|
||||||
|
const fileSizeKB = (file.size / 1024).toFixed(2);
|
||||||
|
ElMessage.error(`${file.name} (${fileSizeKB}KB) 会超过总长度限制,已跳过`);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
totalContentLength += finalContent.length;
|
||||||
|
|
||||||
|
arr.push({
|
||||||
|
uid: crypto.randomUUID(),
|
||||||
|
name: file.name,
|
||||||
|
fileSize: file.size,
|
||||||
|
file,
|
||||||
|
maxWidth: '200px',
|
||||||
|
showDelIcon: true,
|
||||||
|
imgPreview: false,
|
||||||
|
isUploaded: true,
|
||||||
|
fileContent: finalContent,
|
||||||
|
fileType: 'text',
|
||||||
|
});
|
||||||
|
|
||||||
|
// 提示信息
|
||||||
|
if (wasTruncated) {
|
||||||
|
const fileSizeKB = (file.size / 1024).toFixed(2);
|
||||||
|
ElMessage.warning(`${file.name} (${fileSizeKB}KB) 内容过大,已自动截取部分内容`);
|
||||||
|
}
|
||||||
|
|
||||||
|
console.log(`PDF 解析: ${file.name} - 大小: ${(file.size / 1024).toFixed(2)}KB, 总页数: ${result.totalPages}, 已提取: ${result.extractedPages} 页, 内容长度: ${finalContent.length} 字符`);
|
||||||
|
}
|
||||||
|
catch (error) {
|
||||||
|
console.error('解析 PDF 失败:', error);
|
||||||
|
ElMessage.error(`${file.name} 解析失败`);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// 处理文本文件
|
||||||
|
else if (isText) {
|
||||||
|
try {
|
||||||
|
// 读取文本文件内容
|
||||||
|
const content = await readTextFile(file);
|
||||||
|
|
||||||
|
// 限制单个文本文件长度
|
||||||
|
let finalContent = content;
|
||||||
|
let truncated = false;
|
||||||
|
if (content.length > MAX_TEXT_FILE_LENGTH) {
|
||||||
|
finalContent = content.substring(0, MAX_TEXT_FILE_LENGTH);
|
||||||
|
truncated = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 动态裁剪内容以适应剩余空间
|
||||||
|
const remainingSpace = MAX_TOTAL_CONTENT_LENGTH - totalContentLength;
|
||||||
|
if (finalContent.length > remainingSpace && remainingSpace > 1000) {
|
||||||
|
finalContent = finalContent.substring(0, remainingSpace);
|
||||||
|
truncated = true;
|
||||||
|
}
|
||||||
|
else if (remainingSpace <= 1000) {
|
||||||
|
const fileSizeKB = (file.size / 1024).toFixed(2);
|
||||||
|
ElMessage.error(`${file.name} (${fileSizeKB}KB) 会超过总长度限制,已跳过`);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
totalContentLength += finalContent.length;
|
||||||
|
|
||||||
|
arr.push({
|
||||||
|
uid: crypto.randomUUID(),
|
||||||
|
name: file.name,
|
||||||
|
fileSize: file.size,
|
||||||
|
file,
|
||||||
|
maxWidth: '200px',
|
||||||
|
showDelIcon: true,
|
||||||
|
imgPreview: false,
|
||||||
|
isUploaded: true,
|
||||||
|
fileContent: finalContent,
|
||||||
|
fileType: 'text',
|
||||||
|
});
|
||||||
|
|
||||||
|
// 提示信息
|
||||||
|
if (truncated) {
|
||||||
|
const fileSizeKB = (file.size / 1024).toFixed(2);
|
||||||
|
ElMessage.warning(`${file.name} (${fileSizeKB}KB) 内容过大,已自动截取部分内容`);
|
||||||
|
}
|
||||||
|
|
||||||
|
console.log(`文本文件读取: ${file.name} - 大小: ${(file.size / 1024).toFixed(2)}KB, 内容长度: ${content.length} 字符`);
|
||||||
|
}
|
||||||
|
catch (error) {
|
||||||
|
console.error('读取文件失败:', error);
|
||||||
|
ElMessage.error(`${file.name} 读取失败`);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// 不支持的文件类型
|
||||||
|
else {
|
||||||
|
ElMessage.warning(`${file.name} 不是支持的文件类型`);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (arr.length > 0) {
|
||||||
filesStore.setFilesList([...filesStore.filesList, ...arr]);
|
filesStore.setFilesList([...filesStore.filesList, ...arr]);
|
||||||
|
ElMessage.success(`已添加 ${arr.length} 个文件`);
|
||||||
|
}
|
||||||
|
|
||||||
// 重置文件选择器
|
// 重置文件选择器
|
||||||
nextTick(() => reset());
|
nextTick(() => reset());
|
||||||
});
|
});
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 打开文件选择对话框
|
||||||
|
*/
|
||||||
function handleUploadFiles() {
|
function handleUploadFiles() {
|
||||||
open();
|
open();
|
||||||
popoverRef.value.hide();
|
|
||||||
}
|
}
|
||||||
</script>
|
</script>
|
||||||
|
|
||||||
<template>
|
<template>
|
||||||
<div class="files-select">
|
<div class="files-select">
|
||||||
<Popover
|
<!-- 直接点击上传,添加 tooltip 提示 -->
|
||||||
ref="popoverRef"
|
<el-tooltip
|
||||||
placement="top-start"
|
content="上传文件或图片(支持 Excel、Word、PDF、代码文件等,最大3MB)"
|
||||||
:offset="[4, 0]"
|
placement="top"
|
||||||
popover-class="popover-content"
|
|
||||||
:popover-style="popoverStyle"
|
|
||||||
trigger="clickTarget"
|
|
||||||
>
|
>
|
||||||
<template #trigger>
|
|
||||||
<div
|
<div
|
||||||
class="flex items-center gap-4px p-10px rounded-10px cursor-pointer font-size-14px border-1px border-[rgba(0,0,0,0.08)] border-solid hover:bg-[rgba(0,0,0,.04)]"
|
class="flex items-center gap-4px p-10px rounded-10px cursor-pointer font-size-14px border-1px border-[rgba(0,0,0,0.08)] border-solid hover:bg-[rgba(0,0,0,.04)]"
|
||||||
|
@click="handleUploadFiles"
|
||||||
>
|
>
|
||||||
<el-icon>
|
<el-icon>
|
||||||
<Paperclip />
|
<Paperclip />
|
||||||
</el-icon>
|
</el-icon>
|
||||||
</div>
|
</div>
|
||||||
</template>
|
</el-tooltip>
|
||||||
|
|
||||||
<div class="popover-content-box">
|
|
||||||
<div
|
|
||||||
class="popover-content-item flex items-center gap-4px p-10px rounded-10px cursor-pointer font-size-14px hover:bg-[rgba(0,0,0,.04)]"
|
|
||||||
@click="handleUploadFiles"
|
|
||||||
>
|
|
||||||
<el-icon>
|
|
||||||
<Upload />
|
|
||||||
</el-icon>
|
|
||||||
<div class="font-size-14px">
|
|
||||||
上传文件或图片
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<Popover
|
|
||||||
placement="right-end"
|
|
||||||
:offset="[8, 4]"
|
|
||||||
popover-class="popover-content"
|
|
||||||
:popover-style="popoverStyle"
|
|
||||||
trigger="hover"
|
|
||||||
:hover-delay="100"
|
|
||||||
>
|
|
||||||
<template #trigger>
|
|
||||||
<div
|
|
||||||
class="popover-content-item flex items-center gap-4px p-10px rounded-10px cursor-pointer font-size-14px hover:bg-[rgba(0,0,0,.04)]"
|
|
||||||
>
|
|
||||||
<SvgIcon name="code" size="16" />
|
|
||||||
<div class="font-size-14px">
|
|
||||||
上传代码
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<el-icon class="ml-auto">
|
|
||||||
<ArrowRight />
|
|
||||||
</el-icon>
|
|
||||||
</div>
|
|
||||||
</template>
|
|
||||||
|
|
||||||
<div class="popover-content-box">
|
|
||||||
<div
|
|
||||||
class="popover-content-item flex items-center gap-4px p-10px rounded-10px cursor-pointer font-size-14px hover:bg-[rgba(0,0,0,.04)]"
|
|
||||||
@click="
|
|
||||||
() => {
|
|
||||||
ElMessage.warning('暂未开放');
|
|
||||||
}
|
|
||||||
"
|
|
||||||
>
|
|
||||||
代码文件
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div
|
|
||||||
class="popover-content-item flex items-center gap-4px p-10px rounded-10px cursor-pointer font-size-14px hover:bg-[rgba(0,0,0,.04)]"
|
|
||||||
@click="
|
|
||||||
() => {
|
|
||||||
ElMessage.warning('暂未开放');
|
|
||||||
}
|
|
||||||
"
|
|
||||||
>
|
|
||||||
代码文件夹
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</Popover>
|
|
||||||
</div>
|
|
||||||
</Popover>
|
|
||||||
</div>
|
</div>
|
||||||
</template>
|
</template>
|
||||||
|
|
||||||
|
|||||||
@@ -142,6 +142,7 @@ function cleanupPayment() {
|
|||||||
const tabs = [
|
const tabs = [
|
||||||
{ key: 'member', label: '会员套餐' },
|
{ key: 'member', label: '会员套餐' },
|
||||||
{ key: 'token', label: '尊享Token包' },
|
{ key: 'token', label: '尊享Token包' },
|
||||||
|
{ key: 'activation', label: '激活码' },
|
||||||
];
|
];
|
||||||
|
|
||||||
const benefitsData = {
|
const benefitsData = {
|
||||||
@@ -210,8 +211,11 @@ function selectPackage(pkg: any) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
watch(activeTab, () => {
|
watch(activeTab, () => {
|
||||||
|
if (activeTab.value === 'activation') {
|
||||||
|
return;
|
||||||
|
}
|
||||||
const packages = packagesData.value[activeTab.value as 'member' | 'token'];
|
const packages = packagesData.value[activeTab.value as 'member' | 'token'];
|
||||||
if (packages.length > 0) {
|
if (packages && packages.length > 0) {
|
||||||
const firstPackage = packages[0];
|
const firstPackage = packages[0];
|
||||||
selectedId.value = firstPackage.id;
|
selectedId.value = firstPackage.id;
|
||||||
selectedPrice.value = firstPackage.price;
|
selectedPrice.value = firstPackage.price;
|
||||||
@@ -315,6 +319,11 @@ function close() {
|
|||||||
function onClose() {
|
function onClose() {
|
||||||
emit('close');
|
emit('close');
|
||||||
}
|
}
|
||||||
|
|
||||||
|
function goToActivation() {
|
||||||
|
close();
|
||||||
|
userStore.openUserCenter('activationCode');
|
||||||
|
}
|
||||||
</script>
|
</script>
|
||||||
|
|
||||||
<template>
|
<template>
|
||||||
@@ -386,8 +395,36 @@ function onClose() {
|
|||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
|
<!-- 激活码引导页 -->
|
||||||
|
<div v-if="activeTab === 'activation'" class="activation-guide-container">
|
||||||
|
<div class="activation-content">
|
||||||
|
<div class="guide-icon">
|
||||||
|
🎁
|
||||||
|
</div>
|
||||||
|
<h3 class="guide-title">
|
||||||
|
激活码兑换
|
||||||
|
</h3>
|
||||||
|
<p class="guide-desc">
|
||||||
|
如果您持有意心AI的会员激活码或Token兑换码,<br>请点击下方按钮前往控制台进行兑换。
|
||||||
|
</p>
|
||||||
|
<el-button
|
||||||
|
type="primary"
|
||||||
|
size="large"
|
||||||
|
class="redeem-jump-btn"
|
||||||
|
round
|
||||||
|
@click="goToActivation"
|
||||||
|
>
|
||||||
|
前往兑换中心
|
||||||
|
</el-button>
|
||||||
|
|
||||||
|
<div class="guide-tips">
|
||||||
|
<p>💡 兑换成功后权益将立即生效</p>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
<!-- 移动端布局 -->
|
<!-- 移动端布局 -->
|
||||||
<div v-if="isMobile" class="mobile-layout">
|
<div v-else-if="isMobile" class="mobile-layout">
|
||||||
<!-- 商品加载状态(无修改) -->
|
<!-- 商品加载状态(无修改) -->
|
||||||
<div v-if="isLoadingGoods" class="loading-container">
|
<div v-if="isLoadingGoods" class="loading-container">
|
||||||
<el-icon class="is-loading" :size="40">
|
<el-icon class="is-loading" :size="40">
|
||||||
@@ -510,7 +547,12 @@ function onClose() {
|
|||||||
|
|
||||||
<div style="display: flex;justify-content: space-between;margin-top: 15px;">
|
<div style="display: flex;justify-content: space-between;margin-top: 15px;">
|
||||||
<div>
|
<div>
|
||||||
<p>充值后,加客服微信回复账号名,可专享vip售后服务</p>
|
<p style="color: #f97316;font-weight: 800">
|
||||||
|
全站任意充值,每累计充值10元永久优惠尊享包10元,最高可优惠50元
|
||||||
|
</p>
|
||||||
|
<p style="margin-top: 10px;">
|
||||||
|
充值后,加客服微信回复账号名,可专享vip售后服务
|
||||||
|
</p>
|
||||||
<p style="margin-top: 10px;">
|
<p style="margin-top: 10px;">
|
||||||
客服微信号:chengzilaoge520 或扫描右侧二维码
|
客服微信号:chengzilaoge520 或扫描右侧二维码
|
||||||
</p>
|
</p>
|
||||||
@@ -692,7 +734,13 @@ function onClose() {
|
|||||||
|
|
||||||
<div style="display: flex;justify-content: space-between;margin-top: 15px;">
|
<div style="display: flex;justify-content: space-between;margin-top: 15px;">
|
||||||
<div>
|
<div>
|
||||||
<p>充值后,加客服微信回复账号名,可专享vip售后服务</p>
|
<p style="color: #f97316;font-weight: 800">
|
||||||
|
全站任意充值,每累计充值10元永久优惠尊享包10元,最高可优惠50元
|
||||||
|
</p>
|
||||||
|
|
||||||
|
<p style="margin-top: 10px;">
|
||||||
|
充值后,加客服微信回复账号名,可专享vip售后服务
|
||||||
|
</p>
|
||||||
<p style="margin-top: 10px;">
|
<p style="margin-top: 10px;">
|
||||||
客服微信号:chengzilaoge520 或扫描右侧二维码
|
客服微信号:chengzilaoge520 或扫描右侧二维码
|
||||||
</p>
|
</p>
|
||||||
@@ -813,6 +861,73 @@ function onClose() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/* 激活码引导页样式 */
|
||||||
|
.activation-guide-container {
|
||||||
|
padding: 40px 20px;
|
||||||
|
display: flex;
|
||||||
|
justify-content: center;
|
||||||
|
align-items: center;
|
||||||
|
min-height: 400px;
|
||||||
|
background: linear-gradient(to bottom, #fff, #fdfdfd);
|
||||||
|
border-radius: 8px;
|
||||||
|
|
||||||
|
.activation-content {
|
||||||
|
text-align: center;
|
||||||
|
max-width: 400px;
|
||||||
|
|
||||||
|
.guide-icon {
|
||||||
|
font-size: 64px;
|
||||||
|
margin-bottom: 24px;
|
||||||
|
animation: float-icon 3s ease-in-out infinite;
|
||||||
|
}
|
||||||
|
|
||||||
|
.guide-title {
|
||||||
|
font-size: 24px;
|
||||||
|
font-weight: 800;
|
||||||
|
color: #2c3e50;
|
||||||
|
margin-bottom: 16px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.guide-desc {
|
||||||
|
color: #606266;
|
||||||
|
line-height: 1.6;
|
||||||
|
margin-bottom: 32px;
|
||||||
|
font-size: 15px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.redeem-jump-btn {
|
||||||
|
width: 200px;
|
||||||
|
height: 48px;
|
||||||
|
font-size: 16px;
|
||||||
|
font-weight: bold;
|
||||||
|
letter-spacing: 1px;
|
||||||
|
background: linear-gradient(135deg, #ff9a9e 0%, #fecfef 100%);
|
||||||
|
border: none;
|
||||||
|
box-shadow: 0 4px 12px rgba(255, 117, 140, 0.3);
|
||||||
|
transition: all 0.3s;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
transform: translateY(-2px);
|
||||||
|
box-shadow: 0 6px 16px rgba(255, 117, 140, 0.4);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.guide-tips {
|
||||||
|
margin-top: 24px;
|
||||||
|
font-size: 13px;
|
||||||
|
color: #909399;
|
||||||
|
display: flex;
|
||||||
|
justify-content: center;
|
||||||
|
gap: 8px;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@keyframes float-icon {
|
||||||
|
0%, 100% { transform: translateY(0); }
|
||||||
|
50% { transform: translateY(-10px); }
|
||||||
|
}
|
||||||
|
|
||||||
/* 移动端样式(核心新增:主价格/弱化价格样式) */
|
/* 移动端样式(核心新增:主价格/弱化价格样式) */
|
||||||
.mobile-layout {
|
.mobile-layout {
|
||||||
display: flex; flex-direction: column; gap: 24px;
|
display: flex; flex-direction: column; gap: 24px;
|
||||||
|
|||||||
416
Yi.Ai.Vue3/src/components/ToolList/index.vue
Normal file
416
Yi.Ai.Vue3/src/components/ToolList/index.vue
Normal file
@@ -0,0 +1,416 @@
|
|||||||
|
<script setup lang="ts">
|
||||||
|
import { ChromeFilled, ElementPlus, Loading, MagicStick, Search } from '@element-plus/icons-vue';
|
||||||
|
import { computed, onMounted, ref } from 'vue';
|
||||||
|
import { aiChatTool } from '@/api';
|
||||||
|
|
||||||
|
// API返回的工具接口
|
||||||
|
interface ApiToolItem {
|
||||||
|
name: string;
|
||||||
|
description: string;
|
||||||
|
inputSchema: {
|
||||||
|
type: string;
|
||||||
|
properties: Record<string, any>;
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
// 前端工具按钮的数据
|
||||||
|
interface ToolItem {
|
||||||
|
id: number;
|
||||||
|
name: string;
|
||||||
|
apiName: string;
|
||||||
|
icon: any;
|
||||||
|
tip: string;
|
||||||
|
enabled: boolean;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 定义组件事件
|
||||||
|
const emit = defineEmits<{
|
||||||
|
'tools-update': [payload: {
|
||||||
|
selectedToolIds: number[];
|
||||||
|
selectedApiTools: string[];
|
||||||
|
}];
|
||||||
|
}>();
|
||||||
|
|
||||||
|
// 图标映射配置 - 根据API名称映射图标
|
||||||
|
const iconMap: Record<string, any> = {
|
||||||
|
online_search: ChromeFilled, // 在线搜索
|
||||||
|
deep_think: ElementPlus, // 深度思考
|
||||||
|
search: Search, // 搜索
|
||||||
|
web_search: ChromeFilled, // 网页搜索
|
||||||
|
thinking: MagicStick, // 思考
|
||||||
|
default: ElementPlus, // 默认图标
|
||||||
|
};
|
||||||
|
|
||||||
|
// 提示信息映射
|
||||||
|
const tipMap: Record<string, string> = {
|
||||||
|
online_search: '实时搜索最新信息,获取网络资料',
|
||||||
|
deep_think: '深度推理分析,解决复杂问题',
|
||||||
|
web_search: '联网搜索网页内容',
|
||||||
|
thinking: '深入思考和分析问题',
|
||||||
|
default: '点击启用此功能',
|
||||||
|
};
|
||||||
|
|
||||||
|
// 响应式工具列表 - 初始为空,等待接口加载
|
||||||
|
const tools = ref<ToolItem[]>([]);
|
||||||
|
|
||||||
|
// 当前选中的工具ID数组(支持多选)
|
||||||
|
const activeToolIds = ref<number[]>([]);
|
||||||
|
const loading = ref(false);
|
||||||
|
const error = ref<string | null>(null);
|
||||||
|
|
||||||
|
// 从API数据转换为前端格式
|
||||||
|
function transformApiTools(apiTools: ApiToolItem[]): ToolItem[] {
|
||||||
|
return apiTools.map((tool, index) => {
|
||||||
|
const apiName = tool.name;
|
||||||
|
return {
|
||||||
|
id: index + 1,
|
||||||
|
name: tool.description, // 使用中文描述
|
||||||
|
apiName,
|
||||||
|
icon: iconMap[apiName] || iconMap.default,
|
||||||
|
tip: tipMap[apiName] || tipMap.default,
|
||||||
|
enabled: true,
|
||||||
|
};
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取所有选中的工具对象
|
||||||
|
const selectedTools = computed(() => {
|
||||||
|
return tools.value.filter(tool => activeToolIds.value.includes(tool.id));
|
||||||
|
});
|
||||||
|
|
||||||
|
// 获取所有选中的API工具名称
|
||||||
|
const selectedApiTools = computed(() => {
|
||||||
|
return selectedTools.value.map(tool => tool.apiName);
|
||||||
|
});
|
||||||
|
|
||||||
|
// 点击事件处理
|
||||||
|
function handleToolClick(tool: ToolItem) {
|
||||||
|
if (!tool.enabled)
|
||||||
|
return;
|
||||||
|
|
||||||
|
const index = activeToolIds.value.indexOf(tool.id);
|
||||||
|
|
||||||
|
if (index > -1) {
|
||||||
|
activeToolIds.value.splice(index, 1);
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
activeToolIds.value.push(tool.id);
|
||||||
|
}
|
||||||
|
|
||||||
|
console.log('当前选中的工具:', selectedTools.value.map(t => t.name));
|
||||||
|
console.log('对应的API名称:', selectedApiTools.value);
|
||||||
|
|
||||||
|
emitToolsUpdate();
|
||||||
|
}
|
||||||
|
|
||||||
|
// 辅助函数:检查某个工具是否被选中
|
||||||
|
function isActive(toolId: number) {
|
||||||
|
return activeToolIds.value.includes(toolId);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 清空所有选中
|
||||||
|
function clearSelection() {
|
||||||
|
activeToolIds.value = [];
|
||||||
|
emitToolsUpdate();
|
||||||
|
}
|
||||||
|
|
||||||
|
// 全选已启用的工具
|
||||||
|
function selectAll() {
|
||||||
|
activeToolIds.value = tools.value
|
||||||
|
.filter(tool => tool.enabled)
|
||||||
|
.map(tool => tool.id);
|
||||||
|
emitToolsUpdate();
|
||||||
|
}
|
||||||
|
|
||||||
|
// 启用/禁用工具
|
||||||
|
function setToolEnabled(toolId: number, enabled: boolean) {
|
||||||
|
const tool = tools.value.find(t => t.id === toolId);
|
||||||
|
if (tool) {
|
||||||
|
tool.enabled = enabled;
|
||||||
|
if (!enabled && isActive(toolId)) {
|
||||||
|
const index = activeToolIds.value.indexOf(toolId);
|
||||||
|
if (index > -1) {
|
||||||
|
activeToolIds.value.splice(index, 1);
|
||||||
|
emitToolsUpdate();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 根据API名称启用/禁用工具
|
||||||
|
function setToolEnabledByApiName(apiName: string, enabled: boolean) {
|
||||||
|
const tool = tools.value.find(t => t.apiName === apiName);
|
||||||
|
if (tool) {
|
||||||
|
setToolEnabled(tool.id, enabled);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 通知父组件工具状态变化
|
||||||
|
function emitToolsUpdate() {
|
||||||
|
console.log('工具状态已更新:', {
|
||||||
|
selectedToolIds: activeToolIds.value,
|
||||||
|
selectedTools: selectedTools.value,
|
||||||
|
selectedApiTools: selectedApiTools.value,
|
||||||
|
});
|
||||||
|
|
||||||
|
// 触发自定义事件
|
||||||
|
emit('tools-update', {
|
||||||
|
selectedToolIds: activeToolIds.value,
|
||||||
|
selectedApiTools: selectedApiTools.value,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// 从API获取工具列表
|
||||||
|
async function getAiChatToolList() {
|
||||||
|
loading.value = true;
|
||||||
|
error.value = null;
|
||||||
|
|
||||||
|
try {
|
||||||
|
const res = await aiChatTool();
|
||||||
|
|
||||||
|
if (res.data && Array.isArray(res.data)) {
|
||||||
|
console.log('API返回的工具列表:', res.data);
|
||||||
|
|
||||||
|
// 转换API数据
|
||||||
|
const apiTools = transformApiTools(res.data);
|
||||||
|
|
||||||
|
// 更新工具列表
|
||||||
|
tools.value = apiTools;
|
||||||
|
|
||||||
|
// 默认选中第一个可用的工具
|
||||||
|
const firstEnabledTool = apiTools.find(tool => tool.enabled);
|
||||||
|
if (firstEnabledTool && activeToolIds.value.length === 0) {
|
||||||
|
activeToolIds.value = [firstEnabledTool.id];
|
||||||
|
}
|
||||||
|
|
||||||
|
console.log('加载的工具列表:', tools.value);
|
||||||
|
emitToolsUpdate();
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
error.value = '接口返回数据格式不正确';
|
||||||
|
console.error('接口返回数据格式错误:', res);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch (err) {
|
||||||
|
error.value = '获取工具列表失败,请检查网络连接';
|
||||||
|
console.error('获取工具列表失败:', err);
|
||||||
|
}
|
||||||
|
finally {
|
||||||
|
loading.value = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 刷新工具列表
|
||||||
|
async function refreshTools() {
|
||||||
|
await getAiChatToolList();
|
||||||
|
}
|
||||||
|
|
||||||
|
// 组件挂载时获取工具列表
|
||||||
|
onMounted(() => {
|
||||||
|
getAiChatToolList();
|
||||||
|
});
|
||||||
|
|
||||||
|
// 暴露方法给父组件
|
||||||
|
defineExpose({
|
||||||
|
selectedTools,
|
||||||
|
selectedApiTools,
|
||||||
|
clearSelection,
|
||||||
|
selectAll,
|
||||||
|
setToolEnabled,
|
||||||
|
setToolEnabledByApiName,
|
||||||
|
refreshTools,
|
||||||
|
getAiChatToolList,
|
||||||
|
});
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<template>
|
||||||
|
<div class="tools-container">
|
||||||
|
<!-- 加载状态 -->
|
||||||
|
<div v-if="loading" class="loading-state">
|
||||||
|
<el-icon class="loading-icon">
|
||||||
|
<Loading />
|
||||||
|
</el-icon>
|
||||||
|
<span>加载工具中...</span>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<!-- 错误状态 -->
|
||||||
|
<div v-else-if="error" class="error-state">
|
||||||
|
<el-icon class="error-icon">
|
||||||
|
<ElementPlus />
|
||||||
|
</el-icon>
|
||||||
|
<span>{{ error }}</span>
|
||||||
|
<el-button type="text" class="retry-btn" @click="refreshTools">
|
||||||
|
重试
|
||||||
|
</el-button>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<!-- 空状态 -->
|
||||||
|
<div v-else-if="tools.length === 0" class="empty-state">
|
||||||
|
<span>暂无可用工具</span>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<!-- 工具按钮 -->
|
||||||
|
<div
|
||||||
|
v-for="tool in tools"
|
||||||
|
v-else
|
||||||
|
:key="tool.id"
|
||||||
|
class="tool-item"
|
||||||
|
:class="{
|
||||||
|
active: isActive(tool.id),
|
||||||
|
disabled: !tool.enabled,
|
||||||
|
}"
|
||||||
|
:title="tool.tip"
|
||||||
|
@click="handleToolClick(tool)"
|
||||||
|
>
|
||||||
|
<el-icon class="tool-icon">
|
||||||
|
<component :is="tool.icon" />
|
||||||
|
</el-icon>
|
||||||
|
<span class="tool-text">{{ tool.name }}</span>
|
||||||
|
|
||||||
|
<!-- 选中指示器 -->
|
||||||
|
<!-- <span v-if="isActive(tool.id)" class="check-indicator">✓</span> -->
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<!-- 调试信息 -->
|
||||||
|
<div v-if="false" class="debug-info">
|
||||||
|
<div>工具数量: {{ tools.length }}</div>
|
||||||
|
<div>当前选中ID: {{ activeToolIds }}</div>
|
||||||
|
<div>选中的API工具: {{ selectedApiTools }}</div>
|
||||||
|
</div>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<style scoped lang="scss">
|
||||||
|
.tools-container {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: row;
|
||||||
|
gap: 8px;
|
||||||
|
padding: 4px;
|
||||||
|
flex-wrap: wrap;
|
||||||
|
align-items: center;
|
||||||
|
min-height: 44px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.loading-state,
|
||||||
|
.error-state,
|
||||||
|
.empty-state {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 8px;
|
||||||
|
padding: 8px 12px;
|
||||||
|
font-size: 14px;
|
||||||
|
color: #909399;
|
||||||
|
}
|
||||||
|
|
||||||
|
.loading-state {
|
||||||
|
.loading-icon {
|
||||||
|
animation: rotate 1s linear infinite;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.error-state {
|
||||||
|
color: #f56c6c;
|
||||||
|
|
||||||
|
.error-icon {
|
||||||
|
color: #f56c6c;
|
||||||
|
}
|
||||||
|
|
||||||
|
.retry-btn {
|
||||||
|
margin-left: 8px;
|
||||||
|
padding: 0;
|
||||||
|
height: auto;
|
||||||
|
color: #409EFF;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.empty-state {
|
||||||
|
color: #c0c4cc;
|
||||||
|
font-style: italic;
|
||||||
|
}
|
||||||
|
|
||||||
|
@keyframes rotate {
|
||||||
|
from { transform: rotate(0deg); }
|
||||||
|
to { transform: rotate(360deg); }
|
||||||
|
}
|
||||||
|
|
||||||
|
.tool-item {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 6px;
|
||||||
|
padding: 10px 6px;
|
||||||
|
cursor: pointer;
|
||||||
|
font-size: 14px;
|
||||||
|
border: 1px solid rgba(0, 0, 0, 0.08);
|
||||||
|
transition: all 0.2s ease;
|
||||||
|
background: white;
|
||||||
|
position: relative;
|
||||||
|
border-radius: 10px;
|
||||||
|
user-select: none;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: rgba(0, 0, 0, 0.04);
|
||||||
|
transform: translateY(-1px);
|
||||||
|
box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1);
|
||||||
|
}
|
||||||
|
|
||||||
|
&.active {
|
||||||
|
background: rgba(64, 158, 255, 0.1);
|
||||||
|
border-color: #409EFF;
|
||||||
|
color: #409EFF;
|
||||||
|
|
||||||
|
.tool-icon {
|
||||||
|
color: #409EFF;
|
||||||
|
}
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: rgba(64, 158, 255, 0.15);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
&.disabled {
|
||||||
|
opacity: 0.5;
|
||||||
|
cursor: not-allowed;
|
||||||
|
|
||||||
|
&:hover {
|
||||||
|
background: white;
|
||||||
|
transform: none;
|
||||||
|
box-shadow: none;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
&:active:not(.disabled) {
|
||||||
|
transform: translateY(0);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.tool-icon {
|
||||||
|
font-size: 16px;
|
||||||
|
color: #606266;
|
||||||
|
transition: color 0.2s ease;
|
||||||
|
}
|
||||||
|
|
||||||
|
.tool-text {
|
||||||
|
white-space: nowrap;
|
||||||
|
font-size: 13px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.check-indicator {
|
||||||
|
margin-left: 4px;
|
||||||
|
font-weight: bold;
|
||||||
|
font-size: 12px;
|
||||||
|
color: #409EFF;
|
||||||
|
}
|
||||||
|
|
||||||
|
.debug-info {
|
||||||
|
margin-top: 10px;
|
||||||
|
padding: 10px;
|
||||||
|
background: #f5f5f5;
|
||||||
|
border-radius: 4px;
|
||||||
|
font-size: 12px;
|
||||||
|
color: #666;
|
||||||
|
|
||||||
|
div {
|
||||||
|
margin-bottom: 4px;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
</style>
|
||||||
@@ -633,19 +633,10 @@ onMounted(async () => {
|
|||||||
target="_blank"
|
target="_blank"
|
||||||
class="guide-link"
|
class="guide-link"
|
||||||
>
|
>
|
||||||
【意心Ai】AI工具玩法指南
|
点击前往文档:【意心Ai】AI工具玩法指南
|
||||||
<el-icon class="link-icon"><i-ep-top-right /></el-icon>
|
<el-icon class="link-icon"><i-ep-top-right /></el-icon>
|
||||||
</a>
|
</a>
|
||||||
</h1>
|
</h1>
|
||||||
|
|
||||||
<div class="iframe-container">
|
|
||||||
<iframe
|
|
||||||
src="https://ccnetcore.com/article/3a1bc4d1-6a7d-751d-91cc-2817eb2ddcde"
|
|
||||||
class="guide-iframe"
|
|
||||||
loading="lazy"
|
|
||||||
sandbox="allow-scripts allow-same-origin allow-popups"
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
</div>
|
</div>
|
||||||
</el-card>
|
</el-card>
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,483 @@
|
|||||||
|
<script lang="ts" setup>
|
||||||
|
import { ref, onMounted, onUnmounted, nextTick } from 'vue';
|
||||||
|
import { ElMessage } from 'element-plus';
|
||||||
|
import { redeemActivationCode } from '@/api/activationCode';
|
||||||
|
import { MagicStick } from '@element-plus/icons-vue';
|
||||||
|
|
||||||
|
const activationCode = ref('');
|
||||||
|
const loading = ref(false);
|
||||||
|
const canvasRef = ref<HTMLCanvasElement | null>(null);
|
||||||
|
const containerRef = ref<HTMLElement | null>(null);
|
||||||
|
let animationId: number;
|
||||||
|
|
||||||
|
// --- Advanced Physics & Visuals ---
|
||||||
|
|
||||||
|
class Particle {
|
||||||
|
x: number;
|
||||||
|
y: number;
|
||||||
|
vx: number;
|
||||||
|
vy: number;
|
||||||
|
alpha: number;
|
||||||
|
color: string;
|
||||||
|
hue: number;
|
||||||
|
size: number;
|
||||||
|
decay: number;
|
||||||
|
gravity: number;
|
||||||
|
friction: number;
|
||||||
|
brightness: number;
|
||||||
|
flicker: boolean;
|
||||||
|
|
||||||
|
constructor(x: number, y: number, hue: number) {
|
||||||
|
this.x = x;
|
||||||
|
this.y = y;
|
||||||
|
this.hue = hue;
|
||||||
|
|
||||||
|
// Explosive physics
|
||||||
|
const angle = Math.random() * Math.PI * 2;
|
||||||
|
const speed = Math.random() * 15 + 2;
|
||||||
|
this.vx = Math.cos(angle) * speed;
|
||||||
|
this.vy = Math.sin(angle) * speed;
|
||||||
|
|
||||||
|
this.alpha = 1;
|
||||||
|
this.decay = Math.random() * 0.015 + 0.005;
|
||||||
|
this.gravity = 0.05;
|
||||||
|
this.friction = 0.96;
|
||||||
|
|
||||||
|
this.size = Math.random() * 3 + 1;
|
||||||
|
this.brightness = 50; // Standard brightness for white bg visibility (0-100% HSL L value)
|
||||||
|
this.flicker = Math.random() > 0.5;
|
||||||
|
}
|
||||||
|
|
||||||
|
update() {
|
||||||
|
this.vx *= this.friction;
|
||||||
|
this.vy *= this.friction;
|
||||||
|
this.vy += this.gravity;
|
||||||
|
|
||||||
|
this.x += this.vx;
|
||||||
|
this.y += this.vy;
|
||||||
|
|
||||||
|
this.alpha -= this.decay;
|
||||||
|
this.hue += 0.5;
|
||||||
|
}
|
||||||
|
|
||||||
|
draw(ctx: CanvasRenderingContext2D) {
|
||||||
|
ctx.globalAlpha = this.alpha;
|
||||||
|
// On white background:
|
||||||
|
// We want high saturation (100%) and medium lightness (50%) to make colors pop against white.
|
||||||
|
// If lightness is too high (like 80-100), it fades into white.
|
||||||
|
const lightness = this.flicker ? Math.random() * 20 + 40 : this.brightness;
|
||||||
|
ctx.fillStyle = `hsla(${this.hue}, 100%, ${lightness}%, ${this.alpha})`;
|
||||||
|
|
||||||
|
ctx.beginPath();
|
||||||
|
ctx.arc(this.x, this.y, this.size, 0, Math.PI * 2);
|
||||||
|
ctx.fill();
|
||||||
|
ctx.globalAlpha = 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
class Shockwave {
|
||||||
|
x: number;
|
||||||
|
y: number;
|
||||||
|
radius: number;
|
||||||
|
alpha: number;
|
||||||
|
lineWidth: number;
|
||||||
|
hue: number;
|
||||||
|
|
||||||
|
constructor(x: number, y: number, hue: number) {
|
||||||
|
this.x = x;
|
||||||
|
this.y = y;
|
||||||
|
this.hue = hue;
|
||||||
|
this.radius = 0;
|
||||||
|
this.alpha = 1;
|
||||||
|
this.lineWidth = 10;
|
||||||
|
}
|
||||||
|
|
||||||
|
update() {
|
||||||
|
this.radius += 15;
|
||||||
|
this.alpha -= 0.05;
|
||||||
|
this.lineWidth -= 0.2;
|
||||||
|
}
|
||||||
|
|
||||||
|
draw(ctx: CanvasRenderingContext2D) {
|
||||||
|
if (this.alpha <= 0) return;
|
||||||
|
ctx.save();
|
||||||
|
ctx.beginPath();
|
||||||
|
ctx.arc(this.x, this.y, this.radius, 0, Math.PI * 2);
|
||||||
|
// Darker/More saturated shockwave for white background
|
||||||
|
ctx.strokeStyle = `hsla(${this.hue}, 100%, 60%, ${this.alpha})`;
|
||||||
|
ctx.lineWidth = Math.max(0, this.lineWidth);
|
||||||
|
ctx.stroke();
|
||||||
|
ctx.restore();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let particles: Particle[] = [];
|
||||||
|
let shockwaves: Shockwave[] = [];
|
||||||
|
|
||||||
|
function createExplosion(x: number, y: number, hue: number) {
|
||||||
|
const particleCount = 120;
|
||||||
|
for (let i = 0; i < particleCount; i++) {
|
||||||
|
particles.push(new Particle(x, y, hue));
|
||||||
|
}
|
||||||
|
for (let i = 0; i < 60; i++) {
|
||||||
|
particles.push(new Particle(x, y, (hue + 180) % 360));
|
||||||
|
}
|
||||||
|
shockwaves.push(new Shockwave(x, y, hue));
|
||||||
|
}
|
||||||
|
|
||||||
|
function animate() {
|
||||||
|
const canvas = canvasRef.value;
|
||||||
|
if (!canvas) return;
|
||||||
|
const ctx = canvas.getContext('2d');
|
||||||
|
if (!ctx) return;
|
||||||
|
|
||||||
|
// Clear with transparent fade for trails on white
|
||||||
|
// 'destination-out' erases content.
|
||||||
|
// To leave a trail on a white background (canvas is transparent over white gradient):
|
||||||
|
// We need to gently erase what's there.
|
||||||
|
|
||||||
|
ctx.globalCompositeOperation = 'destination-out';
|
||||||
|
ctx.fillStyle = 'rgba(255, 255, 255, 0.1)'; // Alpha controls trail length
|
||||||
|
ctx.fillRect(0, 0, canvas.width, canvas.height);
|
||||||
|
|
||||||
|
// Reset to default drawing
|
||||||
|
ctx.globalCompositeOperation = 'source-over';
|
||||||
|
|
||||||
|
for (let i = particles.length - 1; i >= 0; i--) {
|
||||||
|
particles[i].update();
|
||||||
|
particles[i].draw(ctx);
|
||||||
|
if (particles[i].alpha <= 0) particles.splice(i, 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
for (let i = shockwaves.length - 1; i >= 0; i--) {
|
||||||
|
shockwaves[i].update();
|
||||||
|
shockwaves[i].draw(ctx);
|
||||||
|
if (shockwaves[i].alpha <= 0) shockwaves.splice(i, 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (particles.length > 0 || shockwaves.length > 0) {
|
||||||
|
animationId = requestAnimationFrame(animate);
|
||||||
|
} else {
|
||||||
|
cancelAnimationFrame(animationId);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Shake Effect
|
||||||
|
const isShaking = ref(false);
|
||||||
|
function triggerShake() {
|
||||||
|
isShaking.value = true;
|
||||||
|
setTimeout(() => isShaking.value = false, 500);
|
||||||
|
}
|
||||||
|
|
||||||
|
function triggerCelebration() {
|
||||||
|
const canvas = canvasRef.value;
|
||||||
|
if (!canvas) return;
|
||||||
|
const parent = canvas.parentElement;
|
||||||
|
if (parent) {
|
||||||
|
canvas.width = parent.clientWidth;
|
||||||
|
canvas.height = parent.clientHeight;
|
||||||
|
}
|
||||||
|
|
||||||
|
const cx = canvas.width / 2;
|
||||||
|
const cy = canvas.height / 2;
|
||||||
|
|
||||||
|
// 1. Initial Mega Explosion
|
||||||
|
triggerShake();
|
||||||
|
createExplosion(cx, cy, Math.random() * 360);
|
||||||
|
|
||||||
|
// Start loop
|
||||||
|
animate();
|
||||||
|
|
||||||
|
// 2. Machine Gun Fire sequence
|
||||||
|
let count = 0;
|
||||||
|
const timer = setInterval(() => {
|
||||||
|
count++;
|
||||||
|
const rx = cx + (Math.random() - 0.5) * canvas.width * 0.8;
|
||||||
|
const ry = cy + (Math.random() - 0.5) * canvas.height * 0.8;
|
||||||
|
|
||||||
|
createExplosion(rx, ry, Math.random() * 360);
|
||||||
|
|
||||||
|
if (count % 3 === 0) triggerShake();
|
||||||
|
|
||||||
|
if (count > 25) {
|
||||||
|
clearInterval(timer);
|
||||||
|
}
|
||||||
|
}, 120);
|
||||||
|
}
|
||||||
|
|
||||||
|
async function handleRedeem() {
|
||||||
|
if (!activationCode.value.trim()) {
|
||||||
|
ElMessage.warning('请输入激活码');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
loading.value = true;
|
||||||
|
try {
|
||||||
|
const res = await redeemActivationCode({ code: activationCode.value });
|
||||||
|
triggerCelebration();
|
||||||
|
ElMessage.success({
|
||||||
|
message: '兑换成功!奖励已到账',
|
||||||
|
type: 'success',
|
||||||
|
duration: 3000,
|
||||||
|
showClose: true,
|
||||||
|
});
|
||||||
|
activationCode.value = '';
|
||||||
|
} catch (error: any) {
|
||||||
|
// console.error(error);
|
||||||
|
} finally {
|
||||||
|
loading.value = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
onUnmounted(() => {
|
||||||
|
cancelAnimationFrame(animationId);
|
||||||
|
});
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<template>
|
||||||
|
<div ref="containerRef" class="activation-container" :class="{ 'shake-anim': isShaking }">
|
||||||
|
<!-- Removed Dark overlay -->
|
||||||
|
<canvas ref="canvasRef" class="fireworks-canvas"></canvas>
|
||||||
|
|
||||||
|
<div class="content-wrapper">
|
||||||
|
<div class="gift-icon-wrapper">
|
||||||
|
<div class="gift-box">🎁</div>
|
||||||
|
<div class="gift-glow"></div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<h2 class="title">激活码兑换</h2>
|
||||||
|
<p class="subtitle">开启您的专属惊喜权益</p>
|
||||||
|
|
||||||
|
<div class="input-section">
|
||||||
|
<el-input
|
||||||
|
v-model="activationCode"
|
||||||
|
placeholder="请输入您的激活码"
|
||||||
|
class="activation-input"
|
||||||
|
size="large"
|
||||||
|
:prefix-icon="MagicStick"
|
||||||
|
clearable
|
||||||
|
@keyup.enter="handleRedeem"
|
||||||
|
/>
|
||||||
|
|
||||||
|
<el-button
|
||||||
|
class="redeem-btn"
|
||||||
|
:loading="loading"
|
||||||
|
@click="handleRedeem"
|
||||||
|
>
|
||||||
|
立即兑换
|
||||||
|
</el-button>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="tips-section">
|
||||||
|
<div class="tip-item">
|
||||||
|
<span class="tip-dot">•</span>
|
||||||
|
<span>若激活码内包含意心Ai会员物品,激活后需重新登录生效</span>
|
||||||
|
</div>
|
||||||
|
<div class="tip-item">
|
||||||
|
<span class="tip-dot">•</span>
|
||||||
|
<span>激活成功后,可在充值记录中查看物品是否到账</span>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<style scoped lang="scss">
|
||||||
|
.activation-container {
|
||||||
|
position: relative;
|
||||||
|
width: 100%;
|
||||||
|
height: 100%;
|
||||||
|
min-height: 500px;
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
background: radial-gradient(circle at center, #fffbf0 0%, #fff 100%);
|
||||||
|
border-radius: 12px;
|
||||||
|
overflow: hidden;
|
||||||
|
transition: transform 0.1s;
|
||||||
|
}
|
||||||
|
|
||||||
|
.shake-anim {
|
||||||
|
animation: shake 0.5s cubic-bezier(.36,.07,.19,.97) both;
|
||||||
|
}
|
||||||
|
|
||||||
|
@keyframes shake {
|
||||||
|
10%, 90% { transform: translate3d(-2px, 0, 0); }
|
||||||
|
20%, 80% { transform: translate3d(4px, 0, 0); }
|
||||||
|
30%, 50%, 70% { transform: translate3d(-6px, 0, 0); }
|
||||||
|
40%, 60% { transform: translate3d(6px, 0, 0); }
|
||||||
|
}
|
||||||
|
|
||||||
|
.fireworks-canvas {
|
||||||
|
position: absolute;
|
||||||
|
top: 0;
|
||||||
|
left: 0;
|
||||||
|
width: 100%;
|
||||||
|
height: 100%;
|
||||||
|
pointer-events: none;
|
||||||
|
z-index: 10;
|
||||||
|
}
|
||||||
|
|
||||||
|
.content-wrapper {
|
||||||
|
position: relative;
|
||||||
|
z-index: 11;
|
||||||
|
width: 100%;
|
||||||
|
max-width: 480px;
|
||||||
|
padding: 40px;
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
align-items: center;
|
||||||
|
text-align: center;
|
||||||
|
}
|
||||||
|
|
||||||
|
.gift-icon-wrapper {
|
||||||
|
position: relative;
|
||||||
|
margin-bottom: 24px;
|
||||||
|
animation: float 3s ease-in-out infinite;
|
||||||
|
}
|
||||||
|
|
||||||
|
.gift-box {
|
||||||
|
font-size: 72px;
|
||||||
|
filter: drop-shadow(0 10px 15px rgba(255, 105, 180, 0.3));
|
||||||
|
position: relative;
|
||||||
|
z-index: 2;
|
||||||
|
}
|
||||||
|
|
||||||
|
.gift-glow {
|
||||||
|
position: absolute;
|
||||||
|
top: 50%;
|
||||||
|
left: 50%;
|
||||||
|
transform: translate(-50%, -50%);
|
||||||
|
width: 100px;
|
||||||
|
height: 100px;
|
||||||
|
background: radial-gradient(circle, rgba(255, 215, 0, 0.4) 0%, rgba(255, 255, 255, 0) 70%);
|
||||||
|
border-radius: 50%;
|
||||||
|
z-index: 1;
|
||||||
|
animation: pulse-glow 2s infinite;
|
||||||
|
}
|
||||||
|
|
||||||
|
@keyframes pulse-glow {
|
||||||
|
0% { transform: translate(-50%, -50%) scale(1); opacity: 0.5; }
|
||||||
|
50% { transform: translate(-50%, -50%) scale(1.5); opacity: 1; }
|
||||||
|
100% { transform: translate(-50%, -50%) scale(1); opacity: 0.5; }
|
||||||
|
}
|
||||||
|
|
||||||
|
@keyframes float {
|
||||||
|
0%, 100% { transform: translateY(0); }
|
||||||
|
50% { transform: translateY(-12px); }
|
||||||
|
}
|
||||||
|
|
||||||
|
.title {
|
||||||
|
font-size: 32px;
|
||||||
|
font-weight: 800;
|
||||||
|
background: linear-gradient(45deg, #ff9a9e, #fad0c4, #fad0c4);
|
||||||
|
-webkit-background-clip: text;
|
||||||
|
background-clip: text;
|
||||||
|
color: #2c3e50;
|
||||||
|
margin: 0 0 8px 0;
|
||||||
|
letter-spacing: 2px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.subtitle {
|
||||||
|
font-size: 15px;
|
||||||
|
color: #95a5a6;
|
||||||
|
margin-bottom: 36px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.input-section {
|
||||||
|
width: 100%;
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 24px;
|
||||||
|
margin-bottom: 40px;
|
||||||
|
}
|
||||||
|
|
||||||
|
:deep(.activation-input .el-input__wrapper) {
|
||||||
|
border-radius: 50px;
|
||||||
|
padding: 10px 24px;
|
||||||
|
box-shadow: 0 4px 12px rgba(0, 0, 0, 0.05);
|
||||||
|
border: 2px solid transparent;
|
||||||
|
background-image: linear-gradient(white, white), linear-gradient(to right, #e0e0e0, #e0e0e0);
|
||||||
|
background-origin: border-box;
|
||||||
|
background-clip: padding-box, border-box;
|
||||||
|
transition: all 0.3s ease;
|
||||||
|
}
|
||||||
|
|
||||||
|
:deep(.activation-input .el-input__wrapper:hover),
|
||||||
|
:deep(.activation-input .el-input__wrapper.is-focus) {
|
||||||
|
box-shadow: 0 8px 25px rgba(255, 105, 180, 0.15);
|
||||||
|
background-image: linear-gradient(white, white), linear-gradient(135deg, #ff9a9e, #a18cd1);
|
||||||
|
border: 2px solid transparent;
|
||||||
|
}
|
||||||
|
|
||||||
|
:deep(.activation-input .el-input__inner) {
|
||||||
|
text-align: center;
|
||||||
|
font-size: 18px;
|
||||||
|
letter-spacing: 2px;
|
||||||
|
color: #333;
|
||||||
|
font-weight: 600;
|
||||||
|
}
|
||||||
|
|
||||||
|
.redeem-btn {
|
||||||
|
width: 100%;
|
||||||
|
height: 54px;
|
||||||
|
border-radius: 50px;
|
||||||
|
font-size: 20px;
|
||||||
|
font-weight: 800;
|
||||||
|
letter-spacing: 4px;
|
||||||
|
color: white;
|
||||||
|
border: none;
|
||||||
|
background: linear-gradient(135deg, #ff9a9e 0%, #fecfef 50%, #a18cd1 100%);
|
||||||
|
background-size: 200% 200%;
|
||||||
|
animation: gradient-anim 5s ease infinite;
|
||||||
|
box-shadow: 0 8px 20px rgba(255, 117, 140, 0.4);
|
||||||
|
transition: all 0.3s ease;
|
||||||
|
}
|
||||||
|
|
||||||
|
@keyframes gradient-anim {
|
||||||
|
0% { background-position: 0% 50%; }
|
||||||
|
50% { background-position: 100% 50%; }
|
||||||
|
100% { background-position: 0% 50%; }
|
||||||
|
}
|
||||||
|
|
||||||
|
.redeem-btn:hover {
|
||||||
|
transform: translateY(-3px) scale(1.02);
|
||||||
|
box-shadow: 0 12px 30px rgba(161, 140, 209, 0.6);
|
||||||
|
}
|
||||||
|
|
||||||
|
.redeem-btn:active {
|
||||||
|
transform: translateY(1px);
|
||||||
|
}
|
||||||
|
|
||||||
|
.tips-section {
|
||||||
|
width: 100%;
|
||||||
|
background: rgba(255, 255, 255, 0.8);
|
||||||
|
border-radius: 16px;
|
||||||
|
padding: 20px;
|
||||||
|
border: 1px dashed #dcdfe6;
|
||||||
|
backdrop-filter: blur(5px);
|
||||||
|
}
|
||||||
|
|
||||||
|
.tip-item {
|
||||||
|
display: flex;
|
||||||
|
align-items: flex-start;
|
||||||
|
gap: 10px;
|
||||||
|
font-size: 13px;
|
||||||
|
color: #606266;
|
||||||
|
line-height: 1.6;
|
||||||
|
margin-bottom: 6px;
|
||||||
|
text-align: left;
|
||||||
|
}
|
||||||
|
|
||||||
|
.tip-item:last-child {
|
||||||
|
margin-bottom: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.tip-dot {
|
||||||
|
color: #ff9a9e;
|
||||||
|
font-weight: bold;
|
||||||
|
font-size: 18px;
|
||||||
|
line-height: 14px;
|
||||||
|
}
|
||||||
|
</style>
|
||||||
@@ -797,7 +797,7 @@ function generateShareContent(): string {
|
|||||||
👉 点击链接立即参与我的专属邀请码链接:
|
👉 点击链接立即参与我的专属邀请码链接:
|
||||||
${shareLink}
|
${shareLink}
|
||||||
|
|
||||||
🍀 未注册用户,微信扫码登录,进入用户中心👉每周邀请 即可立即参与!`;
|
🍀 未注册用户,微信扫码登录,进入控制台👉每周邀请 即可立即参与!`;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
|||||||
@@ -549,6 +549,7 @@ onBeforeUnmount(() => {
|
|||||||
</el-option>
|
</el-option>
|
||||||
</el-select>
|
</el-select>
|
||||||
<el-button
|
<el-button
|
||||||
|
v-if="false"
|
||||||
:icon="FullScreen"
|
:icon="FullScreen"
|
||||||
circle
|
circle
|
||||||
plain
|
plain
|
||||||
|
|||||||
@@ -105,6 +105,8 @@ function bindWechat() {
|
|||||||
|
|
||||||
<template>
|
<template>
|
||||||
<div class="user-profile">
|
<div class="user-profile">
|
||||||
|
<!-- 用户卡片 -->
|
||||||
|
<el-card class="profile-card" shadow="hover">
|
||||||
<!-- 顶部标题 -->
|
<!-- 顶部标题 -->
|
||||||
<div class="header">
|
<div class="header">
|
||||||
<h2>
|
<h2>
|
||||||
@@ -112,9 +114,6 @@ function bindWechat() {
|
|||||||
个人信息
|
个人信息
|
||||||
</h2>
|
</h2>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<!-- 用户卡片 -->
|
|
||||||
<el-card class="profile-card" shadow="hover">
|
|
||||||
<!-- 头像和基本信息区域 -->
|
<!-- 头像和基本信息区域 -->
|
||||||
<div class="user-header-section">
|
<div class="user-header-section">
|
||||||
<!-- 头像区域 -->
|
<!-- 头像区域 -->
|
||||||
@@ -138,7 +137,9 @@ function bindWechat() {
|
|||||||
|
|
||||||
<!-- 用户名称和状态 -->
|
<!-- 用户名称和状态 -->
|
||||||
<div class="user-info-quick">
|
<div class="user-info-quick">
|
||||||
<h3 class="user-name">{{ userNick }}</h3>
|
<h3 class="user-name">
|
||||||
|
{{ userNick }}
|
||||||
|
</h3>
|
||||||
<div class="user-tags">
|
<div class="user-tags">
|
||||||
<el-tag v-if="userVipStatus" type="warning" effect="dark" size="large">
|
<el-tag v-if="userVipStatus" type="warning" effect="dark" size="large">
|
||||||
<el-icon><Promotion /></el-icon>
|
<el-icon><Promotion /></el-icon>
|
||||||
@@ -153,8 +154,12 @@ function bindWechat() {
|
|||||||
</div>
|
</div>
|
||||||
<div class="user-stats">
|
<div class="user-stats">
|
||||||
<div class="stat-item">
|
<div class="stat-item">
|
||||||
<div class="stat-value">{{ formatDate(user.creationTime)?.split(' ')[0] || '-' }}</div>
|
<div class="stat-value">
|
||||||
<div class="stat-label">注册时间</div>
|
{{ formatDate(user.creationTime)?.split(' ')[0] || '-' }}
|
||||||
|
</div>
|
||||||
|
<div class="stat-label">
|
||||||
|
注册时间
|
||||||
|
</div>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
@@ -171,7 +176,9 @@ function bindWechat() {
|
|||||||
<el-icon><User /></el-icon>
|
<el-icon><User /></el-icon>
|
||||||
用户名
|
用户名
|
||||||
</div>
|
</div>
|
||||||
<div class="info-value">{{ user.userName || '-' }}</div>
|
<div class="info-value">
|
||||||
|
{{ user.userName || '-' }}
|
||||||
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<!-- 昵称 -->
|
<!-- 昵称 -->
|
||||||
@@ -180,7 +187,9 @@ function bindWechat() {
|
|||||||
<el-icon><Postcard /></el-icon>
|
<el-icon><Postcard /></el-icon>
|
||||||
昵称
|
昵称
|
||||||
</div>
|
</div>
|
||||||
<div class="info-value">{{ userNick }}</div>
|
<div class="info-value">
|
||||||
|
{{ userNick }}
|
||||||
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<!-- 邮箱 -->
|
<!-- 邮箱 -->
|
||||||
@@ -215,7 +224,9 @@ function bindWechat() {
|
|||||||
<!-- 微信绑定 -->
|
<!-- 微信绑定 -->
|
||||||
<div class="info-item full-width">
|
<div class="info-item full-width">
|
||||||
<div class="info-label">
|
<div class="info-label">
|
||||||
<el-icon color="#07C160"><ChatDotRound /></el-icon>
|
<el-icon color="#07C160">
|
||||||
|
<ChatDotRound />
|
||||||
|
</el-icon>
|
||||||
微信绑定
|
微信绑定
|
||||||
</div>
|
</div>
|
||||||
<div class="info-value wechat-binding">
|
<div class="info-value wechat-binding">
|
||||||
|
|||||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user