diff --git a/llm-fe/src/llm-fe/contexts/MessageContext.tsx b/llm-fe/src/llm-fe/contexts/MessageContext.tsx index f2cc115..bc2733b 100644 --- a/llm-fe/src/llm-fe/contexts/MessageContext.tsx +++ b/llm-fe/src/llm-fe/contexts/MessageContext.tsx @@ -1,4 +1,4 @@ -import { createContext, ReactNode, useContext, useEffect, useRef, useState } from "react"; +import { createContext, ReactNode, useCallback, useContext, useEffect, useRef, useState } from "react"; import { WebSocketContext } from "./WebSocketContext"; import { AccountContext } from "./AccountContext"; import { ConversationContext } from "./ConversationContext"; @@ -128,7 +128,7 @@ const MessageProvider = ( {children}: MessageProviderProps) => { } } - const schedulePostStreamRefetch = () => { + const schedulePostStreamRefetch = useCallback(() => { if (refetchTimerRef.current) clearTimeout(refetchTimerRef.current) const conversationId = selectedConversationRef.current if (!conversationId) return @@ -137,7 +137,7 @@ const MessageProvider = ( {children}: MessageProviderProps) => { refetchTimerRef.current = setTimeout(() => { void GetConversationDetails(conversationId) }, 800) - } + }, []) useEffect(() => { GetConversationDetails(selectedConversation) @@ -316,7 +316,7 @@ const MessageProvider = ( {children}: MessageProviderProps) => { /* unsubscribe from channel during cleanup */ unsubscribe(channelName) } - }, [account, subscribe, unsubscribe, conversationDetails, selectedConversation, setSelectedConversation]) + }, [account, subscribe, unsubscribe, conversationDetails, selectedConversation, setSelectedConversation, schedulePostStreamRefetch]) return(