1 line
11 KiB
JSON
1 line
11 KiB
JSON
{"repo": "aibol51/medical-java", "files": {"yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/chat/DeepSeekChatModelTests.java": {"aabedaccec6c52df7a240203ea60f789ef7a11d9": {"t": 1785747155.057404, "n": 2202, "h": "681db718dc5bd132", "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 org.springframework.ai.openai.OpenAiChatModel;\nimport org.springframework.ai.openai.OpenAiChatOptions;\nimport org.springframework.ai.openai.api.OpenAiApi;\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 OpenAiChatModel openAiChatModel = OpenAiChatModel.builder()\n .openAiApi(OpenAiApi.builder()\n .baseUrl(DeepSeekChatModel.BASE_URL)\n .apiKey(\"sk-e52047409b144d97b791a6a46a2d\") // apiKey\n .build())\n .defaultOptions(OpenAiChatOptions.builder()\n .model(\"deepseek-chat\") // 模型\n .temperature(0.7)\n .build())\n .build();\n\n private final DeepSeekChatModel chatModel = new DeepSeekChatModel(openAiChatModel);\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"}, "857b280e186ce33e77010f645b62e162ce9d4cba": {"t": 1785770750.5278237, "n": 2202, "h": "681db718dc5bd132", "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 org.springframework.ai.openai.OpenAiChatModel;\nimport org.springframework.ai.openai.OpenAiChatOptions;\nimport org.springframework.ai.openai.api.OpenAiApi;\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 OpenAiChatModel openAiChatModel = OpenAiChatModel.builder()\n .openAiApi(OpenAiApi.builder()\n .baseUrl(DeepSeekChatModel.BASE_URL)\n .apiKey(\"sk-e52047409b144d97b791a6a46a2d\") // apiKey\n .build())\n .defaultOptions(OpenAiChatOptions.builder()\n .model(\"deepseek-chat\") // 模型\n .temperature(0.7)\n .build())\n .build();\n\n private final DeepSeekChatModel chatModel = new DeepSeekChatModel(openAiChatModel);\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"}}, "yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/chat/BaiChuanChatModelTests.java": {"aabedaccec6c52df7a240203ea60f789ef7a11d9": {"t": 1785747155.1648657, "n": 2327, "h": "de78850acb48e791", "c": "package cn.iocoder.yudao.framework.ai.chat;\n\nimport cn.iocoder.yudao.framework.ai.core.model.baichuan.BaiChuanChatModel;\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 org.springframework.ai.openai.OpenAiChatModel;\nimport org.springframework.ai.openai.OpenAiChatOptions;\nimport org.springframework.ai.openai.api.OpenAiApi;\nimport reactor.core.publisher.Flux;\n\nimport java.util.ArrayList;\nimport java.util.List;\n\n/**\n * {@link BaiChuanChatModel} 集成测试\n *\n * @author 芋道源码\n */\npublic class BaiChuanChatModelTests {\n\n private final OpenAiChatModel openAiChatModel = OpenAiChatModel.builder()\n .openAiApi(OpenAiApi.builder()\n .baseUrl(BaiChuanChatModel.BASE_URL)\n .apiKey(\"sk-61b6766a94c70786ed02673f5e16af3c\") // apiKey\n .build())\n .defaultOptions(OpenAiChatOptions.builder()\n .model(\"Baichuan4-Turbo\") // 模型(https://platform.baichuan-ai.com/docs/api)\n .temperature(0.7)\n .build())\n .build();\n\n private final DeepSeekChatModel chatModel = new DeepSeekChatModel(openAiChatModel);\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"}, "857b280e186ce33e77010f645b62e162ce9d4cba": {"t": 1785770750.7616904, "n": 2327, "h": "de78850acb48e791", "c": "package cn.iocoder.yudao.framework.ai.chat;\n\nimport cn.iocoder.yudao.framework.ai.core.model.baichuan.BaiChuanChatModel;\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 org.springframework.ai.openai.OpenAiChatModel;\nimport org.springframework.ai.openai.OpenAiChatOptions;\nimport org.springframework.ai.openai.api.OpenAiApi;\nimport reactor.core.publisher.Flux;\n\nimport java.util.ArrayList;\nimport java.util.List;\n\n/**\n * {@link BaiChuanChatModel} 集成测试\n *\n * @author 芋道源码\n */\npublic class BaiChuanChatModelTests {\n\n private final OpenAiChatModel openAiChatModel = OpenAiChatModel.builder()\n .openAiApi(OpenAiApi.builder()\n .baseUrl(BaiChuanChatModel.BASE_URL)\n .apiKey(\"sk-61b6766a94c70786ed02673f5e16af3c\") // apiKey\n .build())\n .defaultOptions(OpenAiChatOptions.builder()\n .model(\"Baichuan4-Turbo\") // 模型(https://platform.baichuan-ai.com/docs/api)\n .temperature(0.7)\n .build())\n .build();\n\n private final DeepSeekChatModel chatModel = new DeepSeekChatModel(openAiChatModel);\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"}}}} |