Fix the Reset logic for stopping the streaming

outpostHQ · Oct 10, 2023 · ad44241 · ad44241
1 parent 3088bf5
commit ad44241
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/app/(services)/inference/_components/useInferenceSession.ts b/app/(services)/inference/_components/useInferenceSession.ts
@@ -28,9 +28,12 @@ export function useInferenceSession(
   const [streamMessage, setStreamMessage] = useState("")
 
   const [error, setError] = useState<any>()
-
   const resetSession = useCallback(() => {
     setSession({} as Session)
+    setIsLoading(false)
+    setError("")
+    setIsDisabled(false)
+    setStreamMessage("")
     // toast({ title: "session has been reset" })
   }, [])