remove unneeded from explore

This commit is contained in:
Josh Hawkins 2024-10-13 18:47:43 -05:00
parent 2f0ef1dc69
commit 4ede289a4e

View File

@ -2,7 +2,6 @@ import {
useEmbeddingsReindexProgress, useEmbeddingsReindexProgress,
useEventUpdate, useEventUpdate,
useModelState, useModelState,
useWs,
} from "@/api/ws"; } from "@/api/ws";
import ActivityIndicator from "@/components/indicators/activity-indicator"; import ActivityIndicator from "@/components/indicators/activity-indicator";
import AnimatedCircularProgressBar from "@/components/ui/circular-progress-bar"; import AnimatedCircularProgressBar from "@/components/ui/circular-progress-bar";
@ -193,22 +192,11 @@ export default function Explore() {
// embeddings reindex progress // embeddings reindex progress
const { send: sendReindexCommand } = useWs( const { payload: reindexState } = useEmbeddingsReindexProgress();
"embeddings_reindex_progress",
"embeddingsReindexProgress",
);
useEffect(() => {
sendReindexCommand("embeddingsReindexProgress");
// only run on mount
// eslint-disable-next-line react-hooks/exhaustive-deps
}, []);
const { payload: reindexProgress } = useEmbeddingsReindexProgress();
const embeddingsReindexing = useMemo(() => { const embeddingsReindexing = useMemo(() => {
if (reindexProgress) { if (reindexState) {
switch (reindexProgress.status) { switch (reindexState.status) {
case "indexing": case "indexing":
return true; return true;
case "completed": case "completed":
@ -217,18 +205,10 @@ export default function Explore() {
return undefined; return undefined;
} }
} }
}, [reindexProgress]); }, [reindexState]);
// model states // model states
const { send: sendModelCommand } = useWs("model_state", "modelState");
useEffect(() => {
sendModelCommand("modelState");
// only run on mount
// eslint-disable-next-line react-hooks/exhaustive-deps
}, []);
const { payload: textModelState } = useModelState( const { payload: textModelState } = useModelState(
"jinaai/jina-clip-v1-text_model_fp16.onnx", "jinaai/jina-clip-v1-text_model_fp16.onnx",
); );
@ -274,7 +254,8 @@ export default function Explore() {
if ( if (
config?.semantic_search.enabled && config?.semantic_search.enabled &&
(!textModelState || (!reindexState ||
!textModelState ||
!textTokenizerState || !textTokenizerState ||
!visionModelState || !visionModelState ||
!visionFeatureExtractorState) !visionFeatureExtractorState)
@ -303,24 +284,22 @@ export default function Explore() {
<div className="pt-5 text-center"> <div className="pt-5 text-center">
<AnimatedCircularProgressBar <AnimatedCircularProgressBar
min={0} min={0}
max={reindexProgress.total_objects} max={reindexState.total_objects}
value={reindexProgress.processed_objects} value={reindexState.processed_objects}
gaugePrimaryColor="hsl(var(--selected))" gaugePrimaryColor="hsl(var(--selected))"
gaugeSecondaryColor="hsl(var(--secondary))" gaugeSecondaryColor="hsl(var(--secondary))"
/> />
</div> </div>
<div className="flex w-96 flex-col gap-2 py-5"> <div className="flex w-96 flex-col gap-2 py-5">
{reindexProgress.time_remaining !== null && ( {reindexState.time_remaining !== null && (
<div className="mb-3 flex flex-col items-center justify-center gap-1"> <div className="mb-3 flex flex-col items-center justify-center gap-1">
<div className="text-primary-variant"> <div className="text-primary-variant">
{reindexProgress.time_remaining === -1 {reindexState.time_remaining === -1
? "Starting up..." ? "Starting up..."
: "Estimated time remaining:"} : "Estimated time remaining:"}
</div> </div>
{reindexProgress.time_remaining >= 0 && {reindexState.time_remaining >= 0 &&
(formatSecondsToDuration( (formatSecondsToDuration(reindexState.time_remaining) ||
reindexProgress.time_remaining,
) ||
"Finishing shortly")} "Finishing shortly")}
</div> </div>
)} )}
@ -328,20 +307,20 @@ export default function Explore() {
<span className="text-primary-variant"> <span className="text-primary-variant">
Thumbnails embedded: Thumbnails embedded:
</span> </span>
{reindexProgress.thumbnails} {reindexState.thumbnails}
</div> </div>
<div className="flex flex-row items-center justify-center gap-3"> <div className="flex flex-row items-center justify-center gap-3">
<span className="text-primary-variant"> <span className="text-primary-variant">
Descriptions embedded: Descriptions embedded:
</span> </span>
{reindexProgress.descriptions} {reindexState.descriptions}
</div> </div>
<div className="flex flex-row items-center justify-center gap-3"> <div className="flex flex-row items-center justify-center gap-3">
<span className="text-primary-variant"> <span className="text-primary-variant">
Tracked objects processed: Tracked objects processed:
</span> </span>
{reindexProgress.processed_objects} /{" "} {reindexState.processed_objects} /{" "}
{reindexProgress.total_objects} {reindexState.total_objects}
</div> </div>
</div> </div>
</> </>