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