aidlab/aida_back
7
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

TASK:LLM；

Browse Source
This commit is contained in:
shahaibo
2025-05-19 10:47:02 +08:00
parent a67b487dc0
commit 738144ad22
3 changed files with 20 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/main/java/com/ai/da/controller/LLMController.java
View File

@@ -57,14 +57,15 @@ public class LLMController {
public SseEmitter streamPrompt(@RequestParam String prompt, public SseEmitter streamPrompt(@RequestParam String prompt,
@RequestParam Long projectId, @RequestParam Long projectId,
@RequestParam(required = false) String fileUrl, @RequestParam(required = false) String fileUrl,
@RequestParam(required = false) List<String> imageUrlList,
@RequestParam String token) { @RequestParam String token) {
return llmService.stream(prompt, projectId, fileUrl, token); return llmService.stream(prompt, projectId, fileUrl, imageUrlList, token);
} }
@ApiOperation(value = "对话创建项目") @ApiOperation(value = "对话创建项目")
@GetMapping(value = "/chatCreateProject") @GetMapping(value = "/chatCreateProject")
public Response<Long> chatCreateProject(@RequestParam String prompt) { public Response<Long> chatCreateProject(@RequestParam String prompt, @RequestParam String process) {
return Response.success(llmService.chatCreateProject(prompt)); return Response.success(llmService.chatCreateProject(prompt, process));
} }
@ApiOperation(value = "上传文件") @ApiOperation(value = "上传文件")

4
src/main/java/com/ai/da/service/LLMService.java
View File

@@ -31,9 +31,9 @@ import java.util.Set;
*/ */
public interface LLMService { public interface LLMService {
SseEmitter stream(String prompt, Long projectId, String fileUrl, String token); SseEmitter stream(String prompt, Long projectId, String fileUrl, List<String> imageUrlList, String token);
Long chatCreateProject(String prompt); Long chatCreateProject(String prompt, String process);
List<String> uploadFile(MultipartFile file); List<String> uploadFile(MultipartFile file);

27
src/main/java/com/ai/da/service/impl/LLMServiceImpl.java
View File

@@ -67,7 +67,7 @@ public class LLMServiceImpl implements LLMService {
private final ExecutorService executor = Executors.newCachedThreadPool(); private final ExecutorService executor = Executors.newCachedThreadPool();
@Override @Override
public SseEmitter stream(String prompt, Long projectId, String fileUrl, String token) { public SseEmitter stream(String prompt, Long projectId, String fileUrl, List<String> imageUrlList, String token) {
SseEmitter emitter = new SseEmitter(0L); // 永不超时 SseEmitter emitter = new SseEmitter(0L); // 永不超时
executor.submit(() -> { executor.submit(() -> {
@@ -86,7 +86,7 @@ public class LLMServiceImpl implements LLMService {
JSONObject jsonBodyObject = new JSONObject(); JSONObject jsonBodyObject = new JSONObject();
jsonBodyObject.put("session_id", projectId.toString()); jsonBodyObject.put("session_id", projectId.toString());
jsonBodyObject.put("role", "user"); jsonBodyObject.put("role", "user");
jsonBodyObject.put("image", ""); // 可扩展 jsonBodyObject.put("image", imageUrlList); // 可扩展
jsonBodyObject.put("file", fileUrl != null ? fileUrl : ""); jsonBodyObject.put("file", fileUrl != null ? fileUrl : "");
jsonBodyObject.put("message", prompt); jsonBodyObject.put("message", prompt);
jsonBodyObject.put("enable_thinking", false); jsonBodyObject.put("enable_thinking", false);
@@ -161,7 +161,7 @@ public class LLMServiceImpl implements LLMService {
} }
@Override @Override
public Long chatCreateProject(String prompt) { public Long chatCreateProject(String prompt, String process) {
AuthPrincipalVo userHolder = UserContext.getUserHolder(); AuthPrincipalVo userHolder = UserContext.getUserHolder();
JSONObject jsonObject = pythonService.getProjectParam(prompt); JSONObject jsonObject = pythonService.getProjectParam(prompt);
JSONObject data = jsonObject.getJSONObject("data"); JSONObject data = jsonObject.getJSONObject("data");
@@ -172,16 +172,17 @@ public class LLMServiceImpl implements LLMService {
project.setAccountId(userHolder.getId()); project.setAccountId(userHolder.getId());
project.setName(data.getString("project_name")); project.setName(data.getString("project_name"));
project.setOriginal(1); project.setOriginal(1);
String process = data.getString("process"); // String process = data.getString("process");
if (StringUtils.isEmpty(process)) { // if (StringUtils.isEmpty(process)) {
project.setProcess(DesignProcess.SERIES_DESIGN.name()); // project.setProcess(DesignProcess.SERIES_DESIGN.name());
}else { // }else {
if (DesignProcess.isValidName(process)) { // if (DesignProcess.isValidName(process)) {
project.setProcess(process); // project.setProcess(process);
}else { // }else {
project.setProcess(DesignProcess.SERIES_DESIGN.name()); // project.setProcess(DesignProcess.SERIES_DESIGN.name());
} // }
} // }
project.setProcess(process);
projectMapper.insert(project); projectMapper.insert(project);
Workspace workspace = new Workspace(); Workspace workspace = new Workspace();