package tech.aiflowy.ai.controller;

import com.mybatisflex.core.query.QueryWrapper;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import tech.aiflowy.ai.entity.AiFirstMenu;
import tech.aiflowy.ai.entity.AiKnowledge;
import tech.aiflowy.ai.service.AiDocumentChunkService;
import tech.aiflowy.ai.service.AiFirstMenuService;
import tech.aiflowy.ai.service.AiKnowledgeService;
import tech.aiflowy.ai.service.AiLlmService;
import tech.aiflowy.common.domain.Result;
import tech.aiflowy.common.tree.Tree;
import tech.aiflowy.common.web.controller.BaseCurdController;

import java.util.List;

@RestController
@RequestMapping("/api/v1/aiMenu/FirstMenu")
public class AiFirstMenuController extends BaseCurdController<AiFirstMenuService, AiFirstMenu> {
    public AiFirstMenuController(AiFirstMenuService service) {
        super(service);
    }

    @Autowired
    AiFirstMenuService aiFirstMenuService;


    public Result list(AiFirstMenu entity, Boolean asTree, String sortKey, String sortType) {
        QueryWrapper queryWrapper = QueryWrapper.create(entity, buildOperators(entity));
        queryWrapper.orderBy(AiFirstMenu::getId);
        List<AiFirstMenu> list = Tree.tryToTree(super.service.list(queryWrapper), asTree);
        return Result.success(list);
    }


}