Bitzsoft.Integrations.AI 1.0.0-alpha.4

Bitzsoft.Integrations.AI

AI 服务集成 -- 兼容 OpenAI API 的统一客户端，支持 OpenAI、DeepSeek、通义千问等主流大模型服务商。

功能特性

• 兼容所有 OpenAI Chat Completions API 协议的服务商（OpenAI、DeepSeek、通义千问等）
• 内置自动重试机制，可配置最大重试次数与超时时间
• 支持单轮对话、多轮对话及原始请求三种调用方式
• 支持命名客户端（Named Client），满足多租户或多模型场景
• 基于 IHttpClientFactory 实现，生命周期管理安全可靠
• 完整的 Token 用量追踪（PromptTokens / CompletionTokens / TotalTokens）

安装

dotnet add package Bitzsoft.Integrations.AI

或通过 csproj 引用：

<PackageReference Include="Bitzsoft.Integrations.AI" Version="*" />

配置

在 appsettings.json 中添加以下配置节：

{
  "AI": {
    "ApiUrl": "https://api.openai.com/v1/chat/completions",
    "ApiKey": "sk-your-api-key",
    "DefaultModel": "gpt-4",
    "TimeoutSeconds": 60,
    "MaxRetries": 3
  }
}

注册服务

基础注册

using Bitzsoft.Integrations.AI;
using Bitzsoft.Integrations.AI.Dto;

// Program.cs / Startup.cs
services.AddAIClient(options =>
{
    options.ApiUrl = "https://api.openai.com/v1/chat/completions";
    options.ApiKey = "sk-your-api-key";
    options.DefaultModel = "gpt-4";
    options.TimeoutSeconds = 60;
    options.MaxRetries = 3;
});

多租户 / 多模型注册

using Bitzsoft.Integrations.AI;
using Bitzsoft.Integrations.AI.Dto;

// 注册 GPT-4 客户端
services.AddAIClient("GPT4", options =>
{
    options.ApiUrl = "https://api.openai.com/v1/chat/completions";
    options.ApiKey = "sk-your-openai-key";
    options.DefaultModel = "gpt-4";
});

// 注册 DeepSeek 客户端
services.AddAIClient("DeepSeek", options =>
{
    options.ApiUrl = "https://api.deepseek.com/v1/chat/completions";
    options.ApiKey = "sk-your-deepseek-key";
    options.DefaultModel = "deepseek-chat";
});

使用示例

注入并使用 IAIClient

using Bitzsoft.Integrations.AI;
using Bitzsoft.Integrations.AI.Dto;

public class DocumentService
{
    private readonly IAIClient _aiClient;

    public DocumentService(IAIClient aiClient)
    {
        _aiClient = aiClient;
    }

    /// <summary>
    /// 对文档内容进行摘要提取
    /// </summary>
    public async Task<string> SummarizeAsync(string documentContent, CancellationToken ct = default)
    {
        var result = await _aiClient.SendAsync(
            message: $"请对以下文档内容进行摘要：\n\n{documentContent}",
            systemPrompt: "你是一个专业的文档摘要助手，请用简洁的语言概括文档核心内容。",
            model: null,
            cancellationToken: ct);

        if (result.Error is not null)
        {
            throw new InvalidOperationException($"AI 调用失败：{result.Error.Message}");
        }

        return result.GetContent() ?? string.Empty;
    }

    /// <summary>
    /// 多轮对话场景：结合上下文回答问题
    /// </summary>
    public async Task<string> ChatWithHistoryAsync(
        string userQuestion,
        List<AIMessageItem> history,
        CancellationToken ct = default)
    {
        var messages = new List<AIMessageItem>(history)
        {
            new() { Role = "user", Content = userQuestion }
        };

        var result = await _aiClient.SendMessagesAsync(messages, cancellationToken: ct);

        return result.GetContent() ?? string.Empty;
    }
}

使用原始请求精细控制参数

using Bitzsoft.Integrations.AI;
using Bitzsoft.Integrations.AI.Dto;

public class CreativeWritingService
{
    private readonly IAIClient _aiClient;

    public CreativeWritingService(IAIClient aiClient)
    {
        _aiClient = aiClient;
    }

    public async Task<string> GenerateCreativeContentAsync(string topic, CancellationToken ct = default)
    {
        var request = new AISendDto
        {
            Model = "gpt-4",
            Messages = new List<AIMessageItem>
            {
                new() { Role = "system", Content = "你是一位创意写作专家" },
                new() { Role = "user", Content = $"请围绕「{topic}」写一段创意短文" }
            },
            Temperature = 1.2,
            MaxTokens = 2000,
            EnableSearch = false,
            Stream = false
        };

        var result = await _aiClient.SendRequestAsync(request, ct);

        // 检查 Token 消耗
        if (result.Usage is not null)
        {
            Console.WriteLine($"Token 消耗 - Prompt: {result.Usage.PromptTokens}, " +
                $"Completion: {result.Usage.CompletionTokens}, " +
                $"Total: {result.Usage.TotalTokens}");
        }

        return result.GetContent() ?? string.Empty;
    }
}

相关包