@Component
@RequiredArgsConstructor
public class TagFunctionCalling implements FunctionCalling<TagRequest, R> {

    @Override
    public String functionName() {
        return "tagFunctionCalling";
    }

    /**
     * 判断大模型返回的参数是否正确
     */
    @Override
    public R checkParams(TagRequest tagRequest, PigxUser userDetails, ChatMessageDTO.ExtDetails extDetails) {
        return R.ok();
    }

    /**
     * 业务处理逻辑
     */
    @Override
    public R<String> handle(TagRequest tagRequest, PigxUser userDetails, ChatMessageDTO.ExtDetails extDetails) {
        // 根据大模型返回的标签列表，执行自己的业务逻辑
        
        // 返回给大模型一段话，让大模型返回给用户
        return R.ok("获取成功");
    }

}

@Data
@EqualsAndHashCode(callSuper = true)
@JsonInclude(JsonInclude.Include.NON_NULL)
@JsonClassDescription("标签信息")
class TagRequest extends BaseAiRequest {
    @JsonProperty(required = true, value = "tagList")
    @JsonPropertyDescription("标签列表")
    private List<String> tagList;
}
 
        Copied!