package tech.aiflowy.ai.controller;
|
|
import cn.hutool.core.util.IdUtil;
|
import cn.hutool.core.util.StrUtil;
|
import com.alibaba.fastjson2.JSON;
|
import com.alibaba.fastjson2.JSONArray;
|
import com.alibaba.fastjson2.JSONObject;
|
import org.springframework.web.bind.annotation.GetMapping;
|
import org.springframework.web.bind.annotation.PostMapping;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.RestController;
|
import tech.aiflowy.ai.entity.AiPluginTool;
|
import tech.aiflowy.ai.service.AiPluginToolService;
|
import tech.aiflowy.common.domain.Result;
|
import tech.aiflowy.common.web.controller.BaseCurdController;
|
import tech.aiflowy.common.web.jsonbody.JsonBody;
|
|
import javax.annotation.Resource;
|
import java.math.BigInteger;
|
|
/**
|
* 控制层。
|
*
|
* @author WangGangqiang
|
* @since 2025-04-27
|
*/
|
@RestController
|
@RequestMapping("/api/v1/aiPluginTool")
|
public class AiPluginToolController extends BaseCurdController<AiPluginToolService, AiPluginTool> {
|
public AiPluginToolController(AiPluginToolService service) {
|
super(service);
|
}
|
|
@Resource
|
private AiPluginToolService aiPluginToolService;
|
|
@PostMapping("/tool/save")
|
public Result savePlugin(@JsonBody AiPluginTool aiPluginTool){
|
|
return aiPluginToolService.savePluginTool(aiPluginTool);
|
}
|
|
// 插件工具修改页面查询
|
@PostMapping("/tool/search")
|
public Result searchPlugin(@JsonBody(value = "aiPluginToolId", required = true) BigInteger aiPluginToolId){
|
return aiPluginToolService.searchPlugin(aiPluginToolId);
|
}
|
|
@PostMapping("/toolsList")
|
public Result searchPluginToolByPluginId(@JsonBody(value = "pluginId", required = true) BigInteger pluginId,
|
@JsonBody(value = "botId", required = false) BigInteger botId){
|
return aiPluginToolService.searchPluginToolByPluginId(pluginId, botId);
|
}
|
|
@PostMapping("/tool/update")
|
public Result updatePlugin(@JsonBody AiPluginTool aiPluginTool){
|
|
return aiPluginToolService.updatePlugin(aiPluginTool);
|
}
|
|
@PostMapping("/tool/list")
|
public Result getPluginToolList(@JsonBody(value = "botId", required = true) BigInteger botId){
|
|
return aiPluginToolService.getPluginToolList(botId);
|
}
|
|
@GetMapping("/getTinyFlowData")
|
public Result getTinyFlowData(BigInteger id) {
|
JSONObject nodeData = new JSONObject();
|
AiPluginTool record = aiPluginToolService.getById(id);
|
if (record == null) {
|
return Result.success(nodeData);
|
}
|
nodeData.put("pluginId", record.getId().toString());
|
nodeData.put("pluginName", record.getName());
|
|
JSONArray parameters = new JSONArray();
|
JSONArray outputDefs = new JSONArray();
|
String inputData = record.getInputData();
|
if (StrUtil.isNotEmpty(inputData)) {
|
JSONArray array = JSON.parseArray(inputData);
|
handleArray(array);
|
parameters = array;
|
}
|
String outputData = record.getOutputData();
|
if (StrUtil.isNotEmpty(outputData)) {
|
JSONArray array = JSON.parseArray(outputData);
|
handleArray(array);
|
outputDefs = array;
|
}
|
nodeData.put("parameters", parameters);
|
nodeData.put("outputDefs", outputDefs);
|
return Result.success(nodeData);
|
}
|
|
private void handleArray(JSONArray array) {
|
for (Object o : array) {
|
JSONObject obj = (JSONObject) o;
|
obj.put("id", IdUtil.simpleUUID());
|
obj.put("nameDisabled", true);
|
obj.put("dataTypeDisabled", true);
|
obj.put("deleteDisabled", true);
|
obj.put("addChildDisabled", true);
|
JSONArray children = obj.getJSONArray("children");
|
if (children != null) {
|
handleArray(children);
|
}
|
}
|
}
|
}
|
