Yi.Framework/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/YiFrameworkAiHubDomainModule.cs

using Alipay.EasySDK.Factory;
using Alipay.EasySDK.Kernel;
using Dm.util;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using Volo.Abp.Domain;
using Yi.Framework.AiHub.Domain.AiGateWay;
using Yi.Framework.AiHub.Domain.AiGateWay.Impl.ThorAzureDatabricks.Chats;
using Yi.Framework.AiHub.Domain.AiGateWay.Impl.ThorAzureOpenAI.Chats;
using Yi.Framework.AiHub.Domain.AiGateWay.Impl.ThorAzureOpenAI.Images;
using Yi.Framework.AiHub.Domain.AiGateWay.Impl.ThorClaude.Chats;
using Yi.Framework.AiHub.Domain.AiGateWay.Impl.ThorCustomOpenAI.Chats;
using Yi.Framework.AiHub.Domain.AiGateWay.Impl.ThorDeepSeek.Chats;
using Yi.Framework.AiHub.Domain.AiGateWay.Impl.ThorSiliconFlow.Embeddings;
using Yi.Framework.AiHub.Domain.Managers.Fuwuhao;
using Yi.Framework.AiHub.Domain.Shared;
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi;
using Yi.Framework.Mapster;
using Microsoft.Extensions.DependencyInjection;

namespace Yi.Framework.AiHub.Domain
{
    [DependsOn(
        typeof(YiFrameworkAiHubDomainSharedModule),
        typeof(YiFrameworkMapsterModule),
        typeof(AbpDddDomainModule)
    )]
    public class YiFrameworkAiHubDomainModule : AbpModule
    {
        public override void ConfigureServices(ServiceConfigurationContext context)
        {
            var configuration = context.Services.GetConfiguration();
            var services = context.Services;

            #region OpenAi ChatCompletion

            services.AddKeyedTransient<IChatCompletionService, AzureOpenAiChatCompletionCompletionsService>(
                nameof(AzureOpenAiChatCompletionCompletionsService));
            services.AddKeyedTransient<IChatCompletionService, AzureDatabricksChatCompletionsService>(
                nameof(AzureDatabricksChatCompletionsService));
            services.AddKeyedTransient<IChatCompletionService, DeepSeekChatCompletionsService>(
                nameof(DeepSeekChatCompletionsService));
            services.AddKeyedTransient<IChatCompletionService, OpenAiChatCompletionsService>(
                nameof(OpenAiChatCompletionsService));

            #endregion

            #region Anthropic ChatCompletion

            services.AddKeyedTransient<IAnthropicChatCompletionService, CustomOpenAIAnthropicChatCompletionsService>(
                nameof(CustomOpenAIAnthropicChatCompletionsService));
            services.AddKeyedTransient<IAnthropicChatCompletionService, AnthropicChatCompletionsService>(
                nameof(AnthropicChatCompletionsService));

            #endregion


            #region Image

            services.AddKeyedTransient<IImageService, AzureOpenAIServiceImageService>(
                nameof(AzureOpenAIServiceImageService));

            #endregion

            #region Embedding

            services.AddKeyedTransient<ITextEmbeddingService, SiliconFlowTextEmbeddingService>(
                nameof(SiliconFlowTextEmbeddingService));

            #endregion

            //ai模型特殊性兼容处理
            Configure<SpecialCompatibleOptions>(options =>
            {
                options.Handles.Add(request =>
                {
                    if (request.Model == "o1")
                    {
                        request.Temperature = null;
                    }
                });
                options.Handles.Add(request =>
                {
                    if (request.Model.StartsWith("o3-mini") || request.Model.StartsWith("o4-mini"))
                    {
                        request.MaxCompletionTokens = request.MaxTokens;
                        request.MaxTokens = null;
                        request.Temperature = null;
                    }
                });
                options.Handles.Add(request =>
                {
                    if (request.Stream == true)
                    {
                        request.StreamOptions = new ThorStreamOptions()
                        {
                            IncludeUsage = true
                        };
                    }
                });
                options.Handles.Add(request =>
                {
                    if (request.MaxTokens >= 16384)
                    {
                        request.MaxTokens = 16384;
                    }
                });
                options.AnthropicHandles.add(request =>
                {
                    if (request.MaxTokens is null || request.MaxTokens <= 0)
                    {
                        throw new UserFriendlyException("MaxTokens must be greater than or equal to 0");
                    }
                });
            });


            //配置支付宝支付
            var config = configuration.GetSection("Alipay").Get<Config>();
            Factory.SetOptions(config);

            //配置服务号
            Configure<FuwuhaoOptions>(configuration.GetSection("Fuwuhao"));

            services.AddHttpClient()
                .ConfigureHttpClientDefaults(builder =>
                {
                    builder.ConfigureHttpClient(client =>
                    {
                        client.DefaultRequestHeaders.Add("User-Agent","Apifox/1.0.0 (https://apifox.com)");
                        client.Timeout = TimeSpan.FromMinutes(10);
                    });
                });
        }
    }
}