67 lines
2.1 KiB
Java
67 lines
2.1 KiB
Java
package com.huangge1199.aiagent.controller;
|
|
|
|
import com.huangge1199.aiagent.common.R;
|
|
import io.swagger.v3.oas.annotations.Operation;
|
|
import io.swagger.v3.oas.annotations.tags.Tag;
|
|
import org.springframework.ai.chat.client.ChatClient;
|
|
import org.springframework.ai.chat.model.ChatResponse;
|
|
import org.springframework.core.ParameterizedTypeReference;
|
|
import org.springframework.web.bind.annotation.PostMapping;
|
|
import org.springframework.web.bind.annotation.RequestBody;
|
|
import org.springframework.web.bind.annotation.RequestMapping;
|
|
import org.springframework.web.bind.annotation.RestController;
|
|
|
|
import java.util.List;
|
|
|
|
/**
|
|
* ResControlle
|
|
*
|
|
* @author huangge1199
|
|
* @since 2025/5/17 12:23:36
|
|
*/
|
|
@RestController
|
|
@RequestMapping("/res")
|
|
@Tag(name = "ChatClient返回响应")
|
|
public class ResController {
|
|
|
|
private final ChatClient chatClient;
|
|
|
|
public ResController(ChatClient.Builder builder) {
|
|
this.chatClient = builder.build();
|
|
}
|
|
|
|
record ActorFilms(String actor, List<String> movies) {
|
|
}
|
|
|
|
@PostMapping("/chatRes")
|
|
@Operation(summary = "返回 ChatResponse")
|
|
public R<ChatResponse> chatResRes(@RequestBody String question) {
|
|
ChatResponse chatResponse = chatClient.prompt()
|
|
.user(question)
|
|
.call()
|
|
.chatResponse();
|
|
return R.ok(chatResponse);
|
|
}
|
|
|
|
@PostMapping("/entity")
|
|
@Operation(summary = "返回实体-单个实体")
|
|
public R<ActorFilms> entityRes(@RequestBody String question) {
|
|
ActorFilms actorFilms = chatClient.prompt()
|
|
.user(question)
|
|
.call()
|
|
.entity(ActorFilms.class);
|
|
return R.ok(actorFilms);
|
|
}
|
|
|
|
@PostMapping("/actorFilmsRes")
|
|
@Operation(summary = "返回实体-泛型")
|
|
public R<List<ActorFilms>> actorFilmsRes(@RequestBody String question) {
|
|
List<ActorFilms> actorFilms = chatClient.prompt()
|
|
.user(question)
|
|
.call()
|
|
.entity(new ParameterizedTypeReference<List<ActorFilms>>() {
|
|
});
|
|
return R.ok(actorFilms);
|
|
}
|
|
}
|