Skip to content
Snippets Groups Projects
Commit dbd89bb9 authored by Tony Salomone's avatar Tony Salomone
Browse files

Fix model count to use new endpoint to decide to display wizard.

parent c311ca6e
Branches
Tags
No related merge requests found
/* eslint-disable jsx-a11y/anchor-is-valid */ /* eslint-disable jsx-a11y/anchor-is-valid */
import { useState } from 'react'; import { useState } from 'react';
import useSWRImmutable from 'swr';
import { Button, Sheet, Stack, Typography } from '@mui/joy'; import { Button, Sheet, Stack, Typography } from '@mui/joy';
...@@ -14,13 +15,15 @@ import { ...@@ -14,13 +15,15 @@ import {
PlayCircle, PlayCircle,
PlayCircleIcon, PlayCircleIcon,
} from 'lucide-react'; } from 'lucide-react';
import { useServerStats } from 'renderer/lib/transformerlab-api-sdk'; import * as chatAPI from 'renderer/lib/transformerlab-api-sdk';
import { Link as ReactRouterLink, useNavigate } from 'react-router-dom'; import { Link as ReactRouterLink, useNavigate } from 'react-router-dom';
import DownloadFirstModelModal from './DownloadFirstModelModal'; import DownloadFirstModelModal from './DownloadFirstModelModal';
import HexLogo from './Shared/HexLogo'; import HexLogo from './Shared/HexLogo';
const fetcher = (url) => fetch(url).then((res) => res.json());
function recommendedModel(cpu, os, device) { function recommendedModel(cpu, os, device) {
if (!cpu || !os || !device) return ''; if (!cpu || !os || !device) return '';
...@@ -47,10 +50,22 @@ function typeOfComputer(cpu, os, device) { ...@@ -47,10 +50,22 @@ function typeOfComputer(cpu, os, device) {
} }
export default function Welcome() { export default function Welcome() {
// Check number of downloaded models
let model_count = 0;
const { data: modelCountResponse } = useSWRImmutable(
chatAPI.Endpoints.Models.CountDownloaded(),
fetcher
);
if (modelCountResponse && modelCountResponse!.data) {
model_count = modelCountResponse!.data;
}
// Open DownloadFirstModelModal if the user has no models
const [modelDownloadModalOpen, setModelDownloadModalOpen] = const [modelDownloadModalOpen, setModelDownloadModalOpen] =
useState<boolean>(false); useState<boolean>(model_count == 0);
const { server, isLoading, isError } = useServerStats(); const { server, isLoading, isError } = chatAPI.useServerStats();
const navigate = useNavigate(); const navigate = useNavigate();
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment