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

feat: ai完成stock模块搭建

Browse Source
This commit is contained in:
橙子
2025-03-08 22:14:26 +08:00
parent 337088c908
commit 82865631fc
26 changed files with 1044 additions and 294 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
Yi.Abp.Net8/module/ai-stock/Yi.Framework.Stock.Application.Contracts/Dtos/StockMarket/CreateStockMarketInputDto.cs Normal file
View File

@@ -0,0 +1,23 @@
using System;
namespace Yi.Framework.Stock.Application.Contracts.Dtos.StockMarket;
/// <summary>
/// 创建股市输入DTO
/// </summary>
public class CreateStockMarketInputDto
{
/// <summary>
/// 股市代码
/// </summary>
public string MarketCode { get; set; }
/// <summary>
/// 股市名称
/// </summary>
public string MarketName { get; set; }
/// <summary>
/// 股市描述
/// </summary>
public string Description { get; set; }
}

5
Yi.Abp.Net8/module/ai-stock/Yi.Framework.Stock.Application.Contracts/Dtos/StockPrice/StockPriceRecordDto.cs
View File

@@ -38,5 +38,10 @@ namespace Yi.Framework.Stock.Application.Contracts.Dtos.StockPrice
/// 时间周期类型
/// </summary>
public PeriodTypeEnum PeriodType { get; set; }
/// <summary>
/// 记录时间
/// </summary>
public DateTime RecordTime { get; set; }
}
}

4
Yi.Abp.Net8/module/ai-stock/Yi.Framework.Stock.Application.Contracts/Dtos/StockTransaction/StockTransactionGetListInputDto.cs
View File

@@ -12,12 +12,12 @@ namespace Yi.Framework.Stock.Application.Contracts.Dtos.StockTransaction
/// <summary>
/// 股票代码
/// </summary>
public string StockCode { get; set; }
public string? StockCode { get; set; }
/// <summary>
/// 股票名称
/// </summary>
public string StockName { get; set; }
public string? StockName { get; set; }
/// <summary>
/// 交易类型

6
Yi.Abp.Net8/module/ai-stock/Yi.Framework.Stock.Application.Contracts/IServices/IStockMarketService.cs
View File

@@ -39,5 +39,11 @@ namespace Yi.Framework.Stock.Application.Contracts.IServices
/// <param name="input">卖出股票参数</param>
/// <returns>操作结果</returns>
Task SellStockAsync(SellStockInputDto input);
/// <summary>
/// 生成最新股票记录
/// </summary>
/// <returns>操作结果</returns>
Task GenerateStocksAsync();
}
}

1
Yi.Abp.Net8/module/ai-stock/Yi.Framework.Stock.Application.Contracts/Yi.Framework.Stock.Application.Contracts.csproj
View File

@@ -9,7 +9,6 @@
<PackageReference Include="Volo.Abp.SettingManagement.Application.Contracts" Version="$(AbpVersion)" />
</ItemGroup>
<ItemGroup>
<Folder Include="Dtos\" />
<Folder Include="IServices\" />
</ItemGroup>

30
Yi.Abp.Net8/module/ai-stock/Yi.Framework.Stock.Application/Services/StockMarketService.cs
View File

@@ -12,6 +12,7 @@ using Yi.Framework.Stock.Application.Contracts.IServices;
using Yi.Framework.Stock.Domain.Entities;
using Yi.Framework.SqlSugarCore.Abstractions;
using Yi.Framework.Stock.Domain.Managers;
using Mapster;
namespace Yi.Framework.Stock.Application.Services
{
@@ -34,6 +35,23 @@ namespace Yi.Framework.Stock.Application.Services
_stockMarketManager = stockMarketManager;
}
/// <summary>
/// 创建股市
/// </summary>
[HttpPost("stock/markets")]
[Authorize]
public async Task<StockMarketDto> CreateStockMarketAsync(CreateStockMarketInputDto input)
{
// 使用映射将输入DTO转换为实体
var stockMarket = input.Adapt<StockMarketAggregateRoot>();
// 保存到数据库
var result = await _stockMarketRepository.InsertReturnEntityAsync(stockMarket);
// 使用映射将实体转换为返回DTO
return result.Adapt<StockMarketDto>();
}
/// <summary>
/// 获取股市列表
/// </summary>
@@ -87,6 +105,7 @@ namespace Yi.Framework.Stock.Application.Services
Id = p.Id,
StockId = p.StockId,
CreationTime = p.CreationTime,
RecordTime = p.RecordTime,
CurrentPrice = p.CurrentPrice,
Volume = p.Volume,
Turnover = p.Turnover,
@@ -132,5 +151,16 @@ namespace Yi.Framework.Stock.Application.Services
input.Quantity
);
}
/// <summary>
/// 生成最新股票记录
/// </summary>
[HttpPost("stock/generate")]
[Authorize]
public async Task GenerateStocksAsync()
{
await _stockMarketManager.GenerateStocksAsync();
}
}
}

19
Yi.Abp.Net8/module/ai-stock/Yi.Framework.Stock.Application/Services/StockNewsService.cs
View File

@@ -8,6 +8,7 @@ using Yi.Framework.Stock.Application.Contracts.Dtos.StockNews;
using Yi.Framework.Stock.Application.Contracts.IServices;
using Yi.Framework.Stock.Domain.Entities;
using Yi.Framework.SqlSugarCore.Abstractions;
using Yi.Framework.Stock.Domain.Managers;
namespace Yi.Framework.Stock.Application.Services
{
@@ -17,16 +18,20 @@ namespace Yi.Framework.Stock.Application.Services
public class StockNewsService : ApplicationService, IStockNewsService
{
private readonly ISqlSugarRepository<StockNewsAggregateRoot> _stockNewsRepository;
private readonly NewsManager _newsManager;
public StockNewsService(ISqlSugarRepository<StockNewsAggregateRoot> stockNewsRepository)
public StockNewsService(
ISqlSugarRepository<StockNewsAggregateRoot> stockNewsRepository,
NewsManager newsManager)
{
_stockNewsRepository = stockNewsRepository;
_newsManager = newsManager;
}
/// <summary>
/// 获取股市新闻列表
/// </summary>
[HttpGet("/api/stock/news")]
[HttpGet("/api/app/stock/news")]
public async Task<PagedResultDto<StockNewsDto>> GetStockNewsListAsync(StockNewsGetListInputDto input)
{
RefAsync<int> total = 0;
@@ -61,5 +66,15 @@ namespace Yi.Framework.Stock.Application.Services
return new PagedResultDto<StockNewsDto>(total, list);
}
/// <summary>
/// 生成股市新闻
/// </summary>
/// <returns>生成结果</returns>
[HttpPost("/api/app/stock/news/generate")]
public async Task GenerateNewsAsync()
{
await _newsManager.GenerateNewsAsync();
}
}
}

4
Yi.Abp.Net8/module/ai-stock/Yi.Framework.Stock.Domain/Entities/StockMarketAggregateRoot.cs
View File

@@ -22,7 +22,7 @@ namespace Yi.Framework.Stock.Domain.Entities
/// <summary>
/// 创建时间
/// </summary>
public DateTime CreationTime { get; set; } = DateTime.Now;
public DateTime CreationTime { get; set; }
/// <summary>
/// 创建者
@@ -52,13 +52,11 @@ namespace Yi.Framework.Stock.Domain.Entities
/// <summary>
/// 股市代码
/// </summary>
/// <remarks>如：SH、SZ、HK等</remarks>
public string MarketCode { get; set; } = string.Empty;
/// <summary>
/// 股市名称
/// </summary>
/// <remarks>如：上海证券交易所、深圳证券交易所等</remarks>
public string MarketName { get; set; } = string.Empty;
/// <summary>

10
Yi.Abp.Net8/module/ai-stock/Yi.Framework.Stock.Domain/Entities/StockPriceRecordEntity.cs
View File

@@ -21,11 +21,16 @@ namespace Yi.Framework.Stock.Domain.Entities
/// <remarks>关联到具体的股票</remarks>
public Guid StockId { get; set; }
/// <summary>
/// 创建时间（审计日志）
/// </summary>
public DateTime CreationTime { get; set; }
/// <summary>
/// 记录时间
/// </summary>
/// <remarks>价格记录的时间点</remarks>
public DateTime CreationTime { get; set; }
/// <remarks>价格记录的实际时间点</remarks>
public DateTime RecordTime { get; set; }
/// <summary>
/// 当前价
@@ -63,6 +68,7 @@ namespace Yi.Framework.Stock.Domain.Entities
{
StockId = stockId;
CreationTime = DateTime.Now;
RecordTime = DateTime.Now;
CurrentPrice = currentPrice;
Volume = volume;
Turnover = turnover;

15
Yi.Abp.Net8/module/ai-stock/Yi.Framework.Stock.Domain/Managers/NewsManager.cs
View File

@@ -1,7 +1,9 @@
using Yi.Framework.Stock.Domain.Managers.Plugins;
using Volo.Abp.Domain.Services;
using Volo.Abp.Domain.Services;
using Yi.Framework.SqlSugarCore.Abstractions;
using Yi.Framework.Stock.Domain.Entities;
using Yi.Framework.Stock.Domain.Managers.SemanticKernel;
using Yi.Framework.Stock.Domain.Managers.SemanticKernel.Plugins;
namespace Yi.Framework.Stock.Domain.Managers;
public class NewsManager:DomainService
@@ -19,9 +21,12 @@ public class NewsManager:DomainService
/// </summary>
/// <returns></returns>
public async Task GenerateNewsAsync()
{
_skClient.RegisterPlugins<NewsPlugins>("news");
await _skClient.ChatCompletionAsync("帮我生成一个新闻");
{ var question = """
生成并保存一个新闻，包含新闻标题、新闻内容、新闻简介、新闻来源
内容关于娱乐圈
只用生成一次即可
""";
await _skClient.ChatCompletionAsync(question, ("NewsPlugins","save_news"));
}
public async Task SaveNewsAsync(NewsModel news)

29
Yi.Abp.Net8/module/ai-stock/Yi.Framework.Stock.Domain/Managers/SemanticKernel/Plugins/NewsPlugins.cs
View File

@@ -1,23 +1,24 @@
using System.ComponentModel;
using System.Text.Json.Serialization;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.SemanticKernel;
using Yi.Framework.Stock.Domain.Managers;
namespace Yi.Framework.Stock.Domain.Managers.Plugins;
namespace Yi.Framework.Stock.Domain.Managers.SemanticKernel.Plugins;
public class NewsPlugins
public class NewsPlugins
{
private readonly NewsManager _newsManager;
private readonly IServiceProvider _serviceProvider;
public NewsPlugins(NewsManager newsManager)
public NewsPlugins(IServiceProvider serviceProvider)
{
_newsManager = newsManager;
_serviceProvider = serviceProvider;
}
[KernelFunction("save_news"), Description("生成并且保存一个新闻")]
[KernelFunction("save_news"), Description("生成并保存一个新闻")]
public async Task SaveAsync(NewsModel news)
{
await _newsManager.SaveNewsAsync(news);
var newsManager = _serviceProvider.GetRequiredService<NewsManager>();
await newsManager.SaveNewsAsync(news);
}
}
@@ -29,15 +30,13 @@ public class NewsModel
[JsonPropertyName("content")]
[DisplayName("新闻内容")]
public string? Content { get; set; }
//新闻简介
public string Content { get; set; }
[JsonPropertyName("summary")]
[DisplayName("新闻简介")]
public string? Summary { get; set; }
//新闻来源
public string Summary { get; set; }
[JsonPropertyName("source")]
[DisplayName("新闻来源")]
public string? Source { get; set; }
public string Source { get; set; }
}

28
Yi.Abp.Net8/module/ai-stock/Yi.Framework.Stock.Domain/Managers/SemanticKernel/Plugins/StockPlugins.cs
View File

@@ -1,26 +1,34 @@
using System.ComponentModel;
using System.Text.Json.Serialization;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.SemanticKernel;
namespace Yi.Framework.Stock.Domain.Managers.Plugins;
namespace Yi.Framework.Stock.Domain.Managers.SemanticKernel.Plugins;
public class StockPlugins
{
[KernelFunction("save_stocks"), Description("生成并且保存多个股票记录")]
public async Task<string> SaveAsync(List<StockModel> stockModels)
private readonly IServiceProvider _serviceProvider;
public StockPlugins(IServiceProvider serviceProvider)
{
return "成功";
_serviceProvider = serviceProvider;
}
[KernelFunction("save_stocks"), Description("生成并且保存多个股票记录")]
public async Task SaveAsync(List<StockModel> stockModels)
{
var stockMarketManager= _serviceProvider.GetRequiredService<StockMarketManager>();
await stockMarketManager.SaveStockAsync(stockModels);
}
}
public class StockModel
{
[JsonPropertyName("id")]
public int Id { get; set; }
[DisplayName("股票id")]
public Guid Id { get; set; }
[JsonPropertyName("name")]
public string Name { get; set; }
[JsonPropertyName("is_on")]
public bool? IsOn { get; set; }
[JsonPropertyName("values")]
[DisplayName("股票未来24小时价格")]
public decimal[] Values { get; set; }
}

64
Yi.Abp.Net8/module/ai-stock/Yi.Framework.Stock.Domain/Managers/SemanticKernel/SemanticKernelClient.cs
View File

@@ -1,57 +1,18 @@
using Microsoft.Extensions.Options;
using Microsoft.SemanticKernel;
using Microsoft.SemanticKernel;
using Microsoft.SemanticKernel.ChatCompletion;
using Microsoft.SemanticKernel.Connectors.OpenAI;
using Volo.Abp.DependencyInjection;
namespace Yi.Framework.Stock.Domain.Managers;
namespace Yi.Framework.Stock.Domain.Managers.SemanticKernel;
public class SemanticKernelClient:ITransientDependency
{
private Kernel Kernel { get; }
private readonly IKernelBuilder _kernelBuilder;
private SemanticKernelOptions Options { get; }
public Kernel Kernel { get;}
public SemanticKernelClient(IOptions<SemanticKernelOptions> semanticKernelOption)
public SemanticKernelClient(Kernel kernel)
{
Options = semanticKernelOption.Value;
_kernelBuilder = Kernel.CreateBuilder();
RegisterChatCompletion();
Kernel = _kernelBuilder.Build();
RegisterDefautlPlugins();
this.Kernel = kernel;
}
/// <summary>
/// 注册
/// </summary>
private void RegisterChatCompletion()
{
_kernelBuilder.AddOpenAIChatCompletion(
modelId: Options.ModelId,
apiKey: Options.ApiKey,
httpClient: new HttpClient() { BaseAddress = new Uri(Options.Endpoint) });
}
/// <summary>
/// 插件注册
/// </summary>
private void RegisterDefautlPlugins()
{
//动态导入插件
// this.Kernel.ImportPluginFromPromptDirectory(System.IO.Path.Combine("wwwroot", "plugin","stock"),"stock");
}
/// <summary>
/// 自定义插件
/// </summary>
/// <param name="pluginName"></param>
/// <typeparam name="TPlugin"></typeparam>
public void RegisterPlugins<TPlugin>(string pluginName)
{
this.Kernel.Plugins.AddFromType<TPlugin>(pluginName);
}
/// <summary>
/// 执行插件
/// </summary>
@@ -70,19 +31,24 @@ public class SemanticKernelClient:ITransientDependency
/// 聊天对话,调用方法
/// </summary>
/// <returns></returns>
public async Task<IReadOnlyList<ChatMessageContent>> ChatCompletionAsync(string question)
public async Task<IReadOnlyList<ChatMessageContent>> ChatCompletionAsync(string question,params (string,string)[] functions)
{
OpenAIPromptExecutionSettings openAIPromptExecutionSettings = new()
if (functions is null)
{
FunctionChoiceBehavior = FunctionChoiceBehavior.Auto(),
MaxTokens = Options.MaxTokens
throw new Exception("请选择插件");
}
var openSettings = new OpenAIPromptExecutionSettings()
{
FunctionChoiceBehavior = FunctionChoiceBehavior.Auto(functions.Select(x=>this.Kernel.Plugins.GetFunction(x.Item1, x.Item2)).ToList(),true),
// ToolCallBehavior = ToolCallBehavior.AutoInvokeKernelFunctions,
MaxTokens =1000
};
var chatCompletionService = this.Kernel.GetRequiredService<IChatCompletionService>();
var results =await chatCompletionService.GetChatMessageContentsAsync(
question,
executionSettings: openAIPromptExecutionSettings,
executionSettings: openSettings,
kernel: Kernel);
return results;
}

36
Yi.Abp.Net8/module/ai-stock/Yi.Framework.Stock.Domain/Managers/SemanticKernel/SemanticKernelOptions.cs
View File

@@ -1,29 +1,9 @@
namespace Yi.Framework.Stock.Domain.Managers;
public class SemanticKernelOptions
namespace Yi.Framework.Stock.Domain.Managers.SemanticKernel
{
/// <summary>
/// OpenAI 模型 ID
/// </summary>
public string ModelId { get; set; } = string.Empty;
/// <summary>
/// OpenAI API 密钥
/// </summary>
public string ApiKey { get; set; } = string.Empty;
/// <summary>
/// API 端点地址
/// </summary>
public string Endpoint { get; set; } = string.Empty;
/// <summary>
/// 最大生成令牌数
/// </summary>
public int MaxTokens { get; set; } = 1000;
/// <summary>
/// 插件目录路径
/// </summary>
public string PluginsDirectoryPath { get; set; } = string.Empty;
}
public class SemanticKernelOptions
{
public string ModelId { get; set; }
public string Endpoint { get; set; }
public string ApiKey { get; set; }
}
}

94
Yi.Abp.Net8/module/ai-stock/Yi.Framework.Stock.Domain/Managers/StockMarketManager.cs
View File

@@ -8,7 +8,8 @@ using Yi.Framework.Stock.Domain.Entities;
using Yi.Framework.Stock.Domain.Shared;
using Yi.Framework.Stock.Domain.Shared.Etos;
using Yi.Framework.SqlSugarCore.Abstractions;
using Yi.Framework.Stock.Domain.Managers.Plugins;
using Yi.Framework.Stock.Domain.Managers.SemanticKernel;
using Yi.Framework.Stock.Domain.Managers.SemanticKernel.Plugins;
namespace Yi.Framework.Stock.Domain.Managers
{
@@ -277,27 +278,25 @@ namespace Yi.Framework.Stock.Domain.Managers
/// </summary>
/// <param name="priceRecords">价格记录列表</param>
/// <returns>保存的记录数量</returns>
public async Task<int> BatchSaveStockPriceRecordsAsync(List<StockPriceRecordEntity> priceRecords)
public async Task BatchSaveStockPriceRecordsAsync(List<StockPriceRecordEntity> priceRecords)
{
if (priceRecords == null || !priceRecords.Any())
{
return 0;
return;
}
// 验证数据
foreach (var record in priceRecords)
for (int i = 0; i < priceRecords.Count; i++)
{
var record = priceRecords[i];
if (record.CurrentPrice <= 0)
{
throw new UserFriendlyException($"股票ID {record.StockId} 的价格必须大于0");
}
// 设置创建时间为当前时间（如果未设置）
if (record.CreationTime == default)
{
record.CreationTime = DateTime.Now;
}
// 设置记录时间（当前时间加上i个小时），只记录到年月日小时
record.RecordTime = new DateTime(DateTime.Now.AddHours(i).Year, DateTime.Now.AddHours(i).Month, DateTime.Now.AddHours(i).Day, DateTime.Now.AddHours(i).Hour, 0, 0);
// 计算交易额（如果未设置）
if (record.Turnover == 0 && record.Volume > 0)
{
@@ -306,18 +305,85 @@ namespace Yi.Framework.Stock.Domain.Managers
}
await _stockPriceRecordRepository.InsertManyAsync(priceRecords);
return priceRecords.Count;
}
public async Task SaveStockAsync(List<StockModel> stockModels)
{
if (stockModels == null || !stockModels.Any())
{
return;
}
// 收集所有股票ID
var stockIds = stockModels.Select(m => m.Id).ToList();
// 一次性查询所有相关股票信息
var stockMarkets = await _stockMarketRepository.GetListAsync(s => stockIds.Contains(s.Id));
// 构建字典以便快速查找
var stockMarketsDict = stockMarkets.ToDictionary(s => s.Id);
// 将StockModel转换为StockPriceRecordEntity
var priceRecords = new List<StockPriceRecordEntity>();
foreach (var stockModel in stockModels)
{
if (stockModel.Values == null || !stockModel.Values.Any())
{
continue;
}
// 从字典中查找股票信息，而不是每次查询数据库
if (!stockMarketsDict.TryGetValue(stockModel.Id, out var stockMarket))
{
continue;
}
// 为每个价格点创建一个记录
foreach (var priceValue in stockModel.Values)
{
var priceRecord = new StockPriceRecordEntity
{
StockId = stockMarket.Id,
CurrentPrice = priceValue,
Volume = 0, // 可以根据实际情况设置
Turnover = 0, // 可以根据实际情况设置
PeriodType = PeriodTypeEnum.Hour
};
priceRecords.Add(priceRecord);
}
}
// 批量保存价格记录
if (priceRecords.Any())
{
await BatchSaveStockPriceRecordsAsync(priceRecords);
}
}
/// <summary>
/// 生成最新股票记录
/// </summary>
/// <returns></returns>
public async Task GenerateStocksAsync()
{
_skClient.RegisterPlugins<StockPlugins>("stock");
await _skClient.ChatCompletionAsync("帮我生成多个股市内容");
var question = """
根据给出的模拟新闻，模拟生成多家股票未来价格
以下是近期新闻：
ai势力逐步崛起
大量网红下岗
近期群星演唱会即将开幕
以下是多个股票：
伦敦股票：id:3a1886d5-1479-5402-3bcb-063e989898d1 最后一次价格：188.7
上海股票：id:3a1886d5-4393-606c-b040-52f1ef7d2158 最后一次价格：125.2
纽约股票：id:3a1886d5-6906-8f30-d955-198fbcfe4026 最后一次价格：185.5
根据上面信息，给每个股票返回未来24小时的股票价格，每个小时，整点为单位一条记录，一家共有24条，返回股票id 与 长度为24的价格列表
只用生成一次即可
""";
await _skClient.ChatCompletionAsync(question,("StockPlugins","save_stocks"));
}
}
}

39
Yi.Abp.Net8/module/ai-stock/Yi.Framework.Stock.Domain/YiFrameworkStockDomainModule.cs
View File

@@ -1,14 +1,18 @@
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.SemanticKernel;
using Volo.Abp.Caching;
using Volo.Abp.Domain;
using Yi.Framework.Stock.Domain.Shared;
using Yi.Framework.Mapster;
using Yi.Framework.Stock.Domain.Managers;
using Yi.Framework.Stock.Domain.Managers.SemanticKernel;
using Yi.Framework.Stock.Domain.Managers.SemanticKernel.Plugins;
using Yi.Framework.Stock.Domain.Shared;
namespace Yi.Framework.Stock.Domain
{
[DependsOn(
typeof(YiFrameworkStockDomainSharedModule),
typeof(YiFrameworkMapsterModule),
typeof(AbpDddDomainModule),
typeof(AbpCachingModule)
@@ -17,13 +21,30 @@ namespace Yi.Framework.Stock.Domain
{
public override void ConfigureServices(ServiceConfigurationContext context)
{
Configure<SemanticKernelOptions>((options)=>{
options.Endpoint = "https://api.token-ai.cn/v1";
options.ApiKey = "sk-V6OqmrloXDAiTM2FWoisGgaop72Ngr0fXAnXL8";
options.ModelId = "gpt-4o-mini";
options.MaxTokens = 1000;
options.PluginsDirectoryPath = "plugins";
});
var configuration = context.Services.GetConfiguration();
var services = context.Services;
// 配置绑定
var semanticKernelSection = configuration.GetSection("SemanticKernel");
services.Configure<SemanticKernelOptions>(configuration.GetSection("SemanticKernel"));
services.AddHttpClient();
#pragma warning disable SKEXP0010
// 从配置中获取值
var options = semanticKernelSection.Get<SemanticKernelOptions>();
services.AddKernel()
.AddOpenAIChatCompletion(
modelId: options.ModelId,
endpoint: new Uri(options.Endpoint),
apiKey: options.ApiKey);
#pragma warning restore SKEXP0010
// 添加插件
services.AddSingleton<KernelPlugin>(sp => KernelPluginFactory.CreateFromType<NewsPlugins>(serviceProvider: sp));
services.AddSingleton<KernelPlugin>(sp => KernelPluginFactory.CreateFromType<StockPlugins>(serviceProvider: sp));
// 注册NewsManager
services.AddTransient<NewsManager>();
}
}
}

107
Yi.Abp.Net8/module/chat-hub/Yi.Framework.ChatHub.Domain/Managers/AiManager.cs
View File

@@ -1,11 +1,11 @@
using System.Collections.Generic;
using System.Net;
using Microsoft.Extensions.Options;
using OpenAI;
using OpenAI.Managers;
using OpenAI.ObjectModels;
using OpenAI.ObjectModels.RequestModels;
using OpenAI.ObjectModels.ResponseModels;
// using OpenAI;
// using OpenAI.Managers;
// using OpenAI.ObjectModels;
// using OpenAI.ObjectModels.RequestModels;
// using OpenAI.ObjectModels.ResponseModels;
using Volo.Abp.DependencyInjection;
using Volo.Abp.Domain.Services;
using Yi.Framework.ChatHub.Domain.Shared.Dtos;
@@ -16,58 +16,59 @@ namespace Yi.Framework.ChatHub.Domain.Managers
{
public AiManager(IOptions<AiOptions> options)
{
this.OpenAIService = new OpenAIService(new OpenAiOptions()
{
ApiKey = options.Value.ApiKey,
BaseDomain = options.Value.BaseDomain
});
// this.OpenAIService = new OpenAIService(new OpenAiOptions()
// {
// ApiKey = options.Value.ApiKey,
// BaseDomain = options.Value.BaseDomain
// });
}
private OpenAIService OpenAIService { get; }
// private OpenAIService OpenAIService { get; }
public async IAsyncEnumerable<string> ChatAsStreamAsync(string model, List<AiChatContextDto> aiChatContextDtos)
{
if (aiChatContextDtos.Count == 0)
{
yield return null;
}
List<ChatMessage> messages = aiChatContextDtos.Select(x =>
{
if (x.AnswererType == AnswererTypeEnum.Ai)
{
return ChatMessage.FromSystem(x.Message);
}
else
{
return ChatMessage.FromUser(x.Message);
}
}).ToList();
var completionResult = OpenAIService.ChatCompletion.CreateCompletionAsStream(new ChatCompletionCreateRequest
{
Messages = messages,
Model =model
});
HttpStatusCode? error = null;
await foreach (var result in completionResult)
{
if (result.Successful)
{
yield return result.Choices.FirstOrDefault()?.Message.Content ?? null;
}
else
{
error = result.HttpStatusCode;
break;
}
}
if (error == HttpStatusCode.PaymentRequired)
{
yield return "余额不足,请联系站长充值";
}
throw new NotImplementedException("准备sk重构");
yield break;
// if (aiChatContextDtos.Count == 0)
// {
// yield return null;
// }
//
// List<ChatMessage> messages = aiChatContextDtos.Select(x =>
// {
// if (x.AnswererType == AnswererTypeEnum.Ai)
// {
// return ChatMessage.FromSystem(x.Message);
// }
// else
// {
// return ChatMessage.FromUser(x.Message);
// }
// }).ToList();
// var completionResult = OpenAIService.ChatCompletion.CreateCompletionAsStream(new ChatCompletionCreateRequest
// {
// Messages = messages,
// Model =model
// });
//
// HttpStatusCode? error = null;
// await foreach (var result in completionResult)
// {
// if (result.Successful)
// {
// yield return result.Choices.FirstOrDefault()?.Message.Content ?? null;
// }
// else
// {
// error = result.HttpStatusCode;
// break;
// }
//
// }
// if (error == HttpStatusCode.PaymentRequired)
// {
// yield return "余额不足,请联系站长充值";
//
// }
}
}
}

1
Yi.Abp.Net8/module/chat-hub/Yi.Framework.ChatHub.Domain/Yi.Framework.ChatHub.Domain.csproj
View File

@@ -7,7 +7,6 @@
</ItemGroup>
<ItemGroup>
<PackageReference Include="Betalgo.OpenAI" Version="8.6.1" />
<PackageReference Include="Volo.Abp.AspNetCore.SignalR" Version="$(AbpVersion)" />
<PackageReference Include="Volo.Abp.Ddd.Domain" Version="$(AbpVersion)" />

1
Yi.Abp.Net8/src/Yi.Abp.Application/Services/TestService.cs
View File

@@ -30,7 +30,6 @@ namespace Yi.Abp.Application.Services
public ISqlSugarRepository<BannerAggregateRoot> sqlSugarRepository { get; set; }
/// <summary>
/// 动态Api
/// </summary>
/// <param name="name"></param>
/// <returns></returns>

34
Yi.Abp.Net8/src/Yi.Abp.Web/Jobs/ai-stock/GenerateNewsJob.cs Normal file
View File

@@ -0,0 +1,34 @@
using System;
using System.Threading;
using System.Threading.Tasks;
using Volo.Abp.BackgroundWorkers.Hangfire;
using Yi.Framework.Stock.Domain.Managers;
namespace Yi.Abp.Web.Jobs.ai_stock
{
public class GenerateNewsJob : HangfireBackgroundWorkerBase
{
private NewsManager _newsManager;
public GenerateNewsJob(NewsManager newsManager)
{
_newsManager = newsManager;
RecurringJobId = "AI股票新闻生成";
//每个小时整点执行一次
CronExpression = "0 0 * * * ?";
}
public override async Task DoWorkAsync(CancellationToken cancellationToken = new CancellationToken())
{
// 每次触发只有2/24的概率执行生成新闻
var random = new Random();
var probability = random.Next(0, 24);
if (probability < 2)
{
await _newsManager.GenerateNewsAsync();
}
}
}
}

10
Yi.Abp.Net8/src/Yi.Abp.Web/YiAbpWebModule.cs
View File

@@ -27,6 +27,7 @@ using Volo.Abp.AspNetCore.VirtualFileSystem;
using Volo.Abp.Auditing;
using Volo.Abp.Autofac;
using Volo.Abp.BackgroundJobs.Hangfire;
using Volo.Abp.BackgroundWorkers;
using Volo.Abp.Caching;
using Volo.Abp.MultiTenancy;
using Volo.Abp.Swashbuckle;
@@ -106,6 +107,15 @@ namespace Yi.Abp.Web
var host = context.Services.GetHostingEnvironment();
var service = context.Services;
//本地开发环境，禁用作业执行
if (host.IsDevelopment())
{
Configure<AbpBackgroundWorkerOptions> (options =>
{
options.IsEnabled = false;
});
}
//请求日志
Configure<AbpAuditingOptions>(options =>
{

8
Yi.Abp.Net8/src/Yi.Abp.Web/appsettings.json
View File

@@ -108,5 +108,13 @@
"AiOptions": {
"ApiKey": "",
"BaseDomain": ""
},
//语义内核
"SemanticKernel": {
"ModelId": "gpt-4o",
"Endpoint": "https://xxx.com/v1",
"ApiKey": "sk-xxxxxx"
}
}