Improve reindex completion flag (#14308)

This commit is contained in:
Josh Hawkins 2024-10-12 14:44:01 -05:00 committed by GitHub
parent 3a403392e7
commit 0fc7999780
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 19 additions and 9 deletions

View File

@ -176,6 +176,7 @@ class Embeddings:
"processed_objects": 0,
"total_objects": 0,
"time_remaining": 0,
"status": "indexing",
}
self.requestor.send_data(UPDATE_EMBEDDINGS_REINDEX_PROGRESS, totals)
@ -255,6 +256,8 @@ class Embeddings:
"Embedded %d thumbnails and %d descriptions in %s seconds",
totals["thumbnails"],
totals["descriptions"],
time.time() - st,
round(time.time() - st, 1),
)
totals["status"] = "completed"
self.requestor.send_data(UPDATE_EMBEDDINGS_REINDEX_PROGRESS, totals)

View File

@ -195,13 +195,18 @@ export default function Explore() {
const { payload: reindexProgress } = useEmbeddingsReindexProgress();
const embeddingsReindexing = useMemo(
() =>
reindexProgress
? reindexProgress.total_objects - reindexProgress.processed_objects > 0
: undefined,
[reindexProgress],
);
const embeddingsReindexing = useMemo(() => {
if (reindexProgress) {
switch (reindexProgress.status) {
case "indexing":
return true;
case "completed":
return false;
default:
return undefined;
}
}
}, [reindexProgress]);
// model states
@ -320,7 +325,8 @@ export default function Explore() {
<span className="text-primary-variant">
Tracked objects processed:
</span>
{reindexProgress.processed_objects}
{reindexProgress.processed_objects} /{" "}
{reindexProgress.total_objects}
</div>
</div>
</>

View File

@ -68,6 +68,7 @@ export type EmbeddingsReindexProgressType = {
processed_objects: number;
total_objects: number;
time_remaining: number;
status: string;
};
export type ToggleableSetting = "ON" | "OFF";