hhyykk/ipms-sjy
1
1
Fork 0
mirror of https://gitee.com/hhyykk/ipms-sjy.git synced 2025-10-31 10:18:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

【todo】百度文言一心，解决 todo，删除 doOnComplete、增加 retryTemplate

Browse Source
This commit is contained in:
cherishsince
2024-05-20 13:44:50 +08:00
parent e0d8ac74a4
commit 3a10fedddb
1 changed files with 11 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/model/yiyan/YiYanChatClient.java
View File

@@ -85,22 +85,20 @@ public class YiYanChatClient implements ChatClient, StreamingChatClient {
@Override @Override
public Flux<ChatResponse> stream(Prompt prompt) { public Flux<ChatResponse> stream(Prompt prompt) {
YiYanChatCompletionRequest request = this.createRequest(prompt, true); YiYanChatCompletionRequest request = this.createRequest(prompt, true);
// TODO @fan：return this.retryTemplate.execute(ctx -> { // TODO done @fan：return this.retryTemplate.execute(ctx -> {
return retryTemplate.execute(ctx -> {
// 调用 callWithFunctionSupport 发送请求 // 调用 callWithFunctionSupport 发送请求
Flux<YiYanChatCompletionResponse> response = this.yiYanApi.chatCompletionStream(request); Flux<YiYanChatCompletionResponse> response = this.yiYanApi.chatCompletionStream(request);
// TODO @fan：下面的 doOnComplete 是不是可以删除哈？
response.doOnComplete(new Runnable() {
@Override
public void run() {
String a = ";";
}
});
return response.map(chunk -> { return response.map(chunk -> {
// System.err.println("---".concat(chunk.getResult()));
// TODO @fan：ChatResponseMetadata chatResponseMetadata // TODO @fan：ChatResponseMetadata chatResponseMetadata
return new ChatResponse(List.of(new Generation(chunk.getResult()))); return new ChatResponse(List.of(new Generation(chunk.getResult())));
}); });
});
} }
private YiYanChatCompletionRequest createRequest(Prompt prompt, boolean stream) { private YiYanChatCompletionRequest createRequest(Prompt prompt, boolean stream) {
// 参考 https://cloud.baidu.com/doc/WENXINWORKSHOP/s/clntwmv7t 文档，system 是独立字段 // 参考 https://cloud.baidu.com/doc/WENXINWORKSHOP/s/clntwmv7t 文档，system 是独立字段
// 1.1 获取 user 和 assistant // 1.1 获取 user 和 assistant