package com.yb;
|
/*
|
*@title SimpleTest
|
*@description
|
*@author yb
|
*@version 1.0
|
*@create 2024/5/28 20:44
|
*/
|
|
import jakarta.annotation.Resource;
|
import org.junit.jupiter.api.Test;
|
import org.springframework.ai.chat.model.ChatResponse;
|
import org.springframework.ai.chat.prompt.Prompt;
|
import org.springframework.ai.ollama.OllamaChatModel;
|
import org.springframework.ai.ollama.api.OllamaOptions;
|
import org.springframework.boot.test.context.SpringBootTest;
|
import reactor.core.publisher.Flux;
|
|
@SpringBootTest
|
public class SimpleTest {
|
|
@Resource
|
private OllamaChatModel chatModel;
|
|
// @Test
|
// public void test(){
|
// //使用下载的 gemma:7b 配置模型属性，模型属性配置可以参考 OllamaChatModel 构造函数
|
// OllamaOptions options = OllamaOptions.create().withModel("llama3:8b");
|
// Prompt prompt = new Prompt("你好", options);
|
// //调用聊天模型，获取返回值对象
|
// ChatResponse res = chatModel.call(prompt);
|
// //获取 AI 回答字符串
|
// System.out.println(res.getResult().getOutput().getContent());
|
// }
|
//
|
// /**
|
// * 流式返回数据
|
// */
|
// @Test
|
// public void test2(){
|
// //使用下载的 gemma:7b 配置模型属性，模型属性配置可以参考 OllamaChatModel 构造函数
|
// OllamaOptions options = OllamaOptions.create().withModel("llama3:8b");
|
// Prompt prompt = new Prompt("你好", options);
|
// Flux<ChatResponse> flux = chatModel.stream(prompt);
|
// flux.toStream().forEach(res -> {
|
// System.out.println(res.getResult().getOutput().getContent());
|
// });
|
// }
|
|
@Test
|
public void test3(){
|
String res = chatModel.call("java 排序算法");
|
//获取模型返回结果
|
System.out.println(res);
|
}
|
|
}
|
