Spring/spring-ai
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

optimize: remove useless TODO comment on ZhiPu chat model

Browse Source
This commit is contained in:
GR
2024-10-09 01:12:36 +08:00
committed by Mark Pollack
parent 77586d5f86
commit cc3d338fa3
1 changed files with 4 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
models/spring-ai-zhipuai/src/main/java/org/springframework/ai/zhipuai/ZhiPuAiChatModel.java
View File

@@ -297,14 +297,10 @@ public class ZhiPuAiChatModel extends AbstractToolCallSupport implements ChatMod
return this.stream(new Prompt(toolCallConversation, prompt.getOptions()));
}
return Flux.just(response);
}).doOnError(observation::error).doFinally(s -> {
// TODO: Consider a custom ObservationContext and
// include additional metadata
// if (s == SignalType.CANCEL) {
// observationContext.setAborted(true);
// }
observation.stop();
}).contextWrite(ctx -> ctx.put(ObservationThreadLocalAccessor.KEY, observation));
})
.doOnError(observation::error)
.doFinally(s -> observation.stop())
.contextWrite(ctx -> ctx.put(ObservationThreadLocalAccessor.KEY, observation));
// @formatter:on
return new MessageAggregator().aggregate(flux, observationContext::setResponse);