!1293 【功能新增】AI:讯飞、文多多 PPT API 对接

Merge pull request !1293 from 小新/feature/ai
This commit is contained in:
芋道源码 2025-03-17 13:52:02 +00:00 committed by Gitee
commit 1db3b867aa
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
4 changed files with 1761 additions and 0 deletions

View File

@ -0,0 +1,389 @@
package cn.iocoder.yudao.framework.ai.core.model.wenduoduo.api;
import cn.iocoder.yudao.framework.common.util.json.JsonUtils;
import com.fasterxml.jackson.annotation.JsonFormat;
import com.fasterxml.jackson.annotation.JsonInclude;
import lombok.extern.slf4j.Slf4j;
import org.springframework.core.ParameterizedTypeReference;
import org.springframework.http.HttpRequest;
import org.springframework.http.HttpStatusCode;
import org.springframework.http.MediaType;
import org.springframework.util.LinkedMultiValueMap;
import org.springframework.util.MultiValueMap;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.reactive.function.BodyInserters;
import org.springframework.web.reactive.function.client.ClientResponse;
import org.springframework.web.reactive.function.client.WebClient;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
import java.time.LocalDateTime;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.function.Function;
import java.util.function.Predicate;
/**
* 文多多 API
* <p>
* <p>
* * 对接文多多<a href="https://docmee.cn/open-platform/api">PPT 生成 API</a>
*
* @author xiaoxin
*/
@Slf4j
public class WddPptApi {
public static final String BASE_URL = "https://docmee.cn";
private final WebClient webClient;
private final Predicate<HttpStatusCode> STATUS_PREDICATE = status -> !status.is2xxSuccessful();
private final Function<Object, Function<ClientResponse, Mono<? extends Throwable>>> EXCEPTION_FUNCTION =
reqParam -> response -> response.bodyToMono(String.class).handle((responseBody, sink) -> {
HttpRequest request = response.request();
log.error("[wdd-api] 调用失败!请求方式:[{}],请求地址:[{}],请求参数:[{}],响应数据: [{}]",
request.getMethod(), request.getURI(), reqParam, responseBody);
sink.error(new IllegalStateException("[wdd-api] 调用失败!"));
});
public WddPptApi(String baseUrl) {
this.webClient = WebClient.builder()
.baseUrl(baseUrl)
.defaultHeaders((headers) -> headers.setContentType(MediaType.APPLICATION_JSON))
.build();
}
/**
* 创建 token
*
* @param request 请求信息
* @return token
*/
public String createApiToken(CreateTokenRequest request) {
return this.webClient.post()
.uri("/api/user/createApiToken")
.header("Api-Key", request.apiKey)
.body(Mono.just(request), CreateTokenRequest.class)
.retrieve()
.onStatus(STATUS_PREDICATE, EXCEPTION_FUNCTION.apply(request))
.bodyToMono(ApiResponse.class)
.<String>handle((response, sink) -> {
if (response.code != 0) {
sink.error(new IllegalStateException("创建 token 异常," + response.message));
return;
}
sink.next(response.data.get("token").toString());
})
.block();
}
/**
* 创建任务
*
* @param type 类型
* 1.智能生成主题要求
* 2.上传文件生成
* 3.上传思维导图生成
* 4.通过word精准转ppt
* 5.通过网页链接生成
* 6.粘贴文本内容生成
* 7.Markdown大纲生成
* @param content 内容
* type=1 用户输入主题或要求不超过1000字符
* type=24 不传
* type=3 幕布等分享链接
* type=5 网页链接地址http/https
* type=6 粘贴文本内容不超过20000字符
* type=7 大纲内容markdown
* @param files 文件列表
* 文件列表文件数不超过5个总大小不超过50M
* type=1 上传参考文件非必传支持多个
* type=2 上传文件支持多个
* type=3 上传思维导图xmind/mm/md仅支持一个
* type=4 上传word文件仅支持一个
* type=567 不传
* <p>
* 支持格式doc/docx/pdf/ppt/pptx/txt/md/xls/xlsx/csv/html/epub/mobi/xmind/mm
* @return 任务ID
*/
public ApiResponse createTask(String token, Integer type, String content, List<MultipartFile> files) {
MultiValueMap<String, Object> formData = new LinkedMultiValueMap<>();
formData.add("type", type);
if (content != null) {
formData.add("content", content);
}
if (files != null) {
for (MultipartFile file : files) {
formData.add("file", file.getResource());
}
}
return this.webClient.post()
.uri("/api/ppt/v2/createTask")
.header("token", token)
.contentType(MediaType.MULTIPART_FORM_DATA)
.body(BodyInserters.fromMultipartData(formData))
.retrieve()
.onStatus(STATUS_PREDICATE, EXCEPTION_FUNCTION.apply(formData))
.bodyToMono(ApiResponse.class)
.block();
}
/**
* 获取生成选项
*
* @param lang 语种
* @return 生成选项
*/
public Map<String, Object> getOptions(String lang) {
String uri = "/api/ppt/v2/options";
if (lang != null) {
uri += "?lang=" + lang;
}
return this.webClient.get()
.uri(uri)
.retrieve()
.onStatus(STATUS_PREDICATE, EXCEPTION_FUNCTION.apply(lang))
.bodyToMono(new ParameterizedTypeReference<ApiResponse>() {
})
.<Map<String, Object>>handle((response, sink) -> {
if (response.code != 0) {
sink.error(new IllegalStateException("获取生成选项异常," + response.message));
return;
}
sink.next(response.data);
})
.block();
}
/**
* 分页查询PPT模板
*
* @param token 令牌
* @param request 请求体
* @return 模板列表
*/
public PagePptTemplateInfo getTemplatePage(String token, TemplateQueryRequest request) {
return this.webClient.post()
.uri("/api/ppt/templates")
.header("token", token)
.bodyValue(request)
.retrieve()
.onStatus(STATUS_PREDICATE, EXCEPTION_FUNCTION.apply(request))
.bodyToMono(new ParameterizedTypeReference<PagePptTemplateInfo>() {
})
.block();
}
/**
* 生成大纲内容
*
* @return 大纲内容流
*/
public Flux<Map<String, Object>> createOutline(String token, CreateOutlineRequest request) {
return this.webClient.post()
.uri("/api/ppt/v2/generateContent")
.header("token", token)
.body(Mono.just(request), CreateOutlineRequest.class)
.retrieve()
.onStatus(STATUS_PREDICATE, EXCEPTION_FUNCTION.apply(request))
.bodyToFlux(new ParameterizedTypeReference<>() {
});
}
/**
* 修改大纲内容
*
* @param request 请求体
* @return 大纲内容流
*/
public Flux<Map<String, Object>> updateOutline(String token, UpdateOutlineRequest request) {
return this.webClient.post()
.uri("/api/ppt/v2/updateContent")
.header("token", token)
.body(Mono.just(request), UpdateOutlineRequest.class)
.retrieve()
.onStatus(STATUS_PREDICATE, EXCEPTION_FUNCTION.apply(request))
.bodyToFlux(new ParameterizedTypeReference<>() {
});
}
/**
* 生成PPT
*
* @return PPT信息
*/
public PptInfo create(String token, CreatePptRequest request) {
return this.webClient.post()
.uri("/api/ppt/v2/generatePptx")
.header("token", token)
.body(Mono.just(request), CreatePptRequest.class)
.retrieve()
.onStatus(STATUS_PREDICATE, EXCEPTION_FUNCTION.apply(request))
.bodyToMono(ApiResponse.class)
.<PptInfo>handle((response, sink) -> {
if (response.code != 0) {
sink.error(new IllegalStateException("生成 PPT 异常," + response.message));
return;
}
sink.next(Objects.requireNonNull(JsonUtils.parseObject(JsonUtils.toJsonString(response.data.get("pptInfo")), PptInfo.class)));
})
.block();
}
@JsonInclude(value = JsonInclude.Include.NON_NULL)
public record CreateTokenRequest(
String apiKey,
String uid,
Integer limit
) {
public CreateTokenRequest(String apiKey) {
this(apiKey, null, null);
}
}
/**
* API 通用响应
*/
@JsonInclude(value = JsonInclude.Include.NON_NULL)
public record ApiResponse(
Integer code,
String message,
Map<String, Object> data
) {
}
/**
* 创建任务
*/
@JsonInclude(value = JsonInclude.Include.NON_NULL)
public record CreateTaskRequest(
Integer type,
String content,
List<MultipartFile> files
) {
}
/**
* 生成大纲内容请求
*/
@JsonInclude(value = JsonInclude.Include.NON_NULL)
public record CreateOutlineRequest(
String id,
String length,
String scene,
String audience,
String lang,
String prompt
) {
}
/**
* 修改大纲内容请求
*/
@JsonInclude(value = JsonInclude.Include.NON_NULL)
public record UpdateOutlineRequest(
String id,
String markdown,
String question
) {
}
/**
* 生成PPT请求
*/
@JsonInclude(value = JsonInclude.Include.NON_NULL)
public record CreatePptRequest(
String id,
String templateId,
String markdown
) {
}
@JsonInclude(value = JsonInclude.Include.NON_NULL)
public record PptInfo(
String id,
String name,
String subject,
String coverUrl,
String fileUrl,
String templateId,
String pptxProperty,
String userId,
String userName,
int companyId,
@JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss")
LocalDateTime updateTime,
@JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss")
LocalDateTime createTime,
String createUser,
String updateUser
) {
}
@JsonInclude(value = JsonInclude.Include.NON_NULL)
public record TemplateQueryRequest(
int page,
int size,
Filter filters
) {
@JsonInclude(value = JsonInclude.Include.NON_NULL)
public record Filter(
int type,
String category,
String style,
String themeColor
) {
}
}
@JsonInclude(value = JsonInclude.Include.NON_NULL)
public record PagePptTemplateInfo(
List<PptTemplateInfo> data,
String total
) {
}
@JsonInclude(value = JsonInclude.Include.NON_NULL)
public record PptTemplateInfo(
String id,
int type,
Integer subType,
String layout,
String category,
String style,
String themeColor,
String lang,
boolean animation,
String subject,
String coverUrl,
String fileUrl,
List<String> pageCoverUrls,
String pptxProperty,
int sort,
int num,
Integer imgNum,
int isDeleted,
String userId,
int companyId,
@JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss")
LocalDateTime updateTime,
@JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss")
LocalDateTime createTime,
String createUser,
String updateUser
) {
}
}

View File

@ -0,0 +1,766 @@
package cn.iocoder.yudao.framework.ai.core.model.xunfei.api;
import cn.iocoder.yudao.framework.common.util.json.JsonUtils;
import com.fasterxml.jackson.annotation.JsonInclude;
import lombok.extern.slf4j.Slf4j;
import org.springframework.core.io.ByteArrayResource;
import org.springframework.http.HttpStatusCode;
import org.springframework.http.MediaType;
import org.springframework.util.LinkedMultiValueMap;
import org.springframework.util.MultiValueMap;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.reactive.function.BodyInserters;
import org.springframework.web.reactive.function.client.ClientResponse;
import org.springframework.web.reactive.function.client.WebClient;
import reactor.core.publisher.Mono;
import javax.crypto.Mac;
import javax.crypto.spec.SecretKeySpec;
import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.security.InvalidKeyException;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.Base64;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.function.Function;
import java.util.function.Predicate;
/**
* 讯飞智能PPT生成 API
* <p>
* 对接讯飞<a href="https://www.xfyun.cn/doc/spark/PPTv2.html">智能 PPT 生成 API</a>
*
* @author xiaoxin
*/
@Slf4j
public class XunfeiPptApi {
public static final String BASE_URL = "https://zwapi.xfyun.cn/api/ppt/v2";
private final WebClient webClient;
private final String appId;
private final String apiSecret;
private final Predicate<HttpStatusCode> STATUS_PREDICATE = status -> !status.is2xxSuccessful();
private final Function<Object, Function<ClientResponse, Mono<? extends Throwable>>> EXCEPTION_FUNCTION =
reqParam -> response -> response.bodyToMono(String.class).handle((responseBody, sink) -> {
log.error("[xunfei-ppt-api] 调用失败!请求参数:[{}],响应数据: [{}]", reqParam, responseBody);
sink.error(new IllegalStateException("[xunfei-ppt-api] 调用失败!"));
});
public XunfeiPptApi(String baseUrl, String appId, String apiSecret) {
this.webClient = WebClient.builder()
.baseUrl(baseUrl)
.build();
this.appId = appId;
this.apiSecret = apiSecret;
}
/**
* 获取签名
*
* @return 签名信息
*/
private SignatureInfo getSignature() {
long timestamp = System.currentTimeMillis() / 1000;
String ts = String.valueOf(timestamp);
String signature = generateSignature(appId, apiSecret, timestamp);
return new SignatureInfo(appId, ts, signature);
}
/**
* 生成签名
*
* @param appId 应用ID
* @param apiSecret 应用密钥
* @param timestamp 时间戳
* @return 签名
*/
private String generateSignature(String appId, String apiSecret, long timestamp) {
try {
String auth = md5(appId + timestamp);
return hmacSHA1Encrypt(auth, apiSecret);
} catch (NoSuchAlgorithmException | InvalidKeyException e) {
log.error("[xunfei-ppt-api] 生成签名失败", e);
throw new IllegalStateException("[xunfei-ppt-api] 生成签名失败");
}
}
/**
* HMAC SHA1 加密
*/
private String hmacSHA1Encrypt(String encryptText, String encryptKey)
throws NoSuchAlgorithmException, InvalidKeyException {
SecretKeySpec keySpec = new SecretKeySpec(
encryptKey.getBytes(StandardCharsets.UTF_8), "HmacSHA1");
Mac mac = Mac.getInstance("HmacSHA1");
mac.init(keySpec);
byte[] result = mac.doFinal(encryptText.getBytes(StandardCharsets.UTF_8));
return Base64.getEncoder().encodeToString(result);
}
/**
* MD5 哈希
*/
private String md5(String text) throws NoSuchAlgorithmException {
MessageDigest md = MessageDigest.getInstance("MD5");
byte[] digest = md.digest(text.getBytes(StandardCharsets.UTF_8));
StringBuilder sb = new StringBuilder();
for (byte b : digest) {
sb.append(String.format("%02x", b));
}
return sb.toString();
}
/**
* 获取PPT模板列表
*
* @param style 风格"商务"
* @param pageSize 每页数量
* @return 模板列表
*/
public TemplatePageResponse getTemplatePage(String style, Integer pageSize) {
SignatureInfo signInfo = getSignature();
Map<String, Object> requestBody = new HashMap<>();
requestBody.put("style", style);
requestBody.put("pageSize", pageSize != null ? pageSize.toString() : "10");
return this.webClient.post()
.uri("/template/list")
.header("appId", signInfo.appId)
.header("timestamp", signInfo.timestamp)
.header("signature", signInfo.signature)
.contentType(MediaType.APPLICATION_JSON)
.bodyValue(requestBody)
.retrieve()
.onStatus(STATUS_PREDICATE, EXCEPTION_FUNCTION.apply(requestBody))
.bodyToMono(TemplatePageResponse.class)
.block();
}
/**
* 创建大纲通过文本
*
* @param query 查询文本
* @return 大纲创建响应
*/
public CreateResponse createOutline(String query) {
SignatureInfo signInfo = getSignature();
MultiValueMap<String, Object> formData = new LinkedMultiValueMap<>();
formData.add("query", query);
return this.webClient.post()
.uri("/createOutline")
.header("appId", signInfo.appId)
.header("timestamp", signInfo.timestamp)
.header("signature", signInfo.signature)
.contentType(MediaType.MULTIPART_FORM_DATA)
.body(BodyInserters.fromMultipartData(formData))
.retrieve()
.onStatus(STATUS_PREDICATE, EXCEPTION_FUNCTION.apply(formData))
.bodyToMono(CreateResponse.class)
.block();
}
/**
* 直接创建PPT简化版 - 通过文本
*
* @param query 查询文本
* @return 创建响应
*/
public CreateResponse create(String query) {
CreatePptRequest request = CreatePptRequest.builder()
.query(query)
.build();
return create(request);
}
/**
* 直接创建PPT简化版 - 通过文件
*
* @param file 文件
* @param fileName 文件名
* @return 创建响应
*/
public CreateResponse create(MultipartFile file, String fileName) {
CreatePptRequest request = CreatePptRequest.builder()
.file(file)
.fileName(fileName)
.build();
return create(request);
}
/**
* 直接创建PPT完整版
*
* @param request 请求参数
* @return 创建响应
*/
public CreateResponse create(CreatePptRequest request) {
SignatureInfo signInfo = getSignature();
MultiValueMap<String, Object> formData = buildCreateFormData(request);
return this.webClient.post()
.uri("/create")
.header("appId", signInfo.appId)
.header("timestamp", signInfo.timestamp)
.header("signature", signInfo.signature)
.contentType(MediaType.MULTIPART_FORM_DATA)
.body(BodyInserters.fromMultipartData(formData))
.retrieve()
.onStatus(STATUS_PREDICATE, EXCEPTION_FUNCTION.apply(formData))
.bodyToMono(CreateResponse.class)
.block();
}
/**
* 通过大纲创建PPT简化版
*
* @param outline 大纲内容
* @param query 查询文本
* @return 创建响应
*/
public CreateResponse createPptByOutline(OutlineData outline, String query) {
CreatePptByOutlineRequest request = CreatePptByOutlineRequest.builder()
.outline(outline)
.query(query)
.build();
return createPptByOutline(request);
}
/**
* 通过大纲创建PPT完整版
*
* @param request 请求参数
* @return 创建响应
*/
public CreateResponse createPptByOutline(CreatePptByOutlineRequest request) {
SignatureInfo signInfo = getSignature();
return this.webClient.post()
.uri("/createPptByOutline")
.header("appId", signInfo.appId)
.header("timestamp", signInfo.timestamp)
.header("signature", signInfo.signature)
.contentType(MediaType.APPLICATION_JSON)
.bodyValue(request)
.retrieve()
.onStatus(STATUS_PREDICATE, EXCEPTION_FUNCTION.apply(request))
.bodyToMono(CreateResponse.class)
.block();
}
/**
* 检查PPT生成进度
*
* @param sid 任务ID
* @return 进度响应
*/
public ProgressResponse checkProgress(String sid) {
SignatureInfo signInfo = getSignature();
return this.webClient.get()
.uri(uriBuilder -> uriBuilder
.path("/progress")
.queryParam("sid", sid)
.build())
.header("appId", signInfo.appId)
.header("timestamp", signInfo.timestamp)
.header("signature", signInfo.signature)
.retrieve()
.onStatus(STATUS_PREDICATE, EXCEPTION_FUNCTION.apply(sid))
.bodyToMono(ProgressResponse.class)
.block();
}
/**
* 签名信息
*/
@JsonInclude(value = JsonInclude.Include.NON_NULL)
private record SignatureInfo(
String appId,
String timestamp,
String signature
) {
}
/**
* 模板列表响应
*/
@JsonInclude(value = JsonInclude.Include.NON_NULL)
public record TemplatePageResponse(
boolean flag,
int code,
String desc,
Integer count,
TemplatePageData data
) {
}
/**
* 模板列表数据
*/
@JsonInclude(value = JsonInclude.Include.NON_NULL)
public record TemplatePageData(
String total,
List<TemplateInfo> records,
Integer pageNum
) {
}
/**
* 模板信息
*/
@JsonInclude(value = JsonInclude.Include.NON_NULL)
public record TemplateInfo(
String templateIndexId,
Integer pageCount,
String type,
String color,
String industry,
String style,
String detailImage
) {
}
/**
* 创建响应
*/
@JsonInclude(value = JsonInclude.Include.NON_NULL)
public record CreateResponse(
boolean flag,
int code,
String desc,
Integer count,
CreateResponseData data
) {
}
/**
* 创建响应数据
*/
@JsonInclude(value = JsonInclude.Include.NON_NULL)
public record CreateResponseData(
String sid,
String coverImgSrc,
String title,
String subTitle,
OutlineData outline
) {
}
/**
* 大纲数据结构
*/
@JsonInclude(value = JsonInclude.Include.NON_NULL)
public record OutlineData(
String title,
String subTitle,
List<Chapter> chapters
) {
/**
* 章节结构
*/
@JsonInclude(value = JsonInclude.Include.NON_NULL)
public record Chapter(
String chapterTitle,
List<ChapterContent> chapterContents
) {
/**
* 章节内容
*/
@JsonInclude(value = JsonInclude.Include.NON_NULL)
public record ChapterContent(
String chapterTitle
) {
}
}
/**
* 将大纲对象转换为JSON字符串
*
* @return 大纲JSON字符串
*/
public String toJsonString() {
return JsonUtils.toJsonString(this);
}
}
/**
* 进度响应
*/
@JsonInclude(value = JsonInclude.Include.NON_NULL)
public record ProgressResponse(
int code,
String desc,
ProgressResponseData data
) {
}
/**
* 进度响应数据
*/
@JsonInclude(value = JsonInclude.Include.NON_NULL)
public record ProgressResponseData(
int process,
String pptId,
String pptUrl,
String pptStatus, // PPT构建状态building构建中done已完成build_failed生成失败
String aiImageStatus, // ai配图状态building构建中done已完成
String cardNoteStatus, // 演讲备注状态building构建中done已完成
String errMsg, // 生成PPT的失败信息
Integer totalPages, // 生成PPT的总页数
Integer donePages // 生成PPT的完成页数
) {
/**
* 是否全部完成
*
* @return 是否全部完成
*/
public boolean isAllDone() {
return "done".equals(pptStatus)
&& ("done".equals(aiImageStatus) || aiImageStatus == null)
&& ("done".equals(cardNoteStatus) || cardNoteStatus == null);
}
/**
* 是否失败
*
* @return 是否失败
*/
public boolean isFailed() {
return "build_failed".equals(pptStatus);
}
/**
* 获取进度百分比
*
* @return 进度百分比
*/
public int getProgressPercent() {
if (totalPages == null || totalPages == 0 || donePages == null) {
return process; // 兼容旧版返回
}
return (int) (donePages * 100.0 / totalPages);
}
}
/**
* 通过大纲创建PPT请求参数
*/
@JsonInclude(value = JsonInclude.Include.NON_NULL)
public record CreatePptByOutlineRequest(
String query, // 用户生成PPT要求最多8000字
String outlineSid, // 已生成大纲后响应返回的请求大纲唯一id
OutlineData outline, // 大纲内容
String templateId, // 模板ID
String businessId, // 业务ID非必传
String author, // PPT作者名
Boolean isCardNote, // 是否生成PPT演讲备注
Boolean search, // 是否联网搜索
String language, // 语种
String fileUrl, // 文件地址
String fileName, // 文件名(带文件名后缀)
Boolean isFigure, // 是否自动配图
String aiImage // ai配图类型normaladvanced
) {
/**
* 创建构建器
*
* @return 构建器
*/
public static Builder builder() {
return new Builder();
}
/**
* 构建器类
*/
public static class Builder {
private String query;
private String outlineSid;
private OutlineData outline;
private String templateId;
private String businessId;
private String author;
private Boolean isCardNote;
private Boolean search;
private String language;
private String fileUrl;
private String fileName;
private Boolean isFigure;
private String aiImage;
public Builder query(String query) {
this.query = query;
return this;
}
public Builder outlineSid(String outlineSid) {
this.outlineSid = outlineSid;
return this;
}
public Builder outline(OutlineData outline) {
this.outline = outline;
return this;
}
public Builder templateId(String templateId) {
this.templateId = templateId;
return this;
}
public Builder businessId(String businessId) {
this.businessId = businessId;
return this;
}
public Builder author(String author) {
this.author = author;
return this;
}
public Builder isCardNote(Boolean isCardNote) {
this.isCardNote = isCardNote;
return this;
}
public Builder search(Boolean search) {
this.search = search;
return this;
}
public Builder language(String language) {
this.language = language;
return this;
}
public Builder fileUrl(String fileUrl) {
this.fileUrl = fileUrl;
return this;
}
public Builder fileName(String fileName) {
this.fileName = fileName;
return this;
}
public Builder isFigure(Boolean isFigure) {
this.isFigure = isFigure;
return this;
}
public Builder aiImage(String aiImage) {
this.aiImage = aiImage;
return this;
}
public CreatePptByOutlineRequest build() {
return new CreatePptByOutlineRequest(
query, outlineSid, outline, templateId, businessId, author,
isCardNote, search, language, fileUrl, fileName, isFigure, aiImage
);
}
}
}
/**
* 构建创建PPT的表单数据
*
* @param request 请求参数
* @return 表单数据
*/
private MultiValueMap<String, Object> buildCreateFormData(CreatePptRequest request) {
MultiValueMap<String, Object> formData = new LinkedMultiValueMap<>();
// 添加请求参数
if (request.query() != null) {
formData.add("query", request.query());
}
if (request.file() != null) {
try {
formData.add("file", new ByteArrayResource(request.file().getBytes()) {
@Override
public String getFilename() {
return request.file().getOriginalFilename();
}
});
} catch (IOException e) {
log.error("[xunfei-ppt-api] 文件处理失败", e);
throw new IllegalStateException("[xunfei-ppt-api] 文件处理失败", e);
}
}
if (request.fileUrl() != null) {
formData.add("fileUrl", request.fileUrl());
}
if (request.fileName() != null) {
formData.add("fileName", request.fileName());
}
if (request.templateId() != null) {
formData.add("templateId", request.templateId());
}
if (request.businessId() != null) {
formData.add("businessId", request.businessId());
}
if (request.author() != null) {
formData.add("author", request.author());
}
if (request.isCardNote() != null) {
formData.add("isCardNote", request.isCardNote().toString());
}
if (request.search() != null) {
formData.add("search", request.search().toString());
}
if (request.language() != null) {
formData.add("language", request.language());
}
if (request.isFigure() != null) {
formData.add("isFigure", request.isFigure().toString());
}
if (request.aiImage() != null) {
formData.add("aiImage", request.aiImage());
}
return formData;
}
/**
* 直接生成PPT请求参数
*/
@JsonInclude(value = JsonInclude.Include.NON_NULL)
public record CreatePptRequest(
String query, // 用户生成PPT要求最多8000字
MultipartFile file, // 上传文件
String fileUrl, // 文件地址
String fileName, // 文件名(带文件名后缀)
String templateId, // 模板ID
String businessId, // 业务ID非必传
String author, // PPT作者名
Boolean isCardNote, // 是否生成PPT演讲备注
Boolean search, // 是否联网搜索
String language, // 语种
Boolean isFigure, // 是否自动配图
String aiImage // ai配图类型normaladvanced
) {
/**
* 创建构建器
*
* @return 构建器
*/
public static Builder builder() {
return new Builder();
}
/**
* 构建器类
*/
public static class Builder {
private String query;
private MultipartFile file;
private String fileUrl;
private String fileName;
private String templateId;
private String businessId;
private String author;
private Boolean isCardNote;
private Boolean search;
private String language;
private Boolean isFigure;
private String aiImage;
public Builder query(String query) {
this.query = query;
return this;
}
public Builder file(MultipartFile file) {
this.file = file;
return this;
}
public Builder fileUrl(String fileUrl) {
this.fileUrl = fileUrl;
return this;
}
public Builder fileName(String fileName) {
this.fileName = fileName;
return this;
}
public Builder templateId(String templateId) {
this.templateId = templateId;
return this;
}
public Builder businessId(String businessId) {
this.businessId = businessId;
return this;
}
public Builder author(String author) {
this.author = author;
return this;
}
public Builder isCardNote(Boolean isCardNote) {
this.isCardNote = isCardNote;
return this;
}
public Builder search(Boolean search) {
this.search = search;
return this;
}
public Builder language(String language) {
this.language = language;
return this;
}
public Builder isFigure(Boolean isFigure) {
this.isFigure = isFigure;
return this;
}
public Builder aiImage(String aiImage) {
this.aiImage = aiImage;
return this;
}
public CreatePptRequest build() {
// 验证参数
if (query == null && file == null && fileUrl == null) {
throw new IllegalArgumentException("query、file、fileUrl必填其一");
}
if ((file != null || fileUrl != null) && fileName == null) {
throw new IllegalArgumentException("如果传file或者fileUrlfileName必填");
}
return new CreatePptRequest(
query, file, fileUrl, fileName, templateId, businessId, author,
isCardNote, search, language, isFigure, aiImage
);
}
}
}
}

View File

@ -0,0 +1,306 @@
package cn.iocoder.yudao.framework.ai.ppt.wdd;
import cn.iocoder.yudao.framework.ai.core.model.wenduoduo.api.WddPptApi;
import cn.iocoder.yudao.framework.common.util.json.JsonUtils;
import org.junit.jupiter.api.Disabled;
import org.junit.jupiter.api.Test;
import reactor.core.publisher.Flux;
import java.util.Map;
import java.util.Objects;
/**
* {@link WddPptApi} 集成测试
*
* @author xiaoxin
*/
public class WddPptApiTests {
private final WddPptApi wddPptApi = new WddPptApi("https://docmee.cn");
private final String token = "";
@Test //获取token
@Disabled
public void testCreateApiToken() {
// 准备参数
String apiKey = "";
WddPptApi.CreateTokenRequest request = new WddPptApi.CreateTokenRequest(apiKey);
// 调用方法
String token = wddPptApi.createApiToken(request);
// 打印结果
System.out.println(token);
}
@Test // 创建任务
@Disabled
public void testCreateTask() {
WddPptApi.ApiResponse apiResponse = wddPptApi.createTask(token, 1, "dify 介绍", null);
System.out.println(apiResponse);
}
@Test // 创建大纲
@Disabled
public void testGenerateOutlineRequest() {
WddPptApi.CreateOutlineRequest request = new WddPptApi.CreateOutlineRequest("1901539019628613632", "medium", null, null, null, null);
//调用
Flux<Map<String, Object>> flux = wddPptApi.createOutline(token, request);
StringBuffer contentBuffer = new StringBuffer();
flux.doOnNext(chunk -> {
contentBuffer.append(chunk.get("text"));
if (Objects.equals(Integer.parseInt(String.valueOf(chunk.get("status"))), 4)) {
// status 4最终 markdown 结构树
System.out.println(JsonUtils.toJsonString(chunk.get("result")));
System.out.println(" ########################################################################");
}
}).then().block();
// 打印结果
System.out.println(contentBuffer);
}
@Test // 修改大纲
@Disabled
public void testUpdateOutlineRequest() {
WddPptApi.UpdateOutlineRequest request = new WddPptApi.UpdateOutlineRequest("1901539019628613632", TEST_OUT_LINE_CONTENT, "精简一点,三个章节即可");
//调用
Flux<Map<String, Object>> flux = wddPptApi.updateOutline(token, request);
StringBuffer contentBuffer = new StringBuffer();
flux.doOnNext(chunk -> {
contentBuffer.append(chunk.get("text"));
if (Objects.equals(Integer.parseInt(String.valueOf(chunk.get("status"))), 4)) {
// status 4最终 markdown 结构树
System.out.println(JsonUtils.toJsonString(chunk.get("result")));
System.out.println(" ########################################################################");
}
}).then().block();
// 打印结果
System.out.println(contentBuffer);
}
@Test // 获取 PPT 模版分页
@Disabled
public void testGetPptTemplatePage() {
// 准备参数
WddPptApi.TemplateQueryRequest.Filter filter = new WddPptApi.TemplateQueryRequest.Filter(1, null, null, null);
WddPptApi.TemplateQueryRequest request = new WddPptApi.TemplateQueryRequest(1, 10, filter);
//调用
WddPptApi.PagePptTemplateInfo pptTemplatePage = wddPptApi.getTemplatePage(token, request);
// 打印结果
System.out.println(pptTemplatePage);
}
@Test // 生成 PPT
@Disabled
public void testGeneratePptx() {
// 准备参数
WddPptApi.CreatePptRequest request = new WddPptApi.CreatePptRequest("1901539019628613632", "1805081814809960448", TEST_OUT_LINE_CONTENT);
//调用
WddPptApi.PptInfo pptInfo = wddPptApi.create(token, request);
// 打印结果
System.out.println(pptInfo);
}
private final String TEST_OUT_LINE_CONTENT = """
# Dify新一代AI应用开发平台
## 1 什么是Dify
### 1.1 Dify定义AI应用开发平台
#### 1.1.1 低代码开发
Dify是一个低代码AI应用开发平台旨在简化AI应用的构建过程让开发者无需编写大量代码即可快速创建各种智能应用
#### 1.1.2 核心功能
Dify的核心功能包括数据集成模型选择流程编排和应用部署提供一站式解决方案加速AI应用的落地和迭代
#### 1.1.3 开源与商业
Dify提供开源版本和商业版本满足不同用户的需求开源版本适合个人开发者和小型团队商业版本则提供更强大的功能和技术支持
### 1.2 Dify解决的问题AI开发痛点
#### 1.2.1 开发周期长
传统AI应用开发周期长需要大量的人力和时间投入Dify通过可视化界面和预置组件大幅缩短开发周期
#### 1.2.2 技术门槛高
AI技术门槛高需要专业的知识和技能Dify降低技术门槛让更多开发者能够参与到AI应用的开发中来
#### 1.2.3 部署和维护复杂
AI应用的部署和维护复杂需要专业的运维团队Dify提供自动化的部署和维护工具简化流程降低成本
### 1.3 Dify发展历程
#### 1.3.1 早期探索
Dify的早期版本主要关注于自然语言处理领域的应用通过集成各种NLP模型提供文本分类情感分析等功能
#### 1.3.2 功能扩展
随着用户需求的不断增长Dify的功能逐渐扩展到图像识别语音识别等领域支持更多类型的AI应用
#### 1.3.3 生态建设
Dify积极建设开发者生态提供丰富的文档教程和案例帮助开发者更好地使用Dify平台共同推动AI技术的发展
## 2 Dify的核心功能
### 2.1 数据集成连接各种数据源
#### 2.1.1 支持多种数据源
Dify支持连接各种数据源包括关系型数据库NoSQL数据库文件系统云存储等满足不同场景的数据需求
#### 2.1.2 数据转换和清洗
Dify提供数据转换和清洗功能可以将不同格式的数据转换为统一的格式并去除无效数据提高数据质量
#### 2.1.3 数据安全
Dify注重数据安全采用各种安全措施保护用户的数据包括数据加密访问控制权限管理等
### 2.2 模型选择丰富的AI模型库
#### 2.2.1 预置模型
Dify预置了丰富的AI模型包括自然语言处理图像识别语音识别等领域的模型开发者可以直接使用这些模型无需自行训练极大的简化了开发流程
#### 2.2.2 自定义模型
Dify支持开发者上传自定义模型满足个性化的需求开发者可以将自己训练的模型部署到Dify平台上与其他开发者共享
#### 2.2.3 模型评估
Dify提供模型评估功能可以对不同模型进行评估选择最优的模型提高应用性能
### 2.3 流程编排可视化流程设计器
#### 2.3.1 可视化界面
Dify提供可视化的流程设计器开发者可以通过拖拽组件的方式设计AI应用的流程无需编写代码简单高效
#### 2.3.2 灵活的流程控制
Dify支持灵活的流程控制可以根据不同的条件执行不同的分支实现复杂的业务逻辑
#### 2.3.3 实时调试
Dify提供实时调试功能可以在设计流程的过程中实时查看流程的执行结果及时发现和解决问题
### 2.4 应用部署一键部署和管理
#### 2.4.1 快速部署
Dify提供一键部署功能可以将AI应用快速部署到各种环境包括本地环境云环境容器环境等
#### 2.4.2 自动伸缩
Dify支持自动伸缩可以根据应用的负载自动调整资源保证应用的稳定性和性能
#### 2.4.3 监控和告警
Dify提供监控和告警功能可以实时监控应用的状态并在出现问题时及时告警方便运维人员进行处理
## 3 Dify的特点和优势
### 3.1 低代码降低开发门槛
#### 3.1.1 可视化开发
Dify采用可视化开发模式开发者无需编写大量代码只需通过拖拽组件即可完成AI应用的开发降低了开发门槛
#### 3.1.2 预置组件
Dify预置了丰富的组件包括数据源组件模型组件流程控制组件等开发者可以直接使用这些组件提高开发效率
#### 3.1.3 减少代码量
Dify可以显著减少代码量降低开发难度让更多开发者能够参与到AI应用的开发中来
### 3.2 灵活满足不同场景需求
#### 3.2.1 支持多种数据源
Dify支持多种数据源可以连接各种数据源满足不同场景的数据需求
#### 3.2.2 支持自定义模型
Dify支持自定义模型开发者可以将自己训练的模型部署到Dify平台上满足个性化的需求
#### 3.2.3 灵活的流程控制
Dify支持灵活的流程控制可以根据不同的条件执行不同的分支实现复杂的业务逻辑
### 3.3 高效加速应用落地
#### 3.3.1 快速开发
Dify通过可视化界面和预置组件大幅缩短开发周期加速AI应用的落地
#### 3.3.2 快速部署
Dify提供一键部署功能可以将AI应用快速部署到各种环境提高部署效率
#### 3.3.3 自动化运维
Dify提供自动化的运维工具简化运维流程降低运维成本
### 3.4 开放构建繁荣生态
#### 3.4.1 开源社区
Dify拥有活跃的开源社区开发者可以在社区中交流经验分享资源共同推动Dify的发展
#### 3.4.2 丰富的文档
Dify提供丰富的文档教程和案例帮助开发者更好地使用Dify平台
#### 3.4.3 API支持
Dify提供API支持开发者可以通过API将Dify集成到自己的系统中扩展Dify的功能
## 4 Dify的使用场景
### 4.1 智能客服提升客户服务质量
#### 4.1.1 自动回复
Dify可以用于构建智能客服系统实现自动回复客户的常见问题提高客户服务效率
#### 4.1.2 情感分析
Dify可以对客户的语音或文本进行情感分析判断客户的情绪并根据情绪提供个性化的服务
#### 4.1.3 知识库问答
Dify可以构建知识库问答系统让客户通过提问的方式获取所需的信息提高客户满意度
### 4.2 金融风控提高风险识别能力
#### 4.2.1 欺诈检测
Dify可以用于构建金融风控系统实现欺诈检测识别可疑交易降低风险
#### 4.2.2 信用评估
Dify可以对用户的信用进行评估并根据评估结果提供不同的金融服务
#### 4.2.3 反洗钱
Dify可以用于反洗钱识别可疑资金流动防止犯罪行为
### 4.3 智慧医疗提升医疗服务水平
#### 4.3.1 疾病诊断
Dify可以用于辅助疾病诊断提高诊断准确率缩短诊断时间
#### 4.3.2 药物研发
Dify可以用于药物研发加速新药的发现和开发
#### 4.3.3 智能健康管理
Dify可以构建智能健康管理系统为用户提供个性化的健康建议和服务
### 4.4 智慧城市提升城市管理效率
#### 4.4.1 交通优化
Dify可以用于交通优化提高交通效率缓解交通拥堵
#### 4.4.2 环境监测
Dify可以用于环境监测实时监测空气质量水质等环境指标及时发现和解决环境问题
#### 4.4.3 智能安防
Dify可以用于智能安防提高城市安全水平预防犯罪行为
## 5 Dify的成功案例
### 5.1 Case 1某电商平台的智能客服
#### 5.1.1 项目背景
该电商平台客户服务压力大人工客服成本高需要一种智能化的解决方案
#### 5.1.2 解决方案
使用Dify构建智能客服系统实现自动回复客户的常见问题并根据客户的情绪提供个性化的服务
#### 5.1.3 效果
客户服务效率提高50%客户满意度提高20%人工客服成本降低30%
### 5.2 Case 2某银行的金融风控系统
#### 5.2.1 项目背景
该银行面临日益增长的金融风险需要一种更有效的风险识别和控制手段
#### 5.2.2 解决方案
使用Dify构建金融风控系统实现欺诈检测信用评估和反洗钱等功能提高风险识别能力
#### 5.2.3 效果
欺诈交易识别率提高40%信用评估准确率提高30%洗钱风险降低25%
### 5.3 Case 3某医院的辅助疾病诊断系统
#### 5.3.1 项目背景
该医院医生工作压力大疾病诊断准确率有待提高需要一种辅助诊断工具
#### 5.3.2 解决方案
使用Dify构建辅助疾病诊断系统根据患者的病历和症状提供诊断建议提高诊断准确率
#### 5.3.3 效果
疾病诊断准确率提高20%诊断时间缩短15%医生工作效率提高10%
## 6 Dify的未来展望
### 6.1 技术升级
#### 6.1.1 模型优化
Dify将不断优化预置模型提高模型性能并支持更多类型的AI模型
#### 6.1.2 流程引擎升级
Dify将升级流程引擎提高流程的灵活性和可扩展性支持更复杂的业务逻辑
#### 6.1.3 平台性能优化
Dify将不断优化平台性能提高平台的稳定性和可靠性满足大规模应用的需求
### 6.2 生态建设
#### 6.2.1 社区建设
Dify将继续加强开源社区建设吸引更多开发者参与共同推动Dify的发展
#### 6.2.2 合作伙伴拓展
Dify将拓展合作伙伴与更多的企业和机构合作共同推动AI技术的应用
#### 6.2.3 应用商店
Dify将构建应用商店让开发者可以分享自己的应用用户可以购买和使用这些应用构建繁荣的生态系统
### 6.3 应用领域拓展
#### 6.3.1 智能制造
Dify将拓展到智能制造领域为企业提供智能化的生产管理和质量控制解决方案
#### 6.3.2 智慧农业
Dify将拓展到智慧农业领域为农民提供智能化的种植和养殖管理解决方案
#### 6.3.3 更多领域
Dify将拓展到更多领域为各行各业提供智能化的解决方案推动社会发展
## 7 总结
### 7.1 Dify的价值
#### 7.1.1 降低AI开发门槛
Dify通过低代码的方式让更多开发者能够参与到AI应用的开发中来
#### 7.1.2 加速AI应用落地
Dify提供一站式解决方案加速AI应用的落地和迭代
#### 7.1.3 构建繁荣的AI生态
Dify通过开源社区和应用商店构建繁荣的AI生态系统
### 7.2 共同发展
#### 7.2.1 欢迎加入Dify社区
欢迎更多开发者加入Dify社区共同推动Dify的发展
#### 7.2.2 合作共赢
期待与更多的企业和机构合作共同推动AI技术的应用
#### 7.2.3 共创未来
让我们一起用AI技术改变世界共创美好未来""";
}

View File

@ -0,0 +1,300 @@
package cn.iocoder.yudao.framework.ai.ppt.xunfei;
import cn.iocoder.yudao.framework.ai.core.model.xunfei.api.XunfeiPptApi;
import cn.iocoder.yudao.framework.common.util.json.JsonUtils;
import org.junit.jupiter.api.Disabled;
import org.junit.jupiter.api.Test;
import org.springframework.mock.web.MockMultipartFile;
import org.springframework.web.multipart.MultipartFile;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
/**
* {@link XunfeiPptApi} 集成测试
*
* @author xiaoxin
*/
public class XunfeiPptApiTests {
// 讯飞API配置信息实际使用时请替换为您的应用信息
private static final String APP_ID = "";
private static final String API_SECRET = "";
private final XunfeiPptApi xunfeiPptApi = new XunfeiPptApi(XunfeiPptApi.BASE_URL, APP_ID, API_SECRET);
@Test // 获取PPT模板列表
@Disabled
public void testGetTemplatePage() {
// 调用方法
XunfeiPptApi.TemplatePageResponse response = xunfeiPptApi.getTemplatePage("商务", 10);
// 打印结果
System.out.println("模板列表响应:" + JsonUtils.toJsonString(response));
if (response != null && response.data() != null && response.data().records() != null) {
System.out.println("模板总数:" + response.data().total());
System.out.println("当前页码:" + response.data().pageNum());
System.out.println("模板数量:" + response.data().records().size());
// 打印第一个模板的信息如果存在
if (!response.data().records().isEmpty()) {
XunfeiPptApi.TemplateInfo firstTemplate = response.data().records().get(0);
System.out.println("模板ID" + firstTemplate.templateIndexId());
System.out.println("模板风格:" + firstTemplate.style());
System.out.println("模板颜色:" + firstTemplate.color());
System.out.println("模板行业:" + firstTemplate.industry());
}
}
}
@Test // 创建大纲通过文本
@Disabled
public void testCreateOutline() {
XunfeiPptApi.CreateResponse response = getCreateResponse();
// 打印结果
System.out.println("创建大纲响应:" + JsonUtils.toJsonString(response));
// 保存sid和outline用于后续测试
if (response != null && response.data() != null) {
System.out.println("sid: " + response.data().sid());
if (response.data().outline() != null) {
// 使用OutlineData的toJsonString方法
System.out.println("outline: " + response.data().outline().toJsonString());
// 将outline对象转换为JSON字符串用于后续createPptByOutline测试
String outlineJson = response.data().outline().toJsonString();
System.out.println("可用于createPptByOutline的outline字符串: " + outlineJson);
}
}
}
// 创建大纲通过文本
private XunfeiPptApi.CreateResponse getCreateResponse() {
// 准备参数
String param = "智能体平台 Dify 介绍";
// 调用方法
return xunfeiPptApi.createOutline(param);
}
@Test // 通过大纲创建PPT完整参数
@Disabled
public void testCreatePptByOutlineWithFullParams() {
// 创建大纲对象
XunfeiPptApi.CreateResponse createResponse = getCreateResponse();
// 调用方法
XunfeiPptApi.CreateResponse response = xunfeiPptApi.createPptByOutline(createResponse.data().outline(), "精简一些不要超过6个章节");
// 打印结果
System.out.println("通过大纲创建PPT响应" + JsonUtils.toJsonString(response));
// 保存sid用于后续进度查询
if (response != null && response.data() != null) {
System.out.println("sid: " + response.data().sid());
if (response.data().coverImgSrc() != null) {
System.out.println("封面图片: " + response.data().coverImgSrc());
}
}
}
@Test // 检查PPT生成进度
@Disabled
public void testCheckProgress() {
// 准备参数 - 使用之前创建PPT时返回的sid
String sid = "e96dac09f2ec4ee289f029a5fb874ecd"; // 替换为实际的sid
// 调用方法
XunfeiPptApi.ProgressResponse response = xunfeiPptApi.checkProgress(sid);
// 打印结果
System.out.println("检查进度响应:" + JsonUtils.toJsonString(response));
// 安全地访问响应数据
if (response != null && response.data() != null) {
XunfeiPptApi.ProgressResponseData data = response.data();
// 打印PPT生成状态
System.out.println("PPT构建状态: " + data.pptStatus());
System.out.println("AI配图状态: " + data.aiImageStatus());
System.out.println("演讲备注状态: " + data.cardNoteStatus());
// 打印进度信息
if (data.totalPages() != null && data.donePages() != null) {
System.out.println("总页数: " + data.totalPages());
System.out.println("已完成页数: " + data.donePages());
System.out.println("完成进度: " + data.getProgressPercent() + "%");
} else {
System.out.println("进度: " + data.process() + "%");
}
// 检查是否完成
if (data.isAllDone()) {
System.out.println("PPT生成已完成!");
System.out.println("PPT下载链接: " + data.pptUrl());
}
// 检查是否失败
else if (data.isFailed()) {
System.out.println("PPT生成失败!");
System.out.println("错误信息: " + data.errMsg());
}
// 正在进行中
else {
System.out.println("PPT生成中请稍后再查询...");
}
}
}
@Test // 轮询检查PPT生成进度直到完成
@Disabled
public void testPollCheckProgress() throws InterruptedException {
// 准备参数 - 使用之前创建PPT时返回的sid
String sid = "fa36e926f2ed434987fcb4c1f0776ffb"; // 替换为实际的sid
// 最大轮询次数
int maxPolls = 20;
// 轮询间隔毫秒- 讯飞API限流为3秒一次
long pollInterval = 3500;
for (int i = 0; i < maxPolls; i++) {
System.out.println("" + (i + 1) + "次查询进度...");
// 调用方法
XunfeiPptApi.ProgressResponse response = xunfeiPptApi.checkProgress(sid);
// 安全地访问响应数据
if (response != null && response.data() != null) {
XunfeiPptApi.ProgressResponseData data = response.data();
// 打印进度信息
System.out.println("PPT构建状态: " + data.pptStatus());
if (data.totalPages() != null && data.donePages() != null) {
System.out.println("完成进度: " + data.donePages() + "/" + data.totalPages()
+ " (" + data.getProgressPercent() + "%)");
}
// 检查是否完成
if (data.isAllDone()) {
System.out.println("PPT生成已完成!");
System.out.println("PPT下载链接: " + data.pptUrl());
break;
}
// 检查是否失败
else if (data.isFailed()) {
System.out.println("PPT生成失败!");
System.out.println("错误信息: " + data.errMsg());
break;
}
// 正在进行中继续轮询
else {
System.out.println("PPT生成中等待" + (pollInterval / 1000) + "秒后继续查询...");
Thread.sleep(pollInterval);
}
} else {
System.out.println("查询失败,等待" + (pollInterval / 1000) + "秒后重试...");
Thread.sleep(pollInterval);
}
}
}
@Test // 直接创建PPT通过文本
@Disabled
public void testCreatePptByText() {
// 准备参数
String query = "合肥天气趋势分析包括近5年的气温变化、降水量变化、极端天气事件以及对城市生活的影响";
// 调用方法
XunfeiPptApi.CreateResponse response = xunfeiPptApi.create(query);
// 打印结果
System.out.println("直接创建PPT响应" + JsonUtils.toJsonString(response));
// 保存sid用于后续进度查询
if (response != null && response.data() != null) {
System.out.println("sid: " + response.data().sid());
if (response.data().coverImgSrc() != null) {
System.out.println("封面图片: " + response.data().coverImgSrc());
}
System.out.println("标题: " + response.data().title());
System.out.println("副标题: " + response.data().subTitle());
}
}
@Test // 直接创建PPT通过文件
@Disabled
public void testCreatePptByFile() throws IOException {
// 准备参数
File file = new File("src/test/resources/test.txt"); // 请确保此文件存在
MultipartFile multipartFile = convertFileToMultipartFile(file);
// 调用方法
XunfeiPptApi.CreateResponse response = xunfeiPptApi.create(multipartFile, file.getName());
// 打印结果
System.out.println("通过文件创建PPT响应" + JsonUtils.toJsonString(response));
// 保存sid用于后续进度查询
if (response != null && response.data() != null) {
System.out.println("sid: " + response.data().sid());
if (response.data().coverImgSrc() != null) {
System.out.println("封面图片: " + response.data().coverImgSrc());
}
System.out.println("标题: " + response.data().title());
System.out.println("副标题: " + response.data().subTitle());
}
}
@Test // 直接创建PPT完整参数
@Disabled
public void testCreatePptWithFullParams() throws IOException {
// 准备参数
String query = "合肥天气趋势分析包括近5年的气温变化、降水量变化、极端天气事件以及对城市生活的影响";
// 创建请求对象
XunfeiPptApi.CreatePptRequest request = XunfeiPptApi.CreatePptRequest.builder()
.query(query)
.language("cn")
.isCardNote(true)
.search(true)
.isFigure(true)
.aiImage("advanced")
.author("测试用户")
.build();
// 调用方法
XunfeiPptApi.CreateResponse response = xunfeiPptApi.create(request);
// 打印结果
System.out.println("使用完整参数创建PPT响应" + JsonUtils.toJsonString(response));
// 保存sid用于后续进度查询
if (response != null && response.data() != null) {
String sid = response.data().sid();
System.out.println("sid: " + sid);
if (response.data().coverImgSrc() != null) {
System.out.println("封面图片: " + response.data().coverImgSrc());
}
System.out.println("标题: " + response.data().title());
System.out.println("副标题: " + response.data().subTitle());
// 立即查询一次进度
System.out.println("立即查询进度...");
XunfeiPptApi.ProgressResponse progressResponse = xunfeiPptApi.checkProgress(sid);
if (progressResponse != null && progressResponse.data() != null) {
XunfeiPptApi.ProgressResponseData progressData = progressResponse.data();
System.out.println("PPT构建状态: " + progressData.pptStatus());
if (progressData.totalPages() != null && progressData.donePages() != null) {
System.out.println("完成进度: " + progressData.donePages() + "/" + progressData.totalPages()
+ " (" + progressData.getProgressPercent() + "%)");
}
}
}
}
/**
* 将File转换为MultipartFile
*/
private MultipartFile convertFileToMultipartFile(File file) throws IOException {
FileInputStream input = new FileInputStream(file);
return new MockMultipartFile(
"file",
file.getName(),
"text/plain",
input.readAllBytes()
);
}
}