mirror of
https://gitee.com/chinabugotech/hutool.git
synced 2025-08-18 20:38:02 +08:00
@@ -43,7 +43,11 @@ public enum ModelName {
|
||||
/**
|
||||
* grok
|
||||
*/
|
||||
GROK("grok");
|
||||
GROK("grok"),
|
||||
/**
|
||||
* ollama
|
||||
*/
|
||||
OLLAMA("ollama");
|
||||
|
||||
private final String value;
|
||||
|
||||
|
||||
@@ -192,4 +192,19 @@ public class Models {
|
||||
}
|
||||
}
|
||||
|
||||
// Ollama的模型
|
||||
public enum Ollama {
|
||||
QWEN3_32B("qwen3:32b");
|
||||
|
||||
private final String model;
|
||||
|
||||
Ollama(String model) {
|
||||
this.model = model;
|
||||
}
|
||||
|
||||
public String getModel() {
|
||||
return model;
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@@ -0,0 +1,76 @@
|
||||
/*
|
||||
* Copyright (c) 2025 Hutool Team and hutool.cn
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package cn.hutool.ai.model.ollama;
|
||||
|
||||
/**
|
||||
* Ollama公共类
|
||||
*
|
||||
* @author yangruoyu-yumeisoft
|
||||
* @since 5.8.40
|
||||
*/
|
||||
public class OllamaCommon {
|
||||
|
||||
/**
|
||||
* Ollama模型格式枚举
|
||||
*/
|
||||
public enum OllamaFormat {
|
||||
/**
|
||||
* JSON格式
|
||||
*/
|
||||
JSON("json"),
|
||||
/**
|
||||
* 无格式
|
||||
*/
|
||||
NONE("");
|
||||
|
||||
private final String format;
|
||||
|
||||
OllamaFormat(String format) {
|
||||
this.format = format;
|
||||
}
|
||||
|
||||
public String getFormat() {
|
||||
return format;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Ollama选项常量
|
||||
*/
|
||||
public static class Options {
|
||||
/**
|
||||
* 温度参数
|
||||
*/
|
||||
public static final String TEMPERATURE = "temperature";
|
||||
/**
|
||||
* top_p参数
|
||||
*/
|
||||
public static final String TOP_P = "top_p";
|
||||
/**
|
||||
* top_k参数
|
||||
*/
|
||||
public static final String TOP_K = "top_k";
|
||||
/**
|
||||
* 最大token数
|
||||
*/
|
||||
public static final String NUM_PREDICT = "num_predict";
|
||||
/**
|
||||
* 随机种子
|
||||
*/
|
||||
public static final String SEED = "seed";
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,55 @@
|
||||
/*
|
||||
* Copyright (c) 2025 Hutool Team and hutool.cn
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package cn.hutool.ai.model.ollama;
|
||||
|
||||
import cn.hutool.ai.Models;
|
||||
import cn.hutool.ai.core.BaseConfig;
|
||||
|
||||
/**
|
||||
* Ollama配置类,初始化API接口地址,设置默认的模型
|
||||
*
|
||||
* @author yangruoyu-yumeisoft
|
||||
* @since 5.8.40
|
||||
*/
|
||||
public class OllamaConfig extends BaseConfig {
|
||||
|
||||
private final String API_URL = "http://localhost:11434";
|
||||
|
||||
private final String DEFAULT_MODEL = Models.Ollama.QWEN3_32B.getModel();
|
||||
|
||||
public OllamaConfig() {
|
||||
setApiUrl(API_URL);
|
||||
setModel(DEFAULT_MODEL);
|
||||
}
|
||||
|
||||
public OllamaConfig(String apiUrl) {
|
||||
this();
|
||||
setApiUrl(apiUrl);
|
||||
}
|
||||
|
||||
public OllamaConfig(String apiUrl, String model) {
|
||||
this();
|
||||
setApiUrl(apiUrl);
|
||||
setModel(model);
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getModelName() {
|
||||
return "ollama";
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,39 @@
|
||||
/*
|
||||
* Copyright (c) 2025 Hutool Team and hutool.cn
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package cn.hutool.ai.model.ollama;
|
||||
|
||||
import cn.hutool.ai.core.AIConfig;
|
||||
import cn.hutool.ai.core.AIServiceProvider;
|
||||
|
||||
/**
|
||||
* 创建Ollama服务实现类
|
||||
*
|
||||
* @author yangruoyu-yumeisoft
|
||||
* @since 5.8.40
|
||||
*/
|
||||
public class OllamaProvider implements AIServiceProvider {
|
||||
|
||||
@Override
|
||||
public String getServiceName() {
|
||||
return "ollama";
|
||||
}
|
||||
|
||||
@Override
|
||||
public OllamaService create(final AIConfig config) {
|
||||
return new OllamaServiceImpl(config);
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,151 @@
|
||||
/*
|
||||
* Copyright (c) 2025 Hutool Team and hutool.cn
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package cn.hutool.ai.model.ollama;
|
||||
|
||||
import cn.hutool.ai.core.AIService;
|
||||
import cn.hutool.ai.core.Message;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.function.Consumer;
|
||||
|
||||
/**
|
||||
* Ollama特有的功能
|
||||
*
|
||||
* @author yangruoyu-yumeisoft
|
||||
* @since 5.8.40
|
||||
*/
|
||||
public interface OllamaService extends AIService {
|
||||
|
||||
/**
|
||||
* 生成文本补全
|
||||
*
|
||||
* @param prompt 输入提示
|
||||
* @return AI回答
|
||||
* @since 5.8.40
|
||||
*/
|
||||
String generate(String prompt);
|
||||
|
||||
/**
|
||||
* 生成文本补全-SSE流式输出
|
||||
*
|
||||
* @param prompt 输入提示
|
||||
* @param callback 流式数据回调函数
|
||||
* @since 5.8.40
|
||||
*/
|
||||
void generate(String prompt, Consumer<String> callback);
|
||||
|
||||
/**
|
||||
* 生成文本补全(带选项)
|
||||
*
|
||||
* @param prompt 输入提示
|
||||
* @param format 响应格式
|
||||
* @return AI回答
|
||||
* @since 5.8.40
|
||||
*/
|
||||
String generate(String prompt, String format);
|
||||
|
||||
/**
|
||||
* 生成文本补全(带选项)-SSE流式输出
|
||||
*
|
||||
* @param prompt 输入提示
|
||||
* @param format 响应格式
|
||||
* @param callback 流式数据回调函数
|
||||
* @since 5.8.40
|
||||
*/
|
||||
void generate(String prompt, String format, Consumer<String> callback);
|
||||
|
||||
/**
|
||||
* 生成文本嵌入向量
|
||||
*
|
||||
* @param prompt 输入文本
|
||||
* @return 嵌入向量结果
|
||||
* @since 5.8.40
|
||||
*/
|
||||
String embeddings(String prompt);
|
||||
|
||||
/**
|
||||
* 列出本地可用的模型
|
||||
*
|
||||
* @return 模型列表
|
||||
* @since 5.8.40
|
||||
*/
|
||||
String listModels();
|
||||
|
||||
/**
|
||||
* 显示模型信息
|
||||
*
|
||||
* @param modelName 模型名称
|
||||
* @return 模型信息
|
||||
* @since 5.8.40
|
||||
*/
|
||||
String showModel(String modelName);
|
||||
|
||||
/**
|
||||
* 拉取模型
|
||||
*
|
||||
* @param modelName 模型名称
|
||||
* @return 拉取结果
|
||||
* @since 5.8.40
|
||||
*/
|
||||
String pullModel(String modelName);
|
||||
|
||||
/**
|
||||
* 删除模型
|
||||
*
|
||||
* @param modelName 模型名称
|
||||
* @return 删除结果
|
||||
* @since 5.8.40
|
||||
*/
|
||||
String deleteModel(String modelName);
|
||||
|
||||
/**
|
||||
* 复制模型
|
||||
*
|
||||
* @param source 源模型名称
|
||||
* @param destination 目标模型名称
|
||||
* @return 复制结果
|
||||
* @since 5.8.40
|
||||
*/
|
||||
String copyModel(String source, String destination);
|
||||
|
||||
/**
|
||||
* 简化的对话方法
|
||||
*
|
||||
* @param prompt 对话题词
|
||||
* @return AI回答
|
||||
* @since 5.8.40
|
||||
*/
|
||||
default String chat(String prompt) {
|
||||
final List<Message> messages = new ArrayList<>();
|
||||
messages.add(new Message("user", prompt));
|
||||
return chat(messages);
|
||||
}
|
||||
|
||||
/**
|
||||
* 简化的对话方法-SSE流式输出
|
||||
*
|
||||
* @param prompt 对话题词
|
||||
* @param callback 流式数据回调函数
|
||||
* @since 5.8.40
|
||||
*/
|
||||
default void chat(String prompt, Consumer<String> callback) {
|
||||
final List<Message> messages = new ArrayList<>();
|
||||
messages.add(new Message("user", prompt));
|
||||
chat(messages, callback);
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,273 @@
|
||||
/*
|
||||
* Copyright (c) 2025 Hutool Team and hutool.cn
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package cn.hutool.ai.model.ollama;
|
||||
|
||||
import cn.hutool.ai.AIException;
|
||||
import cn.hutool.ai.core.AIConfig;
|
||||
import cn.hutool.ai.core.BaseAIService;
|
||||
import cn.hutool.ai.core.Message;
|
||||
import cn.hutool.core.bean.BeanPath;
|
||||
import cn.hutool.core.thread.ThreadUtil;
|
||||
import cn.hutool.core.util.StrUtil;
|
||||
import cn.hutool.http.Header;
|
||||
import cn.hutool.http.HttpRequest;
|
||||
import cn.hutool.http.HttpResponse;
|
||||
import cn.hutool.json.JSONObject;
|
||||
import cn.hutool.json.JSONUtil;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.HashMap;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.function.Consumer;
|
||||
|
||||
/**
|
||||
* Ollama服务,AI具体功能的实现
|
||||
*
|
||||
* @author yangruoyu-yumeisoft
|
||||
* @since 5.8.40
|
||||
*/
|
||||
public class OllamaServiceImpl extends BaseAIService implements OllamaService {
|
||||
|
||||
// 对话补全
|
||||
private static final String CHAT_ENDPOINT = "/api/chat";
|
||||
// 文本生成
|
||||
private static final String GENERATE_ENDPOINT = "/api/generate";
|
||||
// 文本嵌入
|
||||
private static final String EMBEDDINGS_ENDPOINT = "/api/embeddings";
|
||||
// 列出模型
|
||||
private static final String LIST_MODELS_ENDPOINT = "/api/tags";
|
||||
// 显示模型信息
|
||||
private static final String SHOW_MODEL_ENDPOINT = "/api/show";
|
||||
// 拉取模型
|
||||
private static final String PULL_MODEL_ENDPOINT = "/api/pull";
|
||||
// 删除模型
|
||||
private static final String DELETE_MODEL_ENDPOINT = "/api/delete";
|
||||
// 复制模型
|
||||
private static final String COPY_MODEL_ENDPOINT = "/api/copy";
|
||||
|
||||
/**
|
||||
* 构造函数
|
||||
*
|
||||
* @param config AI配置
|
||||
*/
|
||||
public OllamaServiceImpl(final AIConfig config) {
|
||||
super(config);
|
||||
}
|
||||
|
||||
@Override
|
||||
public String chat(final List<Message> messages) {
|
||||
final String paramJson = buildChatRequestBody(messages);
|
||||
final HttpResponse response = sendPost(CHAT_ENDPOINT, paramJson);
|
||||
JSONObject responseJson = JSONUtil.parseObj(response.body());
|
||||
Object errorMessage = BeanPath.create("error").get(responseJson);
|
||||
if(errorMessage!=null){
|
||||
throw new RuntimeException(errorMessage.toString());
|
||||
}
|
||||
return BeanPath.create("message.content").get(responseJson).toString();
|
||||
}
|
||||
|
||||
@Override
|
||||
public void chat(final List<Message> messages, final Consumer<String> callback) {
|
||||
Map<String, Object> paramMap = buildChatStreamRequestBody(messages);
|
||||
ThreadUtil.newThread(() -> sendPostStream(CHAT_ENDPOINT, paramMap, callback::accept), "ollama-chat-sse").start();
|
||||
}
|
||||
|
||||
@Override
|
||||
public String generate(String prompt) {
|
||||
final String paramJson = buildGenerateRequestBody(prompt, null);
|
||||
final HttpResponse response = sendPost(GENERATE_ENDPOINT, paramJson);
|
||||
return response.body();
|
||||
}
|
||||
|
||||
@Override
|
||||
public void generate(String prompt, Consumer<String> callback) {
|
||||
Map<String, Object> paramMap = buildGenerateStreamRequestBody(prompt, null);
|
||||
ThreadUtil.newThread(() -> sendPostStream(GENERATE_ENDPOINT, paramMap, callback::accept), "ollama-generate-sse").start();
|
||||
}
|
||||
|
||||
@Override
|
||||
public String generate(String prompt, String format) {
|
||||
final String paramJson = buildGenerateRequestBody(prompt, format);
|
||||
final HttpResponse response = sendPost(GENERATE_ENDPOINT, paramJson);
|
||||
return response.body();
|
||||
}
|
||||
|
||||
@Override
|
||||
public void generate(String prompt, String format, Consumer<String> callback) {
|
||||
Map<String, Object> paramMap = buildGenerateStreamRequestBody(prompt, format);
|
||||
ThreadUtil.newThread(() -> sendPostStream(GENERATE_ENDPOINT, paramMap, callback::accept), "ollama-generate-sse").start();
|
||||
}
|
||||
|
||||
@Override
|
||||
public String embeddings(String prompt) {
|
||||
final String paramJson = buildEmbeddingsRequestBody(prompt);
|
||||
final HttpResponse response = sendPost(EMBEDDINGS_ENDPOINT, paramJson);
|
||||
return response.body();
|
||||
}
|
||||
|
||||
@Override
|
||||
public String listModels() {
|
||||
final HttpResponse response = sendGet(LIST_MODELS_ENDPOINT);
|
||||
return response.body();
|
||||
}
|
||||
|
||||
@Override
|
||||
public String showModel(String modelName) {
|
||||
final String paramJson = buildShowModelRequestBody(modelName);
|
||||
final HttpResponse response = sendPost(SHOW_MODEL_ENDPOINT, paramJson);
|
||||
return response.body();
|
||||
}
|
||||
|
||||
@Override
|
||||
public String pullModel(String modelName) {
|
||||
final String paramJson = buildPullModelRequestBody(modelName);
|
||||
final HttpResponse response = sendPost(PULL_MODEL_ENDPOINT, paramJson);
|
||||
return response.body();
|
||||
}
|
||||
|
||||
@Override
|
||||
public String deleteModel(String modelName) {
|
||||
final String paramJson = buildDeleteModelRequestBody(modelName);
|
||||
final HttpResponse response = sendDeleteRequest(DELETE_MODEL_ENDPOINT, paramJson);
|
||||
return response.body();
|
||||
}
|
||||
|
||||
@Override
|
||||
public String copyModel(String source, String destination) {
|
||||
final String paramJson = buildCopyModelRequestBody(source, destination);
|
||||
final HttpResponse response = sendPost(COPY_MODEL_ENDPOINT, paramJson);
|
||||
return response.body();
|
||||
}
|
||||
|
||||
// 构建chat请求体
|
||||
private String buildChatRequestBody(final List<Message> messages) {
|
||||
final Map<String, Object> paramMap = new HashMap<>();
|
||||
paramMap.put("stream",false);
|
||||
paramMap.put("model", config.getModel());
|
||||
paramMap.put("messages", messages);
|
||||
// 合并其他参数
|
||||
paramMap.putAll(config.getAdditionalConfigMap());
|
||||
|
||||
return JSONUtil.toJsonStr(paramMap);
|
||||
}
|
||||
|
||||
// 构建chatStream请求体
|
||||
private Map<String, Object> buildChatStreamRequestBody(final List<Message> messages) {
|
||||
final Map<String, Object> paramMap = new HashMap<>();
|
||||
paramMap.put("stream", true);
|
||||
paramMap.put("model", config.getModel());
|
||||
paramMap.put("messages", messages);
|
||||
// 合并其他参数
|
||||
paramMap.putAll(config.getAdditionalConfigMap());
|
||||
|
||||
return paramMap;
|
||||
}
|
||||
|
||||
// 构建generate请求体
|
||||
private String buildGenerateRequestBody(final String prompt, final String format) {
|
||||
final Map<String, Object> paramMap = new HashMap<>();
|
||||
paramMap.put("model", config.getModel());
|
||||
paramMap.put("prompt", prompt);
|
||||
if (StrUtil.isNotBlank(format)) {
|
||||
paramMap.put("format", format);
|
||||
}
|
||||
// 合并其他参数
|
||||
paramMap.putAll(config.getAdditionalConfigMap());
|
||||
|
||||
return JSONUtil.toJsonStr(paramMap);
|
||||
}
|
||||
|
||||
// 构建generateStream请求体
|
||||
private Map<String, Object> buildGenerateStreamRequestBody(final String prompt, final String format) {
|
||||
final Map<String, Object> paramMap = new HashMap<>();
|
||||
paramMap.put("stream", true);
|
||||
paramMap.put("model", config.getModel());
|
||||
paramMap.put("prompt", prompt);
|
||||
if (StrUtil.isNotBlank(format)) {
|
||||
paramMap.put("format", format);
|
||||
}
|
||||
// 合并其他参数
|
||||
paramMap.putAll(config.getAdditionalConfigMap());
|
||||
|
||||
return paramMap;
|
||||
}
|
||||
|
||||
// 构建embeddings请求体
|
||||
private String buildEmbeddingsRequestBody(final String prompt) {
|
||||
final Map<String, Object> paramMap = new HashMap<>();
|
||||
paramMap.put("model", config.getModel());
|
||||
paramMap.put("prompt", prompt);
|
||||
// 合并其他参数
|
||||
paramMap.putAll(config.getAdditionalConfigMap());
|
||||
|
||||
return JSONUtil.toJsonStr(paramMap);
|
||||
}
|
||||
|
||||
// 构建showModel请求体
|
||||
private String buildShowModelRequestBody(final String modelName) {
|
||||
final Map<String, Object> paramMap = new HashMap<>();
|
||||
paramMap.put("name", modelName);
|
||||
|
||||
return JSONUtil.toJsonStr(paramMap);
|
||||
}
|
||||
|
||||
// 构建pullModel请求体
|
||||
private String buildPullModelRequestBody(final String modelName) {
|
||||
final Map<String, Object> paramMap = new HashMap<>();
|
||||
paramMap.put("name", modelName);
|
||||
|
||||
return JSONUtil.toJsonStr(paramMap);
|
||||
}
|
||||
|
||||
// 构建deleteModel请求体
|
||||
private String buildDeleteModelRequestBody(final String modelName) {
|
||||
final Map<String, Object> paramMap = new HashMap<>();
|
||||
paramMap.put("name", modelName);
|
||||
|
||||
return JSONUtil.toJsonStr(paramMap);
|
||||
}
|
||||
|
||||
/**
|
||||
* 发送DELETE请求
|
||||
*
|
||||
* @param endpoint 请求端点
|
||||
* @param paramJson 请求参数JSON
|
||||
* @return 响应结果
|
||||
*/
|
||||
private HttpResponse sendDeleteRequest(String endpoint, String paramJson) {
|
||||
try {
|
||||
return HttpRequest.delete(config.getApiUrl() + endpoint)
|
||||
.header(Header.CONTENT_TYPE, "application/json")
|
||||
.header(Header.ACCEPT, "application/json")
|
||||
.body(paramJson)
|
||||
.timeout(config.getTimeout())
|
||||
.execute();
|
||||
} catch (Exception e) {
|
||||
throw new AIException("Failed to send DELETE request: " + e.getMessage(), e);
|
||||
}
|
||||
}
|
||||
|
||||
// 构建copyModel请求体
|
||||
private String buildCopyModelRequestBody(final String source, final String destination) {
|
||||
Map<String, Object> requestBody = new HashMap<>();
|
||||
requestBody.put("source", source);
|
||||
requestBody.put("destination", destination);
|
||||
return JSONUtil.toJsonStr(requestBody);
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,42 @@
|
||||
/*
|
||||
* Copyright (c) 2025 Hutool Team and hutool.cn
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
/**
|
||||
* 对Ollama的封装实现.
|
||||
*
|
||||
* 使用方法:
|
||||
* // 创建AI服务
|
||||
* OllamaService aiService = AIServiceFactory.getAIService(
|
||||
* new AIConfigBuilder(ModelName.OLLAMA.getValue())
|
||||
* .setApiUrl("http://localhost:11434")
|
||||
* .setModel("qwen2.5-coder:32b")
|
||||
* .build(),
|
||||
* OllamaService.class
|
||||
* );
|
||||
*
|
||||
* // 构造上下文
|
||||
* List<Message> messageList=new ArrayList<>();
|
||||
* messageList.add(new Message("system","你是一个疯疯癫癫的机器人"));
|
||||
* messageList.add(new Message("user","你能帮我做什么"));
|
||||
*
|
||||
* // 输出对话结果
|
||||
* System.out.println(aiService.chat(messageList));
|
||||
*
|
||||
* @author yangruoyu-yumeisoft
|
||||
* @since 5.8.40
|
||||
*/
|
||||
|
||||
package cn.hutool.ai.model.ollama;
|
||||
@@ -3,3 +3,4 @@ cn.hutool.ai.model.deepseek.DeepSeekConfig
|
||||
cn.hutool.ai.model.openai.OpenaiConfig
|
||||
cn.hutool.ai.model.doubao.DoubaoConfig
|
||||
cn.hutool.ai.model.grok.GrokConfig
|
||||
cn.hutool.ai.model.ollama.OllamaConfig
|
||||
|
||||
@@ -3,3 +3,4 @@ cn.hutool.ai.model.deepseek.DeepSeekProvider
|
||||
cn.hutool.ai.model.openai.OpenaiProvider
|
||||
cn.hutool.ai.model.doubao.DoubaoProvider
|
||||
cn.hutool.ai.model.grok.GrokProvider
|
||||
cn.hutool.ai.model.ollama.OllamaProvider
|
||||
|
||||
@@ -0,0 +1,256 @@
|
||||
/*
|
||||
* Copyright (c) 2025 Hutool Team and hutool.cn
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package cn.hutool.ai.model.ollama;
|
||||
|
||||
import cn.hutool.ai.AIServiceFactory;
|
||||
import cn.hutool.ai.ModelName;
|
||||
import cn.hutool.ai.core.AIConfigBuilder;
|
||||
import cn.hutool.ai.core.Message;
|
||||
import cn.hutool.core.bean.BeanPath;
|
||||
import cn.hutool.core.thread.ThreadUtil;
|
||||
import cn.hutool.core.util.StrUtil;
|
||||
import cn.hutool.json.JSON;
|
||||
import cn.hutool.json.JSONArray;
|
||||
import cn.hutool.json.JSONUtil;
|
||||
import org.junit.jupiter.api.Disabled;
|
||||
import org.junit.jupiter.api.Test;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.concurrent.atomic.AtomicBoolean;
|
||||
import java.util.concurrent.atomic.AtomicReference;
|
||||
|
||||
import static org.junit.jupiter.api.Assertions.assertNotNull;
|
||||
|
||||
/**
|
||||
* OllamaService
|
||||
*
|
||||
* @author yangruoyu-yumeisoft
|
||||
* @since 5.8.40
|
||||
*/
|
||||
class OllamaServiceTest {
|
||||
// 创建service
|
||||
OllamaService ollamaService = AIServiceFactory.getAIService(
|
||||
new AIConfigBuilder(ModelName.OLLAMA.getValue())
|
||||
// 这里填写Ollama服务的地址
|
||||
.setApiUrl("http://127.0.0.1:11434")
|
||||
// 这里填写使用的模型
|
||||
.setModel("qwen2.5-coder:32b")
|
||||
.build(),
|
||||
OllamaService.class
|
||||
);
|
||||
|
||||
// 假设有一个Java工程师的Agent提示词
|
||||
String javaEngineerPrompt="# 角色 \n" +
|
||||
"你是一位精通Spring Boot 3.0的资深Java全栈工程师,具备以下核心能力: \n" +
|
||||
"- 精通Spring Boot 3.0新特性与最佳实践 \n" +
|
||||
"- 熟练整合Hutool工具包、Redis数据访问、Feign远程调用、FreeMarker模板引擎 \n" +
|
||||
"- 能输出符合工程规范的代码结构和配置文件 \n" +
|
||||
"- 注重代码可读性与注释规范 \n" +
|
||||
"\n" +
|
||||
"# 任务 \n" +
|
||||
"请完成以下编程任务(按优先级排序): \n" +
|
||||
"1. **核心要求** \n" +
|
||||
" - 使用Spring Boot 3.0构建项目 \n" +
|
||||
" - 必须包含以下依赖: \n" +
|
||||
" - `cn.hutool:hutool-all`(最新版) \n" +
|
||||
" - `org.springframework.boot:spring-boot-starter-data-redis` \n" +
|
||||
" - `org.springframework.cloud:spring-cloud-starter-openfeign` \n" +
|
||||
" - `org.springframework.boot:spring-boot-starter-freemarker` \n" +
|
||||
"2. **约束条件** \n" +
|
||||
" - 代码需符合Java 17语法规范 \n" +
|
||||
" - 每个类必须包含Javadoc风格的类注释 \n" +
|
||||
" - 关键方法需添加`@Api`/`@ApiOperation`注解(若涉及接口) \n" +
|
||||
" - Redis操作需使用`RedisTemplate`实现 \n" +
|
||||
"3. **实现流程** \n" +
|
||||
" ``` \n" +
|
||||
" 1. 生成pom.xml依赖配置 \n" +
|
||||
" 2. 创建基础配置类(如RedisConfig) \n" +
|
||||
" 3. 编写Feign客户端接口 \n" +
|
||||
" 4. 实现FreeMarker模板渲染服务 \n" +
|
||||
" 5. 提供完整Controller示例 \n" +
|
||||
" ``` \n" +
|
||||
"\n" +
|
||||
"# 输出要求 \n" +
|
||||
"请以严格Markdown格式输出,每个模块独立代码块: \n" +
|
||||
"```markdown \n" +
|
||||
"## 1. 项目依赖配置(pom.xml片段) \n" +
|
||||
"```xml \n" +
|
||||
"<dependency>...</dependency> \n" +
|
||||
"``` \n" +
|
||||
"\n" +
|
||||
"## 2. Redis配置类 \n" +
|
||||
"```java \n" +
|
||||
"@Configuration \n" +
|
||||
"public class RedisConfig { ... } \n" +
|
||||
"``` \n" +
|
||||
"\n" +
|
||||
"## 3. Feign客户端示例 \n" +
|
||||
"```java \n" +
|
||||
"@FeignClient(name = \"...\") \n" +
|
||||
"public interface ... { ... } \n" +
|
||||
"``` \n" +
|
||||
"\n" +
|
||||
"## 4. FreeMarker模板服务 \n" +
|
||||
"```java \n" +
|
||||
"@Service \n" +
|
||||
"public class TemplateService { ... } \n" +
|
||||
"``` \n" +
|
||||
"\n" +
|
||||
"## 5. 控制器示例 \n" +
|
||||
"```java \n" +
|
||||
"@RestController \n" +
|
||||
"@RequestMapping(\"/example\") \n" +
|
||||
"public class ExampleController { ... } \n" +
|
||||
"``` \n" +
|
||||
"``` \n" +
|
||||
"\n" +
|
||||
"# 示例片段(供格式参考) \n" +
|
||||
"```java \n" +
|
||||
"/** \n" +
|
||||
" * 示例Feign客户端 \n" +
|
||||
" * @since 1.0.0 \n" +
|
||||
" */ \n" +
|
||||
"@FeignClient(name = \"demo-service\", url = \"${demo.service.url}\") \n" +
|
||||
"public interface DemoClient { \n" +
|
||||
"\n" +
|
||||
" @GetMapping(\"/data/{id}\") \n" +
|
||||
" @ApiOperation(\"获取示例数据\") \n" +
|
||||
" ResponseEntity<String> getData(@PathVariable(\"id\") Long id); \n" +
|
||||
"} \n" +
|
||||
"``` \n" +
|
||||
"\n" +
|
||||
"请按此规范输出完整代码结构,确保自动化程序可直接解析生成项目文件。";
|
||||
|
||||
/**
|
||||
* 同步方式调用
|
||||
*/
|
||||
@Test
|
||||
@Disabled
|
||||
void testSimple() {
|
||||
final String answer = ollamaService.chat("写一个疯狂星期四广告词");
|
||||
assertNotNull(answer);
|
||||
}
|
||||
|
||||
/**
|
||||
* 按流方式输出
|
||||
*/
|
||||
@Test
|
||||
@Disabled
|
||||
void testStream() {
|
||||
AtomicBoolean isDone = new AtomicBoolean(false);
|
||||
AtomicReference<String> errorMessage = new AtomicReference<>();
|
||||
ollamaService.chat("写一个疯狂星期四广告词", data -> {
|
||||
// 输出到控制台
|
||||
JSON streamData = JSONUtil.parse(data);
|
||||
if (streamData.getByPath("error") != null) {
|
||||
isDone.set(true);
|
||||
errorMessage.set(streamData.getByPath("error").toString());
|
||||
return;
|
||||
}
|
||||
|
||||
if ("true".equals(streamData.getByPath("done").toString())) {
|
||||
isDone.set(true);
|
||||
}
|
||||
});
|
||||
// 轮询检查结束标志
|
||||
while (!isDone.get()) {
|
||||
ThreadUtil.sleep(100);
|
||||
}
|
||||
if (errorMessage.get() != null) {
|
||||
throw new RuntimeException(errorMessage.get());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 带历史上下文的同步方式调用
|
||||
*/
|
||||
@Test
|
||||
@Disabled
|
||||
void testSimpleWithHistory(){
|
||||
List<Message> messageList=new ArrayList<>();
|
||||
messageList.add(new Message("system",javaEngineerPrompt));
|
||||
messageList.add(new Message("user","帮我写一个Java通过Post方式发送JSON给HTTP接口,请求头带有token"));
|
||||
String result = ollamaService.chat(messageList);
|
||||
assertNotNull(result);
|
||||
}
|
||||
|
||||
@Test
|
||||
@Disabled
|
||||
void testStreamWithHistory(){
|
||||
List<Message> messageList=new ArrayList<>();
|
||||
messageList.add(new Message("system",javaEngineerPrompt));
|
||||
messageList.add(new Message("user","帮我写一个Java通过Post方式发送JSON给HTTP接口,请求头带有token"));
|
||||
AtomicBoolean isDone = new AtomicBoolean(false);
|
||||
AtomicReference<String> errorMessage = new AtomicReference<>();
|
||||
ollamaService.chat(messageList, data -> {
|
||||
// 输出到控制台
|
||||
JSON streamData = JSONUtil.parse(data);
|
||||
if (streamData.getByPath("error") != null) {
|
||||
isDone.set(true);
|
||||
errorMessage.set(streamData.getByPath("error").toString());
|
||||
return;
|
||||
}
|
||||
|
||||
if ("true".equals(streamData.getByPath("done").toString())) {
|
||||
isDone.set(true);
|
||||
}
|
||||
});
|
||||
// 轮询检查结束标志
|
||||
while (!isDone.get()) {
|
||||
ThreadUtil.sleep(100);
|
||||
}
|
||||
if (errorMessage.get() != null) {
|
||||
throw new RuntimeException(errorMessage.get());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 列出所有已经拉取到服务器上的模型
|
||||
*/
|
||||
@Test
|
||||
@Disabled
|
||||
void testListModels(){
|
||||
String models = ollamaService.listModels();
|
||||
JSONArray modelList = JSONUtil.parse(models).getByPath("models", JSONArray.class);
|
||||
}
|
||||
|
||||
/**
|
||||
* 让Ollama拉取模型
|
||||
*/
|
||||
@Test
|
||||
@Disabled
|
||||
void testPullModel(){
|
||||
String result = ollamaService.pullModel("qwen2.5:0.5b");
|
||||
List<String> lines = StrUtil.splitTrim(result, "\n");
|
||||
for (String line : lines) {
|
||||
if(line.contains("error")){
|
||||
throw new RuntimeException(JSONUtil.parse(line).getByPath("error").toString());
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 让Ollama删除已经存在的模型
|
||||
*/
|
||||
@Test
|
||||
@Disabled
|
||||
void testDeleteModel(){
|
||||
// 不会返回任何信息
|
||||
ollamaService.deleteModel("qwen2.5:0.5b");
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user