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

Merge branch 'main' of https://github.com/jina-ai/node-DeepResearch

Browse Source
This commit is contained in:
Han Xiao 2025-03-10 17:35:43 +08:00
commit 53163e2b89
1 changed files with 26 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
src/app.ts
View File

@ -391,16 +391,32 @@ 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?.filter(message => message.role === 'assistant').forEach(message => {
// 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();
} 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();
}});
body.messages?.forEach(message => {
if (message.role === 'assistant') {
// 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();
} 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();
}});
}
} 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 content;
});
} else if (message.role === 'system') {
if (Array.isArray(message.content)) {
message.content = message.content.map((content: any) => `${content.text || content}`).join(' ');
}
}
});