feat: 兼容了用量使用显示

2025-07-22 10:40:23 +08:00
parent b6d670c240
commit a0eb234539
8 changed files with 97 additions and 75 deletions
--- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/YiFrameworkAiHubDomainModule.cs
+++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/YiFrameworkAiHubDomainModule.cs
@@ -1,5 +1,6 @@
 using Microsoft.Extensions.DependencyInjection;
 using Volo.Abp.Domain;
+using Yi.Framework.AiHub.Application.Contracts.Dtos.OpenAi;
 using Yi.Framework.AiHub.Domain.AiGateWay;
 using Yi.Framework.AiHub.Domain.AiGateWay.Impl.ThorAzureDatabricks.Chats;
 using Yi.Framework.AiHub.Domain.AiGateWay.Impl.ThorAzureOpenAI.Chats;
@@ -48,6 +49,16 @@ namespace Yi.Framework.AiHub.Domain
                        request.Temperature = null;
                    }
                });
+                options.Handles.Add(request =>
+                {
+                    if (request.Stream == true)
+                    {
+                        request.StreamOptions = new ThorStreamOptions()
+                        {
+                            IncludeUsage = true
+                        };
+                    }
+                });
            });
        }