3.3.1-release
xuxueli 2 months ago
parent 1869d6b796
commit b48a68a07a

@ -4,7 +4,7 @@ import org.junit.jupiter.api.Test;
import org.springframework.boot.test.context.SpringBootTest;
@SpringBootTest
public class XxlJobExecutorExampleBootApplicationTests {
public class BaseTests {
@Test
public void test() {

@ -0,0 +1,57 @@
package com.xxl.job.executor.test.ollama;
import com.xxl.job.core.executor.impl.XxlJobSpringExecutor;
import jakarta.annotation.Resource;
import org.junit.jupiter.api.Test;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.ai.chat.client.ChatClient;
import org.springframework.ai.chat.client.advisor.MessageChatMemoryAdvisor;
import org.springframework.ai.chat.client.advisor.SimpleLoggerAdvisor;
import org.springframework.ai.chat.memory.MessageWindowChatMemory;
import org.springframework.ai.ollama.OllamaChatModel;
import org.springframework.ai.ollama.api.OllamaChatOptions;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.bean.override.mockito.MockitoBean;
@SpringBootTest
public class OllamaTest {
private static final Logger logger = LoggerFactory.getLogger(OllamaTest.class);
// ignore
@MockitoBean
private XxlJobSpringExecutor xxlJobSpringExecutor;
@Resource
private OllamaChatModel ollamaChatModel;
@Test
public void test() {
String model = "qwen3:0.6b";
String prompt = "背景说明:你是一个研发工程师,擅长解决技术类问题。";
String input = "请写一个java程序实现一个方法输入一个字符串返回字符串的长度。";
// build chat-client
ChatClient ollamaChatClient = ChatClient
.builder(ollamaChatModel)
.defaultAdvisors(MessageChatMemoryAdvisor.builder(MessageWindowChatMemory.builder().build()).build())
.defaultAdvisors(SimpleLoggerAdvisor.builder().build())
.defaultOptions(OllamaChatOptions.builder().model(model).build())
.build();
// call ollama
String response = ollamaChatClient
.prompt(prompt)
.user(input)
.call()
.content();
logger.info("input: {}", input);
logger.info("response: {}", response);
}
}
Loading…
Cancel
Save