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