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

feat(message-handling): fix repeated first token, think tag handling

Browse Source 
closes #889
This commit is contained in:
ItzCrazyKns
2025-10-02 17:15:54 +05:30
parent d611ddaab9
commit 5cc11ac0bf
1 changed files with 21 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -399,6 +399,7 @@ export const ChatProvider = ({
(m, j) =>
j > i &&
m.role === 'source' &&
m.sources &&
(nextUserMessageIndex === -1 || j < nextUserMessageIndex),
) as SourceMessage | undefined;
@@ -417,7 +418,7 @@ export const ChatProvider = ({
const closeThinkTag =
processedMessage.match(/<\/think>/g)?.length || 0;
if (openThinkTag > closeThinkTag) {
if (openThinkTag && !closeThinkTag) {
processedMessage += '</think> <a> </a>';
}
}
@@ -426,7 +427,11 @@ export const ChatProvider = ({
thinkingEnded = true;
}
if (sourceMessage && sourceMessage.sources.length > 0) {
if (
sourceMessage &&
sourceMessage.sources &&
sourceMessage.sources.length > 0
) {
processedMessage = processedMessage.replace(
citationRegex,
(_, capturedContent: string) => {
@@ -635,23 +640,22 @@ export const ChatProvider = ({
},
]);
added = true;
setMessageAppeared(true);
} else {
setMessages((prev) =>
prev.map((message) => {
if (
message.messageId === data.messageId &&
message.role === 'assistant'
) {
return { ...message, content: message.content + data.data };
}
return message;
}),
);
}
setMessages((prev) =>
prev.map((message) => {
if (
message.messageId === data.messageId &&
message.role === 'assistant'
) {
return { ...message, content: message.content + data.data };
}
return message;
}),
);
recievedMessage += data.data;
setMessageAppeared(true);
}
if (data.type === 'messageEnd') {