mirror of
https://github.com/toeverything/AFFiNE.git
synced 2026-07-21 20:16:26 +08:00
+1
-1
@@ -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;
|
||||
|
||||
+20
-11
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user