package com.supervision.rasa.controller;

import com.supervision.rasa.pojo.dto.Text2vecDataVo;
import com.supervision.rasa.pojo.dto.Text2vecMatchesReq;
import com.supervision.rasa.pojo.dto.Text2vecMatchesRes;
import com.supervision.rasa.service.Text2vecService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.RequiredArgsConstructor;
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;

@Api(tags = "text2vec服务")
@RestController
@RequestMapping("text2vec")
@RequiredArgsConstructor
public class Text2vecController {

    private final Text2vecService text2vecService;
    @ApiOperation("更新数据库")
    @PostMapping("updateDataset")
    public boolean talkRasa(@RequestBody List<Text2vecDataVo> text2vecDataVoList){

        return text2vecService.updateDataset(text2vecDataVoList);
    }

    @ApiOperation("获取匹配项")
    @PostMapping("matches")
    public List<Text2vecMatchesRes> matches(@RequestBody Text2vecMatchesReq text2vecMatchesReq){

        return text2vecService.matches(text2vecMatchesReq);
    }
}