38 lines
1.4 KiB
Java
38 lines
1.4 KiB
Java
package com.bruce.rag.controller;
|
|
|
|
import com.bruce.common.domain.model.RequestResult;
|
|
import com.bruce.rag.dto.request.RagDocumentQueryRequest;
|
|
import com.bruce.rag.entity.RagDocument;
|
|
import com.bruce.rag.service.IRagDocumentService;
|
|
import io.swagger.v3.oas.annotations.Operation;
|
|
import io.swagger.v3.oas.annotations.tags.Tag;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.web.bind.annotation.GetMapping;
|
|
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;
|
|
|
|
@Tag(name = "RAG知识库文档管理")
|
|
@RestController
|
|
@RequestMapping("/api/rag/documents")
|
|
public class RagDocumentController {
|
|
|
|
@Autowired
|
|
private IRagDocumentService ragDocumentService;
|
|
|
|
@Operation(summary = "查询全部知识库文档")
|
|
@GetMapping
|
|
public RequestResult<List<RagDocument>> list() {
|
|
return RequestResult.success(ragDocumentService.list());
|
|
}
|
|
|
|
@Operation(summary = "按条件查询知识库文档")
|
|
@PostMapping("/query")
|
|
public RequestResult<List<RagDocument>> query(@RequestBody RagDocumentQueryRequest request) {
|
|
return RequestResult.success(ragDocumentService.query(request));
|
|
}
|
|
}
|