fix: update wrong main view state when use a model (janhq#861)

Signed-off-by: James <[email protected]> Co-authored-by: James <[email protected]>
BLDC888 · Dec 5, 2023 · aa8e700 · aa8e700
1 parent 848bdbc
commit aa8e700
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/web/screens/ExploreModels/ExploreModelItemHeader/index.tsx b/web/screens/ExploreModels/ExploreModelItemHeader/index.tsx
@@ -59,15 +59,17 @@ const ExploreModelItemHeader: React.FC<Props> = ({ model, onClick, open }) => {
     <Button onClick={() => onDownloadClick()}>Download</Button>
   )
 
+  const onUseModelClick = () => {
+    startModel(model.id)
+    setMainViewState(MainViewState.Thread)
+  }
+
   if (isDownloaded) {
     downloadButton = (
       <Button
         themes="success"
         className="min-w-[98px]"
-        onClick={() => {
-          startModel(model.id)
-          setMainViewState(MainViewState.MyModels)
-        }}
+        onClick={onUseModelClick}
       >
         Use
       </Button>