github/node-DeepResearch
1
0
Fork 0
mirror of https://github.com/jina-ai/node-DeepResearch.git synced 2025-12-26 06:28:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: filter out the empty assistant message (#82)

Browse Source
This commit is contained in:
Sha Zhou 2025-03-12 10:54:38 +08:00 committed by GitHub
parent c8fc259dff
commit 0bbc42e73c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 23 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
src/app.ts
View File

@ -391,33 +391,47 @@ app.post('/v1/chat/completions', (async (req: Request, res: Response) => {
console.log('messages', JSON.stringify(body.messages));
// clean <think> from all assistant messages
body.messages?.forEach(message => {
body.messages = body.messages?.filter(message => {
if (message.role === 'assistant') {
// 2 cases message.content can be a string or an array
// 2 cases message.content can be a string or an array
if (typeof message.content === 'string') {
message.content = (message.content as string).replace(/<think>[\s\S]*?<\/think>/g, '').trim();
message.content = (message.content as string).replace(/<think>[\s\S]*?<\/think>/g, '').trim();
// Filter out the message if the content is empty after <think> removal
return message.content !== '';
} else if (Array.isArray(message.content)) {
// find all type: text and clean <think> from .text
message.content.forEach((content: any) => {
if (content.type === 'text') {
content.text = (content.text as string).replace(/<think>[\s\S]*?<\/think>/g, '').trim();
}});
}
});
//Filter out any content objects in the array that now have null/undefined/empty text.
message.content = message.content.filter((content:any) =>
!(content.type === 'text' && content.text === '')
);
//Filter out the message if the array is now empty
return message.content.length > 0;
}
return true; // Keep the message if it's not an assistant message, or if assistant message has non string or array content.
} else if (message.role === 'user' && Array.isArray(message.content)) {
message.content = message.content.map((content: any) => {
if (content.type === 'image_url') {
return {
type: 'image',
image: content.image_url?.url || '',
}
return {
type: 'image',
image: content.image_url?.url || '',
}
}
return content;
});
return true;
} else if (message.role === 'system') {
if (Array.isArray(message.content)) {
message.content = message.content.map((content: any) => `${content.text || content}`).join(' ');
message.content = message.content.map((content: any) => `${content.text || content}`).join(' ');
}
return true;
}
return true; // Keep other messages
});
let {tokenBudget, maxBadAttempts} = getTokenBudgetAndMaxAttempts(