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));
    }
}