集成 OpenAI API 的准备工作
确保开发环境已安装 JDK 17+ 和 Maven/Gradle,并创建一个新的 Spring Boot 3 项目。在 pom.xml 中添加 Spring AI 和 OpenAI 的依赖:
XML
<dependency>
<groupId>org.springframework.ai</groupId>
<artifactId>spring-ai-openai-spring-boot-starter</artifactId>
<version>0.8.1</version>
</dependency>
配置 OpenAI API 密钥
在 application.properties 或 application.yml 中配置 OpenAI 的 API 密钥和模型参数:
properties
spring.ai.openai.api-key=your-api-key
spring.ai.openai.model=gpt-3.5-turbo
创建 AI 服务接口
定义一个服务类封装 AI 交互逻辑。例如,创建一个 AIService 类,注入 OpenAiChatClient 并实现对话功能:
java
import org.springframework.ai.chat.ChatClient;
import org.springframework.stereotype.Service;
@Service
public class AIService {
private final ChatClient chatClient;
public AIService(ChatClient chatClient) {
this.chatClient = chatClient;
}
public String generateResponse(String prompt) {
return chatClient.call(prompt);
}
}
实现控制器暴露 API
创建 REST 控制器提供外部访问接口,接收用户输入并返回 AI 生成的响应:
java
import org.springframework.web.bind.annotation.*;
@RestController
@RequestMapping("/api/ai")
public class AIController {
private final AIService aiService;
public AIController(AIService aiService) {
this.aiService = aiService;
}
@PostMapping("/chat")
public String chat(@RequestBody String prompt) {
return aiService.generateResponse(prompt);
}
}
测试与验证
启动应用后,通过 curl 或 Postman 测试接口:
bash
curl -X POST http://localhost:8080/api/ai/chat -H "Content-Type: text/plain" -d "Spring Boot 3有哪些新特性?"
预期返回 OpenAI 生成的文本响应,例如对 Spring Boot 3 新特性的总结。
高级配置选项
支持自定义模型参数,如温度(控制随机性)和最大 token 数:
properties
spring.ai.openai.options.temperature=0.7
spring.ai.openai.options.max-tokens=500
可通过 OpenAiChatOptions 在运行时动态调整参数:
java
OpenAiChatOptions options = OpenAiChatOptions.builder()
.withTemperature(0.5f)
.withMaxTokens(200)
.build();
chatClient.call(new Prompt("你的提示", options));