remi/Perplexica
1
0
Fork 0
mirror of https://github.com/ItzCrazyKns/Perplexica.git synced 2025-12-16 16:48:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(chat-hook): handle sources

Browse Source
This commit is contained in:
ItzCrazyKns
2025-12-15 23:47:38 +05:30
parent bd3c5f895a
commit 9dd670f46a
1 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/lib/hooks/useChat.tsx
View File

@@ -34,7 +34,7 @@ type ChatContext = {
chatHistory: [string, string][]; chatHistory: [string, string][];
files: File[]; files: File[];
fileIds: string[]; fileIds: string[];
focusMode: string; sources: string[];
chatId: string | undefined; chatId: string | undefined;
optimizationMode: string; optimizationMode: string;
isMessagesLoaded: boolean; isMessagesLoaded: boolean;
@@ -48,7 +48,7 @@ type ChatContext = {
researchEnded: boolean; researchEnded: boolean;
setResearchEnded: (ended: boolean) => void; setResearchEnded: (ended: boolean) => void;
setOptimizationMode: (mode: string) => void; setOptimizationMode: (mode: string) => void;
setFocusMode: (mode: string) => void; setSources: (sources: string[]) => void;
setFiles: (files: File[]) => void; setFiles: (files: File[]) => void;
setFileIds: (fileIds: string[]) => void; setFileIds: (fileIds: string[]) => void;
sendMessage: ( sendMessage: (
@@ -176,7 +176,7 @@ const loadMessages = async (
setMessages: (messages: Message[]) => void, setMessages: (messages: Message[]) => void,
setIsMessagesLoaded: (loaded: boolean) => void, setIsMessagesLoaded: (loaded: boolean) => void,
setChatHistory: (history: [string, string][]) => void, setChatHistory: (history: [string, string][]) => void,
setFocusMode: (mode: string) => void, setSources: (sources: string[]) => void,
setNotFound: (notFound: boolean) => void, setNotFound: (notFound: boolean) => void,
setFiles: (files: File[]) => void, setFiles: (files: File[]) => void,
setFileIds: (fileIds: string[]) => void, setFileIds: (fileIds: string[]) => void,
@@ -234,7 +234,7 @@ const loadMessages = async (
setFileIds(files.map((file: File) => file.fileId)); setFileIds(files.map((file: File) => file.fileId));
setChatHistory(history); setChatHistory(history);
setFocusMode(data.chat.focusMode); setSources(data.chat.sources);
setIsMessagesLoaded(true); setIsMessagesLoaded(true);
}; };
@@ -243,7 +243,7 @@ export const chatContext = createContext<ChatContext>({
chatId: '', chatId: '',
fileIds: [], fileIds: [],
files: [], files: [],
focusMode: '', sources: [],
hasError: false, hasError: false,
isMessagesLoaded: false, isMessagesLoaded: false,
isReady: false, isReady: false,
@@ -260,7 +260,7 @@ export const chatContext = createContext<ChatContext>({
sendMessage: async () => {}, sendMessage: async () => {},
setFileIds: () => {}, setFileIds: () => {},
setFiles: () => {}, setFiles: () => {},
setFocusMode: () => {}, setSources: () => {},
setOptimizationMode: () => {}, setOptimizationMode: () => {},
setChatModelProvider: () => {}, setChatModelProvider: () => {},
setEmbeddingModelProvider: () => {}, setEmbeddingModelProvider: () => {},
@@ -286,7 +286,7 @@ export const ChatProvider = ({ children }: { children: React.ReactNode }) => {
const [files, setFiles] = useState<File[]>([]); const [files, setFiles] = useState<File[]>([]);
const [fileIds, setFileIds] = useState<string[]>([]); const [fileIds, setFileIds] = useState<string[]>([]);
const [focusMode, setFocusMode] = useState('webSearch'); const [sources, setSources] = useState<string[]>(['web']);
const [optimizationMode, setOptimizationMode] = useState('speed'); const [optimizationMode, setOptimizationMode] = useState('speed');
const [isMessagesLoaded, setIsMessagesLoaded] = useState(false); const [isMessagesLoaded, setIsMessagesLoaded] = useState(false);
@@ -436,7 +436,7 @@ export const ChatProvider = ({ children }: { children: React.ReactNode }) => {
setMessages, setMessages,
setIsMessagesLoaded, setIsMessagesLoaded,
setChatHistory, setChatHistory,
setFocusMode, setSources,
setNotFound, setNotFound,
setFiles, setFiles,
setFileIds, setFileIds,
@@ -722,7 +722,7 @@ export const ChatProvider = ({ children }: { children: React.ReactNode }) => {
}, },
chatId: chatId!, chatId: chatId!,
files: fileIds, files: fileIds,
focusMode: focusMode, sources: sources,
optimizationMode: optimizationMode, optimizationMode: optimizationMode,
history: rewrite history: rewrite
? chatHistory.slice(0, messageIndex === -1 ? undefined : messageIndex) ? chatHistory.slice(0, messageIndex === -1 ? undefined : messageIndex)
@@ -774,7 +774,7 @@ export const ChatProvider = ({ children }: { children: React.ReactNode }) => {
chatHistory, chatHistory,
files, files,
fileIds, fileIds,
focusMode, sources,
chatId, chatId,
hasError, hasError,
isMessagesLoaded, isMessagesLoaded,
@@ -785,7 +785,7 @@ export const ChatProvider = ({ children }: { children: React.ReactNode }) => {
optimizationMode, optimizationMode,
setFileIds, setFileIds,
setFiles, setFiles,
setFocusMode, setSources,
setOptimizationMode, setOptimizationMode,
rewrite, rewrite,
sendMessage, sendMessage,