1 line
2.1 KiB
JSON
1 line
2.1 KiB
JSON
{"repo": "aishuijiaodexioshou/acdr-dev", "files": {"ruoyi-vue-pro/yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/chat/DeepSeekChatModelTests.java": {"7cdb16487f45c1dbee92f85cb83e3b60e35feac6": {"t": 1785747157.0642283, "n": 1584, "h": "a09843ae02846042", "c": "package cn.iocoder.yudao.framework.ai.chat;\n\nimport cn.iocoder.yudao.framework.ai.core.model.deepseek.DeepSeekChatModel;\nimport org.junit.jupiter.api.Disabled;\nimport org.junit.jupiter.api.Test;\nimport org.springframework.ai.chat.messages.Message;\nimport org.springframework.ai.chat.messages.SystemMessage;\nimport org.springframework.ai.chat.messages.UserMessage;\nimport org.springframework.ai.chat.model.ChatResponse;\nimport org.springframework.ai.chat.prompt.Prompt;\nimport reactor.core.publisher.Flux;\n\nimport java.util.ArrayList;\nimport java.util.List;\n\n/**\n * {@link DeepSeekChatModel} 集成测试\n *\n * @author 芋道源码\n */\npublic class DeepSeekChatModelTests {\n\n private final DeepSeekChatModel chatModel = new DeepSeekChatModel(\"sk-e94db327cc7d457d99a8de8810fc6b12\");\n\n @Test\n @Disabled\n public void testCall() {\n // 准备参数\n List<Message> messages = new ArrayList<>();\n messages.add(new SystemMessage(\"你是一个优质的文言文作者，用文言文描述着各城市的人文风景。\"));\n messages.add(new UserMessage(\"1 + 1 = ？\"));\n\n // 调用\n ChatResponse response = chatModel.call(new Prompt(messages));\n // 打印结果\n System.out.println(response);\n }\n\n @Test\n @Disabled\n public void testStream() {\n // 准备参数\n List<Message> messages = new ArrayList<>();\n messages.add(new SystemMessage(\"你是一个优质的文言文作者，用文言文描述着各城市的人文风景。\"));\n messages.add(new UserMessage(\"1 + 1 = ？\"));\n\n // 调用\n Flux<ChatResponse> flux = chatModel.stream(new Prompt(messages));\n // 打印结果\n flux.doOnNext(System.out::println).then().block();\n }\n\n}\n"}}}} |