Spring AI Chat

Spring AI 聊天集成

功能描述

提供 AI 聊天交互能力，支持与多种大语言模型进行对话，基于 Spring AI 框架实现。

何时使用

在以下情况调用此技能：

• 创建聊天端点或 API
• 集成新的 AI 聊天模型
• 添加聊天相关功能
• 处理 AI 对话流程

核心功能

• 多模型支持：支持 OpenAI、DeepSeek 等多种 AI 模型
• 聊天对话：支持简单聊天和完整对话流程
• 参数配置：支持温度、token 数量等参数配置
• 流式响应：支持流式输出模式

输入参数

输出格式

{
  "provider": "deepseek",
  "content": "AI 回复内容",
  "model": "deepseek-chat",
  "usage": {
    "promptTokens": 100,
    "completionTokens": 50
  },
  "success": true,
  "errorMessage": null
}

支持模型

API 调用示例

# 简单聊天
curl -X POST http://localhost:8080/api/chat \
  -H "Content-Type: application/json" \
  -d '{
    "message": "你好，AI！"
  }'

# 指定模型
curl -X POST http://localhost:8080/api/chat \
  -H "Content-Type: application/json" \
  -d '{
    "model": "deepseek",
    "message": "解释一下量子计算",
    "temperature": 0.7,
    "maxTokens": 2048
  }'

# 流式响应
curl -X POST http://localhost:8080/api/chat/stream \
  -H "Content-Type: application/json" \
  -d '{
    "message": "写一首诗"
  }'

使用模式

// 注入 ChatService
private final ChatService chatService;

// 简单聊天
ChatResponse response = chatService.chat("你好 AI！");

// 指定模型
ChatResponse response = chatService.chat("deepseek", "你好！");

// 完整请求
ChatRequest request = new ChatRequest("deepseek", "你好", 0.7, 2048, Map.of());
ChatResponse response = chatService.chat(request);

核心组件

配置要求

环境变量

配置文件

# application.yml
server:
  port: ${SERVER_PORT:8080}

spring:
  ai:
    openai:
      base-url: ${DEEPSEEK_BASE_URL:https://api.deepseek.com/v1}
      api-key: ${DEEPSEEK_API_KEY}
      chat:
        options:
          model: deepseek-chat
          temperature: 0.7
          max-tokens: 2048

添加新的聊天功能

1. 在 ChatService.java 中添加方法
2. 在 model 包中创建请求/响应 DTO
3. 在 ChatController.java 中添加控制器端点

扩展指南

添加新的模型提供者

1. 实现 ModelProvider 接口
2. 创建对应的配置类
3. 在 ModelProviderManager 中注册

自定义聊天逻辑

继承 ChatServiceImpl 并重写相关方法，实现自定义业务逻辑。

目录结构

src/main/java/com/bage/study/ai/best/practice/hello/
├── controller/
│   └── ChatController.java
├── service/
│   ├── ChatService.java
│   └── impl/
│       └── ChatServiceImpl.java
├── model/
│   ├── ChatRequest.java
│   └── ChatResponse.java
├── provider/
│   └── ModelProvider.java
└── config/
    └── ChatConfig.java