springboot ai库接入
一 添加mvn库
<dependency>
<groupId>org.springframework.ai</groupId>
<artifactId>spring-ai-openai-spring-boot-starter</artifactId>
<version>1.0.0-SNAPSHOT</version>
</dependency>
二 yml中添加配置
spring:
application:
name: easy-ai
ai:
openai:
# Api密钥
api-key: sk-absdfsadfsdfsdf
# 要连接的URL
base-url: http://www.hichat.site/proxy
chat:
options:
# 模型
model:
# 随机性
temperature: 0.7
@RestController
public class ChatController {
private final OpenAiChatClient chatClient;
@Autowired
public ChatController(OpenAiChatClient chatClient) {
this.chatClient = chatClient;
}
@GetMapping("/v1/chat/completions")
public Flux<ChatResponse> generateStream(@RequestParam(value = "message", defaultValue = "给我打个招呼") String message) {
OpenAiChatOptions openAiChatOptions=new OpenAiChatOptions();
// openAiChatOptions.setModel("claude-3-sonnet-20240229");
openAiChatOptions.setModel("gpt-4-all");
UserMessage userMessage = new UserMessage(message);
Prompt prompt = new Prompt(userMessage,openAiChatOptions);
return chatClient.stream(prompt);
}
}
修改于 2024-07-20 02:35:04