package com.newyetai.ai.controller;

import com.newyetai.common.AlitaResult;
import com.newyetai.sdk.xf.model.ChatReq;
import com.newyetai.sdk.xf.model.TranslateReq;
import com.newyetai.sdk.xf.util.XfUtils;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.springframework.validation.annotation.Validated;
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;

@Api(tags = "ai")
@RestController
@RequestMapping("/ai")
public class AiController {

  @PostMapping("chat")
  @ApiOperation("ai对话")
  public AlitaResult<String> chat(@Validated @RequestBody ChatReq req) {
    return AlitaResult.success(XfUtils.chat(req));
  }

  @PostMapping("translate")
  @ApiOperation("ai翻译")
  public AlitaResult<String> chat(@Validated @RequestBody TranslateReq req) {
    return AlitaResult.success(XfUtils.translate(req));
  }


}