feat: support deepseek and openai-compatible providers

2026-04-29 22:19:00 +08:00
parent 4ac755a7fe
commit 3555d19b26
13 changed files with 761 additions and 190 deletions
@@ -0,0 +1,150 @@
+package com.svnlog.web.service;
+
+import java.nio.file.Path;
+import java.time.Instant;
+
+import org.junit.jupiter.api.Assertions;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.io.TempDir;
+
+import com.svnlog.web.model.TaskInfo;
+import com.svnlog.web.model.TaskStatus;
+
+import okio.Buffer;
+
+class AiWorkflowServiceTest {
+
+    @TempDir
+    Path tempDir;
+
+    @Test
+    void shouldResolveDeepSeekProviderByDefault() {
+        final AiWorkflowService service = new AiWorkflowService(
+                buildOutputFileService(),
+                new SettingsService(buildOutputFileService(), new SvnPresetService()),
+                new AiInputValidator()
+        );
+
+        final AiWorkflowService.AiProviderContext context = service.resolveProviderContext(null);
+
+        Assertions.assertEquals(SettingsService.PROVIDER_DEEPSEEK, context.getProvider());
+        Assertions.assertEquals("deepseek-chat", context.getStageOneModel());
+        Assertions.assertEquals("deepseek-reasoner", context.getStageTwoModel());
+    }
+
+    @Test
+    void shouldResolveOpenAiCompatibleModelsAndUrl() {
+        final OutputFileService outputFileService = buildOutputFileService();
+        final SettingsService settingsService = new SettingsService(outputFileService, new SvnPresetService());
+        settingsService.updateSettings(
+                null,
+                SettingsService.PROVIDER_OPENAI_COMPATIBLE,
+                "http://127.0.0.1:5001/v1/",
+                "sk-openai-test",
+                "deepseek-v4-flash",
+                "deepseek-v4-pro",
+                null,
+                null
+        );
+        final AiWorkflowService service = new AiWorkflowService(outputFileService, settingsService, new AiInputValidator());
+
+        final AiWorkflowService.AiProviderContext context = service.resolveProviderContext(null);
+
+        Assertions.assertEquals(SettingsService.PROVIDER_OPENAI_COMPATIBLE, context.getProvider());
+        Assertions.assertEquals("http://127.0.0.1:5001/v1/chat/completions", context.getApiUrl());
+        Assertions.assertEquals("deepseek-v4-flash", context.getStageOneModel());
+        Assertions.assertEquals("deepseek-v4-pro", context.getStageTwoModel());
+    }
+
+    @Test
+    void shouldFailFastWhenOpenAiCompatibleBaseUrlMissing() {
+        final AiWorkflowService service = new AiWorkflowService(
+                buildOutputFileService(),
+                new StubSettingsService(buildOutputFileService(), "   ", "sk-openai-test"),
+                new AiInputValidator()
+        );
+
+        final IllegalStateException error = Assertions.assertThrows(
+                IllegalStateException.class,
+                () -> service.resolveProviderContext(null)
+        );
+
+        Assertions.assertTrue(error.getMessage().contains("OpenAI兼容 Base URL"));
+    }
+
+    @Test
+    void shouldParseCompatibleStreamWhenOnlyContentIsReturned() throws Exception {
+        final OutputFileService outputFileService = buildOutputFileService();
+        final SettingsService settingsService = new SettingsService(outputFileService, new SvnPresetService());
+        settingsService.updateSettings(
+                null,
+                SettingsService.PROVIDER_OPENAI_COMPATIBLE,
+                "http://127.0.0.1:5001/v1",
+                "sk-openai-test",
+                "deepseek-v4-flash",
+                "deepseek-v4-pro",
+                null,
+                null
+        );
+        final AiWorkflowService service = new AiWorkflowService(outputFileService, settingsService, new AiInputValidator());
+        final AiWorkflowService.AiProviderContext providerContext = service.resolveProviderContext(null);
+        final TaskContext taskContext = new TaskContext(buildTaskInfo(), null, null);
+        final Buffer buffer = new Buffer()
+                .writeUtf8("data: {\"choices\":[{\"delta\":{\"content\":\"{\\\"items\\\":[\"}}]}\n")
+                .writeUtf8("data: {\"choices\":[{\"delta\":{\"content\":\"{\\\"summary\\\":\\\"done\\\"}\"}}]}\n")
+                .writeUtf8("data: {\"choices\":[{\"delta\":{\"content\":\"]}\"},\"finish_reason\":\"stop\"}]}\n")
+                .writeUtf8("data: [DONE]\n");
+
+        final AiWorkflowService.AiStreamResult result = service.readStreamingResponse(
+                buffer,
+                taskContext,
+                providerContext,
+                "deepseek-v4-flash",
+                8000
+        );
+
+        Assertions.assertEquals("{\"items\":[{\"summary\":\"done\"}]}", result.getAnswer());
+        Assertions.assertEquals("stop", result.getFinishReason());
+    }
+
+    private OutputFileService buildOutputFileService() {
+        final OutputFileService outputFileService = new OutputFileService();
+        outputFileService.setOutputRoot(tempDir.resolve("outputs").toString());
+        return outputFileService;
+    }
+
+    private TaskInfo buildTaskInfo() {
+        final TaskInfo taskInfo = new TaskInfo();
+        taskInfo.setTaskId("task-1");
+        taskInfo.setStatus(TaskStatus.RUNNING);
+        taskInfo.setCreatedAt(Instant.now());
+        taskInfo.setUpdatedAt(Instant.now());
+        return taskInfo;
+    }
+
+    private static final class StubSettingsService extends SettingsService {
+        private final String openaiBaseUrl;
+        private final String openaiApiKey;
+
+        private StubSettingsService(OutputFileService outputFileService, String openaiBaseUrl, String openaiApiKey) {
+            super(outputFileService, new SvnPresetService());
+            this.openaiBaseUrl = openaiBaseUrl;
+            this.openaiApiKey = openaiApiKey;
+        }
+
+        @Override
+        public String getProvider() {
+            return SettingsService.PROVIDER_OPENAI_COMPATIBLE;
+        }
+
+        @Override
+        public String getOpenaiBaseUrl() {
+            return openaiBaseUrl;
+        }
+
+        @Override
+        public String getOpenaiApiKey() {
+            return openaiApiKey;
+        }
+    }
+}