Pre Merge pull request !76 from orcasjh/feat-orcarouter
commit
fef96a4f32
@ -0,0 +1,48 @@
|
||||
package com.xxl.job.executor.test.orcarouter;
|
||||
|
||||
import com.xxl.job.core.executor.impl.XxlJobSpringExecutor;
|
||||
import com.xxl.job.executor.config.OrcaRouterChatModel;
|
||||
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.openai.OpenAiChatOptions;
|
||||
import org.springframework.boot.test.context.SpringBootTest;
|
||||
import org.springframework.test.context.bean.override.mockito.MockitoBean;
|
||||
|
||||
@SpringBootTest
|
||||
public class OrcaRouterLiveTest {
|
||||
private static final Logger logger = LoggerFactory.getLogger(OrcaRouterLiveTest.class);
|
||||
|
||||
// ignore
|
||||
@MockitoBean
|
||||
private XxlJobSpringExecutor xxlJobSpringExecutor;
|
||||
|
||||
@Resource
|
||||
private OrcaRouterChatModel orcaRouterChatModel;
|
||||
|
||||
@Test
|
||||
public void chatTest() {
|
||||
|
||||
String prompt = "你是一个研发工程师,擅长解决技术类问题。";
|
||||
String input = "请简单介绍一下你自己";
|
||||
|
||||
// build chat-client
|
||||
ChatClient orcaRouterChatClient = ChatClient
|
||||
.builder(orcaRouterChatModel)
|
||||
.defaultOptions(OpenAiChatOptions.builder().model("orcarouter/auto"))
|
||||
.build();
|
||||
|
||||
// call orcarouter
|
||||
String response = orcaRouterChatClient
|
||||
.prompt(prompt)
|
||||
.user(input)
|
||||
.call()
|
||||
.content();
|
||||
|
||||
logger.info("input: {}", input);
|
||||
logger.info("response: {}", response);
|
||||
}
|
||||
|
||||
}
|
||||
Loading…
Reference in new issue