diff --git a/example/pom.xml b/example/pom.xml index 8e90ab2..6c237cf 100644 --- a/example/pom.xml +++ b/example/pom.xml @@ -47,7 +47,7 @@ io.github.alexcheng1982 spring-ai-dashscope-spring-boot-starter - 1.1.2 + 1.2.0 org.springdoc diff --git a/example/src/main/java/io/github/alexcheng1982/springai/dashscope/example/DemoController.java b/example/src/main/java/io/github/alexcheng1982/springai/dashscope/example/DemoController.java index 54f81e0..337e732 100644 --- a/example/src/main/java/io/github/alexcheng1982/springai/dashscope/example/DemoController.java +++ b/example/src/main/java/io/github/alexcheng1982/springai/dashscope/example/DemoController.java @@ -1,8 +1,9 @@ package io.github.alexcheng1982.springai.dashscope.example; import java.util.List; -import org.springframework.ai.chat.ChatClient; -import org.springframework.ai.embedding.EmbeddingClient; +import org.springframework.ai.chat.client.ChatClient; +import org.springframework.ai.chat.model.ChatModel; +import org.springframework.ai.embedding.EmbeddingModel; import org.springframework.web.bind.annotation.PostMapping; import org.springframework.web.bind.annotation.RequestBody; import org.springframework.web.bind.annotation.RequestMapping; @@ -29,21 +30,22 @@ public record EmbeddingResponse(List result) { } private final ChatClient chatClient; - private final EmbeddingClient embeddingClient; + private final EmbeddingModel embeddingModel; - public DemoController(ChatClient chatClient, - EmbeddingClient embeddingClient) { - this.chatClient = chatClient; - this.embeddingClient = embeddingClient; + public DemoController(ChatModel chatModel, + EmbeddingModel embeddingModel) { + this.chatClient = ChatClient.create(chatModel); + this.embeddingModel = embeddingModel; } @PostMapping("/chat") public ChatResponse chat(@RequestBody ChatRequest request) { - return new ChatResponse(chatClient.call(request.input())); + return new ChatResponse( + chatClient.prompt().user(request.input()).call().content()); } @PostMapping("/embed") public EmbeddingResponse embed(@RequestBody EmbeddingRequest request) { - return new EmbeddingResponse(embeddingClient.embed(request.input())); + return new EmbeddingResponse(embeddingModel.embed(request.input())); } } diff --git a/pom.xml b/pom.xml index 588171b..a56bdd9 100644 --- a/pom.xml +++ b/pom.xml @@ -29,6 +29,7 @@ client spring-boot-autoconfigure spring-boot-starter + example