feat: ai完成stock模块搭建

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

@@ -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"));
         }
     }
 }