79 lines
3.2 KiB
Java
79 lines
3.2 KiB
Java
package com.cloudhandson.vpdbackoffice.web;
|
|
|
|
import com.cloudhandson.vpdbackoffice.domain.vector.VectorIngestCommand;
|
|
import com.cloudhandson.vpdbackoffice.domain.vector.VectorIngestResult;
|
|
import com.cloudhandson.vpdbackoffice.service.VectorKnowledgeService;
|
|
import com.cloudhandson.vpdbackoffice.mapper.UserMapper;
|
|
import org.springframework.dao.DataAccessException;
|
|
import org.springframework.stereotype.Controller;
|
|
import org.springframework.ui.Model;
|
|
import org.springframework.web.bind.annotation.GetMapping;
|
|
import org.springframework.web.bind.annotation.PostMapping;
|
|
import org.springframework.web.bind.annotation.RequestParam;
|
|
import org.springframework.web.servlet.mvc.support.RedirectAttributes;
|
|
|
|
@Controller
|
|
public class VectorKnowledgeController {
|
|
|
|
private final VectorKnowledgeService vectorKnowledgeService;
|
|
private final UserMapper userMapper;
|
|
|
|
public VectorKnowledgeController(VectorKnowledgeService vectorKnowledgeService, UserMapper userMapper) {
|
|
this.vectorKnowledgeService = vectorKnowledgeService;
|
|
this.userMapper = userMapper;
|
|
}
|
|
|
|
@GetMapping("/vector-knowledge")
|
|
public String page(Model model) {
|
|
model.addAttribute("users", userMapper.findAll());
|
|
model.addAttribute("aiEmbeddingConfigured", vectorKnowledgeService.aiEmbeddingConfigured());
|
|
try {
|
|
model.addAttribute("summary", vectorKnowledgeService.summary());
|
|
} catch (DataAccessException exception) {
|
|
RuntimeErrorMessage error = RuntimeErrorMessages.dataAccess(exception);
|
|
model.addAttribute("summary", null);
|
|
model.addAttribute("runtimeError", error);
|
|
}
|
|
return "vector-knowledge";
|
|
}
|
|
|
|
@PostMapping("/vector-knowledge/ingest")
|
|
public String ingest(
|
|
@RequestParam String documentId,
|
|
@RequestParam String title,
|
|
@RequestParam(required = false, defaultValue = "") String sourceUri,
|
|
@RequestParam String content,
|
|
@RequestParam String techTags,
|
|
@RequestParam(defaultValue = "600") int chunkSize,
|
|
@RequestParam(defaultValue = "DEMO") String embeddingMode,
|
|
RedirectAttributes redirectAttributes
|
|
) {
|
|
try {
|
|
VectorIngestResult result = vectorKnowledgeService.ingest(new VectorIngestCommand(
|
|
documentId, title, sourceUri, content, techTags, chunkSize, embeddingMode));
|
|
redirectAttributes.addFlashAttribute("ingestResult", result);
|
|
redirectAttributes.addFlashAttribute("message",
|
|
result.chunkCount() + "개 청크를 저장했습니다. 태그 " + result.tagCount() + "개가 각 청크에 연결되었습니다.");
|
|
} catch (Exception exception) {
|
|
redirectAttributes.addFlashAttribute("errorMessage", exception.getMessage());
|
|
}
|
|
return "redirect:/vector-knowledge";
|
|
}
|
|
|
|
@PostMapping("/vector-knowledge/search")
|
|
public String search(
|
|
@RequestParam long userId,
|
|
@RequestParam String query,
|
|
@RequestParam(defaultValue = "10") int limit,
|
|
@RequestParam(defaultValue = "DEMO") String embeddingMode,
|
|
Model model
|
|
) {
|
|
try {
|
|
model.addAttribute("searchResult", vectorKnowledgeService.search(userId, query, limit, embeddingMode));
|
|
} catch (Exception exception) {
|
|
model.addAttribute("errorMessage", exception.getMessage());
|
|
}
|
|
return "fragments/vector-search-result :: result";
|
|
}
|
|
}
|