66 lines
5.4 KiB
JavaScript
66 lines
5.4 KiB
JavaScript
function buildSystemPrompt(behavior) {
|
|
return behavior + '\n\nRules:\n- Answer only the user question; do not dump unrelated textbook content.\n- For recognizable medical terms, abbreviations, diseases, and acronyms, answer directly without prefacing with "Assuming you meant".\n- For genuinely misspelled or partial terms, use the retrieved sources to infer the closest medical concept when there is a plausible match, then answer directly. Ask for clarification only when the retrieved sources do not indicate any plausible concept.\n- Use the exact source numbers from the retrieved sources; do not renumber citations for order or style.\n- Cite factual claims immediately with numbered citations like [1] or [1, 3].\n- Every clinical recommendation, dose, threshold, lab value, statistic, comparison, contraindication, red flag, and table row must include its own supporting citation.\n- Do not leave a paragraph, bullet, or table row with multiple factual claims supported only by an uncited heading.\n- If a claim is not directly supported by retrieved sources, omit it or say the available sources are insufficient.\n- If retrieved sources mention the medication/intervention only for other diseases, explicitly say the available sources do not support it for the user\'s requested disease.\n- Do not cite a source number that is not provided.\n- If sources disagree or are insufficient, say so.\n- Keep the main answer concise and clinically useful.\n- Use clear markdown with headings, bullets, and tables when useful.\n- When using a table, output a valid GitHub-flavored markdown table with pipe characters and a separator row. Never output tab-separated tables.\n- Put any summary sentence in a separate paragraph after the table, not as a table row.\n- Do not add a final Sources or References section; the UI displays all retrieved sources separately.\n- Do not add generic disclaimers about clinician judgment.';
|
|
}
|
|
|
|
function buildUserPrompt(question, context, history, searchQuery) {
|
|
var hist = history.filter(function(m) { return m && (m.role === 'user' || m.role === 'assistant') && m.content; })
|
|
.map(function(m) { return m.role.toUpperCase() + ': ' + String(m.content).substring(0, 1000); }).join('\n');
|
|
var searchNote = searchQuery && searchQuery !== question ? ('\n\nStandalone retrieval query used:\n' + searchQuery) : '';
|
|
return 'Question:\n' + question + searchNote + '\n\nRecent conversation, if relevant:\n' + (hist || 'None') + '\n\nRetrieved sources:\n' + context + '\n\nWrite the answer now. If the question is a short misspelled or partial term and the sources point to a likely concept, answer the likely concept rather than asking for clarification.';
|
|
}
|
|
|
|
async function finalizeAssistantAnswer(ai, options) {
|
|
options = options || {};
|
|
var answer = stripModelSourcesSection(String(ai && ai.content || '').trim());
|
|
if (shouldRegenerateTruncatedAnswer(answer, ai && ai.finishReason) && typeof options.callAI === 'function') {
|
|
console.warn('[clinical-assistant] answer looked truncated; regenerating final answer', { finishReason: ai && ai.finishReason, chars: answer.length, streamed: Boolean(options.streamed) });
|
|
if (typeof options.onRegenerating === 'function') options.onRegenerating();
|
|
var completed = await options.callAI(options.messages, {
|
|
model: options.chatModel || undefined,
|
|
temperature: 0.15,
|
|
maxTokens: 5000
|
|
});
|
|
answer = stripModelSourcesSection(String(completed.content || '').trim()) || answer;
|
|
ai.model = completed.model || ai.model;
|
|
ai.provider = completed.provider || ai.provider;
|
|
ai.finishReason = completed.finishReason || ai.finishReason;
|
|
}
|
|
return { answer: answer, ai: ai };
|
|
}
|
|
|
|
function stripModelSourcesSection(answer) {
|
|
return cleanDanglingSourceLeadIn(String(answer || '')
|
|
.replace(/\n\s*(---\s*)?(#{1,3}\s*)?(Sources|References)\s*\n[\s\S]*$/i, '')
|
|
.replace(/\n\s*>?\s*(⚠️\s*)?Clinical decision support[^\n]*$/gim, '')
|
|
.trim());
|
|
}
|
|
|
|
function cleanDanglingSourceLeadIn(answer) {
|
|
answer = String(answer || '').trim();
|
|
return answer
|
|
.replace(/([.!?])\s+(?:however,?\s*)?(?:but\s*)?(?:the\s*)?available(?:\s+(?:sources?|retrieved\s+sources?|evidence|material))?\s*$/i, '$1')
|
|
.replace(/([.!?])\s+(?:however,?\s*)?(?:but\s*)?(?:based\s+on\s+)?(?:the\s*)?available\s*$/i, '$1')
|
|
.replace(/(?:^|\n\n)(?:however,?\s*)?(?:but\s*)?(?:the\s*)?available(?:\s+(?:sources?|retrieved\s+sources?|evidence|material))?\s*$/i, '')
|
|
.replace(/(?:^|\n\n)(?:however,?\s*)?(?:but\s*)?(?:based\s+on\s+)?(?:the\s*)?available\s*$/i, '')
|
|
.trim();
|
|
}
|
|
|
|
function shouldRegenerateTruncatedAnswer(answer, finishReason) {
|
|
answer = String(answer || '').trim();
|
|
if (!answer) return false;
|
|
if (finishReason === 'length') return true;
|
|
if (/\b(the|a|an|and|or|but|with|without|for|to|of|in|on|as|by|from|because|therefore|however|some|many|most|few|several|additional|other|further|including|such|available)$/i.test(answer)) return true;
|
|
if (/[,:;\-(]$/.test(answer)) return true;
|
|
var lines = answer.split(/\n+/).map(function(line) { return line.trim(); }).filter(Boolean);
|
|
var last = lines.length ? lines[lines.length - 1] : answer;
|
|
if (last.length > 24 && /[A-Za-z]$/.test(last) && !/[.!?\])"']$/.test(last)) return true;
|
|
return false;
|
|
}
|
|
|
|
module.exports = {
|
|
buildSystemPrompt: buildSystemPrompt,
|
|
buildUserPrompt: buildUserPrompt,
|
|
finalizeAssistantAnswer: finalizeAssistantAnswer,
|
|
stripModelSourcesSection: stripModelSourcesSection,
|
|
shouldRegenerateTruncatedAnswer: shouldRegenerateTruncatedAnswer
|
|
};
|