com.arextest.web.api.service.controller.AIController Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of arex-web-api Show documentation

There is a newer version: 0.7.0.1

package com.arextest.web.api.service.controller;

import java.util.List;
import java.util.Optional;
import lombok.Getter;
import lombok.RequiredArgsConstructor;
import org.springframework.boot.autoconfigure.condition.ConditionalOnBean;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.CrossOrigin;
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.ResponseBody;

import com.arextest.common.model.response.Response;
import com.arextest.common.utils.ResponseUtils;
import com.arextest.web.core.business.ai.AIProvider;
import com.arextest.web.model.contract.contracts.ai.GenReq;

import lombok.extern.slf4j.Slf4j;

/**
 * @author: QizhengMo
 * @date: 2024/3/25 15:15
 */
@Slf4j
@RequestMapping("/api/ai/")
@Controller
@CrossOrigin(origins = "*", maxAge = 3600)
@ConditionalOnBean(value = AIProvider.class)
@RequiredArgsConstructor
public class AIController {
  @Getter
  private final List providers;

  @PostMapping("/generateTestScript")
  @ResponseBody
  public Response generateTestScript(@RequestBody GenReq req) {
    return ResponseUtils.successResponse(getProvider(req.getModelName()).generateScripts(req));
  }

  private AIProvider getProvider(String modelName) {
    Optional provider = providers
        .stream()
        .filter(p -> p.getModelInfo().getModelName().equals(modelName))
        .findFirst();
    return provider.orElse(providers.get(0));
  }
}