diff --git a/pom.xml b/pom.xml index 163598e..8ee3128 100644 --- a/pom.xml +++ b/pom.xml @@ -71,7 +71,7 @@ com.alibaba.cloud.ai spring-ai-alibaba-bom - 1.0.0.2 + 1.0.0.4 pom import diff --git a/src/main/java/com/hanserwei/snailsai/controller/DashScopeController.java b/src/main/java/com/hanserwei/snailsai/controller/DashScopeController.java index 16ca13c..21bb4b5 100644 --- a/src/main/java/com/hanserwei/snailsai/controller/DashScopeController.java +++ b/src/main/java/com/hanserwei/snailsai/controller/DashScopeController.java @@ -2,10 +2,12 @@ package com.hanserwei.snailsai.controller; import jakarta.annotation.Resource; import org.springframework.ai.chat.client.ChatClient; +import org.springframework.http.MediaType; import org.springframework.web.bind.annotation.GetMapping; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RequestParam; import org.springframework.web.bind.annotation.RestController; +import reactor.core.publisher.Flux; @RequestMapping("/dashscope") @RestController @@ -14,11 +16,11 @@ public class DashScopeController { @Resource private ChatClient dashScopeChatClient; - @GetMapping("/chat") - public String chat(@RequestParam("userPrompt") String userPrompt) { + @GetMapping(value = "/chat",produces = MediaType.TEXT_EVENT_STREAM_VALUE) + public Flux chat(@RequestParam("userPrompt") String userPrompt) { return dashScopeChatClient.prompt() .user(userPrompt) - .call() + .stream() .content(); } }