fix: prevent convo overwrite on convo refresh and combine with remote models handling (#1217)

This commit is contained in:
Danny Avila 2023-11-25 09:19:13 -05:00 committed by GitHub
parent ebe62ad250
commit 4dab094855
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 15 additions and 11 deletions

View file

@ -1,31 +1,37 @@
import { useEffect, useRef } from 'react';
import { useParams } from 'react-router-dom';
import { useGetConvoIdQuery } from 'librechat-data-provider';
import { useGetConvoIdQuery, useGetModelsQuery } from 'librechat-data-provider';
import ChatView from '~/components/Chat/ChatView';
import useAuthRedirect from './useAuthRedirect';
import { useSetStorage } from '~/hooks';
import { useNewConvo } from '~/hooks';
import store from '~/store';
export default function ChatRoute() {
const index = 0;
const setStorage = useSetStorage();
const { conversationId } = useParams();
const { conversation, setConversation } = store.useCreateConversationAtom(index);
const { conversation } = store.useCreateConversationAtom(index);
const { isAuthenticated } = useAuthRedirect();
const { newConversation } = useNewConvo();
const hasSetConversation = useRef(false);
const modelsQuery = useGetModelsQuery({ enabled: isAuthenticated });
const initialConvoQuery = useGetConvoIdQuery(conversationId ?? '', {
enabled: isAuthenticated && conversationId !== 'new',
});
useEffect(() => {
if (initialConvoQuery.data && !hasSetConversation.current) {
setStorage(initialConvoQuery.data);
setConversation(initialConvoQuery.data);
if (conversationId === 'new' && modelsQuery.data && !hasSetConversation.current) {
newConversation({ modelsData: modelsQuery.data });
hasSetConversation.current = true;
} else if (initialConvoQuery.data && modelsQuery.data && !hasSetConversation.current) {
newConversation({
template: initialConvoQuery.data,
modelsData: modelsQuery.data,
});
hasSetConversation.current = true;
}
// eslint-disable-next-line react-hooks/exhaustive-deps
}, [initialConvoQuery.data]);
}, [initialConvoQuery.data, modelsQuery.data]);
if (!isAuthenticated) {
return null;

View file

@ -9,13 +9,12 @@ import {
} from 'librechat-data-provider';
import type { ContextType } from '~/common';
import { Nav, MobileNav } from '~/components/Nav';
import { useAuthContext, useServerStream, useConversation, useNewConvo } from '~/hooks';
import { useAuthContext, useServerStream, useConversation } from '~/hooks';
import store from '~/store';
export default function Root() {
const location = useLocation();
const { newConversation } = useConversation();
const { newConversation: newConvo } = useNewConvo();
const { user, isAuthenticated } = useAuthContext();
const [navVisible, setNavVisible] = useState(() => {
const savedNavVisible = localStorage.getItem('navVisible');
@ -44,7 +43,6 @@ export default function Root() {
newConversation({}, undefined, modelsQuery.data);
} else if (modelsQuery.data) {
setModelsConfig(modelsQuery.data);
newConvo({ modelsData: modelsQuery.data });
} else if (modelsQuery.isError) {
console.error('Failed to get models', modelsQuery.error);
}