fix(server): rerank scores calc (#13016)

fix AI-257
This commit is contained in:
DarkSky
2025-07-07 23:05:02 +08:00
committed by GitHub
parent 2d050a317f
commit 181ccf5a45
18 changed files with 329 additions and 109 deletions
@@ -60,7 +60,7 @@ export class EmbeddingProgress extends Entity {
smartRetry(),
mergeMap(value => {
this.progress$.next(value);
if (value && value.embedded === value.total) {
if (value && value.embedded === value.total && value.total > 0) {
this.stopEmbeddingProgressPolling();
}
return EMPTY;
@@ -62,18 +62,25 @@ const EmbeddingCloud: React.FC<{ disabled: boolean }> = ({ disabled }) => {
option: checked ? 'on' : 'off',
});
embeddingService.embeddingEnabled.setEnabled(checked).catch(error => {
const err = UserFriendlyError.fromAny(error);
notify.error({
title:
t[
'com.affine.settings.workspace.indexer-embedding.embedding.switch.error'
](),
message: t[`error.${err.name}`](err.data),
embeddingService.embeddingEnabled
.setEnabled(checked)
.then(() => {
if (checked) {
embeddingService.embeddingProgress.startEmbeddingProgressPolling();
}
})
.catch(error => {
const err = UserFriendlyError.fromAny(error);
notify.error({
title:
t[
'com.affine.settings.workspace.indexer-embedding.embedding.switch.error'
](),
message: t[`error.${err.name}`](err.data),
});
});
});
},
[embeddingService.embeddingEnabled, t]
[embeddingService.embeddingEnabled, embeddingService.embeddingProgress, t]
);
const handleAttachmentUpload = useCallback(
@@ -84,8 +91,10 @@ const EmbeddingCloud: React.FC<{ disabled: boolean }> = ({ disabled }) => {
docType: file.type,
});
embeddingService.additionalAttachments.addAttachments([file]);
// Restart polling to track progress of newly uploaded files
embeddingService.embeddingProgress.startEmbeddingProgressPolling();
},
[embeddingService.additionalAttachments]
[embeddingService.additionalAttachments, embeddingService.embeddingProgress]
);
const handleAttachmentsDelete = useCallback(