介绍

AIUtil 是 Hutool AI 模块的核心工具类，提供了快速访问各种 AI 服务的静态方法。

方法介绍

• 获取 AI 服务

// 获取通用 AI 服务
AIService getAIService(AIConfig config);

// 获取指定类型的 AI 服务
<T extends AIService> T getAIService(AIConfig config, Class<T> clazz);

// 获取 DeepSeek 服务
DeepSeekService getDeepSeekService(AIConfig config);

// 获取豆包服务
DoubaoService getDoubaoService(AIConfig config);

// 获取 Grok 服务
GrokService getGrokService(AIConfig config);

// 获取 OpenAI 服务
OpenaiService getOpenAIService(AIConfig config);

• 快速对话

// 单次对话
String chat(AIConfig config, String prompt)

// 多轮对话
String chat(AIConfig config, List<Message> messages)

使用

• 基本使用

// 快速对话
String chat = AIUtil.chat(new AIConfigBuilder(ModelName.DEEPSEEK.getValue()).setApiKey("your key").build(), "写一首赞美我的诗");

//多轮对话
List<Message> messages = new ArrayList<>();
messages.add(new Message("system","你是财神爷，只会说“我是财神”"));
messages.add(new Message("user","你是谁啊？"));
String chat = AIUtil.chat(new AIConfigBuilder(ModelName.DEEPSEEK.getValue()).setApiKey("your key").build(), messages);

Hutool为每一个支持的大模型都添加了ModelName（AI厂商）以及model的枚举值

• 使用特定服务

// 获取 DeepSeek 服务
DeepSeekService deepSeekService = AIUtil.getDeepSeekService(new AIConfigBuilder(ModelName.DEEPSEEK.getValue()).setApiKey("your key").build());
String beta = deepSeekService.beta("写一个疯狂星期四广告词");

// 获取 DeepSeek 服务，指定服务类
DeepSeekService deepSeekService2 = AIUtil.getAIService(new AIConfigBuilder(ModelName.DEEPSEEK.getValue()).setApiKey("your key").build(), DeepSeekService.class);
String beta2 = deepSeekService.beta("写一个疯狂星期四广告词");