|
|
|
@ -0,0 +1,44 @@
|
|
|
|
|
package Tools.TestUnit;
|
|
|
|
|
import java.util.Arrays;
|
|
|
|
|
import java.lang.System;
|
|
|
|
|
import com.alibaba.dashscope.aigc.generation.Generation;
|
|
|
|
|
import com.alibaba.dashscope.aigc.generation.GenerationParam;
|
|
|
|
|
import com.alibaba.dashscope.aigc.generation.GenerationResult;
|
|
|
|
|
import com.alibaba.dashscope.common.Message;
|
|
|
|
|
import com.alibaba.dashscope.common.Role;
|
|
|
|
|
import com.alibaba.dashscope.exception.ApiException;
|
|
|
|
|
import com.alibaba.dashscope.exception.InputRequiredException;
|
|
|
|
|
import com.alibaba.dashscope.exception.NoApiKeyException;
|
|
|
|
|
|
|
|
|
|
public class TestDeepSeek {
|
|
|
|
|
public static GenerationResult callWithMessage() throws ApiException, NoApiKeyException, InputRequiredException {
|
|
|
|
|
Generation gen = new Generation();
|
|
|
|
|
Message systemMsg = Message.builder()
|
|
|
|
|
.role(Role.SYSTEM.getValue())
|
|
|
|
|
.content("You are a helpful assistant.")
|
|
|
|
|
.build();
|
|
|
|
|
Message userMsg = Message.builder()
|
|
|
|
|
.role(Role.USER.getValue())
|
|
|
|
|
.content("你是谁?")
|
|
|
|
|
.build();
|
|
|
|
|
GenerationParam param = GenerationParam.builder()
|
|
|
|
|
// 若没有配置环境变量,请用百炼API Key将下行替换为:.apiKey("sk-xxx")
|
|
|
|
|
.apiKey("sk-01d13a39e09844038322108ecdbd1bbc")
|
|
|
|
|
// 模型列表:https://help.aliyun.com/zh/model-studio/getting-started/models
|
|
|
|
|
.model("deepseek-r1")
|
|
|
|
|
.messages(Arrays.asList(systemMsg, userMsg))
|
|
|
|
|
.resultFormat(GenerationParam.ResultFormat.MESSAGE)
|
|
|
|
|
.build();
|
|
|
|
|
return gen.call(param);
|
|
|
|
|
}
|
|
|
|
|
public static void main(String[] args) {
|
|
|
|
|
try {
|
|
|
|
|
GenerationResult result = callWithMessage();
|
|
|
|
|
System.out.println(result.getOutput().getChoices().get(0).getMessage().getContent());
|
|
|
|
|
} catch (ApiException | NoApiKeyException | InputRequiredException e) {
|
|
|
|
|
System.err.println("错误信息:"+e.getMessage());
|
|
|
|
|
System.out.println("请参考文档:https://help.aliyun.com/zh/model-studio/developer-reference/error-code");
|
|
|
|
|
}
|
|
|
|
|
System.exit(0);
|
|
|
|
|
}
|
|
|
|
|
}
|