package com.vincent.rsf.server.ai.controller;
|
|
import com.vincent.rsf.framework.common.R;
|
import com.vincent.rsf.server.ai.dto.AiChatRequest;
|
import com.vincent.rsf.server.ai.service.AiChatService;
|
import com.vincent.rsf.server.system.controller.BaseController;
|
import lombok.RequiredArgsConstructor;
|
import org.springframework.http.MediaType;
|
import org.springframework.security.access.prepost.PreAuthorize;
|
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
|
|
@RestController
|
@RequiredArgsConstructor
|
public class AiChatController extends BaseController {
|
|
private final AiChatService aiChatService;
|
|
@PreAuthorize("isAuthenticated()")
|
@GetMapping("/ai/chat/runtime")
|
public R runtime(@RequestParam(required = false) String promptCode,
|
@RequestParam(required = false) Long sessionId) {
|
return R.ok().add(aiChatService.getRuntime(promptCode, sessionId, getLoginUserId(), getTenantId()));
|
}
|
|
@PreAuthorize("isAuthenticated()")
|
@GetMapping("/ai/chat/sessions")
|
public R sessions(@RequestParam(required = false) String promptCode) {
|
return R.ok().add(aiChatService.listSessions(promptCode, getLoginUserId(), getTenantId()));
|
}
|
|
@PreAuthorize("isAuthenticated()")
|
@PostMapping("/ai/chat/session/remove/{sessionId}")
|
public R removeSession(@PathVariable Long sessionId) {
|
aiChatService.removeSession(sessionId, getLoginUserId(), getTenantId());
|
return R.ok("Delete Success").add(sessionId);
|
}
|
|
@PreAuthorize("isAuthenticated()")
|
@PostMapping(value = "/ai/chat/stream", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
|
public SseEmitter stream(@RequestBody AiChatRequest request) {
|
return aiChatService.stream(request, getLoginUserId(), getTenantId());
|
}
|
}
|
